include/timer.h

Go to the documentation of this file.

#ifndef TIMER_H
#define TIMER_H

//== INCLUDES =================================================================

#include <string>
#include <list>
#include <cuda_runtime.h>

//== NAMESPACES ===============================================================

namespace gpufilter {

//== CLASS DEFINITION =========================================================

class base_timer {

public:

    base_timer( const char *type_label,
                size_t data_size = 0,
                const std::string &unit = "" );

    void start();

    void stop();

    float elapsed();

    bool is_stopped() const { return !m_started; }

    size_t data_size() const { return m_data_size; }

    const std::string &unit() const { return m_unit; }

    const char *type_label() { return m_type_label; }

protected:

    virtual void do_start() = 0;

    virtual void do_stop() = 0;

    virtual float do_get_elapsed() const = 0;

private:

    base_timer( const base_timer& bt );

    base_timer& operator = ( const base_timer& bt );

    const char *m_type_label; 

    float m_elapsed; 

    bool m_started; 

    size_t m_data_size; 

    std::string m_unit; 

};

//== CLASS DEFINITION =========================================================

class gpu_timer : public base_timer {

public:

    gpu_timer( size_t data_size = 0,
               const std::string& unit = "",
               bool start = true );

    ~gpu_timer();

private:

    virtual void do_start();

    virtual void do_stop();

    virtual float do_get_elapsed() const;

    cudaEvent_t m_start; 
    cudaEvent_t m_stop; 

};

//== CLASS DEFINITION =========================================================

class cpu_timer : public base_timer {

public:

    cpu_timer( size_t data_size = 0,
               const std::string& unit = "",
               bool start = true );

    ~cpu_timer();

private:

    virtual void do_start();

    virtual void do_stop();

    virtual float do_get_elapsed() const;

    double get_cpu_time() const;

    double m_start_time; 
    double m_stop_time; 

};

//== CLASS DEFINITION =========================================================

class scoped_timer_stop {

public:

    scoped_timer_stop( base_timer& timer );

    ~scoped_timer_stop() { stop(); }

    void stop() { m_timer->stop(); }

    float elapsed() const { return m_timer->elapsed(); }

private:

    base_timer *m_timer; 

};

//== CLASS DEFINITION =========================================================

class timer_pool {

public:

    ~timer_pool() { }

    gpu_timer &gpu_add( const std::string& label,
                        size_t data_size = 0,
                        const std::string& unit = "" );

    cpu_timer &cpu_add( const std::string& label,
                        size_t data_size = 0,
                        const std::string& unit = "" );

    void flush();

private:

    struct timer_data {
        base_timer *timer; 
        std::string label; 
        int level; 
    };

    typedef std::list<timer_data> timer_list; 

    timer_list m_timers; 

};

//== EXTERNS ==================================================================

extern
timer_pool timers;

//=============================================================================
} // namespace gpufilter
//=============================================================================
#endif // TIMER_H
//=============================================================================
//vi: ai sw=4 ts=4