gloss-mc/VideoController.py

import sys, clutter, clutter.cluttergst, gst, pygst, gtk, pygtk
import threading


class VideoController:

    def __init__(self, stage):
        self.stage = stage
        self.overlay = None
        # Primary video texture & sink definition
        self.video_texture = clutter.cluttergst.VideoTexture()
        self.video_sink = clutter.cluttergst.VideoSink(self.video_texture)
        self.video_texture.connect('size-change', self.set_fullscreen)
        self.video_texture.set_position(0,0)
        
        self.pipeline = gst.Pipeline()
        self.osd = osd(stage)
        """
        src = gst.element_factory_make ("videotestsrc")
        
        gst.Bin.add (self.pipeline, filesrc, colorspace, deinterlace, video_sink)
        gst.element_link_many (filesrc, colorspace, deinterlace, video_sink)
        """
        
    def on_key_press_event(self, event):
        if event.keyval == clutter.keysyms.Left:
            self.skip(-20)
        if event.keyval == clutter.keysyms.Right:
            self.skip(20)
            
        #self.osd.enter()
    
    def play_video(self, uri):
        self.stage.add(self.video_texture)
        self.video_texture.set_uri(uri)
        self.video_texture.show()
        
        self.video_texture.set_playing(True)
        self.isPlaying = True
        
        return self.video_texture
    
    def stop_video(self):
        if self.video_texture.get_playing():
            self.isPlaying = False
            self.video_texture.set_playing(False)
            
            timeline = clutter.Timeline(15, 25)
            timeline.connect('completed', self.end_video_event)
            alpha = clutter.Alpha(timeline, clutter.ramp_inc_func)
            behaviour = clutter.BehaviourOpacity(alpha, 255,0)
            behaviour.apply(self.video_texture)
        
            timeline.start()
    def end_video_event(self, data):
        self.stage.remove(self.video_texture) 
        
    def customBin(self):
        self.src = gst.element_factory_make("filesrc", "src");
        self.src.set_property("location", "/home/josh/clutter/toys/gloss/test.mpg")
        self.demux = gst.element_factory_make("ffdemux_mpegts", "demux")
        self.queue1 = gst.element_factory_make("queue", "queue1")
        self.queue2 = gst.element_factory_make("queue", "queue2")
        #self.deinterlace = gst.element_factory_make("ffdeinterlace", "deinterlace")
        self.vdecode = gst.element_factory_make("mpeg2dec", "vdecode")
        self.adecode = gst.element_factory_make("mad", "adecode")
        #self.vsink = gst.element_factory_make("xvimagesink", "vsink")
        self.vsink = self.video_sink #cluttergst.VideoSink(self.video_texture)
        self.asink = gst.element_factory_make("alsasink", "asink")

        
        # add elements to the pipeline
        self.pipeline.add(self.src)
        self.pipeline.add(self.demux)
        self.pipeline.add(self.queue1)
        self.pipeline.add(self.queue2)
        self.pipeline.add(self.vdecode)
        #self.pipeline.add(self.deinterlace)
        self.pipeline.add(self.adecode)
        self.pipeline.add(self.vsink)
        self.pipeline.add(self.asink)

        bus = self.pipeline.get_bus()
        gst.Bus.add_signal_watch (bus) 
       
        # we can't link demux until the audio and video pads are added
        # we need to listen for "pad-added" signals
        self.demux.connect("pad-added", self.on_pad_added)
        
        # link all elements apart from demux
        gst.element_link_many(self.src, self.demux)
        gst.element_link_many(self.queue1, self.vdecode, self.vsink) #self.deinterlace, self.vsink)
        gst.element_link_many(self.queue2, self.adecode, self.asink)

    def on_pad_added(self, element, src_pad):
        caps = src_pad.get_caps()
        name = caps[0].get_name()
        # link demux to vdecode when video/mpeg pad added to demux
        if name == "video/mpeg":
            sink_pad = self.queue1.get_pad("sink")
        elif name == "audio/mpeg":
            sink_pad = self.queue2.get_pad("sink")
        else:
            return
        if not sink_pad.is_linked():
            src_pad.link(sink_pad)
        
    def set_fullscreen(self, texture, width, height):
        texture.set_property("sync-size", False)
        texture.set_position(0, 0)
        xy_ratio = float(height) / float(width)
        #print "XY Ratio: " + str(xy_ratio)
        
        width = int(self.stage.get_width())
        height = int (width * xy_ratio)
        #height = 768
        
        texture.set_size(width, height)
        
    def pause_video(self):
        #Use the overlay to go over show
        if self.overlay == None:
            self.overlay = clutter.Texture()
            pixbuf = gtk.gdk.pixbuf_new_from_file('ui/backdrop.png')
            self.overlay.set_pixbuf(pixbuf)
            self.overlay.set_width(self.stage.get_width())
            self.overlay.set_height(self.stage.get_height())
        self.overlay.set_opacity(0)
        self.overlay.show()

        self.stage.add(self.overlay)
        #self.video_texture.lower_actor(self.overlay)
        #self.overlay.raise_actor(self.video_texture)
        #Fade the overlay in
        timeline_overlay = clutter.Timeline(10,30)
        alpha = clutter.Alpha(timeline_overlay, clutter.ramp_inc_func)
        overlay_behaviour = clutter.BehaviourOpacity(alpha, 0, 230)
        #video_behaviour = clutter.BehaviourOpacity(alpha, 255, 80)
        overlay_behaviour.apply(self.overlay)
        #video_behaviour.apply(self.video_texture)
        timeline_overlay.start()
        
        #Pause the video
        self.video_texture.set_playing(False)
        
    def unpause_video(self):
        #Fade the backdrop in
        timeline_unpause = clutter.Timeline(10,30)
        alpha = clutter.Alpha(timeline_unpause, clutter.ramp_inc_func)
        overlay_behaviour = clutter.BehaviourOpacity(alpha, 230, 0)
        #video_behaviour = clutter.BehaviourOpacity(alpha, 80, 255)
        overlay_behaviour.apply(self.overlay)
        #video_behaviour.apply(self.video_texture)
        timeline_unpause.start()
        
        #Resume the video
        self.video_texture.set_playing(True)
        
    def skip(self, amount):
        if not self.video_texture.get_can_seek():
            return
        
        current_pos = self.video_texture.get_position()
        new_pos = int(current_pos + amount)
        
        if new_pos >= self.video_texture.get_duration():
            new_pos = self.video_texture.get_duration()-1
        if new_pos <= 0:
            new_pos = 1
        
        # There's apparently a collision in the python bindings with the following method. Change this when its fixed in the bindings
        #self.video_texture.set_position(new_pos)
        #Until then use:
        self.video_texture.set_property("position", int(new_pos))
        self.osd.shift_video(self.video_texture, amount)
        
