TestSuite.py

Go to the documentation of this file.

import wx
from proc import *
from plotter import *
from libCatKernel import *
from libCatKernel import *

#import matplotlib
#matplotlib.use('wxagg')

class TestSuite(proc):
    def __init__(self, cat, obj, panel, path):
        proc.__init__(self,cat,obj,panel,path)
        self.loadxrc()
        self.s1=self.getControl("s1")
        self.s2=self.getControl("s2")
        self.s3=self.getControl("s3")
        self.s4=self.getControl("s4")
        self.update()
        self.panel.Bind(wx.EVT_BUTTON, self.onApply, id=xrc.XRCID("Apply"))

    def onApply(self, event):
        v1 = float(self.s1.GetValue())
        v2 = float(self.s2.GetValue())
        v3 = float(self.s3.GetValue())
        v4 = float(self.s4.GetValue())
        self.obj.setSigma(v1,v2,v3,v4)

    def update(self):
        v1=self.obj.sigma(0)
        v2=self.obj.sigma(1)
        v3=self.obj.sigma(2)
        v4=self.obj.sigma(3)

        self.s1.SetValue(str(v1))
        self.s2.SetValue(str(v2))
        self.s3.SetValue(str(v3))
        self.s4.SetValue(str(v4))        

#----------------------------------------------------------------------

def Edit (cat, obj, panel, path):
    return TestSuite(cat, obj, panel, path)

#----------------------------------------------------------------------

def createPlots(cat, obj, run):
    fig = plt.figure(figsize=(10,10))
    fig.suptitle("Run "+str(run), fontsize=14, fontweight='bold')
    for vplot in [1,2,3]:
        ax = fig.add_subplot(3,1,vplot)
    fig.show()    
    return [fig]

def updatePlots(figs, obj):
    fig = figs[0]
    ax1 = fig.add_subplot(3,1,1)
    lines, = ax1.plot([],[], 'bo')
    #Autoscale on unknown axis and known lims on the other
    ax1.set_autoscaley_on(True)
#    ax.set_xlim(self.min_x, self.max_x)
    #Other stuff
    ax1.grid()        
    lines.set_xdata(obj.data(0))
    lines.set_ydata(obj.data(1))
    #Need both of these in order to rescale
    ax1.relim()
    ax1.autoscale_view()

    # plot histo root
    ax2=fig.add_subplot(3,1,2)
    h1 = ax2.hist(obj.data(0))
    ax2.relim()
    ax2.autoscale_view()

    #We need to draw *and* flush
    fig.canvas.draw()
    fig.canvas.flush_events()
    fig.show()
    return fig
    
#----------------------------------------------------------------------

def Plot(cat, obj, parent, id):
    from mpl_toolkits.mplot3d import Axes3D
    from matplotlib import cm
    import matplotlib as mpl
    import matplotlib.pyplot as plt
    import numpy as np

    plot=plotter(parent,id)
    fig=plot.current()
    fig.suptitle('TestSuite - Plot capabilities', fontsize=14, fontweight='bold')
    
    # plot of data set 0
    p1=fig.add_subplot(2,3,1)
    p1.plot(obj.data(0), 'bo', obj.data().vector(1),'k')
    p1.set_title(r'$\sigma_0='+str(obj.sigma(0))+'$')
    
    # plot of data set 1
    p2=fig.add_subplot(2,3,2)
    p2.plot(obj.data(1), 'r--')
    p2.set_title(r'$\sigma_1='+str(obj.sigma(1))+'$')
    
    # plot of data set 0 (histogram)
    plt.subplot(2,3,3)
    n, bins, h=plt.hist(obj.data().vector(0), 10)
    plt.title(obj.data().title(0))    
    plt.grid(True)

    # plot of data set 1 (histogram)    
    plt.subplot(2,3,4)
    plt.hist(obj.data().vector(1), 20)
    plt.title(obj.data().name(1)+" : "+obj.data().title(1))
    plt.grid(True)

    # plot histo root
    p5=fig.add_subplot(2,3,5)
    h1=h1d(p5, obj.hist1d(0), facecolor='red',linestyle='dashed')
    p5.grid(True)



    # plot histo root
    p7=fig.add_subplot(2,3,6)
    h7=obj.hist1d(0)
    print h7.xbins()
    p7.plot(h7.xbins(),h7.bins(),'ro',color='red')
    p7.grid(False)

    # plot histo root
#    p8=fig.add_subplot(4,3,8)
#    f8=Axes3D(fig)
#    h8=obj.hist2d(0)
#    x,y=np.meshgrid(h8.xbins(),h8.ybins())
#    z=np.array(h8.bins())

#    print x
#    print y
#    print z

#    p8.contour(x,y,z)
#    f8.plot_surface(x,y,z)
#    p8.plot



    # return the plot that must be plotted
    return plot