diff service/audioInputLevels/audioInputLevelsPulse.py @ 887:64a2ba088665

audio input levels to graphite Ignore-this: b1422609f3bf51778b8b2284ca914916 darcs-hash:20130802145545-312f9-64fa9ffcb82dc26e67591f28b16633add6299cd0
author drewp <drewp@bigasterisk.com>
date Fri, 02 Aug 2013 07:55:45 -0700
parents
children f84129fbbe38
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/service/audioInputLevels/audioInputLevelsPulse.py	Fri Aug 02 07:55:45 2013 -0700
@@ -0,0 +1,114 @@
+# based on http://freshfoo.com/blog/pulseaudio_monitoring
+from __future__ import division
+import socket, argparse
+from Queue import Queue
+from ctypes import POINTER, c_ubyte, c_void_p, c_ulong, cast
+import galena
+
+# From https://github.com/Valodim/python-pulseaudio
+from pulseaudio import lib_pulseaudio as P
+
+METER_RATE = 1
+MAX_SAMPLE_VALUE = 127
+
+class PeakMonitor(object):
+
+    def __init__(self, source_name, rate):
+        self.source_name = source_name
+        self.rate = rate
+
+        # Wrap callback methods in appropriate ctypefunc instances so
+        # that the Pulseaudio C API can call them
+        self._context_notify_cb = P.pa_context_notify_cb_t(self.context_notify_cb)
+        self._source_info_cb = P.pa_source_info_cb_t(self.source_info_cb)
+        self._stream_read_cb = P.pa_stream_request_cb_t(self.stream_read_cb)
+
+        # stream_read_cb() puts peak samples into this Queue instance
+        self._samples = Queue()
+
+        # Create the mainloop thread and set our context_notify_cb
+        # method to be called when there's updates relating to the
+        # connection to Pulseaudio
+        _mainloop = P.pa_threaded_mainloop_new()
+        _mainloop_api = P.pa_threaded_mainloop_get_api(_mainloop)
+        context = P.pa_context_new(_mainloop_api, 'peak_demo')
+        P.pa_context_set_state_callback(context, self._context_notify_cb, None)
+        P.pa_context_connect(context, None, 0, None)
+        P.pa_threaded_mainloop_start(_mainloop)
+
+    def __iter__(self):
+        while True:
+            yield self._samples.get()
+
+    def context_notify_cb(self, context, _):
+        state = P.pa_context_get_state(context)
+
+        if state == P.PA_CONTEXT_READY:
+            print "Pulseaudio connection ready..."
+            # Connected to Pulseaudio. Now request that source_info_cb
+            # be called with information about the available sources.
+            o = P.pa_context_get_source_info_list(context, self._source_info_cb, None)
+            P.pa_operation_unref(o)
+
+        elif state == P.PA_CONTEXT_FAILED :
+            print "Connection failed"
+
+        elif state == P.PA_CONTEXT_TERMINATED:
+            print "Connection terminated"
+
+    def source_info_cb(self, context, source_info_p, _, __):
+        if not source_info_p:
+            return
+
+        source_info = source_info_p.contents
+
+        if source_info.name == self.source_name:
+            # Found the source we want to monitor for peak levels.
+            # Tell PA to call stream_read_cb with peak samples.
+            print 'setting up peak recording using', source_info.name
+            print 'description:', source_info.description
+            samplespec = P.pa_sample_spec()
+            samplespec.channels = 1
+            samplespec.format = P.PA_SAMPLE_U8
+            samplespec.rate = self.rate
+
+            pa_stream = P.pa_stream_new(context, "peak detect demo", samplespec, None)
+            P.pa_stream_set_read_callback(pa_stream,
+                                          self._stream_read_cb,
+                                          source_info.index)
+            P.pa_stream_connect_record(pa_stream,
+                                       source_info.name,
+                                       None,
+                                       P.PA_STREAM_PEAK_DETECT)
+
+    def stream_read_cb(self, stream, length, index_incr):
+        data = c_void_p()
+        P.pa_stream_peek(stream, data, c_ulong(length))
+        data = cast(data, POINTER(c_ubyte))
+        for i in xrange(length):
+            # When PA_SAMPLE_U8 is used, samples values range from 128
+            # to 255 because the underlying audio data is signed but
+            # it doesn't make sense to return signed peaks.
+            self._samples.put(data[i] - 128)
+        P.pa_stream_drop(stream)
+
+def main():
+
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        '--source', required=True,
+        help='pulseaudio source name (use `pactl list sources | grep Name`)')
+
+    args = parser.parse_args()
+
+    
+    out = galena.Galena(host='bang')
+    prefix = 'system.house.audio.%s' % socket.gethostname()
+    monitor = PeakMonitor(args.source, METER_RATE)
+    for sample in monitor:
+        #print ' %3d %s' % (sample, '>' * sample)
+        out.send(prefix + ".max", sample / 128)
+        
+if __name__ == '__main__':
+    main()