Changeset 32548

Timestamp:

04/18/09 01:27:01 (4 years ago)

Author:

rezoo

Message:

verificator.py add plotIndications()

Files:

• lang/python/tremolo/tremolo/verificator.py (modified) (1 diff)

lang/python/tremolo/tremolo/verificator.py

@@ -146 +146 @@
                         p_list.append(float(n) / float(N))
                 return p_list
+        
+        def plotIndications(self, keys, start, end, main=-1,
+                            margin=(0.1, 0.1, 0.1, 0.1)):
+                """
+                指定された指標群のプロットを行います.
+                
+                keys   : プロットする指標のリスト
+                start  : tの開始範囲
+                end    : tの終了範囲
+                main   : 大きく表示させたい指標のインデックス.-1ならば等間隔に設定
+                margin : 余白の設定.top,right,bottom,leftの順に設定.
+                         指定されない場合はすべてが10%の余白をとる
+                """
+                from pylab import axes
+                from matplotlib.ticker import NullFormatter
+                indication_list = [self.indications[key]["indication"] for key in keys]
+                
+                width  = 1.0 - margin[1] - margin[3]
+                height = 1.0 - margin[0] - margin[2]
+                
+                index_length = main > -1 and len(keys)+1 or len(keys)
+                axes_h = height/float(index_length)
+                
+                p = margin[2]
+                axes_list = []
+                nullfmt = NullFormatter()
+                for i in xrange(len(keys)):
+                        if i == main:
+                                a = axes([margin[3], p, width, axes_h*2.0])
+                                a.xaxis.set_major_formatter(nullfmt)
+                                axes_list.append(a)
+                                p += p + axes_h*2.0
+                        else:
+                                a = axes([margin[3], p, width, axes_h])
+                                a.xaxis.set_major_formatter(nullfmt)
+                                axes_list.append(a)
+                                p += axes_h
+                
+                for (ind, ax) in zip(indication_list, axes_list):
+                        ind.plot(start, end, axes=ax)
+