| Geant4 Cross Reference |
1 // 1 //
2 // ******************************************* 2 // ********************************************************************
3 // * License and Disclaimer 3 // * License and Disclaimer *
4 // * 4 // * *
5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. 9 // * include a list of copyright holders. *
10 // * 10 // * *
11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. *
17 // * 17 // * *
18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. *
24 // ******************************************* 24 // ********************************************************************
25 // 25 //
26 /* 26 /*
27 * MemStat.cc 27 * MemStat.cc
28 * 28 *
29 * Created on: 4 févr. 2014 29 * Created on: 4 févr. 2014
30 * Author: kara 30 * Author: kara
31 */ 31 */
32 32
33 #include "G4MemStat.hh" 33 #include "G4MemStat.hh"
34 34
35 #if ( defined(__MACH__) && defined(__clang__) 35 #if ( defined(__MACH__) && defined(__clang__) && defined(__x86_64__) ) || \
36 ( defined(__MACH__) && defined(__GNUC__) & 36 ( defined(__MACH__) && defined(__GNUC__) && __GNUC__>=4 && __GNUC_MINOR__>=7 ) || \
37 defined(__linux__) || defined(_AIX) 37 defined(__linux__) || defined(_AIX)
38 38
39 #include <unistd.h> 39 #include <unistd.h>
40 40
41 #endif 41 #endif
42 42
43 #include <ios> 43 #include <ios>
44 #include <iostream> 44 #include <iostream>
45 #include <fstream> 45 #include <fstream>
46 #include <string> 46 #include <string>
47 47
48 namespace G4MemStat 48 namespace G4MemStat
49 { 49 {
50 50
51 using std::ios_base; 51 using std::ios_base;
52 using std::ifstream; 52 using std::ifstream;
53 using std::string; 53 using std::string;
54 54
55 MemStat MemoryUsage() 55 MemStat MemoryUsage()
56 { 56 {
57 MemStat output; 57 MemStat output;
58 58
59 #if ( defined(__MACH__) && defined(__clang__) 59 #if ( defined(__MACH__) && defined(__clang__) && defined(__x86_64__) ) || \
60 ( defined(__MACH__) && defined(__GNUC__) & 60 ( defined(__MACH__) && defined(__GNUC__) && __GNUC__>=4 && __GNUC_MINOR__>=7 ) || \
61 defined(__linux__) || defined(_AIX) 61 defined(__linux__) || defined(_AIX)
62 62
63 // 'file' stat seems to give the most reli 63 // 'file' stat seems to give the most reliable results
64 // 64 //
65 ifstream stat_stream("/proc/self/stat", io 65 ifstream stat_stream("/proc/self/stat", ios_base::in);
66 66
67 // dummy vars for leading entries in stat 67 // dummy vars for leading entries in stat that we don't care about
68 // 68 //
69 string pid, comm, state, ppid, pgrp, sessi 69 string pid, comm, state, ppid, pgrp, session, tty_nr;
70 string tpgid, flags, minflt, cminflt, majf 70 string tpgid, flags, minflt, cminflt, majflt, cmajflt;
71 string utime, stime, cutime, cstime, prior 71 string utime, stime, cutime, cstime, priority, nice;
72 string O, itrealvalue, starttime; 72 string O, itrealvalue, starttime;
73 73
74 // the two fields we want 74 // the two fields we want
75 // 75 //
76 unsigned long vsize; 76 unsigned long vsize;
77 long rss; 77 long rss;
78 78
79 stat_stream >> pid >> comm >> state >> ppi 79 stat_stream >> pid >> comm >> state >> ppid >> pgrp >> session >> tty_nr
80 >> tpgid >> flags >> minflt >> cminflt >> 80 >> tpgid >> flags >> minflt >> cminflt >> majflt >> cmajflt >> utime
81 >> stime >> cutime >> cstime >> priority > 81 >> stime >> cutime >> cstime >> priority >> nice >> O >> itrealvalue
82 >> starttime >> vsize >> rss; // don't car 82 >> starttime >> vsize >> rss; // don't care about the rest
83 83
84 stat_stream.close(); 84 stat_stream.close();
85 85
86 long page_size_kb = sysconf(_SC_PAGE_SIZE) 86 long page_size_kb = sysconf(_SC_PAGE_SIZE) / 1024; // in case x86-64 is configured to use 2MB pages
87 output.vmz = vsize / 1024.0; 87 output.vmz = vsize / 1024.0;
88 output.mem = rss * page_size_kb; 88 output.mem = rss * page_size_kb;
89 #endif 89 #endif
90 90
91 return output; 91 return output;
92 } 92 }
93 93
94 std::ostream & operator<<(std::ostream &os, 94 std::ostream & operator<<(std::ostream &os, const MemStat& memStat)
95 { 95 {
96 return os << "( vmz: " << memStat.vmz << " 96 return os << "( vmz: " << memStat.vmz << ", " << "mem: " << memStat.mem
97 << ")"; 97 << ")";
98 } 98 }
99 99
100 } 100 }
101 101