Initial commit

Author: jbaiter

README.rst

@@ -0,0 +1,49 @@
+pyomxplayer
+===========
+Python wrapper module around `OMXPlayer <https://github.com/huceke/omxplayer>`_
+for the Raspberry Pi.
+
+Unlike `other implementations <https://github.com/KenT2/pyomxplayer>`_, this
+module does not rely on any external scripts and FIFOs, but uses the
+`pexpect module <http://pypi.python.org/pypi/pexpect/2.4>`_ for communication
+with the OMXPlayer process.
+
+CPU overhead is rather low (~3% for the Python process on my development RPi)
+and the object-oriented design makes it easy to re-use in other projects.
+
+Installation:
+-------------
+::
+    git clone https://github.com/jbaiter/pyomxplayer.git
+    python pyomxplayer/setup.py install
+
+Example:
+--------
+::
+    >>> from pyomxplayer import OMXPlayer
+    >>> from pprint import pprint
+    >>> omx = OMXPlayer('/tmp/video.mp4')
+    >>> pprint(omx.__dict__)
+    {'_position_thread': <Thread(Thread-5, started 1089234032)>,
+    '_process': <pexpect.spawn object at 0x1a435d0>,
+    'audio': {'bps': 16,
+            'channels': 2,
+            'decoder': 'mp3',
+            'rate': 48000,
+            'streams': 1},
+    'chapters': 0,
+    'current_audio_stream': 1,
+    'current_volume': 0.0,
+    'paused': True,
+    'position': 0.0,
+    'subtitles': 0,
+    'subtitles_visible': False,
+    'video': {'decoder': 'omx-mpeg4',
+            'dimensions': (640, 272),
+            'fps': 23.976025,
+            'profile': 15,
+            'streams': 1}}
+    >>> omx.toggle_pause()
+    >>> omx.position
+    9.43
+    >>> omx.stop()

pyomxplayer.py

@@ -0,0 +1,98 @@
+import pexpect
+import re
+
+from threading import Thread
+from time import sleep
+
+class OMXPlayer(object):
+
+    _FILEPROP_REXP = re.compile(r".*audio streams (\d+) video streams (\d+) chapters (\d+) subtitles (\d+).*")
+    _VIDEOPROP_REXP = re.compile(r".*Video codec ([\w-]+) width (\d+) height (\d+) profile (\d+) fps ([\d.]+).*")
+    _AUDIOPROP_REXP = re.compile(r"Audio codec (\w+) channels (\d+) samplerate (\d+) bitspersample (\d+).*")
+    _STATUS_REXP = re.compile(r"V :\s*([\d.]+).*")
+    _DONE_REXP = re.compile(r"have a nice day.*")
+
+    _LAUNCH_CMD = '/usr/bin/omxplayer -s %s %s'
+    _PAUSE_CMD = 'p'
+    _TOGGLE_SUB_CMD = 's'
+    _QUIT_CMD = 'q'
+
+    paused = False
+    subtitles_visible = True
+
+    def __init__(self, mediafile, args=None, start_playback=False):
+        if not args:
+            args = ""
+        cmd = self._LAUNCH_CMD % (mediafile, args)
+        self._process = pexpect.spawn(cmd)
+        
+        self.video = dict()
+        self.audio = dict()
+        # Get file properties
+        file_props = self._FILEPROP_REXP.match(self._process.readline()).groups()
+        (self.audio['streams'], self.video['streams'],
+         self.chapters, self.subtitles) = [int(x) for x in file_props]
+        # Get video properties
+        video_props = self._VIDEOPROP_REXP.match(self._process.readline()).groups()
+        self.video['decoder'] = video_props[0]
+        self.video['dimensions'] = tuple(int(x) for x in video_props[1:3])
+        self.video['profile'] = int(video_props[3])
+        self.video['fps'] = float(video_props[4])
+        # Get audio properties
+        audio_props = self._AUDIOPROP_REXP.match(self._process.readline()).groups()
+        self.audio['decoder'] = audio_props[0]
+        (self.audio['channels'], self.audio['rate'],
+         self.audio['bps']) = [int(x) for x in audio_props[1:]]
+
+        if self.audio['streams'] > 0:
+            self.current_audio_stream = 1
+            self.current_volume = 0.0
+
+        self._position_thread = Thread(target=self._get_position)
+        self._position_thread.start()
+
+        if not start_playback:
+            self.toggle_pause()
+        self.toggle_subtitles()
+
+
+    def _get_position(self):
+        while True:
+            index = self._process.expect([self._STATUS_REXP,
+                                            pexpect.TIMEOUT,
+                                            pexpect.EOF,
+                                            self._DONE_REXP])
+            if index == 1: continue
+            elif index in (2, 3): break
+            else:
+                self.position = float(self._process.match.group(1))
+            sleep(0.05)
+
+    def toggle_pause(self):
+        if self._process.send(self._PAUSE_CMD):
+            self.paused = not self.paused
+
+    def toggle_subtitles(self):
+        if self._process.send(self._TOGGLE_SUB_CMD):
+            self.subtitles_visible = not self.subtitles_visible
+    def stop(self):
+        self._process.send(self._QUIT_CMD)
+        self._process.terminate(force=True)
+
+    def set_speed(self):
+        raise NotImplementedError
+
+    def set_audiochannel(self, channel_idx):
+        raise NotImplementedError
+
+    def set_subtitles(self, sub_idx):
+        raise NotImplementedError
+
+    def set_chapter(self, chapter_idx):
+        raise NotImplementedError
+
+    def set_volume(self, volume):
+        raise NotImplementedError
+
+    def seek(self, minutes):
+        raise NotImplementedError

setup.py

@@ -0,0 +1,7 @@
+import distutils.core
+
+distutils.core.setup(
+    name="pyomxplayer",
+    packages = ["."],
+    requires = ['pexpect (>= 2.4)'],
+    )