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
  23 void clitk::PrintMemory(bool verbose, std::string s)
  24 {
  25 #if CLITK_MEMORY_INFO == 1
  26   if (verbose) {
  27     static double previous=0;
  28     double mem = GetMemoryUsageInMb();
  29     if (s != "") std::cout << "==> " << s << ": ";
  30     std::cout << mem << "MB (" << mem-previous << "MB)" << std::endl;
  31     previous = mem;
  32     //DD(stat[i].proc_resident/1000/1000);
  33     //DD(stat[i].pid);
  34   }
  35 #endif
  36   }
  37
  38 double clitk::GetMemoryUsageInMb()
  39 {
  40 #if CLITK_MEMORY_INFO == 1
  41   //sleep(1); // wait to refresh memory ? need to let the system refresh the mem
  42   int * entries = new int;
  43   sg_process_stats * stat = new sg_process_stats;
  44   int i=0;
  45   stat = sg_get_process_stats(entries);
  46   // Search the current pid in the list of processes
  47   while (stat[i].pid != getpid()) i++;
  48   // Display total memory size
  49   double mem = stat[i].proc_resident/1024/1024; // in Mb
  50   return mem;
  51 #else
  52   return 0.0;
  53 #endif
  54   }
  55