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"):
        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);

        print("There are " + len(building.groups) + " groups");

        while(1)
        {
            group = input("In which group do you want to reference that light ( type \"s\" to return the main menu )? ");

            if(group == "s"):
                break;

            group = int(group);

            if(group >= 0 && group < len(building.groups)):

                if(!(building.groups[group].flags & 0x200)):
                    building.groups[group].flags &= 0x200;

                building.groups[group].light_refs.append(building.num_lights);
            else:
                print("Bad group.");
        }


    else:
        print("Unknown command");