def create_plot(self):
if self.plot is not None:
self.plot.close()
self.toolbar.set_auto(True)
if self.settings.display == Display.PLOT:
self.plot = Plotter(self.notify, self.figure, self.settings)
elif self.settings.display == Display.SPECT:
self.plot = Spectrogram(self.notify, self.figure, self.settings)
elif self.settings.display == Display.SURFACE:
self.plot = Plotter3d(self.notify, self.figure, self.settings)
elif self.settings.display == Display.STATUS:
self.plot = PlotterStatus(self.notify, self.figure, self.settings)
else:
self.plot = PlotterTime(self.notify, self.figure, self.settings)
self.__set_fonts()
self.toolbar.set_plot(self.plot)
self.toolbar.set_type(self.settings.display)
self.measureTable.set_type(self.settings.display)
self.set_plot_title()
self.figure.subplots_adjust(top=0.85)
self.redraw_plot()
self.plot.scale_plot(True)
self.mouseZoom = MouseZoom(self.toolbar,
plot=self.plot,
callbackHide=self.__hide_overlay)
self.mouseSelect = MouseSelect(self.plot, self.on_select,
self.on_selected)
self.measureTable.show(self.settings.showMeasure)
self.panel.SetFocus()
def __init__(self, parent, spectrum, settings):
self.spectrum = sort_spectrum(spectrum)
self.settings = settings
self.smoothed = None
wx.Dialog.__init__(self, parent=parent, title='Smooth Spectrum')
self.queue = Queue.Queue()
self.timer = wx.Timer(self)
self.Bind(wx.EVT_TIMER, self.__on_timer, self.timer)
self.timer.Start(self.POLL)
self.figure = matplotlib.figure.Figure(facecolor='white')
self.canvas = FigureCanvas(self, -1, self.figure)
settings = copy.copy(settings)
settings.plotFunc = PlotFunc.NONE
self.plot = Plotter(self.queue, self.figure, settings)
textFunc = wx.StaticText(self, label='Window function')
self.choiceFunc = wx.Choice(self, choices=WINFUNC[::2])
self.choiceFunc.SetSelection(WINFUNC[::2].index(settings.smoothFunc))
textRatio = wx.StaticText(self, label='Smoothing')
self.slideRatio = wx.Slider(self, value=settings.smoothRatio,
minValue=2, maxValue=100,
style=wx.SL_INVERSE)
buttonSmooth = wx.Button(self, label='Smooth')
self.Bind(wx.EVT_BUTTON, self.__on_smooth, buttonSmooth)
sizerButtons = wx.StdDialogButtonSizer()
self.buttonOk = wx.Button(self, wx.ID_OK)
self.buttonOk.Disable()
buttonCancel = wx.Button(self, wx.ID_CANCEL)
sizerButtons.AddButton(self.buttonOk)
sizerButtons.AddButton(buttonCancel)
sizerButtons.Realize()
self.Bind(wx.EVT_BUTTON, self.__on_ok, self.buttonOk)
sizerGrid = wx.GridBagSizer(5, 5)
sizerGrid.Add(self.canvas, pos=(0, 0), span=(10, 6),
flag=wx.EXPAND | wx.ALL, border=5)
sizerGrid.Add(textFunc, pos=(0, 6),
flag=wx.ALL, border=5)
sizerGrid.Add(self.choiceFunc, pos=(1, 6), span=(1, 2),
flag=wx.EXPAND | wx.ALL, border=5)
sizerGrid.Add(textRatio, pos=(2, 6),
flag=wx.ALL, border=5)
sizerGrid.Add(self.slideRatio, pos=(3, 6), span=(1, 2),
flag=wx.EXPAND | wx.ALL, border=5)
sizerGrid.Add(buttonSmooth, pos=(4, 6), span=(1, 2),
flag=wx.ALL | wx.ALIGN_CENTRE, border=5)
sizerGrid.Add(sizerButtons, pos=(10, 6), span=(1, 2),
flag=wx.ALIGN_RIGHT | wx.ALL, border=5)
self.SetSizerAndFit(sizerGrid)
self.__draw_plot(self.spectrum)
def __init__(self, parent, spectrum, settings):
self.spectrum = spectrum
self.settings = settings
self.sweeps = None
self.isExporting = False
wx.Dialog.__init__(self, parent=parent, title='Export Plot Sequence')
self.queue = Queue.Queue()
self.timer = wx.Timer(self)
self.Bind(wx.EVT_TIMER, self.__on_timer, self.timer)
self.timer.Start(self.POLL)
self.figure = matplotlib.figure.Figure(facecolor='white')
self.canvas = FigureCanvas(self, -1, self.figure)
self.plot = Plotter(self.queue, self.figure, settings)
textPlot = wx.StaticText(self, label='Plot')
self.checkAxes = wx.CheckBox(self, label='Axes')
self.checkAxes.SetValue(True)
self.Bind(wx.EVT_CHECKBOX, self.__on_axes, self.checkAxes)
self.checkGrid = wx.CheckBox(self, label='Grid')
self.checkGrid.SetValue(True)
self.Bind(wx.EVT_CHECKBOX, self.__on_grid, self.checkGrid)
self.checkBar = wx.CheckBox(self, label='Bar')
self.checkBar.SetValue(True)
self.Bind(wx.EVT_CHECKBOX, self.__on_bar, self.checkBar)
sizerCheck = wx.BoxSizer(wx.HORIZONTAL)
sizerCheck.Add(self.checkAxes, flag=wx.ALL, border=5)
sizerCheck.Add(self.checkGrid, flag=wx.ALL, border=5)
sizerCheck.Add(self.checkBar, flag=wx.ALL, border=5)
textRange = wx.StaticText(self, label='Range')
self.sweepTimeStamps = sorted([timeStamp for timeStamp in spectrum.keys()])
sweepChoices = [format_time(timeStamp, True) for timeStamp in self.sweepTimeStamps]
textStart = wx.StaticText(self, label="Start")
self.choiceStart = wx.Choice(self, choices=sweepChoices)
self.choiceStart.SetSelection(0)
self.Bind(wx.EVT_CHOICE, self.__on_choice, self.choiceStart)
textEnd = wx.StaticText(self, label="End")
self.choiceEnd = wx.Choice(self, choices=sweepChoices)
self.choiceEnd.SetSelection(len(self.sweepTimeStamps) - 1)
self.Bind(wx.EVT_CHOICE, self.__on_choice, self.choiceEnd)
textSweeps = wx.StaticText(self, label='Sweeps')
self.textSweeps = wx.StaticText(self, label="")
textOutput = wx.StaticText(self, label='Output')
self.textSize = wx.StaticText(self)
buttonSize = wx.Button(self, label='Change...')
buttonSize.SetToolTipString('Change exported image size')
self.Bind(wx.EVT_BUTTON, self.__on_imagesize, buttonSize)
self.__show_image_size()
buttonBrowse = wx.Button(self, label='Browse...')
self.Bind(wx.EVT_BUTTON, self.__on_browse, buttonBrowse)
self.editDir = wx.TextCtrl(self)
self.editDir.SetValue(settings.dirExport)
font = textPlot.GetFont()
fontSize = font.GetPointSize()
font.SetPointSize(fontSize + 4)
textPlot.SetFont(font)
textRange.SetFont(font)
textOutput.SetFont(font)
sizerButtons = wx.StdDialogButtonSizer()
buttonOk = wx.Button(self, wx.ID_OK)
buttonCancel = wx.Button(self, wx.ID_CANCEL)
sizerButtons.AddButton(buttonOk)
sizerButtons.AddButton(buttonCancel)
sizerButtons.Realize()
self.Bind(wx.EVT_BUTTON, self.__on_ok, buttonOk)
sizerGrid = wx.GridBagSizer(5, 5)
sizerGrid.Add(self.canvas, pos=(0, 0), span=(10, 6),
flag=wx.EXPAND | wx.ALL, border=5)
sizerGrid.Add(textPlot, pos=(0, 7),
flag=wx.TOP | wx.BOTTOM, border=5)
sizerGrid.Add(sizerCheck, pos=(1, 7), span=(1, 2),
flag=wx.ALL, border=5)
sizerGrid.Add(textRange, pos=(2, 7),
flag=wx.TOP | wx.BOTTOM, border=5)
sizerGrid.Add(textStart, pos=(3, 7),
flag=wx.ALIGN_CENTRE_VERTICAL | wx.ALL, border=5)
sizerGrid.Add(self.choiceStart, pos=(3, 8),
flag=wx.ALL, border=5)
sizerGrid.Add(textEnd, pos=(4, 7),
flag=wx.ALIGN_CENTRE_VERTICAL | wx.ALL, border=5)
sizerGrid.Add(self.choiceEnd, pos=(4, 8),
flag=wx.ALL, border=5)
sizerGrid.Add(textSweeps, pos=(5, 7),
flag=wx.ALIGN_CENTRE_VERTICAL | wx.ALL, border=5)
sizerGrid.Add(self.textSweeps, pos=(5, 8),
flag=wx.ALIGN_CENTRE_VERTICAL | wx.ALL, border=5)
sizerGrid.Add(textOutput, pos=(6, 7),
flag=wx.TOP | wx.BOTTOM, border=5)
sizerGrid.Add(self.textSize, pos=(7, 7),
flag=wx.ALIGN_CENTRE_VERTICAL | wx.ALL, border=5)
sizerGrid.Add(buttonSize, pos=(7, 8),
flag=wx.ALL, border=5)
sizerGrid.Add(self.editDir, pos=(8, 7), span=(1, 2),
flag=wx.ALL | wx.EXPAND, border=5)
sizerGrid.Add(buttonBrowse, pos=(9, 7),
flag=wx.ALL, border=5)
sizerGrid.Add(sizerButtons, pos=(10, 7), span=(1, 2),
flag=wx.ALIGN_RIGHT | wx.ALL, border=5)
self.SetSizerAndFit(sizerGrid)
self.__draw_plot()
