airbnb-ruby-prof 0.0.1

data/ext/ruby_prof/rp_measure_cpu_time.c

@@ -0,0 +1,134 @@
+/* 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 */
+
+#include "ruby_prof.h"
+
+static VALUE cMeasureCpuTime;
+
+/* The _WIN32 check is needed for msys (and maybe cygwin?) */
+#if defined(_WIN32)
+
+static unsigned long long get_cpu_time()
+{
+    LARGE_INTEGER time;
+    QueryPerformanceCounter(&time);
+    return time.QuadPart;
+}
+
+static unsigned long long get_cpu_frequency()
+{
+    static unsigned long long cpu_frequency;
+
+    if(!cpu_frequency) {
+        LARGE_INTEGER cpu_frequency_struct;
+        QueryPerformanceFrequency(&cpu_frequency_struct);
+        cpu_frequency = cpu_frequency_struct.QuadPart;
+    }
+
+    return cpu_frequency;
+}
+
+static double measure_cpu_time()
+{
+    return ((double)get_cpu_time()) / get_cpu_frequency();
+}
+
+#else
+
+#include <sys/resource.h>
+#include <stdint.h>
+#include <time.h>
+
+static unsigned long long get_cpu_time()
+{
+#if defined(__i386__) || defined(__x86_64__)
+    uint32_t a, d;
+    __asm__ volatile("rdtsc" : "=a" (a), "=d" (d));
+    return ((uint64_t)d << 32) + a;
+#elif defined(__powerpc__) || defined(__ppc__)
+    unsigned long long x, y;
+
+    __asm__ __volatile__ ("\n\
+    1:  mftbu   %1\n\
+        mftb    %L0\n\
+        mftbu   %0\n\
+        cmpw    %0,%1\n\
+        bne-    1b"
+        : "=r" (x), "=r" (y));
+
+    return x;
+#endif
+}
+
+static unsigned long long get_cpu_frequency()
+{
+    static unsigned long long cpu_frequency;
+
+    if(!cpu_frequency) {
+        unsigned long long x, y;
+
+        struct timespec ts;
+        ts.tv_sec = 0;
+        ts.tv_nsec = 500000000;
+        x = get_cpu_time();
+        nanosleep(&ts, NULL);
+        y = get_cpu_time();
+        cpu_frequency = (y - x) * 2;
+    }
+
+    return cpu_frequency;
+}
+
+static double measure_cpu_time()
+{
+    struct rusage rusage;
+    getrusage(RUSAGE_SELF, &rusage);
+
+    double seconds = 0;
+
+    seconds += rusage.ru_utime.tv_sec;
+    seconds += rusage.ru_stime.tv_sec;
+
+    seconds += rusage.ru_utime.tv_usec / 1000000.0;
+    seconds += rusage.ru_stime.tv_usec / 1000000.0;
+
+    return seconds;
+}
+#endif
+
+
+get_measurement prof_measurer_cpu_time()
+{
+    return measure_cpu_time;
+}
+
+/* call-seq:
+   measure -> float
+
+Returns the cpu time.*/
+static VALUE
+prof_measure_cpu_time(VALUE self)
+{
+    return rb_float_new(measure_cpu_time());
+}
+
+/* call-seq:
+   cpu_frequency -> int
+
+Returns the cpu's frequency.  This value is needed when
+RubyProf::measure_mode is set to CPU_TIME. */
+static VALUE
+prof_get_cpu_frequency(VALUE self)
+{
+    return ULL2NUM(get_cpu_frequency());
+}
+
+void rp_init_measure_cpu_time()
+{
+    rb_define_const(mProf, "CPU_TIME", INT2NUM(MEASURE_CPU_TIME));
+    rb_define_const(mProf, "CPU_TIME_ENABLED", Qtrue);
+
+    cMeasureCpuTime = rb_define_class_under(mMeasure, "CpuTime", rb_cObject);
+    rb_define_singleton_method(cMeasureCpuTime, "measure", prof_measure_cpu_time, 0);
+    rb_define_singleton_method(cMeasureCpuTime, "frequency", prof_get_cpu_frequency, 0);
+}

data/ext/ruby_prof/rp_measure_gc_runs.c

