divide.hpp

Go to the documentation of this file.

 
#ifndef NIHU_DIVIDE_HPP_INCLUDED
#define NIHU_DIVIDE_HPP_INCLUDED
 
#include "cluster.hpp"
#include "util/crtp_base.hpp"
 
#include <algorithm> // std::max
 
namespace NiHu
{
namespace fmm
{
 
template <class Derived>
class divide_base
{
public:
    NIHU_CRTP_HELPERS
    
    template <class Cluster>
    bool operator()(Cluster const &c) const
    {
        return derived().operator()(c);
    }
};
    
class divide_depth : public divide_base<divide_depth>
{
    size_t m_depth;
 
public:
    divide_depth(size_t depth)
        : m_depth(depth)
    {
    }
 
    template <class Cluster>
    bool operator()(Cluster const &c) const
    {
        return c.get_level() < m_depth;
    }
};
 
class divide_num_nodes : public divide_base<divide_num_nodes>
{
    size_t m_max_nodes;
 
public:
    divide_num_nodes(size_t max_nodes)
        : m_max_nodes(max_nodes)
    {
    }
 
    template <class Cluster>
    bool operator()(Cluster const &c) const
    {
        size_t s = c.get_n_src_nodes();
        size_t r = c.get_n_rec_nodes();
        return std::max(s, r) > m_max_nodes;
    }
};
 
class divide_diameter : public divide_base<divide_diameter>
{
    double m_diameter;
 
public:
    divide_diameter(double d)
        : m_diameter(d)
    {
    }
 
    template <class Cluster>
    bool operator()(Cluster const &c) const
    {
        return c.get_bounding_box().get_diameter() > m_diameter;
    }
};
 
} // end of namespace fmm
} // end of namespace NiHu
 
#endif /* NIHU_DIVIDE_HPP_INCLUDED */