dutMux.py

#!/usr/bin/env python
#
# Script to record webcam and screencast
#
# Copyright 2012 Intel Corporation.
#
# Author: Michael Wood <michael.g.wood@intel.com>
#
# This program is free software; you can redistribute it and/or modify it
# under the terms and conditions of the GNU Lesser General Public License,
# version 2.1, as published by the Free Software Foundation.
#
# This program is distributed in the hope it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
# details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this program; if not, see <http://www.gnu.org/licenses>
#

from gi.repository import GLib
from gi.repository import Gdk
import gst
import time
import subprocess

class Muxer:
    def __init__(self, projectDir, posX, posY):

      self.projectDir = projectDir

      primaryLocation = "\""+projectDir+"/primary-dut.webm\""
      secondaryLocation = "\""+projectDir+"/secondary-dut.webm\""
      outLocation = "\""+projectDir+"/user-testing.webm\""
      finalLocation = "\""+projectDir+"/final.webm\""

      #posX, posY = self.get_primary_video_info (primaryLocation)

      gstPipe = """filesrc
      location="""+secondaryLocation+""" name=filein !
      matroskademux name=demux1 ! queue !
      vp8dec ! videorate force-fps=15/1 !
      video/x-raw-yuv,width=320,height=240,framerate=15/1 !
      queue ! videomixer name=mix
      sink_0::xpos=0 sink_0::ypos=0
      sink_1::xpos="""+str (posX)+"""
      sink_1::ypos="""+str (posY)+""" ! vp8enc
      quality=10 speed=2 threads=4 ! webmmux name=outmux !
      filesink location="""+outLocation+"""
      filesrc  location="""+primaryLocation+"""
      ! matroskademux name=demux2 ! queue !
      vp8dec ! videorate force-fps=15/1 ! videoscale add-borders=1 !
      video/x-raw-yuv,framerate=15/1 ! mix."""

      self.element = gst.parse_launch (gstPipe)

      pipebus = self.element.get_bus ()

      pipebus.add_signal_watch ()
      pipebus.connect ("message", self.pipe1_changed_cb)

      #second pass add audio - we could do this in the above pipeline but due to a bug it doesn't quite work..
      gstPipe = """filesrc
      location="""+secondaryLocation+""" ! queue ! matroskademux !
      vorbisparse ! audio/x-vorbis !  queue ! outmux.audio_0
      filesrc location="""+outLocation+""" ! queue !
      matroskademux ! video/x-vp8 ! queue ! outmux.video_0 webmmux
      name=outmux ! filesink location="""+finalLocation+""""""

      self.element2 = gst.parse_launch (gstPipe)

      pipebus2 = self.element2.get_bus ()

      pipebus2.add_signal_watch ()
      pipebus2.connect ("message", self.pipe2_changed_cb)


    def pipe2_changed_cb (self, bus, message):
      if message.type == gst.MESSAGE_ERROR:
          err, debug = message.parse_error()
          print ("Err: dutMux pipe2: %s" % err, debug)

      if message.type == gst.MESSAGE_EOS:
          print ("Info: Second pass done")

    def pipe1_changed_cb (self, bus, message):
      if message.type == gst.MESSAGE_EOS:
          print ("Info: Done first pass, starting second pass")
          self.element2.set_state (gst.STATE_PLAYING)
      if message.type == gst.MESSAGE_ERROR:
          err, debug = message.parse_error()
          print ("Err: dutMux pipe1: %s" % err, debug)

    def pipe_report (self):
        positionMs, format = self.element.query_position (gst.FORMAT_TIME, None)
        durationMs, format = self.element.query_duration (gst.FORMAT_TIME, None)

        duration = round (durationMs*0.000000001)
        position = round (positionMs*0.000000001)

        print ("position %f" % position)
        print ("duration %f" % duration)

        percentDone = int ((position/duration)*100)

        # HACK because sometimes gstreamer gets the duration wrong :(
        if percentDone > 100:
            percentDone = 100

        print ("Percent done %d" % percentDone)

        return percentDone

    def record (self, start):
      if start == 1:
        print ("Start mux record")
        self.element.set_state (gst.STATE_PLAYING)

        self.element.get_state (gst.CLOCK_TIME_NONE)
      else:
        print ("stop mux record")
        self.element.set_state (3)
        self.element.set_state (1)