Implement rendering capability for event and segment analyzers + add some more analyz...

diff --git a/timeside/analyzer/core.py b/timeside/analyzer/core.py
index 87a905cd62cc7600a5c99f3ccd260fc027f346aa..32e3e2fbddcc25cb44bf269ed905f952704a100e 100644 (file)
--- a/timeside/analyzer/core.py
+++ b/timeside/analyzer/core.py
@@ -767,6 +767,9 @@ class EventObject(object):
     def duration(self):
         return numpy.zeros(len(self))
 
+    def _render_plot(self, ax):
+        ax.stem(self.time, self.data)
+
 
 class SegmentObject(EventObject):
     _time_mode = 'segment'
@@ -799,25 +802,29 @@ class FrameLabelResult(LabelObject, FramewiseObject, AnalyzerResult):
 
 
 class EventValueResult(ValueObject, EventObject, AnalyzerResult):
-    def _render_plot(self, ax):
-        for time, value in (self.time, self.data):
-            ax.axvline(time, ymin=0, ymax=value, color='r')
-            # TODO : check value shape !!!
-
+    pass
 
 class EventLabelResult(LabelObject, EventObject, AnalyzerResult):
-    def _render_plot(self, ax):
-        pass
-
+    pass
 
 class SegmentValueResult(ValueObject, SegmentObject, AnalyzerResult):
     def _render_plot(self, ax):
-        pass
+        import itertools
+        colors = itertools.cycle(['b', 'g', 'r', 'c', 'm', 'y', 'k'])
+        for time, value in (self.time, self.data):
+            ax.axvline(time, ymin=0, ymax=value, color='r')
+            # TODO : check value shape !!!
 
 
 class SegmentLabelResult(LabelObject, SegmentObject, AnalyzerResult):
     def _render_plot(self, ax):
-        pass
+        import itertools
+        colors = itertools.cycle(['b', 'g', 'r', 'c', 'm', 'y', 'k'])
+        ax_color = {}
+        for key in self.label_metadata.label.keys():
+            ax_color[key] = colors.next()
+        for time, duration, label in zip(self.time, self.duration, self.data):
+            ax.axvspan(time, time+duration, color=ax_color[label], alpha=0.3)
 
 
 class AnalyzerResultContainer(dict):
@@ -832,7 +839,6 @@ class AnalyzerResultContainer(dict):
     >>> a.new_result() #doctest: +ELLIPSIS
     FrameValueResult(id_metadata=IdMetadata(id='analyzer', name='Generic analyzer', unit='', description='', date='...', version='...', author='TimeSide', uuid='...'), data_object=DataObject(value=array([], dtype=float64)), audio_metadata=AudioMetadata(uri='...', start=0.0, duration=8.0..., is_segment=False, channels=None, channelsManagement=''), frame_metadata=FrameMetadata(samplerate=44100, blocksize=8192, stepsize=8192), parameters={})
     >>> resContainer = timeside.analyzer.core.AnalyzerResultContainer()
-
     '''
 
 

diff --git a/timeside/grapher/render_analyzers.py b/timeside/grapher/render_analyzers.py
index a43bee1a55e6c0dde32e72d78b98377b7de08b63..063a0a4b0215109643168675de3fdd7aae25273c 100644 (file)
--- a/timeside/grapher/render_analyzers.py
+++ b/timeside/grapher/render_analyzers.py
@@ -94,6 +94,8 @@ class DisplayAnalyzer(Grapher):
             def name():
                 return grapher_name
 
+            __doc__ = """Builds a PIL image representing """ + grapher_name
+
         NewGrapher.__name__ = 'Display'+result_id
 
         return NewGrapher
@@ -119,3 +121,8 @@ DisplayWaveform = DisplayAnalyzer.create(analyzer=wav,
                                                        result_id='waveform_analyzer',
                                                        grapher_id='grapher_waveform',
                                                        grapher_name='Waveform from Analyzer')
+irit4hz = analyzer.IRITSpeech4Hz()
+Display4hzSpeechSegmentation = DisplayAnalyzer.create(analyzer=irit4hz,
+                                                      result_id='irit_speech_4hz.segments',
+                                                      grapher_id='grapher_irit_speech_4hz_segments',
+                                                      grapher_name='Irit 4Hz Speech Segmentation')