start the great refactoring

4 files changed, 154 insertions, 235 deletions

diff --git a/voctogui/lib/audioleveldisplay.py b/voctogui/lib/audioleveldisplay.py
new file mode 100644
index 0000000..8b9bf7e
--- /dev/null
+++ b/voctogui/lib/audioleveldisplay.py
@@ -0,0 +1,25 @@
+import logging
+from gi.repository import Gst, Gtk
+
+class AudioLevelDisplay:
+	""" Displays a Level-Meter of another VideoDisplay into a GtkWidget """
+
+	def __init__(self, drawing_area):
+		self.log = logging.getLogger('AudioLevelDisplay[%s]' % drawing_area.get_name())
+
+		self.drawing_area = drawing_area
+
+		self.levelrms = [0, 0]  # Initialize to []
+		self.drawing_area.connect('draw', self.on_draw)
+
+
+	def on_draw(self, widget, cr):
+		cr.set_source_rgb(1, 1, 1)
+		cr.set_line_width(10)
+
+		cr.move_to(15, 0)
+		cr.line_to(15, self.levelrms[0]*-20)  # Work with 0+ Channels
+		cr.stroke()
+
+	def level_callback(self, peaks, rms):
+		self.levelrms = rms
diff --git a/voctogui/lib/ui.py b/voctogui/lib/ui.py
index 4106967..134bc2c 100644
--- a/voctogui/lib/ui.py
+++ b/voctogui/lib/ui.py
@@ -4,7 +4,17 @@ from gi.repository import Gtk, Gst, Gdk, GLib
 
 from lib.config import Config
 from lib.uibuilder import UiBuilder
+
 from lib.videodisplay import VideoDisplay
+from lib.audioleveldisplay import AudioLevelDisplay
+from lib.warningoverlay import VideoWarningOverlay
+
+# from lib.videopreviews import VideoPreviewsController
+# from lib.audioselector import AudioSelectorController
+
+# from lib.toolbar.composition import CompositionToolbarController
+# from lib.toolbar.streamblank import StreamblankToolbarController
+# from lib.toolbar.specialfunctions import SpecialFunctionsToolbarController
 
 class Ui(UiBuilder):
 	def __init__(self, uifile):
@@ -20,216 +30,44 @@ class Ui(UiBuilder):
 		# Connect Close-Handler
 		self.win.connect('delete-event', Gtk.main_quit)
 
-		self.previews = {}
-		self.preview_players = {}
-
-
-		self.configure_toolbar_accelerators()
-		self.configure_video_main()
-		self.configure_video_previews()
-		self.configure_audio_selector()
-		self.configure_streamblank_selector()
-
-	def configure_toolbar_accelerators(self):
-		accelerators = Gtk.AccelGroup()
-		self.win.add_accel_group(accelerators)
-
-		composites = [
-			'composite-fullscreen',
-			'composite-picture-in-picture',
-			'composite-side-by-side-equal',
-			'composite-side-by-side-preview'
-		]
-
-		for idx, name in enumerate(composites):
-			key, mod = Gtk.accelerator_parse('F%u' % (idx+1))
-			btn = self.find_widget_recursive(self.win, name)
-			btn.set_name(name)
-
-			# Thanks to http://stackoverflow.com/a/19739855/1659732
-			btn.get_child().add_accelerator('clicked', accelerators, key, mod, Gtk.AccelFlags.VISIBLE)
-			btn.connect('toggled', self.composite_btn_toggled)
-
-	def composite_btn_toggled(self, btn):
-		if not btn.get_active():
-			return
-
-		self.log.info("composite_btn_toggled: %s", btn.get_name())
-
-	def configure_video_main(self):
-		self.log.info('Initializing Main Video and Main Audio-Level View')
-
-		video = self.find_widget_recursive(self.win, 'video_main')
-		audiolevel = self.find_widget_recursive(self.win, 'audiolevel_main')
-		self.video_main_player = VideoDisplay(11000, video, audiolevel,
-			playaudio=Config.getboolean('mainvideo', 'playaudio'),
-			allowoverlay=True)
-
-	def configure_video_previews(self):
-		self.log.info('Initializing Video Previews')
-
-		sources = ['cam1', 'cam2', 'grabber']
-		box = self.find_widget_recursive(self.win, 'box_left')
-
-		try:
-			width = Config.getint('previews', 'width')
-			self.log.debug('Preview-Width configured to %u', width)
-		except:
-			width = 320
-			self.log.debug('Preview-Width selected as %u', width)
-
-		try:
-			height = Config.getint('previews', 'height')
-			self.log.debug('Preview-Height configured to %u', width)
-		except:
-			height = width*9/16
-			self.log.debug('Preview-Height calculated to %u', width)
-
-
-		# Accelerators
-		accelerators = Gtk.AccelGroup()
-		self.win.add_accel_group(accelerators)
-
-		group_a = None
-		group_b = None
-
-		for idx, source in enumerate(sources):
-			self.log.info('Initializing Video Preview %s', source)
-
-			preview = self.get_check_widget('widget_preview', clone=True)
-			video = self.find_widget_recursive(preview, 'video')
-
-			video.set_size_request(width, height)
-			box.pack_start(preview, fill=False, expand=False, padding=0)
 
