[ruby-dev:21033] [BigDecimal] modify methods result from Comparable

斎藤です。

BigDecimal中の比較周りのメソッドは、残念ながらその起源である
Comparableの定義とは異なってしまっています。
ということで、それらの修正を含んだパッチです。

もしよければ取り込んで下さい。よろしくお願いします。

# num_coerce_relopがstaticなので、4関数分くらい再発明が必要に
# なっています。既にrb_num_coerce_cmpもありますし、relopの方も
# 公開するのがいいと思うのですが、どうでしょうか。>まつもとさん


Index: bigdecimal.c
===================================================================
RCS file: /src/ruby/ext/bigdecimal/bigdecimal.c,v
retrieving revision 1.17
diff -u -p -r1.17 bigdecimal.c
--- bigdecimal.c        29 Jul 2003 07:52:48 -0000      1.17
+++ bigdecimal.c        30 Jul 2003 16:47:06 -0000
@@ -90,31 +90,39 @@ coerce_rescue(VALUE *x)
 {
     rb_raise(rb_eTypeError, "%s can't be coerced into %s",
         rb_special_const_p(x[1])?
-        rb_str2cstr(rb_inspect(x[1]),0):
-        rb_class2name(CLASS_OF(x[1])),
-        rb_class2name(CLASS_OF(x[0])));
-    return (VALUE)0;
+        RSTRING(rb_inspect(x[1]))->ptr:
+        rb_obj_classname(x[1]),
+        rb_obj_classname(x[0]));
+    return Qnil;
 }

-static void
-do_coerce(VALUE *x, VALUE *y)
+static int
+do_coerce(VALUE *x, VALUE *y, int err)
 {
     VALUE ary;
     VALUE a[2];
     a[0] = *x; a[1] = *y;
-    ary = rb_rescue(coerce_body, (VALUE)a, coerce_rescue, (VALUE)a);
+    ary = rb_rescue(coerce_body, (VALUE)a, err?coerce_rescue:0, (VALUE)a);
     if (TYPE(ary) != T_ARRAY || RARRAY(ary)->len != 2) {
-        rb_raise(rb_eTypeError, "coerce must return [x, y]");
+       if (err) rb_raise(rb_eTypeError, "coerce must return [x, y]");
+       return Qfalse;
     }
     *x = RARRAY(ary)->ptr[0];
     *y = RARRAY(ary)->ptr[1];
+    return Qtrue;
 }

 static VALUE
-DoSomeOne(VALUE x, VALUE y)
+num_coerce_relop(VALUE x, VALUE y)
 {
-    do_coerce(&x, &y);
-    return rb_funcall(x, rb_frame_last_func(), 1, y);
+    VALUE c, x0 = x, y0 = y;
+
+    if (!do_coerce(&x, &y, Qfalse) ||
+       NIL_P(c = rb_funcall(x, rb_frame_last_func(), 1, y))) {
+       rb_cmperr(x0, y0);
+       return Qnil;            /* not reached */
+    }
+    return c;
 }

 static void
@@ -176,8 +184,8 @@ SomeOneMayDoIt:
     if(must) {
         rb_raise(rb_eTypeError, "%s can't be coerced into BigDecimal",
                     rb_special_const_p(v)?
-                    rb_str2cstr(rb_inspect(v),0):
-                    rb_class2name(CLASS_OF(v))
+                    RSTRING(rb_inspect(v))->ptr:
+                    rb_obj_classname(v)
                 );
     }
     return NULL; /* NULL means to coerce */
@@ -489,7 +497,7 @@ BigDecimal_add(VALUE self, VALUE r)

     GUARD_OBJ(a,GetVpValue(self,1));
     b = GetVpValue(r,0);
-    if(!b) return DoSomeOne(self,r);
+    if(!b) return rb_num_coerce_bin(self,r);
     SAVE(b);

     if(VpIsNaN(b)) return b->obj;
@@ -518,7 +526,7 @@ BigDecimal_sub(VALUE self, VALUE r)

     GUARD_OBJ(a,GetVpValue(self,1));
     b = GetVpValue(r,0);
-    if(!b) return DoSomeOne(self,r);
+    if(!b) return rb_num_coerce_bin(self,r);
     SAVE(b);

     if(VpIsNaN(b)) return b->obj;
@@ -539,18 +547,6 @@ BigDecimal_sub(VALUE self, VALUE r)
     return ToValue(c);
 }

