def __init__(self):
Gtk.Window.__init__(self, title="SaltoTools Replay")
self.saltoRecorder = SaltoRecorder()
self.sending_server_ip = "127.0.0.1"
self.listening_port = 14040
self.sending_port = 14050
self.is_recording = False
self.is_playing = False
# Draw the GUI
self.draw_interface()
class SaltoReplayWindow(Gtk.Window):
def __init__(self):
Gtk.Window.__init__(self, title="SaltoTools Replay")
self.saltoRecorder = SaltoRecorder()
self.sending_server_ip = "127.0.0.1"
self.listening_port = 14040
self.sending_port = 14050
self.is_recording = False
self.is_playing = False
# Draw the GUI
self.draw_interface()
def draw_interface(self):
""" Draw widget for our GUI
"""
self.set_resizable(False)
self.set_border_width(20)
self.box = Gtk.Box(spacing=10)
self.add(self.box)
self.table = Gtk.Table(3, 2, True)
self.table.set_row_spacings(20)
self.table.set_col_spacings(20)
# self.add(self.table)
self.box.pack_start(self.table, True, True, 0)
self.countdown_label = Gtk.Label(label=":P")
self.countdown_label.set_use_markup(True)
self.countdown_label.set_markup('<span size="38000"> </span>')
self.recordImage = Gtk.Image()
self.recordImage.set_from_file("lib/Assets/record.png")
self.playImage = Gtk.Image()
self.playImage.set_from_file("lib/Assets/play.png")
self.stopImage = Gtk.Image()
self.stopImage.set_from_file("lib/Assets/stop.png")
self.recordButton = Gtk.Button(image=self.recordImage)
self.recordButton.connect("clicked", self.record)
self.recordButton.set_size_request(170, 50)
self.playButton = Gtk.Button(image=self.playImage)
self.playButton.connect("clicked", self.play)
self.playButton.set_size_request(170, 50)
self.actionButton = Gtk.Button(label="Action!")
self.actionButton.connect("clicked", self.action)
self.actionButton.set_size_request(170, 50)
self.check_passthrough = Gtk.CheckButton("Passthrough")
self.check_passthrough.set_active(True)
self.table.attach(self.countdown_label, 0, 2, 0, 1)
self.table.attach(self.recordButton, 0, 1, 1, 2)
self.table.attach(self.playButton, 1, 2, 1, 2)
self.table.attach(self.check_passthrough, 0, 1, 2, 3)
self.table.attach(self.actionButton, 1, 2, 2, 3)
def action(self, widget):
# Pause for five seconds then continue
self.countdown(5)
# Make a server and a client and pass the data through!
self.action_client = SaltoPlayer(self.sending_server_ip, self.sending_port)
self.saltoRecorder.should_record = False
self.saltoRecorder.passthrough = True
self.saltoRecorder.passthrough_client = self.action_client
self.saltoRecorder.run()
def record(self, widget):
if not self.is_recording:
self.countdown(5, label="Recording ...", label_color="#a20")
self.recordButton.set_image(self.stopImage)
self.saltoRecorder.should_record = True
self.saltoRecorder.run()
self.is_recording = True
else:
self.clear_message()
self.recordButton.set_image(self.recordImage)
self.saltoRecorder.should_record = False
self.saltoRecorder.close()
self.is_recording = False
def play(self, widget):
if not self.is_playing:
self.playButton.set_image(self.stopImage)
self.is_playing = True
filename = self.choose_file()
self.countdown(5, label="Playing")
self.saltoPlayer = SaltoPlayer(self.sending_server_ip, self.sending_port)
self.saltoPlayer.run(filename, self.rundoneCallback)
else:
self.saltoPlayer.close()
self.is_playing = False
self.playButton.set_image(self.playImage)
def rundoneCallback(self):
self.saltoPlayer.close()
self.is_playing = False
self.playButton.set_image(self.playImage)
self.clear_message()
def quit(self, widget, stuff):
""" Let's exit nicely
"""
if self.is_recording:
self.saltoRecorder.close()
if self.is_playing:
self.saltoPlayer.close()
Gtk.main_quit()
######################
# #
# Helper Functions #
# #
######################
def countdown(self, time, callback=None, label=None, label_color=None):
""" Helper function to display a countdown
and call a function after the time has run out
"""
for sec in range(time * 10):
remaining = time - (sec / 10)
sleep(0.1)
self.countdown_label.set_markup('<span size="38000">' + str(remaining) + "</span>")
while Gtk.events_pending():
Gtk.main_iteration()
if not label:
label = "Active"
if not label_color:
label_color = "#2a0"
self.countdown_label.set_markup('<span size="38000" color="' + label_color + '">' + label + "</span>")
if callback:
callback()
return True
def clear_message(self):
""" Clear the message field
"""
self.countdown_label.set_markup('<span size="38000"> </span>')
def choose_file(self):
""" Helper function to display a file picker
and return a path
"""
chooser_dialog = Gtk.FileChooserDialog(
title="Open file",
action=Gtk.FileChooserAction.OPEN,
buttons=["Open", Gtk.ResponseType.OK, "Cancel", Gtk.ResponseType.CANCEL],
)
tfilter = Gtk.FileFilter()
tfilter.add_mime_type("text/plain")
tfilter.add_pattern("*.txt")
tfilter.set_name("Salto Recordings")
chooser_dialog.add_filter(tfilter)
response = chooser_dialog.run()
filename = chooser_dialog.get_filename()
chooser_dialog.hide() # destroy()
# Hack to make the dialog disappear
while Gtk.events_pending():
Gtk.main_iteration()
return filename.decode("utf8")