separable_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_separable_numerical_test_kernel_trial_integrator_hpp
#define fiber_separable_numerical_test_kernel_trial_integrator_hpp

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

#include "bempp/common/config_opencl.hpp"

#include "test_kernel_trial_integrator.hpp"

#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 SeparableNumericalTestKernelTrialIntegrator :
        public TestKernelTrialIntegrator<BasisFunctionType, KernelType, ResultType>
{
public:
    typedef TestKernelTrialIntegrator<BasisFunctionType, KernelType, ResultType> Base;
    typedef typename Base::CoordinateType CoordinateType;
    typedef typename Base::ElementIndexPair ElementIndexPair;

    SeparableNumericalTestKernelTrialIntegrator(
            const arma::Mat<CoordinateType>& localTestQuadPoints,
            const arma::Mat<CoordinateType>& localTrialQuadPoints,
            const std::vector<CoordinateType>& testQuadWeights,
            const std::vector<CoordinateType>& trialQuadWeights,
            const GeometryFactory& testGeometryFactory,
            const GeometryFactory& trialGeometryFactory,
            const RawGridGeometry<CoordinateType>& testRawGeometry,
            const RawGridGeometry<CoordinateType>& trialRawGeometry,
            const CollectionOfShapesetTransformations<CoordinateType>& testTransformations,
            const CollectionOfKernels<KernelType>& kernels,
            const CollectionOfShapesetTransformations<CoordinateType>& trialTransformations,
            const TestKernelTrialIntegral<BasisFunctionType, KernelType, ResultType>& integral,
            const OpenClHandler& openClHandler,
            bool cacheGeometricalData = true);

    virtual ~SeparableNumericalTestKernelTrialIntegrator ();

    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:
    void integrateCpu(
            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;

    void integrateCl(
            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;

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

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

    void precalculateGeometricalData();
    void precalculateGeometricalDataOnSingleGrid(
            const arma::Mat<CoordinateType>& localQuadPoints,
            const GeometryFactory& geometryFactory,
            const RawGridGeometry<CoordinateType>& rawGeometry,
            size_t geomDeps,
            std::vector<GeometricalData<CoordinateType> >& geomData);

    const std::pair<const char*, int> clStrIntegrateRowOrCol () const;

    arma::Mat<CoordinateType> m_localTestQuadPoints;
    arma::Mat<CoordinateType> m_localTrialQuadPoints;
    std::vector<CoordinateType> m_testQuadWeights;
    std::vector<CoordinateType> m_trialQuadWeights;

    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;

    const OpenClHandler& m_openClHandler;
    bool m_cacheGeometricalData;

    std::vector<GeometricalData<CoordinateType> > m_cachedTestGeomData;
    std::vector<GeometricalData<CoordinateType> > m_cachedTrialGeomData;
    mutable tbb::enumerable_thread_specific<GeometricalData<CoordinateType> >
    m_testGeomData, m_trialGeomData;


#ifdef WITH_OPENCL
    cl::Buffer *clTestQuadPoints;
    cl::Buffer *clTrialQuadPoints;
    cl::Buffer *clTestQuadWeights;
    cl::Buffer *clTrialQuadWeights;
#endif
};

} // namespace Fiber

#include "separable_numerical_test_kernel_trial_integrator_imp.hpp"

#endif