import pygst

pygst.require("0.10")

import gst

import gobject

class Media:

    def __init__(self, config):

        # set gstreamer basic options

        self.config = config

        self.pipe = ""

    def setup(self, filename, mux, vcodec, vbitrate,\

              fps, acodec, abitrate, width, height, port):

        self.filename = filename

        self.mux = mux

        self.vcodec = vcodec

        self.vbitrate = int(vbitrate)

        self.fps = int(fps)

        self.acodec = acodec

        self.abitrate = int(abitrate)

        self.width = int(width)

        self.height = int(height)

        self.port = int(port)

        ## Pipelines

        #queue ! videoscale ! video/x-raw-yuv,width=240,height=144\

        #! videorate ! ffenc_h263p bitrate=256000 me-method=2 \

        #! rtph263ppay ! udpsink  host=224.0.0.1 port=5000

        #audio/x-raw-int ! queue ! audioconvert ! faac ! rtpmp4gpay\

        #!  udpsink name=upd_audio host=224.0.0.1 port=5002

        self.pipe = "filesrc location=%s ! decodebin name=d ! queue ! videoscale !"\

                    "video/x-raw-yuv,width=(int)%d,height=(int)%d ! ffenc_h263p bitrate=%d"\

                    " me-method=2 ! rtph263ppay ! udpsink host=224.0.0.1 port=%d d. ! "\

                    "queue  ! audioconvert ! faac ! rtpmp4gpay ! udpsink name=udp_audio "\

                    "host=224.0.0.1 port=%d" % (self.filename, self.width, self.height,\

                                                self.vbitrate, self.port, self.port+2)

        #self.pipe = "filesrc location=/tmp/mpg/cpm.mpg ! decodebin ! ffmpegcolorspace ! ximagesink"

        self.pipeline = gst.parse_launch(self.pipe)

    def play(self):

        print "Trying to play pipeline: %s" % self.pipe

        try:

            if (self.pipeline):

                self.pipeline.set_state(gst.STATE_PLAYING)

        except gobject.GError, e:

            print "Error: " + str(e)

    def stop(self):

        print "Trying to stop pipeline: %s" % self.pipe

        try:

            if (self.pipeline):

                self.pipeline.set_state(gst.STATE_NULL)

        except gobject.GError, e:

            print "Error: " + str(e)