#include <stdio.h>

typedef unsigned char Uint8;
typedef signed char Sint8;
typedef unsigned short Uint16;

typedef struct {
	Uint8 dat[0x100], ptr;
} Stack;

typedef struct Uxn {
	Uint8 ram[0x10000], dev[0x100];
	Stack wst, rst;
} Uxn;

Uxn uxn;

Uint8
emu_dei(Uint8 addr)
{
	return uxn.dev[addr];
}

void
emu_deo(Uint8 addr, Uint8 value)
{
	uxn.dev[addr] = value;
	switch(addr) {
	case 0x18: fputc(uxn.dev[0x18], stdout), fflush(stdout); return;
	case 0x19: fputc(uxn.dev[0x19], stderr), fflush(stderr); return;
	}
}

#define FLP { s = ins & 0x40 ? &uxn.wst : &uxn.rst; }
#define JMP(x) { if(_2) pc = (x); else pc += (Sint8)(x); }
#define PO1(o) { o = s->dat[--*sp]; }
#define PO2(o) { o = s->dat[--*sp] | (s->dat[--*sp] << 0x8); }
#define POx(o) { if(_2) { PO2(o) } else PO1(o) }
#define PU1(y) { s->dat[s->ptr++] = (y); }
#define PU2(y) { tt = (y); s->dat[s->ptr++] = tt >> 0x8; s->dat[s->ptr++] = tt; }
#define PUx(y) { if(_2) { PU2(y) } else PU1(y) }
#define PEK(o, x, r) { if(_2) { r = (x); o = uxn.ram[r++] << 8 | uxn.ram[r]; } else o = uxn.ram[(x)]; }
#define POK(x, y, r) { if(_2) { r = (x); uxn.ram[r++] = y >> 8; uxn.ram[r] = y; } else uxn.ram[(x)] = (y); }
#define DEI(o, p) { if(_2) { o = (emu_dei(p) << 8) | emu_dei(p + 1); } else o = emu_dei(p); }
#define DEO(p, y) { if(_2) { emu_deo(p, y >> 8); emu_deo(p + 1, y); } else emu_deo(p, y); }

int
uxn_eval(Uint16 pc)
{
	if(!pc || uxn.dev[0x0f]) return 0;
	for(;;) {
		Uint16 tt, a, b, c;
		Uint8 t, kp, *sp, ins = uxn.ram[pc++];
		/* 2 */ Uint8 _2 = ins & 0x20;
		/* r */ Stack *s = ins & 0x40 ? &uxn.rst : &uxn.wst;
		/* k */ if(ins & 0x80) kp = s->ptr, sp = &kp; else sp = &s->ptr;
		switch(ins & 0x1f) {
		case 0x00:
			switch(ins) {
			/* BRK */ case 0x00: return 1;
			/* JCI */ case 0x20: PO1(b) if(b) { goto JMI; } pc += 2; break;
			/* JMI */ case 0x40: JMI: pc += uxn.ram[pc++] << 8 | uxn.ram[pc++]; break;
			/* JSI */ case 0x60: PU2(pc + 2) goto JMI;
			/* LI2 */ case 0xa0: case 0xe0: PU1(uxn.ram[pc++]) /* fall */
			/* LIT */ case 0x80: case 0xc0: PU1(uxn.ram[pc++]) break;
			} break;
		/* INC */ case 0x01: POx(a) PUx(a + 1) break;
		/* POP */ case 0x02: POx(a) break;
		/* NIP */ case 0x03: POx(a) POx(b) PUx(a) break;
		/* SWP */ case 0x04: POx(a) POx(b) PUx(a) PUx(b) break;
		/* ROT */ case 0x05: POx(a) POx(b) POx(c) PUx(b) PUx(a) PUx(c) break;
		/* DUP */ case 0x06: POx(a) PUx(a) PUx(a) break;
		/* OVR */ case 0x07: POx(a) POx(b) PUx(b) PUx(a) PUx(b) break;
		/* EQU */ case 0x08: POx(a) POx(b) PU1(b == a) break;
		/* NEQ */ case 0x09: POx(a) POx(b) PU1(b != a) break;
		/* GTH */ case 0x0a: POx(a) POx(b) PU1(b > a) break;
		/* LTH */ case 0x0b: POx(a) POx(b) PU1(b < a) break;
		/* JMP */ case 0x0c: POx(a) JMP(a) break;
		/* JCN */ case 0x0d: POx(a) PO1(b) if(b) JMP(a) break;
		/* JSR */ case 0x0e: POx(a) FLP PU2(pc) JMP(a) break;
		/* STH */ case 0x0f: POx(a) FLP PUx(a) break;
		/* LDZ */ case 0x10: PO1(a) PEK(b, a, t) PUx(b) break;
		/* STZ */ case 0x11: PO1(a) POx(b) POK(a, b, t) break;
		/* LDR */ case 0x12: PO1(a) PEK(b, pc + (Sint8)a, tt) PUx(b) break;
		/* STR */ case 0x13: PO1(a) POx(b) POK(pc + (Sint8)a, b, tt) break;
		/* LDA */ case 0x14: PO2(a) PEK(b, a, tt) PUx(b) break;
		/* STA */ case 0x15: PO2(a) POx(b) POK(a, b, tt) break;
		/* DEI */ case 0x16: PO1(a) DEI(b, a) PUx(b) break;
		/* DEO */ case 0x17: PO1(a) POx(b) DEO(a, b) break;
		/* ADD */ case 0x18: POx(a) POx(b) PUx(b + a) break;
		/* SUB */ case 0x19: POx(a) POx(b) PUx(b - a) break;
		/* MUL */ case 0x1a: POx(a) POx(b) PUx(b * a) break;
		/* DIV */ case 0x1b: POx(a) POx(b) PUx(a ? b / a : 0) break;
		/* AND */ case 0x1c: POx(a) POx(b) PUx(b & a) break;
		/* ORA */ case 0x1d: POx(a) POx(b) PUx(b | a) break;
		/* EOR */ case 0x1e: POx(a) POx(b) PUx(b ^ a) break;
		/* SFT */ case 0x1f: PO1(a) POx(b) PUx(b >> (a & 0xf) << (a >> 4)) break;
		}
	}
}

void
console_input(char c, int type)
{
	uxn.dev[0x12] = c, uxn.dev[0x17] = type;
	uxn_eval(uxn.dev[0x10] << 8 | uxn.dev[0x11]);
}

int
main(int argc, char **argv)
{
	FILE *f;
	int i = 1;
	if(argc < 2) {
		fprintf(stdout, "usage: %s file.rom [args..]\n", argv[0]);
		return 0;
	}
	f = fopen(argv[i++], "rb");
	if(!f) {
		fprintf(stderr, "uxnmin: Error %s\n", argv[1]);
		return 0;
	}
	uxn.dev[0x17] = argc - i;
	fread(&uxn.ram[0x0100], 0xff00, 1, f), fclose(f);
	if(uxn_eval(0x0100) && (uxn.dev[0x10] << 8 | uxn.dev[0x11])) {
		for(; i < argc; i++) {
			char *p = argv[i];
			while(*p) console_input(*p++, 0x2);
			console_input('\n', i == argc - 1 ? 0x4 : 0x3);
		}
		while(!uxn.dev[0x0f]) {
			int c = fgetc(stdin);
			if(c == EOF) {
				console_input(0x00, 0x4);
				break;
			}
			console_input((Uint8)c, 0x1);
		}
	}
	return uxn.dev[0x0f] & 0x7f;
}