Added assemble.c test program.

icculus · Dec 10, 2008 · d52c89c · d52c89c
1 parent 467c077
commit d52c89c
Show file tree

Hide file tree

Showing 2 changed files with 83 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -64,6 +64,8 @@ ADD_EXECUTABLE(testoutput testoutput.c)
 TARGET_LINK_LIBRARIES(testoutput mojoshader)
 ADD_EXECUTABLE(finderrors finderrors.c)
 TARGET_LINK_LIBRARIES(finderrors mojoshader ${SDL_LIBRARY})
+ADD_EXECUTABLE(assemble assemble.c)
+TARGET_LINK_LIBRARIES(assemble mojoshader)
 
 # End of CMakeLists.txt ...
 
diff --git a/assemble.c b/assemble.c
@@ -0,0 +1,81 @@
+/**
+ * MojoShader; generate shader programs from bytecode of compiled
+ *  Direct3D shaders.
+ *
+ * Please see the file LICENSE.txt in the source's root directory.
+ *
+ *  This file written by Ryan C. Gordon.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include "mojoshader.h"
+
+static int assemble(const char *buf, const char *outfile)
+{
+    FILE *io = fopen(outfile, "wb");
+    if (io == NULL)
+    {
+        printf(" ... fopen('%s') failed.\n", outfile);
+        return 0;
+    } // if
+
+    const MOJOSHADER_parseData *pd;
+    int retval = 0;
+
+    pd = MOJOSHADER_assemble(buf, NULL, NULL, NULL);
+    if (pd->error != NULL)
+        printf("ERROR: %s\n", pd->error);
+    else
+    {
+        if (pd->output != NULL)
+        {
+            if (fwrite(pd->output, pd->output_len, 1, io) != 1)
+                printf(" ... fwrite('%s') failed.\n", outfile);
+            else if (fclose(io) == EOF)
+                printf(" ... fclose('%s') failed.\n", outfile);
+            else
+                retval = 1;
+        } // if
+    } // else
+    MOJOSHADER_freeParseData(pd);
+
+    return retval;
+} // assemble
+
+
+int main(int argc, char **argv)
+{
+    int retval = 1;
+
+    if (argc != 3)
+        printf("\n\nUSAGE: %s <d3dasmfile> <outputfile>\n\n", argv[0]);
+    else
+    {
+        const char *infile = argv[1];
+        const char *outfile = argv[2];
+        FILE *io = fopen(infile, "rb");
+        if (io == NULL)
+            printf(" ... fopen('%s') failed.\n", infile);
+        else
+        {
+            char *buf = (char *) malloc(1000000);
+            int rc = fread(buf, 1, 1000000-1, io);
+            fclose(io);
+            if (rc == EOF)
+                printf(" ... fread('%s') failed.\n", infile);
+            else
+            {
+                buf[rc] = '\0';
+                if (assemble(buf, outfile))
+                    retval = 0;
+                free(buf);
+            } // else
+        } // for
+    } // else
+
+    return retval;
+} // main
+
+// end of assemble.c ...
+