[ruby-dev:31886] Re: [m17n] String#encode

なかだです。

At Thu, 6 Sep 2007 14:08:26 +0900,
Nobuyoshi Nakada wrote in [ruby-dev:31744]:
> なぜか"sjis"で登録されているのに"Shift_JIS"と出てきたりするので、
> 別名を付けられるようにしてrb_enc_name()にあわせてみました。

rb_enc_name()は大文字を含む名前を使っているのに、小文字でしか検
索できないというのは非常に不便なので、文字ケースは無視するように
してみました。


Index: encoding.c
===================================================================
--- encoding.c	(revision 13544)
+++ encoding.c	(working copy)
@@ -24,4 +24,5 @@ struct rb_encoding_entry {
 static struct rb_encoding_entry *enc_table;
 static int enc_table_size;
+static st_table *enc_table_alias;
 
 void
@@ -44,10 +45,22 @@ rb_enc_register(const char *name, rb_enc
 
 void
+rb_enc_alias(const char *alias, const char *orig)
+{
+    if (!enc_table_alias) {
+	enc_table_alias = st_init_strcasetable();
+    }
+    st_insert(enc_table_alias, (st_data_t)alias, (st_data_t)orig);
+}
+
+void
 rb_enc_init(void)
 {
-    rb_enc_register("ascii", ONIG_ENCODING_ASCII);
-    rb_enc_register("sjis", ONIG_ENCODING_SJIS);
-    rb_enc_register("euc-jp", ONIG_ENCODING_EUC_JP);
-    rb_enc_register("utf-8", ONIG_ENCODING_UTF8);
+#define enc_register(enc) rb_enc_register(rb_enc_name(enc), enc)
+    ENC_REGISTER(ONIG_ENCODING_ASCII);
+    ENC_REGISTER(ONIG_ENCODING_SJIS);
+    ENC_REGISTER(ONIG_ENCODING_EUC_JP);
+    ENC_REGISTER(ONIG_ENCODING_UTF8);
+    rb_enc_alias("binary", "ascii");
+    rb_enc_alias("sjis", "shift_jis");
 }
 
@@ -64,18 +77,35 @@ rb_enc_from_index(int index)
 }
 
-rb_encoding *
-rb_enc_find(const char *name)
+int
+rb_enc_find_index(const char *name)
 {
     int i;
+    st_data_t alias = 0;
 
+    if (!name) return -1;
     if (!enc_table) {
 	rb_enc_init();
     }
+  find:
     for (i=0; i<enc_table_size; i++) {
-	if (strcmp(name, enc_table[i].name) == 0) {
-	    return enc_table[i].enc;
+	if (strcasecmp(name, enc_table[i].name) == 0) {
+	    return i;
 	}
     }
-    return ONIG_ENCODING_ASCII;
+    if (!alias && enc_table_alias) {
+	if (st_lookup(enc_table_alias, (st_data_t)name, &alias)) {
+	    name = (const char *)alias;
+	    goto find;
+	}
+    }
+    return -1;
+}
+
+rb_encoding *
+rb_enc_find(const char *name)
+{
+    rb_encoding *enc = rb_enc_from_index(rb_enc_find_index(name));
+    if (!enc) enc = ONIG_ENCODING_ASCII;
+    return enc;
 }
 
@@ -164,5 +194,5 @@ rb_enc_get_index(VALUE obj)
     int i;
 
-    enc_check_capable(obj);
+    if (!enc_capable(obj)) return -1;
     i = ENCODING_GET(obj);
     if (i == ENCODING_INLINE_MAX) {
Index: st.c
===================================================================
--- st.c	(revision 13544)
+++ st.c	(working copy)
@@ -53,4 +53,10 @@ static const struct st_hash_type type_st
 };
 
+static int strcasehash(const char *);
+static const struct st_hash_type type_strcasehash = {
+    strcasecmp,
+    strcasehash,
+};
+
 static void rehash(st_table *);
 
@@ -203,4 +209,16 @@ st_init_strtable_with_size(int size)
 }
 
+st_table*
+st_init_strcasetable(void)
+{
+    return st_init_table(&type_strcasehash);
+}
+
+st_table*
+st_init_strcasetable_with_size(int size)
+{
+    return st_init_table_with_size(&type_strcasehash, size);
+}
+
 void
 st_clear(st_table *table)
@@ -815,4 +833,23 @@ strhash(register const char *string)
 }
 
+static int
+strcasehash(register const char *string)
+{
+    register unsigned int hval = FNV1_32A_INIT;
+
+    /*
+     * FNV-1a hash each octet in the buffer
+     */
+    while (*string) {
+	unsigned int c = (unsigned char)*string++;
+	if ((unsigned int)(c - 'A') > ('Z' - 'A')) c += 'a' - 'A';
+	hval ^= c;
+
+	/* multiply by the 32 bit FNV magic prime mod 2^32 */
+	hval *= FNV_32_PRIME;
+    }
+    return hval;
+}
+
 int
 st_numcmp(st_data_t x, st_data_t y)
Index: include/ruby/encoding.h
===================================================================
--- include/ruby/encoding.h	(revision 13544)
+++ include/ruby/encoding.h	(working copy)
@@ -40,6 +40,8 @@ typedef OnigEncodingType rb_encoding;
 int rb_enc_to_index(rb_encoding*);
 int rb_enc_get_index(VALUE obj);
+int rb_enc_find_index(const char *name);
 rb_encoding* rb_enc_get(VALUE);
 rb_encoding* rb_enc_check(VALUE,VALUE);
+void rb_enc_associate_index(VALUE, int);
 void rb_enc_associate(VALUE, rb_encoding*);
 void rb_enc_copy(VALUE, VALUE);
@@ -77,4 +79,5 @@ int rb_enc_codelen(int, rb_encoding*);
 #define rb_enc_prev_char(s,p,enc) (char *)onigenc_get_prev_char_head(enc,(UChar*)s,(UChar*)p)
 
+#define rb_enc_isctype(c,t,enc) ONIGENC_IS_CODE_CTYPE(enc,c,t)
 #define rb_enc_isascii(c,enc) ONIGENC_IS_CODE_ASCII(c)
 #define rb_enc_isalpha(c,enc) ONIGENC_IS_CODE_ALPHA(enc,c)
Index: include/ruby/st.h
===================================================================
--- include/ruby/st.h	(revision 13544)
+++ include/ruby/st.h	(working copy)
@@ -72,4 +72,6 @@ st_table *st_init_numtable_with_size(int
 st_table *st_init_strtable(void);
 st_table *st_init_strtable_with_size(int);
+st_table *st_init_strcasetable(void);
+st_table *st_init_strcasetable_with_size(int);
 int st_delete(st_table *, st_data_t *, st_data_t *);
 int st_delete_safe(st_table *, st_data_t *, st_data_t *, st_data_t);


-- 
--- 僕の前にBugはない。
--- 僕の後ろにBugはできる。
    中田 伸悦