Add countdown panel. Working on issue #5

Author: FreeSlave

Diff for: src/countdown_panel.py

@@ -0,0 +1,57 @@
+import wx
+import wx.lib.newevent
+import datetime
+import time
+
+CountdownPanelEvent, EVT_COUNTDOWN_RANOUT_EVENT = wx.lib.newevent.NewEvent()
+
+class CountdownPanel(wx.Panel):
+    def __init__(self, parent, id=wx.ID_ANY, pos=wx.DefaultPosition, size=wx.DefaultSize, style=wx.TAB_TRAVERSAL, name=wx.PanelNameStr):
+        super(CountdownPanel, self).__init__(parent, id, pos, size, style, name)
+        self.timer = wx.Timer()
+        self.timer.Bind(wx.EVT_TIMER, self._on_timer)
+        self.end_time = 0
+
+        self.label = wx.StaticText(self, label="00:00:00", style=wx.ALIGN_CENTRE_HORIZONTAL)
+
+        sizer = wx.BoxSizer(wx.VERTICAL)
+        sizer.Add((0,0), 1, wx.EXPAND)
+        sizer.Add(self.label, 0, wx.CENTER | wx.ALL)
+        sizer.Add((0,0), 1, wx.EXPAND)
+        self.SetSizer(sizer)
+
+        self.Bind(wx.EVT_SIZE, self._on_resize)
+        self._recalculate_font_size()
+
+    def start_timer(self, timedelta):
+        if self.timer.IsRunning():
+            self.timer.Stop()
+        self.timer.Start(1000)
+        seconds = timedelta
+        if (type(timedelta) is datetime.timedelta):
+            seconds = timedelta.total_seconds()
+        self.end_time = time.time() + seconds
+        self._update_label()
+
+    def _recalculate_font_size(self):
+        font = self.label.GetFont()
+        fontHeight = self.GetSize().height / 4
+        font.SetPixelSize(wx.Size(0, fontHeight))
+        self.label.SetFont(font)
+
+    def _on_resize(self, e):
+        self._recalculate_font_size()
+        e.Skip()
+
+    def _on_timer(self, e):
+        self._update_label()
+
+    def _update_label(self):
+        seconds = int(round(self.end_time - time.time()))
+        if seconds < 0:
+            self.timer.Stop()
+            wx.PostEvent(self.GetEventHandler(), CountdownPanelEvent())
+            return
+        minutes, seconds = divmod(seconds, 60)
+        hours, minutes = divmod(minutes, 60)
+        self.label.SetLabel( "{:0>2}:{:0>2}:{:0>2}".format(hours, minutes, seconds) )

Diff for: src/main.pyw

@@ -189,13 +189,15 @@ class MainFrame(wx.Frame):
         no_show_item = menu_play.Append(wx.ID_ANY, "&No Show")
         menu_play.AppendSeparator()
         play_pause_bg_item = menu_play.Append(wx.ID_ANY, "&Play/Pause Background\tF3")
+        start_countdown_item = menu_play.Append(wx.ID_ANY, "Countdown")
 
         self.Bind(wx.EVT_MENU, self.emergency_stop, emergency_stop_item)
         self.Bind(wx.EVT_MENU, self.show_zad, show_zad_item)
         self.Bind(wx.EVT_MENU, self.play_async, play_track_item)
         self.Bind(wx.EVT_MENU, self.clear_zad, clear_zad_item)
         self.Bind(wx.EVT_MENU, lambda e: self.clear_zad(e, True), no_show_item)
         self.Bind(wx.EVT_MENU, self.play_pause_bg, play_pause_bg_item)
+        self.Bind(wx.EVT_MENU, self.start_countdown, start_countdown_item)
 
         self.SetAcceleratorTable(wx.AcceleratorTable([
             wx.AcceleratorEntry(wx.ACCEL_NORMAL, wx.WXK_ESCAPE, emergency_stop_item.GetId()),
@@ -1061,6 +1063,10 @@ class MainFrame(wx.Frame):
         else:
             self.background_pause(paused=True)
 
+    def start_countdown(self, e=None):
+        self.ensure_proj_win()
+        self.proj_win.switch_to_timer(65) # TODO: dialog with time settings
+
 
 if __name__ == "__main__":
     app = wx.App(False)

Diff for: src/projector.py

@@ -1,5 +1,5 @@
 import wx
-
+import countdown_panel
 
 class ProjectorWindow(wx.Frame):
     def __init__(self, parent, screen=None):
@@ -49,8 +49,14 @@ def on_paint(self, e):
                 dc.DrawBitmap(self.drawable_bitmap, w//2 - drw_w//2, 0)
 
         self.images_panel = ImagesPanel(self)
+
+        self.countdown_panel = countdown_panel.CountdownPanel(self)
+        self.countdown_panel.Bind(countdown_panel.EVT_COUNTDOWN_RANOUT_EVENT, self.on_timer_ranout)
+        self.countdown_panel.Hide()
+
         self.sizer.Add(self.images_panel, 1, wx.EXPAND)
         self.sizer.Add(self.video_panel, 1, wx.EXPAND)  # TODO: Align top
+        self.sizer.Add(self.countdown_panel, 1, wx.EXPAND)
 
         self.SetSizer(self.sizer)
         self.Layout()
@@ -81,3 +87,14 @@ def no_show(self):
         self.images_panel.drawable_bitmap = \
             wx.Bitmap(wx.Image(*self.images_panel.drawable_bitmap.GetSize()))
         self.images_panel.Refresh()
+
+    def switch_to_timer(self, timedelta):
+        self.video_panel.Hide()
+        self.images_panel.Hide()
+        self.countdown_panel.start_timer(timedelta)
+        self.countdown_panel.Show()
+
+    def on_timer_ranout(self, e):
+        self.countdown_panel.Hide()
+        self.images_panel.Show()
+        self.no_show()