chebyshev_cluster.hpp

Go to the documentation of this file.

 
#ifndef NIHU_CHEBYSHEV_CLUSTER_HPP_INCLUDED
#define NIHU_CHEBYSHEV_CLUSTER_HPP_INCLUDED
 
#include "cluster.hpp"
#include "util/misc.hpp"
#include "nd_cheb.hpp"
 
#include <cstddef>
 
namespace NiHu
{
namespace fmm
{
 
template <size_t Dim, class Scalar, size_t FieldDim>
class chebyshev_cluster;
 
template <size_t Dim, class Scalar, size_t FieldDim>
struct cluster_traits<chebyshev_cluster<Dim, Scalar, FieldDim> >
{
    static size_t const dimension = Dim;
    typedef Eigen::Matrix<Scalar, Eigen::Dynamic, 1> multipole_t;
    typedef Eigen::Matrix<Scalar, Eigen::Dynamic, 1> local_t;
};
 
template <size_t Dim, class Scalar, size_t FieldDim = 1>
class chebyshev_cluster
    : public cluster_base<chebyshev_cluster<Dim, Scalar, FieldDim> >
{
public:
    typedef cluster_base<chebyshev_cluster<Dim, Scalar, FieldDim> > base_t;
    static size_t const dimension = base_t::dimension;
    static size_t const field_dimension = FieldDim;
    typedef typename base_t::multipole_t multipole_t;
    typedef typename base_t::local_t local_t;
 
    typedef Eigen::Matrix<double, dimension, Eigen::Dynamic> cheb_nodes_t;
 
    void set_chebyshev_order(size_t order)
    {
        m_chebyshev_order = order;
        m_cheb_nodes = chebnodes<dimension>(m_chebyshev_order, this->get_bounding_box());
    }
 
    size_t get_data_size() const
    {
        return Ipow(m_chebyshev_order, dimension) * field_dimension;
    }
 
    multipole_t zero_multipole() const
    {
        return multipole_t::Zero(get_data_size(), 1);
    }
 
    local_t zero_local() const
    {
        return local_t::Zero(get_data_size(), 1);
    }
 
    cheb_nodes_t const &get_chebyshev_nodes() const
    {
        return m_cheb_nodes;
    }
 
    size_t get_chebyshev_order() const
    {
        return m_chebyshev_order;
    }
 
private:
    size_t m_chebyshev_order;
    cheb_nodes_t m_cheb_nodes;
};
 
} // end of namespace fmm
} // end of namespace NiHu
 
#endif /* NIHU_CHEBYSHEV_CLUSTER_HPP_INCLUDED */