from UniversalClasses import System, State, AffinityRule, TransitionRule
#import SaveFile

import detGen
import oneSidedGen
import nonDetGen
import QuickRotate
import DirectCombine


red = "f03a47"
blue = "3f88c5"
green = "0ead69"
orange = "f39237"
black = "323031"
white = "DFE0E2"
grey = "9EA9A4"
light_blue = "C2DCFE"

def genSquare(value, model):
    rectSys = ""
    
    if model == "Deterministic":
        rectSys = detGen.genRect(value)

    if model == "Single-Transition":
        rectSys = oneSidedGen.genRect(value)

    if model == "Non-Deterministic":
        rectSys = nonDetGen.genRect(value)

    if rectSys == "":
        return

    rectSysB = QuickRotate.rotate(rectSys)
    rectSysC = QuickRotate.rotate(rectSysB)
    rectSysD = QuickRotate.rotate(rectSysC)

    squareSys = DirectCombine.main(rectSys, rectSysB, "*")
    squareSys = DirectCombine.main(squareSys, rectSysC, "#")
    squareSys = DirectCombine.main(squareSys, rectSysD, "$")


    squareSys.add_State(State("pA", black))
    squareSys.add_State(State("pB", black))
    squareSys.add_State(State("pC", black))

    squareSys.add_transition_rule(TransitionRule("N", "c", "N", "pA", "h"))
    squareSys.add_affinity(AffinityRule("pA", "SA*", "h", 1))

    squareSys.add_transition_rule(TransitionRule("N*", "c*", "N*", "pB", "v"))
    squareSys.add_affinity(AffinityRule("pB", "SA#", "v", 1))

    squareSys.add_transition_rule(TransitionRule("c#", "N#", "pC", "N#", "h"))
    squareSys.add_affinity(AffinityRule("SA$", "pC", "h", 1))

    squareSys.add_Initial_State(State("SA*", black))
    squareSys.add_Initial_State(State("SA#", black))
    squareSys.add_Initial_State(State("SA$", black))

    squareSys.add_State(State("pD", black))
    squareSys.add_transition_rule(TransitionRule("c$", "N$", "pD", "N$", "v"))
    squareSys.add_transition_rule(TransitionRule("pD", "0c$", "pD", "pD", "h"))
    squareSys.add_transition_rule(TransitionRule("pD", "0cn$", "pD", "pD", "h"))
    squareSys.add_transition_rule(TransitionRule("pD", "+$", "pD", "pD", "h"))

    squareSys.add_State(State("f", light_blue))
    squareSys.add_Initial_State(State("f", light_blue))
    squareSys.add_affinity(AffinityRule("pD", "f", "h", 1))
    squareSys.add_affinity(AffinityRule("f", "f", "h", 1))
    squareSys.add_affinity(AffinityRule("f", "f", "v", 1))
    

    return squareSys

# Row Generator
# Note: Actual Row Length = length + 1
def simple_row_generator(length=0):
    symbols = ['@', '#', '$', '!']
    loop_counter = 0

    seed_states = []
    initial_states = []
    complete_states = []
    vertical_affinities = []
    horizontal_affinities = []
    vertical_transitions = []
    horizontal_transitions = []

    if(length > 3):
        # For each sub-system...
        for symbol in symbols:
            if(loop_counter == 0):  # System #0; Row
                # Record Seed, Initial, and Complete States
                for i in range(length):
                    tempState = State(str(i+1)+symbol, 'ffffff')
                    if(i == 0):  # If it's the first tile, make it the seed.
                        seed_states.append(tempState)
                    else:  # Otherwise, make it an initial state.
                        initial_states.append(tempState)
                    complete_states.append(tempState)
                # Record necessary affinity rules
                for i in range(length-1):
                    tempRule = AffinityRule(
                        str(i+1)+symbol, str(i+2)+symbol, 'h', 1)
                    horizontal_affinities.append(tempRule)
            elif(loop_counter == 1):  # System #1; Column
                # Record States again
                for i in range(length):
                    tempState = State(str(i+1)+symbol, 'ffffff')
                    initial_states.append(tempState)
                    complete_states.append(tempState)
                # Establish affinity rules again
                for i in range(length-2):
                    tempRule = AffinityRule(
                        str(i+1)+symbol, str(i+2)+symbol, 'v', 1)
                    vertical_affinities.append(tempRule)
                # Connect System #1 to #0
                tempRule = AffinityRule(
                    str(length)+symbols[0], "1"+symbol, 'v', 1)
                vertical_affinities.append(tempRule)
            elif(loop_counter == 2):  # System #2; Row
                # Record States again
                for i in range(length):
                    tempState = State(str(i+1)+symbol, 'ffffff')
                    initial_states.append(tempState)
                    complete_states.append(tempState)
                # Establish affinity rules again
                for i in range(length-2):
                    tempRule = AffinityRule(
                        str(i+2)+symbol, str(i+1)+symbol, 'h', 1)
                    horizontal_affinities.append(tempRule)
                # Connect System #2 to #1
                tempRule = AffinityRule(
                    "1"+symbol, str(length-1) + symbols[1], 'h', 1)
                horizontal_affinities.append(tempRule)
            else:
                # If we're at System #3, then we're just making filler squares.
                tempState = State(symbol, 'ffff00')
                initial_states.append(tempState)
                complete_states.append(tempState)
                tempState = State(symbol+'\'', '00ffff')
                initial_states.append(tempState)
                complete_states.append(tempState)
                tempRule1 = AffinityRule(symbol, symbol, 'v', 1)
                tempRule2 = AffinityRule(symbol, symbol+'\'', 'h', 1)
                tempRule3 = AffinityRule(symbol+'\'', symbol+'\'', 'h', 1)
                vertical_affinities.append(tempRule1)
                horizontal_affinities.append(tempRule2)
                horizontal_affinities.append(tempRule3)
                # Connect System #3 to #2
                tempRule = AffinityRule(
                    symbol, str(length-1)+symbols[2], 'v', 1)
                vertical_affinities.append(tempRule)
            # Increment loop counter.
            loop_counter += 1
        tempSystem = System(1, complete_states, initial_states, seed_states, vertical_affinities,
                            horizontal_affinities, vertical_transitions, horizontal_transitions)
        #SaveFile.main(tempSystem, ["PStest.xml"])


simple_row_generator(25)