From 9ce8d4e3b964774294c88d9414fdf4a92c352709 Mon Sep 17 00:00:00 2001
From: akr <akr@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Fri, 2 Jan 2009 08:02:36 +0000
Subject: [PATCH] * ext/socket/mkconstants.rb: refactored to gather stringizer 
  description.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@21270 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 ChangeLog                 |  5 +++++
 ext/socket/mkconstants.rb | 47 ++++++++++++++++++++++++++++++---------
 2 files changed, 42 insertions(+), 10 deletions(-)

diff --git a/ChangeLog b/ChangeLog
index 7583c6f75a..0dde7a992c 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,8 @@
+Fri Jan  2 17:01:51 2009  Tanaka Akira  <akr@fsij.org>
+
+	* ext/socket/mkconstants.rb: refactored to gather stringizer
+	  description.
+
 Fri Jan  2 15:30:57 2009  Tanaka Akira  <akr@fsij.org>
 
 	* ext/socket/mkconstants.rb: make common prefix optional.
diff --git a/ext/socket/mkconstants.rb b/ext/socket/mkconstants.rb
index 574ad25b71..a90171277e 100644
--- a/ext/socket/mkconstants.rb
+++ b/ext/socket/mkconstants.rb
@@ -121,25 +121,56 @@ ERB.new(<<'EOS', nil, '%').def_method(Object, "gen_name_to_int(str_var, len_var,
     }
 EOS
 
-ERB.new(<<'EOS', nil, '%').def_method(Object, "gen_int_to_name_hash(hash_var, pat)")
+ERB.new(<<'EOS', nil, '%').def_method(Object, "gen_int_to_name_hash(hash_var, pat, prefix_pat)")
     <%=hash_var%> = st_init_numtable();
 % reverse_each_name(pat) {|n|
 #ifdef <%=n%>
     st_insert(<%=hash_var%>, (st_data_t)<%=n%>, (st_data_t)<%=c_str n%>);
 #endif
 % }
+% if prefix_pat
+%  reverse_each_name(pat) {|n|
+%   next if prefix_pat !~ n
+#ifdef <%=n%>
+    st_insert(<%=hash_var%>, (st_data_t)<%=n%>, (st_data_t)<%=c_str $'%>);
+#endif
+%  }
+% end
 EOS
 
-ERB.new(<<'EOS', nil, '%').def_method(Object, "gen_int_to_name(int_var, hash_var)")
+ERB.new(<<'EOS', nil, '%').def_method(Object, "gen_int_to_name_func(func_name, hash_var)")
+static char *
+<%=func_name%>(int val)
+{
     st_data_t name;
-    if (st_lookup(<%=hash_var%>, (st_data_t)<%=int_var%>, &name))
+    if (st_lookup(<%=hash_var%>, (st_data_t)val, &name))
         return (char*)name;
     return NULL;
+}
 EOS
 
+STRINGIZER_DEFS = []
+def def_stringizer(func_name, pat, prefix_optional=nil)
+  prefix_pat = nil
+  if prefix_optional
+    if Regexp === prefix_optional
+      prefix_pat = prefix_optional
+    else
+      prefix_pat = /\A#{Regexp.escape prefix_optional}/
+    end
+  end
+  hash_var = "#{func_name}_hash"
+  decl = "static st_table *#{hash_var};"
+  gen_hash = gen_int_to_name_hash(hash_var, pat, prefix_pat)
+  func = gen_int_to_name_func(func_name, hash_var)
+  STRINGIZER_DEFS << [decl, gen_hash, func]
+end
+
+def_stringizer('family_to_str',  /\AAF_/)
+
 result << ERB.new(<<'EOS', nil, '%').result(binding)
 
-static st_table *family_to_str_hash;
+<%= STRINGIZER_DEFS.map {|decl, gen_hash, func| decl }.join("\n") %>
 
 static void
 init_constants(VALUE mConst)
@@ -159,7 +190,7 @@ init_constants(VALUE mConst)
 #endif
 %   end
 % }
-<%= gen_int_to_name_hash('family_to_str_hash', /\AAF_/) %>
+<%= STRINGIZER_DEFS.map {|decl, gen_hash, func| gen_hash }.join("\n") %>
 }
 
 static int
@@ -210,11 +241,7 @@ udp_optname_to_int(char *str, int len, int *valp)
 <%= gen_name_to_int("str", "len", "valp", /\AUDP_/, "UDP_") %>
 }
 
-static char *
-family_to_str(int val)
-{
-<%= gen_int_to_name("val", "family_to_str_hash") %>
-}
+<%= STRINGIZER_DEFS.map {|decl, gen_hash, func| func }.join("\n") %>
 
 EOS