[macruby-changes] [3690] MacRuby/branches/icu/string.c

[source_changes at macosforge.org]

Revision: 3690
          http://trac.macosforge.org/projects/ruby/changeset/3690
Author:   lsansonetti at apple.com
Date:     2010-03-03 19:07:26 -0800 (Wed, 03 Mar 2010)
Log Message:
-----------
added #transform, an extension to apply ICU transformations

Modified Paths:
--------------
    MacRuby/branches/icu/string.c

Modified: MacRuby/branches/icu/string.c
===================================================================
--- MacRuby/branches/icu/string.c	2010-03-04 01:11:53 UTC (rev 3689)
+++ MacRuby/branches/icu/string.c	2010-03-04 03:07:26 UTC (rev 3690)
@@ -23,6 +23,7 @@
 #include "vm.h"
 
 #include <unicode/unum.h>
+#include <unicode/utrans.h>
 
 VALUE rb_cString;
 VALUE rb_cNSString;
@@ -4343,6 +4344,61 @@
     return beg;
 }
 
+// :nodoc
+static VALUE
+rstr_transform(VALUE str, SEL sel, VALUE transform_pat)
+{
+    StringValue(transform_pat);
+
+    UChar *new_chars = NULL;
+    long new_chars_len = 0;
+    bool need_free = false;
+    rb_str_get_uchars(str, &new_chars, &new_chars_len, &need_free);
+
+    if (new_chars_len == 0) {
+	return Qnil;
+    }
+
+    if (!need_free) {
+	UChar *tmp = (UChar *)malloc(sizeof(UChar) * new_chars_len);
+	memcpy(tmp, new_chars, sizeof(UChar) * new_chars_len);
+	new_chars = tmp;
+    }
+
+    UChar *transform_chars = NULL;
+    long transform_chars_len = 0;
+    need_free = false;
+    rb_str_get_uchars(transform_pat, &transform_chars, &transform_chars_len,
+	    &need_free);
+
+    UErrorCode status = U_ZERO_ERROR;
+    UTransliterator *trans = utrans_openU(transform_chars, transform_chars_len,
+	    UTRANS_FORWARD, NULL, 0, NULL, &status);
+
+    if (trans == NULL) {
+	if (need_free) {
+	    free(transform_chars);
+	}
+	rb_raise(rb_eArgError, "cannot create transliterator");
+    }
+
+    int32_t capacity = (int32_t)new_chars_len;
+    int32_t limit = capacity;
+    utrans_transUChars(trans, new_chars, &capacity, capacity,
+	    0, &limit, &status);
+
+    new_chars_len = (long)capacity;
+
+    VALUE newstr = rb_unicode_str_new(new_chars, new_chars_len);
+
+    if (need_free) {
+	free(transform_chars);
+    }
+    free(new_chars);
+
+    return newstr;
+}
+
 // NSString primitives.
 
 static void
@@ -4512,7 +4568,10 @@
     rb_objc_define_method(rb_cRubyString, "next!", rstr_succ_bang, 0);
     rb_objc_define_method(rb_cRubyString, "upto", rstr_upto, -1);
 
-    // Added for MacRuby (debugging).
+    // MacRuby extensions.
+    rb_objc_define_method(rb_cRubyString, "transform", rstr_transform, 1);
+
+    // MacRuby extensions (debugging).
     rb_objc_define_method(rb_cRubyString, "__chars_count__",
 	    rstr_chars_count, 0);
     rb_objc_define_method(rb_cRubyString, "__getchar__", rstr_getchar, 1);
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://lists.macosforge.org/pipermail/macruby-changes/attachments/20100303/6ba5186b/attachment.html>