Implement BigMath.hypot(x, y, prec)

Author: tompng

lib/bigdecimal/math.rb

@@ -5,6 +5,7 @@
 #--
 # Contents:
 #   sqrt(x, prec)
+#   hypot(x, y, prec)
 #   sin (x, prec)
 #   cos (x, prec)
 #   tan (x, prec)
@@ -45,6 +46,22 @@ def sqrt(x, prec)
     x.sqrt(prec)
   end
 
+  # call-seq:
+  #   hypot(x, y, numeric) -> BigDecimal
+  #
+  # Returns sqrt(x**2 + y**2) to the specified number of digits of
+  # precision, +numeric+.
+  #
+  #   BigMath.hypot(BigDecimal('1'), BigDecimal('2'), 16).to_s
+  #   #=> "0.2236067977499789696409173668333333334e1"
+  #
+  def hypot(x, y, prec)
+    return BigDecimal::NAN if x.nan? || y.nan?
+    return BigDecimal::INFINITY if x.infinite? || y.infinite?
+    prec2 = prec + BigDecimal.double_fig
+    sqrt(x.mult(x, prec2) + y.mult(y, prec2), prec)
+  end
+
   # call-seq:
   #   sin(decimal, numeric) -> BigDecimal
   #

test/bigdecimal/test_bigmath.rb

@@ -9,6 +9,7 @@ class TestBigMath < Test::Unit::TestCase
   # SQRT in 116 (= 100 + double_fig) digits
   SQRT2 = BigDecimal("1.4142135623730950488016887242096980785696718753769480731766797379907324784621070388503875343276415727350138462309123")
   SQRT3 = BigDecimal("1.7320508075688772935274463415058723669428052538103806280558069794519330169088000370811461867572485756756261414154067")
+  SQRT5 = BigDecimal("2.2360679774997896964091736687312762354406183596115257242708972454105209256378048994144144083787822749695081761507738")
   PINF = BigDecimal("+Infinity")
   MINF = BigDecimal("-Infinity")
   NAN = BigDecimal("NaN")
@@ -33,6 +34,17 @@ def test_sqrt
     assert_relative_precision {|n| sqrt(BigDecimal("2e50"), n) }
   end
 
+  def test_hypot
+    assert_in_delta(SQRT2, hypot(BigDecimal("1"), BigDecimal("1"), 100), BigDecimal("1e-100"))
+    assert_in_delta(SQRT5, hypot(SQRT2, SQRT3, 100), BigDecimal("1e-100"))
+    assert_equal(0, hypot(BigDecimal(0), BigDecimal(0), N))
+    assert_equal(PINF, hypot(PINF, SQRT3, N))
+    assert_equal(PINF, hypot(SQRT3, MINF, N))
+    assert_relative_precision {|n| hypot(BigDecimal("1e-30"), BigDecimal("2e-30"), n) }
+    assert_relative_precision {|n| hypot(BigDecimal("1.23"), BigDecimal("4.56"), n) }
+    assert_relative_precision {|n| hypot(BigDecimal("2e30"), BigDecimal("1e30"), n) }
+  end
+
   def test_sin
     assert_in_delta(0.0, sin(BigDecimal("0.0"), N))
     assert_in_delta(Math.sqrt(2.0) / 2, sin(PI(N) / 4, N))