1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
/* Copyright (C) 2005-2013 Shugo Maeda <shugo@ruby-lang.org> and Charlie Savage <cfis@savagexi.com>
Please see the LICENSE file for copyright and distribution information */
/* :nodoc: */
#include "rp_measurement.h"
static VALUE cMeasureMemory;
static double measure_memory(rb_trace_arg_t* trace_arg)
{
static double result = 0;
if (trace_arg)
{
// Only process creation of new objects
rb_event_flag_t event = rb_tracearg_event_flag(trace_arg);
if (event == RUBY_INTERNAL_EVENT_NEWOBJ)
{
// Don't count allocations of internal IMemo objects
VALUE object = rb_tracearg_object(trace_arg);
if (BUILTIN_TYPE(object) != T_IMEMO)
result += rb_obj_memsize_of(object);
}
}
return result;
}
prof_measurer_t* prof_measurer_memory(bool track_allocations)
{
prof_measurer_t* measure = ALLOC(prof_measurer_t);
measure->mode = MEASURE_MEMORY;
measure->measure = measure_memory;
measure->multiplier = 1;
// Need to track allocations to get RUBY_INTERNAL_EVENT_NEWOBJ event
measure->track_allocations = true;
return measure;
}
void rp_init_measure_memory(void)
{
rb_define_const(mProf, "MEMORY", INT2NUM(MEASURE_MEMORY));
cMeasureMemory = rb_define_class_under(mMeasure, "Allocations", rb_cObject);
}
|
