/**

* 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 const char **include_paths = NULL;

static unsigned int include_path_count = 0;

#if MOJOSHADER_DEBUG_MALLOC

static void *Malloc(int len)

{

void *ptr = malloc(len + sizeof (int));

int *store = (int *) ptr;

printf("malloc() %d bytes (%p)\n", len, ptr);

if (ptr == NULL) return NULL;

*store = len;

return (void *) (store + 1);

} // Malloc

static void Free(void *_ptr)

{

int *ptr = (((int *) _ptr) - 1);

int len = *ptr;

printf("free() %d bytes (%p)\n", len, ptr);

free(ptr);

} // Free

#else

#define Malloc NULL

#define Free NULL

#endif

static void fail(const char *err)

{

printf("%s.\n", err);

exit(1);

} // fail

static int open_include(MOJOSHADER_includeType inctype, const char *fname,

const char *parent, const char **outdata,

unsigned int *outbytes, MOJOSHADER_malloc m,

MOJOSHADER_free f, void *d)

{

int i;

for (i = 0; i < include_path_count; i++)

{

const char *path = include_paths[i];

const size_t len = strlen(path) + strlen(fname) + 2;

char *buf = (char *) m(len, d);

if (buf == NULL)

return 0;

snprintf(buf, len, "%s/%s", path, fname);

FILE *io = fopen(buf, "rb");

f(buf, d);

if (io == NULL)

continue;

if (fseek(io, 0, SEEK_END) != -1)

{

const long fsize = ftell(io);

if ((fsize == -1) || (fseek(io, 0, SEEK_SET) == -1))

{

fclose(io);

return 0;

} // if

char *data = (char *) m(fsize, d);

if (data == NULL)

{

fclose(io);

return 0;

} // if

if (fread(data, fsize, 1, io) != 1)

{

f(data, d);

fclose(io);

return 0;

} // if

fclose(io);

*outdata = data;

*outbytes = (unsigned int) fsize;

return 1;

} // if

} // for

return 0;

} // open_include

static void close_include(const char *data, MOJOSHADER_malloc m,

MOJOSHADER_free f, void *d)

{

f((void *) data, d);

} // close_include

const char *outfile,

const MOJOSHADER_preprocessorDefine *defs,

{

const MOJOSHADER_preprocessData *pd;

int retval = 0;

pd = MOJOSHADER_preprocess(fname, buf, len, defs, defcount, open_include,

close_include, Malloc, Free, NULL);

if (pd->error_count > 0)

{

int i;

for (i = 0; i < pd->error_count; i++)

{

printf("%s:%d: ERROR: %s\n",

pd->errors[i].filename ? pd->errors[i].filename : "???",

pd->errors[i].error_position,

pd->errors[i].error);

} // for

} // if

else

{

if (pd->output != NULL)

{

if (fwrite(pd->output, pd->output_len, 1, io) != 1)

printf(" ... fwrite('%s') failed.\n", outfile);

printf(" ... fclose('%s') failed.\n", outfile);

else

retval = 1;

} // if

} // else

MOJOSHADER_freePreprocessData(pd);

return retval;

static int assemble(const char *fname, const char *buf, int len,

const char *outfile,

const MOJOSHADER_preprocessorDefine *defs,

unsigned int defcount, FILE *io)

{

const MOJOSHADER_parseData *pd;

int retval = 0;

pd = MOJOSHADER_assemble(fname, buf, len, NULL, 0, NULL, 0,

defs, defcount, open_include, close_include,

Malloc, Free, NULL);

if (pd->error_count > 0)

{

int i;

for (i = 0; i < pd->error_count; i++)

{

printf("%s:%d: ERROR: %s\n",

pd->errors[i].filename ? pd->errors[i].filename : "???",

pd->errors[i].error_position,

pd->errors[i].error);

} // for

} // if

else

{

if (pd->output != NULL)

{

if (fwrite(pd->output, pd->output_len, 1, io) != 1)

printf(" ... fwrite('%s') failed.\n", outfile);

else if ((outfile != NULL) && (fclose(io) == EOF))

printf(" ... fclose('%s') failed.\n", outfile);

else

retval = 1;

} // if

} // else

MOJOSHADER_freeParseData(pd);

return retval;

} // assemble

typedef enum

{

ACTION_UNKNOWN,

ACTION_PREPROCESS,

ACTION_ASSEMBLE,

} Action;

int main(int argc, char **argv)

{

const char *infile = NULL;

const char *outfile = NULL;

int i;

MOJOSHADER_preprocessorDefine *defs = NULL;

unsigned int defcount = 0;

for (i = 1; i < argc; i++)

{

const char *arg = argv[i];

if (strcmp(arg, "-P") == 0)

{

if (action != ACTION_UNKNOWN)

fail("Multiple actions specified");

action = ACTION_PREPROCESS;

} // if

else if (strcmp(arg, "-A") == 0)

{

if (action != ACTION_UNKNOWN)

fail("Multiple actions specified");

action = ACTION_ASSEMBLE;

} // else if

else if (strcmp(arg, "-o") == 0)

{

if (outfile != NULL)

arg = argv[++i];

if (arg == NULL)

outfile = arg;

} // if

{

arg = argv[++i];

if (arg == NULL)

fail("no path after '-I'");

include_paths = (const char **) realloc(include_paths,

(include_path_count+1) * sizeof (char *));

include_paths[include_path_count] = arg;

include_path_count++;

} // if

else if (strncmp(arg, "-D", 2) == 0)

{

arg += 2;

char *ident = strdup(arg);

char *ptr = strchr(ident, '=');

const char *val = "";

if (ptr)

{

*ptr = '\0';

val = ptr+1;

} // if

defs = (MOJOSHADER_preprocessorDefine *) realloc(defs,

(defcount+1) * sizeof (MOJOSHADER_preprocessorDefine));

defs[defcount].identifier = ident;

defs[defcount].definition = val;

defcount++;

} // else if

else

{

if (infile != NULL)

infile = arg;

} // else

} // for

if (action == ACTION_UNKNOWN)

action = ACTION_ASSEMBLE;

FILE *io = fopen(infile, "rb");

if (io == NULL)

fail("failed to open input file");

fseek(io, 0, SEEK_END);

long fsize = ftell(io);

fseek(io, 0, SEEK_SET);

if (fsize == -1)

fsize = 1000000;

char *buf = (char *) malloc(fsize);

const int rc = fread(buf, 1, fsize, io);

fclose(io);

if (rc == EOF)

fail("failed to read input file");

FILE *outio = outfile ? fopen(outfile, "wb") : stdout;

if (outio == NULL)

fail("failed to open output file");

if (action == ACTION_PREPROCESS)

retval = (!preprocess(infile, buf, rc, outfile, defs, defcount, outio));

else if (action == ACTION_ASSEMBLE)

retval = (!assemble(infile, buf, rc, outfile, defs, defcount, outio));

if ((retval != 0) && (outfile != NULL))

remove(outfile);

free(buf);

free(defs);

free(include_paths);

return retval;

} // main