@@ -0,0 +1,71 @@
+/* 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 "ruby_prof.h"
+
+static VALUE cMeasureGcRuns;
+
+#if defined(HAVE_RB_GC_COLLECTIONS)
+  VALUE rb_gc_collections(void);
+#endif
+
+#if defined(HAVE_RB_GC_COUNT)
+  size_t rb_gc_count(void);
+#endif
+
+#if defined(HAVE_RB_GC_HEAP_INFO)
+  VALUE rb_gc_heap_info(void);
+#endif
+
+
+static double
+measure_gc_runs()
+{
+#if defined(HAVE_RB_GC_COLLECTIONS)
+#define MEASURE_GC_RUNS_ENABLED Qtrue
+  return NUM2INT(rb_gc_collections());
+
+#elif defined(HAVE_RB_GC_COUNT)
+#define MEASURE_GC_RUNS_ENABLED Qtrue
+  return rb_gc_count();
+
+#elif defined(HAVE_RB_GC_HEAP_INFO)
+#define MEASURE_GC_RUNS_ENABLED Qtrue
+  VALUE h = rb_gc_heap_info();
+  return NUM2UINT(rb_hash_aref(h, rb_str_new2("num_gc_passes")));
+
+#else
+#define MEASURE_GC_RUNS_ENABLED Qfalse
+  return 0;
+#endif
+}
+
+get_measurement prof_measurer_gc_runs()
+{
+  return measure_gc_runs;
+}
+
+/* call-seq:
+   measure -> int
+
+Returns the number of GC runs.*/
+static VALUE
+prof_measure_gc_runs(VALUE self)
+{
+#if defined(HAVE_LONG_LONG)
+    return ULL2NUM(measure_gc_runs());
+#else
+    return ULONG2NUM(measure_gc_runs());
+#endif
+}
+
+void rp_init_measure_gc_runs()
+{
+    rb_define_const(mProf, "GC_RUNS", INT2NUM(MEASURE_GC_RUNS));
+    rb_define_const(mProf, "GC_RUNS_ENABLED", MEASURE_GC_RUNS_ENABLED);
+    
+    cMeasureGcRuns = rb_define_class_under(mMeasure, "GcRuns", rb_cObject);
+    rb_define_singleton_method(cMeasureGcRuns, "measure", prof_measure_gc_runs, 0);
+}

data/ext/ruby_prof/rp_measure_gc_time.c

@@ -0,0 +1,58 @@
+/* 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 "ruby_prof.h"
+
+static VALUE cMeasureGcTimes;
+
+#if defined(HAVE_RB_GC_TIME)
+  VALUE rb_gc_time();
+#endif
+
+static double
+measure_gc_time()
+{
+#if defined(HAVE_RB_GC_TOTAL_TIME)
+#define MEASURE_GC_TIME_ENABLED Qtrue
+    return rb_gc_total_time();
+
+#elif defined(HAVE_RB_GC_TIME)
+#define MEASURE_GC_TIME_ENABLED Qtrue
+    const double conversion = 1000000.0;
+#if HAVE_LONG_LONG
+    return NUM2LL(rb_gc_time()) / conversion;
+#else
+    return NUM2LONG(rb_gc_time()) / conversion;
+#endif
+
+#else
+#define MEASURE_GC_TIME_ENABLED Qfalse
+    return 0.0;
+#endif
+}
+
+get_measurement prof_measurer_gc_time()
+{
+  return measure_gc_time;
+}
+
+/* call-seq:
+   measure -> float
+
+Returns the time spent performing GC.*/
+static VALUE
+prof_measure_gc_time(VALUE self)
+{
+    return rb_float_new(measure_gc_time());
+}
+
+void rp_init_measure_gc_time()
+{
+    rb_define_const(mProf, "GC_TIME", INT2NUM(MEASURE_GC_TIME));
+    rb_define_const(mProf, "GC_TIME_ENABLED", MEASURE_GC_TIME_ENABLED);
+
+    cMeasureGcTimes = rb_define_class_under(mMeasure, "GcTime", rb_cObject);
+    rb_define_singleton_method(cMeasureGcTimes, "measure", prof_measure_gc_time, 0);
+}

data/ext/ruby_prof/rp_measure_memory.c

@@ -0,0 +1,75 @@
+/* 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 "ruby_prof.h"
+
+static VALUE cMeasureMemory;
+
+
+#if defined(HAVE_RB_GC_ALLOCATED_SIZE)
+  VALUE rb_gc_allocated_size();
+#endif
+
+#if defined(HAVE_RB_GC_MALLOC_ALLOCATED_SIZE)
+  size_t rb_gc_malloc_allocated_size();
+#endif
+
+#if defined(HAVE_RB_HEAP_TOTAL_MEM)
+  //FIXME: did not find the patch to check prototype, assuming it to return size_t
+  size_t rb_heap_total_mem();
+#endif
+
+static double
+measure_memory()
+{
+#if defined(HAVE_RB_GC_ALLOCATED_SIZE)
+#define MEASURE_MEMORY_ENABLED Qtrue
+#if defined(HAVE_LONG_LONG)
+    return NUM2LL(rb_gc_allocated_size()) / 1024.0;
+#else
+    return NUM2ULONG(rb_gc_allocated_size()) / 1024.0;
+#endif
+
+#elif defined(HAVE_RB_GC_MALLOC_ALLOCATED_SIZE)
+#define MEASURE_MEMORY_ENABLED Qtrue
+    return rb_gc_malloc_allocated_size() / 1024.0;
+
+#elif defined(HAVE_RB_GC_TOTAL_MALLOCED_BYTES)
+#define MEASURE_MEMORY_ENABLED Qtrue
+    return rb_gc_total_malloced_bytes() / 1024.0;
+
+#elif defined(HAVE_RB_HEAP_TOTAL_MEM)
+#define MEASURE_MEMORY_ENABLED Qtrue
+    return rb_heap_total_mem() / 1024.0;
+
+#else
+#define MEASURE_MEMORY_ENABLED Qfalse
+    return 0;
+#endif
+}
+
+get_measurement prof_measurer_memory()
+{
+  return measure_memory;
+}
+
+/* call-seq:
+   measure_process_time -> float
+
+Returns the process time.*/
+static VALUE
+prof_measure_memory(VALUE self)
+{
+    return rb_float_new(measure_memory());
+}
+
+void rp_init_measure_memory()
+{
+    rb_define_const(mProf, "MEMORY", INT2NUM(MEASURE_MEMORY));
+    rb_define_const(mProf, "MEMORY_ENABLED", MEASURE_MEMORY_ENABLED);
+
+    cMeasureMemory = rb_define_class_under(mMeasure, "Memory", rb_cObject);
+    rb_define_singleton_method(cMeasureMemory, "measure", prof_measure_memory, 0);
+}

