juce/source/extras/binarybuilder/Source/Main.cpp

   1 /*
   2   ==============================================================================
   3
   4    Utility to turn a bunch of binary files into a .cpp file and .h file full of
   5    data so they can be built directly into an executable.
   6
   7    Use this code at your own risk! It carries no warranty!
   8
   9   ==============================================================================
  10 */
  11
  12 #include "../JuceLibraryCode/JuceHeader.h"
  13
  14
  15 //==============================================================================
  16 static int addFile (const File& file,
  17                     const String& classname,
  18                     OutputStream& headerStream,
  19                     OutputStream& cppStream)
  20 {
  21     MemoryBlock mb;
  22     file.loadFileAsData (mb);
  23
  24     const String name (file.getFileName().toLowerCase()
  25                            .replaceCharacter (' ', '_')
  26                            .replaceCharacter ('.', '_')
  27                            .retainCharacters ("abcdefghijklmnopqrstuvwxyz_0123456789"));
  28
  29     std::cout << "Adding " << name << ": "
  30               << (int) mb.getSize() << " bytes" << std::endl;
  31
  32     headerStream << "    extern const char*  " << name << ";\r\n"
  33                     "    const int           " << name << "Size = "
  34                  << (int) mb.getSize() << ";\r\n\r\n";
  35
  36     static int tempNum = 0;
  37
  38     cppStream << "static const unsigned char temp" << ++tempNum << "[] = {";
  39
  40     size_t i = 0;
  41     const uint8* const data = (const uint8*) mb.getData();
  42
  43     while (i < mb.getSize() - 1)
  44     {
  45         if ((i % 40) != 39)
  46             cppStream << (int) data[i] << ",";
  47         else
  48             cppStream << (int) data[i] << ",\r\n  ";
  49
  50         ++i;
  51     }
  52
  53     cppStream << (int) data[i] << ",0,0};\r\n";
  54
  55     cppStream << "const char* " << classname << "::" << name
  56               << " = (const char*) temp" << tempNum << ";\r\n\r\n";
  57
  58     return mb.getSize();
  59 }
  60
  61 static bool isHiddenFile (const File& f, const File& root)
  62 {
  63     return f.getFileName().endsWithIgnoreCase (".scc")
  64          || f.getFileName() == ".svn"
  65          || f.getFileName().startsWithChar ('.')
  66          || (f.getSize() == 0 && ! f.isDirectory())
  67          || (f.getParentDirectory() != root && isHiddenFile (f.getParentDirectory(), root));
  68 }
  69
  70
  71 //==============================================================================
  72 int main (int argc, char* argv[])
  73 {
  74     std::cout << "\n BinaryBuilder! Copyright 2007 by Julian Storer - www.rawmaterialsoftware.com\n\n";
  75
  76     if (argc < 4 || argc > 5)
  77     {
  78         std::cout << " Usage: BinaryBuilder  sourcedirectory targetdirectory targetclassname [optional wildcard pattern]\n\n"
  79                      " BinaryBuilder will find all files in the source directory, and encode them\n"
  80                      " into two files called (targetclassname).cpp and (targetclassname).h, which it\n"
  81                      " will write into the target directory supplied.\n\n"
  82                      " Any files in sub-directories of the source directory will be put into the\n"
  83                      " resultant class, but #ifdef'ed out using the name of the sub-directory (hard to\n"
  84                      " explain, but obvious when you try it...)\n";
  85
  86         return 0;
  87     }
  88
  89     const File sourceDirectory (File::getCurrentWorkingDirectory()
  90                                      .getChildFile (String (argv[1]).unquoted()));
  91
  92     if (! sourceDirectory.isDirectory())
  93     {
  94         std::cout << "Source directory doesn't exist: "
  95                   << sourceDirectory.getFullPathName()
  96                   << std::endl << std::endl;
  97
  98         return 0;
  99     }
 100
 101     const File destDirectory (File::getCurrentWorkingDirectory()
 102                                    .getChildFile (String (argv[2]).unquoted()));
 103
 104     if (! destDirectory.isDirectory())
 105     {
 106         std::cout << "Destination directory doesn't exist: "
 107                   << destDirectory.getFullPathName() << std::endl << std::endl;
 108
 109         return 0;
 110     }
 111
 112     String className (argv[3]);
 113     className = className.trim();
 114
 115     const File headerFile (destDirectory.getChildFile (className).withFileExtension (".h"));
 116     const File cppFile    (destDirectory.getChildFile (className).withFileExtension (".cpp"));
 117
 118     std::cout << "Creating " << headerFile.getFullPathName()
 119               << " and " << cppFile.getFullPathName()
 120               << " from files in " << sourceDirectory.getFullPathName()
 121               << "..." << std::endl << std::endl;
 122
 123     Array <File> files;
 124     sourceDirectory.findChildFiles (files, File::findFiles, true,
 125                                     (argc > 4) ? argv[4] : "*");
 126
 127     if (files.size() == 0)
 128     {
 129         std::cout << "Didn't find any source files in: "
 130                   << sourceDirectory.getFullPathName() << std::endl << std::endl;
 131         return 0;
 132     }
 133
 134     headerFile.deleteFile();
 135     cppFile.deleteFile();
 136
 137     ScopedPointer <OutputStream> header (headerFile.createOutputStream());
 138
 139     if (header == 0)
 140     {
 141         std::cout << "Couldn't open "
 142                   << headerFile.getFullPathName() << " for writing" << std::endl << std::endl;
 143         return 0;
 144     }
 145
 146     ScopedPointer <OutputStream> cpp (cppFile.createOutputStream());
 147
 148     if (cpp == 0)
 149     {
 150         std::cout << "Couldn't open "
 151                   << cppFile.getFullPathName() << " for writing" << std::endl << std::endl;
 152         return 0;
 153     }
 154
 155     *header << "/* (Auto-generated binary data file). */\r\n\r\n"
 156                "#ifndef BINARY_" << className.toUpperCase() << "_H\r\n"
 157                "#define BINARY_" << className.toUpperCase() << "_H\r\n\r\n"
 158                "namespace " << className << "\r\n"
 159                "{\r\n";
 160
 161     *cpp << "/* (Auto-generated binary data file). */\r\n\r\n"
 162             "#include \"" << className << ".h\"\r\n\r\n";
 163
 164     int totalBytes = 0;
 165
 166     for (int i = 0; i < files.size(); ++i)
 167     {
 168         const File file (files[i]);
 169
 170         // (avoid source control files and hidden files..)
 171         if (! isHiddenFile (file, sourceDirectory))
 172         {
 173             if (file.getParentDirectory() != sourceDirectory)
 174             {
 175                 *header << "  #ifdef " << file.getParentDirectory().getFileName().toUpperCase() << "\r\n";
 176                 *cpp << "#ifdef " << file.getParentDirectory().getFileName().toUpperCase() << "\r\n";
 177
 178                 totalBytes += addFile (file, className, *header, *cpp);
 179
 180                 *header << "  #endif\r\n";
 181                 *cpp << "#endif\r\n";
 182             }
 183             else
 184             {
 185                 totalBytes += addFile (file, className, *header, *cpp);
 186             }
 187         }
 188     }
 189
 190     *header << "};\r\n\r\n"
 191                "#endif\r\n";
 192
 193     header = 0;
 194     cpp = 0;
 195
 196     std::cout << std::endl << " Total size of binary data: " << totalBytes << " bytes" << std::endl;
 197
 198     return 0;
 199 }