gmp 0.1.7

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,144 @@
1
+ #ifndef _GMPF_H_
2
+ #define _GMPF_H_
3
+
4
+ /*
5
+ * gmpf.h
6
+ *
7
+ * This file contains GMP::F stuff.
8
+ */
9
+
10
+ #include <ruby_gmp.h>
11
+
12
+ #ifdef MPFR
13
+ #define MPFR_SINGLE_FUNCTION(name) \
14
+ static VALUE r_gmpfr_##name(VALUE self) \
15
+ { \
16
+ MP_FLOAT *self_val, *res_val; \
17
+ unsigned long prec; \
18
+ VALUE res; \
19
+ \
20
+ mpf_get_struct_prec(self, self_val, prec); \
21
+ mpf_make_struct_init(res, res_val, prec); \
22
+ mpfr_##name(res_val, self_val, __gmp_default_rounding_mode); \
23
+ \
24
+ return res; \
25
+ }
26
+
27
+ MPFR_SINGLE_FUNCTION(log)
28
+ MPFR_SINGLE_FUNCTION(exp)
29
+ MPFR_SINGLE_FUNCTION(sqrt)
30
+ MPFR_SINGLE_FUNCTION(cos)
31
+ MPFR_SINGLE_FUNCTION(sin)
32
+ MPFR_SINGLE_FUNCTION(tan)
33
+ MPFR_SINGLE_FUNCTION(acos)
34
+ MPFR_SINGLE_FUNCTION(asin)
35
+ MPFR_SINGLE_FUNCTION(atan)
36
+ MPFR_SINGLE_FUNCTION(cosh)
37
+ MPFR_SINGLE_FUNCTION(sinh)
38
+ MPFR_SINGLE_FUNCTION(tanh)
39
+ MPFR_SINGLE_FUNCTION(acosh)
40
+ MPFR_SINGLE_FUNCTION(asinh)
41
+ MPFR_SINGLE_FUNCTION(atanh)
42
+ MPFR_SINGLE_FUNCTION(log1p)
43
+ MPFR_SINGLE_FUNCTION(expm1)
44
+ MPFR_SINGLE_FUNCTION(log2)
45
+ MPFR_SINGLE_FUNCTION(log10)
46
+
47
+ static VALUE r_gmpfr_nan_p(VALUE self)
48
+ {
49
+ MP_FLOAT *self_val;
50
+
51
+ mpf_get_struct(self, self_val);
52
+ if (mpfr_nan_p(self_val)) {
53
+ return Qtrue;
54
+ }
55
+ else {
56
+ return Qfalse;
57
+ }
58
+ }
59
+
60
+ static VALUE r_gmpfr_inf_p(VALUE self)
61
+ {
62
+ MP_FLOAT *self_val;
63
+
64
+ mpf_get_struct(self, self_val);
65
+ if (mpfr_inf_p(self_val)) {
66
+ return Qtrue;
67
+ }
68
+ else {
69
+ return Qfalse;
70
+ }
71
+ }
72
+
73
+ static VALUE r_gmpfr_fin_p(VALUE self)
74
+ {
75
+ if (r_gmpfr_inf_p(self)) {
76
+ return Qfalse;
77
+ }
78
+ else {
79
+ return Qtrue;
80
+ }
81
+ }
82
+
83
+ static VALUE r_gmpfr_number_p(VALUE self)
84
+ {
85
+ MP_FLOAT *self_val;
86
+
87
+ mpf_get_struct(self, self_val);
88
+ if (mpfr_number_p(self_val)) {
89
+ return Qtrue;
90
+ }
91
+ else {
92
+ return Qfalse;
93
+ }
94
+ }
95
+
96
+ static VALUE r_gmpfr_pow(VALUE self, VALUE arg)
97
+ {
98
+ MP_FLOAT *self_val, *res_val, *arg_val_f;
99
+ MP_RAT *arg_val_q;
100
+ MP_INT *arg_val_z;
101
+ unsigned long prec;
102
+ VALUE res;
103
+
104
+ mpf_get_struct_prec(self, self_val, prec);
105
+
106
+ if (GMPF_P(arg)) {
107
+ mpf_get_struct(arg, arg_val_f);
108
+ prec_max(prec, arg_val_f);
109
+ mpf_make_struct_init(res, res_val, prec);
110
+ mpfr_pow(res_val, self_val, arg_val_f, __gmp_default_rounding_mode);
111
+ } else {
112
+ mpf_make_struct_init(res, res_val, prec);
113
+
114
+ if (GMPZ_P(arg)) {
115
+ mpz_get_struct(arg, arg_val_z);
116
+ mpf_set_z(res_val, arg_val_z);
117
+ mpfr_pow(res_val, self_val, res_val, __gmp_default_rounding_mode);
118
+ } else if (GMPQ_P(arg)) {
119
+ mpq_get_struct(arg, arg_val_q);
120
+ mpf_set_q(res_val, arg_val_q);
121
+ mpfr_pow(res_val, self_val, res_val, __gmp_default_rounding_mode);
122
+ } else if (FLOAT_P(arg)) {
123
+ mpf_set_d(res_val, FLT2DBL(arg));
124
+ mpfr_pow(res_val, self_val, res_val, __gmp_default_rounding_mode);
125
+ } else if (FIXNUM_P(arg)) {
126
+ mpfr_pow_si(res_val, self_val, FIX2INT(arg), __gmp_default_rounding_mode);
127
+ } else if (BIGNUM_P(arg)) {
128
+ mpz_temp_from_bignum(arg_val_z, arg);
129
+ mpf_set_z(res_val, arg_val_z);
130
+ mpz_temp_free(arg_val_z);
131
+ mpfr_pow(res_val, self_val, res_val, __gmp_default_rounding_mode);
132
+ } else {
133
+ typeerror(ZQFXBD);
134
+ }
135
+
136
+ }
137
+
138
+ return res;
139
+ }
140
+
141
+ #endif
142
+
143
+
144
+ #endif
@@ -0,0 +1,774 @@
1
+ #include <gmpz.h>
2
+ #include <gmpq.h>
3
+ #include <gmpf.h>
4
+
5
+ /*
6
+ * Document-class: GMP::Q
7
+ *
8
+ * GMP Multiple Precision Rational Number.
9
+ *
10
+ * Instances of this class can store variables of the type mpq_t. This class
11
+ * also contains many methods that act as the functions for mpq_t variables,
12
+ * as well as a few methods that attempt to make this library more Ruby-ish.
13
+ *
14
+ * The following list is just a simple checklist for me, really. A better
15
+ * reference should be found in the rdocs.
16
+ *
17
+ * Ruby method C Extension function GMP function
18
+ * to_d r_gmpq_to_d mpq_get_d
19
+ * to_s r_gmpq_to_s ?
20
+ * add r_gmpq_add mpq_add
21
+ * sub r_gmpq_sub mpq_sub
22
+ * mul r_gmpq_mul mpq_mul
23
+ * div r_gmpq_div mpq_div
24
+ */
25
+
26
+ /**********************************************************************
27
+ * Macros *
28
+ **********************************************************************/
29
+
30
+ #define DEFUN_RAT_CMP(name,CMP_OP) \
31
+ static VALUE r_gmpq_cmp_##name(VALUE self, VALUE arg) \
32
+ { \
33
+ MP_RAT *self_val; \
34
+ mpq_get_struct(self,self_val); \
35
+ return (mpq_cmp_value(self_val, arg) CMP_OP 0)?Qtrue:Qfalse; \
36
+ }
37
+
38
+ #define DEFUN_RAT2INT(fname,mpz_fname) \
39
+ static VALUE r_gmpq_##fname(VALUE self) \
40
+ { \
41
+ MP_RAT *self_val; \
42
+ MP_INT *res_val; \
43
+ VALUE res; \
44
+ \
45
+ mpq_get_struct(self, self_val); \
46
+ mpz_make_struct_init(res, res_val) \
47
+ mpz_fname(res_val, mpq_numref(self_val), mpq_denref(self_val)); \
48
+ return res; \
49
+ }
50
+
51
+
52
+ /**********************************************************************
53
+ * Initializing Rationals *
54
+ **********************************************************************/
55
+
56
+ /*
57
+ * call-seq:
58
+ * GMP::Q.new(arg)
59
+ *
60
+ * Creates a new GMP::Q rational, with arg as its value, converting where
61
+ * necessary.
62
+ */
63
+ VALUE r_gmpqsg_new(int argc, VALUE *argv, VALUE klass)
64
+ {
65
+ MP_RAT *res_val;
66
+ VALUE res;
67
+
68
+ (void)klass;
69
+
70
+ if (argc > 2)
71
+ rb_raise(rb_eArgError, "wrong # of arguments(%d for 0, 1 or 2)", argc);
72
+
73
+ mpq_make_struct (res, res_val);
74
+ mpq_init (res_val);
75
+ rb_obj_call_init(res, argc, argv);
76
+
77
+ return res;
78
+ }
79
+
80
+ /*
81
+ * call-seq:
82
+ * GMP::Q(arg)
83
+ *
84
+ * A convenience method for +GMP::Q.new(arg)+.
85
+ */
86
+ VALUE r_gmpmod_q(int argc, VALUE *argv, VALUE module)
87
+ {
88
+ (void)module;
89
+ return r_gmpqsg_new(argc, argv, cGMP_Q);
90
+ }
91
+
92
+ /*
93
+ * call-seq:
94
+ * rat1.swap rat2
95
+ *
96
+ * Efficiently swaps the contents of +rat1+ with +rat2+.
97
+ */
98
+ VALUE r_gmpq_swap(VALUE self, VALUE arg)
99
+ {
100
+ MP_RAT *self_val, *arg_val;
101
+
102
+ if (!GMPQ_P(arg)) {
103
+ rb_raise(rb_eTypeError, "Can't swap GMP::Q with object of other class");
104
+ }
105
+
106
+ mpq_get_struct(self, self_val);
107
+ mpq_get_struct(arg, arg_val);
108
+ mpq_swap(self_val,arg_val);
109
+
110
+ return Qnil;
111
+ }
112
+
113
+
114
+ /**********************************************************************
115
+ * Rational Conversions *
116
+ **********************************************************************/
117
+
118
+ /*
119
+ * call-seq:
120
+ * rational.to_d
121
+ *
122
+ * Returns +rational+ as an Float if +rational+ fits in a Float.
123
+ *
124
+ * Otherwise returns the least significant part of +rational+, with the same
125
+ * sign as +rational+.
126
+ *
127
+ * If the exponent from the conversion is too big or too small to fit a double
128
+ * then the result is system dependent. For too big an infinity is returned
129
+ * when available. For too small 0.0 is normally returned. Hardware overflow,
130
+ * underflow and denorm traps may or may not occur.
131
+ */
132
+ VALUE r_gmpq_to_d(VALUE self)
133
+ {
134
+ MP_RAT *self_val;
135
+ mpq_get_struct(self, self_val);
136
+
137
+ return rb_float_new(mpq_get_d(self_val));
138
+ }
139
+
140
+ /*
141
+ * Document-method: to_s
142
+ *
143
+ * call-seq:
144
+ * rational.to_s
145
+ *
146
+ * Returns the decimal representation of +rational+, as a Ruby string.
147
+ */
148
+ VALUE r_gmpq_to_s(VALUE self)
149
+ {
150
+ MP_RAT *self_val;
151
+ MP_INT *self_val_num, *self_val_den;
152
+ char *str;
153
+ VALUE res;
154
+ int sizeinbase;
155
+ int offset;
156
+
157
+ Data_Get_Struct(self, MP_RAT, self_val);
158
+
159
+ if (mpz_cmp_ui(mpq_denref(self_val), 1) == 0) {
160
+ str = mpz_get_str(NULL, 10, mpq_numref (self_val));
161
+ res = rb_str_new2(str);
162
+ free (str);
163
+ return res;
164
+ }
165
+
166
+ self_val_num = mpq_numref(self_val);
167
+ self_val_den = mpq_denref(self_val);
168
+
169
+ sizeinbase = mpz_sizeinbase (self_val_num, 10) + mpz_sizeinbase (self_val_den, 10) + 3;
170
+ str = malloc (sizeinbase);
171
+
172
+ mpz_get_str (str, 10, self_val_num);
173
+ offset = strlen (str);
174
+ str[offset] = '/';
175
+ mpz_get_str (str + offset + 1, 10, self_val_den);
176
+ res = rb_str_new2(str);
177
+ free (str);
178
+
179
+ return res;
180
+ }
181
+
182
+
183
+ /**********************************************************************
184
+ * Rational Arithmetic *
185
+ **********************************************************************/
186
+
187
+ /*
188
+ * call-seq:
189
+ * rat1 + rat2
190
+ *
191
+ * Adds +rat1+ to +rat2+. +rat2+ can be
192
+ * * GMP::Z
193
+ * * Fixnum
194
+ * * GMP::Q
195
+ * * GMP::F
196
+ * * Bignum
197
+ */
198
+ VALUE r_gmpq_add(VALUE self, VALUE arg)
199
+ {
200
+ MP_RAT *self_val, *arg_val_q, *res_val;
201
+ MP_INT *arg_val_z, *res_val_num;
202
+ VALUE res;
203
+
204
+ mpq_get_struct(self, self_val);
205
+ mpq_make_struct_init(res, res_val);
206
+
207
+ if (GMPQ_P(arg)) {
208
+ mpq_get_struct(arg,arg_val_q);
209
+ mpq_add(res_val, self_val, arg_val_q);
210
+ } else if (GMPZ_P(arg)) {
211
+ res_val_num = mpq_numref(res_val);
212
+ mpz_set(mpq_denref(res_val), mpq_denref(self_val));
213
+ mpz_get_struct(arg, arg_val_z);
214
+ mpz_mul(res_val_num, mpq_denref(self_val), arg_val_z);
215
+ mpz_add(res_val_num, res_val_num, mpq_numref(self_val));
216
+ } else if (FIXNUM_P(arg)) {
217
+ res_val_num = mpq_numref(res_val);
218
+ mpz_set(mpq_denref(res_val), mpq_denref(self_val));
219
+ mpz_mul_si(res_val_num, mpq_denref(self_val), FIX2INT(arg));
220
+ mpz_add(res_val_num, res_val_num, mpq_numref(self_val));
221
+ } else if (GMPF_P(arg)) {
222
+ return r_gmpf_add(arg,self);
223
+ } else if (BIGNUM_P(arg)) {
224
+ res_val_num = mpq_numref(res_val);
225
+ mpz_set(mpq_denref(res_val), mpq_denref(self_val));
226
+ mpz_set_bignum(res_val_num, arg);
227
+ mpz_mul(res_val_num, res_val_num, mpq_denref(self_val));
228
+ mpz_add(res_val_num, res_val_num, mpq_numref(self_val));
229
+ } else {
230
+ typeerror(ZQFXB);
231
+ }
232
+ return res;
233
+ }
234
+
235
+ /*
236
+ * call-seq:
237
+ * rat1 - rat2
238
+ *
239
+ * Subtracts +rat2+ from +rat1+. +rat2+ can be
240
+ * * GMP::Z
241
+ * * Fixnum
242
+ * * GMP::Q
243
+ * * GMP::F
244
+ * * Bignum
245
+ */
246
+ VALUE r_gmpq_sub(VALUE self, VALUE arg)
247
+ {
248
+ MP_RAT *self_val, *arg_val_q, *res_val;
249
+ MP_INT *arg_val_z, *res_val_num;
250
+ MP_FLOAT *arg_val_f, *res_val_f;
251
+ VALUE res;
252
+ unsigned int prec;
253
+
254
+ mpq_get_struct(self, self_val);
255
+ mpq_make_struct_init(res, res_val);
256
+
257
+ if (GMPQ_P(arg)) {
258
+ mpq_get_struct(arg,arg_val_q);
259
+ mpq_sub (res_val, self_val, arg_val_q);
260
+ } else if (GMPZ_P(arg)) {
261
+ res_val_num = mpq_numref(res_val);
262
+ mpz_set (mpq_denref(res_val), mpq_denref(self_val));
263
+ mpz_get_struct(arg, arg_val_z);
264
+ mpz_mul (res_val_num, mpq_denref(self_val), arg_val_z);
265
+ mpz_neg (res_val_num, res_val_num);
266
+ mpz_add (res_val_num, res_val_num, mpq_numref(self_val));
267
+ } else if (FIXNUM_P(arg)) {
268
+ res_val_num = mpq_numref(res_val);
269
+ mpz_set (mpq_denref(res_val), mpq_denref(self_val));
270
+ mpz_mul_si (res_val_num, mpq_denref(self_val), -FIX2INT(arg));
271
+ mpz_add (res_val_num, res_val_num, mpq_numref(self_val));
272
+ } else if (GMPF_P(arg)) {
273
+ mpf_get_struct_prec (arg, arg_val_f, prec);
274
+ mpf_make_struct_init(res, res_val_f, prec);
275
+ mpf_set_q (res_val_f, self_val);
276
+ mpf_sub (res_val_f, res_val_f, arg_val_f);
277
+ } else if (BIGNUM_P(arg)) {
278
+ res_val_num = mpq_numref(res_val);
279
+ mpz_set (mpq_denref(res_val), mpq_denref(self_val));
280
+ mpz_set_bignum (res_val_num, arg);
281
+ mpz_mul (res_val_num, res_val_num, mpq_denref(self_val));
282
+ mpz_neg (res_val_num, res_val_num);
283
+ mpz_add (res_val_num, res_val_num, mpq_numref(self_val));
284
+ } else {
285
+ typeerror (ZQFXB);
286
+ }
287
+ return res;
288
+ }
289
+
290
+ /*
291
+ * call-seq:
292
+ * rat1 * rat2
293
+ *
294
+ * Multiplies +rat1+ with +rat2+. +rat2+ can be
295
+ * * GMP::Z
296
+ * * Fixnum
297
+ * * GMP::Q
298
+ * * GMP::F
299
+ * * Bignum
300
+ */
301
+ VALUE r_gmpq_mul(VALUE self, VALUE arg)
302
+ {
303
+ MP_RAT *self_val, *arg_val_q, *res_val;
304
+ MP_INT *arg_val_z, *tmp_z;
305
+ VALUE res;
306
+ #if GMP >= 4
307
+ unsigned long tmp_ui;
308
+ #endif
309
+
310
+ mpq_get_struct(self, self_val);
311
+ mpq_make_struct_init(res, res_val);
312
+
313
+ if (GMPQ_P(arg)) {
314
+ mpq_get_struct(arg,arg_val_q);
315
+ mpq_mul(res_val, self_val, arg_val_q);
316
+ } else if (GMPZ_P(arg)) {
317
+ mpz_get_struct(arg,arg_val_z);
318
+ mpz_temp_init(tmp_z);
319
+ mpz_gcd(tmp_z, mpq_denref(self_val), arg_val_z);
320
+ mpz_divexact(mpq_denref(res_val), mpq_denref(self_val), tmp_z);
321
+ mpz_divexact(mpq_numref(res_val), arg_val_z, tmp_z);
322
+ mpz_mul(mpq_numref(res_val), mpq_numref(res_val), mpq_numref(self_val));
323
+ mpz_temp_free(tmp_z);
324
+ } else if (FIXNUM_P(arg)) {
325
+ #if GMP >= 4
326
+ if (FIX2INT(arg) > 0) {
327
+ tmp_ui = mpz_gcd_ui(0, mpq_denref(self_val), FIX2INT(arg));
328
+ } else if (FIX2INT(arg) < 0) {
329
+ tmp_ui = mpz_gcd_ui(0, mpq_denref(self_val), -FIX2INT(arg));
330
+ } else {
331
+ mpz_set_ui(mpq_numref(res_val), 0);
332
+ mpz_set_ui(mpq_denref(res_val), 1);
333
+ return res;
334
+ }
335
+ mpz_divexact_ui(mpq_denref(res_val), mpq_denref(self_val), tmp_ui);
336
+ mpz_mul_ui(mpq_numref(res_val), mpq_numref(self_val), FIX2INT(arg)/tmp_ui);
337
+ #else
338
+ mpz_set(mpq_denref(res_val), mpq_denref(self_val));
339
+ mpz_mul_si(mpq_numref(res_val), mpq_numref(self_val), FIX2INT(arg));
340
+ mpq_canonicalize(res_val);
341
+ #endif
342
+ } else if (GMPF_P(arg)) {
343
+ return r_gmpf_mul(arg, self);
344
+ } else if (BIGNUM_P(arg)) {
345
+ mpz_temp_alloc(tmp_z);
346
+ mpz_set_bignum(tmp_z, arg);
347
+ mpz_gcd(mpq_denref(res_val), mpq_denref(self_val), tmp_z);
348
+ mpz_divexact(mpq_numref(res_val), tmp_z, mpq_denref(res_val));
349
+ mpz_divexact(mpq_denref(res_val), mpq_denref(self_val), mpq_denref(res_val));
350
+ mpz_mul(mpq_numref(res_val), mpq_numref(res_val), mpq_numref(self_val));
351
+ mpz_temp_free(tmp_z);
352
+ } else {
353
+ typeerror(ZQFXB);
354
+ }
355
+ return res;
356
+ }
357
+
358
+ /*
359
+ * call-seq:
360
+ * rat1 / rat2
361
+ *
362
+ * Divides +rat1+ by +rat2+. +rat2+ can be
363
+ * * GMP::Z
364
+ * * Fixnum
365
+ * * GMP::Q
366
+ * * GMP::F
367
+ * * Bignum
368
+ */
369
+ VALUE r_gmpq_div(VALUE self, VALUE arg)
370
+ {
371
+ MP_RAT *self_val, *arg_val_q, *res_val;
372
+ MP_INT *arg_val_z, *tmp_z;
373
+ MP_FLOAT *arg_val_f, *res_val_f;
374
+ VALUE res;
375
+ unsigned long tmp_ui, prec;
376
+
377
+ mpq_get_struct(self, self_val);
378
+ mpq_make_struct_init(res, res_val);
379
+
380
+ if (GMPQ_P(arg)) {
381
+ mpq_get_struct(arg,arg_val_q);
382
+ if (mpz_sgn(mpq_numref(arg_val_q)) == 0)
383
+ rb_raise(rb_eZeroDivError, "divided by 0");
384
+ mpq_div(res_val, self_val, arg_val_q);
385
+ } else if (GMPZ_P(arg)) {
386
+ mpz_get_struct(arg,arg_val_z);
387
+ mpz_temp_init(tmp_z);
388
+ mpz_gcd(tmp_z, mpq_numref(self_val), arg_val_z);
389
+ mpz_divexact(mpq_numref(res_val), mpq_numref(self_val), tmp_z);
390
+ mpz_divexact(mpq_denref(res_val), arg_val_z, tmp_z);
391
+ mpz_mul(mpq_denref(res_val), mpq_denref(res_val), mpq_denref(self_val));
392
+ mpz_temp_free(tmp_z);
393
+ } else if (FIXNUM_P(arg)) {
394
+ if (FIX2INT(arg) == 0)
395
+ rb_raise(rb_eZeroDivError, "divided by 0");
396
+ if (FIX2INT(arg) > 0) {
397
+ tmp_ui = mpz_gcd_ui(0, mpq_numref(self_val), FIX2INT(arg));
398
+ } else {
399
+ tmp_ui = mpz_gcd_ui(0, mpq_numref(self_val), -FIX2INT(arg));
400
+ }
401
+ mpz_divexact_ui(mpq_numref(res_val), mpq_numref(self_val), tmp_ui);
402
+ mpz_mul_ui(mpq_denref(res_val), mpq_denref(self_val), FIX2INT(arg)/tmp_ui);
403
+ } else if (GMPF_P(arg)) {
404
+ mpf_get_struct_prec(arg, arg_val_f, prec);
405
+ mpf_make_struct_init(res, res_val_f, prec);
406
+ mpf_set_q(res_val_f, self_val);
407
+ mpf_div(res_val_f, res_val_f, arg_val_f);
408
+ } else if (BIGNUM_P(arg)) {
409
+ mpz_temp_alloc(tmp_z);
410
+ mpz_set_bignum(tmp_z, arg);
411
+ mpz_gcd(mpq_numref(res_val), mpq_numref(self_val), tmp_z);
412
+ mpz_divexact(mpq_denref(res_val), tmp_z, mpq_numref(res_val));
413
+ mpz_divexact(mpq_numref(res_val), mpq_numref(self_val), mpq_numref(res_val));
414
+ mpz_mul(mpq_denref(res_val), mpq_denref(res_val), mpq_denref(self_val));
415
+ mpz_temp_free(tmp_z);
416
+ } else {
417
+ typeerror(ZQFXB);
418
+ }
419
+ return res;
420
+ }
421
+
422
+ DEFUN_RAT2INT(floor,mpz_fdiv_q)
423
+ DEFUN_RAT2INT(trunc,mpz_tdiv_q)
424
+ DEFUN_RAT2INT(ceil,mpz_cdiv_q)
425
+
426
+ /*
427
+ * Document-method: neg
428
+ *
429
+ * call-seq:
430
+ * -rational
431
+ * rational.neg
432
+ *
433
+ * From the GMP Manual:
434
+ *
435
+ * Returns -+rational+.
436
+ */
437
+ VALUE r_gmpq_neg(VALUE self)
438
+ {
439
+ MP_RAT *self_val, *res_val;
440
+ VALUE res;
441
+ mpq_get_struct(self, self_val);
442
+ mpq_make_struct_init(res, res_val);
443
+ mpq_neg (res_val, self_val);
444
+ return res;
445
+ }
446
+
447
+ /*
448
+ * Document-method: neg!
449
+ *
450
+ * call-seq:
451
+ * rational.neg!
452
+ *
453
+ * From the GMP Manual:
454
+ *
455
+ * Sets +rational+ to -+rational+.
456
+ */
457
+ VALUE r_gmpq_neg_self(VALUE self)
458
+ {
459
+ MP_RAT *self_val;
460
+ mpq_get_struct(self, self_val);
461
+ mpz_neg (mpq_numref(self_val), mpq_numref(self_val));
462
+ return Qnil;
463
+ }
464
+
465
+ /*
466
+ * Document-method: abs
467
+ *
468
+ * call-seq:
469
+ * rational.abs
470
+ *
471
+ * From the GMP Manual:
472
+ *
473
+ * Returns the absolute value of +rational+.
474
+ */
475
+ VALUE r_gmpq_abs(VALUE self)
476
+ {
477
+ MP_RAT *self_val, *res_val;
478
+ VALUE res;
479
+ mpq_get_struct(self, self_val);
480
+ mpq_make_struct_init(res, res_val);
481
+ mpz_abs (mpq_numref(res_val), mpq_numref(self_val));
482
+ mpz_set (mpq_denref(res_val), mpq_denref(self_val));
483
+
484
+ return res;
485
+ }
486
+
487
+ /*
488
+ * Document-method: abs!
489
+ *
490
+ * call-seq:
491
+ * rational.abs!
492
+ *
493
+ * From the GMP Manual:
494
+ *
495
+ * Sets +rational+ to its absolute value.
496
+ */
497
+ VALUE r_gmpq_abs_self(VALUE self)
498
+ {
499
+ MP_RAT *self_val;
500
+ mpq_get_struct(self, self_val);
501
+ mpz_abs (mpq_numref(self_val), mpq_numref(self_val));
502
+ return Qnil;
503
+ }
504
+
505
+ VALUE r_gmpq_inv(VALUE self)
506
+ {
507
+ MP_RAT *self_val, *res_val;
508
+ VALUE res;
509
+
510
+ mpq_get_struct(self, self_val);
511
+ if (mpq_sgn(self_val) == 0)
512
+ rb_raise (rb_eZeroDivError, "divided by 0");
513
+ mpq_make_struct_init(res, res_val);
514
+ mpq_inv (res_val, self_val);
515
+
516
+ return res;
517
+ }
518
+
519
+ VALUE r_gmpq_inv_self(VALUE self)
520
+ {
521
+ MP_RAT *self_val;
522
+ mpq_get_struct(self, self_val);
523
+ if (mpq_sgn(self_val) == 0)
524
+ rb_raise (rb_eZeroDivError, "divided by 0");
525
+ mpq_inv (self_val, self_val);
526
+ return Qnil;
527
+ }
528
+
529
+
530
+ /**********************************************************************
531
+ * Comparing Rationals *
532
+ **********************************************************************/
533
+
534
+ int mpq_cmp_value(MP_RAT *OP, VALUE arg)
535
+ {
536
+ MP_INT *arg_val_z, *tmp_z;
537
+ MP_RAT *arg_val_q;
538
+ int res;
539
+
540
+ if (GMPQ_P(arg)) {
541
+ mpq_get_struct(arg,arg_val_q);
542
+ return mpq_cmp(OP,arg_val_q);
543
+ } else if (GMPZ_P(arg)) {
544
+ mpz_get_struct(arg, arg_val_z);
545
+ mpz_temp_alloc(tmp_z);
546
+ mpz_init(tmp_z);
547
+ mpz_mul(tmp_z, mpq_denref(OP), arg_val_z);
548
+ res = mpz_cmp(mpq_numref(OP),tmp_z);
549
+ mpz_temp_free(tmp_z);
550
+ return res;
551
+ } else if (FIXNUM_P(arg)) {
552
+ mpz_temp_alloc(tmp_z);
553
+ mpz_init(tmp_z);
554
+ mpz_mul_si(tmp_z, mpq_denref(OP), FIX2INT(arg));
555
+ res = mpz_cmp(mpq_numref(OP), tmp_z);
556
+ mpz_temp_free(tmp_z);
557
+ return res;
558
+ } else if (GMPF_P(arg)) {
559
+ not_yet;
560
+ } else if (BIGNUM_P(arg)) {
561
+ mpz_temp_from_bignum(tmp_z, arg);
562
+ mpz_mul(tmp_z, tmp_z, mpq_denref(OP));
563
+ res = mpz_cmp(mpq_numref(OP), tmp_z);
564
+ mpz_temp_free(tmp_z);
565
+ return res;
566
+ } else {
567
+ typeerror(ZQFXB);
568
+ }
569
+ }
570
+
571
+ VALUE r_gmpq_eq(VALUE self, VALUE arg)
572
+ {
573
+ MP_RAT *self_val, *arg_val_q;
574
+ MP_INT *arg_val_z;
575
+
576
+ mpq_get_struct(self,self_val);
577
+ if (GMPQ_P(arg)) {
578
+ mpq_get_struct(arg,arg_val_q);
579
+ return mpq_equal(self_val,arg_val_q)?Qtrue:Qfalse;
580
+ } else if (GMPZ_P(arg)) {
581
+ if (mpz_cmp_ui(mpq_denref(self_val), 1) != 0)
582
+ return Qfalse;
583
+ mpz_get_struct (arg, arg_val_z);
584
+ return (mpz_cmp(mpq_numref(self_val),arg_val_z)==0)?Qtrue:Qfalse;
585
+ } else if (FIXNUM_P(arg)) {
586
+ if (mpz_cmp_ui(mpq_denref(self_val), 1) != 0)
587
+ return Qfalse;
588
+ return (mpz_cmp_ui(mpq_numref(self_val),FIX2INT(arg))==0)?Qtrue:Qfalse;
589
+ } else if (BIGNUM_P(arg)) {
590
+ if (mpz_cmp_ui(mpq_denref(self_val), 1) != 0)
591
+ return Qfalse;
592
+ mpz_temp_from_bignum(arg_val_z, arg);
593
+ if (mpz_cmp (mpq_numref(self_val),arg_val_z) == 0) {
594
+ mpz_temp_free (arg_val_z);
595
+ return Qtrue;
596
+ } else {
597
+ mpz_temp_free (arg_val_z);
598
+ return Qfalse;
599
+ }
600
+ } else {
601
+ return Qfalse;
602
+ }
603
+ }
604
+
605
+ VALUE r_gmpq_cmp(VALUE self, VALUE arg)
606
+ {
607
+ MP_RAT *self_val;
608
+ int res;
609
+ mpq_get_struct (self,self_val);
610
+ res = mpq_cmp_value(self_val, arg);
611
+ if (res > 0)
612
+ return INT2FIX(1);
613
+ else if (res == 0)
614
+ return INT2FIX(0);
615
+ else
616
+ return INT2FIX(-1);
617
+ }
618
+
619
+ DEFUN_RAT_CMP(lt,<)
620
+ DEFUN_RAT_CMP(le,<=)
621
+ DEFUN_RAT_CMP(gt,>)
622
+ DEFUN_RAT_CMP(ge,>=)
623
+
624
+ static VALUE r_gmpq_cmpabs(VALUE self, VALUE arg)
625
+ {
626
+ MP_RAT *arg_val_q, *self_val;
627
+ MP_INT *arg_val_z, *tmp_z;
628
+ int res;
629
+ int sgnt;
630
+
631
+ mpq_get_struct(self, self_val);
632
+
633
+ if (GMPQ_P(arg)) {
634
+ mpq_get_struct(arg,arg_val_q);
635
+ sgnt = 3*mpz_sgn(mpq_numref(self_val)) + mpz_sgn(mpq_numref(arg_val_q));
636
+ switch (sgnt)
637
+ {
638
+ default:
639
+ case 0:
640
+ return INT2FIX(0);
641
+ case 1:
642
+ case -1:
643
+ return INT2FIX(-1);
644
+ case 2:
645
+ tmp_z = mpq_numref(arg_val_q);
646
+ mpz_neg (tmp_z, tmp_z);
647
+ res = mpq_cmp (self_val, arg_val_q);
648
+ mpz_neg (tmp_z, tmp_z);
649
+ return res;
650
+ case -2:
651
+ tmp_z = mpq_numref(arg_val_q);
652
+ mpz_neg (tmp_z, tmp_z);
653
+ res = mpq_cmp (self_val, arg_val_q);
654
+ mpz_neg (tmp_z, tmp_z);
655
+ return res;
656
+ case 3:
657
+ case -3:
658
+ return INT2FIX(1);
659
+ case 4:
660
+ case -4:
661
+ return INT2FIX(mpq_cmp (self_val,arg_val_q));
662
+ }
663
+ } else if (GMPZ_P(arg)) {
664
+ mpz_get_struct(arg, arg_val_z);
665
+ mpz_temp_alloc (tmp_z);
666
+ mpz_init (tmp_z);
667
+ mpz_mul (tmp_z, mpq_denref(self_val), arg_val_z);
668
+ res = mpz_cmpabs (mpq_numref(self_val),tmp_z);
669
+ mpz_temp_free (tmp_z);
670
+ return res;
671
+ } else if (FIXNUM_P(arg)) {
672
+ mpz_temp_alloc (tmp_z);
673
+ mpz_init (tmp_z);
674
+ mpz_mul_si (tmp_z, mpq_denref(self_val), FIX2INT(arg));
675
+ res = mpz_cmpabs (mpq_numref(self_val), tmp_z);
676
+ mpz_temp_free (tmp_z);
677
+ return res;
678
+ } else if (GMPF_P(arg)) {
679
+ not_yet;
680
+ } else if (BIGNUM_P(arg)) {
681
+ mpz_temp_from_bignum (tmp_z, arg);
682
+ mpz_mul (tmp_z, tmp_z, mpq_denref(self_val));
683
+ res = mpz_cmpabs (mpq_numref(self_val), tmp_z);
684
+ mpz_temp_free (tmp_z);
685
+ return res;
686
+ } else {
687
+ typeerror (ZQFXB);
688
+ }
689
+ }
690
+
691
+ /*
692
+ * call-seq:
693
+ * rational.sgn
694
+ *
695
+ * From the GMP Manual:
696
+ *
697
+ * Returns +1 if +rational+ > 0, 0 if +rational+ == 0, and -1 if +rational+ < 0.
698
+ */
699
+ VALUE r_gmpq_sgn(VALUE self)
700
+ {
701
+ MP_RAT *self_val;
702
+ mpq_get_struct(self, self_val);
703
+ return INT2FIX(mpq_sgn(self_val));
704
+ }
705
+
706
+ /**********************************************************************
707
+ * Applying Integer Functions *
708
+ **********************************************************************/
709
+
710
+ VALUE r_gmpq_num(VALUE self)
711
+ {
712
+ MP_RAT *self_val;
713
+ MP_INT *res_val;
714
+ VALUE res;
715
+ mpq_get_struct(self,self_val);
716
+ mpz_make_struct(res, res_val);
717
+ mpz_init_set (res_val, mpq_numref (self_val));
718
+ return res;
719
+ }
720
+
721
+ VALUE r_gmpq_den(VALUE self)
722
+ {
723
+ MP_RAT *self_val;
724
+ MP_INT *res_val;
725
+ VALUE res;
726
+ mpq_get_struct(self,self_val);
727
+ mpz_make_struct(res, res_val);
728
+ mpz_init_set (res_val, mpq_denref (self_val));
729
+ return res;
730
+ }
731
+
732
+
733
+ void init_gmpq()
734
+ {
735
+ mGMP = rb_define_module("GMP");
736
+ rb_define_module_function(mGMP, "Q", r_gmpmod_q, -1);
737
+
738
+ cGMP_Q = rb_define_class_under (mGMP, "Q", rb_cNumeric);
739
+
740
+ // Initializing Rationals
741
+ rb_define_singleton_method(cGMP_Q, "new", r_gmpqsg_new, -1);
742
+ rb_define_method(cGMP_Q, "swap", r_gmpq_swap, 1);
743
+
744
+ // Rational Conversions
745
+ rb_define_method(cGMP_Q, "to_d", r_gmpq_to_d, 0);
746
+ rb_define_method(cGMP_Q, "to_s", r_gmpq_to_s, 0);
747
+
748
+ // Rational Arithmetic
749
+ rb_define_method(cGMP_Q, "+", r_gmpq_add, 1);
750
+ rb_define_method(cGMP_Q, "-", r_gmpq_sub, 1);
751
+ rb_define_method(cGMP_Q, "*", r_gmpq_mul, 1);
752
+ rb_define_method(cGMP_Q, "/", r_gmpq_div, 1);
753
+ rb_define_method(cGMP_Q, "-@", r_gmpq_neg, 0);
754
+ rb_define_method(cGMP_Q, "neg!", r_gmpq_neg_self, 0);
755
+ rb_define_method(cGMP_Q, "inv", r_gmpq_inv, 0);
756
+ rb_define_method(cGMP_Q, "inv!", r_gmpq_inv_self, 0);
757
+ rb_define_method(cGMP_Q, "abs", r_gmpq_abs, 0);
758
+ rb_define_method(cGMP_Q, "abs!", r_gmpq_abs_self, 0);
759
+
760
+ // Comparing Rationals
761
+ rb_define_method(cGMP_Q, "<=>", r_gmpq_cmp, 1);
762
+ rb_define_method(cGMP_Q, ">", r_gmpq_cmp_gt, 1);
763
+ rb_define_method(cGMP_Q, ">=", r_gmpq_cmp_ge, 1);
764
+ rb_define_method(cGMP_Q, "<", r_gmpq_cmp_lt, 1);
765
+ rb_define_method(cGMP_Q, "<=", r_gmpq_cmp_le, 1);
766
+ rb_define_method(cGMP_Q, "==", r_gmpq_eq, 1);
767
+ rb_define_method(cGMP_Q, "sgn", r_gmpq_sgn, 0);
768
+ rb_define_method(cGMP_Q, "cmpabs", r_gmpq_cmpabs, 1);
769
+
770
+ // _unsorted_
771
+ rb_define_method(cGMP_Q, "floor", r_gmpq_floor, 0);
772
+ rb_define_method(cGMP_Q, "ceil", r_gmpq_ceil, 0);
773
+ rb_define_method(cGMP_Q, "trunc", r_gmpq_trunc, 0);
774
+ }