data/ext/ruby_prof/rp_measure_process_time.c

@@ -0,0 +1,69 @@
+/* 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 */
+
+#include "ruby_prof.h"
+#include <time.h>
+
+static VALUE cMeasureProcessTime;
+
+static double
+measure_process_time()
+{
+#if defined(__linux__)
+    struct timespec clock;
+    clock_gettime(CLOCK_PROCESS_CPUTIME_ID , &clock);
+    return clock.tv_sec + (clock.tv_nsec/1000000000.0);
+#elif defined(_win32)
+	FILETIME createTime;
+	FILETIME exitTime;
+	FILETIME sysTime;
+	FILETIME cpuTime;
+
+	ULARGE_INTEGER sysTimeInt;
+	ULARGE_INTEGER cpuTimeInt;
+	ULONGLONG totalTime;
+
+	GetProcessTimes(GetCurrentProcess(), &createTime, &exitTime, &sysTime, &cpuTime); 
+
+	/* Doing this based on MSFT's recommendation in the FILETIME structure documentation at
+	  http://msdn.microsoft.com/en-us/library/ms724284%28VS.85%29.aspx*/
+
+	sysTimeInt.LowPart = sysTime.dwLowDateTime;
+	sysTimeInt.HighPart = sysTime.dwHighDateTime;
+	cpuTimeInt.LowPart = cpuTime.dwLowDateTime;
+	cpuTimeInt.HighPart = cpuTime.dwHighDateTime;
+
+	totalTime = sysTimeInt.QuadPart + cpuTimeInt.QuadPart;
+
+	// Times are in 100-nanosecond time units.  So instead of 10-9 use 10-7
+	return totalTime / 10000000.0;
+#else
+    return ((double)clock()) / CLOCKS_PER_SEC;
+#endif
+}
+
+/* call-seq:
+   measure_process_time -> float
+
+Returns the process time.*/
+static VALUE
+prof_measure_process_time(VALUE self)
+{
+    return rb_float_new(measure_process_time());
+}
+
+get_measurement prof_measurer_process_time()
+{
+  return measure_process_time;
+}
+
+
+void rp_init_measure_process_time()
+{
+    rb_define_const(mProf, "CLOCKS_PER_SEC", INT2NUM(CLOCKS_PER_SEC));
+    rb_define_const(mProf, "PROCESS_TIME", INT2NUM(MEASURE_PROCESS_TIME));
+	rb_define_const(mProf, "PROCESS_TIME_ENABLED", Qtrue);
+
+    cMeasureProcessTime = rb_define_class_under(mMeasure, "ProcessTime", rb_cObject);
+    rb_define_singleton_method(cMeasureProcessTime, "measure", prof_measure_process_time, 0);
+}

data/ext/ruby_prof/rp_measure_wall_time.c

@@ -0,0 +1,43 @@
+/* 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 "ruby_prof.h"
+#if HAVE_GETTIMEOFDAY && !defined(_WIN32)
+#include <sys/time.h>
+#endif
+
+static VALUE cMeasureWallTime;
+
+static double
+measure_wall_time()
+{
+    struct timeval tv;
+    gettimeofday(&tv, NULL);
+    return tv.tv_sec + (tv.tv_usec/1000000.0);
+}
+
+get_measurement prof_measurer_wall_time()
+{
+  return measure_wall_time;
+}
+
+/* Document-method: prof_measure_wall_time
+   call-seq:
+     measure_wall_time -> float
+
+Returns the wall time.*/
+static VALUE
+prof_measure_wall_time(VALUE self)
+{
+    return rb_float_new(measure_wall_time());
+}
+
+void rp_init_measure_wall_time()
+{
+    rb_define_const(mProf, "WALL_TIME", INT2NUM(MEASURE_WALL_TIME));
+    rb_define_const(mProf, "WALL_TIME_ENABLED", Qtrue);
+
+    cMeasureWallTime = rb_define_class_under(mMeasure, "WallTime", rb_cObject);
+    rb_define_singleton_method(cMeasureWallTime, "measure", prof_measure_wall_time, 0);
+}