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
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
