print("Some assembler")
import sys
args=sys.argv
if(len(args)<2):
    print(len(args))
    args.append(input("input filename:"))
file=open(args[1])
filedat=file.read()
file.close()
fdat_upper="""library IEEE;
use IEEE.STD_LOGIC_1164.ALL;
use IEEE.NUMERIC_STD.ALL;

entity biosram is
port (
		 clka  : IN STD_LOGIC;
		 ena   : IN STD_LOGIC;
		 wea   : IN STD_LOGIC;
		 addra : IN STD_LOGIC_VECTOR(13 DOWNTO 0);
		 dina  : IN STD_LOGIC_VECTOR(7 DOWNTO 0);
		 douta : OUT STD_LOGIC_VECTOR(7 DOWNTO 0)
     );
end biosram;

architecture Behavioral of biosram is

	--Declaration of type and signal of a 512 element RAM
	--with each element being 8 bit wide.
	type ram_t is array (0 to """
fdat_mid=""") of std_logic_vector(7 downto 0);
	signal ram : ram_t := ("""
fdat_lower=""");
	signal data_o : std_logic_vector(7 downto 0);
	signal adr    : std_logic_vector(5 downto 0);
begin

	--process for read and write operation.
	process(CLKa)
	begin
		if(rising_edge(CLKa))then
			if(wea='1' and ena='1')then
				ram(to_integer(unsigned(addra))) <= dina;
			end if;
		end if;
	end process;
	douta <= ram(to_integer(unsigned(addra))) when wea='0' and ena='1';

end Behavioral;"""
split=False
com=False
tmp=""
tokens=[]
for i in filedat:
    if i==';':
        com=True
    elif(com):
        if i=='\n' or i=='\r':
            com=False
    elif i=='"':
        split=not split
        if(tmp!=""):
            tokens.append(tmp)
        tmp=""
        if split:
            tokens.append('"')
    elif (i.isspace() or i==',' or i=='\t')and tmp!="'" and not split:
        if(tmp!=""):
            tokens.append(tmp)
        tmp=""
    else:
        tmp+=i
tokens.append(tmp)

names={}
location=0
i=0
regs=["ax","bx","cx","dx","si","di","sp",'pc']
def to_int(s):
    base=10
    if(str(s)[:2]=="0x"):
        base=16
    #exec("y="+s)
    try:
        return int(str(s),base)
    except:
        if(s[0]=="'" and s[2]=="'"):
            return ord(s[1])
def is_int(s):
    base=10
    s=str(s)
    if(str(s)[:2]=="0x"):
        base=16
    try: 
        if(len(s)>=3 and s[0]=="'" and s[2]=="'"):
            return True
        int(s,base)
        return True
    except ValueError:
        return False
while i<len(tokens):
    if tokens[i].endswith(':'):
        names[tokens[i].replace(':','')]=location
        tokens.remove(tokens[i])
        i-=1
    else:
        tokens[i]=tokens[i].lower()
        if tokens[i]=="org":
            i=i+1
            location=to_int(tokens[i])
        elif tokens[i]=='"':
            i+=1
            location+=len(tokens[i])
        elif tokens[i]=="jmp":
            i=i+1
            if(tokens[i] in regs):
                location+=1
            else:
                location+=5
        elif tokens[i]=="jpz":
            i=i+1
            location+=5
        elif tokens[i]=="jnz":
            i=i+1
            location+=5
        elif tokens[i]=="call":
            location+=5
            i+=1
        elif tokens[i]=="lodsb":
            location+=1
        elif tokens[i]=="ex":
            location+=1
            i+=2
        elif tokens[i]=="stosb":
            location+=1
        elif tokens[i]in["ret","rtz",'nop']:
            location+=1
        elif tokens[i] in ["add","sub","mul","div","or","and","xor"]:
            if(tokens[i+2] in regs):
                location+=1
            else:
                location+=5
            i+=2
        elif tokens[i] in ["not","neg","dec","inc","push","pop"]:
            location+=1
            i+=1
        elif tokens[i]=="mov":
            if( tokens[i+2] in regs):
                location+=1
            else:
                location+=5
            i+=2
        elif tokens[i] in ["out",'in']:
            if( tokens[i+2] in regs):
                location+=1
            else:
                location+=2
            i+=2
        elif tokens[i]=="db":
            location+=1
            i+=1
            if(tokens[i]=="'"):
                i+=2
        else:
            print("error: invalid:"+tokens[i])
    i+=1
print("Should be "+str(location)+" bytes if starting from address 0x0000")
location=0
coe=""
out=[]
def evaluate(x):  
    if(x in names.keys()):
        out.append(names[x]&255)
        out.append((names[x]>>8)&255)
        out.append((names[x]>>16)&255)
        out.append((names[x]>>24)&255)
    elif is_int(x):
        out.append(to_int(x)&255)
        out.append((to_int(x)>>8)&255)
        out.append((to_int(x)>>16)&255)
        out.append((to_int(x)>>24)&255)
    elif x=='$':
        out.append(to_int(location)&255)
        out.append((to_int(location)>>8)&255)
        out.append((to_int(location)>>16)&255)
        out.append((to_int(location)>>24)&255)
    else:
        print("Error: '"+x+"' is not a valid number, identifier, or symbol.")
