コード例 #1
0
ファイル: view_wx.py プロジェクト: jjdenis/Compy
 def OnInit(self):
     self.frame = self.set_frame()
     self.canvas = self.set_canvas(self.frame)
     self.sound = wx.Sound('res/sonido.mp3')
     self.timer = self.set_timer(self.canvas)
     self.pinta = Pinta(self.canvas)
     self.esc_commands = EscapeCommands(self.pinta.poke,
                                        self.envia_comando,
                                        self.cierra_por_esc,
                                        self.get_img_of_sceen)
     return True
コード例 #2
0
ファイル: view_wx.py プロジェクト: jjdenis/Compy
class GUIwx(wx.App):
    def __init__(self, queue1, queue2):
        self.queue1 = queue1
        self.queue2 = queue2
        self.key_pressed = None
        self.frame = None
        self.canvas = None
        self.sound = None
        self.timer = None
        self.dc = None
        # self.app = wx.App(False)
        # Create a new app, don't redirect stdout/stderr to a window.
        self.cierra_por_esc_when_turn = False
        self.pinta = None
        self.esc_commands = None
        wx.App.__init__(self, False)

    def OnInit(self):
        self.frame = self.set_frame()
        self.canvas = self.set_canvas(self.frame)
        self.sound = wx.Sound('res/sonido.mp3')
        self.timer = self.set_timer(self.canvas)
        self.pinta = Pinta(self.canvas)
        self.esc_commands = EscapeCommands(self.pinta.poke,
                                           self.envia_comando,
                                           self.cierra_por_esc,
                                           self.get_img_of_sceen)
        return True

    def run(self):
        self.frame.Show(True)  # Show the frame.
        self.MainLoop()

    def stop(self):
        self.OnClose(1)

    def OnClose(self, event):
        self.timer.Stop()
        self.frame.Destroy()

    def set_frame(self):
        frame = wx.Frame(None, wx.ID_ANY)  # A Frame is a top-level window.
        frame.SetClientSize((TOTAL_CNVS_PTS_X, TOTAL_CNVS_PTS_Y))
        frame.Bind(wx.EVT_CLOSE, self.OnClose)
        frame.Move((30, 30))  # Pone la ventana arriba a la izquierda
        frame.SetTitle(TITLE)
        return frame

    def set_canvas(self, frame):
        canvas = wx.ScrolledWindow(frame)
        canvas.SetClientSize((TOTAL_CNVS_PTS_X, TOTAL_CNVS_PTS_Y))
        canvas.SetBackgroundStyle(wx.BG_STYLE_CUSTOM)
        canvas.Refresh()
        canvas.Bind(wx.EVT_KEY_DOWN, self.on_key)
        canvas.Bind(wx.EVT_KEY_UP, self.on_key_release)
        return canvas

    def set_timer(self, canvas):
        timer = wx.Timer(canvas)
        canvas.Bind(wx.EVT_TIMER, self.on_timer, timer)
        timer.Start(10)  # ms
        return timer

    def on_key(self, event):
        key = event.GetKeyCode()
        key = self.esc_commands.run(key)

        # self.sound.Play(wx.SOUND_ASYNC)
        # self.pinta_bloque(0, 0, color="blue")
        # self.pinta_bloque(5, 5, color="blue")
        if key != self.key_pressed:
            self.envia_comando('key_pressed', key, time())
            self.key_pressed = key

    def on_key_release(self, event):
        key = event.GetKeyCode()
        self.envia_comando('key_released', key, time())
        self.key_pressed = None

    def envia_comando(self, *args):
        commands = list(args)
        self.queue2.put(commands)

    def on_timer(self, event):

        while not self.queue1.empty():
            msg = self.queue1.get()
            comando = msg[0]
            args = msg[1:]

            if comando == 'poke':
                self.pinta.poke(*args)

            elif comando == 'reset_canvas':
                self.pinta.reset_canvas(fm_color=args[0], bg_color=args[1])

            elif comando == 'close_window':
                self.esc_commands.run(ESCAPE)

            else:
                pass

        if self.cierra_por_esc_when_turn:
            self.stop()

    def cierra_por_esc(self):
        self.envia_comando('closing', None, None)
        self.cierra_por_esc_when_turn = True

    def get_img_of_sceen(self):
        os.system('screencapture scr.png')
        screen = wx.Bitmap('scr.png')
        os.remove('scr.png')
        rect = self.frame.GetRect()
        bitmap = screen.GetSubBitmap(rect)
        img = bitmap.ConvertToImage()
        return img