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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
/************************************************
random.c -
$Author: matz $
$Date: 1994/08/12 04:47:48 $
created at: Fri Dec 24 16:39:21 JST 1993
Copyright (C) 1994 Yukihiro Matsumoto
************************************************/
#include "ruby.h"
static int first = 1;
static char state[256];
static VALUE
Fsrand(obj, args)
VALUE obj, args;
{
int seed, old;
#ifdef HAVE_RANDOM
static int saved_seed;
#endif
if (rb_scan_args(args, "01", &seed) == 0) {
seed = time(0);
}
else {
seed = NUM2INT(seed);
}
#ifdef HAVE_RANDOM
if (first == 1) {
initstate(1, state, sizeof state);
first = 0;
}
else {
setstate(state);
}
srandom(seed);
old = saved_seed;
saved_seed = seed;
return int2inum(old);
#else
old = srand(seed);
return int2inum(old);
#endif
}
static VALUE
Frand(obj, max)
VALUE obj, max;
{
int val;
#ifdef HAVE_RANDOM
if (first == 1) {
initstate(1, state, sizeof state);
first = 0;
}
val = random() % NUM2INT(max);
#else
val = rand() % NUM2INT(max);
#endif
if (val < 0) val = -val;
return int2inum(val);
}
Init_Random()
{
extern VALUE C_Kernel;
rb_define_method(C_Kernel, "srand", Fsrand, -2);
rb_define_method(C_Kernel, "rand", Frand, 1);
}
|