sabme ua / mruby-mbed

mbed I/F binding for mruby

Dependents:   mruby_mbed_web mirb_mbed

mbed-mruby

How to use

Class

mrbgems/mruby-random/random.c@1:8ccd1d494a4b, 2015-04-13 (annotated)

Committer:
mzta
Date:
Mon Apr 13 05:20:15 2015 +0000
Revision:
1:8ccd1d494a4b
Parent:
0:158c61bb030f 
- code refactoring.; - add SPI, SPISlave, I2C class to mruby-mbed (Incomplete).

Who changed what in which revision?

UserRevisionLine numberNew contents of line
mzta 0:158c61bb030f 1 /*
mzta 0:158c61bb030f 2 ** random.c - Random module
mzta 0:158c61bb030f 3 **
mzta 0:158c61bb030f 4 ** See Copyright Notice in mruby.h
mzta 0:158c61bb030f 5 */
mzta 0:158c61bb030f 6
mzta 0:158c61bb030f 7 #include "mruby.h"
mzta 0:158c61bb030f 8 #include "mruby/variable.h"
mzta 0:158c61bb030f 9 #include "mruby/class.h"
mzta 0:158c61bb030f 10 #include "mruby/data.h"
mzta 0:158c61bb030f 11 #include "mruby/array.h"
mzta 0:158c61bb030f 12 #include "mt19937ar.h"
mzta 0:158c61bb030f 13
mzta 0:158c61bb030f 14 #include <time.h>
mzta 0:158c61bb030f 15
mzta 0:158c61bb030f 16 static char const MT_STATE_KEY[] = "$mrb_i_mt_state";
mzta 0:158c61bb030f 17
mzta 0:158c61bb030f 18 static const struct mrb_data_type mt_state_type = {
mzta 0:158c61bb030f 19 MT_STATE_KEY, mrb_free,
mzta 0:158c61bb030f 20 };
mzta 0:158c61bb030f 21
mzta 0:158c61bb030f 22 static mrb_value mrb_random_rand(mrb_state *mrb, mrb_value self);
mzta 0:158c61bb030f 23 static mrb_value mrb_random_srand(mrb_state *mrb, mrb_value self);
mzta 0:158c61bb030f 24
mzta 0:158c61bb030f 25 static void
mzta 0:158c61bb030f 26 mt_srand(mt_state *t, unsigned long seed)
mzta 0:158c61bb030f 27 {
mzta 0:158c61bb030f 28 mrb_random_init_genrand(t, seed);
mzta 0:158c61bb030f 29 }
mzta 0:158c61bb030f 30
mzta 0:158c61bb030f 31 static unsigned long
mzta 0:158c61bb030f 32 mt_rand(mt_state *t)
mzta 0:158c61bb030f 33 {
mzta 0:158c61bb030f 34 return mrb_random_genrand_int32(t);
mzta 0:158c61bb030f 35 }
mzta 0:158c61bb030f 36
mzta 0:158c61bb030f 37 static double
mzta 0:158c61bb030f 38 mt_rand_real(mt_state *t)
mzta 0:158c61bb030f 39 {
mzta 0:158c61bb030f 40 return mrb_random_genrand_real1(t);
mzta 0:158c61bb030f 41 }
mzta 0:158c61bb030f 42
mzta 0:158c61bb030f 43 static mrb_value
mzta 0:158c61bb030f 44 mrb_random_mt_srand(mrb_state *mrb, mt_state *t, mrb_value seed)
mzta 0:158c61bb030f 45 {
mzta 0:158c61bb030f 46 if (mrb_nil_p(seed)) {
mzta 0:158c61bb030f 47 seed = mrb_fixnum_value((mrb_int)(time(NULL) + mt_rand(t)));
mzta 0:158c61bb030f 48 if (mrb_fixnum(seed) < 0) {
mzta 0:158c61bb030f 49 seed = mrb_fixnum_value(0 - mrb_fixnum(seed));
mzta 0:158c61bb030f 50 }
mzta 0:158c61bb030f 51 }
mzta 0:158c61bb030f 52
mzta 0:158c61bb030f 53 mt_srand(t, (unsigned) mrb_fixnum(seed));
mzta 0:158c61bb030f 54
mzta 0:158c61bb030f 55 return seed;
mzta 0:158c61bb030f 56 }
mzta 0:158c61bb030f 57
mzta 0:158c61bb030f 58 static mrb_value
mzta 0:158c61bb030f 59 mrb_random_mt_rand(mrb_state *mrb, mt_state *t, mrb_value max)
mzta 0:158c61bb030f 60 {
mzta 0:158c61bb030f 61 mrb_value value;
mzta 0:158c61bb030f 62
mzta 0:158c61bb030f 63 if (mrb_fixnum(max) == 0) {
mzta 0:158c61bb030f 64 value = mrb_float_value(mrb, mt_rand_real(t));
mzta 0:158c61bb030f 65 }
mzta 0:158c61bb030f 66 else {
mzta 0:158c61bb030f 67 value = mrb_fixnum_value(mt_rand(t) % mrb_fixnum(max));
mzta 0:158c61bb030f 68 }
mzta 0:158c61bb030f 69
mzta 0:158c61bb030f 70 return value;
mzta 0:158c61bb030f 71 }
mzta 0:158c61bb030f 72
mzta 0:158c61bb030f 73 static mrb_value
mzta 0:158c61bb030f 74 get_opt(mrb_state* mrb)
mzta 0:158c61bb030f 75 {
mzta 0:158c61bb030f 76 mrb_value arg;
mzta 0:158c61bb030f 77
mzta 0:158c61bb030f 78 arg = mrb_nil_value();
mzta 0:158c61bb030f 79 mrb_get_args(mrb, "|o", &arg);
mzta 0:158c61bb030f 80
mzta 0:158c61bb030f 81 if (!mrb_nil_p(arg)) {
mzta 0:158c61bb030f 82 if (!mrb_fixnum_p(arg)) {
mzta 0:158c61bb030f 83 mrb_raise(mrb, E_ARGUMENT_ERROR, "invalid argument type");
mzta 0:158c61bb030f 84 }
mzta 0:158c61bb030f 85 arg = mrb_check_convert_type(mrb, arg, MRB_TT_FIXNUM, "Fixnum", "to_int");
mzta 0:158c61bb030f 86 if (mrb_fixnum(arg) < 0) {
mzta 0:158c61bb030f 87 arg = mrb_fixnum_value(0 - mrb_fixnum(arg));
mzta 0:158c61bb030f 88 }
mzta 0:158c61bb030f 89 }
mzta 0:158c61bb030f 90 return arg;
mzta 0:158c61bb030f 91 }
mzta 0:158c61bb030f 92
mzta 0:158c61bb030f 93 static mrb_value
mzta 0:158c61bb030f 94 get_random(mrb_state *mrb) {
mzta 0:158c61bb030f 95 return mrb_const_get(mrb,
mzta 0:158c61bb030f 96 mrb_obj_value(mrb_class_get(mrb, "Random")),
mzta 0:158c61bb030f 97 mrb_intern_lit(mrb, "DEFAULT"));
mzta 0:158c61bb030f 98 }
mzta 0:158c61bb030f 99
mzta 0:158c61bb030f 100 static mt_state *
mzta 0:158c61bb030f 101 get_random_state(mrb_state *mrb)
mzta 0:158c61bb030f 102 {
mzta 0:158c61bb030f 103 mrb_value random_val = get_random(mrb);
mzta 0:158c61bb030f 104 return DATA_GET_PTR(mrb, random_val, &mt_state_type, mt_state);
mzta 0:158c61bb030f 105 }
mzta 0:158c61bb030f 106
mzta 0:158c61bb030f 107 static mrb_value
mzta 0:158c61bb030f 108 mrb_random_g_rand(mrb_state *mrb, mrb_value self)
mzta 0:158c61bb030f 109 {
mzta 0:158c61bb030f 110 mrb_value random = get_random(mrb);
mzta 0:158c61bb030f 111 return mrb_random_rand(mrb, random);
mzta 0:158c61bb030f 112 }
mzta 0:158c61bb030f 113
mzta 0:158c61bb030f 114 static mrb_value
mzta 0:158c61bb030f 115 mrb_random_g_srand(mrb_state *mrb, mrb_value self)
mzta 0:158c61bb030f 116 {
mzta 0:158c61bb030f 117 mrb_value random = get_random(mrb);
mzta 0:158c61bb030f 118 return mrb_random_srand(mrb, random);
mzta 0:158c61bb030f 119 }
mzta 0:158c61bb030f 120
mzta 0:158c61bb030f 121 static mrb_value
mzta 0:158c61bb030f 122 mrb_random_init(mrb_state *mrb, mrb_value self)
mzta 0:158c61bb030f 123 {
mzta 0:158c61bb030f 124 mrb_value seed;
mzta 0:158c61bb030f 125 mt_state *t;
mzta 0:158c61bb030f 126
mzta 0:158c61bb030f 127 /* avoid memory leaks */
mzta 0:158c61bb030f 128 t = (mt_state*)DATA_PTR(self);
mzta 0:158c61bb030f 129 if (t) {
mzta 0:158c61bb030f 130 mrb_free(mrb, t);
mzta 0:158c61bb030f 131 }
mzta 0:158c61bb030f 132 mrb_data_init(self, NULL, &mt_state_type);
mzta 0:158c61bb030f 133
mzta 0:158c61bb030f 134 t = (mt_state *)mrb_malloc(mrb, sizeof(mt_state));
mzta 0:158c61bb030f 135 t->mti = N + 1;
mzta 0:158c61bb030f 136
mzta 0:158c61bb030f 137 seed = get_opt(mrb);
mzta 0:158c61bb030f 138 seed = mrb_random_mt_srand(mrb, t, seed);
mzta 0:158c61bb030f 139 if (mrb_nil_p(seed)) {
mzta 0:158c61bb030f 140 t->has_seed = FALSE;
mzta 0:158c61bb030f 141 }
mzta 0:158c61bb030f 142 else {
mzta 0:158c61bb030f 143 mrb_assert(mrb_fixnum_p(seed));
mzta 0:158c61bb030f 144 t->has_seed = TRUE;
mzta 0:158c61bb030f 145 t->seed = mrb_fixnum(seed);
mzta 0:158c61bb030f 146 }
mzta 0:158c61bb030f 147
mzta 0:158c61bb030f 148 mrb_data_init(self, t, &mt_state_type);
mzta 0:158c61bb030f 149
mzta 0:158c61bb030f 150 return self;
mzta 0:158c61bb030f 151 }
mzta 0:158c61bb030f 152
mzta 0:158c61bb030f 153 static void
mzta 0:158c61bb030f 154 mrb_random_rand_seed(mrb_state *mrb, mt_state *t)
mzta 0:158c61bb030f 155 {
mzta 0:158c61bb030f 156 if (!t->has_seed) {
mzta 0:158c61bb030f 157 mrb_random_mt_srand(mrb, t, mrb_nil_value());
mzta 0:158c61bb030f 158 }
mzta 0:158c61bb030f 159 }
mzta 0:158c61bb030f 160
mzta 0:158c61bb030f 161 static mrb_value
mzta 0:158c61bb030f 162 mrb_random_rand(mrb_state *mrb, mrb_value self)
mzta 0:158c61bb030f 163 {
mzta 0:158c61bb030f 164 mrb_value max;
mzta 0:158c61bb030f 165 mt_state *t = DATA_GET_PTR(mrb, self, &mt_state_type, mt_state);
mzta 0:158c61bb030f 166
mzta 0:158c61bb030f 167 max = get_opt(mrb);
mzta 0:158c61bb030f 168 mrb_random_rand_seed(mrb, t);
mzta 0:158c61bb030f 169 return mrb_random_mt_rand(mrb, t, max);
mzta 0:158c61bb030f 170 }
mzta 0:158c61bb030f 171
mzta 0:158c61bb030f 172 static mrb_value
mzta 0:158c61bb030f 173 mrb_random_srand(mrb_state *mrb, mrb_value self)
mzta 0:158c61bb030f 174 {
mzta 0:158c61bb030f 175 mrb_value seed;
mzta 0:158c61bb030f 176 mrb_value old_seed;
mzta 0:158c61bb030f 177 mt_state *t = DATA_GET_PTR(mrb, self, &mt_state_type, mt_state);
mzta 0:158c61bb030f 178
mzta 0:158c61bb030f 179 seed = get_opt(mrb);
mzta 0:158c61bb030f 180 seed = mrb_random_mt_srand(mrb, t, seed);
mzta 0:158c61bb030f 181 old_seed = t->has_seed? mrb_fixnum_value(t->seed) : mrb_nil_value();
mzta 0:158c61bb030f 182 if (mrb_nil_p(seed)) {
mzta 0:158c61bb030f 183 t->has_seed = FALSE;
mzta 0:158c61bb030f 184 }
mzta 0:158c61bb030f 185 else {
mzta 0:158c61bb030f 186 mrb_assert(mrb_fixnum_p(seed));
mzta 0:158c61bb030f 187 t->has_seed = TRUE;
mzta 0:158c61bb030f 188 t->seed = mrb_fixnum(seed);
mzta 0:158c61bb030f 189 }
mzta 0:158c61bb030f 190
mzta 0:158c61bb030f 191 return old_seed;
mzta 0:158c61bb030f 192 }
mzta 0:158c61bb030f 193
mzta 0:158c61bb030f 194 /*
mzta 0:158c61bb030f 195 * call-seq:
mzta 0:158c61bb030f 196 * ary.shuffle! -> ary
mzta 0:158c61bb030f 197 *
mzta 0:158c61bb030f 198 * Shuffles elements in self in place.
mzta 0:158c61bb030f 199 */
mzta 0:158c61bb030f 200
mzta 0:158c61bb030f 201 static mrb_value
mzta 0:158c61bb030f 202 mrb_ary_shuffle_bang(mrb_state *mrb, mrb_value ary)
mzta 0:158c61bb030f 203 {
mzta 0:158c61bb030f 204 mrb_int i;
mzta 0:158c61bb030f 205 mt_state *random = NULL;
mzta 0:158c61bb030f 206
mzta 0:158c61bb030f 207 if (RARRAY_LEN(ary) > 1) {
mzta 0:158c61bb030f 208 mrb_get_args(mrb, "|d", &random, &mt_state_type);
mzta 0:158c61bb030f 209
mzta 0:158c61bb030f 210 if (random == NULL) {
mzta 0:158c61bb030f 211 random = get_random_state(mrb);
mzta 0:158c61bb030f 212 }
mzta 0:158c61bb030f 213 mrb_random_rand_seed(mrb, random);
mzta 0:158c61bb030f 214
mzta 0:158c61bb030f 215 mrb_ary_modify(mrb, mrb_ary_ptr(ary));
mzta 0:158c61bb030f 216
mzta 0:158c61bb030f 217 for (i = RARRAY_LEN(ary) - 1; i > 0; i--) {
mzta 0:158c61bb030f 218 mrb_int j;
mzta 0:158c61bb030f 219 mrb_value tmp;
mzta 0:158c61bb030f 220
mzta 0:158c61bb030f 221 j = mrb_fixnum(mrb_random_mt_rand(mrb, random, mrb_fixnum_value(RARRAY_LEN(ary))));
mzta 0:158c61bb030f 222
mzta 0:158c61bb030f 223 tmp = RARRAY_PTR(ary)[i];
mzta 0:158c61bb030f 224 mrb_ary_ptr(ary)->ptr[i] = RARRAY_PTR(ary)[j];
mzta 0:158c61bb030f 225 mrb_ary_ptr(ary)->ptr[j] = tmp;
mzta 0:158c61bb030f 226 }
mzta 0:158c61bb030f 227 }
mzta 0:158c61bb030f 228
mzta 0:158c61bb030f 229 return ary;
mzta 0:158c61bb030f 230 }
mzta 0:158c61bb030f 231
mzta 0:158c61bb030f 232 /*
mzta 0:158c61bb030f 233 * call-seq:
mzta 0:158c61bb030f 234 * ary.shuffle -> new_ary
mzta 0:158c61bb030f 235 *
mzta 0:158c61bb030f 236 * Returns a new array with elements of self shuffled.
mzta 0:158c61bb030f 237 */
mzta 0:158c61bb030f 238
mzta 0:158c61bb030f 239 static mrb_value
mzta 0:158c61bb030f 240 mrb_ary_shuffle(mrb_state *mrb, mrb_value ary)
mzta 0:158c61bb030f 241 {
mzta 0:158c61bb030f 242 mrb_value new_ary = mrb_ary_new_from_values(mrb, RARRAY_LEN(ary), RARRAY_PTR(ary));
mzta 0:158c61bb030f 243 mrb_ary_shuffle_bang(mrb, new_ary);
mzta 0:158c61bb030f 244
mzta 0:158c61bb030f 245 return new_ary;
mzta 0:158c61bb030f 246 }
mzta 0:158c61bb030f 247
mzta 0:158c61bb030f 248 /*
mzta 0:158c61bb030f 249 * call-seq:
mzta 0:158c61bb030f 250 * ary.sample -> obj
mzta 0:158c61bb030f 251 * ary.sample(n) -> new_ary
mzta 0:158c61bb030f 252 *
mzta 0:158c61bb030f 253 * Choose a random element or +n+ random elements from the array.
mzta 0:158c61bb030f 254 *
mzta 0:158c61bb030f 255 * The elements are chosen by using random and unique indices into the array
mzta 0:158c61bb030f 256 * in order to ensure that an element doesn't repeat itself unless the array
mzta 0:158c61bb030f 257 * already contained duplicate elements.
mzta 0:158c61bb030f 258 *
mzta 0:158c61bb030f 259 * If the array is empty the first form returns +nil+ and the second form
mzta 0:158c61bb030f 260 * returns an empty array.
mzta 0:158c61bb030f 261 */
mzta 0:158c61bb030f 262
mzta 0:158c61bb030f 263 static mrb_value
mzta 0:158c61bb030f 264 mrb_ary_sample(mrb_state *mrb, mrb_value ary)
mzta 0:158c61bb030f 265 {
mzta 0:158c61bb030f 266 mrb_int n = 0;
mzta 0:158c61bb030f 267 mrb_bool given;
mzta 0:158c61bb030f 268 mt_state *random = NULL;
mzta 0:158c61bb030f 269 mrb_int len = RARRAY_LEN(ary);
mzta 0:158c61bb030f 270
mzta 0:158c61bb030f 271 mrb_get_args(mrb, "|i?d", &n, &given, &random, &mt_state_type);
mzta 0:158c61bb030f 272 if (random == NULL) {
mzta 0:158c61bb030f 273 random = get_random_state(mrb);
mzta 0:158c61bb030f 274 }
mzta 0:158c61bb030f 275 mrb_random_rand_seed(mrb, random);
mzta 0:158c61bb030f 276 mt_rand(random);
mzta 0:158c61bb030f 277 if (!given) { /* pick one element */
mzta 0:158c61bb030f 278 switch (len) {
mzta 0:158c61bb030f 279 case 0:
mzta 0:158c61bb030f 280 return mrb_nil_value();
mzta 0:158c61bb030f 281 case 1:
mzta 0:158c61bb030f 282 return RARRAY_PTR(ary)[0];
mzta 0:158c61bb030f 283 default:
mzta 0:158c61bb030f 284 return RARRAY_PTR(ary)[mt_rand(random) % len];
mzta 0:158c61bb030f 285 }
mzta 0:158c61bb030f 286 }
mzta 0:158c61bb030f 287 else {
mzta 0:158c61bb030f 288 mrb_value result;
mzta 0:158c61bb030f 289 mrb_int i, j;
mzta 0:158c61bb030f 290
mzta 0:158c61bb030f 291 if (n < 0) mrb_raise(mrb, E_ARGUMENT_ERROR, "negative sample number");
mzta 0:158c61bb030f 292 if (n > len) n = len;
mzta 0:158c61bb030f 293 result = mrb_ary_new_capa(mrb, n);
mzta 0:158c61bb030f 294 for (i=0; i<n; i++) {
mzta 0:158c61bb030f 295 mrb_int r;
mzta 0:158c61bb030f 296
mzta 0:158c61bb030f 297 for (;;) {
mzta 0:158c61bb030f 298 retry:
mzta 0:158c61bb030f 299 r = mt_rand(random) % len;
mzta 0:158c61bb030f 300
mzta 0:158c61bb030f 301 for (j=0; j<i; j++) {
mzta 0:158c61bb030f 302 if (mrb_fixnum(RARRAY_PTR(result)[j]) == r) {
mzta 0:158c61bb030f 303 goto retry; /* retry if duplicate */
mzta 0:158c61bb030f 304 }
mzta 0:158c61bb030f 305 }
mzta 0:158c61bb030f 306 break;
mzta 0:158c61bb030f 307 }
mzta 0:158c61bb030f 308 mrb_ary_push(mrb, result, mrb_fixnum_value(r));
mzta 0:158c61bb030f 309 }
mzta 0:158c61bb030f 310 for (i=0; i<n; i++) {
mzta 0:158c61bb030f 311 mrb_ary_set(mrb, result, i, RARRAY_PTR(ary)[mrb_fixnum(RARRAY_PTR(result)[i])]);
mzta 0:158c61bb030f 312 }
mzta 0:158c61bb030f 313 return result;
mzta 0:158c61bb030f 314 }
mzta 0:158c61bb030f 315 }
mzta 0:158c61bb030f 316
mzta 0:158c61bb030f 317
mzta 0:158c61bb030f 318 void mrb_mruby_random_gem_init(mrb_state *mrb)
mzta 0:158c61bb030f 319 {
mzta 0:158c61bb030f 320 struct RClass *random;
mzta 0:158c61bb030f 321 struct RClass *array = mrb->array_class;
mzta 0:158c61bb030f 322
mzta 0:158c61bb030f 323 mrb_define_method(mrb, mrb->kernel_module, "rand", mrb_random_g_rand, MRB_ARGS_OPT(1));
mzta 0:158c61bb030f 324 mrb_define_method(mrb, mrb->kernel_module, "srand", mrb_random_g_srand, MRB_ARGS_OPT(1));
mzta 0:158c61bb030f 325
mzta 0:158c61bb030f 326 random = mrb_define_class(mrb, "Random", mrb->object_class);
mzta 0:158c61bb030f 327 MRB_SET_INSTANCE_TT(random, MRB_TT_DATA);
mzta 0:158c61bb030f 328 mrb_define_class_method(mrb, random, "rand", mrb_random_g_rand, MRB_ARGS_OPT(1));
mzta 0:158c61bb030f 329 mrb_define_class_method(mrb, random, "srand", mrb_random_g_srand, MRB_ARGS_OPT(1));
mzta 0:158c61bb030f 330
mzta 0:158c61bb030f 331 mrb_define_method(mrb, random, "initialize", mrb_random_init, MRB_ARGS_OPT(1));
mzta 0:158c61bb030f 332 mrb_define_method(mrb, random, "rand", mrb_random_rand, MRB_ARGS_OPT(1));
mzta 0:158c61bb030f 333 mrb_define_method(mrb, random, "srand", mrb_random_srand, MRB_ARGS_OPT(1));
mzta 0:158c61bb030f 334
mzta 0:158c61bb030f 335 mrb_define_method(mrb, array, "shuffle", mrb_ary_shuffle, MRB_ARGS_OPT(1));
mzta 0:158c61bb030f 336 mrb_define_method(mrb, array, "shuffle!", mrb_ary_shuffle_bang, MRB_ARGS_OPT(1));
mzta 0:158c61bb030f 337 mrb_define_method(mrb, array, "sample", mrb_ary_sample, MRB_ARGS_OPT(2));
mzta 0:158c61bb030f 338
mzta 0:158c61bb030f 339 mrb_const_set(mrb, mrb_obj_value(random), mrb_intern_lit(mrb, "DEFAULT"),
mzta 0:158c61bb030f 340 mrb_obj_new(mrb, random, 0, NULL));
mzta 0:158c61bb030f 341 }
mzta 0:158c61bb030f 342
mzta 0:158c61bb030f 343 void mrb_mruby_random_gem_final(mrb_state *mrb)
mzta 0:158c61bb030f 344 {
mzta 0:158c61bb030f 345 }
mzta 0:158c61bb030f 346