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   }