Exemplo n.º 1
0
    def __init__(self, parent=None):
        QtGui.QWidget.__init__(self, parent)
        #self.timeScale = 1.0
        #self.scale = 1.0
        self.offset = 0.0
        self.ui = Ui_Form()
        self.ui.setupUi(self)
        self.ui.functionText.setFontFamily('Courier')
        self.ui.errorText.setVisible(False)

        self.advancedMode = None  ## if True, then the current state was generated in advanced
        ## mode. Otherwise, it was generated from the simple stim tree.
        self.lockMode = False  ## used to temporarily block changes to self.advancedMode

        self.pSpace = None  ## cached sequence parameter space

        self.cache = {}  ## cached waveforms
        self.cacheRate = None
        self.cacheNPts = None

        self.meta = {  ## holds some extra information about signals (units, expected scale and range, etc)
            ## mostly information useful in configuring SpinBoxes
            'x': {},
            'y': {},
            'xy': {}  ## values that are the product of x and y values
        }

        ## variables that are added into the function evaluation namespace.
        self.extraParams = {}

        ## Simple stim generator
        self.stimParams = StimParamSet()
        self.ui.stimulusTree.setParameters(self.stimParams, showTop=False)
        self.stimParams.sigTreeStateChanged.connect(self.stimParamsChanged)

        ## advanced stim generator
        self.seqParams = SequenceParamSet()
        self.ui.seqTree.setParameters(self.seqParams, showTop=False)

        self.setAdvancedMode(False)
        self.setError()

        self.updateWidgets()

        self.seqParams.sigTreeStateChanged.connect(self.seqParamsChanged)
        self.ui.functionText.textChanged.connect(self.funcChanged)

        self.ui.updateBtn.clicked.connect(self.update)
        self.ui.autoUpdateCheck.clicked.connect(self.autoUpdateClicked)

        self.ui.errorBtn.clicked.connect(self.updateWidgets)
        self.ui.helpBtn.clicked.connect(self.updateWidgets)
        self.ui.advancedBtn.toggled.connect(self.updateWidgets)
        self.ui.forceAdvancedBtn.clicked.connect(self.forceAdvancedClicked)
        self.ui.forceSimpleBtn.clicked.connect(self.forceSimpleClicked)
Exemplo n.º 2
0
    def __init__(self, parent=None):
        QtGui.QWidget.__init__(self, parent)
        #self.timeScale = 1.0
        #self.scale = 1.0
        self.offset = 0.0
        self.ui = Ui_Form()
        self.ui.setupUi(self)
        self.ui.functionText.setFontFamily('Courier')
        self.ui.errorText.setVisible(False)
        
        self.advancedMode = None  ## if True, then the current state was generated in advanced
                                  ## mode. Otherwise, it was generated from the simple stim tree.
        self.lockMode = False     ## used to temporarily block changes to self.advancedMode
        
        self.pSpace = None    ## cached sequence parameter space
        
        self.cache = {}       ## cached waveforms
        self.cacheRate = None
        self.cacheNPts = None

        
        
        self.meta = {  ## holds some extra information about signals (units, expected scale and range, etc)
                       ## mostly information useful in configuring SpinBoxes
            'x': {},
            'y': {},
            'xy': {}  ## values that are the product of x and y values
        }
        
        ## variables that are added into the function evaluation namespace.
        self.extraParams = {}
        
        ## Simple stim generator
        self.stimParams = StimParamSet()
        self.ui.stimulusTree.setParameters(self.stimParams, showTop=False)
        self.stimParams.sigTreeStateChanged.connect(self.stimParamsChanged)
        
        ## advanced stim generator
        self.seqParams = SequenceParamSet()
        self.ui.seqTree.setParameters(self.seqParams, showTop=False)
        
        self.setAdvancedMode(False)
        self.setError()

        self.updateWidgets()
        
        self.seqParams.sigTreeStateChanged.connect(self.seqParamsChanged)
        self.ui.functionText.textChanged.connect(self.funcChanged)
        
        self.ui.updateBtn.clicked.connect(self.update)
        self.ui.autoUpdateCheck.clicked.connect(self.autoUpdateClicked)

        self.ui.errorBtn.clicked.connect(self.updateWidgets)
        self.ui.helpBtn.clicked.connect(self.updateWidgets)
        self.ui.advancedBtn.toggled.connect(self.updateWidgets)
        self.ui.forceAdvancedBtn.clicked.connect(self.forceAdvancedClicked)
        self.ui.forceSimpleBtn.clicked.connect(self.forceSimpleClicked)