i=0
while i<len(tokens):
    tokens[i]=tokens[i].lower()
    if tokens[i]=="org":
        i=i+1
        location=to_int(tokens[i])
    elif tokens[i]=="db":
        i=i+1
        if(tokens[i]=="'"):
            i+=2
            out.append(to_int(["'",tokens[i-1],"'"]))
        else:
            out.append(to_int(tokens[i]))
        location+=1
    elif tokens[i]=='"':
        i+=1
        location+=len(tokens[i])
        for x in tokens[i]:
            out.append(ord(x))
    elif tokens[i]=="jmp":
        i=i+1
        if(tokens[i] in regs):
            location+=1
            out.append(0x07|(regs.index(tokens[i])<<4))
        else:
            out.append(0x0F)
            evaluate(tokens[i])
            location+=5
    elif tokens[i]=="jpz":
        i=i+1
        out.append(0x2C)
        evaluate(tokens[i])
        location+=5
    elif tokens[i]=="jnz":
        i=i+1
        out.append(0x3C)
        evaluate(tokens[i])
        location+=5
    elif tokens[i]=="call":
        i+=1
        out.append(0x2D)
        evaluate(tokens[i])
        location+=5
    elif tokens[i] in ["add","sub","mul","div","or","and","xor"]:
        try:
            q2=["add","sub","mul","div","or","and","xor"].index(tokens[i])
        except:
            print("Error: internal assembler error (*OUCH*)")
        if(tokens[i+2] in regs):
            i+=2
            try:
               q=regs.index(tokens[i])
            except:
                print("Error: '"+tokens[i]+"' is not a valid 16-bit register.")
            out.append(0x80|q|(q2<<4))
            location+=1
        else:
            i+=1
            if(tokens[i]!='ax'):
                print("Error: invalid operands.")
            i+=1
            out.append(0x88|q2)
            evaluate(tokens[i])
            location+=5
    elif tokens[i]in["not","neg","dec","inc","push","pop"]:
        q=0x98+(["inc","dec","not","neg","push","pop"].index(tokens[i])<<4)
        i+=1
        try:
            out.append(regs.index(tokens[i])|q)
        except:
            print("Error: '"+tokens[i]+"' is not a valid 16-bit register.")
        location+=1
    elif tokens[i]=="lodsb":
        out.append(0x18)
        location+=1
    elif tokens[i]=="stosb":
        out.append(0x1C)
        location+=1
    elif tokens[i]=="ret":
        out.append(0x2E)
        location+=1
    elif tokens[i]=="rtz":
        out.append(0x3E)
        location+=1
    elif tokens[i]=="nop":
        out.append(0)
        location+=1
    elif tokens[i]=="mov":
        if( tokens[i+2] in regs):
            i+=1
            try:
                out.append(regs.index(tokens[i])|(regs.index(tokens[i+1])<<4))
            except:
                print("Error: '"+tokens[i+1]+"' is not a valid 16-bit register.")
            location+=1
            i+=1
        else:
            i+=1
            try:
                out.append(0x08|regs.index(tokens[i]))
                evaluate(tokens[i+1])
            except:
                print("Error: '"+tokens[i]+"' is not a valid 16-bit register.")
            i+=1
            location+=5
    elif tokens[i]=="ex":
        i+=1
        if(tokens[i]=='ax'):
            i+=1
            try:
                out.append(0x78|regs.index(tokens[i]))
            except:
                print("Error: '"+tokens[i+1]+"' is not a valid 16-bit register.")
            location+=1
            i+=1
        elif tokens[i]=='si' and tokens[i+1]=='di':
            i+=1
            location+=5
            out.append(0x2F)
    elif tokens[i]=="out":
        if is_int(tokens[i+2]):
            i+=2
            out.append(0x38)
            out.append(to_int(tokens[i])&255)
            location+=2
        else:
            out.append(0x39)
            location+=1
    elif tokens[i]=="in":
        if is_int(tokens[i+2]):
            i+=2
            out.append(0x3A)
            out.append(to_int(tokens[i])&255)
            location+=2
        else:
            out.append(0x3B)
            location+=1
    else:
        print("error: invalid:"+tokens[i])
    i+=1
print("Looks good!")
print(str(len(out))+" bytes in size.")
"""if("bios.asm" in args or "boot.asm" in args):
    print("generating output...")
    COE=""
    for i in out:
        COE+='"'+(bin(i)[2:]).zfill(8)+'",'
    fout=fdat_upper+str(len(out)-1)+fdat_mid+COE[:len(COE)-1]+fdat_lower
    f=open("biosram.vhd",'w')
    f.write(fout)
    f.close()
    COE="memory_initialization_radix=16;\nmemory_initialization_vector=\n"
    for i in out:
        COE+=(hex(i)[2:]).zfill(2)+',\n'
    fout=COE[:len(COE)-2]+';'
    f=open("mem.coe",'w')
    f.write(fout)
    f.close()"""
f=open("boot.bin",'wb')
f.write(bytes(out))
f.close()