-			player = VideoDisplay(13000 + idx, video)
+		# Create Audio-Level Display
+		drawing_area = self.find_widget_recursive(self.win, 'audiolevel_main')
+		self.audio_level_display = AudioLevelDisplay(drawing_area)
 
-			self.find_widget_recursive(preview, 'label').set_label(source)
-			btn_a = self.find_widget_recursive(preview, 'btn_a')
-			btn_b = self.find_widget_recursive(preview, 'btn_b')
 
-			btn_a.set_name("%c %u" % ('a', idx))
-			btn_b.set_name("%c %u" % ('b', idx))
+		# Create Main-Video Overlay Controller
+		self.video_warning_overlay = VideoWarningOverlay()
 
-			if not group_a:
-				group_a = btn_a
-			else:
-				btn_a.join_group(group_a)
 
+		# Create Main-Video Display
+		drawing_area = self.find_widget_recursive(self.win, 'video_main')
+		self.main_video_display = VideoDisplay(drawing_area,
+			port=11000,
+			play_audio=True,
+			draw_callback=self.video_warning_overlay.draw_callback,
+			level_callback=self.audio_level_display.level_callback)
 
-			if not group_b:
-				group_b = btn_b
-			else:
-				btn_b.join_group(group_b)
 
-			btn_a.connect('toggled', self.preview_btn_toggled)
-			btn_b.connect('toggled', self.preview_btn_toggled)
+		# Setup Preview Controller
+		# drawing_area = self.find_widget_recursive(self.win, 'box_left')
+		# self.video_previews_controller = VideoPreviewsController(drawing_area)
 
-			key, mod = Gtk.accelerator_parse('%u' % (idx+1))
-			btn_a.add_accelerator('activate', accelerators, key, mod, Gtk.AccelFlags.VISIBLE)
+		# drawing_area = self.find_widget_recursive(self.win, 'combo_audio')
+		# self.audio_selector_controller = AudioSelectorController(drawing_area)#
 
-			key, mod = Gtk.accelerator_parse('<Ctrl>%u' % (idx+1))
-			btn_b.add_accelerator('activate', accelerators, key, mod, Gtk.AccelFlags.VISIBLE)
 
-			self.preview_players[source] = player
-			self.previews[source] = preview
+		# # Setup Toolbar Controllers
+		# toolbar = self.find_widget_recursive(self.win, 'toolbar')
+		# self.composition_toolbar_controller = CompositionToolbarController(toolbar)
 
-	def preview_btn_toggled(self, btn):
-		if not btn.get_active():
-			return
+		# self.streamblank_toolbar_controller = StreamblankToolbarController(toolbar,
+		# 	warning_overlay=self.video_warning_overlay)
 
