#!/usr/bin/python

# freeseer - vga/presentation capture software
#
#  Copyright (C) 2010  Free and Open Source Software Learning Centre
#  http://fosslc.org
#
#  This program is free software: you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation, either version 3 of the License, or
#  (at your option) any later version.
#
#  This program is distributed in the hope that 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 General Public License
#  along with this program.  If not, see <http://www.gnu.org/licenses/>.

# For support, questions, suggestions or any other inquiries, visit:
# the #fosslc channel on IRC (freenode.net)

import gobject, pygst
pygst.require("0.10")
import gst

class FreeSeeR:
    def __init__(self, window_id):
        self.window_id = window_id
        
        self.viddrv = 'v4lsrc'
        self.viddev = '/dev/video0'
        self.filename = 'default.mkv'
        self.video_codec = 'theoraenc'
        self.audio_codec = 'vorbisenc'
        
        self.player = gst.Pipeline("player")

        # GST Video
        self.vidsrc = gst.element_factory_make(self.viddrv, "vidsrc")
        self.cspace = gst.element_factory_make("ffmpegcolorspace", "cspace")
        self.vidtee = gst.element_factory_make("tee", "vidtee")
        self.vidqueue1 = gst.element_factory_make("queue", "vidqueue1")
        self.vidqueue2 = gst.element_factory_make("queue", "vidqueue2")
        self.vidcodec = gst.element_factory_make(self.video_codec, "vidcodec")
        self.vidsink = gst.element_factory_make("autovideosink", "vidsink")

        # GST Sound
        self.sndsrc = gst.element_factory_make("alsasrc", "sndsrc")
        self.sndqueue = gst.element_factory_make("queue", "sndqueue")
        self.audioconvert = gst.element_factory_make("audioconvert", "audioconvert")
        self.sndcodec = gst.element_factory_make(self.audio_codec, "sndcodec")

        # GST Muxer
        self.mux = gst.element_factory_make("matroskamux", "mux")
        self.filesink = gst.element_factory_make("filesink", "filesink")
        self.filesink.set_property("location", 'default.mkv')

        # GST Add Components
        self.player.add(self.vidsrc, self.cspace, self.vidtee, self.vidqueue1, self.vidqueue2, self.vidcodec, self.vidsink)
        self.player.add(self.sndsrc, self.sndqueue, self.audioconvert, self.sndcodec)
        self.player.add(self.mux, self.filesink)

        # GST Link Components
        gst.element_link_many(self.vidsrc, self.cspace, self.vidtee)
        gst.element_link_many(self.vidtee, self.vidqueue1, self.vidcodec, self.mux)
        gst.element_link_many(self.vidtee, self.vidqueue2, self.vidsink)
        gst.element_link_many(self.sndsrc, self.sndqueue, self.audioconvert, self.sndcodec, self.mux)
        gst.element_link_many(self.mux, self.filesink)

        bus = self.player.get_bus()
        bus.add_signal_watch()
        bus.enable_sync_message_emission()
        bus.connect("message", self.on_message)
        bus.connect("sync-message::element", self.on_sync_message)

    def on_message(self, bus, message):
        t = message.type
        if t == gst.MESSAGE_EOS:
            self.player.set_state(gst.STATE_NULL)
        elif t == gst.MESSAGE_ERROR:
            err, debug = message.parse_error()
            print "Error: %s" % err, debug
            self.player.set_state(gst.STATE_NULL)

    def on_sync_message(self, bus, message):
        if message.structure is None:
            return
        message_name = message.structure.get_name()
        if message_name == "prepare-xwindow-id":
            imagesink = message.src
            imagesink.set_property("force-aspect-ratio", True)
            imagesink.set_xwindow_id(self.window_id)

    def change_videosrc(self, new_source, new_device):
        self.viddrv = new_source
        self.viddev = new_device
        self.player.remove(self.vidsrc)
        self.vidsrc = gst.element_factory_make(self.viddrv, "vidsrc")
        self.player.add(self.vidsrc)
        if (self.viddrv == 'v4lsrc'):
            self.vidsrc.set_property("device", self.viddev)
        elif (self.viddrv == 'v4l2src'):
            self.vidsrc.set_property("device", self.viddev)
        gst.element_link_many(self.vidsrc, self.cspace)

    def record(self, filename):
        self.filename = filename
        self.filesink.set_property("location", self.filename)
        self.player.set_state(gst.STATE_PLAYING)

    def stop(self):
        self.player.set_state(gst.STATE_NULL)

    def change_video_codec(self, new_vcodec):
        self.video_codec = new_vcodec
        self.player.remove(self.vidcodec)
        self.vidcodec = gst.element_factory_make(self.video_codec, "vidcodec")
        self.player.add(self.vidcodec)
        gst.element_link_many(self.vidqueue1, self.vidcodec, self.mux)

    def change_audio_codec(self, new_acodec):
        self.audio_codec = new_acodec
        self.player.remove(self.sndcodec)
        self.sndcodec = gst.element_factory_make(self.audio_codec, "sndcodec")
        self.player.add(self.sndcodec)
        gst.element_link_many(self.audioconvert, self.sndcodec, self.mux)