root / code / trunk / backend / agents / video.py

# Copyright (C) 2007 SPARTA, Inc.
# This software is licensed under the GPLv3 license, included in
# ./GPLv3-LICENSE.txt in the source distribution

import os
import agent
from distributions import *
from seerplatform import run, spawn

class StreamingVideoAgent(agent.TrafgenAgent):

        agenttype = 'VIDEO'

        def __init__(self):
                agent.TrafgenAgent.__init__(self)
                self.addVarType('file', 'string', 'run1150')
                self.addVarType('type', 'string', 'rtp')
                self.addVarType('time', 'string', '100')

        def serverExec(self):
                if (self.whiteboard.incAndGet('VIDEO_SERVER_COUNT') == 1):
                        self.log.info("Starting video server")
                        spawn("SEERVideoServer %s/data" % (os.environ.get('SEERBASE')), self.log)

        def serverStop(self):
                if (self.whiteboard.decAndGet('VIDEO_SERVER_COUNT')  == 0):
                        self.log.info("Stopping video server")
                        run("killall SEERVideoServer", self.log)

        def clientInit(self):
                """ Save a copy of current variables for client executions """
                self.file = self.getVar("file") ## filename to use 'minus the .ts part'
                self.type = self.getVar("type") ## udp, rtp or tcp
                self.time = self.getVar("time") ## length to keep streaming

        def clientExec(self, src, dst, size):
                cmd = "openRTSP -d %d " % (int(eval(self.time)))
                if (src is not None):
                        cmd += "-I %s " % (src)

                if (self.type == 'tcp'):
                        cmd += "-t rtsp://%s:8554/%s.rtp " % (dst, self.file)
                else:
                        cmd += "rtsp://%s:8554/%s.%s " % (dst, self.file, self.type)

                # We print this on stdout so it goes with redirected stdout
                print("Starting " + cmd)
                subpid = spawn(cmd, None)