geometry.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_geometry_hpp
#define bempp_geometry_hpp

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

#include "geometry_type.hpp"

#include "../common/armadillo_fwd.hpp"
#include "../fiber/_3d_array.hpp"

namespace Fiber
{

template <typename CoordinateType> struct GeometricalData;
} // namespace Fiber

namespace Bempp
{

class Geometry
{
public:
    virtual ~Geometry() {}

    virtual int dim() const = 0;

    virtual int dimWorld() const = 0;

    void setup(const arma::Mat<double>& corners,
               const arma::Col<char>& auxData);
    void setup(const arma::Mat<float>& corners,
               const arma::Col<char>& auxData);

    virtual GeometryType type() const = 0;

    virtual bool affine() const = 0;

    virtual int cornerCount() const = 0;

    void getCorners(arma::Mat<double>& c) const;
    void getCorners(arma::Mat<float>& c) const;

    void local2global(const arma::Mat<double>& local,
                      arma::Mat<double>& global) const;
    void local2global(const arma::Mat<float>& local,
                      arma::Mat<float>& global) const;

    void global2local(const arma::Mat<double>& global,
                      arma::Mat<double>& local) const;
    void global2local(const arma::Mat<float>& global,
                      arma::Mat<float>& local) const;

    void getIntegrationElements(const arma::Mat<double>& local,
                                arma::Row<double>& int_element) const;
    void getIntegrationElements(const arma::Mat<float>& local,
                                arma::Row<float>& int_element) const;

    virtual double volume() const = 0;

    void getCenter(arma::Col<double>& c) const;
    void getCenter(arma::Col<float>& c) const;

    void getJacobiansTransposed(
            const arma::Mat<double>& local,
            arma::Cube<double>& jacobian_t) const;
    void getJacobiansTransposed(
            const arma::Mat<float>& local,
            arma::Cube<float>& jacobian_t) const;
    void getJacobiansTransposed(
            const arma::Mat<double>& local,
            Fiber::_3dArray<double>& jacobian_t) const;
    void getJacobiansTransposed(
            const arma::Mat<float>& local,
            Fiber::_3dArray<float>& jacobian_t) const;

    void getJacobianInversesTransposed(
            const arma::Mat<double>& local,
            arma::Cube<double>& jacobian_inv_t) const;
    void getJacobianInversesTransposed(
            const arma::Mat<float>& local,
            arma::Cube<float>& jacobian_inv_t) const;
    void getJacobianInversesTransposed(
            const arma::Mat<double>& local,
            Fiber::_3dArray<double>& jacobian_inv_t) const;
    void getJacobianInversesTransposed(
            const arma::Mat<float>& local,
            Fiber::_3dArray<float>& jacobian_inv_t) const;

    void getNormals(const arma::Mat<double>& local,
                    arma::Mat<double>& normal) const;
    void getNormals(const arma::Mat<float>& local,
                    arma::Mat<float>& normal) const;

    void getData(size_t what, const arma::Mat<double>& local,
                 Fiber::GeometricalData<double>& data) const;
    void getData(size_t what, const arma::Mat<float>& local,
                 Fiber::GeometricalData<float>& data) const;

private:
    // Virtual functions to be implemented in subclasses
    virtual void setupImpl(
            const arma::Mat<double>& corners,
            const arma::Col<char>& auxData) = 0;
    virtual void getCornersImpl(arma::Mat<double>& c) const = 0;
    virtual void local2globalImpl(
            const arma::Mat<double>& local,
            arma::Mat<double>& global) const = 0;
    virtual void global2localImpl(
            const arma::Mat<double>& global,
            arma::Mat<double>& local) const = 0;
    virtual void getIntegrationElementsImpl(
            const arma::Mat<double>& local,
            arma::Row<double>& int_element) const = 0;
    virtual void getCenterImpl(arma::Col<double>& c) const = 0;
    virtual void getJacobiansTransposedImpl(
            const arma::Mat<double>& local,
            Fiber::_3dArray<double>& jacobian_t) const = 0;
    virtual void getJacobianInversesTransposedImpl(
            const arma::Mat<double>& local,
            Fiber::_3dArray<double>& jacobian_inv_t) const = 0;
    virtual void getNormalsImpl(
            const arma::Mat<double>& local,
            arma::Mat<double>& normal) const = 0;
    virtual void getDataImpl(
            size_t what, const arma::Mat<double>& local,
            Fiber::GeometricalData<double>& data) const = 0;

    // Helper functions for implementation
    template <typename T1, typename T2>
    void convertMat(const arma::Mat<T1>& in, arma::Mat<T2>& out) const;
    template <typename T1, typename T2>
    void convertCube(const arma::Cube<T1>& in, arma::Cube<T2>& out) const;
    template <typename T1, typename T2>
    void convertCube(const Fiber::_3dArray<T1>& in, Fiber::_3dArray<T2>& out) const;
    template <typename T1, typename T2>
    void convertCube(const Fiber::_3dArray<T1>& in, arma::Cube<T2>& out) const;
};

} // namespace Bempp

#include "geometry_imp.hpp"

#endif