Acquisition.py

Go to the documentation of this file.

import wx
from proc import *
from plotter import *
from libCatKernel import *
from matplotlib.path import Path
import matplotlib.patches as patches

class Acquisition(proc):
    def __init__(self, cat, obj, panel, path):
        proc.__init__(self,cat,obj,panel,path)
        self.loadxrc()
        self.panel.Bind(wx.EVT_TOGGLEBUTTON,self.onSoftTrig,id=xrc.XRCID("softtrigger"))
        self.panel.Bind(wx.EVT_TOGGLEBUTTON,self.onTree,id=xrc.XRCID("tree"))
        self.panel.Bind(wx.EVT_TOGGLEBUTTON,self.onTrend,id=xrc.XRCID("trend"))
        self.panel.Bind(wx.EVT_TOGGLEBUTTON,self.onSample,id=xrc.XRCID("sample"))
        self.panel.Bind(wx.EVT_SPINCTRL,self.onDeadtime,id=xrc.XRCID("acquisition_deadtime"))
        self.panel.Bind(wx.EVT_BUTTON,self.onApply,id=xrc.XRCID("apply"))
        self.panel.Bind(wx.EVT_SPINCTRL, self.onDepth, id=xrc.XRCID("depth"))
        self.panel.Bind(wx.EVT_SPINCTRL, self.onNSample, id=xrc.XRCID("nsample"))
        self.panel.Bind(wx.EVT_SPINCTRL, self.onNBins, id=xrc.XRCID("nbins"))
        self.panel.Bind(wx.EVT_TEXT, self.onMin, id=xrc.XRCID("min"))
        self.panel.Bind(wx.EVT_TEXT, self.onMax, id=xrc.XRCID("max"))

        self.update()

    def onApply(self,event):
        channels=0
        if (self.getControl("chl0").GetValue()) : channels=channels|0b1
        else : channels=channels&0b11111110
        if (self.getControl("chl1").GetValue()) : channels=channels|0b10
        else : channels=channels&0b11111101
        if (self.getControl("chl2").GetValue()) : channels=channels|0b100
        else : channels=channels&0b11111011
        if (self.getControl("chl3").GetValue()) : channels=channels|0b1000
        else : channels=channels&0b11110111
        if (self.getControl("chl4").GetValue()) : channels=channels|0b10000
        else : channels=channels&0b11101111
        if (self.getControl("chl5").GetValue()) : channels=channels|0b100000
        else : channels=channels&0b11011111
        if (self.getControl("chl6").GetValue()) : channels=channels|0b1000000
        else : channels=channels&0b10111111
        if (self.getControl("chl7").GetValue()) : channels=channels|0b10000000
        else : channels=channels&0b1111111
        self.obj.setChannels(channels)            
        
    def onSoftTrig(self,event):
        self.obj.setTrigger(self.getControl("softtrigger").GetValue())
        if self.getControl("softtrigger").GetValue():
            self.getControl("softtrigger").SetValue(1)
            self.getControl("softtrigger").SetLabel("Soft. Trig.")
        else:
            self.getControl("softtrigger").SetValue(0)
            self.getControl("softtrigger").SetLabel("Ext. Trig.")

    def onTree(self,event):
        self.obj.setStoreTree(self.getControl("tree").GetValue())

    def onTrend(self,event):
        self.obj.setStoreTrend(self.getControl("trend").GetValue())

    def onSample(self,event):
        self.obj.setStoreSample(self.getControl("sample").GetValue())

    def onDeadtime(self,event):
        self.obj.setDeadTime(self.getControl("acquisition_deadtime").GetValue())
    def onDepth(self,event):
        self.obj.setDepth(self.getControl("depth").GetValue()) 

    def onNSample(self,event):
        self.obj.setSample(self.getControl("nsample").GetValue()) 

    def onNBins(self,event):
        self.obj.setNBins(self.getControl("nbins").GetValue())  

    def onMin(self,event):
        self.obj.setMinRange(float(self.getControl("min").GetValue()))  

    def onMax(self,event):
        self.obj.setMaxRange(float(self.getControl("max").GetValue()))  

    def update(self):
        if self.obj.trigger():
            self.getControl("softtrigger").SetValue(1)
            self.getControl("softtrigger").SetLabel("Soft. Trig.")
        else:
            self.getControl("softtrigger").SetValue(0)
            self.getControl("softtrigger").SetLabel("Ext. Trig.")

        channels=self.obj.channels()
        if (channels&1) : self.getControl("chl0").SetValue(True)
        else : self.getControl("chl0").SetValue(False)
        if (channels&0b10) : self.getControl("chl1").SetValue(True)
        else : self.getControl("chl1").SetValue(False)
        if (channels&0b100) : self.getControl("chl2").SetValue(True)
        else : self.getControl("chl2").SetValue(False)
        if (channels&0b1000) : self.getControl("chl3").SetValue(True)
        else : self.getControl("chl3").SetValue(False)
        if (channels&0b10000) : self.getControl("chl4").SetValue(True)
        else : self.getControl("chl4").SetValue(False)
        if (channels&0b100000) : self.getControl("chl5").SetValue(True)
        else : self.getControl("chl5").SetValue(False)
        if (channels&0b1000000) : self.getControl("chl6").SetValue(True)
        else : self.getControl("chl6").SetValue(False)
        if (channels&0b10000000) : self.getControl("chl7").SetValue(True)
        else : self.getControl("chl7").SetValue(False)
        self.getControl("depth").SetValue(self.obj.depth())
        self.getControl("tree").SetValue(self.obj.storeTree())
        self.getControl("trend").SetValue(self.obj.storeTrend())
        self.getControl("sample").SetValue(self.obj.storeSample())
        self.getControl("acquisition_deadtime").SetValue(self.obj.deadTime())
        self.getControl("depth").SetValue(self.obj.depth())
        self.getControl("nsample").SetValue(self.obj.sample())
        self.getControl("nbins").SetValue(self.obj.nBins())
        self.getControl("min").SetValue(str(self.obj.minRange()))
        self.getControl("max").SetValue(str(self.obj.maxRange()))
         
