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