| 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
81 >> stime >> cutime >> cstime >> priority > << 81 >> utime >> stime >> cutime >> cstime >> priority >> nice
82 >> starttime >> vsize >> rss; // don't car << 82 >> O >> itrealvalue >> 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 } <<
99 98
100 } 99 }
101 100