fmm_timer.h

Go to the documentation of this file.

 
#ifndef NIHU_FMM_TIMER_H_INCLUDED
#define NIHU_FMM_TIMER_H_INCLUDED
 
#include "fmm_operator.hpp"
#include "util/timer.h"
 
#include <array>
#include <iostream>
#include <vector>
 
namespace NiHu
{
namespace fmm
{
 
class fmm_timer
{
    typedef NiHu::cpu_time timer_t;
    
public:
    enum {
        M2M = op_tags::tag2idx(op_tags::m2m()),
        L2L = op_tags::tag2idx(op_tags::l2l()),
        M2L = op_tags::tag2idx(op_tags::m2l()),
        P2M = op_tags::tag2idx(op_tags::p2m()),
        P2L = op_tags::tag2idx(op_tags::p2l()),
        L2P = op_tags::tag2idx(op_tags::l2p()),
        M2P = op_tags::tag2idx(op_tags::m2p()),
        P2P = op_tags::tag2idx(op_tags::p2p()),
        NUM_TIME_INDICES = op_tags::num_tags(),
    } time_index_t;
 
private:
    std::vector<std::array<long long, NUM_TIME_INDICES> > m_times;
    timer_t::time_point_t m_t0;
 
public:
    fmm_timer(size_t nLevel)
        : m_times(nLevel)
    {
    }
 
    void reset(void)
    {
        for (auto &it : m_times)
            it[M2M] = it[L2L] = it[M2L] = it[P2M] = it[P2L] = it[L2P] = it[M2P] = it[P2P] = 0;
    }
 
    typename timer_t::time_point_t tic(void)
    {
        return m_t0 = timer_t::tic();
    }
 
    void toc(size_t level, int type)
    {
        m_times[level][type] += (long long) (timer_t::toc(m_t0) * 1e6);
    }
 
    std::ostream &print(std::ostream &os = std::cout) const;
 
    std::vector<std::array<long long, NUM_TIME_INDICES> > const & get_times(void) const
    {
        return m_times;
    }
};
 
} // namespace fmm
} // namesapce NiHu
 
#endif /* NIHU_FMM_TIMER_H_INCLUDED */