/* dla2csv.c - convert dla.xex save files to CSV.
	Rather bloated and slow by Atari standards. */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>

#include "dlaver.h"

#define HEIGHT 170
#define WIDTH 256

/* cc65 doesn't fold constants, it won't let us do this:
	#define INBUF_SIZE (WIDTH * HEIGHT)
	...it has to be a *constant*. */
#define INBUF_SIZE 5440

#define STRINGBUF_SIZE 256

/* cc65 doesn't "localize" \b to the Atari backspace character, so: */
#ifdef __ATARI__
	#define BS '\x7e'
#else
	#define BS '\b'
#endif

char inbuf[INBUF_SIZE];
char stringbuf[STRINGBUF_SIZE];
FILE *inf, *outf;

/* don't use "\n" or "\r\n" on the right hand side, use explicit
	hex codes, to avoid cc65 tampering with them. */
char *eoltypes[][2] = {
	{ "Atari ($9B)", "\x9b" },
	{ "Unix (\\n)", "\x0a" },
	{ "MS (\\r\\n)", "\x0d\x0a" },
	{ NULL, NULL }
};

/* read a string from stdin (E: on the Atari).
	exit on EOF (or Break key on the Atari).
	on DOS 2.0S, we end up back at the DUP menu... but
	SpartaDOS 3.2d locks up with a black screen. so don't
	suggest using this...
 */
void readstring(void) {
	memset(stringbuf, 0, STRINGBUF_SIZE);
	if(fgets(stringbuf, STRINGBUF_SIZE - 1, stdin) == NULL)
		exit(1);
}

/* prompt for a filename, try to open it. if there's an error,
	show error message and retry. will not return until it
	opens the file. */
FILE *prompt_filename(const char *name, const char *mode) {
	FILE *f = NULL;
	putchar('\n');
	while(f == NULL) {
		printf("%s file: ", name);
		fflush(stdout);
		readstring();
		stringbuf[strlen(stringbuf) - 1] = '\0'; /* kill trailing \n */
		if(strlen(stringbuf) == 0) continue;
#ifdef __ATARI__
		/* if there's no device spec (D: or D1: etc), prepend D: */
		if(!strchr(stringbuf, ':')) {
			memmove(stringbuf+2, stringbuf, strlen(stringbuf) + 1);
			stringbuf[0] = 'D';
			stringbuf[1] = ':';
		}
#endif
		f = fopen(stringbuf, mode);
		if(!f) perror(stringbuf);
	}
	return f;
}

int prompt_yn(char *prompt, int default_y) {
	char *yn = "y/N";
	if(default_y) yn = "Y/n";

	printf("%s[%s]? ", prompt, yn);
	fflush(stdout);
	readstring();

	switch(stringbuf[0]) {
		case 'y':
		case 'Y':
			return 1;
			break;
		case 'n':
		case 'N':
			return 0;
			break;
		default:
			break;
	}
	return default_y;
}

/* prompt for and read EOL type, retry if needed. will not return
	until a valid number was entered. */
char *prompt_eol(void) {
	int i;

	putchar('\n');
	for(i = 0; eoltypes[i][0] != NULL; i++) {
		printf("%d:%s  ", i + 1, eoltypes[i][0]);
	}
	putchar('\n');

	i = -1;
	while(i == -1) {
		printf("Line ending type[1]? ");
		fflush(stdout);
		readstring();
		if(stringbuf[0] == '\n') {
			i = 0;
		} else {
			i = stringbuf[0] - 49; /* ascii 1-3 => 0-2 */
			if(i < 0 || i > 2)
				i = -1;
		}
	}
	return eoltypes[i][1];
}

void backspace3(void) {
	putchar(BS);
	putchar(BS);
	putchar(BS);
}

int main(int argc, char **argv) {
	char *inp, *eol;
	int i, bytes = 0, x, y, xmask, err;

#ifdef __ATARI__
	/* cc65's startup code turns off caps lock. turn it back on,
		since we're typing DOS filenames. in BASIC this would be:
		POKE 702,64 */
	*((char *)0x02be) = 0x40;

	/* clear the screen */
	putchar(0x7d);
#endif

	printf("DLA to CSV converter v" VERSION ".\n");

	while(1) {
		err = 0;

		/* in case of short read on 2nd or later conversion: */
		memset(inbuf, 0, INBUF_SIZE);

		/* read whole input file into memory */
		inf = prompt_filename("Input DLA", "rb");
		printf("Reading...");
		fflush(stdout);
		bytes = fread(inbuf, 1, INBUF_SIZE, inf);
		if(bytes <= 0) {
			perror(stringbuf);
			continue;
		}
		printf("Read %d bytes.\n", bytes);

		/* warn if the file size is wrong... */
		if(bytes < INBUF_SIZE) {
			printf("Warning: File too short!\n");
			err = 1;
		} else if(fgetc(inf) != EOF) {
			printf("Warning: File too long!\n");
			err = 1;
		} else {
			printf("File size is OK.\n");
		}
		fclose(inf);

		/* a DLA file never has non-zero bytes at the beginning. */
		for(i = 0; i < 32; i++) {
			if(inbuf[i]) {
				printf("Warning: File doesn't look like a DLA!\n");
				err = 1;
				break;
			}
		}

		/* if anything looks wrong, we shouldn't proceed... but the user
			is boss, so give him the choice to shoot himself in the foot. */
		if(err) {
			if(!prompt_yn("Try to convert anyway", 0))
				continue;
		}

		eol = prompt_eol();

		/* convert in a loop. in case of write error, this allows retry
			without re-reading the input file. */
		i = err = 0;
		while(!i) {
			outf = prompt_filename("Output CSV", "wb");

			printf("\nConverting...   ");
			fflush(stdout);

			/* CSV file header row (column names) */
			fprintf(outf, "x,y%s", eol);

			/* write output file one line at a time */
			inp = inbuf;
			xmask = 0x80;
			bytes = 0;
			for(y = 0; y < HEIGHT; y++) {
				backspace3();
				printf("%02d%%", y * 100 / HEIGHT); /* percentage */
				fflush(stdout);
				for(x = 0; x < WIDTH; x++) {
					if(*inp & xmask) {
						if(fprintf(outf, "%d,%d%s", x, y, eol) < 0) {
							#ifdef __ATARI__
							/* dirty hack alert: cc65 gives us the
								wrong error here, when the disk fills up. */
							if(errno == EBADF) errno = ENOSPC;
							#endif
							printf("errno is %d\n", errno);
							perror(stringbuf);
							err = 1;
							break;
						}
						bytes++;
					}
					xmask >>= 1;
					if(!xmask) {
						xmask = 0x80;
						inp++;
					}
				}
				if(err) break;
			}
			fclose(outf);
			if(err) {
				if(!prompt_yn("Conversion failed, try again", 1))
					i = 1;
			} else {
				i = 1;
				backspace3();
				printf("100%%\n%d particles.\n", bytes);
			}
		}

		if(!prompt_yn("\nConvert another file", 1))
			return 0;
	}
}