userproject/include/timer.h

Go to the documentation of this file.

#pragma once

// Standard C++ includes
#include <chrono>
#include <fstream>
#include <iostream>
#include <string>

//------------------------------------------------------------------------
// Timer
//------------------------------------------------------------------------
class Timer
{
public:
    Timer(const std::string &message, const std::string &filename = "")
    :   m_Start(std::chrono::high_resolution_clock::now()), m_Message(message), m_Filename(filename)
    {
    }

    ~Timer()
    {
        const double duration = get();
        std::cout << m_Message << duration << " seconds" << std::endl;

        // If we specified a filename, write duration
        if(!m_Filename.empty()) {
            std::ofstream output(m_Filename, std::ios::app);
            output << duration << std::endl;
        }
    }

    //------------------------------------------------------------------------
    // Public API
    //------------------------------------------------------------------------
    double get() const
    {
        auto now = std::chrono::high_resolution_clock::now();
        std::chrono::duration<double> duration = now - m_Start;
        return duration.count();
    }

private:
    //------------------------------------------------------------------------
    // Members
    //------------------------------------------------------------------------
    const std::chrono::time_point<std::chrono::high_resolution_clock> m_Start;
    const std::string m_Message;
    const std::string m_Filename;
};


//------------------------------------------------------------------------
// TimerAccumulate
//------------------------------------------------------------------------
class TimerAccumulate
{
public:
    TimerAccumulate(double &accumulator) : m_Start(std::chrono::high_resolution_clock::now()), m_Accumulator(accumulator)
    {}

    ~TimerAccumulate()
    {
        m_Accumulator += get();
    }

    //------------------------------------------------------------------------
    // Public API
    //------------------------------------------------------------------------
    double get() const
    {
        auto now = std::chrono::high_resolution_clock::now();
        std::chrono::duration<double> duration = now - m_Start;
        return duration.count();
    }

private:
    //------------------------------------------------------------------------
    // Members
    //------------------------------------------------------------------------
    std::chrono::time_point<std::chrono::high_resolution_clock> m_Start;
    double &m_Accumulator;
};