Python EmPlotter.isClosed方法代码示例

# 需要导入模块: from pyworkflow.em.plotter import EmPlotter [as 别名]
# 或者: from pyworkflow.em.plotter.EmPlotter import isClosed [as 别名]

#.........这里部分代码省略.........
        xValues=xValues[0] # From [array(...)] to array(...)
        yValues=yValues[0]

        useMeasureLog = self.useMeasureLog()
        useTimeLog = self.useTimeLog()

        if not (useMeasureLog or useTimeLog):
            return xValues, yValues

        # If log will be used either for time or measure var
        # we need to filter elements larger than 0
        newXValues = []
        newYValues = []

        def _value(v, useLog):
            if useLog:
                return math.log10(v) if v > 0 else None
            return v

        for x, y in izip(xValues, yValues):
            x = _value(x, useTimeLog)
            y = _value(y, useMeasureLog)

            if x is not None and y is not None:
                newXValues.append(x)
                newYValues.append(y)

        return newXValues, newYValues

    def _onPlotClick(self, e=None):
        sampleKeys = self.samplesTree.selection()

        if sampleKeys:
            if self.plotter is None or self.plotter.isClosed():
                self.plotter = EmPlotter()
                doShow = True
                ax = self.plotter.createSubPlot("Plot", self.getTimeLabel(),
                                                self.getMeasureLabel())
                self.plotDict = {}
            else:
                doShow = False
                ax = self.plotter.getLastSubPlot()

            samples = [self.experiment.samples[k] for k in sampleKeys]
            for s in samples:
                if not s.sampleName in self.plotDict:
                    x, y = self.getPlotValues(s)
                    ax.plot(x, y, label=s.sampleName)
                    self.plotDict[s.sampleName] = True
            leg = ax.legend()
            if leg:
                leg.draggable()

            if doShow:
                self.plotter.show()
            else:
                self.plotter.draw()
        else:
            self.showInfo("Please select some sample(s) to plot.")

    def _onPlotSummaryClick(self, e=None):
        sampleKeys = self.samplesTree.selection()
        n = len(sampleKeys)

        if n == 1:
            self.showInfo("Please select several samples to plot.")

# 需要导入模块: from pyworkflow.em.plotter import EmPlotter [as 别名]
# 或者: from pyworkflow.em.plotter.EmPlotter import isClosed [as 别名]

#.........这里部分代码省略.........

    def useMeasureLog(self):
        return self.measureWidget[2].get()

    def getUnits(self, varName):
        return self.experiment.variables[varName].getUnitsString()

    def getLabel(self, varName, useLog):
        varLabel = '%s [%s]' % (varName, self.getUnits(varName))
        if useLog:
            varLabel = "log10(%s)" % varLabel
        return varLabel

    def getTimeLabel(self):
        return self.getLabel(self.varNameX, self.useTimeLog())

    def getMeasureLabel(self):
        return self.getLabel(self.varNameY, self.useMeasureLog())

    def computePlotValues(self, xValues, yValues):
        useMeasureLog = self.useMeasureLog()
        useTimeLog = self.useTimeLog()

        if not (useMeasureLog or useTimeLog):
            newXValues = xValues
            newYValues = yValues
        else:
            # If log will be used either for time or measure var
            # we need to filter elements larger than 0
            newXValues = []
            newYValues = []

            def _value(v, useLog):
                if useLog:
                    return math.log10(v) if v > 0 else None
                return v

            for x, y in izip(xValues, yValues):
                x = _value(x, useTimeLog)
                y = _value(y, useMeasureLog)

                if x is not None and y is not None:
                    newXValues.append(x)
                    newYValues.append(y)

        return newXValues, newYValues

    def _updateModel(self):
        """ This function should be called whenever the sample changes """
        pass

    def _onLogChanged(self, *args):
        # We will treat a log change as a sample change to plot
        self._onSampleChanged()

    def _onSampleChanged(self, e=None):
        sampleKeys = self.samplesTree.selection()

        if sampleKeys:
            # When the sample is changed we need to re-compute (with log or not)
            # the x, y values
            self.sample = self.experiment.samples[sampleKeys[0]]
            self.xValues, self.yValues = self.sample.getXYValues(self.varNameX,
                                                                 self.varNameY)
            self.newXValues, self.newYValues = self.computePlotValues(self.xValues[0],
                                                                      self.yValues[0])
            self._updateModel()
            self.computeFit()
            self.plotResults()
        else:
            dialog.showInfo("Warning","Please select some sample(s) to plot.",self)

    def plotResults(self):
        if self.plotter is None or self.plotter.isClosed():
            self.plotter = EmPlotter()
            doShow = True
        else:
            doShow = False
            ax = self.plotter.getLastSubPlot()
            self.plotter.clear()

        ax = self.plotter.createSubPlot("Sample: %s" % self.sample.sampleName,
                                        self.getTimeLabel(),
                                        self.getMeasureLabel())
        self.newXPValues, self.newYPValues = self.computePlotValues(self.xpValues[0],
                                                                    self.ypValues[0])
        ax.plot(self.newXValues, self.newYValues, 'x', label="Observations")
        ax.plot(self.newXPValues, self.newYPValues, label="Fit")
        ax.legend()

        if doShow:
            self.plotter.show()
        else:
            self.plotter.draw()

    def destroy(self):
        """Destroy the window"""
        if not (self.plotter is None or self.plotter.isClosed()):
            self.plotter.close()
        dialog.Dialog.destroy(self)