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