timer.h

Go to the documentation of this file.

// base/timer.h

// Copyright 2009-2011  Ondrej Glembek;  Microsoft Corporation

// See ../../COPYING for clarification regarding multiple authors
//
// 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

// THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED
// WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE,
// MERCHANTABLITY OR NON-INFRINGEMENT.
// See the Apache 2 License for the specific language governing permissions and
// limitations under the License.
#ifndef KALDI_BASE_TIMER_H_
#define KALDI_BASE_TIMER_H_

#include "base/kaldi-utils.h"
#include "base/kaldi-error.h"


#if defined(_MSC_VER) || defined(MINGW)

namespace kaldi {
class Timer {
 public:
  Timer() { Reset(); }

  // You can initialize with bool to control whether or not you want the time to
  // be set when the object is created.
  explicit Timer(bool set_timer) { if (set_timer) Reset(); }

  void Reset() {
    QueryPerformanceCounter(&time_start_);
  }
  double Elapsed() const {
    LARGE_INTEGER time_end;
    LARGE_INTEGER freq;
    QueryPerformanceCounter(&time_end);

    if (QueryPerformanceFrequency(&freq) == 0) {
      //  Hardware does not support this.
      return 0.0;
    }
    return (static_cast<double>(time_end.QuadPart) -
            static_cast<double>(time_start_.QuadPart)) /
           (static_cast<double>(freq.QuadPart));
  }
 private:
  LARGE_INTEGER time_start_;
};


#else
#include <sys/time.h>
#include <unistd.h>

namespace kaldi {
class Timer {
 public:
  Timer() { Reset(); }

  // You can initialize with bool to control whether or not you want the time to
  // be set when the object is created.
  explicit Timer(bool set_timer) { if (set_timer) Reset(); }

  void Reset() { gettimeofday(&this->time_start_, &time_zone_); }

  double Elapsed() const {
    struct timeval time_end;
    struct timezone time_zone;
    gettimeofday(&time_end, &time_zone);
    double t1, t2;
    t1 =  static_cast<double>(time_start_.tv_sec) +
          static_cast<double>(time_start_.tv_usec)/(1000*1000);
    t2 =  static_cast<double>(time_end.tv_sec) +
          static_cast<double>(time_end.tv_usec)/(1000*1000);
    return t2-t1;
  }

 private:
  struct timeval time_start_;
  struct timezone time_zone_;
};

#endif

class Profiler {
 public:
  // Caution: the 'const char' should always be a string constant; for speed,
  // internally the profiling code uses the address of it as a lookup key.
  Profiler(const char *function_name): name_(function_name) { }
  ~Profiler();
 private:
  Timer tim_;
  const char *name_;
};

//  To add timing info for a function, you just put
//  KALDI_PROFILE;
//  at the beginning of the function.  Caution: this doesn't
//  include the class name.
#define KALDI_PROFILE Profiler _profiler(__func__)



}  // namespace kaldi


#endif  // KALDI_BASE_TIMER_H_