-		self.log.info('preview_btn_toggled: %s', btn.get_name())
-
-	def configure_audio_selector(self):
-		self.log.info('Initializing Audio Selector')
-
-		combo = self.find_widget_recursive(self.win, 'combo_audio')
-		combo.set_sensitive(True)
-
-		# FIXME access via combo_audio?
-		liststore = self.get_check_widget('liststore_audio')
-		liststore.clear()
-
-		row = liststore.append()
-		liststore.set(row, [0], ['foobar'])
-
-		row = liststore.append('')
-		liststore.set(row, [0], ['moofar'])
-
-		combo.set_active_id('moofar')
-
-	def configure_streamblank_selector(self):
-		livebtn = self.get_check_widget('stream_live')
-		blankbtn = self.get_check_widget('stream_blank')
-		toolbar = blankbtn.get_parent()
-		pos = toolbar.get_item_index(blankbtn)
-
-		self.blink_btn = None
-		self.blink_btn_state = False
-
-		livebtn.connect('toggled', self.streamblank_button_toggled)
-		livebtn.set_name('live')
-
-		GLib.timeout_add_seconds(1, self.blink_streamblank_button)
-
-		for idx, name in enumerate(['pause', 'nostream']):
-			if idx == 0:
-				new_btn = blankbtn
-			else:
-				new_icon = Gtk.Image.new_from_pixbuf(blankbtn.get_icon_widget().get_pixbuf())
-				new_btn = Gtk.RadioToolButton(group=livebtn)
-				new_btn.set_icon_widget(new_icon)
-				toolbar.insert(new_btn, pos+1)
-
-			new_btn.set_label("Stream %s" % name)
-			new_btn.connect('toggled', self.streamblank_button_toggled)
-			new_btn.set_name(name)
-
-	def blink_streamblank_button(self):
-		self.blink_btn_state = not self.blink_btn_state
-		return True
-
-	def streamblank_button_toggled(self, btn):
-		if not btn.get_active():
-			return
-
-		if btn.get_name() != 'live':
-			self.blink_btn = btn
-			self.blink_btn_state = False
-			self.streamblank_mode = btn.get_name()
-			self.video_main_player.set_overlay_callback(self.draw_streamblank_warning)
-		else:
-			self.blink_btn = None
-			self.video_main_player.set_overlay_callback(None)
-
-	def draw_streamblank_warning(self, cairooverlay, cr, timestamp, duration):
-		h = 1080/20
-		w = 1920
-
-		if self.blink_btn_state:
-			cr.set_source_rgba(1.0, 0.0, 0.0, 0.8)
-		else:
-			cr.set_source_rgba(1.0, 0.5, 0.0, 0.8)
-
-		cr.rectangle(0, 0, w, h)
-		cr.fill()
-
-		text = "Stream is Blanked: {}".format(self.streamblank_mode)
-
-		cr.set_font_size(h*0.75)
-		xbearing, ybearing, txtwidth, txtheight, xadvance, yadvance = cr.text_extents(text)
-
-		cr.move_to(w/2 - txtwidth/2, h*0.75)
-		cr.set_source_rgba(1.0, 1.0, 1.0, 1.0)
-		cr.show_text(text)
+		# self.special_functions_toolbar_controller = SpecialFunctionsToolbarController(toolbar,
+		# 	video_display=self.main_video_display)
 
 
 	def show(self):
-		self.log.info('Running Video-Playback Pipelines')
-
-		self.video_main_player.run()
-		for name, player in self.preview_players.items():
-			player.run()
-
 		self.log.info('Showing Main-Window')
 		self.win.show_all()
diff --git a/voctogui/lib/videodisplay.py b/voctogui/lib/videodisplay.py
index 42724ac..00648ef 100644
--- a/voctogui/lib/videodisplay.py
+++ b/voctogui/lib/videodisplay.py
@@ -1,52 +1,65 @@
 import logging
-from gi.repository import Gst, Gtk
+from gi.repository import Gst
 
 class VideoDisplay:
 	""" Displays a Voctomix-Video-Stream into a GtkWidget """
 
-	def __init__(self, port, videowidget, audiolevelwidget=None, playaudio=False, allowoverlay=False):
-		self.log = logging.getLogger('VideoDisplay[%u]' % port)
+	def __init__(self, drawing_area, port, play_audio=False, draw_callback=None, level_callback=None):
+		self.log = logging.getLogger('VideoDisplay[%s]' % drawing_area.get_name())
 
+		self.drawing_area = drawing_area
+		self.draw_callback = draw_callback
+		self.level_callback = level_callback
+
+		# Setup Server-Connection, Demuxing and Decoding
 		pipeline = """
 			videotestsrc !
 			timeoverlay !
-			video/x-raw,width=1920,height=1080 !
+			video/x-raw,width=1920,height=1080,framerate=25/1 !
 		""".format(
 			port=port
 		)
-		if allowoverlay:
+
+		# If an overlay is required, add an cairooverlay-Element into the Video-Path
+		if self.draw_callback:
 			pipeline += """
 				videoconvert !
 				cairooverlay name=overlay !
 				videoconvert !
 			"""
 
+		# Video Display
 		pipeline += """
 			xvimagesink name=v
 		"""
 
-		if audiolevelwidget or playaudio:
+
+		# If an Audio-Path is required, add an Audio-Path through a level-Element
+		if self.level_callback or play_audio:
 			pipeline += """
 				audiotestsrc wave=blue-noise !
 				audio/x-raw !
 				level name=lvl interval=50000000 !
 			"""
 
-			if playaudio:
+			# If Playback is requested, push fo alsa
+			if play_audio:
 				pipeline += """
 					alsasink
 				"""
+
+			# Otherwise just trash the Audio
 			else:
 				pipeline += """
 					fakesink
 				"""
 
-		self.log.info('launching gstreamer-pipeline for widget %s "%s":\n%s', videowidget.get_name(), Gtk.Buildable.get_name(videowidget), pipeline)
-
+		self.log.debug('Creating Display-Pipeline:\n%s', pipeline)
 		self.pipeline = Gst.parse_launch(pipeline)
 
-		videowidget.realize()
-		self.xid = videowidget.get_property('window').get_xid()
+		self.drawing_area.realize()
+		self.xid = self.drawing_area.get_property('window').get_xid()
+		self.log.debug('Realized Drawing-Area with xid %u', self.xid)
 
 		bus = self.pipeline.get_bus()
 		bus.add_signal_watch()
