class SniffSettingsWidget(QWidget): sniff_setting_edited = pyqtSignal() sniff_file_edited = pyqtSignal() sniff_parameters_changed = pyqtSignal(dict) def __init__(self, device_name: str, project_manager: ProjectManager, signal=None, backend_handler=None, network_raw_mode=False, signals=None, parent=None): super().__init__(parent) self.ui = Ui_SniffSettings() self.ui.setupUi(self) signals = signals if signals is not None else [] self.project_manager = project_manager for encoding in self.project_manager.decodings: self.ui.comboBox_sniff_encoding.addItem(encoding.name) self.bootstrap(project_manager.device_conf, signal, enforce_default=True) self.sniffer = ProtocolSniffer(samples_per_symbol=self.ui.spinbox_sniff_SamplesPerSymbol.value(), center=self.ui.spinbox_sniff_Center.value(), center_spacing=self.ui.spinBoxCenterSpacing.value(), noise=self.ui.spinbox_sniff_Noise.value(), tolerance=self.ui.spinbox_sniff_ErrorTolerance.value(), modulation_type=self.ui.combox_sniff_Modulation.currentText(), bits_per_symbol=self.ui.spinBoxBitsPerSymbol.value(), device=device_name, backend_handler=BackendHandler() if backend_handler is None else backend_handler, network_raw_mode=network_raw_mode) self.sniffer.adaptive_noise = self.ui.checkBoxAdaptiveNoise.isChecked() self.sniffer.automatic_center = self.ui.checkBoxAutoCenter.isChecked() self.__set_center_offset_visibility() self.create_connects() self.ui.comboBox_sniff_encoding.currentIndexChanged.emit(self.ui.comboBox_sniff_encoding.currentIndex()) self.ui.comboBox_sniff_viewtype.setCurrentIndex(settings.read('default_view', 0, int)) # Auto Complete like a Boss completer = QCompleter() completer.setModel(QDirModel(completer)) self.ui.lineEdit_sniff_OutputFile.setCompleter(completer) self.signals = signals if len(signals) == 0: self.ui.label_sniff_Signal.hide() self.ui.btn_sniff_use_signal.hide() self.ui.comboBox_sniff_signal.hide() else: for signal in signals: self.ui.comboBox_sniff_signal.addItem(signal.name) def __set_center_offset_visibility(self): visible = self.ui.spinBoxBitsPerSymbol.value() > 1 self.ui.labelCenterSpacing.setVisible(visible) self.ui.spinBoxCenterSpacing.setVisible(visible) def bootstrap(self, conf_dict: dict, signal=None, enforce_default=False): def set_val(widget, key: str, default): try: value = conf_dict[key] except KeyError: value = default if enforce_default else None if value is not None: if hasattr(widget, "setValue"): widget.setValue(value) elif hasattr(widget, "setCurrentIndex"): widget.setCurrentIndex(value) set_val(self.ui.spinbox_sniff_SamplesPerSymbol, "samples_per_symbol", signal.samples_per_symbol if signal else 100) set_val(self.ui.spinbox_sniff_Center, "center", signal.center if signal else 0.02) set_val(self.ui.spinBoxCenterSpacing, "center_spacing", signal.center_spacing if signal else 0.1) set_val(self.ui.spinbox_sniff_ErrorTolerance, "tolerance", signal.tolerance if signal else 5) set_val(self.ui.spinbox_sniff_Noise, "noise", signal.noise_threshold_relative if signal else 0.001) self.ui.combox_sniff_Modulation.setCurrentText(conf_dict.get("modulation_type", signal.modulation_type if signal else "FSK")) set_val(self.ui.spinBoxBitsPerSymbol, "bits_per_symbol", signal.bits_per_symbol if signal else 1) self.ui.comboBox_sniff_encoding.setCurrentText(conf_dict.get("decoding_name", "")) self.ui.checkBoxAdaptiveNoise.setChecked(bool(conf_dict.get("adaptive_noise", False))) self.ui.checkBoxAutoCenter.setChecked(bool(conf_dict.get("automatic_center", False))) self.ui.spinbox_sniff_Center.setDisabled(self.ui.checkBoxAutoCenter.isChecked()) self.emit_editing_finished_signals() def create_connects(self): self.ui.spinbox_sniff_Noise.editingFinished.connect(self.on_noise_edited) self.ui.spinbox_sniff_Center.editingFinished.connect(self.on_center_edited) self.ui.spinBoxCenterSpacing.editingFinished.connect(self.on_center_spacing_edited) self.ui.spinbox_sniff_SamplesPerSymbol.editingFinished.connect(self.on_samples_per_symbol_edited) self.ui.spinbox_sniff_ErrorTolerance.editingFinished.connect(self.on_tolerance_edited) self.ui.combox_sniff_Modulation.currentTextChanged.connect(self.on_modulation_changed) self.ui.spinBoxBitsPerSymbol.editingFinished.connect(self.on_spin_box_bits_per_symbol_editing_finished) self.ui.comboBox_sniff_viewtype.currentIndexChanged.connect(self.on_view_type_changed) self.ui.lineEdit_sniff_OutputFile.editingFinished.connect(self.on_line_edit_output_file_editing_finished) self.ui.comboBox_sniff_encoding.currentIndexChanged.connect(self.on_combobox_sniff_encoding_index_changed) self.ui.checkBox_sniff_Timestamp.clicked.connect(self.on_checkbox_sniff_timestamp_clicked) self.ui.btn_sniff_use_signal.clicked.connect(self.on_btn_sniff_use_signal_clicked) self.ui.checkBoxAdaptiveNoise.clicked.connect(self.on_check_box_adaptive_noise_clicked) self.ui.checkBoxAutoCenter.clicked.connect(self.on_check_box_auto_center_clicked) def emit_editing_finished_signals(self): self.ui.spinbox_sniff_Noise.editingFinished.emit() self.ui.spinbox_sniff_Center.editingFinished.emit() self.ui.spinBoxCenterSpacing.editingFinished.emit() self.ui.spinbox_sniff_SamplesPerSymbol.editingFinished.emit() self.ui.spinBoxBitsPerSymbol.editingFinished.emit() self.ui.spinbox_sniff_ErrorTolerance.editingFinished.emit() self.ui.lineEdit_sniff_OutputFile.editingFinished.emit() self.ui.checkBoxAdaptiveNoise.clicked.emit() def emit_sniff_parameters_changed(self): self.sniff_parameters_changed.emit(dict(samples_per_symbol=self.sniffer.signal.samples_per_symbol, center=self.sniffer.signal.center, center_spacing=self.sniffer.signal.center_spacing, noise=self.sniffer.signal.noise_threshold, tolerance=self.sniffer.signal.tolerance, modulation_type=self.sniffer.signal.modulation_type, bits_per_symbol=self.sniffer.signal.bits_per_symbol, decoding_name=self.sniffer.decoder.name, adaptive_noise=self.sniffer.adaptive_noise, automatic_center=self.sniffer.automatic_center)) @pyqtSlot() def on_noise_edited(self): self.sniffer.signal.noise_threshold_relative = self.ui.spinbox_sniff_Noise.value() self.sniff_setting_edited.emit() @pyqtSlot() def on_center_edited(self): self.sniffer.signal.center = self.ui.spinbox_sniff_Center.value() self.sniff_setting_edited.emit() @pyqtSlot() def on_center_spacing_edited(self): self.sniffer.signal.center_spacing = self.ui.spinBoxCenterSpacing.value() self.sniff_setting_edited.emit() @pyqtSlot() def on_samples_per_symbol_edited(self): self.sniffer.signal.samples_per_symbol = self.ui.spinbox_sniff_SamplesPerSymbol.value() self.sniff_setting_edited.emit() @pyqtSlot() def on_spin_box_bits_per_symbol_editing_finished(self): self.sniffer.signal.bits_per_symbol = self.ui.spinBoxBitsPerSymbol.value() self.__set_center_offset_visibility() self.sniff_setting_edited.emit() @pyqtSlot() def on_tolerance_edited(self): self.sniffer.signal.tolerance = self.ui.spinbox_sniff_ErrorTolerance.value() self.sniff_setting_edited.emit() @pyqtSlot(str) def on_modulation_changed(self, new_modulation: str): self.sniffer.signal.silent_set_modulation_type(new_modulation) self.sniff_setting_edited.emit() @pyqtSlot() def on_view_type_changed(self): self.sniff_setting_edited.emit() @pyqtSlot(int) def on_combobox_sniff_encoding_index_changed(self, index: int): if self.sniffer.decoder != self.project_manager.decodings[index]: self.sniffer.set_decoder_for_messages(self.project_manager.decodings[index]) self.sniffer.decoder = self.project_manager.decodings[index] self.sniff_setting_edited.emit() @pyqtSlot() def on_line_edit_output_file_editing_finished(self): self.ui.lineEdit_sniff_OutputFile.setStyleSheet("") text = self.ui.lineEdit_sniff_OutputFile.text() if text and not text.endswith(".txt"): text += ".txt" self.ui.lineEdit_sniff_OutputFile.setText(text) if text and not os.path.isfile(text): try: open(text, "w").close() except Exception as e: self.ui.lineEdit_sniff_OutputFile.setStyleSheet("color:red;") return self.sniffer.sniff_file = text self.sniff_file_edited.emit() @pyqtSlot() def on_checkbox_sniff_timestamp_clicked(self): self.sniff_setting_edited.emit() @pyqtSlot() def on_btn_sniff_use_signal_clicked(self): try: signal = self.signals[self.ui.comboBox_sniff_signal.currentIndex()] except IndexError: return self.ui.spinbox_sniff_SamplesPerSymbol.setValue(signal.samples_per_symbol) self.ui.spinbox_sniff_Center.setValue(signal.center) self.ui.spinbox_sniff_Noise.setValue(signal.noise_threshold_relative) self.ui.spinbox_sniff_ErrorTolerance.setValue(signal.tolerance) self.ui.combox_sniff_Modulation.setCurrentText(signal.modulation_type) self.emit_editing_finished_signals() @pyqtSlot() def on_check_box_adaptive_noise_clicked(self): self.sniffer.adaptive_noise = self.ui.checkBoxAdaptiveNoise.isChecked() @pyqtSlot() def on_check_box_auto_center_clicked(self): self.sniffer.automatic_center = self.ui.checkBoxAutoCenter.isChecked() self.ui.spinbox_sniff_Center.setDisabled(self.ui.checkBoxAutoCenter.isChecked())
class SniffSettingsWidget(QWidget): sniff_setting_edited = pyqtSignal() sniff_file_edited = pyqtSignal() sniff_parameters_changed = pyqtSignal(dict) def __init__(self, device_name: str, project_manager: ProjectManager, signal=None, backend_handler=None, network_raw_mode=False, signals=None, parent=None): super().__init__(parent) self.ui = Ui_SniffSettings() self.ui.setupUi(self) signals = signals if signals is not None else [] self.project_manager = project_manager for encoding in self.project_manager.decodings: self.ui.comboBox_sniff_encoding.addItem(encoding.name) self.bootstrap(project_manager.device_conf, signal, enforce_default=True) self.sniffer = ProtocolSniffer(bit_len=self.ui.spinbox_sniff_BitLen.value(), center=self.ui.spinbox_sniff_Center.value(), noise=self.ui.spinbox_sniff_Noise.value(), tolerance=self.ui.spinbox_sniff_ErrorTolerance.value(), modulation_type=self.ui.combox_sniff_Modulation.currentIndex(), device=device_name, backend_handler=BackendHandler() if backend_handler is None else backend_handler, network_raw_mode=network_raw_mode) self.create_connects() self.ui.comboBox_sniff_encoding.currentIndexChanged.emit(self.ui.comboBox_sniff_encoding.currentIndex()) self.ui.comboBox_sniff_viewtype.setCurrentIndex(constants.SETTINGS.value('default_view', 0, int)) # Auto Complete like a Boss completer = QCompleter() completer.setModel(QDirModel(completer)) self.ui.lineEdit_sniff_OutputFile.setCompleter(completer) self.signals = signals if len(signals) == 0: self.ui.label_sniff_Signal.hide() self.ui.btn_sniff_use_signal.hide() self.ui.comboBox_sniff_signal.hide() else: for signal in signals: self.ui.comboBox_sniff_signal.addItem(signal.name) def bootstrap(self, conf_dict: dict, signal=None, enforce_default=False): def set_val(widget, key: str, default): try: value = conf_dict[key] except KeyError: value = default if enforce_default else None if value is not None: if hasattr(widget, "setValue"): widget.setValue(value) elif hasattr(widget, "setCurrentIndex"): widget.setCurrentIndex(value) set_val(self.ui.spinbox_sniff_BitLen, "bit_len", signal.bit_len if signal else 100) set_val(self.ui.spinbox_sniff_Center, "center", signal.qad_center if signal else 0.02) set_val(self.ui.spinbox_sniff_ErrorTolerance, "tolerance", signal.tolerance if signal else 5) set_val(self.ui.spinbox_sniff_Noise, "noise", signal.noise_threshold if signal else 0.001) set_val(self.ui.combox_sniff_Modulation, "modulation_index", signal.modulation_type if signal else 1) self.ui.comboBox_sniff_encoding.setCurrentText(conf_dict.get("decoding_name", "")) self.ui.checkBoxAdaptiveNoise.setChecked(bool(conf_dict.get("adaptive_noise", False))) self.emit_editing_finished_signals() def create_connects(self): self.ui.spinbox_sniff_Noise.editingFinished.connect(self.on_noise_edited) self.ui.spinbox_sniff_Center.editingFinished.connect(self.on_center_edited) self.ui.spinbox_sniff_BitLen.editingFinished.connect(self.on_bit_len_edited) self.ui.spinbox_sniff_ErrorTolerance.editingFinished.connect(self.on_tolerance_edited) self.ui.combox_sniff_Modulation.currentIndexChanged.connect(self.on_modulation_changed) self.ui.comboBox_sniff_viewtype.currentIndexChanged.connect(self.on_view_type_changed) self.ui.lineEdit_sniff_OutputFile.editingFinished.connect(self.on_line_edit_output_file_editing_finished) self.ui.comboBox_sniff_encoding.currentIndexChanged.connect(self.on_combobox_sniff_encoding_index_changed) self.ui.checkBox_sniff_Timestamp.clicked.connect(self.on_checkbox_sniff_timestamp_clicked) self.ui.btn_sniff_use_signal.clicked.connect(self.on_btn_sniff_use_signal_clicked) self.ui.checkBoxAdaptiveNoise.clicked.connect(self.on_check_box_adaptive_noise_clicked) def emit_editing_finished_signals(self): self.ui.spinbox_sniff_Noise.editingFinished.emit() self.ui.spinbox_sniff_Center.editingFinished.emit() self.ui.spinbox_sniff_BitLen.editingFinished.emit() self.ui.spinbox_sniff_ErrorTolerance.editingFinished.emit() self.ui.lineEdit_sniff_OutputFile.editingFinished.emit() self.ui.checkBoxAdaptiveNoise.clicked.emit() def emit_sniff_parameters_changed(self): self.sniff_parameters_changed.emit(dict(bit_len=self.sniffer.signal.bit_len, center=self.sniffer.signal.qad_center, noise=self.sniffer.signal.noise_threshold, tolerance=self.sniffer.signal.tolerance, modulation_index=self.sniffer.signal.modulation_type, decoding_name=self.sniffer.decoder.name, adaptive_noise=self.sniffer.adaptive_noise)) @pyqtSlot() def on_noise_edited(self): self.sniffer.signal._noise_threshold = self.ui.spinbox_sniff_Noise.value() self.sniff_setting_edited.emit() @pyqtSlot() def on_center_edited(self): self.sniffer.signal.qad_center = self.ui.spinbox_sniff_Center.value() self.sniff_setting_edited.emit() @pyqtSlot() def on_bit_len_edited(self): self.sniffer.signal.bit_len = self.ui.spinbox_sniff_BitLen.value() self.sniff_setting_edited.emit() @pyqtSlot() def on_tolerance_edited(self): self.sniffer.signal.tolerance = self.ui.spinbox_sniff_ErrorTolerance.value() self.sniff_setting_edited.emit() @pyqtSlot(int) def on_modulation_changed(self, new_index: int): self.sniffer.signal.silent_set_modulation_type(new_index) self.sniff_setting_edited.emit() @pyqtSlot() def on_view_type_changed(self): self.sniff_setting_edited.emit() @pyqtSlot(int) def on_combobox_sniff_encoding_index_changed(self, index: int): if self.sniffer.decoder != self.project_manager.decodings[index]: self.sniffer.set_decoder_for_messages(self.project_manager.decodings[index]) self.sniffer.decoder = self.project_manager.decodings[index] self.sniff_setting_edited.emit() @pyqtSlot() def on_line_edit_output_file_editing_finished(self): self.ui.lineEdit_sniff_OutputFile.setStyleSheet("") text = self.ui.lineEdit_sniff_OutputFile.text() if text and not text.endswith(".txt"): text += ".txt" self.ui.lineEdit_sniff_OutputFile.setText(text) if text and not os.path.isfile(text): try: open(text, "w").close() except Exception as e: self.ui.lineEdit_sniff_OutputFile.setStyleSheet("color:red;") return self.sniffer.sniff_file = text self.sniff_file_edited.emit() @pyqtSlot() def on_checkbox_sniff_timestamp_clicked(self): self.sniff_setting_edited.emit() @pyqtSlot() def on_btn_sniff_use_signal_clicked(self): try: signal = self.signals[self.ui.comboBox_sniff_signal.currentIndex()] except IndexError: return self.ui.spinbox_sniff_BitLen.setValue(signal.bit_len) self.ui.spinbox_sniff_Center.setValue(signal.qad_center) self.ui.spinbox_sniff_Noise.setValue(signal.noise_threshold) self.ui.spinbox_sniff_ErrorTolerance.setValue(signal.tolerance) self.ui.combox_sniff_Modulation.setCurrentIndex(signal.modulation_type) self.emit_editing_finished_signals() @pyqtSlot() def on_check_box_adaptive_noise_clicked(self): self.sniffer.adaptive_noise = self.ui.checkBoxAdaptiveNoise.isChecked()
class ProtocolSniffDialogController(SendRecvDialogController): protocol_accepted = pyqtSignal(list) def __init__(self, project_manager, noise, center, bit_length, tolerance, modulation_type_index, encodings, encoding_index=0, parent=None, testing_mode=False): super().__init__(project_manager, is_tx=False, parent=parent, testing_mode=testing_mode) self.set_sniff_ui_items_visible(True) self.graphics_view = self.ui.graphicsView_sniff_Preview self.ui.stackedWidget.setCurrentWidget(self.ui.page_sniff) self.hide_send_ui_items() self.hide_receive_ui_items() self.ui.sliderYscale.hide() self.ui.label_y_scale.hide() self.ui.spinbox_sniff_Noise.setValue(noise) self.ui.spinbox_sniff_Center.setValue(center) self.ui.spinbox_sniff_BitLen.setValue(bit_length) self.ui.spinbox_sniff_ErrorTolerance.setValue(tolerance) self.ui.combox_sniff_Modulation.setCurrentIndex(modulation_type_index) self.sniffer = ProtocolSniffer(bit_length, center, noise, tolerance, modulation_type_index, self.ui.cbDevice.currentText(), self.backend_handler) # set really in on_device_started self.scene_manager = None # type: LiveSceneManager self.init_device() self.set_bandwidth_status() self.graphics_view.setScene(self.scene_manager.scene) self.graphics_view.scene_manager = self.scene_manager # Auto Complete like a Boss completer = QCompleter() completer.setModel(QDirModel(completer)) self.ui.lineEdit_sniff_OutputFile.setCompleter(completer) self.setWindowTitle(self.tr("Sniff Protocol")) self.setWindowIcon(QIcon.fromTheme(":/icons/data/icons/sniffer.svg")) self.encodings = encodings for encoding in self.encodings: self.ui.comboBox_sniff_encoding.addItem(encoding.name) self.create_connects() if encoding_index > -1: self.ui.comboBox_sniff_encoding.setCurrentIndex(encoding_index) self.ui.comboBox_sniff_viewtype.setCurrentIndex( constants.SETTINGS.value('default_view', 0, int)) @property def device(self): if hasattr(self, "sniffer"): return self.sniffer.rcv_device else: return None @device.setter def device(self, value): if hasattr(self, "sniffer"): self.sniffer.rcv_device = value else: pass @property def view_type(self): return self.ui.comboBox_sniff_viewtype.currentIndex() def create_connects(self): super().create_connects() self.ui.btnAccept.clicked.connect(self.on_btn_accept_clicked) self.sniffer.qt_signals.data_sniffed.connect(self.on_data_sniffed) self.sniffer.qt_signals.sniff_device_errors_changed.connect( self.on_device_errors_changed) self.ui.spinbox_sniff_Noise.editingFinished.connect( self.on_noise_edited) self.ui.spinbox_sniff_Center.editingFinished.connect( self.on_center_edited) self.ui.spinbox_sniff_BitLen.editingFinished.connect( self.on_bit_len_edited) self.ui.spinbox_sniff_ErrorTolerance.editingFinished.connect( self.on_tolerance_edited) self.ui.combox_sniff_Modulation.currentIndexChanged.connect( self.on_modulation_changed) self.ui.comboBox_sniff_viewtype.currentIndexChanged.connect( self.on_view_type_changed) self.ui.lineEdit_sniff_OutputFile.editingFinished.connect( self.on_line_edit_output_file_editing_finished) self.ui.comboBox_sniff_encoding.currentIndexChanged.connect( self.on_combobox_sniff_encoding_index_changed) def set_device_ui_items_visibility(self, device_name: str, adjust_gains=True): super().set_device_ui_items_visibility(device_name, adjust_gains) visible = device_name != NetworkSDRInterfacePlugin.NETWORK_SDR_NAME for item in ("spinbox_sniff_Noise", "combox_sniff_Modulation", "label_sniff_Modulation", "graphicsView_sniff_Preview", "spinbox_sniff_Center", "spinbox_sniff_BitLen", "spinbox_sniff_ErrorTolerance", "label_sniff_Noise", "label_sniff_Center", "label_sniff_BitLength", "label_sniff_Tolerance"): getattr(self.ui, item).setVisible(visible) def init_device(self): dev_name = self.ui.cbDevice.currentText() self.sniffer.device_name = dev_name self._create_device_connects() self.scene_manager = SniffSceneManager(np.array([]), parent=self) def emit_editing_finished_signals(self): super().emit_editing_finished_signals() self.ui.spinbox_sniff_Noise.editingFinished.emit() self.ui.spinbox_sniff_Center.editingFinished.emit() self.ui.spinbox_sniff_BitLen.editingFinished.emit() self.ui.spinbox_sniff_ErrorTolerance.editingFinished.emit() def update_view(self): if super().update_view(): self.scene_manager.end = self.device.current_index self.scene_manager.init_scene() self.scene_manager.show_full_scene() self.graphics_view.update() @pyqtSlot() def on_device_started(self): self.scene_manager.data_array = self.device.data.real if hasattr( self.device.data, "real") else None super().on_device_started() self.ui.btnStart.setEnabled(False) self.set_device_ui_items_enabled(False) @pyqtSlot() def on_noise_edited(self): self.sniffer.signal._noise_threshold = self.ui.spinbox_sniff_Noise.value( ) self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(self.view_type)) @pyqtSlot() def on_center_edited(self): self.sniffer.signal.qad_center = self.ui.spinbox_sniff_Center.value() self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(self.view_type)) @pyqtSlot() def on_bit_len_edited(self): self.sniffer.signal.bit_len = self.ui.spinbox_sniff_BitLen.value() self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(self.view_type)) @pyqtSlot() def on_tolerance_edited(self): self.sniffer.signal.tolerance = self.ui.spinbox_sniff_ErrorTolerance.value( ) self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(self.view_type)) @pyqtSlot(int) def on_modulation_changed(self, new_index: int): self.sniffer.signal.silent_set_modulation_type(new_index) self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(self.view_type)) @pyqtSlot() def on_start_clicked(self): super().on_start_clicked() self.sniffer.sniff() @pyqtSlot() def on_stop_clicked(self): self.sniffer.stop() @pyqtSlot() def on_clear_clicked(self): self.ui.btnClear.setEnabled(False) self.ui.txtEd_sniff_Preview.clear() self.scene_manager.clear_path() self.device.current_index = 0 self.sniffer.clear() @pyqtSlot(int) def on_data_sniffed(self, from_index: int): new_data = self.sniffer.decoded_to_string(self.view_type, start=from_index) if new_data.strip(): self.ui.txtEd_sniff_Preview.appendPlainText(new_data) self.ui.txtEd_sniff_Preview.verticalScrollBar().setValue( self.ui.txtEd_sniff_Preview.verticalScrollBar().maximum()) @pyqtSlot(int) def on_view_type_changed(self, new_index: int): self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(new_index)) @pyqtSlot() def on_btn_accept_clicked(self): self.protocol_accepted.emit(self.sniffer.messages) self.close() @pyqtSlot(str) def on_device_errors_changed(self, txt: str): self.ui.txtEditErrors.append(txt) @pyqtSlot() def on_line_edit_output_file_editing_finished(self): text = self.ui.lineEdit_sniff_OutputFile.text() if text and not text.endswith(".txt"): text += ".txt" self.ui.lineEdit_sniff_OutputFile.setText(text) self.sniffer.sniff_file = text self.ui.btnAccept.setDisabled(bool(self.sniffer.sniff_file)) @pyqtSlot(int) def on_combobox_sniff_encoding_index_changed(self, index: int): if self.sniffer.decoder != self.encodings[index]: self.sniffer.set_decoder_for_messages(self.encodings[index]) self.sniffer.decoder = self.encodings[index] self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(self.view_type))
class SniffSettingsWidget(QWidget): sniff_setting_edited = pyqtSignal() sniff_file_edited = pyqtSignal() sniff_parameters_changed = pyqtSignal(dict) def __init__(self, device_name: str, project_manager: ProjectManager, signal=None, backend_handler=None, network_raw_mode=False, real_time=False, parent=None): super().__init__(parent) self.ui = Ui_SniffSettings() self.ui.setupUi(self) conf = project_manager.device_conf bit_length = conf.get("bit_len", signal.bit_len if signal else 100) modulation_type_index = conf.get("modulation_index", signal.modulation_type if signal else 1) tolerance = conf.get("tolerance", signal.tolerance if signal else 5) noise = conf.get("noise", signal.noise_threshold if signal else 0.001) center = conf.get("center", signal.qad_center if signal else 0.02) decoding_name = conf.get("decoding_name", "") self.sniffer = ProtocolSniffer(bit_len=bit_length, center=center, noise=noise, tolerance=tolerance, modulation_type=modulation_type_index, device=device_name, backend_handler=BackendHandler() if backend_handler is None else backend_handler, network_raw_mode=network_raw_mode, real_time=real_time) self.ui.spinbox_sniff_Noise.setValue(noise) self.ui.spinbox_sniff_Center.setValue(center) self.ui.spinbox_sniff_BitLen.setValue(bit_length) self.ui.spinbox_sniff_ErrorTolerance.setValue(tolerance) self.ui.combox_sniff_Modulation.setCurrentIndex(modulation_type_index) self.project_manager = project_manager for encoding in self.project_manager.decodings: self.ui.comboBox_sniff_encoding.addItem(encoding.name) self.create_connects() if decoding_name: self.ui.comboBox_sniff_encoding.setCurrentText(decoding_name) self.ui.comboBox_sniff_viewtype.setCurrentIndex(constants.SETTINGS.value('default_view', 0, int)) # Auto Complete like a Boss completer = QCompleter() completer.setModel(QDirModel(completer)) self.ui.lineEdit_sniff_OutputFile.setCompleter(completer) def create_connects(self): self.ui.spinbox_sniff_Noise.editingFinished.connect(self.on_noise_edited) self.ui.spinbox_sniff_Center.editingFinished.connect(self.on_center_edited) self.ui.spinbox_sniff_BitLen.editingFinished.connect(self.on_bit_len_edited) self.ui.spinbox_sniff_ErrorTolerance.editingFinished.connect(self.on_tolerance_edited) self.ui.combox_sniff_Modulation.currentIndexChanged.connect(self.on_modulation_changed) self.ui.comboBox_sniff_viewtype.currentIndexChanged.connect(self.on_view_type_changed) self.ui.lineEdit_sniff_OutputFile.editingFinished.connect(self.on_line_edit_output_file_editing_finished) self.ui.comboBox_sniff_encoding.currentIndexChanged.connect(self.on_combobox_sniff_encoding_index_changed) self.ui.checkBox_sniff_Timestamp.clicked.connect(self.on_checkbox_sniff_timestamp_clicked) def emit_editing_finished_signals(self): self.ui.spinbox_sniff_Noise.editingFinished.emit() self.ui.spinbox_sniff_Center.editingFinished.emit() self.ui.spinbox_sniff_BitLen.editingFinished.emit() self.ui.spinbox_sniff_ErrorTolerance.editingFinished.emit() self.ui.lineEdit_sniff_OutputFile.editingFinished.emit() def emit_sniff_parameters_changed(self): self.sniff_parameters_changed.emit(dict(bit_len=self.sniffer.signal.bit_len, center=self.sniffer.signal.qad_center, noise=self.sniffer.signal.noise_threshold, tolerance=self.sniffer.signal.tolerance, modulation_index=self.sniffer.signal.modulation_type, decoding_name=self.sniffer.decoder.name)) @pyqtSlot() def on_noise_edited(self): self.sniffer.signal._noise_threshold = self.ui.spinbox_sniff_Noise.value() self.sniff_setting_edited.emit() @pyqtSlot() def on_center_edited(self): self.sniffer.signal.qad_center = self.ui.spinbox_sniff_Center.value() self.sniff_setting_edited.emit() @pyqtSlot() def on_bit_len_edited(self): self.sniffer.signal.bit_len = self.ui.spinbox_sniff_BitLen.value() self.sniff_setting_edited.emit() @pyqtSlot() def on_tolerance_edited(self): self.sniffer.signal.tolerance = self.ui.spinbox_sniff_ErrorTolerance.value() self.sniff_setting_edited.emit() @pyqtSlot(int) def on_modulation_changed(self, new_index: int): self.sniffer.signal.silent_set_modulation_type(new_index) self.sniff_setting_edited.emit() @pyqtSlot() def on_view_type_changed(self): self.sniff_setting_edited.emit() @pyqtSlot(int) def on_combobox_sniff_encoding_index_changed(self, index: int): if self.sniffer.decoder != self.project_manager.decodings[index]: self.sniffer.set_decoder_for_messages(self.project_manager.decodings[index]) self.sniffer.decoder = self.project_manager.decodings[index] self.sniff_setting_edited.emit() @pyqtSlot() def on_line_edit_output_file_editing_finished(self): self.ui.lineEdit_sniff_OutputFile.setStyleSheet("") text = self.ui.lineEdit_sniff_OutputFile.text() if text and not text.endswith(".txt"): text += ".txt" self.ui.lineEdit_sniff_OutputFile.setText(text) if text and not os.path.isfile(text): try: open(text, "w").close() except Exception as e: self.ui.lineEdit_sniff_OutputFile.setStyleSheet("color:red;") return self.sniffer.sniff_file = text self.sniff_file_edited.emit() @pyqtSlot() def on_checkbox_sniff_timestamp_clicked(self): self.sniff_setting_edited.emit()