#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "instruktion.c"
#include "stack.c"
#include "program.c"
#include "codeReader.c"
#include "SDA.c"
#include "debugMenu.c"
#include "bigint.h"
#include "record.c"

// Debug
int debug = 0;

// Stack
struct stack stack;
#define SIZE 64

//Register
struct stack reg;

// Program
struct program program;

// SDA
struct sda sda;
unsigned fp;

void version(void) {
    printf("Ninja Virtual Machine version %i (compiled %s, %s)\n", 0, __DATE__, __TIME__);
}

void help(void) {
    printf("Usage: ./njvm [options] <code file>\n\t--debug\tstart virtual machine in debug mode\n\t--version\tshow version and exit\n\t--help\t\tshow this help and exit\n");
}

void execute(struct program program) {
    int i;
    char charInput;
    StackSlot tempSlot;
    ObjRef tempObj;
    ObjRef tempObj2;
    int tempInt;
    for (i = 0; i < *program.size; ++i) {
        if (debug == 1) debugMenu(fp, stack, &debug, i);
        switch (program.program[i] >> 24) {
            case HALT:
                if (debug == 1) printf("halt\n");
                goto end;
            case PUSHC:
                if (debug == 1) printf("pushc: %i\n", IMMEDIATE(program.program[i]));
                bigFromInt(SIGN_EXTEND(IMMEDIATE(program.program[i])));
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case ADD:
                if (debug == 1) printf("add: %i + %i\n", peek(stack, 2), peek(stack, 1));
                bip.op2 = pop(stack).u.objRef;
                bip.op1 = pop(stack).u.objRef;
                bigAdd();
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case SUB:
                if (debug == 1) printf("sub: %i - %i\n", peek(stack, 2), peek(stack, 1));
                bip.op2 = pop(stack).u.objRef;
                bip.op1 = pop(stack).u.objRef;
                bigSub();
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case MUL:
                if (debug == 1) printf("mul: %i * %i\n", peek(stack, 2), peek(stack, 1));
                bip.op2 = pop(stack).u.objRef;
                bip.op1 = pop(stack).u.objRef;
                bigMul();
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case DIV:
                if (debug == 1) printf("div: %i / %i\n", peek(stack, 2), peek(stack, 1));
                bip.op2 = pop(stack).u.objRef;
                bip.op1 = pop(stack).u.objRef;
                bigDiv();
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case MOD:
                if (debug == 1) printf("mod: %i %% %i\n", peek(stack, 2), peek(stack, 1));
                bip.op2 = pop(stack).u.objRef;
                bip.op1 = pop(stack).u.objRef;
                bigDiv();
                push(stack, stackSlotWithObjRef(bip.rem));
                break;
            case RDINT:
                if (debug == 1) printf("rdint\n");
                bigRead(stdin);
                push(stack, stackSlotWithObjRef(bip.res));
                if (debug == 1) printf("pushed %i\n", peek(stack, 1));
                break;
            case WRINT:
                if (debug == 1) printf("wrint: %i\n", peek(stack, 1));
                bip.op1 = pop(stack).u.objRef;
                bigPrint(stdout);
                break;
            case RDCHR:
                if (debug == 1) printf("rdchr\n");
                scanf("%c", &charInput);
                bigFromInt(charInput);
                push(stack, stackSlotWithObjRef(bip.res));
                if (debug == 1) printf("pushed %c\n", charInput);
                break;
            case WRCHR:
                if (debug == 1) printf("wrchr: %c\n", peek(stack, 1));
                bip.op1 = pop(stack).u.objRef;
                printf("%c", bigToInt());
                break;
            case PUSHG:
                if (debug == 1) printf("pushg: %i\n", SIGN_EXTEND(IMMEDIATE(program.program[i])));
                push(stack, stackSlotWithObjRef(getSDA(SIGN_EXTEND(IMMEDIATE(program.program[i])), sda)));
                break;
            case POPG:
                if (debug == 1) printf("popg: %i\n", SIGN_EXTEND(IMMEDIATE(program.program[i])));
                setSDA(SIGN_EXTEND(IMMEDIATE(program.program[i])), pop(stack).u.objRef, sda);
                break;
            case ASF:
                if (debug == 1) printf("asf: %i\n", SIGN_EXTEND(IMMEDIATE(program.program[i])));
                push(stack, stackSlotWitchNumber(fp));
                fp = *stack.current;
                *stack.current = *stack.current + SIGN_EXTEND(IMMEDIATE(program.program[i]));
                break;
            case RSF:
                if (debug == 1) printf("rsf: %i\n", SIGN_EXTEND(IMMEDIATE(program.program[i])));
                *stack.current = fp;
                if (debug == 1) printf("pop: %i\n", peek(stack, 1));
                tempSlot = pop(stack);
                fp = tempSlot.u.number;
                break;
            case POPL:
                if (debug == 1) printf("popl: %i\n", SIGN_EXTEND(IMMEDIATE(program.program[i])));
                stack.stack[fp + SIGN_EXTEND(IMMEDIATE(program.program[i]))] = pop(stack);
                break;
            case PUSHL:
                if (debug == 1) printf("pushl: %i\n", SIGN_EXTEND(IMMEDIATE(program.program[i])));
                push(stack, stack.stack[fp + SIGN_EXTEND(IMMEDIATE(program.program[i]))]);
                break;
            case NE:
                if (debug == 1) printf("ne: %i != %i\n", peek(stack, 2), peek(stack, 1));
                bip.op2 = pop(stack).u.objRef;
                bip.op1 = pop(stack).u.objRef;
                bigFromInt(bigCmp() != 0);
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case EQ:
                if (debug == 1) printf("eq: %i == %i\n", peek(stack, 2), peek(stack, 1));
                bip.op2 = pop(stack).u.objRef;
                bip.op1 = pop(stack).u.objRef;
                bigFromInt(bigCmp() == 0);
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case LT:
                if (debug == 1) printf("lt: %i < %i\n", peek(stack, 2), peek(stack, 1));
                bip.op2 = pop(stack).u.objRef;
                bip.op1 = pop(stack).u.objRef;
                bigFromInt(bigCmp() < 0);
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case LE:
                if (debug == 1) printf("le: %i <= %i\n", peek(stack, 2), peek(stack, 1));
                bip.op2 = pop(stack).u.objRef;
                bip.op1 = pop(stack).u.objRef;
                bigFromInt(bigCmp() <= 0);
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case GT:
                if (debug == 1) printf("gt: %i > %i\n", peek(stack, 2), peek(stack, 1));
                bip.op2 = pop(stack).u.objRef;
                bip.op1 = pop(stack).u.objRef;
                bigFromInt(bigCmp() > 0);
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case GE:
                if (debug == 1) printf("ge: %i >= %i\n", peek(stack, 2), peek(stack, 1));
                bip.op2 = pop(stack).u.objRef;
                bip.op1 = pop(stack).u.objRef;
                bigFromInt(bigCmp() >= 0);
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case BRF:
                if (debug == 1) printf("brf: %i\n", SIGN_EXTEND(IMMEDIATE(program.program[i])));
                if (debug == 1) printf("pop: %i\n", peek(stack, 1));
                bip.op1 = pop(stack).u.objRef;
                int b = bigToInt();
                if (b == false) {
                    i = SIGN_EXTEND(IMMEDIATE(program.program[i])) - 1;
                    if (debug == 1) printf("new i: %i\n", i);
                } else if (b != true) {
                    printf("Error: brf: %i\n", b);
                    exit(EXIT_FAILURE);
                }
                break;
            case BRT:
                if (debug == 1) printf("brt: %i\n", SIGN_EXTEND(IMMEDIATE(program.program[i])));
                if (debug == 1) printf("pop: %i\n", peek(stack, 1));
                bip.op1 = pop(stack).u.objRef;
                b = bigToInt();
                if (b == true) {
                    i = SIGN_EXTEND(IMMEDIATE(program.program[i])) - 1;
                    if (debug == 1) printf("new i: %i\n", i);
                } else if (b != false) {
                    printf("Error: brt: %i\n", b);
                    exit(EXIT_FAILURE);
                }
                break;
            case JMP:
                if (debug == 1) printf("jmp: %i\n", SIGN_EXTEND(IMMEDIATE(program.program[i])));
                i = SIGN_EXTEND(IMMEDIATE(program.program[i])) - 1;
                if (debug == 1) printf("new i: %i\n", i);
                break;
            case CALL:
                if (debug == 1) printf("call: %i\n", SIGN_EXTEND(IMMEDIATE(program.program[i])));
                push(stack, stackSlotWitchNumber(i));
                if (debug == 1) printf("push: %i\n", i + 1);
                i = SIGN_EXTEND(IMMEDIATE(program.program[i])) - 1;
                if (debug == 1) printf("new i: %i\n", i);
                break;
            case RET:
                if (debug == 1) printf("ret\n");
                if (debug == 1) printf("pop: %i\n", peek(stack, 1));
                i = getIntValfromStackSlot(pop(stack));
                if (debug == 1) printf("new i: %i\n", i);
                break;
            case DROP:
                if (debug == 1) printf("drop\n");
                *stack.current = *stack.current - SIGN_EXTEND(IMMEDIATE(program.program[i]));
                break;
            case DUP:
                if (debug == 1) printf("dup\n");
                tempObj = pop(stack).u.objRef;
                push(stack, stackSlotWithObjRef(tempObj));
                push(stack, stackSlotWithObjRef(tempObj));
                break;
            case POPR:
                if (debug == 1) printf("popr");
                push(reg, pop(stack));
                break;
            case PUSHR:
                if (debug == 1) printf("pushr");
                push(stack, pop(reg));
                break;
            case NEW:
                if (debug == 1) printf("new");
                push(stack, stackSlotWithObjRef(newRecord(SIGN_EXTEND(IMMEDIATE(program.program[i])))));
                break;
            case GETF:
                if (debug == 1) printf("getf");
                tempObj = pop(stack).u.objRef;
                push(stack, stackSlotWithObjRef(getField(tempObj,SIGN_EXTEND(IMMEDIATE(program.program[i])))));
                break;
            case PUTF:
                if (debug == 1) printf("putf");
                tempObj = pop(stack).u.objRef;
                tempObj2 = pop(stack).u.objRef;
                setField(tempObj2, SIGN_EXTEND(IMMEDIATE(program.program[i])),tempObj);
                break;
            case NEWA:
                if(debug == 1) printf("newa");
                bip.op1 = pop(stack).u.objRef;
                push(stack, stackSlotWithObjRef(newRecord(bigToInt())));
                break;
            case GETFA:
                if(debug == 1) printf("getfa");
                bip.op1 = pop(stack).u.objRef;
                tempInt = bigToInt();
                tempObj = pop(stack).u.objRef;
                push(stack, stackSlotWithObjRef(getField(tempObj,bigToInt())));
                break;
            case PUTFA:
                if (debug == 1) printf("putfa");
                tempObj = pop(stack).u.objRef; // Value
                tempObj2 = pop(stack).u.objRef; // Index
                bip.op1 = tempObj2;
                tempInt = bigToInt();
                setField(pop(stack).u.objRef, tempInt,tempObj);
                break;
            case GETSZ:
                if (debug == 1) printf("getsz");
                tempObj = pop(stack).u.objRef;
                if(IS_PRIMITIVE(tempObj))  bigFromInt(-1);
                else bigFromInt(GET_ELEMENT_COUNT(tempObj));
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case PUSHN:
                if (debug == 1) printf("pushn");
                push(stack, stackSlotWithObjRef(NULL));
                break;
            case REFEQ:
                if (debug == 1) printf("refeq");
                if(pop(stack).u.objRef == pop(stack).u.objRef) bigFromInt(true);
                else bigFromInt(false);
                push(stack, stackSlotWithObjRef(bip.res));
                break;
            case REFNE:
                if (debug == 1) printf("refeq");
                if(pop(stack).u.objRef != pop(stack).u.objRef) bigFromInt(true);
                else bigFromInt(false);
                push(stack, stackSlotWithObjRef(bip.res));
                break;
        }
    }
    end:
    return;
}

