Stopwatch.h

Go to the documentation of this file.

// n2p2 - A neural network potential package
// Copyright (C) 2018 Andreas Singraber (University of Vienna)
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
#ifndef STOPWATCH_H
#define STOPWATCH_H
 
#if !(defined(__linux__) || defined(__MACH__))
#pragma message("WARNING: Platform not supported.")
#define N2P2_NO_TIME
#endif
 
#ifdef N2P2_NO_TIME
#pragma message("WARNING: Compiling dummy Stopwatch class (-DN2P2_NO_TIME).")
#endif
 
#include <ctime>
#ifdef __MACH__
#include <mach/mach_time.h>
#endif
 
namespace nnp
{
 
class Stopwatch
{
 
public:
    Stopwatch();
    void                      start(bool newLoop = true);
    double                    stop();
    double                    loop();
    double                    getTotal() const {return timeTotal;};
    double                    getLoop() const {return timeLoop;};
    void                      reset();
 
private:
    enum State
    {
        STOPPED,
        RUNNING
    };
 
    State               state;
    bool                resetLoop;
    static const double NSEC;
    double              timeTotal;
    double              timeLoop;
#ifdef N2P2_NO_TIME
#elif __linux__
    timespec            time;
#elif __MACH__
    uint64_t            time;
#endif
 
    void                stopTime();
    double              updateTime();
};
 
}
 
#endif