コード例 #1
0
    def __init__(self, parent):
        """
        generated sscan_gui_autogenerated file use wxFormBuilder V3.4-Beta
        """
        sscan_gui_autogenerated.MainFrame.__init__(self, parent)

        # save e0 value
        self.e0 = 0.0
        self.eNow = 0.0
        self.e0EdgeEnergy = 7112.0

        self.xEnergy = []
        self.yTrans = []

        # for each panels to display spectra, generate a graphics canvas object
        self.make_canvas('canvasTrans', self.m_plotTrans)
        self.make_canvas('canvasFluo', self.m_plotFluo)

        self.scan2 = Scan('lww:scan2')                            # scan record
        self.scan2.add_callback('FAZE', self.scanUpdateCallBack)   # monitor scan status
        self.scan2.add_callback('PAUS', self.pauseCallBack)        # state change PAUSE/RESUME button
        self.scan2.add_callback('BUSY', self.scanBusyCallBack)     # scanning or stop status

        self.count = Scaler('BL10C:scaler1')                 # scaler record
        self.count.add_callback('CNT', self.UpdateCountCallBack)   # monitor current scan point

        # make sense now energy PV for monitor current Energy, direct access to the DCM Energy
        self.eNowPV = epics.PV(self.scan2.get('P1PV')+'.RBV', callback=self.eNowCallBack)
        self.scanAbortPV = epics.PV("lww:AbortScans.PROC")  # scan Abort PV
        self.scanPositionerPV = epics.PV("mobiis:m2")        # scan positioner(P1PV) PV

        # update display from IOC values to GUI-----------------------------------------------------
        self.m_totalScanPoint.SetLabel(str(self.scan2.NPTS))  # total scan point in IOC
        self.m_currentScanPoint.SetLabel(str(self.scan2.current_point))  # current scan point in IOC

        #-------------------- set Region names ---------------------
        # Region Start names
        self.m_PreEdgeStart.myName = "Pre-Edge"
        self.m_XanesStart.myName = "XANES"
        self.m_Xafs1Start.myName = "XAFS1"
        self.m_Xafs2Start.myName = "XAFS2"
        self.m_Xafs3Start.myName = "XAFS3"
        self.m_Xafs4Start.myName = "XAFS4"
        # Region Stop names
        self.m_PreEdgeStop.myName = "Pre-Edge"
        self.m_XanesStop.myName = "XANES"
        self.m_Xafs1Stop.myName = "XAFS1"
        self.m_Xafs2Stop.myName = "XAFS2"
        self.m_Xafs3Stop.myName = "XAFS3"
        self.m_Xafs4Stop.myName = "XAFS4"
        # Region Step names
        self.m_PreEdgeStep.myName = "Pre-Edge"
        self.m_XanesStep.myName = "XANES"
        self.m_Xafs1Step.myName = "XAFS1"
        self.m_Xafs2Step.myName = "XAFS2"
        self.m_Xafs3Step.myName = "XAFS3"
        self.m_Xafs4Step.myName = "XAFS4"
        # Region Npts names
        self.m_PreEdgeNpts.myName = "Pre-Edge"
        self.m_XanesNpts.myName = "XANES"
        self.m_Xafs1Npts.myName = "XAFS1"
        self.m_Xafs2Npts.myName = "XAFS2"
        self.m_Xafs3Npts.myName = "XAFS3"
        self.m_Xafs4Npts.myName = "XAFS4"
        # Region Unit names
        self.m_PreEdgeUnits.myName = "Pre-Edge"
        self.m_XanesUnits.myName = "XANES"
        self.m_Xafs1Units.myName = "XAFS1"
        self.m_Xafs2Units.myName = "XAFS2"
        self.m_Xafs3Units.myName = "XAFS3"
        self.m_Xafs4Units.myName = "XAFS4"

        # make region widgets
        self.reg_settings = []
        # append PreEdge Region items
        self.reg_settings.append((self.m_PreEdgeStart, self.m_PreEdgeStop,
                                  self.m_PreEdgeStep,  self.m_PreEdgeNpts,
                                  self.m_PreEdgeUnits))
        # Append XANES Region items
        self.reg_settings.append((self.m_XanesStart, self.m_XanesStop,
                                  self.m_XanesStep,  self.m_XanesNpts,
                                  self.m_XanesUnits))
        # Append XAFS1 Region items
        self.reg_settings.append((self.m_Xafs1Start, self.m_Xafs1Stop,
                                  self.m_Xafs1Step,  self.m_Xafs1Npts,
                                  self.m_Xafs1Units))
        # Append XAFS2 Region items
        self.reg_settings.append((self.m_Xafs2Start, self.m_Xafs2Stop,
                                  self.m_Xafs2Step,  self.m_Xafs2Npts,
                                  self.m_Xafs2Units))
        # Append XAFS3 Region items
        self.reg_settings.append((self.m_Xafs3Start, self.m_Xafs3Stop,
                                  self.m_Xafs3Step,  self.m_Xafs3Npts,
                                  self.m_Xafs3Units))
        '''
        # Append XAFS4 Region items
        self.reg_settings.append((self.m_Xafs4Start, self.m_Xafs4Stop,
                                  self.m_Xafs4Step,  self.m_Xafs4Npts,
                                  self.m_Xafs4Units))
        '''

        self.m_Xafs4Start.Disable()
        self.m_Xafs4Stop.Disable()
        self.m_Xafs4Step.Disable()
        self.m_Xafs4Npts.Disable()
        self.m_Xafs4Time.Disable()
        self.m_Xafs4Units.Disable()
