From 7e1390938266616f52ffb247317c269468d1c43b Mon Sep 17 00:00:00 2001 From: Thomas Vander Stichele Date: Tue, 26 May 2009 21:31:21 +0000 Subject: [PATCH] * morituri/common/encode.py (added): * examples/encode.py (added): Add an Encode Task, hardcoded to flac for now. Add an example, that also sets tags. --- ChangeLog | 7 +++ examples/encode.py | 98 +++++++++++++++++++++++++++++++ morituri/common/encode.py | 118 ++++++++++++++++++++++++++++++++++++++ 3 files changed, 223 insertions(+) create mode 100644 examples/encode.py create mode 100644 morituri/common/encode.py diff --git a/ChangeLog b/ChangeLog index 81c9cb1..cc46b16 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,10 @@ +2009-05-26 Thomas Vander Stichele + + * morituri/common/encode.py (added): + * examples/encode.py (added): + Add an Encode Task, hardcoded to flac for now. + Add an example, that also sets tags. + 2009-05-26 Thomas Vander Stichele * morituri/common/task.py: diff --git a/examples/encode.py b/examples/encode.py new file mode 100644 index 0000000..ede39ba --- /dev/null +++ b/examples/encode.py @@ -0,0 +1,98 @@ +# -*- Mode: Python; test-case-name: morituri.test.test_header -*- +# vi:si:et:sw=4:sts=4:ts=4 + +# Morituri - for those about to RIP + +# Copyright (C) 2009 Thomas Vander Stichele + +# This file is part of morituri. +# +# morituri 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. +# +# morituri 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 morituri. If not, see . + +import os +import sys +import optparse +import tempfile +import pickle +import shutil + +import gobject +gobject.threads_init() + +import gst + +import gtk + + +from morituri.common import task, taskgtk, common, encode + +def gtkmain(runner, taskk): + runner.connect('stop', lambda _: gtk.main_quit()) + + window = gtk.Window() + window.add(runner) + window.show_all() + + runner.run(taskk) + + gtk.main() + +def climain(runner, taskk): + runner.run(taskk) + +class Listener(object): + def __init__(self, persister): + self._persister = persister + + def progressed(self, task, value): + pass + + def described(self, task, description): + pass + + def started(self, task): + pass + + def stopped(self, task): + self._persister.object[task.path] = task.trm + print task.path, task.trm + self._persister.persist() + + +def main(argv): + parser = optparse.OptionParser() + + default = 'cli' + parser.add_option('-r', '--runner', + action="store", dest="runner", + help="runner ('cli' or 'gtk', defaults to %s)" % default, + default=default) + + options, args = parser.parse_args(argv[1:]) + + taglist = gst.TagList() + taglist[gst.TAG_ARTIST] = 'Thomas' + taglist[gst.TAG_TITLE] = 'Yes' + taskk = encode.EncodeTask(args[0], args[1], taglist=taglist) + + if options.runner == 'cli': + runner = task.SyncRunner() + function = climain + elif options.runner == 'gtk': + runner = taskgtk.GtkProgressRunner() + function = gtkmain + + function(runner, taskk) + +main(sys.argv) diff --git a/morituri/common/encode.py b/morituri/common/encode.py new file mode 100644 index 0000000..d8e0723 --- /dev/null +++ b/morituri/common/encode.py @@ -0,0 +1,118 @@ +# -*- Mode: Python; test-case-name: morituri.test.test_common_encode -*- +# vi:si:et:sw=4:sts=4:ts=4 + +# Morituri - for those about to RIP + +# Copyright (C) 2009 Thomas Vander Stichele + +# This file is part of morituri. +# +# morituri 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. +# +# morituri 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 morituri. If not, see . + +import os +import struct +import zlib + +import gst + +from morituri.common import common, task + +from morituri.common import log +log.init() + +class EncodeTask(task.Task): + """ + I am a task that encodes a .wav file. + I set tags too. + """ + + description = 'Encoding' + + def __init__(self, inpath, outpath, taglist=None): + """ + """ + self._inpath = inpath + self._outpath = outpath + self._taglist = taglist + + def start(self, runner): + task.Task.start(self, runner) + self._pipeline = gst.parse_launch(''' + filesrc location="%s" ! + decodebin name=decoder ! audio/x-raw-int ! + flacenc name=muxer ! + filesink location="%s" name=sink''' % (self._inpath, self._outpath)) + + + # set tags + if self._taglist: + muxer = self._pipeline.get_by_name('muxer') + muxer.merge_tags(self._taglist, gst.TAG_MERGE_APPEND) + + self.debug('pausing pipeline') + self._pipeline.set_state(gst.STATE_PAUSED) + self._pipeline.get_state() + self.debug('paused pipeline') + + # get length + decoder = self._pipeline.get_by_name('decoder') + self.debug('query duration') + length, format = muxer.query_duration(gst.FORMAT_DEFAULT) + # wavparse 0.10.14 returns in bytes + if format == gst.FORMAT_BYTES: + self.debug('query returned in BYTES format') + length /= 4 + self.debug('total length: %r', length) + self._length = length + + # add a probe so we can track progress + sinkpad = muxer.get_pad('sink') + srcpad = sinkpad.get_peer() + srcpad.add_buffer_probe(self._probe_handler, False) + + # add eos handling + bus = self._pipeline.get_bus() + bus.add_signal_watch() + bus.connect('message::eos', self._eos_cb) + + self.debug('scheduling setting to play') + # since set_state returns non-False, adding it as timeout_add + # will repeatedly call it, and block the main loop; so + # gobject.timeout_add(0L, self._pipeline.set_state, gst.STATE_PLAYING) + # would not work. + + def play(): + self._pipeline.set_state(gst.STATE_PLAYING) + return False + self.runner.schedule(0, play) + + #self._pipeline.set_state(gst.STATE_PLAYING) + self.debug('scheduled setting to play') + + def _probe_handler(self, pad, buffer, ret): + # marshal to main thread + self.runner.schedule(0, self.setProgress, + float(buffer.offset) / self._length) + return True + + def _eos_cb(self, bus, message): + self.debug('eos, scheduling stop') + self.runner.schedule(0, self.stop) + + def stop(self): + self.debug('stopping') + self.debug('setting state to NULL') + self._pipeline.set_state(gst.STATE_NULL) + self.debug('set state to NULL') + task.Task.stop(self)