+++ /dev/null
-# Ogg Theora support.
-#
-# Copyright 2006 Joe Wreschnig <piman@sacredchao.net>
-#
-# This program is free software; you can redistribute it and/or modify
-# it under the terms of the GNU General Public License version 2 as
-# published by the Free Software Foundation.
-#
-# $Id: oggtheora.py 4275 2008-06-01 06:32:37Z piman $
-
-"""Read and write Ogg Theora comments.
-
-This module handles Theora files wrapped in an Ogg bitstream. The
-first Theora stream found is used.
-
-Based on the specification at http://theora.org/doc/Theora_I_spec.pdf.
-"""
-
-__all__ = ["OggTheora", "Open", "delete"]
-
-import struct
-
-from mutagen._vorbis import VCommentDict
-from mutagen.ogg import OggPage, OggFileType, error as OggError
-
-class error(OggError): pass
-class OggTheoraHeaderError(error): pass
-
-class OggTheoraInfo(object):
- """Ogg Theora stream information.
-
- Attributes:
- length - file length in seconds, as a float
- fps - video frames per second, as a float
- """
-
- length = 0
-
- def __init__(self, fileobj):
- page = OggPage(fileobj)
- while not page.packets[0].startswith("\x80theora"):
- page = OggPage(fileobj)
- if not page.first:
- raise OggTheoraHeaderError(
- "page has ID header, but doesn't start a stream")
- data = page.packets[0]
- vmaj, vmin = struct.unpack("2B", data[7:9])
- if (vmaj, vmin) != (3, 2):
- raise OggTheoraHeaderError(
- "found Theora version %d.%d != 3.2" % (vmaj, vmin))
- fps_num, fps_den = struct.unpack(">2I", data[22:30])
- self.fps = fps_num / float(fps_den)
- self.bitrate, = struct.unpack(">I", data[37:40] + "\x00")
- self.serial = page.serial
-
- def pprint(self):
- return "Ogg Theora, %.2f seconds, %d bps" % (self.length, self.bitrate)
-
-class OggTheoraCommentDict(VCommentDict):
- """Theora comments embedded in an Ogg bitstream."""
-
- def __init__(self, fileobj, info):
- pages = []
- complete = False
- while not complete:
- page = OggPage(fileobj)
- if page.serial == info.serial:
- pages.append(page)
- complete = page.complete or (len(page.packets) > 1)
- data = OggPage.to_packets(pages)[0][7:]
- super(OggTheoraCommentDict, self).__init__(data + "\x01")
-
- def _inject(self, fileobj):
- """Write tag data into the Theora comment packet/page."""
-
- fileobj.seek(0)
- page = OggPage(fileobj)
- while not page.packets[0].startswith("\x81theora"):
- page = OggPage(fileobj)
-
- old_pages = [page]
- while not (old_pages[-1].complete or len(old_pages[-1].packets) > 1):
- page = OggPage(fileobj)
- if page.serial == old_pages[0].serial:
- old_pages.append(page)
-
- packets = OggPage.to_packets(old_pages, strict=False)
-
- packets[0] = "\x81theora" + self.write(framing=False)
-
- new_pages = OggPage.from_packets(packets, old_pages[0].sequence)
- OggPage.replace(fileobj, old_pages, new_pages)
-
-class OggTheora(OggFileType):
- """An Ogg Theora file."""
-
- _Info = OggTheoraInfo
- _Tags = OggTheoraCommentDict
- _Error = OggTheoraHeaderError
- _mimes = ["video/x-theora"]
-
- def score(filename, fileobj, header):
- return (header.startswith("OggS") *
- (("\x80theora" in header) + ("\x81theora" in header)))
- score = staticmethod(score)
-
-Open = OggTheora
-
-def delete(filename):
- """Remove tags from a file."""
- OggTheora(filename).delete()