From: yomguy <yomguy@parisson.com>
Date: Fri, 18 Jun 2010 22:32:12 +0000 (+0000)
Subject: fix spec default
X-Git-Tag: 0.3.2~133
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=a43d514fd124c8561aeac6ee220899845d7ac507;p=timeside.git

fix spec default
---

diff --git a/timeside/grapher/core.py b/timeside/grapher/core.py
index 4eb518b..a06c6a1 100644
--- a/timeside/grapher/core.py
+++ b/timeside/grapher/core.py
@@ -266,8 +266,12 @@ class SpectrogramImage(object):
         self.filename = filename
         self.color_scheme = color_scheme
 
+        if isinstance(color_scheme, dict):
+            colors = color_scheme['spectrogram']
+        else:
+            colors = default_color_schemes[color_scheme]['spectrogram']
+
         self.image = Image.new("P", (self.image_height, self.image_width))
-        colors = color_schemes[self.color_scheme]['spectrogram']
         self.image.putpalette(interpolate_colors(colors, True))
 
         self.samples_per_pixel = self.nframes / float(self.image_width)
diff --git a/timeside/grapher/spectrogram.py b/timeside/grapher/spectrogram.py
index 0962842..de898b8 100644
--- a/timeside/grapher/spectrogram.py
+++ b/timeside/grapher/spectrogram.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 Spectrogram(Processor):
@@ -30,15 +30,9 @@ class Spectrogram(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 = 1500
-        if height:
-            self.height = height
-        else:
-            self.height = 200
+    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