try:
    from reaper_python import *
except ImportError:
    pass
try:
    import tkinter
    from tkinter import ttk
except ImportError:
    pass
try:
    import debug
except ImportError:
    pass
try:
    from rs_midi import RSMidi
except ImportError:
    pass
try:
    from random import choice
except ImportError:
    pass
try:
    import reaChord_data
except ImportError:
    pass
try:
    from rs_statemanager import RSStateManager
except ImportError:
    pass

class ChordSection(RSStateManager):

    #rc = reaChord_functions.RC() # comment out - just for autocomplete
    def __init__(self, parent, rc):

        self.settings = {
            "draw": 0, "channel": 1, "velocity": 96, "highlight": 1, "invert_above": 6
        }
        self.stateManager_Start("ChordSection", self.settings)
        self.parent = parent
        self.rc = rc
        self.init_widgets()

    def draw(self, midiTake, song, sectionLength):
        if not(self.drawOrNot.get()):  return
        currentPos = 0

        for i, sect in enumerate(song["Structure"]):
            chords = song[str(sect)]
            chordLength = sectionLength / len(chords)
            for i, chord in enumerate(chords):
                self.drawChord(midiTake, chord, currentPos, chordLength, i)
                currentPos += chordLength
    ##  return currentPos

    def drawChord(self,midiTake, chord, currentPos, chordLength, chordIndex): #velocity,x,position,startPosition,rd,chordattack,selectNote):

        velocity = self.velocity.get()
        chordattack = self.chordAttack.get()
        chordinvert = self.chordInv.get()
        selectNote = self.highlight.get()
        
        if chordinvert == 'All (default)': 
            randomChPosL = ["rootPosition", "inversion1st", "inversion2nd"]
            chordPos = choice(randomChPosL)
        elif chordinvert == '1st only': 
            randomChPosL = ["inversion1st"]
            chordPos = choice(randomChPosL)
        elif chordinvert == '2nd only': 
            randomChPosL = ["inversion2nd"]
            chordPos = choice(randomChPosL)
        elif chordinvert == 'Root + 1st': 
            randomChPosL = ["rootPosition","inversion1st"]
            chordPos = choice(randomChPosL)
        elif chordinvert == 'Root + 2nd': 
            randomChPosL = ["rootPosition", "inversion2nd"]
            chordPos = choice(randomChPosL)
        else:
            chordPos = "rootPosition"

        step = 0

        if chordattack == 'Fast': step=120 #fast
        elif chordattack == 'Slow': step=480 #slow
        channel = self.settings["channel"]
        noteOffset = -12
                
        # root position
        if chordPos == "rootPosition":
            note1Pitch = int(self.rc.ChordDict[chord][0]) + noteOffset
            note2Pitch = int(self.rc.ChordDict[chord][1]) + noteOffset
            note3Pitch = int(self.rc.ChordDict[chord][2]) + noteOffset
            
        # first inversion
        if chordPos == "inversion1st":
            note1Pitch = int(self.rc.ChordDict[chord][1]) + noteOffset - 12
            note2Pitch = int(self.rc.ChordDict[chord][2]) + noteOffset - 12
            note3Pitch = int(self.rc.ChordDict[chord][0]) + noteOffset

        # second inversion
        if chordPos == "inversion2nd":
            note1Pitch = int(self.rc.ChordDict[chord][2]) + noteOffset - 12
            note2Pitch = int(self.rc.ChordDict[chord][0]) + noteOffset
            note3Pitch = int(self.rc.ChordDict[chord][1]) + noteOffset

        RSMidi.addNote(midiTake, channel, int(velocity), int(currentPos), note1Pitch, int(chordLength), selectNote)
        RSMidi.addNote(midiTake, channel, int(velocity), int(currentPos+step), note2Pitch, int(chordLength)-step, selectNote)
        RSMidi.addNote(midiTake, channel, int(velocity), int(currentPos+(step*2)), note3Pitch, int(chordLength)-(step*2), selectNote)
        
        
        return currentPos

    def init_widgets(self):
        self._label_1 = tkinter.Label(self.parent,
            text = "Draw:",
        )
        self._label_2 = tkinter.Label(self.parent,
            text = "Highlight:",
        )
        self._label_3 = tkinter.Label(self.parent,
            text = "Channel:",
        )
        self._label_4 = tkinter.Label(self.parent,
            text = "Velocity:",
        )
        self._label_5 = tkinter.Label(self.parent,
            text = "Attack:",
        )
        self._label_6 = tkinter.Label(self.parent,
            text = "Inversions:",
        )
        
        self.drawOrNot = self.newControlIntVar("draw", 1)
        self.cDrawOrNot = tkinter.Checkbutton(self.parent, variable=self.drawOrNot)
        self.highlight = self.newControlIntVar("highlight", 1)
        self.cHighlight = tkinter.Checkbutton(self.parent, variable=self.highlight)
        self.channel = self.newControlIntVar("channel", 1)
        self.sChannel = tkinter.Scale(self.parent, from_=1, to=16, resolution=1, showvalue=0, orient="horizontal", variable=self.channel)
        self.scaleTxt1 = ttk.Label(self.parent, textvariable=self.channel)
        self.velocity = self.newControlIntVar("velocity", 96)
        self.sVelocity = tkinter.Scale(self.parent, from_=1, to=127, resolution=1, showvalue=0, orient="horizontal", variable=self.velocity)
        self.scaleTxt2 = ttk.Label(self.parent, textvariable=self.velocity)
        self.chordAttack = self.newControlStrVar("chordattack", self.rc.ChordAttack[1])
        self.cbChordAttack = ttk.Combobox(self.parent, values=self.rc.ChordAttack, state='readonly', width=13, textvariable=self.chordAttack)
        
        self.chordInv = self.newControlStrVar("chordinv", self.rc.ChordInversions[1])
        self.cbChordInversions = ttk.Combobox(self.parent, values=self.rc.ChordInversions, state='readonly', width=13, textvariable=self.chordInv)
                
        # ChordInversions = ('','All (default)','1st inversion','2nd inversion','None')

        #self.cbChordAttack.current(0)

        # Geometry Management
        self._label_1.grid(
            in_    = self.parent,
            column = 1,
            row    = 1,
            columnspan = 1,
            ipadx = 10,
            ipady = 5,
            padx = 0,
            pady = 5,
            rowspan = 1,
            sticky = "w"
        )
        self._label_2.grid(
            in_    = self.parent,
            column = 1,
            row    = 2,
            columnspan = 1,
            ipadx = 10,
            ipady = 5,
            padx = 0,
            pady = 5,
            rowspan = 1,
            sticky = "w"
        )
        self._label_3.grid(
            in_    = self.parent,
            column = 1,
            row    = 3,
            columnspan = 1,
            ipadx = 10,
            ipady = 5,
            padx = 0,
            pady = 5,
            rowspan = 1,
            sticky = "w"
        )
        self._label_4.grid(
            in_    = self.parent,
            column = 1,
            row    = 4,
            columnspan = 1,
            ipadx = 10,
            ipady = 5,
            padx = 0,
            pady = 5,
            rowspan = 1,
            sticky = "w"
        )
        self._label_5.grid(
            in_    = self.parent,
            column = 1,
            row    = 5,
            columnspan = 2,
            ipadx = 10,
            ipady = 0,
            padx = 0,
            pady = 10,
            rowspan = 1,
            sticky = "w"
        )
        self._label_6.grid(
            in_    = self.parent,
            column = 1,
            row    = 6,
            columnspan = 2,
            ipadx = 10,
            ipady = 0,
            padx = 0,
            pady = 0,
            rowspan = 1,
            sticky = "w"
        )
        self.scaleTxt1.grid(
            in_    = self.parent,
            column = 2,
            row    = 3,
            columnspan = 1,
            ipadx = 0,
            ipady = 0,
            padx = 2,
            pady = 0,
            rowspan = 1,
            sticky = "e"
        )
        self.scaleTxt2.grid(
            in_    = self.parent,
            column = 2,
            row    = 4,
            columnspan = 1,
            ipadx = 0,
            ipady = 0,
            padx = 2,
            pady = 0,
            rowspan = 1,
            sticky = "e"
        )
        self.cDrawOrNot.grid(
            in_    = self.parent,
            column = 2,
            row    = 1,
            columnspan = 1,
            ipadx = 0,
            ipady = 0,
            padx = 0,
            pady = 0,
            rowspan = 1,
            sticky = "w"
        )
        self.cHighlight.grid(
            in_    = self.parent,
            column = 2,
            row    = 2,
            columnspan = 1,
            ipadx = 0,
            ipady = 0,
            padx = 0,
            pady = 0,
            rowspan = 1,
            sticky = "w"
        )
        self.sVelocity.grid(
            in_    = self.parent,
            column = 2,
            row    = 4,
            columnspan = 1,
            ipadx = 0,
            ipady = 0,
            padx = 0,
            pady = 0,
            rowspan = 1,
            sticky = "w"
        )
        self.cbChordAttack.grid(
            in_    = self.parent,
            column = 2,
            row    = 5,
            columnspan = 1,
            ipadx = 0,
            ipady = 0,
            padx = 0,
            pady = 0,
            rowspan = 1,
            sticky = "w"
        )
        
        self.cbChordInversions.grid(
            in_    = self.parent,
            column = 2,
            row    = 6,
            columnspan = 1,
            ipadx = 0,
            ipady = 0,
            padx = 0,
            pady = 0,
            rowspan = 1,
            sticky = "w"
        )       
 
        self.sChannel.grid(
            in_    = self.parent,
            column = 2,
            row    = 3,
            columnspan = 1,
            ipadx = 0,
            ipady = 0,
            padx = 0,
            pady = 0,
            rowspan = 1,
            sticky = "w"
        )
        self.parent.grid_rowconfigure(1, weight = 0, minsize = 22, pad = 0)
        self.parent.grid_rowconfigure(2, weight = 0, minsize = 22, pad = 0)
        self.parent.grid_rowconfigure(3, weight = 0, minsize = 22, pad = 0)
        self.parent.grid_rowconfigure(4, weight = 0, minsize = 22, pad = 0)
        self.parent.grid_rowconfigure(5, weight = 0, minsize = 22, pad = 0)
        self.parent.grid_rowconfigure(6, weight = 0, minsize = 22, pad = 0)
        self.parent.grid_columnconfigure(1, weight = 0, minsize = 80, pad = 0)
        self.parent.grid_columnconfigure(2, weight = 0, minsize = 135, pad = 0)

    def msg(self, m):
        if (reaChord_data.debug):
            RPR_ShowConsoleMsg(str(m)+'\n')