Add bba #embed support

Signed-off-by: Clifford Wolf <clifford@clifford.at>

1 files changed, 26 insertions, 2 deletions

diff --git a/bba/main.cc b/bba/main.cc
index d4d16e12..5940797e 100644
--- a/bba/main.cc
+++ b/bba/main.cc
@@ -72,6 +72,7 @@ int main(int argc, char **argv)
     bool verbose = false;
     bool bigEndian;
     bool writeC = false;
+    bool writeE = false;
     char buffer[512];
 
     namespace po = boost::program_options;
@@ -82,7 +83,8 @@ int main(int argc, char **argv)
     options.add_options()("debug,d", "debug output");
     options.add_options()("be,b", "big endian");
     options.add_options()("le,l", "little endian");
-    options.add_options()("c,c", "write c strings");
+    options.add_options()("c,c", "write C strings");
+    options.add_options()("e,e", "write #embed C");
     options.add_options()("files", po::value<std::vector<std::string>>(), "file parameters");
     pos.add("files", -1);
 
@@ -115,13 +117,19 @@ int main(int argc, char **argv)
     }
     if (vm.count("c"))
         writeC = true;
+    if (vm.count("e"))
+        writeE = true;
 
+    if (writeC && writeE) {
+        printf("Incompatible modes\n");
+        exit(-1);
+    }
     if (vm.count("files") == 0) {
         printf("File parameters are mandatory\n");
         exit(-1);
     }
     std::vector<std::string> files = vm["files"].as<std::vector<std::string>>();
-    if (files.size() != 2) {
+    if (files.size() != (writeE ? 3 : 2)) {
         printf("Input and output parameters must be set\n");
         exit(-1);
     }
@@ -433,6 +441,22 @@ int main(int argc, char **argv)
 
         for (auto &s : postText)
             fprintf(fileOut, "%s\n", s.c_str());
+    } else if (writeE) {
+        for (auto &s : preText)
+            fprintf(fileOut, "%s\n", s.c_str());
+
+        fprintf(fileOut, "const char %s[%d] =\n", streams[0].name.c_str(), int(data.size()));
+        char *bin_basename_buf = strdup(files.at(2).c_str());
+        fprintf(fileOut, "#embed \"%s\"\n", basename(bin_basename_buf));
+        fprintf(fileOut, ";\n");
+
+        for (auto &s : postText)
+            fprintf(fileOut, "%s\n", s.c_str());
+
+        FILE *fileBin = fopen(files.at(2).c_str(), "wb");
+        assert(fileBin != nullptr);
+        fwrite(data.data(), int(data.size()), 1, fileBin);
+        fclose(fileBin);
     } else {
         fwrite(data.data(), int(data.size()), 1, fileOut);
     }