from cozmo_fsm import *

class LearnColor(StateMachineProgram):

    class Test(StateNode):
        def onclick(self,event):
            print('event=',event)
            x = int(round(event.xdata))
            y = int(round(event.ydata))
            print(self.im[y][x])
            for i in range(0,50):
                self.im[y,x+i] = (255,255,0)
            plt.imshow(self.im)
            plt.show(block=False)

        def start(self,event=None):
            super().start()
            self.im = cv2.cvtColor(cv2.imread('/tmp/foo.jpg'), cv2.COLOR_BGR2RGB)
            self.fig = plt.figure()
            ax = self.fig.add_subplot(111)
            axim = plt.imshow(self.im)
            print(axim)
            cid = self.fig.canvas.mpl_connect('button_press_event', self.onclick)
            plt.show(block=False)
            print('shown')

    def setup(self):
        """
            self.Test()
        """
        
        # Code generated by genfsm on Fri Mar 23 01:43:18 2018:
        
        test1 = self.Test() .set_name("test1") .set_parent(self)
        
        return self