weak_form_aca_assembly_helper.hpp

// Copyright (C) 2011-2012 by the BEM++ Authors
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

#ifndef bempp_weak_form_aca_assembly_helper_hpp
#define bempp_weak_form_aca_assembly_helper_hpp

#include "../common/common.hpp"

#include "ahmed_aux_fwd.hpp"
#include "../common/armadillo_fwd.hpp"
#include "../common/shared_ptr.hpp"
#include "../common/types.hpp"
#include "../fiber/scalar_traits.hpp"

#include <tbb/atomic.h>
#include <tbb/concurrent_unordered_map.h>
#include <vector>

class cluster;
namespace Fiber
{

template <typename ResultType> class LocalAssemblerForIntegralOperators;
} // namespace Fiber

namespace Bempp
{

class AssemblyOptions;
template <typename ResultType> class DiscreteBoundaryOperator;
template <typename BasisFunctionType> class LocalDofListsCache;
template <typename BasisFunctionType> class Space;
template <typename BasisFunctionType, typename ResultType>
class WeakFormAcaAssemblyHelper
{
public:
    typedef DiscreteBoundaryOperator<ResultType> DiscreteLinOp;
    typedef Fiber::LocalAssemblerForIntegralOperators<ResultType> LocalAssembler;
    typedef typename Fiber::ScalarTraits<ResultType>::RealType CoordinateType;
    typedef typename Fiber::ScalarTraits<ResultType>::RealType MagnitudeType;
    typedef typename AhmedTypeTraits<ResultType>::Type AhmedResultType;

    WeakFormAcaAssemblyHelper(const Space<BasisFunctionType>& testSpace,
                              const Space<BasisFunctionType>& trialSpace,
                              const std::vector<unsigned int>& p2oTestDofs,
                              const std::vector<unsigned int>& p2oTrialDofs,
                              const std::vector<LocalAssembler*>& assemblers,
                              const std::vector<const DiscreteLinOp*>& sparseTermsToAdd,
                              const std::vector<ResultType>& denseTermsMultipliers,
                              const std::vector<ResultType>& sparseTermsMultipliers,
                              const AssemblyOptions& options);

    void cmpbl(unsigned b1, unsigned n1, unsigned b2, unsigned n2,
               AhmedResultType* data,
               const cluster* c1 = 0, const cluster* c2 = 0,
               bool countAccessedEntries = true) const;

    void cmpblsym(unsigned b1, unsigned n1, AhmedResultType* data,
                  const cluster* c1 = 0,
                  bool countAccessedEntries = true) const;

    MagnitudeType scale(unsigned b1, unsigned n1, unsigned b2, unsigned n2,
                        const cluster* c1 = 0, const cluster* c2 = 0) const;

    MagnitudeType relativeScale(unsigned b1, unsigned n1, unsigned b2, unsigned n2,
                                const cluster* c1 = 0, const cluster* c2 = 0) const;

    size_t accessedEntryCount() const;

    void resetAccessedEntryCount();

private:
    MagnitudeType estimateMinimumDistance(
            const cluster* c1, const cluster* c2) const;

private:
    const Space<BasisFunctionType>& m_testSpace;
    const Space<BasisFunctionType>& m_trialSpace;
    const std::vector<unsigned int>& m_p2oTestDofs;
    const std::vector<unsigned int>& m_p2oTrialDofs;
    const std::vector<LocalAssembler*>& m_assemblers;
    const std::vector<const DiscreteLinOp*>& m_sparseTermsToAdd;
    const std::vector<ResultType>& m_denseTermsMultipliers;
    const std::vector<ResultType>& m_sparseTermsMultipliers;
    const AssemblyOptions& m_options;
    bool m_indexWithGlobalDofs;
    bool m_uniformQuadratureOrder;

    shared_ptr<LocalDofListsCache<BasisFunctionType> >
    m_testDofListsCache, m_trialDofListsCache;

    mutable tbb::atomic<size_t> m_accessedEntryCount;

    typedef tbb::concurrent_unordered_map<
    std::pair<const cluster*, const cluster*>, CoordinateType> DistanceMap;
    mutable DistanceMap m_distancesCache;

};

} // namespace Bempp

#endif