From: yomguy <yomguy@parisson.com>
Date: Fri, 18 Jun 2010 22:10:49 +0000 (+0000)
Subject: fix grapher core defaults
X-Git-Tag: 0.3.2~136
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=70c3b982538be7c798e7b629668f372fe9638838;p=timeside.git

fix grapher core defaults
---

diff --git a/INSTALL b/INSTALL
index 9b57714..c622564 100644
--- a/INSTALL
+++ b/INSTALL
@@ -106,6 +106,7 @@ For example, a normalization and a waveform, from a python shell::
 >>> grapher  =  timeside.grapher.Waveform('image.png')
 >>> encoder  =  timeside.encoder.WavEncoder('normalized.wav')
 >>> (decoder | grapher | encoder).run()
+>>> grapher.render()
 
 
 See the website for more examples and information about the TimeSide API:
diff --git a/timeside/grapher/core.py b/timeside/grapher/core.py
index 21b8fe1..4eb518b 100644
--- a/timeside/grapher/core.py
+++ b/timeside/grapher/core.py
@@ -131,7 +131,7 @@ class WaveformImage(object):
     Adds pixels iteratively thanks to the adapter providing fixed size frame buffers.
     Peaks are colored relative to the spectral centroids of each frame packet. """
 
-    def __init__(self, image_width, image_height, nframes, samplerate, fft_size, bg_color=(0,0,0), color_scheme=None, filename=None):
+    def __init__(self, image_width, image_height, nframes, samplerate, fft_size, bg_color, color_scheme, filename=None):
         self.image_width = image_width
         self.image_height = image_height
         self.nframes = nframes
@@ -143,10 +143,8 @@ class WaveformImage(object):
 
         if isinstance(color_scheme, dict):
             colors = color_scheme['waveform']
-        elif color_scheme:
-            colors = default_color_schemes[self.color_scheme]['waveform']
         else:
-            colors = default_color_schemes['default']['waveform']
+            colors = default_color_schemes[color_scheme]['waveform']
 
         self.color_lookup = interpolate_colors(colors)
 
diff --git a/timeside/grapher/waveform.py b/timeside/grapher/waveform.py
index fb7016e..0f78054 100644
--- a/timeside/grapher/waveform.py
+++ b/timeside/grapher/waveform.py
@@ -21,7 +21,7 @@
 
 from timeside.core import Processor, implements, interfacedoc, FixedSizeInputAdapter
 from timeside.api import IGrapher
-from timeside.grapher import *
+from timeside.grapher.core import *
 
 
 class Waveform(Processor):
@@ -30,15 +30,9 @@ class Waveform(Processor):
     FFT_SIZE = 0x400
 
     @interfacedoc
-    def __init__(self, width=None, height=None, output=None, bg_color=None, color_scheme=None):
-        if width:
-            self.width = width
-        else:
-            self.width = 1024
-        if height:
-            self.height = height
-        else:
-            self.height = 256
+    def __init__(self, width=1024, height=256, output=None, bg_color=(0,0,0), color_scheme='default'):
+        self.width = width
+        self.height = height
         self.bg_color = bg_color
         self.color_scheme = color_scheme
         self.filename = output