mrkn	2010-08-28 17:52:14 +0900 (Sat, 28 Aug 2010)

  New Revision: 29127

  http://svn.ruby-lang.org/cgi-bin/viewvc.cgi?view=rev&revision=29127

  Log:
    * ext/bigdecimal/bigdecimal.c (BigDecimal_save_exception_mode,
      BigDecimal_save_rounding_mode, BigDecimal_save_limit): added.
    * test/bigdecimal/test_bigdecimal.rb: added tests for the above
      features.

  Modified files:
    trunk/ChangeLog
    trunk/ext/bigdecimal/bigdecimal.c
    trunk/test/bigdecimal/test_bigdecimal.rb

Index: ChangeLog
===================================================================
--- ChangeLog	(revision 29126)
+++ ChangeLog	(revision 29127)
@@ -1,3 +1,11 @@
+Sat Aug 28 17:39:33 2010  Kenta Murata  <mrkn@m...>
+
+	* ext/bigdecimal/bigdecimal.c (BigDecimal_save_exception_mode,
+	  BigDecimal_save_rounding_mode, BigDecimal_save_limit): added.
+
+	* test/bigdecimal/test_bigdecimal.rb: added tests for the above
+	  features.
+
 Sat Aug 28 08:11:05 2010  Tanaka Akira  <akr@f...>
 
 	* ext/pathname/pathname.c (path_blockdev_p): Pathname#blockdev?
Index: ext/bigdecimal/bigdecimal.c
===================================================================
--- ext/bigdecimal/bigdecimal.c	(revision 29126)
+++ ext/bigdecimal/bigdecimal.c	(revision 29127)
@@ -1800,6 +1800,48 @@
     return INT2FIX(s);
 }
 
+/* call-seq:
+ * BigDecimal.save_exception_mode { ... }
+ */
+static VALUE
+BigDecimal_save_exception_mode(VALUE self)
+{
+    unsigned short const exception_mode = VpGetException();
+    int state;
+    VALUE ret = rb_protect(rb_yield, Qnil, &state);
+    VpSetException(exception_mode);
+    if (state) rb_jump_tag(state);
+    return ret;
+}
+
+/* call-seq:
+ * BigDecimal.save_rounding_mode { ... }
+ */
+static VALUE
+BigDecimal_save_rounding_mode(VALUE self)
+{
+    unsigned short const round_mode = VpGetRoundMode();
+    int state;
+    VALUE ret = rb_protect(rb_yield, Qnil, &state);
+    VpSetRoundMode(round_mode);
+    if (state) rb_jump_tag(state);
+    return Qnil;
+}
+
+/* call-seq:
+ * BigDecimal.save_limit { ... }
+ */
+static VALUE
+BigDecimal_save_limit(VALUE self)
+{
+    size_t const limit = VpGetPrecLimit();
+    int state;
+    VALUE ret = rb_protect(rb_yield, Qnil, &state);
+    VpSetPrecLimit(limit);
+    if (state) rb_jump_tag(state);
+    return Qnil;
+}
+
 /* Document-class: BigDecimal
  * BigDecimal provides arbitrary-precision floating point decimal arithmetic.
  *
@@ -1926,6 +1968,10 @@
     rb_define_singleton_method(rb_cBigDecimal, "_load", BigDecimal_load, 1);
     rb_define_singleton_method(rb_cBigDecimal, "ver", BigDecimal_version, 0);
 
+    rb_define_singleton_method(rb_cBigDecimal, "save_exception_mode", BigDecimal_save_exception_mode, 0);
+    rb_define_singleton_method(rb_cBigDecimal, "save_rounding_mode", BigDecimal_save_rounding_mode, 0);
+    rb_define_singleton_method(rb_cBigDecimal, "save_limit", BigDecimal_save_limit, 0);
+
     /* Constants definition */
 
     /*
Index: test/bigdecimal/test_bigdecimal.rb
===================================================================
--- test/bigdecimal/test_bigdecimal.rb	(revision 29126)
+++ test/bigdecimal/test_bigdecimal.rb	(revision 29127)
@@ -77,6 +77,42 @@
     end
   end
 
+  def test_save_exception_mode
+    BigDecimal.mode(BigDecimal::EXCEPTION_OVERFLOW, false)
+    mode = BigDecimal.mode(BigDecimal::EXCEPTION_OVERFLOW)
+    BigDecimal.save_exception_mode do
+      BigDecimal.mode(BigDecimal::EXCEPTION_OVERFLOW, true)
+    end
+    assert_equal(mode, BigDecimal.mode(BigDecimal::EXCEPTION_OVERFLOW))
+
+    BigDecimal.mode(BigDecimal::ROUND_MODE, BigDecimal::ROUND_FLOOR)
+    BigDecimal.save_exception_mode do
+      BigDecimal.mode(BigDecimal::ROUND_MODE, BigDecimal::ROUND_HALF_EVEN)
+    end
+    assert_equal(BigDecimal::ROUND_HALF_EVEN, BigDecimal.mode(BigDecimal::ROUND_MODE))
+  end
+
+  def test_save_rounding_mode
+    BigDecimal.mode(BigDecimal::ROUND_MODE, BigDecimal::ROUND_FLOOR)
+    BigDecimal.save_rounding_mode do
+      BigDecimal.mode(BigDecimal::ROUND_MODE, BigDecimal::ROUND_HALF_EVEN)
+    end
+    assert_equal(BigDecimal::ROUND_FLOOR, BigDecimal.mode(BigDecimal::ROUND_MODE))
+  end
+
+  def test_save_limit
+    begin
+      old = BigDecimal.limit
+      BigDecimal.limit(100)
+      BigDecimal.save_limit do
+        BigDecimal.limit(200)
+      end
+      assert_equal(100, BigDecimal.limit);
+    ensure
+      BigDecimal.limit(old)
+    end
+  end
+
   def test_exception_nan
     _test_mode(BigDecimal::EXCEPTION_NaN) { BigDecimal.new("NaN") }
   end

--
ML: ruby-changes@q...
Info: http://www.atdot.net/~ko1/quickml/