Exemplo n.º 3
0
class StimGenerator(QtGui.QWidget):

    sigDataChanged = QtCore.Signal(
    )  ## Emitted when the output of getSingle() is expected to have changed
    sigStateChanged = QtCore.Signal(
    )  ## Emitted when the output of saveState() is expected to have changed
    sigParametersChanged = QtCore.Signal(
    )  ## Emitted when the sequence parameter space has changed
    sigFunctionChanged = QtCore.Signal(
    )  ## Emitted when the waveform-generating function has changed

    def __init__(self, parent=None):
        QtGui.QWidget.__init__(self, parent)
        #self.timeScale = 1.0
        #self.scale = 1.0
        self.offset = 0.0
        self.ui = Ui_Form()
        self.ui.setupUi(self)
        self.ui.functionText.setFontFamily('Courier')
        self.ui.errorText.setVisible(False)

        self.advancedMode = None  ## if True, then the current state was generated in advanced
        ## mode. Otherwise, it was generated from the simple stim tree.
        self.lockMode = False  ## used to temporarily block changes to self.advancedMode

        self.pSpace = None  ## cached sequence parameter space

        self.cache = {}  ## cached waveforms
        self.cacheRate = None
        self.cacheNPts = None

        self.meta = {  ## holds some extra information about signals (units, expected scale and range, etc)
            ## mostly information useful in configuring SpinBoxes
            'x': {},
            'y': {},
            'xy': {}  ## values that are the product of x and y values
        }

        ## variables that are added into the function evaluation namespace.
        self.extraParams = {}

        ## Simple stim generator
        self.stimParams = StimParamSet()
        self.ui.stimulusTree.setParameters(self.stimParams, showTop=False)
        self.stimParams.sigTreeStateChanged.connect(self.stimParamsChanged)

        ## advanced stim generator
        self.seqParams = SequenceParamSet()
        self.ui.seqTree.setParameters(self.seqParams, showTop=False)

        self.setAdvancedMode(False)
        self.setError()

        self.updateWidgets()

        self.seqParams.sigTreeStateChanged.connect(self.seqParamsChanged)
        self.ui.functionText.textChanged.connect(self.funcChanged)

        self.ui.updateBtn.clicked.connect(self.update)
        self.ui.autoUpdateCheck.clicked.connect(self.autoUpdateClicked)

        self.ui.errorBtn.clicked.connect(self.updateWidgets)
        self.ui.helpBtn.clicked.connect(self.updateWidgets)
        self.ui.advancedBtn.toggled.connect(self.updateWidgets)
        self.ui.forceAdvancedBtn.clicked.connect(self.forceAdvancedClicked)
        self.ui.forceSimpleBtn.clicked.connect(self.forceSimpleClicked)

    def setEvalNames(self, **kargs):
        """Make variables accessible for use by evaluated functions."""
        self.extraParams.update(kargs)
        self.clearCache()
        self.autoUpdate()

    def delEvalName(self, name):
        del self.extraParams[name]
        self.clearCache()
        self.autoUpdate()

    def widgetGroupInterface(self):
        return (self.sigStateChanged, StimGenerator.saveState,
                StimGenerator.loadState)

    #def setTimeScale(self, s):
    #"""Set the scale factor for X axis. See setScale for description."""
    #if self.timeScale != s:
    #self.timeScale = s
    #self.clearCache()
    #self.autoUpdate()

    #def setScale(self, s):
    #"""Set the scale factor to be applied to all generated data.
    #This allows, for example, to write waveform functions with values
    #in units of mV and have the resulting data come out in units of V.
    #pulse(10, 10, 100) => gives pulse 100 units tall, but a scale
    #factor of 1e-3 converts it to 0.1 units
    #This should become obsolete--instead we would write the function like
    #pulse(10*ms, 10*ms, 100*mV)
    #This is more verbose but far less ambiguous.
    #"""
    #if self.scale != s:
    #self.scale = s
    #self.clearCache()
    #self.autoUpdate()

    def setOffset(self, o):
        """Set the offset to be added to all generated data.
        This allows, for example, writing a pulse waveform such that 0 is 
        always assumed to mean the current holding value."""
        if self.offset != o:
            self.offset = o
            self.clearCache()
            self.autoUpdate()

    def setMeta(self, axis, **args):
        """Set meta data for X, Y, and XY axes. This is used primarily to configure
        SpinBoxes to display the correct units, limits, step sizes, etc.
        Meta data is _updated_, not replaced. Any previously specified keys will 
        still be present.
        
        Suggested args are:
            suffix='units', dec=True, minStep=1e-3, step=1, limits=(min, max)        
        """
        self.meta[axis].update(args)
        self.stimParams.setMeta(axis, self.meta[axis])

    def clearCache(self):
        self.cache = {}

    def functionString(self):
        return str(self.ui.functionText.toPlainText())

    def update(self):
        ## Let others know that waveform generation has changed.
        ## Note: it's generally better to call autoUpdate instead.
        if self.test():
            self.sigDataChanged.emit()

    def autoUpdate(self):
        if self.ui.autoUpdateCheck.isChecked():
            self.update()

    def autoUpdateClicked(self):
        self.autoUpdate()
        self.sigStateChanged.emit()

    #def errorBtnClicked(self, b):
    #self.updateWidgets()
    ##if b:  ## resize error text box if it is too small
    ##height = self.ui.advSplitter.height()
    ##sizes = self.ui.advSplitter.sizes()
    ##if sizes[2] < height/3.:
    ##diff = (height/3.) - sizes[2]
    ##sizes[2] = height/3.
    ##r = float(sizes[0]) / (sizes[0]+sizes[1])
    ##sizes[0] -= diff * r
    ##sizes[1] -= diff * (1-r)
    ##self.ui.advSplitter.setSizes(sizes)

    def forceSimpleClicked(self):
        self.ui.advancedBtn.setChecked(False)
        self.setAdvancedMode(False)

    def forceAdvancedClicked(self):
        self.ui.advancedBtn.setChecked(True)
        self.setAdvancedMode(True)

    def updateWidgets(self):
        ## show/hide widgets depending on the current mode.
        errVis = self.ui.errorBtn.isChecked()
        self.ui.errorText.setVisible(errVis)
        if errVis or str(self.ui.errorText.toPlainText()) != '':
            self.ui.errorBtn.show()
        else:
            self.ui.errorBtn.hide()

        if self.ui.helpBtn.isChecked():
            self.ui.stack.setCurrentIndex(3)
            return
        if self.ui.advancedBtn.isChecked():
            self.ui.stack.setCurrentIndex(2)
        else:
            if self.advancedMode:
                self.ui.stack.setCurrentIndex(1)
            else:
                self.ui.stack.setCurrentIndex(0)

    def setAdvancedMode(self, adv):
        if self.lockMode or self.advancedMode == adv:
            return
        self.ui.advancedBtn.setChecked(adv)

        if not adv:
            self.stimParamsChanged()  ## to clear out advanced-mode settings
        self.advancedMode = adv
        self.updateWidgets()

    def funcChanged(self):
        ## called when the function string changes
        self.clearCache()
        self.setAdvancedMode(True)

        if self.test():  # test function. If ok, auto-update
            self.autoUpdate()
            self.sigFunctionChanged.emit()
        self.sigStateChanged.emit()

    def seqParamsChanged(self, *args):
        ## called when advanced sequence parameter tree has changed

        #print "seqParamsChanged:", args, "\n"
        ## should filter out some uninteresting events here..

        self.setAdvancedMode(True)
        self.clearCache()
        self.pSpace = None
        if self.test():
            self.autoUpdate()
        self.sigParametersChanged.emit()
        self.sigStateChanged.emit()

    def stimParamsChanged(self, param=None, changes=None):
        ## called when the simple stim generator tree changes
        try:
            funcStr, params = self.stimParams.compile()
        except:
            self.setError(str(sys.exc_info()[1]))
            raise

        #print "stimParamsChanged:", changes, "\n"
        ## Update advanced-mode widgets to reflect (and ultimately enact) changes
        try:
            self.lockMode = True  ## don't let anyone change the mode until we're done
            block = self.signalsBlocked()
            self.blockSignals(
                True)  ## avoid emitting dataChanged signals twice
            try:
                self.seqParams.setState(params)
            finally:
                self.blockSignals(block)
            self.sigParametersChanged.emit()

            self.ui.functionText.setPlainText(funcStr)
        finally:
            self.lockMode = False

    def test(self):
        try:
            self.paramSpace()
            self.setError()
        except:
            self.setError("Error in parameter list:\n" +
                          str(sys.exc_info()[1]))
            return False
        try:
            self.getSingle(1, 1, params={'test': True})
            self.setError()
            return True
        except:
            self.setError("Error in function:\n" + str(sys.exc_info()[1]))
            return False

    def saveState(self):
        """ Return a dict structure with the state of the widget """
        #print "Saving state:", self.functionString()
        state = {
            'function': self.functionString(),
            'params': self.seqParams.getState(),
            'autoUpdate': self.ui.autoUpdateCheck.isChecked(),
            'advancedMode': self.advancedMode,
        }
        if not self.advancedMode:
            state['stimuli'] = self.stimParams.getState()
        return state

    def loadState(self, state):
        """set the parameters with the new state"""
        try:
            block = self.signalsBlocked()
            self.blockSignals(True)  ## avoid emitting multiple signals
            if 'function' in state:
                #self.ui.advancedBtn.setChecked(True)
                self.ui.functionText.setPlainText(state['function'])
            if 'params' in state:
                if isinstance(state['params'],
                              basestring):  ## for backward compatibility
                    state['params'] = seqListParse(state['params'])
                #self.ui.advancedBtn.setChecked(True)
                #self.ui.paramText.setPlainText(state['params'])
                self.seqParams.setState(state['params'])
            if 'stimuli' in state:
                self.stimParams.setState(state['stimuli'])
            if 'autoUpdate' in state:
                #self.ui.advancedBtn.setChecked(False)
                self.ui.autoUpdateCheck.setChecked(state['autoUpdate'])
            if 'advancedMode' in state:
                self.setAdvancedMode(state['advancedMode'])
        finally:
            self.blockSignals(block)
            self.sigDataChanged.emit()
            self.sigStateChanged.emit()
            self.sigParametersChanged.emit()
            self.sigFunctionChanged.emit()

    def paramSpace(self):
        """Return an ordered dict describing the parameter space"""
        ## return looks like:
        ## {
        ##   'param1': (singleVal, [sequence]),
        ##   'param2': (singleVal, [sequence]),
        ##   ...
        ## }

        if self.pSpace is None:
            #self.pSpace = seqListParse(self.paramString()) # get the sequence(s) and the targets
            self.pSpace = self.seqParams.compile()
        return self.pSpace

    def listSequences(self):
        """ return an ordered dict of the sequence parameter names and values in the same order as that
        of the axes returned by get Sequence"""
        ps = self.paramSpace()

        #l = [(k, (ps[k][1]*self.scale)+self.offset) for k in ps.keys() if ps[k][1] != None]
        l = [(k, ps[k][1]) for k in ps.keys() if ps[k][1] is not None]
        d = OrderedDict(l)

        ## d should look like: { 'param1': [val1, val2, ...],  ...  }
        return d

    def flatParamSpace(self):
        """return a list of every point in the parameter space"""
        l = self.listSequences()
        shape = tuple(l.values())
        ar = ones(shape)
        return argwhere(ar)

    def setError(self, msg=None):
        if msg is None or msg == '':
            self.ui.errorText.setText('')
            self.ui.errorBtn.setStyleSheet('')
            #self.ui.errorBtn.hide()
        else:
            self.ui.errorText.setText(msg)
            self.ui.errorBtn.setStyleSheet(
                'QToolButton {border: 2px solid #F00; border-radius: 3px}')
            #self.ui.errorBtn.show()
        self.updateWidgets()

    def getSingle(self, rate, nPts, params=None):
        """
        Return a single generated waveform (possibly cached) with the given sample rate
        number of samples, and sequence parameters.        
        """
        if params is None:
            params = {}

        if self.cacheRate != rate or self.cacheNPts != nPts:
            self.clearCache()

        paramKey = tuple(params.items())
        if paramKey in self.cache:
            return self.cache[paramKey]

        self.cacheRate = rate
        self.cacheNPts = nPts

        ## create namespace with generator functions.
        ##   - iterates over all functions provided in waveforms module
        ##   - wrap each function to automatically provide rate and nPts arguments
        ns = {}
        #arg = {'rate': rate * self.timeScale, 'nPts': nPts}
        arg = {'rate': rate, 'nPts': nPts}
        ns.update(arg)  ## copy rate and nPts to eval namespace
        for i in dir(waveforms):
            obj = getattr(waveforms, i)
            if type(obj) is types.FunctionType:
                ns[i] = self.makeWaveFunction(i, arg)

        ## add current sequence parameter values into namespace
        seq = self.paramSpace(
        )  # -- this is where the Laser bug was happening -- seq becomes 'Pulse_sum', but params was {'power.Pulse_sum': x}, so the default value is always used instead (fixed by removing 'power.' before the params are sent to stimGenerator, but perhaps there is a better place to fix this)
        for k in seq:
            if k in params:  ## select correct value from sequence list
                try:
                    ns[k] = float(seq[k][1][params[k]])
                except IndexError:
                    print "Requested value %d for param %s, but only %d in the param list." % (
                        params[k], str(k), len(seq[k][1]))
                    raise
            else:  ## just use single value
                ns[k] = float(seq[k][0])

        ## add units into namespace
        ns.update(units.allUnits)

        ## add extra parameters to namespace
        ns.update(self.extraParams)

        ## evaluate and return
        fn = self.functionString()

        ## build global namespace with numpy imported
        #gns = {}
        ns['np'] = np

        #print "function: '%s'" % fn
        if fn.strip() == '':
            ret = np.zeros(nPts)
        else:
            try:  # first try eval() without line breaks for backward compatibility
                ret = eval(fn.replace('\n', ''), ns, {})
            except SyntaxError:  # next try exec() as contents of a function
                try:
                    run = "\noutput=fn()\n"
                    code = "def fn():\n" + "\n".join(
                        ["    " + l for l in fn.split('\n')]) + run
                    #print "--- Code: ---"
                    #print code
                    #print "-------------"
                    lns = {}
                    exec(code, ns, lns)
                    ret = lns['output']
                except SyntaxError as err:
                    err.lineno -= 1
                    raise err

        if isinstance(ret, ndarray):
            #ret *= self.scale
            ret += self.offset
            #print "===eval===", ret.min(), ret.max(), self.scale
        elif ret is not None:
            raise TypeError("Function must return ndarray or None.")

        if 'message' in arg:
            self.setError(arg['message'])
        else:
            self.setError()

        self.cache[paramKey] = ret
        return ret

    def makeWaveFunction(self, name, arg):
        ## Creates a copy of a wave function (such as steps or pulses) with the first parameter filled in
        ## Must be in its own function so that obj is properly scoped to the lambda function.
        obj = getattr(waveforms, name)
        return lambda *args, **kwargs: obj(arg, *args, **kwargs)
