plotter.py

Go to the documentation of this file.

import wx
import wx.aui
#import matplotlib as mpl
#import matplotlib.pyplot as plt


#from matplotlib.path import Path
#import matplotlib.patches as patches

#import matplotlib as mpl
#mpl.use('WXAgg')

#import matplotlib.backends.backend_wxagg as mwx

#from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as Canvas
#from matplotlib.backends.backend_wxagg import NavigationToolbar2Wx as Toolbar

import matplotlib.patches as patches
from   matplotlib.path import Path
import matplotlib.pyplot as plt
import numpy as np 


class plotter(wx.Panel):
        def __init__(self, parent,figid):
                if (parent==None):
                        self.figure = plt.figure(figid)
                else:
                        wx.Panel.__init__(self, parent, id=-1)
                        self.figid  = figid
                        self.figure = plt.figure(figid)
                        self.figure.subplots_adjust(top=0.9,bottom=0.05,right=0.95,left=0.05)
                        self.canvas = Canvas(self, -1, self.figure)
                        self.toolbar = Toolbar(self.canvas)
                        self.toolbar.Realize()
                        sizer = wx.BoxSizer(wx.VERTICAL)
                        sizer.Add(self.canvas,1,wx.EXPAND)
                        sizer.Add(self.toolbar, 0 , wx.LEFT | wx.EXPAND)
                        self.SetSizer(sizer)
                        
    def current(self):
            return self.figure

    def destroy(self):
        self.canvas.Destroy()

def h1d(fig, histo ,
        alpha=0.5,
        facecolor='green',
        edgecolor='yellow',
        linestyle='dashed'):
        import numpy as np
        left=np.array(histo.edges()[:-1])
        right=np.array(histo.edges()[1:])
        bottom = np.zeros(len(left))
        top=bottom+histo.bins()
        nrects=len(left)
        nverts = nrects*(1+3+1)
        verts = np.zeros((nverts, 2))
        codes = np.ones(nverts, int) * Path.LINETO
        codes[0::5] = Path.MOVETO
        codes[4::5] = Path.CLOSEPOLY
        verts[0::5,0] = left
        verts[0::5,1] = bottom
        verts[1::5,0] = left
        verts[1::5,1] = top
        verts[2::5,0] = right
        verts[2::5,1] = top
        verts[3::5,0] = right
        verts[3::5,1] = bottom
        barpath = Path(verts, codes)
        patch = patches.PathPatch(barpath,
                                  facecolor=facecolor,
                                  edgecolor=edgecolor,
                                  alpha=alpha,
                                  linestyle=linestyle)
        max=histo.maxY()
        fig.add_patch(patch)
        fig.set_xlim(histo.minX(), histo.maxX())
        if (max>0):
                fig.set_ylim(0.,max*1.05)
        else:
                fig.set_ylim(max*1.05,0.)