#include <stdio.h>
#include <fcntl.h>
#include <string.h>
#ifdef WIN32
	#include <io.h>
#endif

static unsigned char buf[1024];

static char *filenameof(char *fname) {
	char *p;

#ifdef WIN32
	if (fname[1] == ':')
		fname = fname+2;
	p = strrchr(fname, '\\');
	if (p) fname = p+1;
#endif
	p = strrchr(fname, '/');
	if (p) fname = p+1;
	p = strchr(fname, '.');
	if (p) *p = 0;
	return fname;
}

int main(int argc, char * argv[])
{
char *		opt_progname;
char *		opt_inputfile;
char *		opt_outputfile;
char *		opt_arrayname;
int			opt_breakblocks;
char *		opt_static;
char *		opt_const;
FILE *		f_input;
FILE *		f_output;
unsigned	blocknum;
size_t		len;
size_t		i;

	/* Default values for our parameters */
	opt_progname = argv[0];
	opt_inputfile = 0;
	opt_outputfile = 0;
	opt_arrayname = 0;
	opt_breakblocks = 0;
	opt_static = "";
	opt_const = "";

	/* Read the arguments */
	while(*++argv) {
		if (argv[0][0] == '-') {
			while (*++(argv[0])) {
				switch(argv[0][0]) {
				case 'b':		opt_breakblocks = 1;		break;
				case 'c':		opt_const = "const ";		break;
				case 's':		opt_static = "static ";		break;
				case 'n':		opt_arrayname = *++argv;	goto nextarg;
				default:
					fprintf(stderr, "Unknown flag -%c\n", argv[0][0]);
					goto usage;
				}
			}
		} else if (!opt_inputfile)
			opt_inputfile = argv[0];
		else if (!opt_outputfile)
			opt_outputfile = argv[0];
		else {
			usage:
			fprintf(stderr, "Usage:\n\t%s -?\n"
							"\t%s [-bs] [-n name] [inputfile] [outputfile]\n"
							"\t\t-?\tThis help\n"
							"\t\t-b\tBreak the arrays for compilers that won't handle large arrays\n"
							"\t\t-c\tDeclare the arrays as const (useful to ensure they end up in Flash)\n"
							"\t\t-s\tDeclare the arrays as static\n"
							"\t\t-n name\tUse \"name\" as the name of the array\n"
					, opt_progname, opt_progname);
			return 1;
		}
	nextarg:	;
	}

	/* Open the input file */
	if (opt_inputfile) {
		f_input = fopen(opt_inputfile,
#ifdef WIN32
			"rb");
#else
			"r");
#endif
		if (!f_input) {
			fprintf(stderr, "Could not open input file '%s'\n", opt_inputfile);
			goto usage;
		}
	} else {
		f_input = stdin;
#ifdef WIN32
		_setmode(_fileno(stdin), _O_BINARY);
#endif
	}

	/* Open the output file */
	if (opt_outputfile) {
		f_output = fopen(opt_outputfile, "w");
		if (!f_output) {
			fprintf(stderr, "Could not open output file '%s'\n", opt_outputfile);
			goto usage;
		}
	} else
		f_output = stdout;

	/* Set the array name */
	if (!opt_arrayname) {
		if (opt_inputfile)
			opt_arrayname = filenameof(opt_inputfile);
		if (!opt_arrayname || !opt_arrayname[0])
			opt_arrayname = "filearray";
	}

	/* Read the file processing 1K at a time */
	blocknum = 0;
	while((len = fread(buf, 1, sizeof(buf), f_input))) {
		if (!blocknum++)
			fprintf(f_output, "%s%sunsigned char %s[] = {", opt_static, opt_const, opt_arrayname);
		else if (opt_breakblocks)
			fprintf(f_output, "\n};\n%s%sunsigned char %s_p%u[] = {", opt_static, opt_const, opt_arrayname, blocknum);
		for(i = 0; i < len; i++) {
			fprintf(f_output, (i & 0x0F) ? ", 0x%02X" : "\n\t0x%02X", buf[i]);
		}
	}
	fprintf(f_output, "\n};\n");

	/* Clean up */
	if (ferror(f_input))
		fprintf(stderr, "Input file read error\n");
	if (ferror(f_output))
		fprintf(stderr, "Output file write error - disk full?\n");
	if (f_input != stdin)
		fclose(f_input);
	if (f_output != stdout)
		fclose(f_output);

	return 0;
}