/*=========================================================================
Program: vv http://www.creatis.insa-lyon.fr/rio/vv
- Authors belong to:
+ Authors belong to:
- University of LYON http://www.universite-lyon.fr/
- Léon Bérard cancer center http://www.centreleonberard.fr
- CREATIS CNRS laboratory http://www.creatis.insa-lyon.fr
// clitk include
#include "clitkCommon.h"
#include "clitkMemoryUsage.h"
-#include <unistd.h>
+#ifndef _WIN32
+# include <unistd.h>
+#endif
-void clitk::PrintMemory(bool verbose, std::string s)
+void clitk::PrintMemory(bool verbose, std::string s)
{
#if CLITK_MEMORY_INFO == 1
if (verbose) {
#endif
}
-double clitk::GetMemoryUsageInMb()
+double clitk::GetMemoryUsageInMb()
{
#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);
+ size_t entries;
+ sg_process_stats *stat;
+ int i=0;
+ sg_init(1);
+
+ stat = sg_get_process_stats(&entries);
+ if(stat == NULL)
+ return 0.0;
+
// 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
+ while (stat->pid != getpid() && i < entries)
+ {
+ ++i;
+ ++stat;
+ }
+
+ // Display total memory size
+ double mem = stat->proc_resident/1000/1000; // in MB
return mem;
#else
return 0.0;
#endif
}
-