mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-23 00:36:51 +00:00
examples/remuxer.py: Code dump, work in progress...
Original commit message from CVS: 2006-04-07 Andy Wingo <wingo@pobox.com> * examples/remuxer.py: Code dump, work in progress...
This commit is contained in:
parent
2afe4a0c32
commit
6bdf0cec7b
2 changed files with 159 additions and 18 deletions
|
@ -1,3 +1,7 @@
|
|||
2006-04-07 Andy Wingo <wingo@pobox.com>
|
||||
|
||||
* examples/remuxer.py: Code dump, work in progress...
|
||||
|
||||
2006-04-07 Edward Hervey <edward@fluendo.com>
|
||||
|
||||
* testsuite/test_ghostpad.py:
|
||||
|
|
|
@ -51,6 +51,9 @@ class GstPlayer:
|
|||
def set_location(self, location):
|
||||
self.player.set_property('uri', location)
|
||||
|
||||
def get_location(self):
|
||||
return self.player.get_property('uri')
|
||||
|
||||
def query_position(self):
|
||||
"Returns a (position, duration) tuple"
|
||||
try:
|
||||
|
@ -253,24 +256,33 @@ class ProgressDialog(gtk.Dialog):
|
|||
progress.show()
|
||||
vbox.pack_start(progress, False)
|
||||
|
||||
self.progresstext = label = gtk.Label('<i>%s</i>' % task)
|
||||
self.progresstext = label = gtk.Label('')
|
||||
label.set_use_markup(True)
|
||||
label.set_alignment(0.0, 0.0)
|
||||
label.show()
|
||||
vbox.pack_start(label)
|
||||
self.set_task(task)
|
||||
|
||||
def set_task(self, task):
|
||||
self.progresstext.set_markup('<i>%s</i>' % task)
|
||||
|
||||
UNKNOWN = 0
|
||||
SUCCESS = 1
|
||||
FAILURE = 2
|
||||
CANCELLED = 3
|
||||
|
||||
class RemuxProgressDialog(ProgressDialog):
|
||||
def __init__(self, parent, start, stop):
|
||||
ProgressDialog.__init__(self,
|
||||
"Writing to disk",
|
||||
('Your ears need to be periodically cleaned to '
|
||||
'maintain a decent level of hearing. The '
|
||||
'process will be quick and painless.'),
|
||||
'Removing excess ear wax with a golf pencil',
|
||||
('Writing the selected segment of your '
|
||||
'media file to disk. This may take some '
|
||||
'time depending on the file size.'),
|
||||
'Starting media pipeline',
|
||||
parent,
|
||||
gtk.DIALOG_MODAL | gtk.DIALOG_DESTROY_WITH_PARENT,
|
||||
(gtk.STOCK_CANCEL, gtk.RESPONSE_REJECT,
|
||||
gtk.STOCK_CLOSE, gtk.RESPONSE_ACCEPT))
|
||||
(gtk.STOCK_CANCEL, CANCELLED,
|
||||
gtk.STOCK_CLOSE, SUCCESS))
|
||||
self.start = start
|
||||
self.stop = stop
|
||||
self.update_position(start)
|
||||
|
@ -285,33 +297,150 @@ class RemuxProgressDialog(ProgressDialog):
|
|||
self.progress.set_fraction(1.0 - float(remaining) / (self.stop - self.start))
|
||||
|
||||
def set_completed(self, completed):
|
||||
self.set_response_sensitive(gtk.RESPONSE_ACCEPT, completed)
|
||||
self.set_response_sensitive(CANCELLED, not completed)
|
||||
self.set_response_sensitive(SUCCESS, completed)
|
||||
|
||||
class Remuxer(gst.Pipeline):
|
||||
|
||||
__gsignals__ = {'done': (gobject.SIGNAL_RUN_LAST, None, (int,))}
|
||||
|
||||
def __init__(self, fromuri, touri, start, stop):
|
||||
gst.Pipeline.__init__(self)
|
||||
self.__gobject_init__()
|
||||
|
||||
assert start >= 0
|
||||
assert stop > start
|
||||
|
||||
self.src = gst.element_make_from_uri(gst.URI_SRC, fromuri)
|
||||
self.remuxbin = RemuxBin()
|
||||
self.sink = gst.element_make_from_uri(gst.URI_SINK, touri)
|
||||
self.resolution = UNKNOWN
|
||||
|
||||
self.add(self.src, self.remuxbin, self.sink)
|
||||
|
||||
self.src.link(self.remuxbin)
|
||||
self.remuxbin.link(self.sink)
|
||||
|
||||
self.window = None
|
||||
self.pdialog = None
|
||||
|
||||
self.start_time = start
|
||||
self.stop_time = stop
|
||||
|
||||
def _start_queries(self):
|
||||
pass
|
||||
|
||||
def _stop_queries(self):
|
||||
pass
|
||||
|
||||
def _bus_watch(self, bus, message):
|
||||
if message.type == gst.MESSAGE_ERROR:
|
||||
print 'error', message
|
||||
self._stop_queries()
|
||||
m = gtk.MessageDialog(self.window,
|
||||
gtk.DIALOG_MODAL|gtk.DIALOG_DESTROY_WITH_PARENT,
|
||||
gtk.MESSAGE_ERROR,
|
||||
gtk.BUTTONS_CLOSE,
|
||||
"Error processing file")
|
||||
txt = 'There was an error processing your file: %r' % message
|
||||
m.format_secondary_text(txt)
|
||||
m.run()
|
||||
m.destroy()
|
||||
self.response(FAILURE)
|
||||
elif message.type == gst.MESSAGE_WARNING:
|
||||
print 'warning', message
|
||||
elif message.type == gst.MESSAGE_EOS:
|
||||
print 'eos, woot'
|
||||
self.pdialog.set_task('Finished')
|
||||
self.pdialog.update_position(self.stop_time)
|
||||
self._stop_queries()
|
||||
self.pdialog.set_completed(True)
|
||||
elif message.type == gst.MESSAGE_STATE_CHANGED:
|
||||
if message.src == self:
|
||||
print message
|
||||
old, new, pending = message.parse_state_changed()
|
||||
if ((old, new, pending) ==
|
||||
(gst.STATE_READY, gst.STATE_PAUSED,
|
||||
gst.STATE_VOID_PENDING)):
|
||||
self.pdialog.set_task('Processing file')
|
||||
self.pdialog.update_position(self.start_time)
|
||||
self._start_queries()
|
||||
self.set_state(gst.STATE_PLAYING)
|
||||
|
||||
def response(self, response):
|
||||
assert self.resolution == UNKNOWN
|
||||
self.resolution = response
|
||||
self.set_state(gst.STATE_NULL)
|
||||
self.pdialog.destroy()
|
||||
self.pdialog = None
|
||||
self.emit('done', response)
|
||||
|
||||
def start(self, main_window):
|
||||
bus = self.get_bus()
|
||||
bus.add_signal_watch()
|
||||
bus.connect('message', self._bus_watch)
|
||||
self.window = main_window
|
||||
if self.window:
|
||||
# can be None if we are debugging...
|
||||
self.window.set_sensitive(False)
|
||||
self.pdialog = RemuxProgressDialog(main_window, self.start_time,
|
||||
self.stop_time)
|
||||
self.pdialog.show()
|
||||
self.pdialog.connect('response', lambda w, r: self.response(r))
|
||||
self.set_state(gst.STATE_PAUSED)
|
||||
|
||||
def run(self, main_window):
|
||||
self.start(None)
|
||||
loop = gobject.MainLoop()
|
||||
self.connect('done', lambda *x: gobject.idle_add(loop.quit))
|
||||
loop.run()
|
||||
return self.resolution
|
||||
|
||||
class RemuxBin(gst.Bin):
|
||||
def __init__(self):
|
||||
self.__gobject_init__()
|
||||
|
||||
self.parsers = self._find_parsers()
|
||||
|
||||
self.demux = gst.element_factory_make('oggdemux')
|
||||
self.mux = gst.element_factory_make('oggmux')
|
||||
self.sink = gst.element_make_from_uri(gst.URI_SINK, touri)
|
||||
|
||||
self.add(self.src, self.demux, self.mux, self.sink)
|
||||
self.add(self.demux, self.mux)
|
||||
|
||||
self.src.link(self.demux)
|
||||
self.mux.link(self.sink)
|
||||
self.add_pad(gst.GhostPad('sink', self.demux.get_pad('sink')))
|
||||
self.add_pad(gst.GhostPad('src', self.mux.get_pad('src')))
|
||||
|
||||
self.demux.connect('pad-added', self._new_demuxed_pad)
|
||||
self.demux.connect('no-more-pads', self._no_more_pads)
|
||||
|
||||
def _find_parsers(self):
|
||||
registry = gst.registry_get_default()
|
||||
ret = {}
|
||||
for f in registry.get_feature_list(gst.ElementFactory):
|
||||
if f.get_klass().find('Parser') >= 0:
|
||||
for t in f.get_static_pad_templates():
|
||||
if t.direction == gst.PAD_SINK:
|
||||
for s in t.get_caps():
|
||||
ret[s.get_name()] = f.get_name()
|
||||
break
|
||||
return ret
|
||||
|
||||
def _new_demuxed_pad(self, element, pad):
|
||||
pad.link(self.mux.get_pad('sink_%d'))
|
||||
format = pad.get_caps()[0].get_name()
|
||||
|
||||
if format not in self.parsers:
|
||||
self.async_error("Unsupported media type: %s", format)
|
||||
return
|
||||
|
||||
parser = gst.element_factory_make(self.parsers[format])
|
||||
self.add(parser)
|
||||
parser.set_state(gst.STATE_PAUSED)
|
||||
pad.link(parser.get_compatible_pad(pad))
|
||||
parser.link(self.mux)
|
||||
|
||||
def _no_more_pads(self, element):
|
||||
# this is when we should commit to paused or something
|
||||
pass
|
||||
|
||||
def run(self):
|
||||
self.set_state(gst.STATE_PLAYING)
|
||||
|
||||
|
||||
class PlayerWindow(gtk.Window):
|
||||
UPDATE_INTERVAL = 500
|
||||
|
@ -406,7 +535,7 @@ class PlayerWindow(gtk.Window):
|
|||
buttonbox.show()
|
||||
table.attach(buttonbox, 2, 3, 1, 2, 0, 0)
|
||||
|
||||
button = gtk.Button("_Render to disk")
|
||||
button = gtk.Button("_Write to disk")
|
||||
button.set_property('image',
|
||||
gtk.image_new_from_stock(gtk.STOCK_SAVE_AS,
|
||||
gtk.ICON_SIZE_BUTTON))
|
||||
|
@ -415,6 +544,14 @@ class PlayerWindow(gtk.Window):
|
|||
|
||||
self.cutin.connect('notify::time', lambda *x: self.check_cutout())
|
||||
self.cutout.connect('notify::time', lambda *x: self.check_cutin())
|
||||
button.connect('clicked', lambda *x: self.do_remux())
|
||||
|
||||
def do_remux(self):
|
||||
in_uri = self.player.get_location()
|
||||
out_uri = in_uri[:-4] + '-remuxed.ogg'
|
||||
r = Remuxer(in_uri, out_uri,
|
||||
self.cutin.get_time(), self.cutout.get_time())
|
||||
r.run(self)
|
||||
|
||||
def check_cutout(self):
|
||||
if self.cutout.get_time() <= self.cutin.get_time():
|
||||
|
|
Loading…
Reference in a new issue