minimalistic start of a level-meter

1 files changed, 48 insertions, 6 deletions

diff --git a/voctogui/lib/videodisplay.py b/voctogui/lib/videodisplay.py
index 6537b78..b98daf1 100644
--- a/voctogui/lib/videodisplay.py
+++ b/voctogui/lib/videodisplay.py
@@ -4,23 +4,40 @@ from gi.repository import Gst, Gtk
 class VideoDisplay:
 	""" Displays a Voctomix-Video-Stream into a GtkWidget """
 
-	def __init__(self, port, widget):
+	def __init__(self, port, videowidget, audiolevelwidget=None, playaudio=False):
 		self.log = logging.getLogger('VideoDisplay[%u]' % port)
 
 		pipeline = """
 			videotestsrc !
 			timeoverlay !
 			video/x-raw,width=1920,height=1080 !
-			xvimagesink
+			xvimagesink name=v
 		""".format(
 			port=port
 		)
-		self.log.info('launching gstreamer-pipeline for widget %s "%s":\n%s', widget.get_name(), Gtk.Buildable.get_name(widget), pipeline)
+
+		if audiolevelwidget or playaudio:
+			pipeline += """
+				audiotestsrc wave=blue-noise !
+				audio/x-raw !
+				level name=lvl interval=50000000 !
+			"""
+
+			if playaudio:
+				pipeline += """
+					alsasink
+				"""
+			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.pipeline = Gst.parse_launch(pipeline)
 
-		widget.realize()
-		self.xid = widget.get_property('window').get_xid()
+		videowidget.realize()
+		self.xid = videowidget.get_property('window').get_xid()
 
 		bus = self.pipeline.get_bus()
 		bus.add_signal_watch()
@@ -29,6 +46,12 @@ class VideoDisplay:
 		bus.connect('message::error', self.on_error)
 		bus.connect("sync-message::element", self.on_syncmsg)
 
+		if audiolevelwidget:
+			self.levelrms = [0, 0]
+			self.audiolevelwidget = audiolevelwidget
+			self.audiolevelwidget.connect('draw', self.on_level_draw)
+			bus.connect("message::element", self.on_level)
+
 	def run(self):
 		self.pipeline.set_state(Gst.State.PLAYING)
 
@@ -38,4 +61,23 @@ class VideoDisplay:
 				msg.src.set_window_handle(self.xid)
 
 	def on_error(self, bus, msg):
-			self.log.error('on_error():', msg.parse_error())
+		self.log.error('on_error():', msg.parse_error())
+
+	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':
+			return
+
+		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()