]> Creatis software - clitk.git/blob - common/clitkMemoryUsage.cxx
Add code to write dicom sequence tag in gateSimulation2Dicom
[clitk.git] / common / clitkMemoryUsage.cxx
1 /*=========================================================================
2   Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
3
4   Authors belong to:
5   - University of LYON              http://www.universite-lyon.fr/
6   - Léon Bérard cancer center       http://www.centreleonberard.fr
7   - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr
8
9   This software is distributed WITHOUT ANY WARRANTY; without even
10   the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
11   PURPOSE.  See the copyright notices for more information.
12
13   It is distributed under dual licence
14
15   - BSD        See included LICENSE.txt file
16   - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
17   ===========================================================================**/
18
19 // clitk include
20 #include "clitkCommon.h"
21 #include "clitkMemoryUsage.h"
22 #ifndef _WIN32
23 #  include <unistd.h>
24 #endif
25
26 void clitk::PrintMemory(bool verbose, std::string s)
27 {
28 #if CLITK_MEMORY_INFO == 1
29   if (verbose) {
30     static double previous=0;
31     double mem = GetMemoryUsageInMb();
32     if (s != "") std::cout << "==> " << s << ": ";
33     std::cout << mem << "MB (" << mem-previous << "MB)" << std::endl;
34     previous = mem;
35     //DD(stat[i].proc_resident/1000/1000);
36     //DD(stat[i].pid);
37   }
38 #endif
39   }
40
41 double clitk::GetMemoryUsageInMb()
42 {
43 #if CLITK_MEMORY_INFO == 1
44   //sleep(1); // wait to refresh memory ? need to let the system refresh the mem
45   size_t entries;
46   sg_process_stats *stat;
47   int i=0;
48   sg_init(1);
49
50   stat = sg_get_process_stats(&entries);
51   if(stat == NULL)
52     return 0.0;
53
54   // Search the current pid in the list of processes
55   while (stat->pid != getpid() && i < entries)
56   {
57     ++i;
58     ++stat;
59   }
60
61   // Display total memory size
62   double mem = stat->proc_resident/1000/1000; // in MB
63   return mem;
64 #else
65   return 0.0;
66 #endif
67   }