class FrameMain(wx.Frame): def __init__(self, title, pool): self.grid = True self.pool = pool self.lock = threading.Lock() self.sdr = None self.threadScan = None self.threadUpdate = None self.stopAtEnd = False self.stopScan = False self.dlgCal = None self.menuOpen = None self.menuSave = None self.menuExport = None self.menuProperties = None self.menuPref = None self.menuAdvPref = None self.menuClearSelect = None self.menuShowMeasure = None self.menuDevices = None self.menuStart = None self.menuStop = None self.menuStopEnd = None self.menuCompare = None self.menuCal = None self.popupMenu = None self.popupMenuStart = None self.popupMenuStop = None self.popupMenuStopEnd = None self.popupMenuRangeLim = None self.popupMenuPointsLim = None self.popupMenuClearSelect = None self.popupMenuShowMeasure = None self.graph = None self.toolbar = None self.canvas = None self.mouseZoom = None self.mouseSelect = None self.buttonStart = None self.buttonStop = None self.controlGain = None self.choiceMode = None self.choiceDwell = None self.choiceNfft = None self.spinCtrlStart = None self.spinCtrlStop = None self.checkUpdate = None self.checkGrid = None self.spectrum = {} self.scanInfo = ScanInfo() self.isSaved = True self.settings = Settings() self.devices = get_devices(self.settings.devices) self.filename = "" self.oldCal = 0 wx.Frame.__init__(self, None, title=title) self.Bind(wx.EVT_CLOSE, self.on_exit) self.status = Statusbar(self) self.SetStatusBar(self.status) add_colours() self.create_widgets() self.create_menu() self.create_popup_menu() self.set_control_state(True) self.Show() displaySize = wx.DisplaySize() toolbarSize = self.toolbar.GetBestSize() self.SetClientSize((toolbarSize[0] + 10, displaySize[1] / 2)) self.SetMinSize((displaySize[0] / 4, displaySize[1] / 4)) self.Connect(-1, -1, EVT_THREAD_STATUS, self.on_event) self.SetDropTarget(DropTarget(self)) def create_widgets(self): panel = wx.Panel(self) self.graph = PanelGraph(panel, self, self.settings, self.on_motion) self.toolbar = wx.Panel(panel) self.buttonStart = wx.Button(self.toolbar, wx.ID_ANY, "Start") self.buttonStop = wx.Button(self.toolbar, wx.ID_ANY, "Stop") self.buttonStart.SetToolTip(wx.ToolTip("Start scan")) self.buttonStop.SetToolTip(wx.ToolTip("Stop scan")) self.Bind(wx.EVT_BUTTON, self.on_start, self.buttonStart) self.Bind(wx.EVT_BUTTON, self.on_stop, self.buttonStop) textRange = wx.StaticText(self.toolbar, label="Range (MHz)", style=wx.ALIGN_CENTER) textStart = wx.StaticText(self.toolbar, label="Start") textStop = wx.StaticText(self.toolbar, label="Stop") self.spinCtrlStart = wx.SpinCtrl(self.toolbar) self.spinCtrlStop = wx.SpinCtrl(self.toolbar) self.spinCtrlStart.SetToolTip(wx.ToolTip("Start frequency")) self.spinCtrlStop.SetToolTip(wx.ToolTip("Stop frequency")) self.spinCtrlStart.SetRange(F_MIN, F_MAX - 1) self.spinCtrlStop.SetRange(F_MIN + 1, F_MAX) self.Bind(wx.EVT_SPINCTRL, self.on_spin, self.spinCtrlStart) self.Bind(wx.EVT_SPINCTRL, self.on_spin, self.spinCtrlStop) textGain = wx.StaticText(self.toolbar, label="Gain (dB)") self.controlGain = wx.Choice(self.toolbar, choices=[""]) textMode = wx.StaticText(self.toolbar, label="Mode") self.choiceMode = wx.Choice(self.toolbar, choices=MODE[::2]) self.choiceMode.SetToolTip(wx.ToolTip("Scanning mode")) textDwell = wx.StaticText(self.toolbar, label="Dwell") self.choiceDwell = wx.Choice(self.toolbar, choices=DWELL[::2]) self.choiceDwell.SetToolTip(wx.ToolTip("Scan time per step")) textNfft = wx.StaticText(self.toolbar, label="FFT size") self.choiceNfft = wx.Choice(self.toolbar, choices=map(str, NFFT)) self.choiceNfft.SetToolTip(wx.ToolTip("Higher values for greater" "precision")) textDisplay = wx.StaticText(self.toolbar, label="Display") self.choiceDisplay = wx.Choice(self.toolbar, choices=DISPLAY[::2]) self.Bind(wx.EVT_CHOICE, self.on_choice, self.choiceDisplay) self.choiceDisplay.SetToolTip(wx.ToolTip("Spectrogram available in" "continuous mode")) grid = wx.GridBagSizer(5, 5) grid.Add(self.buttonStart, pos=(0, 0), span=(3, 1), flag=wx.ALIGN_CENTER) grid.Add(self.buttonStop, pos=(0, 1), span=(3, 1), flag=wx.ALIGN_CENTER) grid.Add((20, 1), pos=(0, 2)) grid.Add(textRange, pos=(0, 3), span=(1, 4), flag=wx.ALIGN_CENTER) grid.Add(textStart, pos=(1, 3), flag=wx.ALIGN_CENTER) grid.Add(self.spinCtrlStart, pos=(1, 4)) grid.Add(textStop, pos=(1, 5), flag=wx.ALIGN_CENTER) grid.Add(self.spinCtrlStop, pos=(1, 6)) grid.Add(textGain, pos=(0, 7), flag=wx.ALIGN_CENTER) grid.Add(self.controlGain, pos=(1, 7), flag=wx.ALIGN_CENTER) grid.Add((20, 1), pos=(0, 8)) grid.Add(textMode, pos=(0, 9), flag=wx.ALIGN_CENTER) grid.Add(self.choiceMode, pos=(1, 9), flag=wx.ALIGN_CENTER) grid.Add(textDwell, pos=(0, 10), flag=wx.ALIGN_CENTER) grid.Add(self.choiceDwell, pos=(1, 10), flag=wx.ALIGN_CENTER) grid.Add(textNfft, pos=(0, 11), flag=wx.ALIGN_CENTER) grid.Add(self.choiceNfft, pos=(1, 11), flag=wx.ALIGN_CENTER) grid.Add((20, 1), pos=(0, 12)) grid.Add(textDisplay, pos=(0, 13), flag=wx.ALIGN_CENTER) grid.Add(self.choiceDisplay, pos=(1, 13), flag=wx.ALIGN_CENTER) self.set_controls() self.set_gain_control() self.toolbar.SetSizer(grid) self.toolbar.Layout() sizer = wx.BoxSizer(wx.VERTICAL) sizer.Add(self.graph, 1, wx.EXPAND) sizer.Add(self.toolbar, 0, wx.EXPAND) panel.SetSizer(sizer) panel.Layout() def create_menu(self): menuFile = wx.Menu() self.menuOpen = menuFile.Append(wx.ID_OPEN, "&Open...", "Open plot") recent = wx.Menu() self.settings.fileHistory.UseMenu(recent) self.settings.fileHistory.AddFilesToMenu() menuFile.AppendMenu(wx.ID_ANY, "&Recent Files", recent) menuFile.AppendSeparator() self.menuSave = menuFile.Append(wx.ID_SAVE, "&Save As...", "Save plot") self.menuExport = menuFile.Append(wx.ID_ANY, "&Export...", "Export plot") menuFile.AppendSeparator() self.menuProperties = menuFile.Append(wx.ID_ANY, "&Properties...", "Show properties") menuFile.AppendSeparator() menuExit = menuFile.Append(wx.ID_EXIT, "E&xit", "Exit the program") menuEdit = wx.Menu() self.menuPref = menuEdit.Append(wx.ID_ANY, "&Preferences...", "Preferences") self.menuAdvPref = menuEdit.Append(wx.ID_ANY, "&Advanced preferences...", "Advanced preferences") self.menuDevices = menuEdit.Append(wx.ID_ANY, "&Devices...", "Device selection and configuration") menuView = wx.Menu() self.menuClearSelect = menuView.Append(wx.ID_ANY, "Clear selection", "Clear current selection") self.graph.add_menu_clear_select(self.menuClearSelect) self.menuShowMeasure = menuView.Append( wx.ID_ANY, "Show &measurements", "Show measurements window", kind=wx.ITEM_CHECK ) self.menuShowMeasure.Check(self.settings.showMeasure) menuScan = wx.Menu() self.menuStart = menuScan.Append(wx.ID_ANY, "&Start", "Start scan") self.menuStop = menuScan.Append(wx.ID_ANY, "S&top", "Stop scan immediately") self.menuStopEnd = menuScan.Append(wx.ID_ANY, "Stop at &end", "Complete current sweep " "before stopping") menuTools = wx.Menu() self.menuCompare = menuTools.Append(wx.ID_ANY, "&Compare...", "Compare plots") self.menuCal = menuTools.Append( wx.ID_ANY, "&Auto Calibration...", "Automatically calibrate to a known frequency" ) menuHelp = wx.Menu() menuHelpLink = menuHelp.Append(wx.ID_HELP, "&Help...", "Link to help") menuUpdate = menuHelp.Append(wx.ID_ANY, "&Check for updates...", "Check for updates to the program") menuAbout = menuHelp.Append(wx.ID_ABOUT, "&About...", "Information about this program") menuBar = wx.MenuBar() menuBar.Append(menuFile, "&File") menuBar.Append(menuEdit, "&Edit") menuBar.Append(menuView, "&View") menuBar.Append(menuScan, "&Scan") menuBar.Append(menuTools, "&Tools") menuBar.Append(menuHelp, "&Help") self.SetMenuBar(menuBar) self.Bind(wx.EVT_MENU, self.on_open, self.menuOpen) self.Bind(wx.EVT_MENU_RANGE, self.on_file_history, id=wx.ID_FILE1, id2=wx.ID_FILE9) self.Bind(wx.EVT_MENU, self.on_save, self.menuSave) self.Bind(wx.EVT_MENU, self.on_export, self.menuExport) self.Bind(wx.EVT_MENU, self.on_properties, self.menuProperties) self.Bind(wx.EVT_MENU, self.on_exit, menuExit) self.Bind(wx.EVT_MENU, self.on_pref, self.menuPref) self.Bind(wx.EVT_MENU, self.on_adv_pref, self.menuAdvPref) self.Bind(wx.EVT_MENU, self.on_devices, self.menuDevices) self.Bind(wx.EVT_MENU, self.on_clear_select, self.menuClearSelect) self.Bind(wx.EVT_MENU, self.on_show_measure, self.menuShowMeasure) self.Bind(wx.EVT_MENU, self.on_start, self.menuStart) self.Bind(wx.EVT_MENU, self.on_stop, self.menuStop) self.Bind(wx.EVT_MENU, self.on_stop_end, self.menuStopEnd) self.Bind(wx.EVT_MENU, self.on_compare, self.menuCompare) self.Bind(wx.EVT_MENU, self.on_cal, self.menuCal) self.Bind(wx.EVT_MENU, self.on_about, menuAbout) self.Bind(wx.EVT_MENU, self.on_help, menuHelpLink) self.Bind(wx.EVT_MENU, self.on_update, menuUpdate) idF1 = wx.wx.NewId() self.Bind(wx.EVT_MENU, self.on_help, id=idF1) accelTable = wx.AcceleratorTable([(wx.ACCEL_NORMAL, wx.WXK_F1, idF1)]) self.SetAcceleratorTable(accelTable) def create_popup_menu(self): self.popupMenu = wx.Menu() self.popupMenuStart = self.popupMenu.Append(wx.ID_ANY, "&Start", "Start scan") self.popupMenuStop = self.popupMenu.Append(wx.ID_ANY, "S&top", "Stop scan immediately") self.popupMenuStopEnd = self.popupMenu.Append( wx.ID_ANY, "Stop at &end", "Complete current sweep " "before stopping" ) self.popupMenu.AppendSeparator() self.popupMenuRangeLim = self.popupMenu.Append( wx.ID_ANY, "Set range to current zoom", "Set scanning range to the " "current zoom" ) self.popupMenu.AppendSeparator() self.popupMenuPointsLim = self.popupMenu.Append( wx.ID_ANY, "Limit points", "Limit points to " "increase plot speed", kind=wx.ITEM_CHECK ) self.popupMenuPointsLim.Check(self.settings.pointsLimit) self.popupMenu.AppendSeparator() self.popupmenuClearSelect = self.popupMenu.Append(wx.ID_ANY, "Clear selection", "Clear current selection") self.graph.add_menu_clear_select(self.popupmenuClearSelect) self.popupMenuShowMeasure = self.popupMenu.Append( wx.ID_ANY, "Show &measurements", "Show measurements window", kind=wx.ITEM_CHECK ) self.popupMenuShowMeasure.Check(self.settings.showMeasure) self.Bind(wx.EVT_MENU, self.on_start, self.popupMenuStart) self.Bind(wx.EVT_MENU, self.on_stop, self.popupMenuStop) self.Bind(wx.EVT_MENU, self.on_stop_end, self.popupMenuStopEnd) self.Bind(wx.EVT_MENU, self.on_range_lim, self.popupMenuRangeLim) self.Bind(wx.EVT_MENU, self.on_points_lim, self.popupMenuPointsLim) self.Bind(wx.EVT_MENU, self.on_clear_select, self.popupmenuClearSelect) self.Bind(wx.EVT_MENU, self.on_show_measure, self.popupMenuShowMeasure) self.Bind(wx.EVT_CONTEXT_MENU, self.on_popup_menu) def on_popup_menu(self, event): pos = event.GetPosition() pos = self.ScreenToClient(pos) self.PopupMenu(self.popupMenu, pos) def on_open(self, _event): if self.save_warn(Warn.OPEN): return dlg = wx.FileDialog(self, "Open a scan", self.settings.dirScans, self.filename, File.RFS, wx.OPEN) if dlg.ShowModal() == wx.ID_OK: self.open(dlg.GetDirectory(), dlg.GetFilename()) dlg.Destroy() def on_file_history(self, event): selection = event.GetId() - wx.ID_FILE1 path = self.settings.fileHistory.GetHistoryFile(selection) self.settings.fileHistory.AddFileToHistory(path) dirname, filename = os.path.split(path) self.open(dirname, filename) def on_save(self, _event): dlg = wx.FileDialog( self, "Save a scan", self.settings.dirScans, self.filename, File.RFS, wx.SAVE | wx.OVERWRITE_PROMPT ) if dlg.ShowModal() == wx.ID_OK: self.status.set_general("Saving") self.filename = os.path.splitext(dlg.GetFilename())[0] dirname = dlg.GetDirectory() self.settings.dirScans = dirname save_plot(dirname, dlg.GetFilename(), self.scanInfo, self.spectrum) self.saved(True) self.status.set_general("Finished") self.settings.fileHistory.AddFileToHistory(os.path.join(dirname, dlg.GetFilename())) dlg.Destroy() def on_export(self, _event): dlg = wx.FileDialog( self, "Export a scan", self.settings.dirExport, self.filename, File.get_export_filters(), wx.SAVE | wx.OVERWRITE_PROMPT, ) if dlg.ShowModal() == wx.ID_OK: self.status.set_general("Exporting") dirname = dlg.GetDirectory() self.settings.dirExport = dirname export_plot(dirname, dlg.GetFilename(), dlg.GetFilterIndex(), self.spectrum) self.status.set_general("Finished") dlg.Destroy() def on_properties(self, _event): if len(self.spectrum) > 0: self.scanInfo.timeFirst = min(self.spectrum) self.scanInfo.timeLast = max(self.spectrum) dlg = DialogProperties(self, self.scanInfo) dlg.ShowModal() dlg.Destroy() def on_exit(self, _event): self.Unbind(wx.EVT_CLOSE) if self.save_warn(Warn.EXIT): self.Bind(wx.EVT_CLOSE, self.on_exit) return self.stop_scan() self.wait_background() self.get_controls() self.graph.close() self.settings.dwell = DWELL[1::2][self.choiceDwell.GetSelection()] self.settings.nfft = NFFT[self.choiceNfft.GetSelection()] self.settings.devices = self.devices self.settings.save() self.Close(True) def on_pref(self, _event): self.get_controls() dlg = DialogPrefs(self, self.settings) if dlg.ShowModal() == wx.ID_OK: self.graph.create_plot() self.set_control_state(True) self.set_controls() dlg.Destroy() def on_adv_pref(self, _event): dlg = DialogAdvPrefs(self, self.settings) if dlg.ShowModal() == wx.ID_OK: self.set_control_state(True) dlg.Destroy() def on_devices(self, _event): self.get_controls() self.devices = self.refresh_devices() dlg = DialogDevices(self, self.devices, self.settings) if dlg.ShowModal() == wx.ID_OK: self.devices = dlg.get_devices() self.settings.index = dlg.get_index() self.set_control_state(True) self.set_controls() dlg.Destroy() def on_compare(self, _event): dlg = DialogCompare(self, self.settings.dirScans, self.filename) dlg.ShowModal() dlg.Destroy() def on_clear_select(self, _event): self.graph.clear_selection() def on_show_measure(self, event): show = event.Checked() self.menuShowMeasure.Check(show) self.popupMenuShowMeasure.Check(show) self.settings.showMeasure = show self.graph.show_measureTable(show) self.Layout() def on_cal(self, _event): self.dlgCal = DialogAutoCal(self, self.settings.calFreq, self.auto_cal) self.dlgCal.ShowModal() def on_about(self, _event): dlg = DialogAbout(self) dlg.ShowModal() dlg.Destroy() def on_help(self, _event): webbrowser.open("http://eartoearoak.com/software/rtlsdr-scanner") def on_update(self, _event): if self.threadUpdate is None: self.status.set_general("Checking for updates") self.threadUpdate = Thread(target=self.update_check) self.threadUpdate.start() def on_spin(self, event): control = event.GetEventObject() if control == self.spinCtrlStart: self.spinCtrlStop.SetRange(self.spinCtrlStart.GetValue() + 1, F_MAX) def on_choice(self, _event): self.get_controls() self.graph.create_plot() def on_start(self, _event): if self.settings.start >= self.settings.stop: wx.MessageBox("Stop frequency must be greater that start", "Warning", wx.OK | wx.ICON_WARNING) return self.get_controls() self.graph.clear_plots() self.devices = self.refresh_devices() if len(self.devices) == 0: wx.MessageBox("No devices found", "Error", wx.OK | wx.ICON_ERROR) else: self.spectrum.clear() self.start_scan() def on_stop(self, _event): self.stopScan = True self.stopAtEnd = False self.stop_scan() def on_stop_end(self, _event): self.stopAtEnd = True def on_range_lim(self, _event): xmin, xmax = self.graph.get_axes().get_xlim() xmin = int(xmin) xmax = math.ceil(xmax) if xmax < xmin + 1: xmax = xmin + 1 self.settings.start = xmin self.settings.stop = xmax self.set_controls() def on_points_lim(self, _event): self.settings.pointsLimit = self.popupMenuPointsLim.IsChecked() self.set_plot(self.spectrum, self.settings.annotate) def on_motion(self, event): xpos = event.xdata ypos = event.ydata text = "" if xpos is None or ypos is None or len(self.spectrum) == 0: return if self.settings.display == Display.PLOT: timeStamp = max(self.spectrum) spectrum = self.spectrum[timeStamp] elif self.settings.display == Display.SPECT: timeStamp = num2epoch(ypos) if timeStamp in self.spectrum: spectrum = self.spectrum[timeStamp] else: nearest = min(self.spectrum.keys(), key=lambda k: abs(k - timeStamp)) spectrum = self.spectrum[nearest] else: spectrum = None if spectrum is not None and len(spectrum) > 0: x = min(spectrum.keys(), key=lambda freq: abs(freq - xpos)) if xpos <= max(spectrum.keys(), key=float): y = spectrum[x] text = "f = {0:.6f}MHz, p = {1:.2f}dB".format(x, y) else: text = "f = {0:.6f}MHz".format(xpos) self.status.SetStatusText(text, 1) def on_event(self, event): status = event.data.get_status() freq = event.data.get_freq() data = event.data.get_data() if status == Event.STARTING: self.status.set_general("Starting") elif status == Event.STEPS: self.stepsTotal = (freq + 1) * 2 self.steps = self.stepsTotal self.status.set_progress(0) self.status.show_progress() elif status == Event.CAL: self.auto_cal(Cal.DONE) elif status == Event.INFO: if self.threadScan is not None: self.sdr = self.threadScan.get_sdr() if data is not None: self.devices[self.settings.index].tuner = data self.scanInfo.tuner = data elif status == Event.DATA: self.saved(False) cal = self.devices[self.settings.index].calibration self.pool.apply_async( anaylse_data, (freq, data, cal, self.settings.nfft, self.settings.overlap, self.settings.winFunc), callback=self.on_process_done, ) self.progress() elif status == Event.STOPPED: self.cleanup() self.status.set_general("Stopped") elif status == Event.FINISHED: self.threadScan = None elif status == Event.ERROR: self.cleanup() self.status.set_general("Error: {0}".format(data)) if self.dlgCal is not None: self.dlgCal.Destroy() self.dlgCal = None elif status == Event.PROCESSED: offset = self.settings.devices[self.settings.index].offset if self.settings.alert: alert = self.settings.alertLevel else: alert = None Thread( target=update_spectrum, name="Update", args=( self, self.lock, self.settings.start, self.settings.stop, freq, data, offset, self.spectrum, not self.settings.retainScans, alert, ), ).start() elif status == Event.LEVEL: wx.Bell() elif status == Event.UPDATED: if data and self.settings.liveUpdate: self.set_plot( self.spectrum, self.settings.annotate and self.settings.retainScans and self.settings.mode == Mode.CONTIN, ) self.progress() elif status == Event.DRAW: self.graph.draw() elif status == Event.VER_UPD: self.update_checked(True, freq, data) elif status == Event.VER_NOUPD: self.update_checked(False) elif status == Event.VER_UPDFAIL: self.update_checked(failed=True) wx.YieldIfNeeded() def on_process_done(self, data): timeStamp, freq, scan = data post_event(self, EventThreadStatus(Event.PROCESSED, freq, (timeStamp, scan))) def open(self, dirname, filename): if not os.path.exists(os.path.join(dirname, filename)): wx.MessageBox("File not found", "Error", wx.OK | wx.ICON_ERROR) return self.filename = os.path.splitext(filename)[0] self.settings.dirScans = dirname self.status.set_general("Opening: {0}".format(filename)) self.scanInfo, spectrum = open_plot(dirname, filename) if len(spectrum) > 0: self.spectrum.clear() self.scanInfo.setToSettings(self.settings) self.spectrum = spectrum self.saved(True) self.set_controls() self.set_control_state(True) self.set_plot(spectrum, self.settings.annotate) self.graph.scale_plot(True) self.status.set_general("Finished") self.settings.fileHistory.AddFileToHistory(os.path.join(dirname, filename)) else: self.status.set_general("Open failed") def auto_cal(self, status): freq = self.dlgCal.get_freq() if self.dlgCal is not None: if status == Cal.START: self.spinCtrlStart.SetValue(int(freq)) self.spinCtrlStop.SetValue(math.ceil(freq)) self.oldCal = self.devices[self.settings.index].calibration self.devices[self.settings.index].calibration = 0 self.get_controls() self.spectrum.clear() if not self.start_scan(isCal=True): self.dlgCal.reset_cal() elif status == Cal.DONE: ppm = self.calc_ppm(freq) self.dlgCal.set_cal(ppm) self.set_control_state(True) elif status == Cal.OK: self.devices[self.settings.index].calibration = self.dlgCal.get_cal() self.settings.calFreq = freq self.dlgCal = None elif status == Cal.CANCEL: self.dlgCal = None if len(self.devices) > 0: self.devices[self.settings.index].calibration = self.oldCal def calc_ppm(self, freq): with self.lock: timeStamp = max(self.spectrum) spectrum = self.spectrum[timeStamp].copy() for x, y in spectrum.iteritems(): spectrum[x] = (((x - freq) * (x - freq)) + 1) * y peak = max(spectrum, key=spectrum.get) return ((freq - peak) / freq) * 1e6 def start_scan(self, isCal=False): if self.settings.mode == Mode.SINGLE: if self.save_warn(Warn.SCAN): return False if not self.threadScan: self.set_control_state(False) samples = calc_samples(self.settings.dwell) self.status.set_info("") self.scanInfo.setFromSettings(self.settings) time = datetime.datetime.utcnow().replace(microsecond=0) self.scanInfo.time = time.isoformat() + "Z" self.scanInfo.lat = None self.scanInfo.lon = None self.scanInfo.desc = "" self.stopAtEnd = False self.stopScan = False self.threadScan = ThreadScan(self, self.sdr, self.settings, self.settings.index, samples, isCal) self.filename = "Scan {0:.1f}-{1:.1f}MHz".format(self.settings.start, self.settings.stop) self.graph.set_plot_title() return True def stop_scan(self): if self.threadScan: self.status.set_general("Stopping") self.threadScan.abort() self.threadScan.join() if self.sdr is not None: self.sdr.close() self.set_control_state(True) def progress(self): self.steps -= 1 if self.steps > 0 and not self.stopScan: self.status.set_progress((self.stepsTotal - self.steps) * 100 / self.stepsTotal) self.status.show_progress() self.status.set_general("Scanning") else: self.status.hide_progress() self.set_plot(self.spectrum, self.settings.annotate) if self.stopScan: self.status.set_general("Stopped") self.cleanup() elif self.settings.mode == Mode.SINGLE: self.status.set_general("Finished") self.cleanup() else: if self.settings.mode == Mode.CONTIN: if self.dlgCal is None and not self.stopAtEnd: self.limit_spectrum() self.start_scan() else: self.status.set_general("Stopped") self.cleanup() def cleanup(self): if self.sdr is not None: self.sdr.close() self.sdr = None self.status.hide_progress() self.steps = 0 self.threadScan = None self.set_control_state(True) self.stopAtEnd = False self.stopScan = True def limit_spectrum(self): with self.lock: while len(self.spectrum) >= self.settings.retainMax: timeStamp = min(self.spectrum) del self.spectrum[timeStamp] def saved(self, isSaved): self.isSaved = isSaved title = "RTLSDR Scanner - " + self.filename if not isSaved: title += "*" self.SetTitle(title) def set_plot(self, spectrum, annotate): if len(spectrum) > 0: total = count_points(spectrum) if total > 0: spectrum = sort_spectrum(spectrum) extent = Extent(spectrum) self.graph.set_plot(spectrum, self.settings.pointsLimit, self.settings.pointsMax, extent, annotate) def set_control_state(self, state): hasDevices = len(self.devices) > 0 self.spinCtrlStart.Enable(state) self.spinCtrlStop.Enable(state) self.controlGain.Enable(state) self.choiceMode.Enable(state) self.choiceDwell.Enable(state) self.choiceNfft.Enable(state) self.buttonStart.Enable(state and hasDevices) self.buttonStop.Enable(not state and hasDevices) self.menuOpen.Enable(state) self.menuSave.Enable(state and len(self.spectrum) > 0) self.menuExport.Enable(state and len(self.spectrum) > 0) self.menuStart.Enable(state) self.menuStop.Enable(not state) self.menuPref.Enable(state) self.menuAdvPref.Enable(state) self.menuDevices.Enable(state) self.menuCal.Enable(state) self.popupMenuStop.Enable(not state) self.popupMenuStart.Enable(state) if self.settings.mode == Mode.CONTIN: self.menuStopEnd.Enable(not state) self.popupMenuStopEnd.Enable(not state) else: self.menuStopEnd.Enable(False) self.popupMenuStopEnd.Enable(False) self.popupMenuRangeLim.Enable(state) def set_controls(self): self.spinCtrlStart.SetValue(self.settings.start) self.spinCtrlStop.SetValue(self.settings.stop) self.choiceMode.SetSelection(MODE[1::2].index(self.settings.mode)) dwell = calc_real_dwell(self.settings.dwell) self.choiceDwell.SetSelection(DWELL[1::2].index(dwell)) self.choiceNfft.SetSelection(NFFT.index(self.settings.nfft)) self.choiceDisplay.SetSelection(DISPLAY[1::2].index(self.settings.display)) def set_gain_control(self): grid = self.controlGain.GetContainingSizer() if len(self.devices) > 0: self.controlGain.Destroy() device = self.devices[self.settings.index] if device.isDevice: gains = device.get_gains_str() self.controlGain = wx.Choice(self.toolbar, choices=gains) gain = device.get_closest_gain_str(device.gain) self.controlGain.SetStringSelection(gain) else: self.controlGain = NumCtrl(self.toolbar, integerWidth=3, fractionWidth=1) font = self.controlGain.GetFont() dc = wx.WindowDC(self.controlGain) dc.SetFont(font) size = dc.GetTextExtent("####.#") self.controlGain.SetMinSize((size[0] * 1.2, -1)) self.controlGain.SetValue(device.gain) grid.Add(self.controlGain, pos=(1, 7), flag=wx.ALIGN_CENTER) grid.Layout() def get_controls(self): self.settings.start = self.spinCtrlStart.GetValue() self.settings.stop = self.spinCtrlStop.GetValue() self.settings.mode = MODE[1::2][self.choiceMode.GetSelection()] self.settings.dwell = DWELL[1::2][self.choiceDwell.GetSelection()] self.settings.nfft = NFFT[self.choiceNfft.GetSelection()] self.settings.display = DISPLAY[1::2][self.choiceDisplay.GetSelection()] if len(self.devices) > 0: device = self.devices[self.settings.index] try: if device.isDevice: device.gain = float(self.controlGain.GetStringSelection()) else: device.gain = self.controlGain.GetValue() except ValueError: device.gain = 0 def save_warn(self, warnType): if self.settings.saveWarn and not self.isSaved: dlg = DialogSaveWarn(self, warnType) code = dlg.ShowModal() if code == wx.ID_YES: self.on_save(None) if self.isSaved: return False else: return True elif code == wx.ID_NO: return False else: return True return False def update_check(self): local = get_version_timestamp(True) try: remote = get_version_timestamp_repo() except IOError: post_event(self, EventThreadStatus(Event.VER_UPDFAIL)) return if remote > local: post_event(self, EventThreadStatus(Event.VER_UPD, local, remote)) else: post_event(self, EventThreadStatus(Event.VER_NOUPD)) def update_checked(self, updateFound=False, local=None, remote=None, failed=False): self.threadUpdate = None self.status.set_general("") if failed: icon = wx.ICON_ERROR message = "Update check failed" else: icon = wx.ICON_INFORMATION if updateFound: message = "Update found\n\n" message += "Local: " + time.strftime("%c", time.localtime(local)) message += "\nRemote: " + time.strftime("%c", time.localtime(remote)) else: message = "No updates found" dlg = wx.MessageDialog(self, message, "Update", wx.OK | icon) dlg.ShowModal() dlg.Destroy() def refresh_devices(self): self.settings.devices = get_devices(self.devices, self.status) if self.settings.index > len(self.devices) - 1: self.settings.index = 0 self.settings.save() return self.settings.devices def wait_background(self): self.Disconnect(-1, -1, EVT_THREAD_STATUS, self.on_event) if self.threadScan: self.threadScan.abort() self.threadScan.join() self.threadScan = None self.pool.close() self.pool.join()