import time

class osd:

    def __init__(self, stage):
        self.stage = stage
        self.timerRunning = False
        
        self.bar_group = clutter.Group()
    
        self.background = clutter.Texture()
        self.background.set_pixbuf( gtk.gdk.pixbuf_new_from_file("ui/osd_bar3.png") )
        self.background.set_opacity(255)
        self.background.set_width(stage.get_width())
        self.bar_group.add(self.background)
        self.bar_group.show_all()
        
    def enter(self):
        self.stage.add(self.bar_group)
        self.bar_group.show()
        
        self.bar_group.set_position(0, self.stage.get_height())
        bar_position_y = int(self.stage.get_height() - self.background.get_height())
        
        knots = (\
                (self.bar_group.get_x(), self.bar_group.get_y()),\
                (self.bar_group.get_x(), bar_position_y) \
                )
                
        self.timeline = clutter.Timeline(25, 50)
        self.alpha = clutter.Alpha(self.timeline, clutter.ramp_inc_func)
        self.enter_behaviour_path = clutter.BehaviourPath(self.alpha, knots)
        
        self.enter_behaviour_path.apply(self.bar_group)
        
        self.timeline.start()
        
        self.timer = threading.Timer(3.0, self.exit)
        self.timer.start()
        
    def exit(self):
        
        knots = (\
                (self.bar_group.get_x(), self.bar_group.get_y()),\
                (self.bar_group.get_x(), int(self.stage.get_height())) \
                )
                
        self.timeline = clutter.Timeline(25, 50)
        self.timeline.connect('completed', self.exit_end_event)
        self.alpha = clutter.Alpha(self.timeline, clutter.ramp_inc_func)
        self.exit_behaviour_path = clutter.BehaviourPath(self.alpha, knots)
        
        self.exit_behaviour_path.apply(self.bar_group)
        self.timeline.start()
    def exit_end_event(self, data):
        self.stage.remove(self.bar_group)
    
    #Is called when the video is skipped forwards or backwards
    def shift_video(self, video, shift_amount):
        #Firstly check whether the label is already there from last time
        if self.timerRunning:
            self.timer.cancel()
            self.timer = threading.Timer(1.5, self.label_exit)
            self.timerRunning = True
            self.timer.start()
            return
            
        shiftDistance = 100
        
        self.shift_label = clutter.Label()
        self.shift_label.set_font_name("Lucida Grande 60")
        self.shift_label.set_opacity(0)
        self.shift_label.set_color(clutter.color_parse('White'))
    
        #Set the string for the fast forward / rewind as well as the 
        if shift_amount > 0:
            self.shift_label.set_text("+" + str(shift_amount) + "s >")
            shift_label_x = int(self.stage.get_width() - self.shift_label.get_width() - shiftDistance)
            direction = 1
        else:
            self.shift_label.set_text("< " + str(shift_amount) + "s")
            shift_label_x = int(0 + shiftDistance)
            direction = -1
        
        shift_label_y = int(self.stage.get_height() - self.shift_label.get_height()) 
        self.shift_label.set_position( shift_label_x, shift_label_y )
        incoming_label_knots = (\
            ( shift_label_x, shift_label_y ),\
            ( int(shift_label_x + (shiftDistance*direction)), shift_label_y )\
            )
        
        self.incoming_text_timeline = clutter.Timeline(20, 60)
        alpha = clutter.Alpha(self.incoming_text_timeline, clutter.ramp_inc_func)
        behaviour1 = clutter.BehaviourPath(alpha, incoming_label_knots)
        behaviour2 = clutter.BehaviourOpacity(alpha, 0, 120)
        
        behaviour1.apply(self.shift_label)
        behaviour2.apply(self.shift_label)
        self.stage.add(self.shift_label)
        self.shift_label.show()
        
        self.timer = threading.Timer(1.5, self.label_exit)
        self.timerRunning = True
        self.timer.start()
        
        self.incoming_text_timeline.start()
        #print time.strftime("%H:%M:%S", time.gmtime(amount))
        
    def label_exit(self):
        self.timerRunning = False
        #Check which way this label needs to go
        if self.shift_label.get_text()[0] == "<":
            end_x = int(self.shift_label.get_width() * -1)
        else:
            end_x = int(self.stage.get_width())
        
        (starting_pos_x, starting_pos_y) = self.shift_label.get_abs_position()
        outgoing_label_knots = (\
        ( starting_pos_x, starting_pos_y ),\
        ( end_x, starting_pos_y )\
        )
        
        self.outgoing_text_timeline = clutter.Timeline(20, 60)
        self.outgoing_text_timeline.connect('completed', self.removeLabel)
        alpha = clutter.Alpha(self.outgoing_text_timeline, clutter.ramp_inc_func)
        behaviour1 = clutter.BehaviourPath(alpha, outgoing_label_knots)
        behaviour2 = clutter.BehaviourOpacity(alpha, self.shift_label.get_opacity() , 0)
        
        behaviour1.apply(self.shift_label)
        behaviour2.apply(self.shift_label)
        
        self.outgoing_text_timeline.start()
        
    def removeLabel(self, data):
        self.stage.remove(self.shift_label)