CurrentMeasurement.py

Go to the documentation of this file.

import wx
from proc import *
from plotter import *
from libCatKernel import *
import matplotlib as mpl
import matplotlib.patches as patches
from   matplotlib.path import Path
import matplotlib.pyplot as plt
import numpy as np

class CurrentMeasurement(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 CurrentMeasurement(cat, obj, panel, path)

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

def createPlot(cat, obj, run):
    fig = plt.figure(figsize=(10,10))
    fig.suptitle("Run "+str(run), fontsize=14, fontweight='bold')
    ndevices = obj.element().numberOfDevices()
    c=['r','b','g','c']
    for i in range(ndevices):
        for ch in range(4):
            ax = fig.add_subplot(4,ndevices,ndevices*ch+1+i)
            lines, = ax.plot([],[], color = c[ch], marker='^')
            if ch==0: plt.title("Dev "+str(i+1))
            if i==0: plt.ylabel("voltage (V)")
            if ch==3: plt.xlabel("time (s)")
            ax.set_autoscaley_on(False)
            ax.set_ylim(0.,0.25)
            ax.grid()        
    fig.show()    
    return fig

def updatePlot(fig, obj):
    ndevices = obj.element().numberOfDevices()
    c=['r','b','g','c']
    for i in range(ndevices):
        for ch in range(4):
            ax = fig.add_subplot(4,ndevices,ndevices*ch+1+i)
            lines, = ax.plot([],[], color = c[ch], marker='^')
            lines.set_xdata(obj.data(0+5*i))
            lines.set_ydata(obj.data(ch+1+5*i))
            ax.relim()
            ax.autoscale_view()
            ax.set_ylim(bottom=0.)

    fig.canvas.draw()
    fig.canvas.flush_events()

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