import sys
from pywmo import Wmo

wmo_name = input('Enter WMO Filename : ');
building = Wmo(wmo_name);

nLight = building.num_lights;

print("There are " + str(nLight) + " lights in this WMO.");

while(1):
	command = input("What do you want to do ?\n\"a\" - Add a light\n\"q\" - Exit\n");

	if(command == "q"):
		print("Goodbye !");
		sys.exit();
	elif(command == "a"):
		group = input("In which group do you want to reference that light ?")

		print("Light definition")

		light = { }

		flag1 = input("LightType : ");
		flag2 = input("type : ");
		flag3 = input("useAtten : ");
		flag4 = input("pad : ");

		r = input("R : ");
		g = input("G : ");
		b = input("B : ");
		a = input("A : ");

		x = input("X : ");
		y = input("Y : ");
		z = input("Z : ");

		intensity = input("intensity : ");
		attenuation_start = input("attenuation_start : ");
		attenuation_stop = input("attenuation_stop : ");

		unknown1 = input("Unknown 1 : ");
		unknown2 = input("Unknown 1 : ");
		unknown3 = input("Unknown 1 : ");
		unknown4 = input("Unknown 1 : ");

		light["flag1"] = int(flag1);
		light["flag2"] = int(flag2);
		light["flag3"] = int(flag3);
		light["flag4"] = int(flag4);

		light["color"] = (float(r), float(g), float(b), float(a));
		light["position"] = (float(x), float(y), float(z));

		light["attenuation_start"] = float(attenuation_start);
		light["attenuation_stop"] = float(attenuation_stop);

		light["unknown1"] = float(unknown1);
		light["unknown2"] = float(unknown2);
		light["unknown3"] = float(unknown3);
		light["unknown4"] = float(unknown4);

		building.num_lights += 1;
		building.lights.append(light);

	else:
		print("Unknown command");