def decoder_update(self): # Only allow {0, 1} signaltype = self.ui.combobox_signals.currentIndex() inpt_txt = self.ui.inpt.text() if signaltype == 0: if inpt_txt.count("0") + inpt_txt.count("1") < len(inpt_txt): self.ui.inpt.setText(self.old_inpt_txt) else: self.old_inpt_txt = inpt_txt # Write decoded bits bit = self.e.str2bit(self.ui.inpt.text()) decoded = self.e.bit2str(self.e.decode(bit)) errors = "[Decoding Errors = " + str(self.e.analyze(bit)[0]) + "]" self.ui.decoding_errors_label.setText(errors) self.ui.output.setText(decoded) if len(decoded) > 0: if signaltype == 0: temp_signal = SignalSceneManager.create_rectangle(inpt_txt)[0] self.ui.graphicsView_signal.setScene(temp_signal) self.ui.graphicsView_signal.update() temp_decoded = SignalSceneManager.create_rectangle(decoded)[0] self.ui.graphicsView_decoded.setScene(temp_decoded) self.ui.graphicsView_decoded.update()
def __init__(self, project_manager, modulated_data, parent=None, testing_mode=False): super().__init__(project_manager, is_tx=True, parent=parent, testing_mode=testing_mode) self.graphics_view = self.ui.graphicsViewSend self.ui.stackedWidget.setCurrentWidget(self.ui.page_send) self.hide_receive_ui_items() self.ui.btnStart.setIcon(QIcon.fromTheme("media-playback-start")) self.setWindowTitle("Send Signal") self.setWindowIcon(QIcon.fromTheme("media-playback-start")) self.ui.btnStart.setToolTip("Send data") self.ui.btnStop.setToolTip("Stop sending") self.device_is_sending = False if modulated_data is not None: # modulated_data is none in continuous send mode self.ui.progressBar.setMaximum(len(modulated_data)) samp_rate = self.ui.spinBoxSampleRate.value() signal = Signal.from_samples(modulated_data, "Modulated Preview", samp_rate) self.scene_manager = SignalSceneManager(signal, parent=self) self.send_indicator = self.scene_manager.scene.addRect(0, -2, 0, 4, QPen(QColor(Qt.transparent), Qt.FlatCap), QBrush(constants.SEND_INDICATOR_COLOR)) self.send_indicator.stackBefore(self.scene_manager.scene.selection_area) self.scene_manager.init_scene() self.graphics_view.set_signal(signal) self.graphics_view.sample_rate = samp_rate self.init_device() self.graphics_view.setScene(self.scene_manager.scene) self.graphics_view.scene_manager = self.scene_manager self.create_connects()
def decoder_update(self): # Only allow {0, 1} signaltype = self.ui.combobox_signals.currentIndex() inpt_txt = self.ui.inpt.text() if signaltype == 0: if inpt_txt.count("0") + inpt_txt.count("1") < len(inpt_txt): self.ui.inpt.setText(self.old_inpt_txt) else: self.old_inpt_txt = inpt_txt # Write decoded bits bit = self.e.str2bit(self.ui.inpt.text()) decoded = self.e.bit2str(self.e.decode(bit)) errors = "[Decoding Errors = " + str(self.e.analyze(bit)[0]) + "]" self.ui.decoding_errors_label.setText(errors) self.ui.output.setText(decoded) if len(decoded) > 0: if signaltype == 0: temp_signal = SignalSceneManager.create_rectangle(inpt_txt)[0] self.ui.graphicsView_signal.setScene(temp_signal) self.ui.graphicsView_signal.update() temp_decoded = SignalSceneManager.create_rectangle(decoded)[0] self.ui.graphicsView_decoded.setScene(temp_decoded) self.ui.graphicsView_decoded.update()
def __init__(self, freq, samp_rate, bw, gain, device: str, modulated_data, parent=None, testing_mode=False): self.is_tx = True super().__init__(freq, samp_rate, bw, gain, device, parent=parent, testing_mode=testing_mode) self.update_interval = 25 self.graphics_view = self.ui.graphicsViewSend self.ui.stackedWidget.setCurrentIndex(1) self.hide_receive_ui_items() self.ui.btnStart.setIcon(QIcon.fromTheme("media-playback-start")) self.setWindowTitle("Send signal") self.ui.btnStart.setToolTip("Send data") self.ui.btnStop.setToolTip("Stop sending") self.ui.progressBar.setMaximum(len(modulated_data)) self.device_is_sending = False signal = Signal.from_samples(modulated_data, "Modulated Preview", samp_rate) self.scene_manager = SignalSceneManager(signal, parent=self) self.send_indicator = self.scene_manager.scene.addRect( 0, -2, 0, 4, QPen(QColor(Qt.transparent), Qt.FlatCap), QBrush(constants.SEND_INDICATOR_COLOR)) self.send_indicator.stackBefore( self.scene_manager.scene.selection_area) self.scene_manager.init_scene() self.graphics_view.set_signal(signal) self.graphics_view.sample_rate = samp_rate self.init_device() self.graphics_view.setScene(self.scene_manager.scene) self.graphics_view.scene_manager = self.scene_manager self.create_connects()
def draw_signal(self, signal, proto_analyzer): if signal is None: return self.horizontalScrollBar().blockSignals(True) self.scene_creator = SignalSceneManager(signal, self) self.scene_creator.init_scene() self.setScene(self.scene_creator.scene) self.scene_creator.show_full_scene() self.fitInView(self.sceneRect()) self.signal_loaded.emit(proto_analyzer) self.signal = signal self.proto_analyzer = proto_analyzer self.horizontalScrollBar().blockSignals(False)
def draw_signal(self, signal, proto_analyzer): if signal is None: return self.signal = signal # type: Signal self.proto_analyzer = proto_analyzer # type: ProtocolAnalyzer self.scene_manager = SignalSceneManager(signal, self) self.plot_data(self.signal.real_plot_data) self.show_full_scene() self.auto_fit_view() self.signal_loaded.emit(self.proto_analyzer)
class SendDialogController(SendRecvDialogController): def __init__(self, freq, samp_rate, bw, gain, device: str, modulated_data, parent=None, testing_mode=False): self.is_tx = True super().__init__(freq, samp_rate, bw, gain, device, parent=parent, testing_mode=testing_mode) self.update_interval = 25 self.graphics_view = self.ui.graphicsViewSend self.ui.stackedWidget.setCurrentIndex(1) self.hide_receive_ui_items() self.ui.btnStart.setIcon(QIcon.fromTheme("media-playback-start")) self.setWindowTitle("Send signal") self.ui.btnStart.setToolTip("Send data") self.ui.btnStop.setToolTip("Stop sending") self.ui.progressBar.setMaximum(len(modulated_data)) self.device_is_sending = False signal = Signal.from_samples(modulated_data, "Modulated Preview", samp_rate) self.scene_manager = SignalSceneManager(signal, parent=self) self.send_indicator = self.scene_manager.scene.addRect( 0, -2, 0, 4, QPen(QColor(Qt.transparent), Qt.FlatCap), QBrush(constants.SEND_INDICATOR_COLOR)) self.send_indicator.stackBefore( self.scene_manager.scene.selection_area) self.scene_manager.init_scene() self.graphics_view.set_signal(signal) self.graphics_view.sample_rate = samp_rate self.init_device() self.graphics_view.setScene(self.scene_manager.scene) self.graphics_view.scene_manager = self.scene_manager self.create_connects() def create_connects(self): super().create_connects() self.graphics_view.save_as_clicked.connect( self.on_graphics_view_save_as_clicked) self.scene_manager.signal.data_edited.connect( self.on_signal_data_edited) self.ui.spinBoxNRepeat.editingFinished.connect( self.on_num_repeats_changed) def __update_send_indicator(self, width: int): y, h = self.ui.graphicsViewSend.view_rect().y( ), self.ui.graphicsViewSend.view_rect().height() self.send_indicator.setRect(0, y - h, width, 2 * h + abs(y)) def update_view(self): if super().update_view(): self.__update_send_indicator(self.device.current_index) if not self.device.sending_finished: self.ui.lblCurrentRepeatValue.setText( str(self.device.current_iteration + 1)) else: self.ui.lblCurrentRepeatValue.setText("Done") def init_device(self): device_name = self.ui.cbDevice.currentText() num_repeats = self.ui.spinBoxNRepeat.value() sts = self.scene_manager.signal._fulldata self.device = VirtualDevice(self.backend_handler, device_name, Mode.send, bw=1e6, freq=433.92e6, gain=40, samp_rate=1e6, samples_to_send=sts, device_ip="192.168.10.2", sending_repeats=num_repeats, parent=self) self._create_device_connects() @pyqtSlot() def on_graphics_view_save_as_clicked(self): filename = FileOperator.get_save_file_name("signal.complex") if filename: try: self.scene_manager.signal.save_as(filename) except Exception as e: QMessageBox.critical(self, self.tr("Error saving signal"), e.args[0]) @pyqtSlot() def on_signal_data_edited(self): signal = self.scene_manager.signal self.ui.progressBar.setMaximum(signal.num_samples) self.device.samples_to_send = signal.data self.scene_manager.init_scene() self.ui.graphicsViewSend.redraw_view() @pyqtSlot() def on_start_clicked(self): super().on_start_clicked() if self.ui.progressBar.value() >= self.ui.progressBar.maximum() - 1: self.on_clear_clicked() if self.device_is_sending: self.device.stop("Sending paused by user") else: self.device.start() @pyqtSlot() def on_stop_clicked(self): self.device.stop("Stopped receiving: Stop button clicked") super().on_stop_clicked() self.on_clear_clicked() @pyqtSlot() def on_device_stopped(self): super().on_device_stopped() self.ui.btnStart.setIcon(QIcon.fromTheme("media-playback-start")) self.ui.btnStart.setToolTip("Start sending") self.device_is_sending = False @pyqtSlot() def on_device_started(self): super().on_device_started() self.device_is_sending = True self.ui.btnStart.setEnabled(True) self.ui.btnStart.setIcon(QIcon.fromTheme("media-playback-pause")) self.set_device_ui_items_enabled(False) @pyqtSlot() def on_num_repeats_changed(self): self.device.num_sending_repeats = self.ui.spinBoxNRepeat.value() @pyqtSlot() def on_clear_clicked(self): self.__update_send_indicator(0) self.reset()
def __init__(self, proto_analyzer: ProtocolAnalyzer, undo_stack: QUndoStack, project_manager, proto_bits=None, parent=None): super().__init__(parent) self.undo_stack = undo_stack self.ui = Ui_SignalFrame() self.ui.setupUi(self) fixed_font = QFontDatabase.systemFont(QFontDatabase.FixedFont) fixed_font.setPointSize(QApplication.instance().font().pointSize()) self.ui.txtEdProto.setFont(fixed_font) self.ui.txtEdProto.participants = project_manager.participants self.ui.txtEdProto.messages = proto_analyzer.messages self.ui.gvSignal.participants = project_manager.participants self.setAttribute(Qt.WA_DeleteOnClose) self.project_manager = project_manager self.proto_analyzer = proto_analyzer self.signal = proto_analyzer.signal if self.proto_analyzer is not None else None # type: Signal self.dsp_filter = Filter([0.1] * 10, FilterType.moving_average) self.set_filter_button_caption() self.filter_dialog = FilterDialogController(self.dsp_filter, parent=self) self.proto_selection_timer = QTimer() # For Update Proto Selection from ROI self.proto_selection_timer.setSingleShot(True) self.proto_selection_timer.setInterval(1) # Disabled because never used (see also set_protocol_visibilty()) self.ui.chkBoxSyncSelection.hide() if self.signal is not None: self.filter_menu = QMenu() self.apply_filter_to_selection_only = self.filter_menu.addAction(self.tr("Apply only to selection")) self.apply_filter_to_selection_only.setCheckable(True) self.apply_filter_to_selection_only.setChecked(False) self.configure_filter_action = self.filter_menu.addAction("Configure filter...") self.configure_filter_action.setIcon(QIcon.fromTheme("configure")) self.configure_filter_action.triggered.connect(self.on_configure_filter_action_triggered) self.ui.btnFilter.setMenu(self.filter_menu) if self.signal.qad_demod_file_loaded: self.ui.lSignalTyp.setText("Quad-Demod Signal (*.wav)") elif self.signal.wav_mode: self.ui.lSignalTyp.setText("Realpart Signal (*.wav)") else: self.ui.lSignalTyp.setText("Complex Signal") self.ui.gvLegend.hide() self.ui.lineEditSignalName.setText(self.signal.name) self.ui.lSamplesInView.setText("{0:,}".format(self.signal.num_samples)) self.ui.lSamplesTotal.setText("{0:,}".format(self.signal.num_samples)) self.sync_protocol = self.ui.chkBoxSyncSelection.isChecked() self.ui.chkBoxSyncSelection.hide() self.ui.splitter.setSizes([self.ui.splitter.height(), 0]) self.protocol_selection_is_updateable = True self.scene_manager = SignalSceneManager(self.signal, self) self.ui.gvSignal.scene_manager = self.scene_manager self.ui.gvSignal.setScene(self.scene_manager.scene) self.jump_sync = True self.on_btn_show_hide_start_end_clicked() self.refresh_signal_information(block=True) self.create_connects() self.set_protocol_visibility() self.ui.chkBoxShowProtocol.setChecked(True) self.set_qad_tooltip(self.signal.noise_threshold) self.ui.btnSaveSignal.hide() self.show_protocol(refresh=False) else: self.ui.btnFilter.setDisabled(True) suffix = "" if not proto_analyzer.filename: suffix = "" elif proto_analyzer.filename.endswith(".proto"): suffix = " (*.proto)" elif proto_analyzer.filename.endswith(".txt"): suffix = " (*.txt)" self.ui.lSignalTyp.setText("Protocol"+suffix) scene, nsamples = SignalSceneManager.create_rectangle(proto_bits) self.ui.lSamplesInView.setText("{0:n}".format(int(nsamples))) self.ui.lSamplesTotal.setText("{0:n}".format(int(nsamples))) self.ui.gvSignal.setScene(scene) self.ui.spinBoxSelectionStart.setMaximum(nsamples) self.ui.spinBoxSelectionEnd.setMaximum(nsamples) self.ui.btnReplay.hide() self.create_connects() self.ui.gvSignal.sel_area_active = True self.ui.btnSaveSignal.hide()
class SignalFrameController(QFrame): closed = pyqtSignal(QWidget) signal_created = pyqtSignal(Signal) drag_started = pyqtSignal(QPoint) frame_dropped = pyqtSignal(QPoint) files_dropped = pyqtSignal(list) not_show_again_changed = pyqtSignal() signal_drawing_finished = pyqtSignal() apply_to_all_clicked = pyqtSignal(Signal) sort_action_clicked = pyqtSignal() @property def proto_view(self): return self.ui.txtEdProto.cur_view def __init__(self, proto_analyzer: ProtocolAnalyzer, undo_stack: QUndoStack, project_manager, proto_bits=None, parent=None): super().__init__(parent) self.undo_stack = undo_stack self.ui = Ui_SignalFrame() self.ui.setupUi(self) fixed_font = QFontDatabase.systemFont(QFontDatabase.FixedFont) fixed_font.setPointSize(QApplication.instance().font().pointSize()) self.ui.txtEdProto.setFont(fixed_font) self.ui.txtEdProto.participants = project_manager.participants self.ui.txtEdProto.messages = proto_analyzer.messages self.ui.gvSignal.participants = project_manager.participants self.setAttribute(Qt.WA_DeleteOnClose) self.project_manager = project_manager self.proto_analyzer = proto_analyzer self.signal = proto_analyzer.signal if self.proto_analyzer is not None else None # type: Signal self.dsp_filter = Filter([0.1] * 10, FilterType.moving_average) self.set_filter_button_caption() self.filter_dialog = FilterDialogController(self.dsp_filter, parent=self) self.proto_selection_timer = QTimer() # For Update Proto Selection from ROI self.proto_selection_timer.setSingleShot(True) self.proto_selection_timer.setInterval(1) # Disabled because never used (see also set_protocol_visibilty()) self.ui.chkBoxSyncSelection.hide() if self.signal is not None: self.filter_menu = QMenu() self.apply_filter_to_selection_only = self.filter_menu.addAction(self.tr("Apply only to selection")) self.apply_filter_to_selection_only.setCheckable(True) self.apply_filter_to_selection_only.setChecked(False) self.configure_filter_action = self.filter_menu.addAction("Configure filter...") self.configure_filter_action.setIcon(QIcon.fromTheme("configure")) self.configure_filter_action.triggered.connect(self.on_configure_filter_action_triggered) self.ui.btnFilter.setMenu(self.filter_menu) if self.signal.qad_demod_file_loaded: self.ui.lSignalTyp.setText("Quad-Demod Signal (*.wav)") elif self.signal.wav_mode: self.ui.lSignalTyp.setText("Realpart Signal (*.wav)") else: self.ui.lSignalTyp.setText("Complex Signal") self.ui.gvLegend.hide() self.ui.lineEditSignalName.setText(self.signal.name) self.ui.lSamplesInView.setText("{0:,}".format(self.signal.num_samples)) self.ui.lSamplesTotal.setText("{0:,}".format(self.signal.num_samples)) self.sync_protocol = self.ui.chkBoxSyncSelection.isChecked() self.ui.chkBoxSyncSelection.hide() self.ui.splitter.setSizes([self.ui.splitter.height(), 0]) self.protocol_selection_is_updateable = True self.scene_manager = SignalSceneManager(self.signal, self) self.ui.gvSignal.scene_manager = self.scene_manager self.ui.gvSignal.setScene(self.scene_manager.scene) self.jump_sync = True self.on_btn_show_hide_start_end_clicked() self.refresh_signal_information(block=True) self.create_connects() self.set_protocol_visibility() self.ui.chkBoxShowProtocol.setChecked(True) self.set_qad_tooltip(self.signal.noise_threshold) self.ui.btnSaveSignal.hide() self.show_protocol(refresh=False) else: self.ui.btnFilter.setDisabled(True) suffix = "" if not proto_analyzer.filename: suffix = "" elif proto_analyzer.filename.endswith(".proto"): suffix = " (*.proto)" elif proto_analyzer.filename.endswith(".txt"): suffix = " (*.txt)" self.ui.lSignalTyp.setText("Protocol"+suffix) scene, nsamples = SignalSceneManager.create_rectangle(proto_bits) self.ui.lSamplesInView.setText("{0:n}".format(int(nsamples))) self.ui.lSamplesTotal.setText("{0:n}".format(int(nsamples))) self.ui.gvSignal.setScene(scene) self.ui.spinBoxSelectionStart.setMaximum(nsamples) self.ui.spinBoxSelectionEnd.setMaximum(nsamples) self.ui.btnReplay.hide() self.create_connects() self.ui.gvSignal.sel_area_active = True self.ui.btnSaveSignal.hide() def create_connects(self): self.ui.btnCloseSignal.clicked.connect(self.on_btn_close_signal_clicked) self.ui.btnReplay.clicked.connect(self.on_btn_replay_clicked) self.ui.btnAutoDetect.clicked.connect(self.on_btn_autodetect_clicked) self.ui.btnInfo.clicked.connect(self.on_info_btn_clicked) self.ui.btnShowHideStartEnd.clicked.connect(self.on_btn_show_hide_start_end_clicked) self.filter_dialog.filter_accepted.connect(self.on_filter_dialog_filter_accepted) if self.signal is not None: self.ui.gvSignal.save_clicked.connect(self.save_signal) self.signal.bit_len_changed.connect(self.ui.spinBoxInfoLen.setValue) self.signal.qad_center_changed.connect(self.on_signal_qad_center_changed) self.signal.noise_threshold_changed.connect(self.on_noise_threshold_changed) self.signal.modulation_type_changed.connect(self.show_modulation_type) self.signal.tolerance_changed.connect(self.ui.spinBoxTolerance.setValue) self.signal.protocol_needs_update.connect(self.refresh_protocol) self.signal.data_edited.connect(self.refresh_signal) # Crop/Delete Mute etc. self.signal.sample_rate_changed.connect(self.__set_duration) self.signal.sample_rate_changed.connect(self.show_protocol) # Update times self.signal.saved_status_changed.connect(self.on_signal_data_changed_before_save) self.ui.btnSaveSignal.clicked.connect(self.save_signal) self.signal.name_changed.connect(self.ui.lineEditSignalName.setText) self.ui.gvLegend.resized.connect(self.on_gv_legend_resized) self.ui.gvSignal.sel_area_width_changed.connect(self.start_proto_selection_timer) self.ui.gvSignal.sel_area_start_end_changed.connect(self.start_proto_selection_timer) self.proto_selection_timer.timeout.connect(self.update_protocol_selection_from_roi) self.ui.lineEditSignalName.editingFinished.connect(self.change_signal_name) self.proto_analyzer.qt_signals.protocol_updated.connect(self.on_protocol_updated) self.ui.btnFilter.clicked.connect(self.on_btn_filter_clicked) self.ui.gvSignal.set_noise_clicked.connect(self.on_set_noise_in_graphic_view_clicked) self.ui.gvSignal.save_as_clicked.connect(self.save_signal_as) self.ui.gvSignal.create_clicked.connect(self.create_new_signal) self.ui.gvSignal.zoomed.connect(self.on_signal_zoomed) self.ui.gvSignal.sel_area_start_end_changed.connect(self.update_selection_area) self.ui.gvSignal.sep_area_changed.connect(self.set_qad_center) self.ui.gvSignal.sep_area_moving.connect(self.update_legend) self.ui.sliderYScale.valueChanged.connect(self.handle_slideryscale_value_changed) self.ui.spinBoxXZoom.valueChanged.connect(self.on_spinbox_x_zoom_value_changed) self.project_manager.project_updated.connect(self.on_participant_changed) self.ui.txtEdProto.participant_changed.connect(self.on_participant_changed) self.ui.gvSignal.participant_changed.connect(self.on_participant_changed) self.proto_selection_timer.timeout.connect(self.update_number_selected_samples) self.ui.cbSignalView.currentIndexChanged.connect(self.on_cb_signal_view_index_changed) self.ui.cbModulationType.currentIndexChanged.connect(self.on_combobox_modulation_type_index_changed) self.ui.cbProtoView.currentIndexChanged.connect(self.on_combo_box_proto_view_index_changed) self.ui.chkBoxShowProtocol.stateChanged.connect(self.set_protocol_visibility) self.ui.chkBoxSyncSelection.stateChanged.connect(self.handle_protocol_sync_changed) self.ui.txtEdProto.proto_view_changed.connect(self.show_protocol) self.ui.txtEdProto.show_proto_clicked.connect(self.update_roi_from_protocol_selection) self.ui.txtEdProto.show_proto_clicked.connect(self.zoom_to_roi) self.ui.txtEdProto.selectionChanged.connect(self.update_roi_from_protocol_selection) self.ui.txtEdProto.deletion_wanted.connect(self.ui.gvSignal.on_delete_action_triggered) self.ui.spinBoxSelectionStart.valueChanged.connect(self.on_spinbox_selection_start_value_changed) self.ui.spinBoxSelectionEnd.valueChanged.connect(self.on_spinbox_selection_end_value_changed) self.ui.spinBoxCenterOffset.editingFinished.connect(self.on_spinbox_center_editing_finished) self.ui.spinBoxTolerance.editingFinished.connect(self.on_spinbox_tolerance_editing_finished) self.ui.spinBoxNoiseTreshold.editingFinished.connect(self.on_spinbox_noise_threshold_editing_finished) self.ui.spinBoxInfoLen.editingFinished.connect(self.on_spinbox_infolen_editing_finished) def refresh_signal_information(self, block=True): self.ui.spinBoxTolerance.blockSignals(block) self.ui.spinBoxCenterOffset.blockSignals(block) self.ui.spinBoxInfoLen.blockSignals(block) self.ui.spinBoxNoiseTreshold.blockSignals(block) self.ui.btnAutoDetect.blockSignals(block) self.ui.spinBoxTolerance.setValue(self.signal.tolerance) self.ui.spinBoxCenterOffset.setValue(self.signal.qad_center) self.ui.spinBoxInfoLen.setValue(self.signal.bit_len) self.ui.spinBoxNoiseTreshold.setValue(self.signal.noise_threshold) self.ui.btnAutoDetect.setChecked(self.signal.auto_detect_on_modulation_changed) self.show_modulation_type() self.ui.spinBoxTolerance.blockSignals(False) self.ui.spinBoxCenterOffset.blockSignals(False) self.ui.spinBoxInfoLen.blockSignals(False) self.ui.spinBoxNoiseTreshold.blockSignals(False) self.ui.btnAutoDetect.blockSignals(False) def set_empty_frame_visibilities(self): self.ui.lInfoLenText.hide() self.ui.spinBoxInfoLen.hide() self.ui.spinBoxCenterOffset.hide() self.ui.spinBoxTolerance.hide() self.ui.chkBoxShowProtocol.hide() self.ui.cbProtoView.hide() self.ui.lErrorTolerance.hide() self.ui.lSignalViewText.hide() self.ui.chkBoxSyncSelection.hide() self.ui.txtEdProto.hide() self.ui.gvLegend.hide() self.ui.cbSignalView.hide() self.ui.cbModulationType.hide() self.ui.btnSaveSignal.hide() def update_number_selected_samples(self): self.ui.lNumSelectedSamples.setText(str(abs(int(self.ui.gvSignal.selection_area.width)))) self.__set_duration() try: sel_messages = self.ui.gvSignal.selected_messages except AttributeError: sel_messages = [] if len(sel_messages) == 1: self.ui.labelRSSI.setText("RSSI: {}".format(Formatter.big_value_with_suffix(sel_messages[0].rssi))) else: self.ui.labelRSSI.setText("") def change_signal_name(self): self.signal.name = self.ui.lineEditSignalName.text() def __set_duration(self): # On Signal Sample Rate changed try: num_samples = int(self.ui.lNumSelectedSamples.text()) except ValueError: return if self.signal: t = num_samples / self.signal.sample_rate self.ui.lDuration.setText(Formatter.science_time(t)) def handle_slideryscale_value_changed(self): try: gvs = self.ui.gvSignal yscale = self.ui.sliderYScale.value() current_factor = gvs.sceneRect().height() / gvs.view_rect().height() gvs.scale(1, yscale / current_factor) x, w = self.ui.gvSignal.view_rect().x(), self.ui.gvSignal.view_rect().width() gvs.centerOn(x + w / 2, gvs.y_center) except ZeroDivisionError: pass def mousePressEvent(self, event): if event.button() == Qt.LeftButton: self.drag_started.emit(self.mapToParent(event.pos())) drag = QDrag(self) mimeData = QMimeData() mimeData.setText("Move Signal") pixmap = QPixmap(self.rect().size()) self.render(pixmap, QPoint(), QRegion(self.rect())) drag.setPixmap(pixmap) drag.setMimeData(mimeData) drag.exec_() def set_filter_button_caption(self): self.ui.btnFilter.setText("Filter ({0})".format(self.dsp_filter.filter_type.value)) def dragMoveEvent(self, event): event.accept() def dragEnterEvent(self, event): event.acceptProposedAction() def dropEvent(self, event: QDropEvent): if len(event.mimeData().urls()) == 0: self.frame_dropped.emit(self.mapToParent(event.pos())) else: self.files_dropped.emit(event.mimeData().urls()) def create_new_signal(self, start, end): if start != end: new_signal = self.signal.create_new(start, end) self.signal_created.emit(new_signal) else: Errors.empty_selection() def my_close(self): settings = constants.SETTINGS not_show = settings.value('not_show_close_dialog', False, type=bool) if not not_show: cb = QCheckBox("Do not show this again.") msgbox = QMessageBox(QMessageBox.Question, "Confirm close", "Are you sure you want to close?") msgbox.addButton(QMessageBox.Yes) msgbox.addButton(QMessageBox.No) msgbox.setDefaultButton(QMessageBox.No) msgbox.setCheckBox(cb) reply = msgbox.exec() not_show_again = bool(cb.isChecked()) settings.setValue("not_show_close_dialog", not_show_again) self.not_show_again_changed.emit() if reply != QMessageBox.Yes: return self.closed.emit(self) def save_signal(self): if len(self.signal.filename) > 0: self.signal.save() else: self.save_signal_as() def save_signal_as(self): filename = FileOperator.get_save_file_name(self.signal.filename, wav_only=self.signal.wav_mode) if filename: try: self.signal.save_as(filename) except Exception as e: QMessageBox.critical(self, self.tr("Error saving signal"), e.args[0]) def draw_signal(self, full_signal=False): gv_legend = self.ui.gvLegend gv_legend.ysep = -self.signal.qad_center self.scene_manager.scene_type = self.ui.cbSignalView.currentIndex() self.scene_manager.init_scene() if full_signal: self.ui.gvSignal.show_full_scene() else: self.ui.gvSignal.redraw_view() legend = LegendScene() legend.setBackgroundBrush(constants.BGCOLOR) legend.setSceneRect(0, self.scene_manager.scene.sceneRect().y(), gv_legend.width(), self.scene_manager.scene.sceneRect().height()) legend.draw_one_zero_arrows(-self.signal.qad_center) gv_legend.setScene(legend) num_samples = self.signal.num_samples self.ui.spinBoxSelectionStart.setMaximum(num_samples) self.ui.spinBoxSelectionEnd.setMaximum(num_samples) self.ui.gvSignal.nsamples = num_samples self.ui.gvSignal.sel_area_active = True self.ui.gvSignal.y_sep = -self.signal.qad_center def restore_protocol_selection(self, sel_start, sel_end, start_message, end_message, old_protoview): if old_protoview == self.proto_view: return self.protocol_selection_is_updateable = False sel_start = int(self.proto_analyzer.convert_index(sel_start, old_protoview, self.proto_view, True)[0]) sel_end = int(math.ceil(self.proto_analyzer.convert_index(sel_end, old_protoview, self.proto_view, True)[1])) c = self.ui.txtEdProto.textCursor() c.setPosition(0) cur_message = 0 i = 0 text = self.ui.txtEdProto.toPlainText() while cur_message < start_message: if text[i] == "\n": cur_message += 1 i += 1 c.movePosition(QTextCursor.Right, QTextCursor.MoveAnchor, i) c.movePosition(QTextCursor.Right, QTextCursor.MoveAnchor, sel_start) text = text[i:] i = 0 while cur_message < end_message: if text[i] == "\n": cur_message += 1 i += 1 c.movePosition(QTextCursor.Right, QTextCursor.KeepAnchor, i) c.movePosition(QTextCursor.Right, QTextCursor.KeepAnchor, sel_end) self.ui.txtEdProto.setTextCursor(c) self.protocol_selection_is_updateable = True def update_protocol(self): self.ui.txtEdProto.setEnabled(False) self.proto_analyzer.get_protocol_from_signal() def show_protocol(self, old_view=-1, refresh=False): if not self.proto_analyzer: return if not self.ui.chkBoxShowProtocol.isChecked(): return if old_view == -1: old_view = self.ui.cbProtoView.currentIndex() if self.proto_analyzer.messages is None or refresh: self.update_protocol() else: # Keep things synchronized and restore selection self.ui.txtEdProto.blockSignals(True) self.ui.cbProtoView.blockSignals(True) self.ui.cbProtoView.setCurrentIndex(self.proto_view) self.ui.cbProtoView.blockSignals(False) start_message = 0 sel_start = self.ui.txtEdProto.textCursor().selectionStart() text = self.ui.txtEdProto.toPlainText()[:sel_start] sel_start = 0 read_pause = False for t in text: if t == "\t": read_pause = True if not read_pause: sel_start += 1 if t == "\n": sel_start = 0 start_message += 1 read_pause = False sel_end = self.ui.txtEdProto.textCursor().selectionEnd() text = self.ui.txtEdProto.toPlainText()[self.ui.txtEdProto.textCursor().selectionStart():sel_end] end_message = 0 sel_end = 0 read_pause = False for t in text: if t == "\t": read_pause = True if not read_pause: sel_end += 1 if t == "\n": sel_end = 0 end_message += 1 read_pause = False self.ui.txtEdProto.setHtml(self.proto_analyzer.plain_to_html(self.proto_view)) try: self.restore_protocol_selection(sel_start, sel_end, start_message, end_message, old_view) except TypeError: # Without try/except: segfault (TypeError) when changing sample_rate in info dialog of signal pass self.ui.txtEdProto.blockSignals(False) def eliminate(self): self.proto_selection_timer.stop() self.ui.verticalLayout.removeItem(self.ui.additionalInfos) if self.signal is not None: # Avoid memory leaks self.scene_manager.eliminate() self.signal.eliminate() self.proto_analyzer.eliminate() self.ui.gvSignal.scene_manager.eliminate() self.ui.gvLegend.eliminate() self.ui.gvSignal.eliminate() self.scene_manager = None self.signal = None self.proto_analyzer = None self.ui.layoutWidget.setParent(None) self.ui.layoutWidget.deleteLater() self.setParent(None) self.deleteLater() @pyqtSlot() def on_signal_zoomed(self): gvs = self.ui.gvSignal self.ui.lSamplesInView.setText("{0:n}".format(int(gvs.view_rect().width()))) self.ui.spinBoxXZoom.blockSignals(True) self.ui.spinBoxXZoom.setValue(int(gvs.sceneRect().width() / gvs.view_rect().width() * 100)) self.ui.spinBoxXZoom.blockSignals(False) @pyqtSlot(int) def on_spinbox_x_zoom_value_changed(self, value: int): gvs = self.ui.gvSignal zoom_factor = value / 100 current_factor = gvs.sceneRect().width() / gvs.view_rect().width() gvs.zoom(zoom_factor / current_factor) @pyqtSlot() def on_btn_close_signal_clicked(self): self.my_close() @pyqtSlot() def on_set_noise_in_graphic_view_clicked(self): self.setCursor(Qt.WaitCursor) start = self.ui.gvSignal.selection_area.x end = start + self.ui.gvSignal.selection_area.width new_thresh = self.signal.calc_noise_threshold(start, end) self.ui.spinBoxNoiseTreshold.setValue(new_thresh) self.ui.spinBoxNoiseTreshold.editingFinished.emit() self.unsetCursor() @pyqtSlot() def on_noise_threshold_changed(self): self.ui.spinBoxNoiseTreshold.setValue(self.signal.noise_threshold) minimum = self.signal.noise_min_plot maximum = self.signal.noise_max_plot if self.ui.cbSignalView.currentIndex() == 0: # Draw Noise only in Analog View self.ui.gvSignal.scene().draw_noise_area(minimum, maximum - minimum) @pyqtSlot(int) def on_spinbox_selection_start_value_changed(self, value: int): if self.ui.gvSignal.sel_area_active: self.ui.gvSignal.set_selection_area(x=value) self.ui.gvSignal.selection_area.finished = True self.ui.gvSignal.emit_sel_area_width_changed() @pyqtSlot(int) def on_spinbox_selection_end_value_changed(self, value: int): if self.ui.gvSignal.sel_area_active: self.ui.gvSignal.set_selection_area(w=value - self.ui.spinBoxSelectionStart.value()) self.ui.gvSignal.selection_area.finished = True self.ui.gvSignal.emit_sel_area_width_changed() @pyqtSlot() def on_protocol_updated(self): self.ui.gvSignal.redraw_view() # Participants may have changed self.ui.txtEdProto.setEnabled(True) self.ui.txtEdProto.setHtml(self.proto_analyzer.plain_to_html(self.proto_view)) @pyqtSlot(float) def update_legend(self, y_sep): if self.ui.gvLegend.isVisible(): self.ui.gvLegend.ysep = y_sep self.ui.gvLegend.refresh() self.ui.spinBoxCenterOffset.blockSignals(True) self.ui.spinBoxCenterOffset.setValue(-y_sep) self.ui.spinBoxCenterOffset.blockSignals(False) @pyqtSlot() def handle_protocol_sync_changed(self): self.sync_protocol = self.ui.chkBoxSyncSelection.isChecked() @pyqtSlot() def set_protocol_visibility(self): checked = self.ui.chkBoxShowProtocol.isChecked() if checked: self.show_protocol() self.ui.cbProtoView.setEnabled(True) # Disabled because never used # self.ui.chkBoxSyncSelection.show() self.ui.txtEdProto.show() else: self.ui.txtEdProto.hide() self.ui.chkBoxSyncSelection.hide() self.ui.cbProtoView.setEnabled(False) @pyqtSlot() def on_cb_signal_view_index_changed(self): self.setCursor(Qt.WaitCursor) self.ui.gvSignal.redraw_view(reinitialize=True) if self.ui.cbSignalView.currentIndex() > 0: self.ui.gvLegend.y_scene = self.scene_manager.scene.sceneRect().y() self.ui.gvLegend.scene_height = self.scene_manager.scene.sceneRect().height() self.ui.gvLegend.refresh() else: self.ui.gvLegend.hide() self.ui.gvSignal.auto_fit_view() self.ui.gvSignal.refresh_selection_area() self.handle_slideryscale_value_changed() # YScale auf neue Sicht übertragen self.unsetCursor() @pyqtSlot() def on_btn_autodetect_clicked(self): self.signal.auto_detect_on_modulation_changed = bool(self.ui.btnAutoDetect.isChecked()) if self.ui.btnAutoDetect.isChecked(): self.signal.auto_detect() @pyqtSlot() def on_btn_replay_clicked(self): project_manager = self.project_manager try: dialog = SendDialogController(project_manager, modulated_data=self.signal.data, parent=self) except OSError as e: logger.error(repr(e)) return if dialog.has_empty_device_list: Errors.no_device() dialog.close() return dialog.recording_parameters.connect(project_manager.set_recording_parameters) dialog.show() dialog.graphics_view.show_full_scene(reinitialize=True) @pyqtSlot(int, int) def update_selection_area(self, start, end): self.update_number_selected_samples() self.ui.spinBoxSelectionStart.blockSignals(True) self.ui.spinBoxSelectionStart.setValue(start) self.ui.spinBoxSelectionStart.blockSignals(False) self.ui.spinBoxSelectionEnd.blockSignals(True) self.ui.spinBoxSelectionEnd.setValue(end) self.ui.spinBoxSelectionEnd.blockSignals(False) @pyqtSlot() def refresh_protocol(self): self.show_protocol(refresh=True) @pyqtSlot(int) def on_combo_box_proto_view_index_changed(self, index: int): old_view = self.ui.txtEdProto.cur_view self.ui.txtEdProto.cur_view = index self.show_protocol(old_view=old_view) @pyqtSlot(float) def set_qad_center(self, th): self.ui.spinBoxCenterOffset.setValue(th) self.ui.spinBoxCenterOffset.editingFinished.emit() def set_roi_from_protocol_analysis(self, start_message, start_pos, end_message, end_pos, view_type): if not self.proto_analyzer: return if not self.ui.chkBoxShowProtocol.isChecked(): self.ui.chkBoxShowProtocol.setChecked(True) self.set_protocol_visibility() self.ui.cbProtoView.setCurrentIndex(view_type) if view_type == 1: # Hex View start_pos *= 4 end_pos *= 4 elif view_type == 2: # ASCII View start_pos *= 8 end_pos *= 8 sample_pos, num_samples = self.proto_analyzer.get_samplepos_of_bitseq(start_message, start_pos, end_message, end_pos, True) self.protocol_selection_is_updateable = False if sample_pos != -1: if self.jump_sync and self.sync_protocol: self.ui.gvSignal.centerOn(sample_pos, self.ui.gvSignal.y_center) self.ui.gvSignal.set_selection_area(sample_pos, num_samples) self.ui.gvSignal.centerOn(sample_pos + num_samples, self.ui.gvSignal.y_center) else: self.ui.gvSignal.set_selection_area(sample_pos, num_samples) self.ui.gvSignal.zoom_to_selection(sample_pos, sample_pos + num_samples) else: self.ui.gvSignal.set_selection_area(0, 0) self.protocol_selection_is_updateable = True self.update_protocol_selection_from_roi() @pyqtSlot() def update_roi_from_protocol_selection(self): text_edit = self.ui.txtEdProto start_pos, end_pos = text_edit.textCursor().selectionStart(), text_edit.textCursor().selectionEnd() if start_pos == end_pos == -1: return forward_selection = text_edit.textCursor().anchor() <= text_edit.textCursor().position() if start_pos > end_pos: start_pos, end_pos = end_pos, start_pos text = text_edit.toPlainText() start_message = text[:start_pos].count("\n") end_message = start_message + text[start_pos:end_pos].count("\n") newline_pos = text[:start_pos].rfind("\n") if newline_pos != -1: start_pos -= (newline_pos + 1) newline_pos = text[:end_pos].rfind("\n") if newline_pos != -1: end_pos -= (newline_pos + 1) factor = 1 if text_edit.cur_view == 0 else 4 if text_edit.cur_view == 1 else 8 start_pos *= factor end_pos *= factor try: include_last_pause = False s = text_edit.textCursor().selectionStart() e = text_edit.textCursor().selectionEnd() if s > e: s, e = e, s selected_text = text[s:e] last_newline = selected_text.rfind("\n") if last_newline == -1: last_newline = 0 if selected_text.endswith(" "): end_pos -= 1 elif selected_text.endswith(" \t"): end_pos -= 2 if "[" in selected_text[last_newline:]: include_last_pause = True sample_pos, num_samples = self.proto_analyzer.get_samplepos_of_bitseq(start_message, start_pos, end_message, end_pos, include_last_pause) except IndexError: return self.ui.gvSignal.blockSignals(True) if sample_pos != -1: if self.jump_sync and self.sync_protocol: self.ui.gvSignal.centerOn(sample_pos, self.ui.gvSignal.y_center) self.ui.gvSignal.set_selection_area(sample_pos, num_samples) if forward_selection: # Forward Selection --> Center ROI to End of Selection self.ui.gvSignal.centerOn(sample_pos + num_samples, self.ui.gvSignal.y_center) else: # Backward Selection --> Center ROI to Start of Selection self.ui.gvSignal.centerOn(sample_pos, self.ui.gvSignal.y_center) else: self.ui.gvSignal.set_selection_area(sample_pos, num_samples) else: self.ui.gvSignal.set_selection_area(0, 0) self.ui.gvSignal.blockSignals(False) self.update_number_selected_samples() def zoom_to_roi(self): roi = self.ui.gvSignal.selection_area start, end = roi.x, roi.x + roi.width self.ui.gvSignal.zoom_to_selection(start, end) @pyqtSlot() def start_proto_selection_timer(self): self.proto_selection_timer.start() @pyqtSlot() def update_protocol_selection_from_roi(self): protocol = self.proto_analyzer if protocol is None or protocol.messages is None or not self.ui.chkBoxShowProtocol.isChecked(): return start = self.ui.gvSignal.selection_area.x w = self.ui.gvSignal.selection_area.width if w < 0: start += w w = -w c = self.ui.txtEdProto.textCursor() self.jump_sync = False self.ui.txtEdProto.blockSignals(True) try: start_message, start_index, end_message, end_index = protocol.get_bitseq_from_selection(start, w) except IndexError: c.clearSelection() self.ui.txtEdProto.setTextCursor(c) self.jump_sync = True self.ui.txtEdProto.blockSignals(False) return if start_message == -1 or end_index == -1 or start_index == -1 or end_message == -1: c.clearSelection() self.ui.txtEdProto.setTextCursor(c) self.jump_sync = True self.ui.txtEdProto.blockSignals(False) return start_index = int(protocol.convert_index(start_index, 0, self.proto_view, True)[0]) end_index = int(math.ceil(protocol.convert_index(end_index, 0, self.proto_view, True)[1])) + 1 text = self.ui.txtEdProto.toPlainText() n = 0 message_pos = 0 c.setPosition(0) for i, t in enumerate(text): message_pos += 1 if t == "\n": n += 1 message_pos = 0 if n == start_message and message_pos == start_index: c.setPosition(i + 1, QTextCursor.MoveAnchor) if n == end_message and message_pos == end_index: c.setPosition(i, QTextCursor.KeepAnchor) break self.ui.txtEdProto.setTextCursor(c) self.ui.txtEdProto.blockSignals(False) self.jump_sync = True def refresh_signal(self, draw_full_signal=False): self.ui.gvSignal.sel_area_active = False self.draw_signal(draw_full_signal) self.ui.lSamplesInView.setText("{0:n}".format(int(self.ui.gvSignal.view_rect().width()))) self.ui.lSamplesTotal.setText("{0:n}".format(self.signal.num_samples)) self.update_number_selected_samples() self.set_qad_tooltip(self.signal.noise_threshold) self.ui.gvSignal.sel_area_active = True @pyqtSlot(float) def on_signal_qad_center_changed(self, qad_center): self.ui.gvSignal.y_sep = -qad_center self.ui.gvLegend.ysep = -qad_center if self.ui.cbSignalView.currentIndex() > 0: self.scene_manager.scene.draw_sep_area(-qad_center) self.ui.gvLegend.refresh() self.ui.spinBoxCenterOffset.blockSignals(False) self.ui.spinBoxCenterOffset.setValue(qad_center) def on_spinbox_noise_threshold_editing_finished(self): if self.signal is not None and self.signal.noise_threshold != self.ui.spinBoxNoiseTreshold.value(): noise_action = ChangeSignalParameter(signal=self.signal, protocol=self.proto_analyzer, parameter_name="noise_threshold", parameter_value=self.ui.spinBoxNoiseTreshold.value()) self.undo_stack.push(noise_action) self.disable_auto_detection() def set_qad_tooltip(self, noise_threshold): self.ui.cbSignalView.setToolTip( "<html><head/><body><p>Choose the view of your signal.</p><p>The quadrature demodulation uses a <b>threshold of magnitude,</b> to <span style=\" font-weight:600;\">supress noise</span>. All samples with a magnitude lower than this threshold will be eliminated (set to <span style=\" font-style:italic;\">-127</span>) after demod.</p><p>Tune this value by selecting a <span style=\" font-style:italic;\">noisy area</span> and mark it as noise using <b>context menu</b>.</p><p>Current noise threshold is: <b>" + str( noise_threshold) + "</b></p></body></html>") def contextMenuEvent(self, event: QContextMenuEvent): if self.signal is None: return menu = QMenu() apply_to_all_action = menu.addAction(self.tr("Apply values (BitLen, 0/1-Threshold, Tolerance) to all signals")) menu.addSeparator() auto_detect_action = menu.addAction(self.tr("Auto-Detect signal parameters")) action = menu.exec_(self.mapToGlobal(event.pos())) if action == apply_to_all_action: self.setCursor(Qt.WaitCursor) self.apply_to_all_clicked.emit(self.signal) self.unsetCursor() elif action == auto_detect_action: self.setCursor(Qt.WaitCursor) self.signal.auto_detect() self.unsetCursor() def show_modulation_type(self): self.ui.cbModulationType.blockSignals(True) self.ui.cbModulationType.setCurrentIndex(self.signal.modulation_type) self.ui.cbModulationType.blockSignals(False) def disable_auto_detection(self): """ Disable auto detection when user manually edited a value :return: """ if self.signal.auto_detect_on_modulation_changed: self.signal.auto_detect_on_modulation_changed = False self.ui.btnAutoDetect.setChecked(False) def on_participant_changed(self): if self.proto_analyzer: self.proto_analyzer.qt_signals.protocol_updated.emit() @pyqtSlot() def on_info_btn_clicked(self): sdc = SignalDetailsController(self.signal, self) sdc.show() @pyqtSlot(int) def on_combobox_modulation_type_index_changed(self, index: int): modulation_action = ChangeSignalParameter(signal=self.signal, protocol=self.proto_analyzer, parameter_name="modulation_type", parameter_value=index) self.undo_stack.push(modulation_action) @pyqtSlot() def on_signal_data_changed_before_save(self): font = self.ui.lineEditSignalName.font() # type: QFont if self.signal.changed: font.setBold(True) self.ui.btnSaveSignal.show() else: font.setBold(False) self.ui.btnSaveSignal.hide() self.ui.lineEditSignalName.setFont(font) @pyqtSlot() def on_gv_legend_resized(self): if self.ui.gvLegend.isVisible(): self.ui.gvLegend.y_zoom_factor = self.ui.gvSignal.transform().m22() self.ui.gvLegend.refresh() self.ui.gvLegend.translate(0, 1) # Resize verschiebt sonst Pfeile @pyqtSlot() def on_btn_show_hide_start_end_clicked(self): show = self.ui.btnShowHideStartEnd.isChecked() if show: self.ui.btnShowHideStartEnd.setIcon(QIcon.fromTheme("arrow-down-double")) self.ui.verticalLayout.insertItem(2, self.ui.additionalInfos) else: self.ui.btnShowHideStartEnd.setIcon(QIcon.fromTheme("arrow-up-double")) self.ui.verticalLayout.removeItem(self.ui.additionalInfos) for i in range(self.ui.additionalInfos.count()): try: self.ui.additionalInfos.itemAt(i).widget().setVisible(show) except AttributeError: pass @pyqtSlot() def on_spinbox_tolerance_editing_finished(self): if self.signal.tolerance != self.ui.spinBoxTolerance.value(): self.ui.spinBoxTolerance.blockSignals(True) tolerance_action = ChangeSignalParameter(signal=self.signal, protocol=self.proto_analyzer, parameter_name="tolerance", parameter_value=self.ui.spinBoxTolerance.value()) self.undo_stack.push(tolerance_action) self.ui.spinBoxTolerance.blockSignals(False) self.disable_auto_detection() @pyqtSlot() def on_spinbox_infolen_editing_finished(self): if self.signal.bit_len != self.ui.spinBoxInfoLen.value(): self.ui.spinBoxInfoLen.blockSignals(True) bitlen_action = ChangeSignalParameter(signal=self.signal, protocol=self.proto_analyzer, parameter_name="bit_len", parameter_value=self.ui.spinBoxInfoLen.value()) self.undo_stack.push(bitlen_action) self.ui.spinBoxInfoLen.blockSignals(False) self.disable_auto_detection() @pyqtSlot() def on_spinbox_center_editing_finished(self): if self.signal.qad_center != self.ui.spinBoxCenterOffset.value(): self.ui.spinBoxCenterOffset.blockSignals(True) center_action = ChangeSignalParameter(signal=self.signal, protocol=self.proto_analyzer, parameter_name="qad_center", parameter_value=self.ui.spinBoxCenterOffset.value()) self.undo_stack.push(center_action) self.disable_auto_detection() @pyqtSlot() def refresh(self, draw_full_signal=False): self.refresh_signal(draw_full_signal=draw_full_signal) self.refresh_signal_information(block=True) self.show_protocol(refresh=True) @pyqtSlot() def on_btn_filter_clicked(self): if self.apply_filter_to_selection_only.isChecked(): start, end = self.ui.gvSignal.selection_area.start, self.ui.gvSignal.selection_area.end else: start, end = 0, self.signal.num_samples filter_action = EditSignalAction(signal=self.signal, mode=EditAction.filter, start=start, end=end, dsp_filter=self.dsp_filter, protocol=self.proto_analyzer) self.undo_stack.push(filter_action) @pyqtSlot() def on_configure_filter_action_triggered(self): self.filter_dialog.set_dsp_filter(self.dsp_filter) self.filter_dialog.exec() @pyqtSlot(Filter) def on_filter_dialog_filter_accepted(self, dsp_filter: Filter): if dsp_filter is not None: self.dsp_filter = dsp_filter self.set_filter_button_caption()
def __init__(self, proto_analyzer: ProtocolAnalyzer, undo_stack: QUndoStack, project_manager, proto_bits=None, parent=None): super().__init__(parent) self.undo_stack = undo_stack self.ui = Ui_SignalFrame() self.ui.setupUi(self) self.ui.txtEdProto.setFont( QFontDatabase.systemFont(QFontDatabase.FixedFont)) self.ui.txtEdProto.participants = project_manager.participants self.ui.txtEdProto.messages = proto_analyzer.messages self.ui.gvSignal.participants = project_manager.participants self.ui.btnMinimize.setIcon(QIcon(":/icons/data/icons/downarrow.png")) self.is_minimized = False self.setAttribute(Qt.WA_DeleteOnClose) self.project_manager = project_manager self.proto_analyzer = proto_analyzer self.signal = proto_analyzer.signal if self.proto_analyzer is not None else None # type: Signal self.proto_selection_timer = QTimer( ) # For Update Proto Selection from ROI self.proto_selection_timer.setSingleShot(True) self.proto_selection_timer.setInterval(1) # Disabled because never used (see also set_protocol_visibilty()) self.ui.chkBoxSyncSelection.hide() self.ui.btnMinimize.hide() if self.signal is not None: if self.signal.qad_demod_file_loaded: self.ui.lSignalTyp.setText("Quad-Demod Signal (*.wav)") elif self.signal.wav_mode: self.ui.lSignalTyp.setText("Realpart Signal (*.wav)") else: self.ui.lSignalTyp.setText("Complex Signal") self.ui.gvLegend.hide() self.ui.lineEditSignalName.setText(self.signal.name) self.ui.lSamplesInView.setText("{0:,}".format( self.signal.num_samples)) self.ui.lSamplesTotal.setText("{0:,}".format( self.signal.num_samples)) self.sync_protocol = self.ui.chkBoxSyncSelection.isChecked() self.ui.chkBoxSyncSelection.hide() self.ui.splitter.setSizes([self.ui.splitter.height(), 0]) self.protocol_selection_is_updateable = True self.scene_manager = SignalSceneManager(self.signal, self) self.ui.gvSignal.scene_manager = self.scene_manager self.ui.gvSignal.setScene(self.scene_manager.scene) self.jump_sync = True self.on_show_hide_start_end_clicked() self.refresh_signal_information(block=True) self.create_connects() self.set_protocol_visibility() self.ui.chkBoxShowProtocol.setChecked(True) self.set_qad_tooltip(self.signal.noise_threshold) self.ui.btnSaveSignal.hide() self.show_protocol(refresh=False) else: self.ui.lSignalTyp.setText("Protocol (*.proto)") scene, nsamples = SignalSceneManager.create_rectangle(proto_bits) self.ui.lSamplesInView.setText("{0:n}".format(int(nsamples))) self.ui.lSamplesTotal.setText("{0:n}".format(int(nsamples))) self.ui.gvSignal.setScene(scene) self.ui.spinBoxSelectionStart.setMaximum(nsamples) self.ui.spinBoxSelectionEnd.setMaximum(nsamples) self.ui.btnReplay.hide() self.create_connects() self.ui.gvSignal.sel_area_active = True self.ui.btnSaveSignal.hide() self.minimize_maximize()
class SignalFrameController(QFrame): closed = pyqtSignal(QWidget) signal_created = pyqtSignal(Signal) drag_started = pyqtSignal(QPoint) frame_dropped = pyqtSignal(QPoint) files_dropped = pyqtSignal(list) not_show_again_changed = pyqtSignal() signal_drawing_finished = pyqtSignal() apply_to_all_clicked = pyqtSignal(Signal) sort_action_clicked = pyqtSignal() @property def proto_view(self): return self.ui.txtEdProto.cur_view def __init__(self, proto_analyzer: ProtocolAnalyzer, undo_stack: QUndoStack, project_manager, proto_bits=None, parent=None): super().__init__(parent) self.undo_stack = undo_stack self.ui = Ui_SignalFrame() self.ui.setupUi(self) self.ui.txtEdProto.setFont( QFontDatabase.systemFont(QFontDatabase.FixedFont)) self.ui.txtEdProto.participants = project_manager.participants self.ui.txtEdProto.messages = proto_analyzer.messages self.ui.gvSignal.participants = project_manager.participants self.ui.btnMinimize.setIcon(QIcon(":/icons/data/icons/downarrow.png")) self.is_minimized = False self.setAttribute(Qt.WA_DeleteOnClose) self.project_manager = project_manager self.proto_analyzer = proto_analyzer self.signal = proto_analyzer.signal if self.proto_analyzer is not None else None # type: Signal self.proto_selection_timer = QTimer( ) # For Update Proto Selection from ROI self.proto_selection_timer.setSingleShot(True) self.proto_selection_timer.setInterval(1) # Disabled because never used (see also set_protocol_visibilty()) self.ui.chkBoxSyncSelection.hide() self.ui.btnMinimize.hide() if self.signal is not None: if self.signal.qad_demod_file_loaded: self.ui.lSignalTyp.setText("Quad-Demod Signal (*.wav)") elif self.signal.wav_mode: self.ui.lSignalTyp.setText("Realpart Signal (*.wav)") else: self.ui.lSignalTyp.setText("Complex Signal") self.ui.gvLegend.hide() self.ui.lineEditSignalName.setText(self.signal.name) self.ui.lSamplesInView.setText("{0:,}".format( self.signal.num_samples)) self.ui.lSamplesTotal.setText("{0:,}".format( self.signal.num_samples)) self.sync_protocol = self.ui.chkBoxSyncSelection.isChecked() self.ui.chkBoxSyncSelection.hide() self.ui.splitter.setSizes([self.ui.splitter.height(), 0]) self.protocol_selection_is_updateable = True self.scene_manager = SignalSceneManager(self.signal, self) self.ui.gvSignal.scene_manager = self.scene_manager self.ui.gvSignal.setScene(self.scene_manager.scene) self.jump_sync = True self.on_show_hide_start_end_clicked() self.refresh_signal_information(block=True) self.create_connects() self.set_protocol_visibility() self.ui.chkBoxShowProtocol.setChecked(True) self.set_qad_tooltip(self.signal.noise_threshold) self.ui.btnSaveSignal.hide() self.show_protocol(refresh=False) else: self.ui.lSignalTyp.setText("Protocol (*.proto)") scene, nsamples = SignalSceneManager.create_rectangle(proto_bits) self.ui.lSamplesInView.setText("{0:n}".format(int(nsamples))) self.ui.lSamplesTotal.setText("{0:n}".format(int(nsamples))) self.ui.gvSignal.setScene(scene) self.ui.spinBoxSelectionStart.setMaximum(nsamples) self.ui.spinBoxSelectionEnd.setMaximum(nsamples) self.ui.btnReplay.hide() self.create_connects() self.ui.gvSignal.sel_area_active = True self.ui.btnSaveSignal.hide() self.minimize_maximize() def create_connects(self): self.ui.btnCloseSignal.clicked.connect( self.on_btn_close_signal_clicked) self.ui.btnReplay.clicked.connect(self.on_btn_replay_clicked) self.ui.btnAutoDetect.clicked.connect(self.on_btn_autodetect_clicked) self.ui.btnInfo.clicked.connect(self.on_info_btn_clicked) self.ui.btnShowHideStartEnd.clicked.connect( self.on_show_hide_start_end_clicked) self.ui.btnMinimize.clicked.connect(self.minimize_maximize) if self.signal is not None: self.ui.gvSignal.save_clicked.connect(self.save_signal) self.signal.bit_len_changed.connect( self.ui.spinBoxInfoLen.setValue) self.signal.qad_center_changed.connect( self.on_signal_qad_center_changed) self.signal.noise_threshold_changed.connect( self.on_noise_threshold_changed) self.signal.modulation_type_changed.connect( self.show_modulation_type) self.signal.tolerance_changed.connect( self.ui.spinBoxTolerance.setValue) self.signal.protocol_needs_update.connect(self.refresh_protocol) self.signal.data_edited.connect( self.refresh_signal) # Crop/Delete Mute etc. self.signal.sample_rate_changed.connect(self.__set_duration) self.signal.sample_rate_changed.connect( self.show_protocol) # Update times self.signal.saved_status_changed.connect( self.on_signal_data_changed_before_save) self.ui.btnSaveSignal.clicked.connect(self.save_signal) self.signal.name_changed.connect( self.ui.lineEditSignalName.setText) self.ui.gvLegend.resized.connect(self.on_gv_legend_resized) self.ui.gvSignal.sel_area_width_changed.connect( self.start_proto_selection_timer) self.ui.gvSignal.sel_area_start_end_changed.connect( self.start_proto_selection_timer) self.proto_selection_timer.timeout.connect( self.update_protocol_selection_from_roi) self.ui.lineEditSignalName.editingFinished.connect( self.change_signal_name) self.proto_analyzer.qt_signals.protocol_updated.connect( self.on_protocol_updated) self.ui.gvSignal.set_noise_clicked.connect( self.on_set_noise_in_graphic_view_clicked) self.ui.gvSignal.save_as_clicked.connect(self.save_signal_as) self.ui.gvSignal.create_clicked.connect(self.create_new_signal) self.ui.gvSignal.zoomed.connect(self.on_signal_zoomed) self.ui.gvSignal.sel_area_start_end_changed.connect( self.update_selection_area) self.ui.gvSignal.sep_area_changed.connect(self.set_qad_center) self.ui.gvSignal.sep_area_moving.connect(self.update_legend) self.ui.sliderYScale.valueChanged.connect( self.handle_slideryscale_value_changed) self.ui.spinBoxXZoom.valueChanged.connect( self.on_spinbox_x_zoom_value_changed) self.project_manager.project_updated.connect( self.on_participant_changed) self.ui.txtEdProto.participant_changed.connect( self.on_participant_changed) self.ui.gvSignal.participant_changed.connect( self.on_participant_changed) self.proto_selection_timer.timeout.connect( self.update_number_selected_samples) self.ui.cbSignalView.currentIndexChanged.connect( self.on_cb_signal_view_index_changed) self.ui.cbModulationType.currentIndexChanged.connect( self.on_combobox_modulation_type_index_changed) self.ui.cbProtoView.currentIndexChanged.connect( self.on_combo_box_proto_view_index_changed) self.ui.chkBoxShowProtocol.stateChanged.connect( self.set_protocol_visibility) self.ui.chkBoxSyncSelection.stateChanged.connect( self.handle_protocol_sync_changed) self.ui.txtEdProto.proto_view_changed.connect(self.show_protocol) self.ui.txtEdProto.show_proto_clicked.connect( self.update_roi_from_protocol_selection) self.ui.txtEdProto.show_proto_clicked.connect(self.zoom_to_roi) self.ui.txtEdProto.selectionChanged.connect( self.update_roi_from_protocol_selection) self.ui.txtEdProto.deletion_wanted.connect( self.ui.gvSignal.on_delete_action_triggered) self.ui.spinBoxSelectionStart.valueChanged.connect( self.on_spinbox_selection_start_value_changed) self.ui.spinBoxSelectionEnd.valueChanged.connect( self.on_spinbox_selection_end_value_changed) self.ui.spinBoxCenterOffset.editingFinished.connect( self.on_spinbox_center_editing_finished) self.ui.spinBoxTolerance.editingFinished.connect( self.on_spinbox_tolerance_editing_finished) self.ui.spinBoxNoiseTreshold.editingFinished.connect( self.on_spinbox_noise_threshold_editing_finished) self.ui.spinBoxInfoLen.editingFinished.connect( self.on_spinbox_infolen_editing_finished) def refresh_signal_information(self, block=True): self.ui.spinBoxTolerance.blockSignals(block) self.ui.spinBoxCenterOffset.blockSignals(block) self.ui.spinBoxInfoLen.blockSignals(block) self.ui.spinBoxNoiseTreshold.blockSignals(block) self.ui.btnAutoDetect.blockSignals(block) self.ui.spinBoxTolerance.setValue(self.signal.tolerance) self.ui.spinBoxCenterOffset.setValue(self.signal.qad_center) self.ui.spinBoxInfoLen.setValue(self.signal.bit_len) self.ui.spinBoxNoiseTreshold.setValue(self.signal.noise_threshold) self.ui.btnAutoDetect.setChecked( self.signal.auto_detect_on_modulation_changed) self.show_modulation_type() self.ui.spinBoxTolerance.blockSignals(False) self.ui.spinBoxCenterOffset.blockSignals(False) self.ui.spinBoxInfoLen.blockSignals(False) self.ui.spinBoxNoiseTreshold.blockSignals(False) self.ui.btnAutoDetect.blockSignals(False) def set_empty_frame_visibilities(self): self.ui.lInfoLenText.hide() self.ui.spinBoxInfoLen.hide() self.ui.spinBoxCenterOffset.hide() self.ui.spinBoxTolerance.hide() self.ui.chkBoxShowProtocol.hide() self.ui.cbProtoView.hide() self.ui.lErrorTolerance.hide() self.ui.lSignalViewText.hide() self.ui.chkBoxSyncSelection.hide() self.ui.txtEdProto.hide() self.ui.gvLegend.hide() self.ui.cbSignalView.hide() self.ui.cbModulationType.hide() self.ui.btnSaveSignal.hide() self.ui.btnMinimize.hide() def update_number_selected_samples(self): self.ui.lNumSelectedSamples.setText( str(abs(int(self.ui.gvSignal.selection_area.width)))) self.__set_duration() sel_messages = self.ui.gvSignal.selected_messages if len(sel_messages) == 1: self.ui.labelRSSI.setText("RSSI: {}".format( Formatter.big_value_with_suffix(sel_messages[0].rssi))) else: self.ui.labelRSSI.setText("") def change_signal_name(self): self.signal.name = self.ui.lineEditSignalName.text() def __set_duration(self): # On Signal Sample Rate changed try: num_samples = int(self.ui.lNumSelectedSamples.text()) except ValueError: return if self.signal: t = num_samples / self.signal.sample_rate self.ui.lDuration.setText(Formatter.science_time(t)) def handle_slideryscale_value_changed(self): try: gvs = self.ui.gvSignal yscale = self.ui.sliderYScale.value() current_factor = gvs.sceneRect().height() / gvs.view_rect().height( ) gvs.scale(1, yscale / current_factor) x, w = self.ui.gvSignal.view_rect().x( ), self.ui.gvSignal.view_rect().width() gvs.centerOn(x + w / 2, gvs.y_center) except ZeroDivisionError: pass def mousePressEvent(self, event): if event.button() == Qt.LeftButton: self.drag_started.emit(self.mapToParent(event.pos())) drag = QDrag(self) mimeData = QMimeData() mimeData.setText("Move Signal") pixmap = QPixmap(self.rect().size()) self.render(pixmap, QPoint(), QRegion(self.rect())) drag.setPixmap(pixmap) drag.setMimeData(mimeData) drag.exec_() def dragMoveEvent(self, event): event.accept() def dragEnterEvent(self, event): event.acceptProposedAction() def dropEvent(self, event: QDropEvent): if len(event.mimeData().urls()) == 0: self.frame_dropped.emit(self.mapToParent(event.pos())) else: self.files_dropped.emit(event.mimeData().urls()) def create_new_signal(self, start, end): if start < end: new_signal = self.signal.create_new(start, end) self.signal_created.emit(new_signal) else: Errors.empty_selection() def my_close(self): settings = constants.SETTINGS not_show = settings.value('not_show_close_dialog', False, type=bool) if not not_show: cb = QCheckBox("Do not show this again.") msgbox = QMessageBox(QMessageBox.Question, "Confirm close", "Are you sure you want to close?") msgbox.addButton(QMessageBox.Yes) msgbox.addButton(QMessageBox.No) msgbox.setDefaultButton(QMessageBox.No) msgbox.setCheckBox(cb) reply = msgbox.exec() not_show_again = bool(cb.isChecked()) settings.setValue("not_show_close_dialog", not_show_again) self.not_show_again_changed.emit() if reply != QMessageBox.Yes: return self.closed.emit(self) def save_signal(self): if len(self.signal.filename) > 0: self.signal.save() else: self.save_signal_as() def save_signal_as(self): filename = FileOperator.get_save_file_name( self.signal.filename, wav_only=self.signal.wav_mode) if filename: try: self.signal.save_as(filename) except Exception as e: QMessageBox.critical(self, self.tr("Error saving signal"), e.args[0]) def show_autocrop_range(self): start = self.signal.get_signal_start() end = self.signal.get_signal_end() self.ui.gvSignal.set_selection_area(start, end - start) self.ui.gvSignal.selection_area.finished = True self.ui.gvSignal.emit_sel_area_width_changed() def draw_signal(self, full_signal=False): gvs = self.ui.gvSignal gv_legend = self.ui.gvLegend gv_legend.ysep = -self.signal.qad_center # Save current visible region for restoring it after drawing y, h = gvs.sceneRect().y(), gvs.sceneRect().height() x, w = gvs.view_rect().x(), gvs.view_rect().width() self.scene_manager.scene_type = self.ui.cbSignalView.currentIndex() self.scene_manager.init_scene() if full_signal: gvs.show_full_scene() else: gvs.redraw_view() legend = LegendScene() legend.setBackgroundBrush(constants.BGCOLOR) legend.setSceneRect(0, self.scene_manager.scene.sceneRect().y(), gv_legend.width(), self.scene_manager.scene.sceneRect().height()) legend.draw_one_zero_arrows(-self.signal.qad_center) gv_legend.setScene(legend) num_samples = self.signal.num_samples self.ui.spinBoxSelectionStart.setMaximum(num_samples) self.ui.spinBoxSelectionEnd.setMaximum(num_samples) gvs.nsamples = num_samples gvs.sel_area_active = True gvs.y_sep = -self.signal.qad_center if not full_signal: # Restore Zoom w = w if w < self.signal.num_samples else self.signal.num_samples gvs.fitInView(QRectF(x, y, w, h)) gvs.centerOn(x + w / 2, gvs.y_center) def restore_protocol_selection(self, sel_start, sel_end, start_message, end_message, old_protoview): if old_protoview == self.proto_view: return self.protocol_selection_is_updateable = False sel_start = int( self.proto_analyzer.convert_index(sel_start, old_protoview, self.proto_view, True)[0]) sel_end = int( math.ceil( self.proto_analyzer.convert_index(sel_end, old_protoview, self.proto_view, True)[1])) c = self.ui.txtEdProto.textCursor() c.setPosition(0) cur_message = 0 i = 0 text = self.ui.txtEdProto.toPlainText() while cur_message < start_message: if text[i] == "\n": cur_message += 1 i += 1 c.movePosition(QTextCursor.Right, QTextCursor.MoveAnchor, i) c.movePosition(QTextCursor.Right, QTextCursor.MoveAnchor, sel_start) text = text[i:] i = 0 while cur_message < end_message: if text[i] == "\n": cur_message += 1 i += 1 c.movePosition(QTextCursor.Right, QTextCursor.KeepAnchor, i) c.movePosition(QTextCursor.Right, QTextCursor.KeepAnchor, sel_end) self.ui.txtEdProto.setTextCursor(c) self.protocol_selection_is_updateable = True def update_protocol(self): self.ui.txtEdProto.setEnabled(False) QApplication.processEvents() self.proto_analyzer.get_protocol_from_signal() def show_protocol(self, old_view=-1, refresh=False): if not self.proto_analyzer: return if not self.ui.chkBoxShowProtocol.isChecked(): return if old_view == -1: old_view = self.ui.cbProtoView.currentIndex() if self.proto_analyzer.messages is None or refresh: self.update_protocol() else: # Keep things synchronized and restore selection self.ui.txtEdProto.blockSignals(True) self.ui.cbProtoView.blockSignals(True) self.ui.cbProtoView.setCurrentIndex(self.proto_view) self.ui.cbProtoView.blockSignals(False) start_message = 0 sel_start = self.ui.txtEdProto.textCursor().selectionStart() text = self.ui.txtEdProto.toPlainText()[:sel_start] sel_start = 0 read_pause = False for t in text: if t == "\t": read_pause = True if not read_pause: sel_start += 1 if t == "\n": sel_start = 0 start_message += 1 read_pause = False sel_end = self.ui.txtEdProto.textCursor().selectionEnd() text = self.ui.txtEdProto.toPlainText( )[self.ui.txtEdProto.textCursor().selectionStart():sel_end] end_message = 0 sel_end = 0 read_pause = False for t in text: if t == "\t": read_pause = True if not read_pause: sel_end += 1 if t == "\n": sel_end = 0 end_message += 1 read_pause = False self.ui.txtEdProto.setHtml( self.proto_analyzer.plain_to_html(self.proto_view)) # self.ui.txtEdProto.setPlainText(self.proto_analyzer.plain_to_string(self.proto_view)) self.restore_protocol_selection(sel_start, sel_end, start_message, end_message, old_view) self.ui.txtEdProto.blockSignals(False) @pyqtSlot() def on_signal_zoomed(self): gvs = self.ui.gvSignal self.ui.lSamplesInView.setText("{0:n}".format( int(gvs.view_rect().width()))) self.ui.spinBoxXZoom.blockSignals(True) self.ui.spinBoxXZoom.setValue( int(gvs.sceneRect().width() / gvs.view_rect().width() * 100)) self.ui.spinBoxXZoom.blockSignals(False) @pyqtSlot(int) def on_spinbox_x_zoom_value_changed(self, value: int): gvs = self.ui.gvSignal zoom_factor = value / 100 current_factor = gvs.sceneRect().width() / gvs.view_rect().width() gvs.zoom(zoom_factor / current_factor) @pyqtSlot() def on_btn_close_signal_clicked(self): self.my_close() @pyqtSlot() def on_set_noise_in_graphic_view_clicked(self): self.setCursor(Qt.WaitCursor) start = self.ui.gvSignal.selection_area.x end = start + self.ui.gvSignal.selection_area.width new_thresh = self.signal.calc_noise_threshold(start, end) self.ui.spinBoxNoiseTreshold.setValue(new_thresh) self.ui.spinBoxNoiseTreshold.editingFinished.emit() self.unsetCursor() @pyqtSlot() def on_noise_threshold_changed(self): self.ui.spinBoxNoiseTreshold.setValue(self.signal.noise_threshold) minimum = self.signal.noise_min_plot maximum = self.signal.noise_max_plot if self.ui.cbSignalView.currentIndex() == 0: # Draw Noise only in Analog View self.ui.gvSignal.scene().draw_noise_area(minimum, maximum - minimum) @pyqtSlot(int) def on_spinbox_selection_start_value_changed(self, value: int): if self.ui.gvSignal.sel_area_active: self.ui.gvSignal.set_selection_area(x=value) self.ui.gvSignal.selection_area.finished = True self.ui.gvSignal.emit_sel_area_width_changed() @pyqtSlot(int) def on_spinbox_selection_end_value_changed(self, value: int): if self.ui.gvSignal.sel_area_active: self.ui.gvSignal.set_selection_area( w=value - self.ui.spinBoxSelectionStart.value()) self.ui.gvSignal.selection_area.finished = True self.ui.gvSignal.emit_sel_area_width_changed() @pyqtSlot() def on_protocol_updated(self): self.ui.gvSignal.redraw_view() # Participants may have changed self.ui.txtEdProto.setEnabled(True) cur_scroll = self.ui.txtEdProto.verticalScrollBar().value() self.ui.txtEdProto.setHtml( self.proto_analyzer.plain_to_html(self.proto_view)) self.ui.txtEdProto.verticalScrollBar().setValue(cur_scroll) @pyqtSlot(float) def update_legend(self, y_sep): if self.ui.gvLegend.isVisible(): self.ui.gvLegend.ysep = y_sep self.ui.gvLegend.refresh() self.ui.spinBoxCenterOffset.blockSignals(True) self.ui.spinBoxCenterOffset.setValue(-y_sep) self.ui.spinBoxCenterOffset.blockSignals(False) @pyqtSlot() def handle_protocol_sync_changed(self): self.sync_protocol = self.ui.chkBoxSyncSelection.isChecked() @pyqtSlot() def set_protocol_visibility(self): checked = self.ui.chkBoxShowProtocol.isChecked() if checked: self.show_protocol() self.ui.cbProtoView.setEnabled(True) # Disabled because never used # self.ui.chkBoxSyncSelection.show() self.ui.txtEdProto.show() else: self.ui.txtEdProto.hide() self.ui.chkBoxSyncSelection.hide() self.ui.cbProtoView.setEnabled(False) @pyqtSlot() def on_cb_signal_view_index_changed(self): self.setCursor(Qt.WaitCursor) self.ui.gvSignal.redraw_view(reinitialize=True) if self.ui.cbSignalView.currentIndex() > 0: self.ui.gvLegend.y_scene = self.scene_manager.scene.sceneRect().y() self.ui.gvLegend.scene_height = self.scene_manager.scene.sceneRect( ).height() self.ui.gvLegend.refresh() else: self.ui.gvLegend.hide() QApplication.processEvents() self.ui.gvSignal.auto_fit_view() self.handle_slideryscale_value_changed( ) # YScale auf neue Sicht übertragen self.unsetCursor() @pyqtSlot() def on_btn_autodetect_clicked(self): self.signal.auto_detect_on_modulation_changed = bool( self.ui.btnAutoDetect.isChecked()) if self.ui.btnAutoDetect.isChecked(): self.signal.auto_detect() @pyqtSlot() def on_btn_replay_clicked(self): project_manager = self.project_manager dialog = SendDialogController(project_manager.frequency, project_manager.sample_rate, project_manager.bandwidth, project_manager.gain, project_manager.device, modulated_data=self.signal.data, parent=self) if dialog.has_empty_device_list: Errors.no_device() dialog.close() return dialog.recording_parameters.connect( project_manager.set_recording_parameters) dialog.show() dialog.graphics_view.show_full_scene(reinitialize=True) @pyqtSlot(int, int) def update_selection_area(self, start, end): self.ui.lNumSelectedSamples.setText(str(end - start)) self.__set_duration() self.ui.spinBoxSelectionStart.blockSignals(True) self.ui.spinBoxSelectionStart.setValue(start) self.ui.spinBoxSelectionStart.blockSignals(False) self.ui.spinBoxSelectionEnd.blockSignals(True) self.ui.spinBoxSelectionEnd.setValue(end) self.ui.spinBoxSelectionEnd.blockSignals(False) @pyqtSlot() def refresh_protocol(self): self.show_protocol(refresh=True) @pyqtSlot(int) def on_combo_box_proto_view_index_changed(self, index: int): old_view = self.ui.txtEdProto.cur_view self.ui.txtEdProto.cur_view = index self.show_protocol(old_view=old_view) @pyqtSlot(float) def set_qad_center(self, th): self.ui.spinBoxCenterOffset.setValue(th) self.ui.spinBoxCenterOffset.editingFinished.emit() def set_roi_from_protocol_analysis(self, start_message, start_pos, end_message, end_pos, view_type): if not self.proto_analyzer: return if not self.ui.chkBoxShowProtocol.isChecked(): self.ui.chkBoxShowProtocol.setChecked(True) self.set_protocol_visibility() self.ui.cbProtoView.setCurrentIndex(view_type) if view_type == 1: # Hex View start_pos *= 4 end_pos *= 4 elif view_type == 2: # ASCII View start_pos *= 8 end_pos *= 8 sample_pos, num_samples = self.proto_analyzer.get_samplepos_of_bitseq( start_message, start_pos, end_message, end_pos, True) self.protocol_selection_is_updateable = False if sample_pos != -1: if self.jump_sync and self.sync_protocol: self.ui.gvSignal.centerOn(sample_pos, self.ui.gvSignal.y_center) self.ui.gvSignal.set_selection_area(sample_pos, num_samples) self.ui.gvSignal.centerOn(sample_pos + num_samples, self.ui.gvSignal.y_center) else: self.ui.gvSignal.set_selection_area(sample_pos, num_samples) self.ui.gvSignal.zoom_to_selection(sample_pos, sample_pos + num_samples) else: self.ui.gvSignal.set_selection_area(0, 0) self.protocol_selection_is_updateable = True self.update_protocol_selection_from_roi() @pyqtSlot() def update_roi_from_protocol_selection(self): text_edit = self.ui.txtEdProto end_pos = text_edit.textCursor().selectionEnd() start_pos = text_edit.textCursor().selectionStart() forward_selection = text_edit.textCursor().anchor( ) <= text_edit.textCursor().position() if start_pos > end_pos: start_pos, end_pos = end_pos, start_pos start_message = text_edit.toPlainText()[:start_pos].count("\n") end_message = start_message + text_edit.toPlainText( )[start_pos:end_pos].count("\n") newline_pos = text_edit.toPlainText()[:start_pos].rfind("\n") if newline_pos != -1: start_pos -= (newline_pos + 1) newline_pos = text_edit.toPlainText()[:end_pos].rfind("\n") if newline_pos != -1: end_pos -= (newline_pos + 1) if text_edit.cur_view == 1: # Hex View start_pos *= 4 end_pos *= 4 elif text_edit.cur_view == 2: # ASCII View start_pos *= 8 end_pos *= 8 try: include_last_pause = False s = text_edit.textCursor().selectionStart() e = text_edit.textCursor().selectionEnd() if s > e: s, e = e, s selected_text = text_edit.toPlainText()[s:e] last_newline = selected_text.rfind("\n") if last_newline == -1: last_newline = 0 if selected_text.endswith(" "): end_pos -= 1 elif selected_text.endswith(" \t"): end_pos -= 2 if "[" in selected_text[last_newline:]: include_last_pause = True sample_pos, num_samples = self.proto_analyzer.get_samplepos_of_bitseq( start_message, start_pos, end_message, end_pos, include_last_pause) except IndexError: return self.ui.gvSignal.blockSignals(True) if sample_pos != -1: if self.jump_sync and self.sync_protocol: self.ui.gvSignal.centerOn(sample_pos, self.ui.gvSignal.y_center) self.ui.gvSignal.set_selection_area(sample_pos, num_samples) if forward_selection: # Forward Selection --> Center ROI to End of Selection self.ui.gvSignal.centerOn(sample_pos + num_samples, self.ui.gvSignal.y_center) else: # Backward Selection --> Center ROI to Start of Selection self.ui.gvSignal.centerOn(sample_pos, self.ui.gvSignal.y_center) else: self.ui.gvSignal.set_selection_area(sample_pos, num_samples) else: self.ui.gvSignal.set_selection_area(0, 0) self.ui.gvSignal.blockSignals(False) self.update_number_selected_samples() def zoom_to_roi(self): roi = self.ui.gvSignal.selection_area start, end = roi.x, roi.x + roi.width self.ui.gvSignal.zoom_to_selection(start, end) @pyqtSlot() def start_proto_selection_timer(self): self.proto_selection_timer.start() @pyqtSlot() def update_protocol_selection_from_roi(self): protocol = self.proto_analyzer if protocol.messages is None or not self.ui.chkBoxShowProtocol.isChecked( ): return start = self.ui.gvSignal.selection_area.x w = self.ui.gvSignal.selection_area.width if w < 0: start += w w = -w c = self.ui.txtEdProto.textCursor() self.jump_sync = False self.ui.txtEdProto.blockSignals(True) try: start_message, start_index, end_message, end_index = protocol.get_bitseq_from_selection( start, w, self.signal.bit_len) except IndexError: c.clearSelection() self.ui.txtEdProto.setTextCursor(c) self.jump_sync = True self.ui.txtEdProto.blockSignals(False) return if start_message == -1 or end_index == -1 or start_index == -1 or end_message == -1: c.clearSelection() self.ui.txtEdProto.setTextCursor(c) self.jump_sync = True self.ui.txtEdProto.blockSignals(False) return start_index = int( protocol.convert_index(start_index, 0, self.proto_view, True)[0]) end_index = int( math.ceil( protocol.convert_index(end_index, 0, self.proto_view, True)[1])) text = self.ui.txtEdProto.toPlainText() n = 0 message_pos = 0 c.setPosition(0) for i, t in enumerate(text): message_pos += 1 if t == "\n": n += 1 message_pos = 0 if n == start_message and message_pos == start_index: c.setPosition(i + 1, QTextCursor.MoveAnchor) if n == end_message and message_pos == end_index: c.setPosition(i, QTextCursor.KeepAnchor) break self.ui.txtEdProto.setTextCursor(c) self.ui.txtEdProto.blockSignals(False) self.jump_sync = True def refresh_signal(self, draw_full_signal=False): gvs = self.ui.gvSignal gvs.sel_area_active = False self.draw_signal(draw_full_signal) self.ui.lSamplesInView.setText("{0:n}".format( int(gvs.view_rect().width()))) self.ui.lSamplesTotal.setText("{0:n}".format(self.signal.num_samples)) selected = 0 if not self.ui.gvSignal.selection_area.is_empty: selected = self.ui.gvSignal.selection_area.width self.ui.lNumSelectedSamples.setText(str(selected)) self.__set_duration() self.set_qad_tooltip(self.signal.noise_threshold) gvs.sel_area_active = True @pyqtSlot(float) def on_signal_qad_center_changed(self, qad_center): self.ui.gvSignal.y_sep = -qad_center self.ui.gvLegend.ysep = -qad_center if self.ui.cbSignalView.currentIndex() > 0: self.scene_manager.scene.draw_sep_area(-qad_center) self.ui.gvLegend.refresh() self.ui.spinBoxCenterOffset.blockSignals(False) self.ui.spinBoxCenterOffset.setValue(qad_center) def on_spinbox_noise_threshold_editing_finished(self): if self.signal is not None and self.signal.noise_threshold != self.ui.spinBoxNoiseTreshold.value( ): noise_action = ChangeSignalParameter( signal=self.signal, protocol=self.proto_analyzer, parameter_name="noise_threshold", parameter_value=self.ui.spinBoxNoiseTreshold.value()) self.undo_stack.push(noise_action) self.disable_auto_detection() def minimize_maximize(self): elements = vars(self.ui) if not self.is_minimized: for name, widget in elements.items(): if name != "btnMinimize" and type(widget) not in (QHBoxLayout, QVBoxLayout, QGridLayout) \ and name not in ( "lSignalNr", "lineEditSignalName", "btnCloseSignal", "lSignalTyp", "btnSaveSignal"): widget.hide() self.ui.btnMinimize.setIcon( QIcon(":/icons/data/icons/uparrow.png")) self.is_minimized = True self.setFixedHeight(65) else: show_start_end = self.ui.btnShowHideStartEnd.text() == "-" for name, widget in elements.items(): if type(widget) in (QHBoxLayout, QVBoxLayout, QGridLayout): continue if not self.ui.chkBoxShowProtocol.isChecked() and name in ( "txtEdProto", "chkBoxSyncSelection"): continue if not show_start_end and name in ( "lStart", "spinBoxSelectionStart", "lEnd", "spinBoxSelectionEnd", "lSamplesInView", "lStrich", "lSamplesTotal", "lSamplesViewText", "btnSaveSignal"): continue if not self.signal.changed and name == "btnSaveSignal": continue if name == "gvLegend": continue widget.show() self.ui.btnMinimize.setIcon( QIcon(":/icons/data/icons/downarrow.png")) self.is_minimized = False self.setMinimumHeight(0) self.setMaximumHeight(20000) if self.ui.cbSignalView.currentIndex() > 0: self.ui.gvLegend.refresh() if self.signal is None: self.set_empty_frame_visibilities() def set_qad_tooltip(self, noise_threshold): self.ui.cbSignalView.setToolTip( "<html><head/><body><p>Choose the view of your signal.</p><p>The quadrature demodulation uses a <span style=\" text-decoration: underline;\">threshold of magnitude,</span> to <span style=\" font-weight:600;\">supress noise</span>. All samples with a magnitude lower than this threshold will be eliminated (set to <span style=\" font-style:italic;\">-127</span>) after demod.</p><p>Tune this value by selecting a <span style=\" font-style:italic;\">noisy area</span> and mark it as noise using <span style=\" text-decoration: underline;\">context menu</span>.</p><p>Current noise threshold is: <b>" + str(noise_threshold) + "</b></p></body></html>") def contextMenuEvent(self, event: QContextMenuEvent): if self.signal is None: return menu = QMenu() apply_to_all_action = menu.addAction( self. tr("Apply values (BitLen, 0/1-Threshold, Tolerance) to all signals" )) menu.addSeparator() auto_detect_action = menu.addAction( self.tr("Auto-Detect signal parameters")) action = menu.exec_(self.mapToGlobal(event.pos())) if action == apply_to_all_action: self.setCursor(Qt.WaitCursor) self.apply_to_all_clicked.emit(self.signal) self.unsetCursor() elif action == auto_detect_action: self.setCursor(Qt.WaitCursor) self.signal.auto_detect() self.unsetCursor() def show_modulation_type(self): self.ui.cbModulationType.blockSignals(True) self.ui.cbModulationType.setCurrentIndex(self.signal.modulation_type) self.ui.cbModulationType.blockSignals(False) def disable_auto_detection(self): """ Disable auto detection when user manually edited a value :return: """ if self.signal.auto_detect_on_modulation_changed: self.signal.auto_detect_on_modulation_changed = False self.ui.btnAutoDetect.setChecked(False) def on_participant_changed(self): if self.proto_analyzer: self.proto_analyzer.qt_signals.protocol_updated.emit() @pyqtSlot() def on_info_btn_clicked(self): sdc = SignalDetailsController(self.signal, self) sdc.show() @pyqtSlot(int) def on_combobox_modulation_type_index_changed(self, index: int): modulation_action = ChangeSignalParameter( signal=self.signal, protocol=self.proto_analyzer, parameter_name="modulation_type", parameter_value=index) self.undo_stack.push(modulation_action) @pyqtSlot() def on_signal_data_changed_before_save(self): font = self.ui.lineEditSignalName.font() # type: QFont if self.signal.changed: font.setBold(True) self.ui.btnSaveSignal.show() else: font.setBold(False) self.ui.btnSaveSignal.hide() self.ui.lineEditSignalName.setFont(font) @pyqtSlot() def on_gv_legend_resized(self): if self.ui.gvLegend.isVisible(): self.ui.gvLegend.y_zoom_factor = self.ui.gvSignal.transform().m22() self.ui.gvLegend.refresh() self.ui.gvLegend.translate(0, 1) # Resize verschiebt sonst Pfeile @pyqtSlot() def on_show_hide_start_end_clicked(self): if self.ui.btnShowHideStartEnd.text() == "+": self.ui.btnShowHideStartEnd.setText("-") self.ui.verticalLayout.insertItem(2, self.ui.additionalInfos) self.ui.lStart.show() self.ui.lEnd.show() self.ui.lSamplesInView.show() self.ui.lStrich.show() self.ui.lSamplesTotal.show() self.ui.lSamplesViewText.show() self.ui.spinBoxSelectionStart.show() self.ui.spinBoxSelectionEnd.show() else: self.ui.btnShowHideStartEnd.setText("+") self.ui.lStart.hide() self.ui.lEnd.hide() self.ui.lSamplesInView.hide() self.ui.lStrich.hide() self.ui.lSamplesTotal.hide() self.ui.lSamplesViewText.hide() self.ui.spinBoxSelectionStart.hide() self.ui.spinBoxSelectionEnd.hide() self.ui.verticalLayout.removeItem(self.ui.additionalInfos) @pyqtSlot() def on_spinbox_tolerance_editing_finished(self): if self.signal.tolerance != self.ui.spinBoxTolerance.value(): self.ui.spinBoxTolerance.blockSignals(True) tolerance_action = ChangeSignalParameter( signal=self.signal, protocol=self.proto_analyzer, parameter_name="tolerance", parameter_value=self.ui.spinBoxTolerance.value()) self.undo_stack.push(tolerance_action) self.ui.spinBoxTolerance.blockSignals(False) self.disable_auto_detection() @pyqtSlot() def on_spinbox_infolen_editing_finished(self): if self.signal.bit_len != self.ui.spinBoxInfoLen.value(): self.ui.spinBoxInfoLen.blockSignals(True) bitlen_action = ChangeSignalParameter( signal=self.signal, protocol=self.proto_analyzer, parameter_name="bit_len", parameter_value=self.ui.spinBoxInfoLen.value()) self.undo_stack.push(bitlen_action) self.ui.spinBoxInfoLen.blockSignals(False) self.disable_auto_detection() @pyqtSlot() def on_spinbox_center_editing_finished(self): if self.signal.qad_center != self.ui.spinBoxCenterOffset.value(): self.ui.spinBoxCenterOffset.blockSignals(True) center_action = ChangeSignalParameter( signal=self.signal, protocol=self.proto_analyzer, parameter_name="qad_center", parameter_value=self.ui.spinBoxCenterOffset.value()) self.undo_stack.push(center_action) self.disable_auto_detection() @pyqtSlot() def refresh(self, draw_full_signal=False): self.refresh_signal(draw_full_signal=draw_full_signal) self.refresh_signal_information(block=True) self.show_protocol(refresh=True)
def __init__(self, proto_analyzer: ProtocolAnalyzer, undo_stack: QUndoStack, project_manager, proto_bits=None, parent=None): super().__init__(parent) self.undo_stack = undo_stack self.ui = Ui_SignalFrame() self.ui.setupUi(self) fixed_font = QFontDatabase.systemFont(QFontDatabase.FixedFont) fixed_font.setPointSize(QApplication.instance().font().pointSize()) self.ui.txtEdProto.setFont(fixed_font) self.ui.txtEdProto.participants = project_manager.participants self.ui.txtEdProto.messages = proto_analyzer.messages self.ui.gvSignal.participants = project_manager.participants self.setAttribute(Qt.WA_DeleteOnClose) self.project_manager = project_manager self.proto_analyzer = proto_analyzer self.signal = proto_analyzer.signal if self.proto_analyzer is not None else None # type: Signal self.dsp_filter = Filter([0.1] * 10, FilterType.moving_average) self.set_filter_button_caption() self.filter_dialog = FilterDialogController(self.dsp_filter, parent=self) self.proto_selection_timer = QTimer( ) # For Update Proto Selection from ROI self.proto_selection_timer.setSingleShot(True) self.proto_selection_timer.setInterval(1) # Disabled because never used (see also set_protocol_visibilty()) self.ui.chkBoxSyncSelection.hide() if self.signal is not None: self.filter_menu = QMenu() self.apply_filter_to_selection_only = self.filter_menu.addAction( self.tr("Apply only to selection")) self.apply_filter_to_selection_only.setCheckable(True) self.apply_filter_to_selection_only.setChecked(False) self.configure_filter_action = self.filter_menu.addAction( "Configure filter...") self.configure_filter_action.setIcon(QIcon.fromTheme("configure")) self.configure_filter_action.triggered.connect( self.on_configure_filter_action_triggered) self.ui.btnFilter.setMenu(self.filter_menu) if self.signal.qad_demod_file_loaded: self.ui.lSignalTyp.setText("Quad-Demod Signal (*.wav)") elif self.signal.wav_mode: self.ui.lSignalTyp.setText("Realpart Signal (*.wav)") else: self.ui.lSignalTyp.setText("Complex Signal") self.ui.gvLegend.hide() self.ui.lineEditSignalName.setText(self.signal.name) self.ui.lSamplesInView.setText("{0:,}".format( self.signal.num_samples)) self.ui.lSamplesTotal.setText("{0:,}".format( self.signal.num_samples)) self.sync_protocol = self.ui.chkBoxSyncSelection.isChecked() self.ui.chkBoxSyncSelection.hide() self.ui.splitter.setSizes([self.ui.splitter.height(), 0]) self.protocol_selection_is_updateable = True self.scene_manager = SignalSceneManager(self.signal, self) self.ui.gvSignal.scene_manager = self.scene_manager self.ui.gvSignal.setScene(self.scene_manager.scene) self.jump_sync = True self.on_btn_show_hide_start_end_clicked() self.refresh_signal_information(block=True) self.create_connects() self.set_protocol_visibility() self.ui.chkBoxShowProtocol.setChecked(True) self.set_qad_tooltip(self.signal.noise_threshold) self.ui.btnSaveSignal.hide() self.show_protocol(refresh=False) else: self.ui.btnFilter.setDisabled(True) suffix = "" if not proto_analyzer.filename: suffix = "" elif proto_analyzer.filename.endswith(".proto"): suffix = " (*.proto)" elif proto_analyzer.filename.endswith(".txt"): suffix = " (*.txt)" self.ui.lSignalTyp.setText("Protocol" + suffix) scene, nsamples = SignalSceneManager.create_rectangle(proto_bits) self.ui.lSamplesInView.setText("{0:n}".format(int(nsamples))) self.ui.lSamplesTotal.setText("{0:n}".format(int(nsamples))) self.ui.gvSignal.setScene(scene) self.ui.spinBoxSelectionStart.setMaximum(nsamples) self.ui.spinBoxSelectionEnd.setMaximum(nsamples) self.ui.btnReplay.hide() self.create_connects() self.ui.gvSignal.sel_area_active = True self.ui.btnSaveSignal.hide()
class SignalFrameController(QFrame): closed = pyqtSignal(QWidget) signal_created = pyqtSignal(Signal) drag_started = pyqtSignal(QPoint) frame_dropped = pyqtSignal(QPoint) files_dropped = pyqtSignal(list) not_show_again_changed = pyqtSignal() signal_drawing_finished = pyqtSignal() apply_to_all_clicked = pyqtSignal(Signal) sort_action_clicked = pyqtSignal() def __init__(self, proto_analyzer: ProtocolAnalyzer, undo_stack: QUndoStack, project_manager, proto_bits=None, parent=None): super().__init__(parent) self.ui = Ui_SignalFrame() self.ui.setupUi(self) self.ui.txtEdProto.setFont(FontHelper.getMonospaceFont()) self.ui.btnMinimize.setIcon(QIcon(":/icons/data/icons/downarrow.png")) self.is_minimized = False self.common_zoom = False self.undo_stack = undo_stack self.setAttribute(Qt.WA_DeleteOnClose) self.project_manager = project_manager self.proto_analyzer = proto_analyzer self.signal = proto_analyzer.signal if self.proto_analyzer is not None else None """:type: Signal """ self.redraw_timer = QTimer() self.redraw_timer.setSingleShot(True) self.redraw_timer.setInterval(100) self.proto_selection_timer = QTimer() # For Update Proto Selection from ROI self.proto_selection_timer.setSingleShot(True) self.proto_selection_timer.setInterval(1) if self.signal is not None: if self.signal.qad_demod_file_loaded: self.ui.lSignalTyp.setText("Quad-Demod Signal (*.wav)") elif self.signal.wav_mode: self.ui.lSignalTyp.setText("Realpart Signal (*.wav)") else: self.ui.lSignalTyp.setText("Complex Signal") self.ui.gvLegend.hide() self.ui.lineEditSignalName.setText(self.signal.name) self.ui.lSamplesInView.setText("{0:,}".format(self.signal.num_samples)) self.ui.lSamplesTotal.setText("{0:,}".format(self.signal.num_samples)) self.sync_protocol = self.ui.chkBoxSyncSelection.isChecked() self.ui.chkBoxSyncSelection.hide() self.ui.splitter.setSizes([self.ui.splitter.height(), 0]) self.protocol_selection_is_updateable = True self.scene_creator = SignalSceneManager(self.signal, self) self.ui.gvSignal.setScene(self.scene_creator.scene) self.jump_sync = True self.handle_show_hide_start_end_clicked() self.refresh_signal_informations(block=True) self.create_connects() self.set_protocol_visibilty() self.ui.chkBoxShowProtocol.setChecked(True) self.set_qad_tooltip(self.signal.noise_treshold) self.ui.btnSaveSignal.hide() self.show_protocol(refresh=False) else: self.ui.lSignalTyp.setText("Protocol (*.txt)") scene, nsamples = SignalSceneManager.create_rectangle(proto_bits) self.ui.lSamplesInView.setText("{0:n}".format(int(nsamples))) self.ui.lSamplesTotal.setText("{0:n}".format(int(nsamples))) self.ui.gvSignal.setScene(scene) self.ui.btnReplay.hide() self.create_connects() self.ui.btnSaveSignal.hide() self.minimize_maximize() @property def spectrogram_is_active(self) -> bool: return self.ui.stackedWidget.currentWidget() == self.ui.pageSpectrogram def create_connects(self): self.ui.spinBoxSelectionStart.valueChanged.connect(self.set_selection_start) self.ui.spinBoxSelectionEnd.valueChanged.connect(self.set_selection_end) self.ui.gvSignal.set_noise_clicked.connect(self.update_noise_min_max) self.ui.btnCloseSignal.clicked.connect(self.my_close) self.ui.btnAutoDetect.clicked.connect(self.on_btn_autodetect_clicked) if self.signal is not None: self.ui.gvSignal.save_clicked.connect(self.save_signal) self.signal.bit_len_changed.connect(self.ui.spinBoxInfoLen.setValue) self.signal.qad_center_changed.connect(self.update_qad_center_view) self.signal.qad_center_changed.connect(self.ui.spinBoxCenterOffset.setValue) self.signal.noise_treshold_changed.connect(self.redraw_noise) self.signal.modulation_type_changed.connect(self.show_modulation_type) self.signal.tolerance_changed.connect(self.ui.spinBoxTolerance.setValue) self.signal.protocol_needs_update.connect(self.refresh_protocol) self.signal.full_refresh_needed.connect(self.refresh) self.signal.sample_rate_changed.connect(self.__set_duration) self.signal.sample_rate_changed.connect(self.show_protocol) # Update times self.ui.gvSignal.horizontalScrollBar().valueChanged.connect(self.on_signal_scrolled) self.signal.saved_status_changed.connect(self.handle_signal_data_changed_before_save) self.ui.btnSaveSignal.clicked.connect(self.save_signal) self.signal.name_changed.connect(self.ui.lineEditSignalName.setText) self.ui.gvLegend.resized.connect(self.handle_gv_legend_resized) self.ui.gvSignal.selection_width_changed.connect(self.start_proto_selection_timer) self.ui.gvSignal.sel_area_start_end_changed.connect(self.start_proto_selection_timer) self.proto_selection_timer.timeout.connect(self.update_protocol_selection_from_roi) self.spectrogram_update_timer.timeout.connect(self.on_spectrogram_update_timer_timeout) self.ui.lineEditSignalName.editingFinished.connect(self.change_signal_name) self.proto_analyzer.qt_signals.protocol_updated.connect(self.on_protocol_updated) self.redraw_timer.timeout.connect(self.redraw_signal) self.ui.btnReplay.clicked.connect(self.on_btn_replay_clicked) self.ui.gvSignal.save_as_clicked.connect(self.save_signal_as) self.ui.gvSignal.create_clicked.connect(self.create_new_signal) self.ui.gvSignal.show_crop_range_clicked.connect(self.show_autocrop_range) self.ui.gvSignal.crop_clicked.connect(self.crop_signal) self.ui.gvSignal.zoomed.connect(self.handle_signal_zoomed) self.ui.sliderYScale.valueChanged.connect(self.handle_slideryscale_value_changed) self.ui.spinBoxXZoom.valueChanged.connect(self.handle_spinbox_xzoom_value_changed) self.ui.btnInfo.clicked.connect(self.on_info_btn_clicked) self.proto_selection_timer.timeout.connect(self.update_nselected_samples) self.ui.gvSignal.sel_area_start_end_changed.connect(self.update_selection_area) self.ui.cbSignalView.currentIndexChanged.connect(self.on_cb_signal_view_index_changed) self.ui.chkBoxShowProtocol.stateChanged.connect(self.set_protocol_visibilty) self.ui.gvSignal.sep_area_changed.connect(self.set_qad_center) self.ui.txtEdProto.proto_view_changed.connect(self.show_protocol) self.ui.txtEdProto.show_proto_clicked.connect(self.update_roi_from_protocol_selection) self.ui.txtEdProto.show_proto_clicked.connect(self.zoom_to_roi) self.ui.gvSignal.sep_area_moving.connect(self.update_legend) self.ui.txtEdProto.selectionChanged.connect(self.update_roi_from_protocol_selection) self.ui.chkBoxSyncSelection.stateChanged.connect(self.handle_protocol_sync_changed) self.ui.gvSignal.deletion_wanted.connect(self.delete_selection) self.ui.gvSignal.mute_wanted.connect(self.mute_selection) self.ui.spinBoxCenterOffset.editingFinished.connect(self.on_spinBoxCenter_editingFinished) self.ui.spinBoxTolerance.editingFinished.connect(self.on_spinBoxTolerance_editingFinished) self.ui.spinBoxNoiseTreshold.editingFinished.connect(self.on_spinBoxNoiseTreshold_editingFinished) self.ui.spinBoxInfoLen.editingFinished.connect(self.on_spinBoxInfoLen_editingFinished) self.ui.btnShowHideStartEnd.clicked.connect(self.handle_show_hide_start_end_clicked) self.ui.cbProtoView.currentIndexChanged.connect(self.handle_proto_infos_index_changed) self.ui.txtEdProto.deletion_wanted.connect(self.delete_from_protocol_selection) self.ui.btnMinimize.clicked.connect(self.minimize_maximize) @property def proto_view(self): return self.ui.txtEdProto.cur_view @property def signal_widgets(self): splitter = self.parent() for i in range(splitter.count() - 1): yield (splitter.widget(i)) def refresh_signal_informations(self, block=True): self.ui.spinBoxTolerance.blockSignals(block) self.ui.spinBoxCenterOffset.blockSignals(block) self.ui.spinBoxInfoLen.blockSignals(block) self.ui.spinBoxNoiseTreshold.blockSignals(block) self.ui.btnAutoDetect.blockSignals(block) self.ui.spinBoxTolerance.setValue(self.signal.tolerance) self.ui.spinBoxCenterOffset.setValue(self.signal.qad_center) self.ui.spinBoxInfoLen.setValue(self.signal.bit_len) self.ui.spinBoxNoiseTreshold.setValue(self.signal.noise_treshold) self.ui.btnAutoDetect.setChecked(self.signal.auto_detect_on_modulation_changed) self.show_modulation_type() self.ui.spinBoxTolerance.blockSignals(False) self.ui.spinBoxCenterOffset.blockSignals(False) self.ui.spinBoxInfoLen.blockSignals(False) self.ui.spinBoxNoiseTreshold.blockSignals(False) self.ui.btnAutoDetect.blockSignals(False) def set_empty_frame_visibilities(self): self.ui.lInfoLenText.hide() self.ui.spinBoxInfoLen.hide() self.ui.spinBoxCenterOffset.hide() self.ui.spinBoxTolerance.hide() self.ui.chkBoxShowProtocol.hide() self.ui.cbProtoView.hide() self.ui.lErrorTolerance.hide() self.ui.lSignalViewText.hide() self.ui.chkBoxSyncSelection.hide() self.ui.txtEdProto.hide() self.ui.gvLegend.hide() self.ui.cbSignalView.hide() #self.ui.btnCloseSignal.hide() self.ui.btnSaveSignal.hide() self.ui.btnMinimize.hide() @pyqtSlot(float) def update_legend(self, y_sep): if self.ui.gvLegend.isVisible(): self.ui.gvLegend.ysep = y_sep self.ui.gvLegend.refresh() self.ui.spinBoxCenterOffset.blockSignals(True) self.ui.spinBoxCenterOffset.setValue(-y_sep) self.ui.spinBoxCenterOffset.blockSignals(False) def emit_signal_drawing_finished(self): self.signal_drawing_finished.emit() @pyqtSlot() def handle_protocol_sync_changed(self): self.sync_protocol = self.ui.chkBoxSyncSelection.isChecked() @pyqtSlot() def set_protocol_visibilty(self): checked = self.ui.chkBoxShowProtocol.isChecked() if checked: self.show_protocol() self.ui.cbProtoView.setEnabled(True) self.ui.chkBoxSyncSelection.show() self.ui.txtEdProto.show() else: self.ui.txtEdProto.hide() self.ui.chkBoxSyncSelection.hide() self.ui.cbProtoView.setEnabled(False) @pyqtSlot() def on_cb_signal_view_index_changed(self): self.setCursor(Qt.WaitCursor) ind = self.ui.cbSignalView.currentIndex() self.scene_creator.scene_type = ind gvs = self.ui.gvSignal if ind > 0: self.signal.modulation_type = ind - 1 vr = self.ui.gvSignal.view_rect() self.scene_creator.init_scene() self.scene_creator.show_scene_section(vr.x(), vr.x() + vr.width()) if ind > 0: self.ui.gvLegend.y_scene = self.scene_creator.scene.sceneRect().y() self.ui.gvLegend.scene_height = self.scene_creator.scene.sceneRect().height() self.ui.gvLegend.refresh() else: self.ui.gvLegend.hide() QApplication.processEvents() gvs.autofit_view() self.handle_slideryscale_value_changed() # YScale auf neue Sicht übertragen self.unsetCursor() @pyqtSlot() def on_btn_autodetect_clicked(self): self.signal.auto_detect_on_modulation_changed = bool(self.ui.btnAutoDetect.isChecked()) if self.ui.btnAutoDetect.isChecked(): self.signal.auto_detect() @pyqtSlot() def on_btn_replay_clicked(self): pmngr = self.project_manager dialog = SendRecvDialogController(pmngr.frequency, pmngr.sample_rate, pmngr.bandwidth, pmngr.gain, pmngr.device, Mode.send, modulated_data=self.signal.data, parent=self) if dialog.has_empty_device_list: Errors.no_device() dialog.close() return dialog.recording_parameters.connect(pmngr.set_recording_parameters) dialog.show() def update_nselected_samples(self): self.ui.lNumSelectedSamples.setText(str(abs(int(self.ui.gvSignal.selection_area.width)))) self.__set_duration() @pyqtSlot(int, int) def update_selection_area(self, start, end): # if start > end: # start, end = end, start self.ui.lNumSelectedSamples.setText(str(end - start)) self.__set_duration() self.ui.spinBoxSelectionStart.blockSignals(True) self.ui.spinBoxSelectionStart.setValue(start) self.ui.spinBoxSelectionStart.blockSignals(False) self.ui.spinBoxSelectionEnd.blockSignals(True) self.ui.spinBoxSelectionEnd.setValue(end) self.ui.spinBoxSelectionEnd.blockSignals(False) def change_signal_name(self): self.signal.name = self.ui.lineEditSignalName.text()
class ZoomAndDropableGraphicView(ZoomableGraphicView): signal_loaded = pyqtSignal(ProtocolAnalyzer) def __init__(self, parent=None): self.scene_creator = None """:type: SignalSceneManager """ self.signal_tree_root = None """type signal_tree_root: ProtocolTreeItem""" self.signal = None self.proto_analyzer = None super().__init__(parent) def dragEnterEvent(self, QDragEnterEvent): QDragEnterEvent.acceptProposedAction() def dropEvent(self, event: QDropEvent): mimedata = event.mimeData() data_str = str(mimedata.text()) indexes = list(data_str.split("/")[:-1]) signal = None proto_analyzer = None for index in indexes: row, column, parent = map(int, index.split(",")) if parent == -1: parent = self.signal_tree_root else: parent = self.signal_tree_root.child(parent) node = parent.child(row) if node.protocol is not None and node.protocol.signal is not None: signal = node.protocol.signal proto_analyzer = node.protocol break if signal is None: return self.draw_signal(signal, proto_analyzer) def draw_signal(self, signal, proto_analyzer): if signal is None: return self.horizontalScrollBar().blockSignals(True) self.scene_creator = SignalSceneManager(signal, self) self.scene_creator.init_scene() self.setScene(self.scene_creator.scene) self.scene_creator.show_full_scene() self.fitInView(self.sceneRect()) self.signal_loaded.emit(proto_analyzer) self.signal = signal self.proto_analyzer = proto_analyzer self.horizontalScrollBar().blockSignals(False) def handle_signal_zoomed_or_scrolled(self): if self.scene_creator is not None: x1 = self.view_rect().x() x2 = x1 + self.view_rect().width() self.scene_creator.show_scene_section(x1, x2)
def __init__(self, freq, samp_rate, bw, gain, device: str, mode: Mode, modulated_data=None, parent=None): super().__init__(parent) self.ui = Ui_SendRecvDialog() self.ui.setupUi(self) self.setAttribute(Qt.WA_DeleteOnClose) self.graphics_view = self.ui.graphicsViewSend if mode == Mode.send else self.ui.graphicsViewReceive self.backend_handler = BackendHandler() if mode == Mode.spectrum: self.update_interval = 1 else: self.update_interval = 50 if mode == Mode.send and modulated_data is None: raise ValueError("I need modulated data to send!") if mode == Mode.receive or mode == Mode.spectrum: self.ui.spinBoxNRepeat.hide() self.ui.labelNRepeat.hide() self.ui.lblCurrentRepeatValue.hide() self.ui.lblRepeatText.hide() self.ui.lSamplesSentText.hide() self.ui.progressBar.hide() self.ui.stackedWidget.setCurrentIndex(0) else: self.ui.stackedWidget.setCurrentIndex(1) if mode == Mode.send or mode == Mode.spectrum: self.ui.lSamplesCaptured.hide() self.ui.lSamplesCapturedText.hide() self.ui.lSignalSize.hide() self.ui.lSignalSizeText.hide() self.ui.lTime.hide() self.ui.lTimeText.hide() self.ui.btnSave.hide() if mode == Mode.spectrum: self.setWindowTitle("Spectrum analyzer") if mode == Mode.send: self.ui.btnStart.setIcon(QIcon.fromTheme("media-playback-start")) self.setWindowTitle("Send signal") self.ui.btnStart.setToolTip("Send data") self.ui.btnStop.setToolTip("Stop sending") self.ui.progressBar.setMaximum(len(modulated_data)) self.device_is_sending = False self.ui.btnStop.setEnabled(False) self.ui.btnClear.setEnabled(False) self.ui.btnSave.setEnabled(False) self.start = 0 self.already_saved = True self.bw_sr_are_locked = constants.SETTINGS.value("lock_bandwidth_sample_rate", True, bool) self.ui.spinBoxFreq.setValue(freq) self.ui.spinBoxSampleRate.setValue(samp_rate) self.ui.spinBoxBandwidth.setValue(bw) self.ui.spinBoxGain.setValue(gain) self.ui.spinBoxNRepeat.setValue(constants.SETTINGS.value('num_sending_repeats', 1, type=int)) self.ui.cbDevice.clear() items = [] for device_name in self.backend_handler.DEVICE_NAMES: dev = self.backend_handler.device_backends[device_name.lower()] if mode == Mode.send and dev.is_enabled and dev.supports_tx: items.append(device_name) elif mode in (Mode.receive, Mode.spectrum) and dev.is_enabled and dev.supports_rx: items.append(device_name) if mode == Mode.send and PluginManager().is_plugin_enabled("NetworkSDRInterface"): items.append(NetworkSDRInterfacePlugin.NETWORK_SDR_NAME) self.ui.cbDevice.addItems(items) if device in items: self.ui.cbDevice.setCurrentIndex(items.index(device)) dev_name = self.ui.cbDevice.currentText() nrep = self.ui.spinBoxNRepeat.value() self.device = VirtualDevice(self.backend_handler, dev_name, mode, bw, freq, gain, samp_rate, samples_to_send=modulated_data, device_ip=self.ui.lineEditIP.text(), sending_repeats=nrep, parent=self) self.ui.lineEditIP.setVisible(dev_name == "USRP") self.ui.labelIP.setVisible(dev_name == "USRP") self.recorded_files = [] self.timer = QTimer(self) if mode == Mode.receive: self.scene_manager = LiveSceneManager(np.array([]), parent=self) # set really in on_device_started elif mode == Mode.send: signal = Signal.from_samples(modulated_data, "Modulated Preview", samp_rate) self.scene_manager = SignalSceneManager(signal, parent=self) self.send_indicator = self.scene_manager.scene.addRect(0, -2, 0, 4, QPen(QColor(Qt.transparent), Qt.FlatCap), QBrush(constants.SEND_INDICATOR_COLOR)) self.send_indicator.stackBefore(self.scene_manager.scene.selection_area) self.scene_manager.init_scene() self.graphics_view.set_signal(signal) self.graphics_view.sample_rate = samp_rate else: self.scene_manager = FFTSceneManager(parent=self, graphic_view=self.graphics_view) self.graphics_view.setScene(self.scene_manager.scene) self.graphics_view.scene_manager = self.scene_manager ipRange = "(?:[0-1]?[0-9]?[0-9]|2[0-4][0-9]|25[0-5])" ipRegex = QRegExp("^" + ipRange + "\\." + ipRange + "\\." + ipRange + "\\." + ipRange + "$") self.ui.lineEditIP.setValidator(QRegExpValidator(ipRegex)) self.create_connects() self.ui.btnLockBWSR.setChecked(self.bw_sr_are_locked) self.on_btn_lock_bw_sr_clicked()
class SendDialogController(SendRecvDialogController): def __init__(self, project_manager, modulated_data, parent=None, testing_mode=False): super().__init__(project_manager, is_tx=True, parent=parent, testing_mode=testing_mode) self.graphics_view = self.ui.graphicsViewSend self.ui.stackedWidget.setCurrentWidget(self.ui.page_send) self.hide_receive_ui_items() self.ui.btnStart.setIcon(QIcon.fromTheme("media-playback-start")) self.setWindowTitle("Send Signal") self.setWindowIcon(QIcon.fromTheme("media-playback-start")) self.ui.btnStart.setToolTip("Send data") self.ui.btnStop.setToolTip("Stop sending") self.device_is_sending = False if modulated_data is not None: # modulated_data is none in continuous send mode self.ui.progressBar.setMaximum(len(modulated_data)) samp_rate = self.ui.spinBoxSampleRate.value() signal = Signal.from_samples(modulated_data, "Modulated Preview", samp_rate) self.scene_manager = SignalSceneManager(signal, parent=self) self.send_indicator = self.scene_manager.scene.addRect(0, -2, 0, 4, QPen(QColor(Qt.transparent), Qt.FlatCap), QBrush(constants.SEND_INDICATOR_COLOR)) self.send_indicator.stackBefore(self.scene_manager.scene.selection_area) self.scene_manager.init_scene() self.graphics_view.set_signal(signal) self.graphics_view.sample_rate = samp_rate self.init_device() self.graphics_view.setScene(self.scene_manager.scene) self.graphics_view.scene_manager = self.scene_manager self.create_connects() def create_connects(self): super().create_connects() self.graphics_view.save_as_clicked.connect(self.on_graphics_view_save_as_clicked) self.scene_manager.signal.data_edited.connect(self.on_signal_data_edited) def _update_send_indicator(self, width: int): y, h = self.ui.graphicsViewSend.view_rect().y(), self.ui.graphicsViewSend.view_rect().height() self.send_indicator.setRect(0, y - h, width, 2 * h + abs(y)) def update_view(self): if super().update_view(): self._update_send_indicator(self.device.current_index) if not self.device.sending_finished: self.ui.lblCurrentRepeatValue.setText(str(self.device.current_iteration + 1)) else: self.ui.lblCurrentRepeatValue.setText("Done") def init_device(self): device_name = self.ui.cbDevice.currentText() num_repeats = self.ui.spinBoxNRepeat.value() sts = self.scene_manager.signal._fulldata self.device = VirtualDevice(self.backend_handler, device_name, Mode.send, samples_to_send=sts, device_ip="192.168.10.2", sending_repeats=num_repeats, parent=self) self._create_device_connects() @pyqtSlot() def on_graphics_view_save_as_clicked(self): filename = FileOperator.get_save_file_name("signal.complex") if filename: try: self.scene_manager.signal.save_as(filename) except Exception as e: QMessageBox.critical(self, self.tr("Error saving signal"), e.args[0]) @pyqtSlot() def on_signal_data_edited(self): signal = self.scene_manager.signal self.ui.progressBar.setMaximum(signal.num_samples) self.device.samples_to_send = signal.data self.scene_manager.init_scene() self.ui.graphicsViewSend.redraw_view() @pyqtSlot() def on_start_clicked(self): super().on_start_clicked() if self.ui.progressBar.value() >= self.ui.progressBar.maximum() - 1: self.on_clear_clicked() if self.device_is_sending: self.device.stop("Sending paused by user") else: self.device.start() @pyqtSlot() def on_stop_clicked(self): self.device.stop("Stopped receiving: Stop button clicked") super().on_stop_clicked() self.on_clear_clicked() @pyqtSlot() def on_device_stopped(self): super().on_device_stopped() self.ui.btnStart.setIcon(QIcon.fromTheme("media-playback-start")) self.ui.btnStart.setToolTip("Start sending") self.device_is_sending = False @pyqtSlot() def on_device_started(self): super().on_device_started() self.device_is_sending = True self.ui.btnStart.setEnabled(True) self.ui.btnStart.setIcon(QIcon.fromTheme("media-playback-pause")) self.set_device_ui_items_enabled(False) @pyqtSlot() def on_clear_clicked(self): self._update_send_indicator(0) self.reset()