More vtkSmartPointer

[clitk.git] / common / clitkMemoryUsage.cxx

diff --git a/common/clitkMemoryUsage.cxx b/common/clitkMemoryUsage.cxx
index 7e0079e80b987ee5acc2ba33eaea3df10dbafa25..ecabfc1af97eacfd51c65ae4afd3ce5b296f1d7d 100644 (file)
--- a/common/clitkMemoryUsage.cxx
+++ b/common/clitkMemoryUsage.cxx
@@ -20,9 +20,6 @@
 #include "clitkCommon.h"
 #include "clitkMemoryUsage.h"
 
-// itk include
-#include "itkMemoryUsageObserver.h"
-
 void clitk::PrintMemory(bool verbose, std::string s) 
 {
 #if CLITK_MEMORY_INFO == 1
@@ -32,13 +29,27 @@ void clitk::PrintMemory(bool verbose, std::string s)
     if (s != "") std::cout << "==> " << s << ": ";
     std::cout << mem << "MB (" << mem-previous << "MB)" << std::endl;
     previous = mem;
+    //DD(stat[i].proc_resident/1000/1000);
+    //DD(stat[i].pid);
   }
 #endif
-}
+  }
 
 double clitk::GetMemoryUsageInMb() 
 {
-  itk::MemoryUsageObserver memUsage;
-  return memUsage.GetMemoryUsage() / 1024.;
-}
+#if CLITK_MEMORY_INFO == 1
+  //sleep(1); // wait to refresh memory ? need to let the system refresh the mem
+  int * entries = new int;
+  sg_process_stats * stat = new sg_process_stats;
+  int i=0; 
+  stat = sg_get_process_stats(entries);
+  // Search the current pid in the list of processes
+  while (stat[i].pid != getpid()) i++;
+  // Display total memory size 
+  double mem = stat[i].proc_resident/1024/1024; // in Mb
+  return mem;
+#else
+  return 0.0;
+#endif
+  }