コード例 #2
0
class SscanBaseMainFrame(sscan_gui_autogenerated.MainFrame):
    def __init__(self, parent):
        """
        generated sscan_gui_autogenerated file use wxFormBuilder V3.4-Beta
        """
        sscan_gui_autogenerated.MainFrame.__init__(self, parent)

        # save e0 value
        self.e0 = 0.0
        self.eNow = 0.0
        self.e0EdgeEnergy = 7112.0

        self.xEnergy = []
        self.yTrans = []

        # for each panels to display spectra, generate a graphics canvas object
        self.make_canvas('canvasTrans', self.m_plotTrans)
        self.make_canvas('canvasFluo', self.m_plotFluo)

        self.scan2 = Scan('lww:scan2')                            # scan record
        self.scan2.add_callback('FAZE', self.scanUpdateCallBack)   # monitor scan status
        self.scan2.add_callback('PAUS', self.pauseCallBack)        # state change PAUSE/RESUME button
        self.scan2.add_callback('BUSY', self.scanBusyCallBack)     # scanning or stop status

        self.count = Scaler('BL10C:scaler1')                 # scaler record
        self.count.add_callback('CNT', self.UpdateCountCallBack)   # monitor current scan point

        # make sense now energy PV for monitor current Energy, direct access to the DCM Energy
        self.eNowPV = epics.PV(self.scan2.get('P1PV')+'.RBV', callback=self.eNowCallBack)
        self.scanAbortPV = epics.PV("lww:AbortScans.PROC")  # scan Abort PV
        self.scanPositionerPV = epics.PV("mobiis:m2")        # scan positioner(P1PV) PV

        # update display from IOC values to GUI-----------------------------------------------------
        self.m_totalScanPoint.SetLabel(str(self.scan2.NPTS))  # total scan point in IOC
        self.m_currentScanPoint.SetLabel(str(self.scan2.current_point))  # current scan point in IOC

        #-------------------- set Region names ---------------------
        # Region Start names
        self.m_PreEdgeStart.myName = "Pre-Edge"
        self.m_XanesStart.myName = "XANES"
        self.m_Xafs1Start.myName = "XAFS1"
        self.m_Xafs2Start.myName = "XAFS2"
        self.m_Xafs3Start.myName = "XAFS3"
        self.m_Xafs4Start.myName = "XAFS4"
        # Region Stop names
        self.m_PreEdgeStop.myName = "Pre-Edge"
        self.m_XanesStop.myName = "XANES"
        self.m_Xafs1Stop.myName = "XAFS1"
        self.m_Xafs2Stop.myName = "XAFS2"
        self.m_Xafs3Stop.myName = "XAFS3"
        self.m_Xafs4Stop.myName = "XAFS4"
        # Region Step names
        self.m_PreEdgeStep.myName = "Pre-Edge"
        self.m_XanesStep.myName = "XANES"
        self.m_Xafs1Step.myName = "XAFS1"
        self.m_Xafs2Step.myName = "XAFS2"
        self.m_Xafs3Step.myName = "XAFS3"
        self.m_Xafs4Step.myName = "XAFS4"
        # Region Npts names
        self.m_PreEdgeNpts.myName = "Pre-Edge"
        self.m_XanesNpts.myName = "XANES"
        self.m_Xafs1Npts.myName = "XAFS1"
        self.m_Xafs2Npts.myName = "XAFS2"
        self.m_Xafs3Npts.myName = "XAFS3"
        self.m_Xafs4Npts.myName = "XAFS4"
        # Region Unit names
        self.m_PreEdgeUnits.myName = "Pre-Edge"
        self.m_XanesUnits.myName = "XANES"
        self.m_Xafs1Units.myName = "XAFS1"
        self.m_Xafs2Units.myName = "XAFS2"
        self.m_Xafs3Units.myName = "XAFS3"
        self.m_Xafs4Units.myName = "XAFS4"

        # make region widgets
        self.reg_settings = []
        # append PreEdge Region items
        self.reg_settings.append((self.m_PreEdgeStart, self.m_PreEdgeStop,
                                  self.m_PreEdgeStep,  self.m_PreEdgeNpts,
                                  self.m_PreEdgeUnits))
        # Append XANES Region items
        self.reg_settings.append((self.m_XanesStart, self.m_XanesStop,
                                  self.m_XanesStep,  self.m_XanesNpts,
                                  self.m_XanesUnits))
        # Append XAFS1 Region items
        self.reg_settings.append((self.m_Xafs1Start, self.m_Xafs1Stop,
                                  self.m_Xafs1Step,  self.m_Xafs1Npts,
                                  self.m_Xafs1Units))
        # Append XAFS2 Region items
        self.reg_settings.append((self.m_Xafs2Start, self.m_Xafs2Stop,
                                  self.m_Xafs2Step,  self.m_Xafs2Npts,
                                  self.m_Xafs2Units))
        # Append XAFS3 Region items
        self.reg_settings.append((self.m_Xafs3Start, self.m_Xafs3Stop,
                                  self.m_Xafs3Step,  self.m_Xafs3Npts,
                                  self.m_Xafs3Units))
        '''
        # Append XAFS4 Region items
        self.reg_settings.append((self.m_Xafs4Start, self.m_Xafs4Stop,
                                  self.m_Xafs4Step,  self.m_Xafs4Npts,
                                  self.m_Xafs4Units))
        '''

        self.m_Xafs4Start.Disable()
        self.m_Xafs4Stop.Disable()
        self.m_Xafs4Step.Disable()
        self.m_Xafs4Npts.Disable()
        self.m_Xafs4Time.Disable()
        self.m_Xafs4Units.Disable()

    def StartScan(self, event):
        """
        Generate xafs step scan positions(P1PA) and START scan
        """
        self.e0EdgeEnergy = float(self.m_SetEdgeEnergy.Value)
        self.scanPosGen(self.e0EdgeEnergy, self.reg_settings)

        self.scan2._print()

        # we need DCM pre postion set to -200[eV] before scan start(e0-startE-200ev)
        prePosition = ((self.e0EdgeEnergy-abs(float(self.m_PreEdgeStart.Value))) / 1000.)-0.2
        self.scanPositionerPV.put(value=prePosition, use_complete=True)

        # set expouse time to scaler
        self.count.CountTime(float(self.m_setCountTime.Value))

        self.scan2.run(wait=True)

        self.m_totalScanPoint.SetLabel(str(self.scan2.NPTS))
        self.xEnergy = []
        self.yTrans = []
        return

    def PauseScan(self, event):
        self.scan2.PAUS = 1  # set PAUSE
        return

    def ResumeScan(self, event):
        self.scan2.PAUS = 0  # set RESUME
        return

    def AbortScan(self, event):
        self.scanAbortPV.put(1)
        return

    def OnChoiceElement(self, event):
        tempEdgeEnergy = ELEM_DICT[self.m_choiceElement.GetStringSelection()]
        self.m_SetEdgeEnergy.SetValue(str(tempEdgeEnergy))
        self.e0 = tempEdgeEnergy
        return

    def enterE0(self, event):
        self.e0 = float(self.m_SetEdgeEnergy.Value)

        for key, element in ELEM_DICT.items():
            if self.e0 == float(element):
                self.m_choiceElement.SetStringSelection(key)
        return

    #
    # Region of Interest event(Start/Stop/Step/Npts/Time/Units)
    #
    def onStart(self, event):
        rowKey = event.GetEventObject().myName
        self.onVal(self, index=ROW_FIND.get(rowKey), label='start',
                   value=unicode(event.GetEventObject().Value))
        return

    def onStop(self, event):
        rowKey = event.GetEventObject().myName
        self.onVal(self, index=ROW_FIND.get(rowKey), label='stop',
                   value=unicode(event.GetEventObject().Value))
        return

    def onStep(self, event):
        rowKey = event.GetEventObject().myName
        self.onVal(self, index=ROW_FIND.get(rowKey), label='step',
                   value=unicode(event.GetEventObject().Value))
        return

    def onNpts(self, event):
        rowKey = event.GetEventObject().myName
        self.onVal(self, index=ROW_FIND.get(rowKey), label='npts',
                   value=unicode(event.GetEventObject().Value))
        return

    def onUnits(self, event):
        rowKey = event.GetEventObject().myName

        index = ROW_FIND.get(rowKey)
        wids = self.reg_settings[index]
        self.setStepNpts(wids, label='units')
        return

    def OnSize_TransCanvas(self, event):
        size = self.m_plotTrans.GetSize()
        self.canvasTrans.SetSize(size)

    def pauseCallBack(self, **kwargs):
        if kwargs['char_value'] == 'PAUSE':
            self.pause_button.Disable()
            self.resume_button.Enable()
        else:
            self.pause_button.Enable()
            self.resume_button.Disable()

    def scanBusyCallBack(self, **kwargs):
        if kwargs['value'] == 1:
            self.pause_button.Enable()
            self.resume_button.Disable()
            self.scan_button.Disable()
        elif kwargs['value'] == 0:  # scan DONE! or Aborted!
            # Todo: we need DCM move to e0 position after scan DONE or IDLE
            # ddd = self.e0EdgeEnergy / 1000.0
            # self.scanPositionerPV.put(value=ddd, use_complete=True)
            self.pause_button.Disable()
            self.resume_button.Disable()
            self.scan_button.Enable()

    def scanUpdateCallBack(self, **kwargs):
        """
        callback method, monitor sscan status
        """
        mode = kwargs['char_value']    # current status of sscan record.
        self.m_staticText3.LabelText = mode

        if mode == "WAIT:DETCTRS":
            self.m_staticText3.ForegroundColour = 'blue'
            self.m_currentScanPoint.SetLabel(str(self.scan2.current_point))

        elif mode == 'WAIT:MOTORS':
            self.m_staticText3.ForegroundColour = 'red'

        # Todo: we need DCM move to e0 position after scan DONE or IDLE
        #if mode == "IDLE":
        #    ddd = self.e0EdgeEnergy / 1000.0
        #    self.scanPositionerPV.put(value=ddd) # , use_complete=True)

        #else:
        #    self.m_staticText3.ForegroundColour = 'black'

    def eNowCallBack(self, **kwargs):
        """
        callback for now energy display from DCM
        read current energy from DCM. display now energy [eV] unit to GUI
        """
        tempEnow = 1000.0 * float(kwargs['value'])
        self.eNow = tempEnow
        self.m_staticStatusGainE.LabelText = unicode("%5.2f") % tempEnow
        return

    def UpdateCountCallBack(self, **kwargs):
        """
        callback method, display counter values(scaler1_calc(s) value)
        :param kwargs:
        :return:
        """
        if kwargs['value'] == 0:
            temp = self.count.Read(use_calc=True)

            self.m_staticStatusCountIo.LabelText = str(temp[1])  # Io count
            self.m_staticStatusCountIf.LabelText = str(temp[2])  # If count
            self.m_staticStatusCountIt.LabelText = str(temp[3])  # It count
            #self.m_staticStatusCountIr.LabelText = str(temp[4])  # Ir count
            self.m_staticStatusCountIr.LabelText = str(round(temp[7], 5))  # Ir count

            energyTemp = self.eNow - self.e0EdgeEnergy
            self.xEnergy.append(energyTemp)

            TransTemp = round(temp[7], 4)
            self.yTrans.append(TransTemp)

            # transmittance plot
            canvasID=self.plotSpectrum(self.xEnergy, self.yTrans, 'E [eV]',
                                       'Transmittance', self.canvasTrans)
            self.canvasID_panelTrans = canvasID

    def etok(self, energy):
        return np.sqrt(energy/XAFS_K2E)

    def ktoe(self, k):
        return k*k*XAFS_K2E

    def krange(self, start, stop, step):
        i = 1
        kstart = np.square((start + step) / k2e)
        kstop = np.square(stop / k2e)
        while(kstart < kstop):
            yield kstart
            i += 1
            kstart = np.square((start + (i*step)) / k2e)

    """
       Update Step/Npts on XAFS region GUI
    """
    def setStepNpts(self, wids, label, fix_npts=False):
        """
        Set step / npts for start/stop/step/npts/units list of widgets
        """
        start = float(wids[0].GetValue())  # start
        stop = float(wids[1].GetValue())  # stop
        step = float(wids[2].GetValue())  # step
        # read Units value eV or K
        un = wids[4]
        if hasattr(un, 'GetStringSelection'):
            units = un.GetStringSelection()
        else:
            units = un.GetLabel()

        if label == 'npts' or fix_npts:
            npts = float(max(2, wids[3].GetValue()))  # npts
        else:
            try:
                if units == u'K':  # [keV] units
                    kstop = abs(self.etok(stop))
                    kstart = abs(self.etok(start))
                    npts = len(list(self.krange(start=kstart, stop=kstop, step=abs(step))))
                else:  # [eV] units
                    npts = float(max(2, 1 + int(0.1 + abs(stop-start)/abs(step))))
            except ZeroDivisionError:
                npts = 3

        if units == u'eV' or npts == 3:
            wids[2].SetValue(unicode(round((stop-start)/(npts-1), 3)))  # changed by LWW , act=False)
        if not fix_npts:
            try:
                wids[3].SetValue(unicode(npts))  # changed by LWW , act=False)
            except AttributeError:
                pass

    """
       Update XAFS region GUI values from user event
    """
    def onVal(self, evt=None, index=0, label=None, value=None, **kws):
        """
        XAFS onVal
        """
        ## if not self._initialized:
        ##     return
        wids = self.reg_settings[index]

        e0_off = 0

        if 0 == self.m_absRel.GetSelection():  # absolute
            e0_off = self.m_SetEdgeEnergy.GetValue()

        if label in ('start', 'stop', 'step', 'npts'):
            self.setStepNpts(wids, label)
            if label == 'stop' and index < len(self.reg_settings)-1:
                self.reg_settings[index+1][0].SetValue(unicode(value))  # by LWW, act=False)
                self.setStepNpts(self.reg_settings[index+1], label)
            elif label == 'start' and index > 0:
                self.reg_settings[index-1][1].SetValue(unicode(value))  # by LWW, act=False)
                self.setStepNpts(self.reg_settings[index-1], label)

    def scanPosGen(self, e0, regions):
        """
        e0 = XAFS Edge Energy
        regions = Region of Interest items
                  pre-edge(start/stop/step/npts(time)/units)
                  xanes(start/stop/step/npts(time)/units)
                             ..... to XAFS4
        Generation format :
            eOut = scanPosGen(5989, [(-35, -5, 2, 1), (-5, 25, 0.1, 1), (25, 255, 2, 1)])
        """
        eOut = []
        reg = []

        # Todo: we need find disable regions and not generate tuple or position
        # generate region tuples
        for wids in regions:
            # wids[0]=start, wids[1]=stop, wids[2]=step, wids[3]=npts, wids[4]=units
            reg.append((float(wids[0].GetValue()), float(wids[1].GetValue()),
                        float(wids[2].GetValue()), float(wids[3].GetValue()), wids[4]))

        # generate all step position
        for start, stop, step, npts, units in reg:
            # get units and check unit fist
            if hasattr(units, 'GetStringSelection'):  # eV or keV
                unitValue = units.GetStringSelection()
            else:
                unitValue = units.GetLabel()   # eV only

            if unitValue == u'K':  # [keV] units
                kstop = abs(self.etok(stop))
                kstart = abs(self.etok(start))
                tempEout = list(self.krange(start=kstart, stop=kstop, step=abs(step)))
                for i in tempEout:
                    eOut.append(e0 + i)
            else:
                for ipt in range(int((stop-start) / step + 0.1)):
                    eOut.append(e0 + start + step * ipt)

        #Todo: please erase P1PA values after put new values.
        # self.scan2.table = eOut   # write new position to scan2.P1PA
        # DCM IOC need keV unit
        self.scan2.table = list(np.divide(eOut, 1000.))  # write new position to scan2.P1PA
        # set epics scan2.NPTS
        self.scan2.npts = len(eOut)
        # set epics scan2.P1SM (scan mode, 0=Linear, 1=Table, 2=Fly)
        self.scan2.mode = 1


    """---------------------------------------------------------------------------------
    Make canvs for PLOT PANEL
    """
    def make_canvas(self, canvas_name, parent_panel):
        """
        Create plot canvas for parent Panel and make it accessible as an attribute .canvas_name
        :param canvas_name: string containing canvas/attribute name
        :param parent_panel: parent wx.Panel for canvas
        :return: canvas reference
        """
        canvas = plot.PlotCanvas(parent_panel)
        canvas.SetInitialSize(size=parent_panel.GetSize())
        canvas.SetEnableZoom(True)
        canvas.SetEnableGrid(True)

        ## canvas.SetEnableDrag(True)

        canvas.SetGridColour('#CCCCCC')
        canvas.Layout()
        setattr(self, canvas_name, canvas)
        return canvas

    def plotSpectrum(self, x, y, xtitle, ytitle, canvas, moreYf=None, moreYr=None, vlines=None, vline_ymax=None):
        """
        send list of (x/y) tuples to plot output in destination(here called "where")
        :param x: x data(1st data)
        :param y: y data
        :param xtitle: x axis title
        :param ytitle: y axis title
        :param canvas: canvas object reference
        :param moreYf: optional additional ydata serise(2nd data)
        :param moreYr: another additional ydata serise(3rd data)
        :param vlines: an list of x-coordinates to place vertical(green) lines
                        at extending over the current canvas
        :param vline_ymax: if ylines are supplied, supplying this determines their max.
        :return: ID of the canvas object drawn on(useful for clearing a canvas
                  from outside this function; needs XXX = wx.FindWindowById(canvasID)
                  at external point to find this canvas; then clear with XXX.Clear() )
        """
        # clear corresponding canvas first
        canvas.Clear()
        # zip data arrays together into list of tuples (wx.plot requires tuples)
        # then define "line" objects to carry the data into the plot
        plotData = zip(x, y)

        line = plot.PolyLine(plotData, colour='black', width=1)
        maker = plot.PolyMarker(plotData, colour='black', width=0, size=1)

        if moreYf is not None:
            plotMoreData = zip(x, moreYf)
            #moreLine = plot.PolyLine(plotMoreData, colour='blue', width=1)
            moreLine = plot.PolyMarker(plotMoreData, colour='blue', width=1)

        if moreYr is not None:
            plotMoreData2 = zip(x, moreYr)
            #moreLine2 = plot.PolyLine(plotMoreData2, colour='green', width=1)
            moreLine2 = plot.PolyMarker(plotMoreData2, colour='green', width=1)

        serise = [line]

        if moreYf is not None:
            serise.append(moreLine)
        if moreYr is not None:
            serise.append(moreLine2)

        if vlines is not None:
            # Now just plot somthing to ensure that autoscaling show the full range
            #line = plot.PolyLine([(0,1), (self.det.detector_data.mca_bins-1, 1)],
            #                     solour='white', widt=1)
            line = plot.PolyMarker([(0,1), (self.det.detector_data.mca_bins-1, 1)],
                                 solour='white', widt=1)
            serise.append(line)

            # plot ROI range limit lines
            ymax = canvas.ClientSize[1]

            if vline_ymax is not None:
                canvas.SetYSpec('min')
                ymax = vline_ymax
            for x in vlines:
                #line_x = plot.PolyLine([(x,1), (x,ymax)], colour='green', width=1)
                line_x = plot.PolyMarker([(x,1), (x,ymax)], colour='green', width=1)
                serise.append(line_x)

        serise.append(maker)

        gc = plot.PlotGraphics(serise, '', xtitle, ytitle)
        canvas.Draw(gc)
        return canvas.GetId()