stopwatch.h

/******************************************************************************
 *   Copyright (C) 2006-2024 by the GIMLi development team                    *
 *   Carsten Rücker carsten@resistivity.net                                   *
 *                                                                            *
 *   Licensed under the Apache License, Version 2.0 (the "License");          *
 *   you may not use this file except in compliance with the License.         *
 *   You may obtain a copy of the License at                                  *
 *                                                                            *
 *       http://www.apache.org/licenses/LICENSE-2.0                           *
 *                                                                            *
 *   Unless required by applicable law or agreed to in writing, software      *
 *   distributed under the License is distributed on an "AS IS" BASIS,        *
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. *
 *   See the License for the specific language governing permissions and      *
 *   limitations under the License.                                           *
 *                                                                            *
 ******************************************************************************/
 
#ifndef _GIMLI_STOPWATCH__H
#define _GIMLI_STOPWATCH__H
 
#include "gimli.h"
 
#include <sys/timeb.h>
#include <chrono>
 
#if defined(__i386__)
static __inline__ size_t rdtsc__(void){
    size_t x;
    __asm__ volatile (".byte 0x0f, 0x31" : "=A" (x));
    return x;
}
#elif defined(__x86_64__)
static __inline__ size_t rdtsc__(void){
    unsigned hi, lo;
    __asm__ __volatile__ ("rdtsc" : "=a"(lo), "=d"(hi));
    return ((size_t)lo)|(((size_t)hi)<<32);
}
#else
static inline size_t rdtsc__(void){
    return 0;
}
#endif
 
namespace GIMLI{
 
class DLLEXPORT CycleCounter{
public:
    CycleCounter() : var_(0) {}
 
    ~CycleCounter(){}
 
    inline void tic(){ var_ = rdtsc__(); }
 
    inline size_t toc() const { return (rdtsc__() - var_); }
 
protected:
 
    size_t var_;
};
 
class DLLEXPORT Stopwatch {
public:
    Stopwatch(bool start=true);
 
    ~Stopwatch();
 
    void start();
 
    void stop(bool verbose=false);

    void restart();

    void reset();

    double duration(bool restart=false);

    size_t cycles(bool restart=false);
 
    const CycleCounter & cycleCounter() const { return _cCounter; }

    void store(bool restart=false);

    const RVector & stored() const { return *this->_store;}
 
 
protected:
    enum watchstate {undefined,halted,running} _state;
 
    std::chrono::time_point<std::chrono::high_resolution_clock> _start, _stop;
    RVector *_store;
    
    CycleCounter _cCounter;
};
 
#define TIC__ std::cout.precision(12); GIMLI::Stopwatch __swatch__(true);
#define TOC__ std::cout << __swatch__.duration(true) << std::endl;
#define toc__ __swatch__.duration()
 
} // namespace GIMLI
 
#endif // _GIMLI_STOPWATCH__H