/* Portable CPU-timer (User + Sys); also WALL CLOCK-timer */

#if  defined  (HAVE_CONFIG_H)
#  include "config.h"
#endif

#include <stdio.h>
#include <unistd.h>
#include <time.h>
#include <sys/types.h>
#include <sys/time.h>
#if defined (HAVE_SYS_TIMES_H)
#include <sys/times.h>
#endif
/* #include <sys/param.h> */
#ifdef __XT3__
#include <catamount/dclock.h>
#endif

#ifdef __XT3__
static double clock0=-1.0;
#endif

/* extern clock_t times (struct tms *buffer); */
#define clock_ticks ( (double) sysconf(_SC_CLK_TCK) )

int util_cputime(double *user_time, double *system_time)
{
#ifdef __XT3__
  *user_time   = dclock()-clock0;
  *system_time = (double) 0;
#else
#if defined (HAVE_SYS_TIMES_H)
  struct tms tbuf;
  if (times(&tbuf) == (clock_t)-1) return ((int) (-1)); 

  *user_time   = ((double) tbuf.tms_utime) / clock_ticks; 
  *system_time = ((double) tbuf.tms_stime) / clock_ticks;
#else
  *user_time   = 0; 
  *system_time = 0;
#endif
#endif

  return (0);
}

double util_walltime(void)
{
  double time_in_secs = 0;
#if defined (HAVE_SYS_TIMES_H)
  static double time_init = 0.;

  struct timeval tbuf;
  if (gettimeofday(&tbuf,NULL) == -1) perror("UTIL_WALLTIME");

  if (time_init == 0.)
    time_init = (double)tbuf.tv_sec + ((double)tbuf.tv_usec / 1000000.0);

  time_in_secs =
    (double)tbuf.tv_sec + ((double)tbuf.tv_usec / 1000000.0) - time_init;
#endif

  return (time_in_secs);
}
 
/****************************************************************************/

#ifdef _AIX

/*
 * High-Resolution Time
 * pwr4: measurement (util_read_real_time) overhead ~ 0.07 us
 *       conversion  (util_diff_real_time) overhead ~ 0.3 us
 */

#include <sys/systemcfg.h>

static double aix_rt_tconv, aix_rt_fhigh;

void util_init_real_time(void)
{
  double tb_top, tb_bot;

  if (_system_configuration.implementation == POWER_RS2) {
    aix_rt_fhigh = 1.0;
    aix_rt_tconv = 1.0;
  } else  {
    /* powerpc family */ 
    aix_rt_fhigh = 4.294967296;
    tb_top = (double) _system_configuration.Xint;
    tb_bot = (double) _system_configuration.Xfrac;
    aix_rt_tconv = tb_top/tb_bot; 
  }
}

void util_get_real_time_size(int *rt_size) 
{ 
  /* fortran out:  integer*4:: rt_size(4) */
  *rt_size = (int) sizeof(struct timebasestruct);
}

void util_read_real_time(void *it) 
{ 
  /* fortran out:  integer*4:: it(4)
   * raw values - not yet scaled to real time
   */
  read_real_time( (struct timebasestruct*) it, TIMEBASE_SZ );
}

void util_diff_real_time(void *it1, void *it2, double *t) 
{
  /* fortran in:  integer*4:: it1(4), it2(4)
   * fortran out: real*8:: t
   * t is the real time diff between measurements it1 and it2
   */
  
  struct timebasestruct *tb1, *tb2;
  
  tb1 = (struct timebasestruct *) it1;
  tb2 = (struct timebasestruct *) it2;
  
  *t = aix_rt_tconv*(aix_rt_fhigh*( (double) (tb2->tb_high - tb1->tb_high) ) 
      +1.0e-9*( (double)tb2->tb_low - (double)tb1->tb_low ));
} 

#elif defined (_HIGH_RESOLUTION_TIMER) && (defined (SX) || defined (ES)) 

#define CPU_CLOCK 2.0e-9    /* SX-6: 500 MHz */

long long int irtc(void);   /* provided as assembler routine */   

void util_init_real_time() 
{
}

void util_get_real_time_size(int *rt_size) 
{ 
  *rt_size = (int) sizeof(double);
}

void util_read_real_time(void *it) 
{ 
  double *t;
  t = (double *) it;
  *t = CPU_CLOCK*irtc();
}

void util_diff_real_time(void *it1, void *it2, double *t) 
{
  double *t1, *t2;
  t1 = (double*) it1;
  t2 = (double*) it2;
  *t = *t2 - *t1;
} 

#elif defined (_HIGH_RESOLUTION_TIMER) && defined (LINUX)

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <asm/msr.h>

static double cpu_clock;

void util_init_real_time() 
{
  double freq = 0.0;
  FILE *cpuinfo;
  char line[256];
  
  if ((cpuinfo = fopen ("/proc/cpuinfo", "r")) == NULL) {
    fprintf (stderr, "Couldn't open /proc/cpuinfo ...\n");
    exit(-1);
  }

  while (! feof(cpuinfo)) {
    (void) fgets (line, 255, cpuinfo);
    if (strstr(line, "cpu MHz") != NULL) {
      sscanf (line, "cpu MHz         : %lf", &freq);
      break;
    }
  }
  if (freq == 0.0) {
    fprintf (stderr, "Couldn't find cpu MHz in /proc/cpuinfo ...\n");
    exit(-1);
  }

  cpu_clock = freq*10e6;
}

void util_get_real_time_size(int *rt_size)
{ 
  *rt_size = (int) sizeof(double);
}

void util_read_real_time(void *it)
{ 
  long long llt;
  double *t;
  t = (double *) it;
  rdtscll(llt);
  *t = (double) llt/cpu_freq;
}

void util_diff_real_time(void *it1, void *it2, double *t)
{
  double *t1, *t2;
  t1 = (double*) it1;
  t2 = (double*) it2;
  *t = *t2 - *t1;
} 

#else

/* fall back to util_walltime */

void util_init_real_time(void)
{
}

void util_get_real_time_size(int *rt_size)
{ 
  *rt_size = (int) sizeof(double);
}

void util_read_real_time(void *it)
{ 
  double *t;
  t = (double *) it;
  *t=util_walltime();
}

void util_diff_real_time(void *it1, void *it2, double *t)
{
  double *t1, *t2;
  t1 = (double*) it1;
  t2 = (double*) it2;
  *t = *t2 - *t1;
} 

#endif





