memusage.c

来自「用于2维的射线追踪」· C语言 代码 · 共 46 行

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#ifdef __APPLE__
	#include <malloc/malloc.h>
#else
	#include <malloc.h>
#endif

#include "memusage.h"

/** \brief get the current memory usage in MBytes */

#ifdef __APPLE__
/*
float get_mem_usage (void) {
   task_basic_info_data_t ti;
   mach_msg_type_number_t count;
   vm_statistics_data_t vm_info;
   vm_size_t       vsize, rsize;
   task_t          a_task = mach_task_self();

   count = TASK_BASIC_INFO_COUNT;
   task_info(a_task, TASK_BASIC_INFO, (task_info_t) & ti, &count);
   rsize = ti.resident_size;
   vsize = ti.virtual_size;

   return((float)(rsize / (1024. * 1024.)));
}
*/
float get_mem_usage (void) {
   malloc_statistics_t memusage;
   return ((float)(memusage.size_in_use)/1024.);
}

#else
float get_mem_usage (void) {
	struct mallinfo m_info;

	m_info = mallinfo();
	return((float)(m_info.uordblks+m_info.usmblks)/(1024.*1024.));
}
#endif