nonseparable_numerical_test_kernel_trial_integrator.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 fiber_nonseparable_numerical_test_kernel_trial_integrator_hpp
#define fiber_nonseparable_numerical_test_kernel_trial_integrator_hpp

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

#include "test_kernel_trial_integrator.hpp"

#include <tbb/concurrent_unordered_map.h>
#include <tbb/enumerable_thread_specific.h>

namespace Fiber
{

class OpenClHandler;
template <typename CoordinateType> class CollectionOfShapesetTransformations;
template <typename ValueType> class CollectionOfKernels;
template <typename CoordinateType> class RawGridGeometry;
template <typename BasisFunctionType, typename KernelType, typename ResultType>
class TestKernelTrialIntegral;
template <typename BasisFunctionType, typename KernelType,
          typename ResultType, typename GeometryFactory>
class NonseparableNumericalTestKernelTrialIntegrator :
        public TestKernelTrialIntegrator<BasisFunctionType, KernelType, ResultType>
{
public:
    typedef TestKernelTrialIntegrator<BasisFunctionType, KernelType, ResultType> Base;
    typedef typename Base::CoordinateType CoordinateType;
    typedef typename Base::ElementIndexPair ElementIndexPair;

    NonseparableNumericalTestKernelTrialIntegrator(
            const arma::Mat<CoordinateType>& localTestQuadPoints,
            const arma::Mat<CoordinateType>& localTrialQuadPoints,
            const std::vector<CoordinateType> quadWeights,
            const GeometryFactory& testGeometryFactory,
            const GeometryFactory& trialGgeometryFactory,
            const RawGridGeometry<CoordinateType>& testRawGeometry,
            const RawGridGeometry<CoordinateType>& trialRawGeometry,
            const CollectionOfShapesetTransformations<CoordinateType>& testTransformations,
            const CollectionOfKernels<KernelType>& kernel,
            const CollectionOfShapesetTransformations<CoordinateType>& trialTransformations,
            const TestKernelTrialIntegral<BasisFunctionType, KernelType, ResultType>& integral,
            const OpenClHandler& openClHandler);
    virtual ~NonseparableNumericalTestKernelTrialIntegrator();

    virtual void integrate(
            CallVariant callVariant,
            const std::vector<int>& elementIndicesA,
            int elementIndexB,
            const Shapeset<BasisFunctionType>& basisA,
            const Shapeset<BasisFunctionType>& basisB,
            LocalDofIndex localDofIndexB,
            const std::vector<arma::Mat<ResultType>*>& result) const;

    virtual void integrate(
            const std::vector<ElementIndexPair>& elementIndexPairs,
            const Shapeset<BasisFunctionType>& testShapeset,
            const Shapeset<BasisFunctionType>& trialShapeset,
            const std::vector<arma::Mat<ResultType>*>& result) const;

private:
    enum ElementType {TEST, TRIAL};

    const BasisData<BasisFunctionType>&
    basisData(ElementType type, const Shapeset<BasisFunctionType>& shapeset) const;

    arma::Mat<CoordinateType> m_localTestQuadPoints;
    arma::Mat<CoordinateType> m_localTrialQuadPoints;
    std::vector<CoordinateType> m_quadWeights;

    const GeometryFactory& m_testGeometryFactory;
    const GeometryFactory& m_trialGeometryFactory;
    const RawGridGeometry<CoordinateType>& m_testRawGeometry;
    const RawGridGeometry<CoordinateType>& m_trialRawGeometry;

    const CollectionOfShapesetTransformations<CoordinateType>& m_testTransformations;
    const CollectionOfKernels<KernelType>& m_kernels;
    const CollectionOfShapesetTransformations<CoordinateType>& m_trialTransformations;
    const TestKernelTrialIntegral<BasisFunctionType, KernelType, ResultType>& m_integral;

    typedef tbb::concurrent_unordered_map<const Shapeset<BasisFunctionType>*,
    BasisData<BasisFunctionType>* > BasisDataCache;
    mutable BasisDataCache m_cachedTestBasisData;
    mutable BasisDataCache m_cachedTrialBasisData;

    const OpenClHandler& m_openClHandler;
    // thread-local static data for integrate() -- allocation and deallocation of GeometricalData
    // is very time-consuming due to the presence of arma::Cube objects.
    mutable tbb::enumerable_thread_specific<GeometricalData<CoordinateType> >
    m_testGeomData, m_trialGeomData;
};

} // namespace Fiber

#include "nonseparable_numerical_test_kernel_trial_integrator_imp.hpp"

#endif