* lib/mathn.rb: a hack to provide canonicalization. This must be

	  temporary, but this seems to be not bad for the time being.



git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@19600 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 90 insertions, 0 deletions

diff --git a/lib/mathn.rb b/lib/mathn.rb
index e918608b0d..401b162b4c 100644
--- a/lib/mathn.rb
+++ b/lib/mathn.rb
@@ -18,19 +18,76 @@ unless defined?(Math.exp!)
   Math = CMath
 end
 
+class Object
+
+  def canon
+    if Rational === self
+      if denominator == 1
+	return numerator
+      end
+    elsif Complex === self
+      if Integer === imag && imag == 0
+	return real
+      end
+    end
+    self
+  end
+
+  private :canon
+
+end
+
+class Numeric
+
+  class << self
+
+    def def_canon(*ids)
+      for id in ids
+	module_eval <<-"end;"
+	  alias_method :__#{id.object_id}__, :#{id.to_s}
+	  private :__#{id.object_id}__
+	  def #{id.to_s}(*args, &block)
+	    __#{id.object_id}__(*args, &block).__send__(:canon)
+	  end
+	end;
+      end
+    end
+
+  end
+
+end
+
 class Fixnum
   remove_method :/
   alias / quo
+
+  def_canon *(instance_methods - Object.methods - [:canon])
+
 end
 
 class Bignum
   remove_method :/
   alias / quo
+
+  def_canon *(instance_methods - Object.methods - [:canon])
+
 end
 
+alias RationalOrig Rational
+private :RationalOrig
+def Rational(*args) RationalOrig(*args).__send__(:canon) end
+
 class Rational
   Unify = true
 
+  class << self
+    alias convert_orig convert
+    private :convert_orig
+    def convert(*args) convert_orig(*args).__send__(:canon) end
+  end
+
+  def_canon *(instance_methods - Object.methods - [:canon])
+
   alias power! **
 
   def ** (other)
@@ -169,6 +226,39 @@ module Math
   module_function :rsqrt
 end
 
+alias ComplexOrig Complex
+private :ComplexOrig
+def Complex(*args) ComplexOrig(*args).__send__(:canon) end
+
 class Complex
   Unify = true
+
+  class << self
+    alias convert_orig convert
+    private :convert_orig
+    def convert(*args) convert_orig(*args).__send__(:canon) end
+  end
+
+  def_canon *(instance_methods - Object.methods - [:canon])
+
+end
+
+class NilClass
+
+  def to_r() 0 end
+  def to_c() 0 end
+
+end
+
+class Integer
+
+  def to_r() self end
+  def to_c() self end
+
+end
+
+class Float
+
+  def_canon *(instance_methods - Object.methods - [:canon])
+
 end