#----------------------------------------------------------------------

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

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

def Plot(cat, obj, parent, id):
    
    if (not obj.storeTrend() and not obj.storeSample):
        return None
    
    from mpl_toolkits.mplot3d import Axes3D
    import matplotlib.ticker as ticker
    from matplotlib import cm
    import matplotlib as mpl
    import matplotlib.pyplot as plt
    import numpy as np

    plot=plotter(parent,id)
    fig=plot.current()

    title = 'Acquisition - Channel(s) '

    channels=obj.channels()
    nch=0
    for i in range(0,8): 
        if (channels>>i&1): 
            nch+=1
            title+=' '+str(i)

    fig.suptitle(title, fontsize=14, fontweight='bold')
    
    col=nch
    row=0
    if obj.storeSample() : row+=1
    if obj.storeTrend() : row+=2

    index=0

    if obj.storeTrend() : 
        for c in range(0,col):
            pl=fig.add_subplot(row,col,c+1)     
#            h=h1d(pl,obj.hist1d(index), facecolor='blue',linestyle='dashed')
            pl.plot(obj.hist1d(index).xbins(),obj.hist1d(index).bins(),'ro',color='blue')
            pl.xaxis.set_ticks([0,obj.depth()/2,obj.depth()-1])
            pl.grid(True)
            index+=1
            pl=fig.add_subplot(row,col,c+col+1)     
#            h=h1d(pl,obj.hist1d(index), facecolor='green',linestyle='dashed')
            pl.plot(obj.hist1d(index).xbins(),obj.hist1d(index).bins(),'ro',color='green')
            pl.xaxis.set_ticks([0,obj.depth()/2,obj.depth()-1])
            pl.grid(True)
            index+=1
            if (row==3): index+=1

    if obj.storeSample() : 
        if (row==3): index=2
        for c in range(0,col):
            pl=fig.add_subplot(row,col,c+(row-1)*col+1)     
            h=h1d(pl,obj.hist1d(index), facecolor='red',linestyle='dashed')
            pl.xaxis.set_ticks([obj.hist1d(index).minX(),(obj.hist1d(index).maxX()+obj.hist1d(index).minX())/2.,obj.hist1d(index).maxX()])
            pl.grid(True)
            index+=1
            if (row==3): index+=2

    return plot