getcpu.c

/* ------- file: -------------------------- gettime.c ---------------

       Version:       rh2.0
       Author:        Han Uitenbroek (huitenbroek@nso.edu)
       Last modified: Thu Nov  1 12:51:38 2007 --

       --------------------------                      ----------RH-- */

/* --- Prints elapsed time in seconds:    action == TIME_POLL.
       Clock is started if called with:   action == TIME_START.
       Borrowed heavily from MULTI's ctime().

       Works on:    SUN sparc  (SunOs 4.x & 5.x)
                    SGI mips   (IRIX 5.2)
                    DEC alpha  (OSF1 3.2 & 4.0)
                    i386       (Linux 1.x)
                    i586       (Linux 2.0)
                    --                                 -------------- */

#include <stdio.h>
#include <string.h>
#include <time.h>
#include <sys/times.h>
#include <unistd.h>

#if defined(SunOS5)
#include <sys/time.h>

#if defined(_POSIX_C_SOURCE)
hrtime_t gethrtime(void);
#endif

#define  NANOSECOND  1.0E-9
#endif

#include "rh.h"
#include "statistics.h"
#include "error.h"
#include "inputs.h"

#define  TAB_POSITION 24
#define  DOTS   "...................................."
#define  SPACES "                                    "
#define  TIME_DOT_OUT "time.out"

/* --- Function prototypes --                          -------------- */


/* --- Global variables --                             -------------- */

extern ProgramStats stats;
extern CommandLine  commandline;
extern char messageStr[];

/* ------- begin -------------------------- getTime.c --------------- */

void getCPU(int level, enum CPUaction action, char *label)
{
  register int n;

  int Nblanck, Nspace, Ndot;

#if defined(SunOS5)
  static hrtime_t CPU[N_TIME_LEVELS];
  static double   scale = NANOSECOND;
  hrtime_t CPUtime;
#else
  static clock_t CPU[N_TIME_LEVELS];
  static double  scale = 1.0 / CLOCKS_PER_SEC;
  clock_t CPUtime;
#endif

  if (level == 0 && action == TIME_START && !commandline.showkeywords) {
    stats.fp_CPU = fopen(TIME_DOT_OUT, "w");
    setvbuf(stats.fp_CPU, NULL, _IOLBF, BUFSIZ);
  }

  if (!stats.printCPU || !stats.fp_CPU)
    return;
  else {

#if defined(SunOS5)
    /* --- Use high-resolution time on SUN Solaris machines -- ------ */

    CPUtime = gethrtime();
#else
    if ((CPUtime = clock()) == ((clock_t) -1)) {
      sprintf(messageStr, "Cannot poll resources");
      Error(ERROR_LEVEL_1, "getCPU", messageStr);
    }
#endif

    switch (action) {
    case TIME_START:
      CPU[level] = CPUtime;
      return;
    case TIME_POLL:
      CPUtime = CPUtime - CPU[level];
      break;
    case TIME_ADD:
      break;
    }
  }
  /* --- Format output --                              -------------- */

  Nspace  = (level < 4) ? level + 1 : 4;
  Ndot    = TAB_POSITION - Nspace - strlen(label);
  Nblanck = (level > 2) ? 0 : 3 - level;

  fprintf(stats.fp_CPU, "%.*s%.*s%s: %7.3f [sec]\n", Nspace, SPACES,
	  Ndot, DOTS, label, CPUtime * scale);
  for (n = 0;  n < Nblanck;  n++) fprintf(stats.fp_CPU, "\n");
}

/* ------- end ---------------------------- getTime.c --------------- */

/* ------- begin -------------------------- printTotalCPU ----------- */

void printTotalCPU()
{
  double user, system, total;
  struct tms tms;

  times(&tms);
  user   = ((double) tms.tms_utime) / sysconf(_SC_CLK_TCK);
  system = ((double) tms.tms_stime) / sysconf(_SC_CLK_TCK);
  total  = user + system;
  
  fprintf(stats.fp_CPU, "\nTotal_time____: %8.3f [sec] ", total);
  fprintf(stats.fp_CPU, " (user: %4.1f%%, system: %4.1f%%)\n\n",
	  100*user/total, 100*system/total);
}
/* ------- end ---------------------------- printTotalCPU ----------- */