-static S_INT
-BigDecimalCmp(VALUE self, VALUE r)
-{
-    ENTER(5);
-    Real *a, *b;
-    GUARD_OBJ(a,GetVpValue(self,1));
-    b = GetVpValue(r,0);
-    if(!b) return DoSomeOne(self,r);
-    SAVE(b);
-    return VpComp(a, b);
-}
-
 static VALUE
 BigDecimal_zero(VALUE self)
 {
@@ -569,8 +565,14 @@ static VALUE
 BigDecimal_comp(VALUE self, VALUE r)
 {
     S_INT e;
-    e = BigDecimalCmp(self, r);
-    if(e==999) return rb_float_new(VpGetDoubleNaN());
+    ENTER(5);
+    Real *a, *b;
+    GUARD_OBJ(a,GetVpValue(self,1));
+    b = GetVpValue(r,0);
+    if(!b) return rb_num_coerce_cmp(self,r);
+    SAVE(b);
+    e = VpComp(a, b);
+    if(e==999) return Qnil;
     return INT2FIX(e);
 }

@@ -581,7 +583,7 @@ BigDecimal_eq(VALUE self, VALUE r)
     Real *a, *b;
     GUARD_OBJ(a,GetVpValue(self,1));
     b = GetVpValue(r,0);
-    if(!b) return Qfalse; /* Not comparable */
+    if(!b) return Qnil; /* Not comparable */
     SAVE(b);
     return VpComp(a, b)? Qfalse:Qtrue;
 }
@@ -602,7 +604,12 @@ static VALUE
 BigDecimal_lt(VALUE self, VALUE r)
 {
     S_INT e;
-    e = BigDecimalCmp(self, r);
+    ENTER(1);
+    Real *a, *b;
+    GUARD_OBJ(a,GetVpValue(self,1));
+    b = GetVpValue(r,0);
+    if(!b) return num_coerce_relop(self, r);
+    e = VpComp(a, b);
     if(e==999) return Qfalse;
     return(e < 0) ? Qtrue : Qfalse;
 }
@@ -611,7 +618,12 @@ static VALUE
 BigDecimal_le(VALUE self, VALUE r)
 {
     S_INT e;
-    e = BigDecimalCmp(self, r);
+    ENTER(1);
+    Real *a, *b;
+    GUARD_OBJ(a,GetVpValue(self,1));
+    b = GetVpValue(r,0);
+    if(!b) return num_coerce_relop(self, r);
+    e = VpComp(a, b);
     if(e==999) return Qfalse;
     return(e <= 0) ? Qtrue : Qfalse;
 }
@@ -620,7 +632,12 @@ static VALUE
 BigDecimal_gt(VALUE self, VALUE r)
 {
     S_INT e;
-    e = BigDecimalCmp(self, r);
+    ENTER(1);
+    Real *a, *b;
+    GUARD_OBJ(a,GetVpValue(self,1));
+    b = GetVpValue(r,0);
+    if(!b) return num_coerce_relop(self, r);
+    e = VpComp(a, b);
     if(e==999) return Qfalse;
     return(e > 0) ? Qtrue : Qfalse;
 }
@@ -629,7 +646,12 @@ static VALUE
 BigDecimal_ge(VALUE self, VALUE r)
 {
     S_INT e;
-    e = BigDecimalCmp(self, r);
+    ENTER(1);
+    Real *a, *b;
+    GUARD_OBJ(a,GetVpValue(self,1));
+    b = GetVpValue(r,0);
+    if(!b) return num_coerce_relop(self, r);
+    e = VpComp(a, b);
     if(e==999) return Qfalse;
     return(e >= 0) ? Qtrue : Qfalse;
 }
@@ -654,7 +676,7 @@ BigDecimal_mult(VALUE self, VALUE r)

     GUARD_OBJ(a,GetVpValue(self,1));
     b = GetVpValue(r,0);
-    if(!b) return DoSomeOne(self,r);
+    if(!b) return rb_num_coerce_bin(self,r);
     SAVE(b);

     mx = a->Prec + b->Prec;
@@ -673,7 +695,7 @@ BigDecimal_divide(Real **c, Real **res,

     GUARD_OBJ(a,GetVpValue(self,1));
     b = GetVpValue(r,0);
-    if(!b) return DoSomeOne(self,r);
+    if(!b) return rb_num_coerce_bin(self,r);
     SAVE(b);
     *div = b;
     mx =(a->MaxPrec + b->MaxPrec + 1) * VpBaseFig();
@@ -717,7 +739,7 @@ BigDecimal_DoDivmod(VALUE self, VALUE r,

     GUARD_OBJ(a,GetVpValue(self,1));
     b = GetVpValue(r,0);
-    if(!b) return DoSomeOne(self,r);
+    if(!b) return rb_num_coerce_bin(self,r);
     SAVE(b);

     mx = a->Prec;
@@ -759,7 +781,7 @@ BigDecimal_divremain(VALUE self, VALUE r

     GUARD_OBJ(a,GetVpValue(self,1));
     b = GetVpValue(r,0);
-    if(!b) return DoSomeOne(self,r);
+    if(!b) return rb_num_coerce_bin(self,r);
     SAVE(b);

     mx  =(a->MaxPrec + b->MaxPrec) *VpBaseFig();


--
斎藤ただし