@@ -55,41 +68,26 @@ class VideoDisplay:
 		bus.connect('message::error', self.on_error)
 		bus.connect("sync-message::element", self.on_syncmsg)
 
-		self.draw_callback = None
-
-		if audiolevelwidget:
-			self.levelrms = [0, 0]
-			self.audiolevelwidget = audiolevelwidget
-			self.audiolevelwidget.connect('draw', self.on_level_draw)
+		if self.level_callback:
 			bus.connect("message::element", self.on_level)
 
-	def run(self):
+		if self.draw_callback:
+			self.pipeline.get_by_name('overlay').connect('draw', self.on_draw)
+
+		self.log.debug('Launching Display-Pipeline')
 		self.pipeline.set_state(Gst.State.PLAYING)
 
-	def set_overlay_callback(self, callback):
-		if callback:
-			if not self.draw_callback:
-				self.draw_callback = self.pipeline.get_by_name('overlay').connect('draw', callback)
-		else:
-			print('disconnect')
-			self.pipeline.get_by_name('overlay').disconnect(self.draw_callback)
-			self.draw_callback = None
 
 	def on_syncmsg(self, bus, msg):
 		if msg.get_structure().get_name() == "prepare-window-handle":
-				self.log.info('setting xvimagesink window-handle to %s', self.xid)
+				self.log.info('Setting xvimagesink window-handle to %s', self.xid)
 				msg.src.set_window_handle(self.xid)
 
-	def on_error(self, bus, msg):
-		self.log.error('on_error(): %s', msg.parse_error())
+	def on_error(self, bus, message):
+		self.log.debug('Received Error-Signal on Display-Pipeline')
+		(error, debug) = message.parse_error()
+		self.log.debug('Error-Details: #%u: %s', error.code, debug)
 
-	def on_level_draw(self, widget, cr):
-		cr.set_source_rgb(1, 1, 1)
-		cr.set_line_width(10)
-
-		cr.move_to(15, 0)
-		cr.line_to(15, self.levelrms[0]*-20)
-		cr.stroke()
 
 	def on_level(self, bus, msg):
 		if msg.src.name != 'lvl':
@@ -98,6 +96,9 @@ class VideoDisplay:
 		if msg.type != Gst.MessageType.ELEMENT:
 			return
 
-		self.levelpeaks = msg.get_structure().get_value('peak')
-		self.levelrms = msg.get_structure().get_value('rms')
-		self.audiolevelwidget.queue_draw()
+		peaks = msg.get_structure().get_value('peak')
+		rms = msg.get_structure().get_value('rms')
+		self.level_callback(peaks, rms)
+
+	def on_draw(self, cairooverlay, cr, timestamp, duration):
+		self.draw_callback(cr, timestamp, duration)
diff --git a/voctogui/lib/warningoverlay.py b/voctogui/lib/warningoverlay.py
new file mode 100644
index 0000000..73e1b33
--- /dev/null
+++ b/voctogui/lib/warningoverlay.py
@@ -0,0 +1,55 @@
+import logging
+from gi.repository import GLib
+
+class VideoWarningOverlay:
+	""" Displays a Warning-Overlay above the Video-Feed of another VideoDisplay """
+
+	def __init__(self):
+		self.log = logging.getLogger('VideoWarningOverlay')
+
+		self.text = None
+		self.enabled = False
+		self.blink_state = False
+
+		GLib.timeout_add_seconds(1, self.on_blink_callback)
+
+
+	def on_blink_callback(self):
+		self.blink_state = not self.blink_state
+		return True
+
+	def enable(self, text=None):
+		self.text = text
+		self.enabled = True
+
+	def set_text(self, text=None):
+		self.text = text
+
+	def disable(self):
+		self.enabled = False
+
+	def draw_callback(self, cr, timestamp, duration):
+		if not self.enabled:
+			return
+
+		w = 1920
+		h = 1080/20
+
+		if self.blink_state:
+			cr.set_source_rgba(1.0, 0.0, 0.0, 0.8)
+		else:
+			cr.set_source_rgba(1.0, 0.5, 0.0, 0.8)
+
+		cr.rectangle(0, 0, w, h)
+		cr.fill()
+
+		text = "Stream is Blanked"
+		if self.text:
+			text += ": "+self.text
+
+		cr.set_font_size(h*0.75)
+		xbearing, ybearing, txtwidth, txtheight, xadvance, yadvance = cr.text_extents(text)
+
+		cr.move_to(w/2 - txtwidth/2, h*0.75)
+		cr.set_source_rgba(1.0, 1.0, 1.0, 1.0)
+		cr.show_text(text)