/* -----------------------------------------------------------------------
   Copyright (2010) Sandia Corporation.  Under the terms of Contract
   DE-AC04-94AL85000 with Sandia Corporation, the U.S. Government retains
   certain rights in this software.  This software is distributed under
   the Simplified BSD License.
   ----------------------------------------------------------------------- */

#ifndef HIP_TIMER_H
#define HIP_TIMER_H


#include <hip/hip_runtime.h>
#include "hip_macros.h"
#include "hip_device.h"

namespace ucl_hip {

/// Class for timing CUDA Driver events
class UCL_Timer {
 public:
  inline UCL_Timer() : _total_time(0.0f), _initialized(false) { }
  inline UCL_Timer(UCL_Device &dev) : _total_time(0.0f), _initialized(false)
    { init(dev); }

  inline ~UCL_Timer() { clear(); }

  /// Clear any data associated with timer
  /** \note init() must be called to reuse timer after a clear() **/
  inline void clear() {
    if (_initialized) {
      CU_DESTRUCT_CALL(hipEventDestroy(start_event));
      CU_DESTRUCT_CALL(hipEventDestroy(stop_event));
      _initialized=false;
      _total_time=0.0;
    }
  }

  /// Initialize default command queue for timing
  inline void init(UCL_Device &dev) { init(dev, dev.cq()); }

  /// Initialize command queue for timing
  inline void init(UCL_Device &dev, command_queue &cq) {
    clear();
    _cq=cq;
    _initialized=true;
    CU_SAFE_CALL( hipEventCreateWithFlags(&start_event,0) );
    CU_SAFE_CALL( hipEventCreateWithFlags(&stop_event,0) );
  }

  /// Start timing on command queue
  inline void start() { CU_SAFE_CALL(hipEventRecord(start_event,_cq)); }

  /// Stop timing on command queue
  inline void stop() { CU_SAFE_CALL(hipEventRecord(stop_event,_cq)); }

  /// Block until the start event has been reached on device
  inline void sync_start()
    { CU_SAFE_CALL(hipEventSynchronize(start_event)); }

  /// Block until the stop event has been reached on device
  inline void sync_stop()
    { CU_SAFE_CALL(hipEventSynchronize(stop_event)); }

  /// Set the time elapsed to zero (not the total_time)
  inline void zero() {
    CU_SAFE_CALL(hipEventRecord(start_event,_cq));
    CU_SAFE_CALL(hipEventRecord(stop_event,_cq));
  }

  /// Set the total time to zero
  inline void zero_total() { _total_time=0.0; }

  /// Add time from previous start and stop to total
  /** Forces synchronization **/
  inline double add_to_total()
    { double t=time(); _total_time+=t; return t/1000.0; }

  /// Add a user specified time to the total (ms)
  inline void add_time_to_total(const double t) { _total_time+=t; }

  /// Return the time (ms) of last start to stop - Forces synchronization
  inline double time() {
    float timer;
    CU_SAFE_CALL(hipEventSynchronize(stop_event));
    CU_SAFE_CALL( hipEventElapsedTime(&timer,start_event,stop_event) );
    return timer;
  }

  /// Return the time (s) of last start to stop - Forces synchronization
  inline double seconds() { return time()/1000.0; }

  /// Return the total time in ms
  inline double total_time() { return _total_time; }

  /// Return the total time in seconds
  inline double total_seconds() { return _total_time/1000.0; }

 private:
  hipEvent_t start_event, stop_event;
  hipStream_t _cq;
  double _total_time;
  bool _initialized;
};

} // namespace

#endif