Exemplo n.º 4
0
class StimGenerator(QtGui.QWidget):

    sigDataChanged = QtCore.Signal()  ## Emitted when the output of getSingle() is expected to have changed
    sigStateChanged = QtCore.Signal()  ## Emitted when the output of saveState() is expected to have changed
    sigParametersChanged = QtCore.Signal()  ## Emitted when the sequence parameter space has changed
    sigFunctionChanged = QtCore.Signal()  ## Emitted when the waveform-generating function has changed

    def __init__(self, parent=None):
        QtGui.QWidget.__init__(self, parent)
        # self.timeScale = 1.0
        # self.scale = 1.0
        self.offset = 0.0
        self.ui = Ui_Form()
        self.ui.setupUi(self)
        self.ui.functionText.setFontFamily("Courier")
        self.ui.errorText.setVisible(False)

        self.advancedMode = None  ## if True, then the current state was generated in advanced
        ## mode. Otherwise, it was generated from the simple stim tree.
        self.lockMode = False  ## used to temporarily block changes to self.advancedMode

        self.pSpace = None  ## cached sequence parameter space

        self.cache = {}  ## cached waveforms
        self.cacheRate = None
        self.cacheNPts = None

        self.meta = {  ## holds some extra information about signals (units, expected scale and range, etc)
            ## mostly information useful in configuring SpinBoxes
            "x": {},
            "y": {},
            "xy": {},  ## values that are the product of x and y values
        }

        ## variables that are added into the function evaluation namespace.
        self.extraParams = {}

        ## Simple stim generator
        self.stimParams = StimParamSet()
        self.ui.stimulusTree.setParameters(self.stimParams, showTop=False)
        self.stimParams.sigTreeStateChanged.connect(self.stimParamsChanged)

        ## advanced stim generator
        self.seqParams = SequenceParamSet()
        self.ui.seqTree.setParameters(self.seqParams, showTop=False)

        self.setAdvancedMode(False)
        self.setError()

        self.updateWidgets()

        self.seqParams.sigTreeStateChanged.connect(self.seqParamsChanged)
        self.ui.functionText.textChanged.connect(self.funcChanged)

        self.ui.updateBtn.clicked.connect(self.update)
        self.ui.autoUpdateCheck.clicked.connect(self.autoUpdateClicked)

        self.ui.errorBtn.clicked.connect(self.updateWidgets)
        self.ui.helpBtn.clicked.connect(self.updateWidgets)
        self.ui.advancedBtn.toggled.connect(self.updateWidgets)
        self.ui.forceAdvancedBtn.clicked.connect(self.forceAdvancedClicked)
        self.ui.forceSimpleBtn.clicked.connect(self.forceSimpleClicked)

    def setEvalNames(self, **kargs):
        """Make variables accessible for use by evaluated functions."""
        self.extraParams.update(kargs)
        self.clearCache()
        self.autoUpdate()

    def delEvalName(self, name):
        del self.extraParams[name]
        self.clearCache()
        self.autoUpdate()

    def widgetGroupInterface(self):
        return (self.sigStateChanged, StimGenerator.saveState, StimGenerator.loadState)

    # def setTimeScale(self, s):
    # """Set the scale factor for X axis. See setScale for description."""
    # if self.timeScale != s:
    # self.timeScale = s
    # self.clearCache()
    # self.autoUpdate()

    # def setScale(self, s):
    # """Set the scale factor to be applied to all generated data.
    # This allows, for example, to write waveform functions with values
    # in units of mV and have the resulting data come out in units of V.
    # pulse(10, 10, 100) => gives pulse 100 units tall, but a scale
    # factor of 1e-3 converts it to 0.1 units
    # This should become obsolete--instead we would write the function like
    # pulse(10*ms, 10*ms, 100*mV)
    # This is more verbose but far less ambiguous.
    # """
    # if self.scale != s:
    # self.scale = s
    # self.clearCache()
    # self.autoUpdate()

    def setOffset(self, o):
        """Set the offset to be added to all generated data.
        This allows, for example, writing a pulse waveform such that 0 is 
        always assumed to mean the current holding value."""
        if self.offset != o:
            self.offset = o
            self.clearCache()
            self.autoUpdate()

    def setMeta(self, axis, **args):
        """Set meta data for X, Y, and XY axes. This is used primarily to configure
        SpinBoxes to display the correct units, limits, step sizes, etc.
        Meta data is _updated_, not replaced. Any previously specified keys will 
        still be present.
        
        Suggested args are:
            suffix='units', dec=True, minStep=1e-3, step=1, limits=(min, max)        
        """
        self.meta[axis].update(args)
        self.stimParams.setMeta(axis, self.meta[axis])

    def clearCache(self):
        self.cache = {}

    def functionString(self):
        return str(self.ui.functionText.toPlainText())

    def update(self):
        ## Let others know that waveform generation has changed.
        ## Note: it's generally better to call autoUpdate instead.
        if self.test():
            self.sigDataChanged.emit()

    def autoUpdate(self):
        if self.ui.autoUpdateCheck.isChecked():
            self.update()

    def autoUpdateClicked(self):
        self.autoUpdate()
        self.sigStateChanged.emit()

    # def errorBtnClicked(self, b):
    # self.updateWidgets()
    ##if b:  ## resize error text box if it is too small
    ##height = self.ui.advSplitter.height()
    ##sizes = self.ui.advSplitter.sizes()
    ##if sizes[2] < height/3.:
    ##diff = (height/3.) - sizes[2]
    ##sizes[2] = height/3.
    ##r = float(sizes[0]) / (sizes[0]+sizes[1])
    ##sizes[0] -= diff * r
    ##sizes[1] -= diff * (1-r)
    ##self.ui.advSplitter.setSizes(sizes)

    def forceSimpleClicked(self):
        self.ui.advancedBtn.setChecked(False)
        self.setAdvancedMode(False)

    def forceAdvancedClicked(self):
        self.ui.advancedBtn.setChecked(True)
        self.setAdvancedMode(True)

    def updateWidgets(self):
        ## show/hide widgets depending on the current mode.
        errVis = self.ui.errorBtn.isChecked()
        self.ui.errorText.setVisible(errVis)
        if errVis or str(self.ui.errorText.toPlainText()) != "":
            self.ui.errorBtn.show()
        else:
            self.ui.errorBtn.hide()

        if self.ui.helpBtn.isChecked():
            self.ui.stack.setCurrentIndex(3)
            return
        if self.ui.advancedBtn.isChecked():
            self.ui.stack.setCurrentIndex(2)
        else:
            if self.advancedMode:
                self.ui.stack.setCurrentIndex(1)
            else:
                self.ui.stack.setCurrentIndex(0)

    def setAdvancedMode(self, adv):
        if self.lockMode or self.advancedMode == adv:
            return
        self.ui.advancedBtn.setChecked(adv)

        if not adv:
            self.stimParamsChanged()  ## to clear out advanced-mode settings
        self.advancedMode = adv
        self.updateWidgets()

    def funcChanged(self):
        ## called when the function string changes
        self.clearCache()
        self.setAdvancedMode(True)

        if self.test():  # test function. If ok, auto-update
            self.autoUpdate()
            self.sigFunctionChanged.emit()
        self.sigStateChanged.emit()

    def seqParamsChanged(self, *args):
        ## called when advanced sequence parameter tree has changed

        # print "seqParamsChanged:", args, "\n"
        ## should filter out some uninteresting events here..

        self.setAdvancedMode(True)
        self.clearCache()
        self.pSpace = None
        if self.test():
            self.autoUpdate()
        self.sigParametersChanged.emit()
        self.sigStateChanged.emit()

    def stimParamsChanged(self, param=None, changes=None):
        ## called when the simple stim generator tree changes
        try:
            funcStr, params = self.stimParams.compile()
        except:
            self.setError(str(sys.exc_info()[1]))
            raise

        # print "stimParamsChanged:", changes, "\n"
        ## Update advanced-mode widgets to reflect (and ultimately enact) changes
        try:
            self.lockMode = True  ## don't let anyone change the mode until we're done
            self.blockSignals(True)  ## avoid emitting dataChanged signals twice
            try:
                self.seqParams.setState(params)
            finally:
                self.blockSignals(False)
            self.sigParametersChanged.emit()

            self.ui.functionText.setPlainText(funcStr)
        finally:
            self.lockMode = False

    def test(self):
        try:
            self.paramSpace()
            self.setError()
        except:
            self.setError("Error in parameter list:\n" + str(sys.exc_info()[1]))
            return False
        try:
            self.getSingle(1, 1, params={"test": True})
            self.setError()
            return True
        except:
            self.setError("Error in function:\n" + str(sys.exc_info()[1]))
            return False

    def saveState(self):
        """ Return a dict structure with the state of the widget """
        # print "Saving state:", self.functionString()
        state = {
            "function": self.functionString(),
            "params": self.seqParams.getState(),
            "autoUpdate": self.ui.autoUpdateCheck.isChecked(),
            "advancedMode": self.advancedMode,
        }
        if not self.advancedMode:
            state["stimuli"] = self.stimParams.getState()
        return state

    def loadState(self, state):
        """set the parameters with the new state"""
        try:
            self.blockSignals(True)  ## avoid emitting multiple signals
            if "function" in state:
                # self.ui.advancedBtn.setChecked(True)
                self.ui.functionText.setPlainText(state["function"])
            if "params" in state:
                if isinstance(state["params"], basestring):  ## for backward compatibility
                    state["params"] = seqListParse(state["params"])
                # self.ui.advancedBtn.setChecked(True)
                # self.ui.paramText.setPlainText(state['params'])
                self.seqParams.setState(state["params"])
            if "stimuli" in state:
                self.stimParams.setState(state["stimuli"])
            if "autoUpdate" in state:
                # self.ui.advancedBtn.setChecked(False)
                self.ui.autoUpdateCheck.setChecked(state["autoUpdate"])
            if "advancedMode" in state:
                self.setAdvancedMode(state["advancedMode"])
        finally:
            self.blockSignals(False)
            self.sigDataChanged.emit()
            self.sigStateChanged.emit()
            self.sigParametersChanged.emit()
            self.sigFunctionChanged.emit()

    def paramSpace(self):
        """Return an ordered dict describing the parameter space"""
        ## return looks like:
        ## {
        ##   'param1': (singleVal, [sequence]),
        ##   'param2': (singleVal, [sequence]),
        ##   ...
        ## }

        if self.pSpace is None:
            # self.pSpace = seqListParse(self.paramString()) # get the sequence(s) and the targets
            self.pSpace = self.seqParams.compile()
        return self.pSpace

    def listSequences(self):
        """ return an ordered dict of the sequence parameter names and values in the same order as that
        of the axes returned by get Sequence"""
        ps = self.paramSpace()

        # l = [(k, (ps[k][1]*self.scale)+self.offset) for k in ps.keys() if ps[k][1] != None]
        l = [(k, ps[k][1]) for k in ps.keys() if ps[k][1] != None]
        d = OrderedDict(l)

        ## d should look like: { 'param1': [val1, val2, ...],  ...  }
        return d

    def flatParamSpace(self):
        """return a list of every point in the parameter space"""
        l = self.listSequences()
        shape = tuple(l.values())
        ar = ones(shape)
        return argwhere(ar)

    def setError(self, msg=None):
        if msg is None or msg == "":
            self.ui.errorText.setText("")
            self.ui.errorBtn.setStyleSheet("")
            # self.ui.errorBtn.hide()
        else:
            self.ui.errorText.setText(msg)
            self.ui.errorBtn.setStyleSheet("QToolButton {border: 2px solid #F00; border-radius: 3px}")
            # self.ui.errorBtn.show()
        self.updateWidgets()

    def getSingle(self, rate, nPts, params=None):
        """
        Return a single generated waveform (possibly cached) with the given sample rate
        number of samples, and sequence parameters.        
        """
        if params is None:
            params = {}

        if self.cacheRate != rate or self.cacheNPts != nPts:
            self.clearCache()

        paramKey = tuple(params.items())
        if paramKey in self.cache:
            return self.cache[paramKey]

        self.cacheRate = rate
        self.cacheNPts = nPts

        ## create namespace with generator functions.
        ##   - iterates over all functions provided in waveforms module
        ##   - wrap each function to automatically provide rate and nPts arguments
        ns = {}
        # arg = {'rate': rate * self.timeScale, 'nPts': nPts}
        arg = {"rate": rate, "nPts": nPts}
        ns.update(arg)  ## copy rate and nPts to eval namespace
        for i in dir(waveforms):
            obj = getattr(waveforms, i)
            if type(obj) is types.FunctionType:
                ns[i] = self.makeWaveFunction(i, arg)

        ## add current sequence parameter values into namespace
        seq = (
            self.paramSpace()
        )  # -- this is where the Laser bug was happening -- seq becomes 'Pulse_sum', but params was {'power.Pulse_sum': x}, so the default value is always used instead (fixed by removing 'power.' before the params are sent to stimGenerator, but perhaps there is a better place to fix this)
        for k in seq:
            if k in params:  ## select correct value from sequence list
                try:
                    ns[k] = float(seq[k][1][params[k]])
                except IndexError:
                    print "Requested value %d for param %s, but only %d in the param list." % (
                        params[k],
                        str(k),
                        len(seq[k][1]),
                    )
                    raise
            else:  ## just use single value
                ns[k] = float(seq[k][0])

        ## add units into namespace
        ns.update(units.allUnits)

        ## add extra parameters to namespace
        ns.update(self.extraParams)

        ## evaluate and return
        fn = self.functionString()

        ## build global namespace with numpy imported
        # gns = {}
        ns["np"] = np

        # print "function: '%s'" % fn
        if fn.strip() == "":
            ret = np.zeros(nPts)
        else:
            try:  # first try eval() without line breaks for backward compatibility
                ret = eval(fn.replace("\n", ""), ns, {})
            except SyntaxError:  # next try exec() as contents of a function
                try:
                    run = "\noutput=fn()\n"
                    code = "def fn():\n" + "\n".join(["    " + l for l in fn.split("\n")]) + run
                    # print "--- Code: ---"
                    # print code
                    # print "-------------"
                    lns = {}
                    exec (code, ns, lns)
                    ret = lns["output"]
                except SyntaxError as err:
                    err.lineno -= 1
                    raise err

        if isinstance(ret, ndarray):
            # ret *= self.scale
            ret += self.offset
            # print "===eval===", ret.min(), ret.max(), self.scale
        elif ret is not None:
            raise TypeError("Function must return ndarray or None.")

        if "message" in arg:
            self.setError(arg["message"])
        else:
            self.setError()

        self.cache[paramKey] = ret
        return ret

    def makeWaveFunction(self, name, arg):
        ## Creates a copy of a wave function (such as steps or pulses) with the first parameter filled in
        ## Must be in its own function so that obj is properly scoped to the lambda function.
        obj = getattr(waveforms, name)
        return lambda *args, **kwargs: obj(arg, *args, **kwargs)