void tests(void) {
}

int main(int argc, char *argv[]) {

    // Initialize the Stack
    int size = SIZE;
    int current = 0;
    stack.size = &size;
    stack.current = &current;
    stack.stack = malloc(SIZE * 10024);

    // Initialize the registery
    int rSize = 1000;
    int rCurrent = 0;
    StackSlot r[1000];
    reg.size = &rSize;
    reg.current = &rCurrent;
    reg.stack = r;

    // Initialize ProgrammSpeicher
    int psize = 1000;
    int saveProgram = 0;
    unsigned int p[1000];
    program.size = &psize;
    program.program = p;
    program.saveProgram = &saveProgram;

    int run = 0;
    int sizeSDA;

    if (argc > 1) {
        for (int i = 1; i < argc; ++i) {
            if (strcmp(argv[i], "--debug") == 0) {
                debug = 1;
            } else if (strcmp(argv[i], "--version") == 0) {
                version();
                return 0;
            } else if (strcmp(argv[i], "--help") == 0) {
                help();
                return 0;
            } else if (strcmp(argv[i], "--stack") == 0) {
                i++;
                // TODO: implement stack size
            } else if (strcmp(argv[i], "--heap") == 0) {
                i++;
                // TODO: implement heap size
            } else if (strcmp(argv[i], "--gcpurge") == 0) {
                i++;
                // TODO: implement gcpurge
            } else {
                sizeSDA = fromFile(argv[i], program);
                run = 1;
            }
        }
    }

    if (debug) {
        tests();
    }

    if (run) {
        printf("Ninja Virtual Machine started\n");
        ObjRef s[sizeSDA];
        sda.size = &sizeSDA;
        sda.sda = s;
        if (debug == 1) printProgram(program);
        execute(program);
        printf("Ninja Virtual Machine stopped\n");
    } else {
        printf("Error: no code file specified\n");
        return 1;
    }
}