p2x_indexed.hpp

Go to the documentation of this file.

 
#ifndef NIHU_P2X_INDEXED_HPP_INCLUDED
#define NIHU_P2X_INDEXED_HPP_INCLUDED
 
#include "fmm_operator.hpp"
#include <type_traits>
 
namespace NiHu
{
namespace fmm
{
 
template <class Operator, class It>
class p2x_indexed
    : public fmm_operator<typename std::decay<Operator>::type::fmm_tag>
 
{
public:
    typedef typename std::decay<Operator>::type operator_t;
    typedef It iterator_t;
 
    typedef typename operator_t::test_input_t test_input_t;
    typedef size_t trial_input_t;
    typedef typename operator_t::result_t result_t;
 
    p2x_indexed(Operator &&op, It begin, It end)
        : m_op(std::forward<Operator>(op))
        , m_begin(begin)
        , m_end(end)
    {
    }
 
    result_t operator()(test_input_t const &ti, size_t idx) const
    {
        return m_op(ti, m_begin[idx]);
    }
 
    size_t num_src() const
    {
        return m_end - m_begin;
    }
 
private:
    Operator m_op;
    iterator_t m_begin;
    iterator_t m_end;
};
 
template <class Operator, class It>
auto create_p2x_indexed(Operator &&op, It begin, It end)
{
    return p2x_indexed<Operator, It>(std::forward<Operator>(op), begin, end);
}
 
} // end of namespace fmm
} // namespace NiHu
 
#endif /* NIHU_P2X_INDEXED_HPP_INCLUDED */