File: rusage.hh

package info (click to toggle)
packup 0.6-3.2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 832 kB
  • sloc: cpp: 10,857; makefile: 111; sh: 6
file content (86 lines) | stat: -rw-r--r-- 3,278 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
/******************************************************************************\
 *    This file is part of packup.                                            *
 *                                                                            *
 *    packup is free software: you can redistribute it and/or modify          *
 *    it under the terms of the GNU General Public License as published by    *
 *    the Free Software Foundation, either version 3 of the License, or       *
 *    (at your option) any later version.                                     *
 *                                                                            *
 *    packup is distributed in the hope that it will be useful,               *
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of          *
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the           *
 *    GNU General Public License for more details.                            *
 *                                                                            *
 *    You should have received a copy of the GNU General Public License       *
 *    along with packup.  If not, see <http://www.gnu.org/licenses/>.         *            
\******************************************************************************/           
/*----------------------------------------------------------------------------*\
 * Version: $Id: rusage.hh 73 2007-07-26 15:16:48Z jpms $
 *
 * Author: jpms
 * 
 * Description: Obtain time and memory resource usage stats.
 *              Adapted from minisat and from MCSAT.
 *
 *                                    Copyright (c) 2006, Joao Marques-Silva
\*----------------------------------------------------------------------------*/

#ifndef _RUSAGE_HH_
#define _RUSAGE_HH_ 1

#include <sys/time.h>
#include <sys/resource.h>
#include <unistd.h>
#include <stdio.h>

#include <iostream>

namespace RUSAGE {
  static inline double read_cpu_time();
  static inline long read_mem_stats(int fields);
  static inline double read_mem_used();
  static inline void print_cpu_time(const char* msg, ostream& outs=std::cout);
  static inline void print_mem_used(const char* msg, ostream& outs=std::cout);
}


static inline double RUSAGE::read_cpu_time()
{
  struct rusage ru; getrusage(RUSAGE_SELF, &ru);
  return (double)ru.ru_utime.tv_sec + (double)ru.ru_utime.tv_usec / 1000000;
}

#define NSTRSZ 256
static inline long RUSAGE::read_mem_stats(int fields)
{
  char    name[NSTRSZ];
  pid_t pid = getpid();
  sprintf(name, "/proc/%d/statm", pid);
  FILE*   finp = fopen(name, "rb");
  if (finp == NULL) { /* assert(0); */ return 0; }
  int value;
  for (; fields >= 0; fields--) { fscanf(finp, "%d", &value); }
  fclose(finp);
  return value;
}

#define __1MBYTE__ 1048576

static inline double RUSAGE::read_mem_used()
{
  return ((long)read_mem_stats(0) * (long)getpagesize() * 1.0) / __1MBYTE__;
}

static inline void RUSAGE::print_cpu_time(const char* msg, ostream& outs)
{
  outs << msg << ": "<< RUSAGE::read_cpu_time() << endl;
}

static inline void RUSAGE::print_mem_used(const char* msg, ostream& outs)
{
  outs << msg << ": " << RUSAGE::read_mem_used() << endl;
}

#endif /* _RUSAGE_HH_ */

/*----------------------------------------------------------------------------*/