import sys
import struct

from assembler.lexer import Lexer, LexerError
from assembler.validator import ValidatorError
from assembler.vs.keywords import find_keyword
from assembler.vs.parser import Parser, ParserError
from assembler.vs.emitter import emit_instruction, emit_dual_math_instruction
from assembler.vs.validator import validate_instruction, Instruction, DualMathInstruction
from assembler.error import print_error

sample = b"""
temp[0].xyzw = VE_ADD    const[1].xyzw     const[1].0000     const[1].0000
temp[1].xyzw = VE_ADD    const[1].xyzw     const[1].0000     const[1].0000
temp[0].x    = VE_MAD    const[0].x___     temp[1].x___      temp[0].y___
temp[0].x    = VE_FRC    temp[0].x___      temp[0].0000      temp[0].0000
temp[0].x    = VE_MAD    temp[0].x___      const[1].z___     const[1].w___
temp[0].y    = ME_COS    temp[0].xxxx      temp[0].0000      temp[0].0000
temp[0].x    = ME_SIN    temp[0].xxxx      temp[0].0000      temp[0].0000
temp[0].yz   = VE_MUL    input[0]._xy_     temp[0]._yy_      temp[0].0000
out[0].xz    = VE_MAD    input[0].-y-_-0-_ temp[0].x_0_      temp[0].y_0_
out[0].yw    = VE_MAD    input[0]._x_0     temp[0]._x_0      temp[0]._z_1
"""

def frontend_inner(filename, buf):
    lexer = Lexer(filename, buf, find_keyword)
    tokens = list(lexer.lex_tokens())
    parser = Parser(tokens)
    for ins in parser.instructions():
        ins = validate_instruction(ins)
        if type(ins) is Instruction:
            yield list(emit_instruction(ins))
        elif type(ins) is DualMathInstruction:
            yield list(emit_dual_math_instruction(ins))
        else:
            assert False, type(ins)

def frontend(filename, buf):
    try:
        yield from frontend_inner(filename, buf)
    except ParserError as e:
        print_error(e)
        raise
    except LexerError as e:
        print_error(e)
        raise
    except ValidatorError as e:
        print_error(e)
        raise

if __name__ == "__main__":
    assert len(sys.argv) in {2, 3}
    input_filename = sys.argv[1]
    binary = len(sys.argv) == 3
    if binary:
        output_filename = sys.argv[2]

    with open(input_filename, 'rb') as f:
        buf = f.read()

    code_gen = list(frontend(input_filename, buf))

    if not binary:
        for cw in code_gen:
            print(f"0x{cw[0]:08x}, 0x{cw[1]:08x}, 0x{cw[2]:08x}, 0x{cw[3]:08x},")
    else:
        with open(output_filename, 'wb') as f:
            for cw in code_gen:
                data = struct.pack("<IIII", *cw)
                f.write(data)