1 files changed, 38 insertions, 5 deletions

diff --git a/src/tools/bin2cpp.cpp b/src/tools/bin2cpp.cpp
index e0eab28..65e48ed 100644
--- a/src/tools/bin2cpp.cpp
+++ b/src/tools/bin2cpp.cpp
@@ -4,6 +4,7 @@
 #include <bu/streamstack.h>
 #include <bu/strfilter.h>
 #include <bu/taf.h>
+#include <bu/autoconfig.h>
 
 #ifdef BU_HAS_DEFLATE
 #include <bu/deflate.h>
@@ -18,6 +19,16 @@
 #include <bu/base64.h>
 #include <bu/hex.h>
 
+Bu::String cescape( const Bu::String &sIn )
+{
+    Bu::String sOut;
+    for( Bu::String::const_iterator i = sIn.begin(); i; i++ )
+    {
+        sOut += Bu::String("\\x%1").arg( (int)*i, Bu::Fmt(2).hex() );
+    }
+    return sOut;
+}
+
 using namespace Bu;
 
 class Options : public OptParser
@@ -223,12 +234,34 @@ int main( int argc, char *argv[] )
         << "\t{" << fSrc.nl;
 
     int idx = 0;
-    for( Bu::StringList::iterator i = opt.slInput.begin(); i; i++ )
     {
-        fSrc << "\t\tcase " << Bu::__calcHashCode( *i ) << "UL: // \"" 
-            << *i << "\"" << fSrc.nl
-            << "\t\t\treturn aFile[" << idx << "];" << fSrc.nl;
-        idx++;
+        typedef Bu::List<Bu::StringList::iterator> IterList;
+        typedef Bu::Hash<uint32_t, IterList> IterHashList;
+        typedef Bu::Hash<Bu::String, int> IdxHash;
+
+        IdxHash hIdx;
+        IterHashList hCodes;
+        for( Bu::StringList::iterator i = opt.slInput.begin(); i; i++ )
+        {
+            uint32_t uc = Bu::__calcHashCode( *i );
+            if( !hCodes.has( uc ) )
+            {
+                hCodes.insert( uc, IterList() );
+            }
+            hCodes.get( uc ).append( i );
+            hIdx.insert( *i, idx++ );
+        }
+        for( IterHashList::iterator r = hCodes.begin(); r; r++ )
+        {
+            fSrc << "\t\tcase " << r.getKey() << "UL:" << fSrc.nl;
+            for( IterList::iterator i = r.getValue().begin(); i; i++ )
+            {
+                fSrc << "\t\t\tif( sName == \"" << cescape(*(*i)) << "\" ) // "
+                    << *(*i) << fSrc.nl
+                    << "\t\t\t\treturn aFile[" << hIdx.get(*(*i)) << "];" << fSrc.nl;
+            }
+            fSrc << "\t\t\tbreak;" << fSrc.nl;
+        }
     }
     fSrc << "\t}" << fSrc.nl
         << "\tthrow Bu::ExceptionBase(\"No file matching \\\"%s\\\" found.\", sName.getStr() );" << fSrc.nl