from pywinauto.application import Application

class GuiControl:

    def __init__(self):

        self.visualizer = None

        self.visualizer_dlg = None

        self.recorder = None

        self.recorder_dlg = None

    def set_windows_record(self):

        self.visualizer = Application().start(r"C:\Program Files\Ultraleap\TrackingVisualizer\TrackingVisualizer.exe")

        self.visualizer_dlg = self.visualizer.window()

        self.recorder = Application().connect(class_name="wxWindowNR")

        self.recorder_dlg = self.recorder.window()

        visualizer_dlg = self.visualizer_dlg

        recorder_dlg = self.recorder_dlg

        # toggle off the help menu

        visualizer_dlg.set_keyboard_focus().send_keystrokes("h---f")

        # get maximum screen size

        visualizer_dlg.maximize()

        max_size = visualizer_dlg.rectangle()

        visualizer_dlg.restore()

        size = GuiControl.position_record(max_size, main=True)

        visualizer_dlg.move_window(x=-size['x'], y=size['y'], width=size['width'], height=size['height'], repaint=True)

        # visualizer_dlg.set_transparency(alpha=120)

        size = GuiControl.position_record(max_size, main=False)

        recorder_dlg.set_focus()

        recorder_dlg.move_window(x=-size['x'], y=size['y'], width=size['width'], height=size['height'], repaint=True)

        recorder_dlg.Stop.set_focus()

    def end_record(self):

        self.visualizer.kill()

        self.recorder_dlg.restore()

        self.recorder_dlg.maximize()

    @staticmethod

    def position_record(max_size, main):

        max_width = max_size.right - max_size.left - 20

        max_height = max_size.bottom - max_size.top - 20

        if main:

            return {'x': -int(max_width / 3),

                    'y': 0,

                    'width': int(max_width / 3 * 2),

                    'height': int(max_height)}

        return {'x': 0,

                'y': 0,

                'width': int(max_width / 3),

                'height': int(max_height)}