sabme ua / mruby-mbed

mbed I/F binding for mruby

Dependents:   mruby_mbed_web mirb_mbed

mbed-mruby

How to use

Class

mrbgems/mruby-objectspace/mruby_objectspace.c@0:158c61bb030f, 2015-03-25 (annotated)

Committer:
mzta
Date:
Wed Mar 25 17:36:16 2015 +0000
Revision:
0:158c61bb030f
mirb_mbed initial commit;

Who changed what in which revision?

UserRevisionLine numberNew contents of line
mzta 0:158c61bb030f 1 #include "mruby.h"
mzta 0:158c61bb030f 2 #include "mruby/gc.h"
mzta 0:158c61bb030f 3 #include "mruby/hash.h"
mzta 0:158c61bb030f 4 #include "mruby/class.h"
mzta 0:158c61bb030f 5
mzta 0:158c61bb030f 6 struct os_count_struct {
mzta 0:158c61bb030f 7 mrb_int total;
mzta 0:158c61bb030f 8 mrb_int freed;
mzta 0:158c61bb030f 9 mrb_int counts[MRB_TT_MAXDEFINE+1];
mzta 0:158c61bb030f 10 };
mzta 0:158c61bb030f 11
mzta 0:158c61bb030f 12 static void
mzta 0:158c61bb030f 13 os_count_object_type(mrb_state *mrb, struct RBasic *obj, void *data)
mzta 0:158c61bb030f 14 {
mzta 0:158c61bb030f 15 struct os_count_struct *obj_count;
mzta 0:158c61bb030f 16 obj_count = (struct os_count_struct*)data;
mzta 0:158c61bb030f 17
mzta 0:158c61bb030f 18 obj_count->total++;
mzta 0:158c61bb030f 19
mzta 0:158c61bb030f 20 if (is_dead(mrb, obj)) {
mzta 0:158c61bb030f 21 obj_count->freed++;
mzta 0:158c61bb030f 22 }
mzta 0:158c61bb030f 23 else {
mzta 0:158c61bb030f 24 obj_count->counts[obj->tt]++;
mzta 0:158c61bb030f 25 }
mzta 0:158c61bb030f 26 }
mzta 0:158c61bb030f 27
mzta 0:158c61bb030f 28 /*
mzta 0:158c61bb030f 29 * call-seq:
mzta 0:158c61bb030f 30 * ObjectSpace.count_objects([result_hash]) -> hash
mzta 0:158c61bb030f 31 *
mzta 0:158c61bb030f 32 * Counts objects for each type.
mzta 0:158c61bb030f 33 *
mzta 0:158c61bb030f 34 * It returns a hash, such as:
mzta 0:158c61bb030f 35 * {
mzta 0:158c61bb030f 36 * :TOTAL=>10000,
mzta 0:158c61bb030f 37 * :FREE=>3011,
mzta 0:158c61bb030f 38 * :T_OBJECT=>6,
mzta 0:158c61bb030f 39 * :T_CLASS=>404,
mzta 0:158c61bb030f 40 * # ...
mzta 0:158c61bb030f 41 * }
mzta 0:158c61bb030f 42 *
mzta 0:158c61bb030f 43 * If the optional argument +result_hash+ is given,
mzta 0:158c61bb030f 44 * it is overwritten and returned. This is intended to avoid probe effect.
mzta 0:158c61bb030f 45 *
mzta 0:158c61bb030f 46 */
mzta 0:158c61bb030f 47
mzta 0:158c61bb030f 48 static mrb_value
mzta 0:158c61bb030f 49 os_count_objects(mrb_state *mrb, mrb_value self)
mzta 0:158c61bb030f 50 {
mzta 0:158c61bb030f 51 struct os_count_struct obj_count = { 0 };
mzta 0:158c61bb030f 52 enum mrb_vtype i;
mzta 0:158c61bb030f 53 mrb_value hash;
mzta 0:158c61bb030f 54
mzta 0:158c61bb030f 55 if (mrb_get_args(mrb, "|H", &hash) == 0) {
mzta 0:158c61bb030f 56 hash = mrb_hash_new(mrb);
mzta 0:158c61bb030f 57 }
mzta 0:158c61bb030f 58
mzta 0:158c61bb030f 59 if (!mrb_test(mrb_hash_empty_p(mrb, hash))) {
mzta 0:158c61bb030f 60 mrb_hash_clear(mrb, hash);
mzta 0:158c61bb030f 61 }
mzta 0:158c61bb030f 62
mzta 0:158c61bb030f 63 mrb_objspace_each_objects(mrb, os_count_object_type, &obj_count);
mzta 0:158c61bb030f 64
mzta 0:158c61bb030f 65 mrb_hash_set(mrb, hash, mrb_symbol_value(mrb_intern_lit(mrb, "TOTAL")), mrb_fixnum_value(obj_count.total));
mzta 0:158c61bb030f 66 mrb_hash_set(mrb, hash, mrb_symbol_value(mrb_intern_lit(mrb, "FREE")), mrb_fixnum_value(obj_count.freed));
mzta 0:158c61bb030f 67
mzta 0:158c61bb030f 68 for (i = MRB_TT_FALSE; i < MRB_TT_MAXDEFINE; i++) {
mzta 0:158c61bb030f 69 mrb_value type;
mzta 0:158c61bb030f 70 switch (i) {
mzta 0:158c61bb030f 71 #define COUNT_TYPE(t) case (MRB_T ## t): type = mrb_symbol_value(mrb_intern_lit(mrb, #t)); break;
mzta 0:158c61bb030f 72 COUNT_TYPE(T_FALSE);
mzta 0:158c61bb030f 73 COUNT_TYPE(T_FREE);
mzta 0:158c61bb030f 74 COUNT_TYPE(T_TRUE);
mzta 0:158c61bb030f 75 COUNT_TYPE(T_FIXNUM);
mzta 0:158c61bb030f 76 COUNT_TYPE(T_SYMBOL);
mzta 0:158c61bb030f 77 COUNT_TYPE(T_UNDEF);
mzta 0:158c61bb030f 78 COUNT_TYPE(T_FLOAT);
mzta 0:158c61bb030f 79 COUNT_TYPE(T_CPTR);
mzta 0:158c61bb030f 80 COUNT_TYPE(T_OBJECT);
mzta 0:158c61bb030f 81 COUNT_TYPE(T_CLASS);
mzta 0:158c61bb030f 82 COUNT_TYPE(T_MODULE);
mzta 0:158c61bb030f 83 COUNT_TYPE(T_ICLASS);
mzta 0:158c61bb030f 84 COUNT_TYPE(T_SCLASS);
mzta 0:158c61bb030f 85 COUNT_TYPE(T_PROC);
mzta 0:158c61bb030f 86 COUNT_TYPE(T_ARRAY);
mzta 0:158c61bb030f 87 COUNT_TYPE(T_HASH);
mzta 0:158c61bb030f 88 COUNT_TYPE(T_STRING);
mzta 0:158c61bb030f 89 COUNT_TYPE(T_RANGE);
mzta 0:158c61bb030f 90 COUNT_TYPE(T_EXCEPTION);
mzta 0:158c61bb030f 91 COUNT_TYPE(T_FILE);
mzta 0:158c61bb030f 92 COUNT_TYPE(T_ENV);
mzta 0:158c61bb030f 93 COUNT_TYPE(T_DATA);
mzta 0:158c61bb030f 94 COUNT_TYPE(T_FIBER);
mzta 0:158c61bb030f 95 #undef COUNT_TYPE
mzta 0:158c61bb030f 96 default:
mzta 0:158c61bb030f 97 type = mrb_fixnum_value(i); break;
mzta 0:158c61bb030f 98 }
mzta 0:158c61bb030f 99 if (obj_count.counts[i])
mzta 0:158c61bb030f 100 mrb_hash_set(mrb, hash, type, mrb_fixnum_value(obj_count.counts[i]));
mzta 0:158c61bb030f 101 }
mzta 0:158c61bb030f 102
mzta 0:158c61bb030f 103 return hash;
mzta 0:158c61bb030f 104 }
mzta 0:158c61bb030f 105
mzta 0:158c61bb030f 106 struct os_each_object_data {
mzta 0:158c61bb030f 107 mrb_value block;
mzta 0:158c61bb030f 108 struct RClass *target_module;
mzta 0:158c61bb030f 109 mrb_int count;
mzta 0:158c61bb030f 110 };
mzta 0:158c61bb030f 111
mzta 0:158c61bb030f 112 static void
mzta 0:158c61bb030f 113 os_each_object_cb(mrb_state *mrb, struct RBasic *obj, void *ud)
mzta 0:158c61bb030f 114 {
mzta 0:158c61bb030f 115 struct os_each_object_data *d = (struct os_each_object_data*)ud;
mzta 0:158c61bb030f 116
mzta 0:158c61bb030f 117 /* filter dead objects */
mzta 0:158c61bb030f 118 if (is_dead(mrb, obj)) {
mzta 0:158c61bb030f 119 return;
mzta 0:158c61bb030f 120 }
mzta 0:158c61bb030f 121
mzta 0:158c61bb030f 122 /* filter internal objects */
mzta 0:158c61bb030f 123 switch (obj->tt) {
mzta 0:158c61bb030f 124 case MRB_TT_ENV:
mzta 0:158c61bb030f 125 case MRB_TT_ICLASS:
mzta 0:158c61bb030f 126 return;
mzta 0:158c61bb030f 127 default:
mzta 0:158c61bb030f 128 break;
mzta 0:158c61bb030f 129 }
mzta 0:158c61bb030f 130
mzta 0:158c61bb030f 131 /* filter half baked (or internal) objects */
mzta 0:158c61bb030f 132 if (!obj->c) return;
mzta 0:158c61bb030f 133
mzta 0:158c61bb030f 134 /* filter class kind if target module defined */
mzta 0:158c61bb030f 135 if (d->target_module && !mrb_obj_is_kind_of(mrb, mrb_obj_value(obj), d->target_module)) {
mzta 0:158c61bb030f 136 return;
mzta 0:158c61bb030f 137 }
mzta 0:158c61bb030f 138
mzta 0:158c61bb030f 139 mrb_yield(mrb, d->block, mrb_obj_value(obj));
mzta 0:158c61bb030f 140 ++d->count;
mzta 0:158c61bb030f 141 }
mzta 0:158c61bb030f 142
mzta 0:158c61bb030f 143 /*
mzta 0:158c61bb030f 144 * call-seq:
mzta 0:158c61bb030f 145 * ObjectSpace.each_object([module]) {|obj| ... } -> fixnum
mzta 0:158c61bb030f 146 *
mzta 0:158c61bb030f 147 * Calls the block once for each object in this Ruby process.
mzta 0:158c61bb030f 148 * Returns the number of objects found.
mzta 0:158c61bb030f 149 * If the optional argument +module+ is given,
mzta 0:158c61bb030f 150 * calls the block for only those classes or modules
mzta 0:158c61bb030f 151 * that match (or are a subclass of) +module+.
mzta 0:158c61bb030f 152 *
mzta 0:158c61bb030f 153 * If no block is given, ArgumentError is raised.
mzta 0:158c61bb030f 154 *
mzta 0:158c61bb030f 155 */
mzta 0:158c61bb030f 156
mzta 0:158c61bb030f 157 static mrb_value
mzta 0:158c61bb030f 158 os_each_object(mrb_state *mrb, mrb_value self)
mzta 0:158c61bb030f 159 {
mzta 0:158c61bb030f 160 mrb_value cls = mrb_nil_value();
mzta 0:158c61bb030f 161 struct os_each_object_data d;
mzta 0:158c61bb030f 162 mrb_get_args(mrb, "&|C", &d.block, &cls);
mzta 0:158c61bb030f 163
mzta 0:158c61bb030f 164 if (mrb_nil_p(d.block)) {
mzta 0:158c61bb030f 165 mrb_raise(mrb, E_ARGUMENT_ERROR, "Expected block in ObjectSpace.each_object.");
mzta 0:158c61bb030f 166 }
mzta 0:158c61bb030f 167
mzta 0:158c61bb030f 168 d.target_module = mrb_nil_p(cls) ? NULL : mrb_class_ptr(cls);
mzta 0:158c61bb030f 169 d.count = 0;
mzta 0:158c61bb030f 170 mrb_objspace_each_objects(mrb, os_each_object_cb, &d);
mzta 0:158c61bb030f 171 return mrb_fixnum_value(d.count);
mzta 0:158c61bb030f 172 }
mzta 0:158c61bb030f 173
mzta 0:158c61bb030f 174 void
mzta 0:158c61bb030f 175 mrb_mruby_objectspace_gem_init(mrb_state *mrb)
mzta 0:158c61bb030f 176 {
mzta 0:158c61bb030f 177 struct RClass *os = mrb_define_module(mrb, "ObjectSpace");
mzta 0:158c61bb030f 178 mrb_define_class_method(mrb, os, "count_objects", os_count_objects, MRB_ARGS_OPT(1));
mzta 0:158c61bb030f 179 mrb_define_class_method(mrb, os, "each_object", os_each_object, MRB_ARGS_OPT(1));
mzta 0:158c61bb030f 180 }
mzta 0:158c61bb030f 181
mzta 0:158c61bb030f 182 void
mzta 0:158c61bb030f 183 mrb_mruby_objectspace_gem_final(mrb_state *mrb)
mzta 0:158c61bb030f 184 {
mzta 0:158c61bb030f 185 }
mzta 0:158c61bb030f 186