* numeric.c (int_chr): take an optional encoding parameter.

  [ruby-core:12816]


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

1 files changed, 30 insertions, 10 deletions

diff --git a/numeric.c b/numeric.c
index 4e30d4e23b..dbe273bbcf 100644
--- a/numeric.c
+++ b/numeric.c
@@ -11,6 +11,7 @@
 **********************************************************************/
 
 #include "ruby/ruby.h"
+#include "ruby/encoding.h"
 #include <ctype.h>
 #include <math.h>
 #include <stdio.h>
@@ -1802,25 +1803,44 @@ int_pred(VALUE num)
 
 /*
  *  call-seq:
- *     int.chr    => string
+ *     int.chr([encoding])    => string
  *
- *  Returns a string containing the ASCII character represented by the
- *  receiver's value.
+ *  Returns a string containing the character represented by the
+ *  receiver's value according to +encoding+.
  *
  *     65.chr    #=> "A"
  *     230.chr   #=> "\346"
+ *     255.chr(Encoding::UTF_8)   #=> "\303\277"
  */
 
 static VALUE
-int_chr(VALUE num)
+int_chr(int argc, VALUE *argv, VALUE num)
 {
     char c;
+    int n;
     long i = NUM2LONG(num);
-
-    if (i < 0 || 0xff < i)
-	rb_raise(rb_eRangeError, "%ld out of char range", i);
-    c = i;
-    return rb_str_new(&c, 1);
+    rb_encoding *enc;
+    VALUE str;
+
+    switch (argc) {
+      case 0:
+	if (i < 0 || 0xff < i) {
+	  out_of_range:
+	    rb_raise(rb_eRangeError, "%ld out of char range", i);
+	}
+	c = i;
+	return rb_str_new(&c, 1);
+      case 1:
+	break;
+      default:
+	rb_raise(rb_eArgError, "wrong number of arguments (%d for 0 or 1)", argc);
+	break;
+    }
+    enc = rb_to_encoding(argv[0]);
+    if (i < 0 || (n = rb_enc_codelen(i, enc)) <= 0) goto out_of_range;
+    str = rb_enc_str_new(0, n, enc);
+    rb_enc_mbcput(i, RSTRING_PTR(str), enc);
+    return str;
 }
 
 /********************************************************************
@@ -3081,7 +3101,7 @@ Init_Numeric(void)
     rb_define_method(rb_cInteger, "succ", int_succ, 0);
     rb_define_method(rb_cInteger, "next", int_succ, 0);
     rb_define_method(rb_cInteger, "pred", int_pred, 0);
-    rb_define_method(rb_cInteger, "chr", int_chr, 0);
+    rb_define_method(rb_cInteger, "chr", int_chr, -1);
     rb_define_method(rb_cInteger, "to_i", int_to_i, 0);
     rb_define_method(rb_cInteger, "to_int", int_to_i, 0);
     rb_define_method(rb_cInteger, "floor", int_to_i, 0);