def __init__(self, compare_frame_controller: CompareFrameController, project_manager: ProjectManager, parent=None): super().__init__(parent) self.ui = Ui_GeneratorTab() self.ui.setupUi(self) self.project_manager = project_manager self.ui.treeProtocols.setHeaderHidden(True) self.tree_model = GeneratorTreeModel(compare_frame_controller) self.tree_model.set_root_item(compare_frame_controller.proto_tree_model.rootItem) self.tree_model.controller = self self.ui.treeProtocols.setModel(self.tree_model) self.table_model = GeneratorTableModel(compare_frame_controller.proto_tree_model.rootItem, [Modulator("Modulation")], compare_frame_controller.decodings) self.table_model.controller = self self.ui.tableMessages.setModel(self.table_model) self.label_list_model = GeneratorListModel(None) self.ui.listViewProtoLabels.setModel(self.label_list_model) self.network_sdr_button_orig_tooltip = self.ui.btnNetworkSDRSend.toolTip() self.set_network_sdr_send_button_visibility() self.set_rfcat_button_visibility() self.network_sdr_plugin = NetworkSDRInterfacePlugin() self.rfcat_plugin = RfCatPlugin() self.init_rfcat_plugin() self.refresh_modulators() self.on_selected_modulation_changed() self.set_fuzzing_ui_status() self.ui.prBarGeneration.hide() self.create_connects(compare_frame_controller)
def init_rfcat_plugin(self): self.set_rfcat_button_visibility() self.rfcat_plugin = RfCatPlugin() self.rfcat_plugin.current_send_message_changed.connect( self.on_send_message_changed) self.ui.btnRfCatSend.setEnabled(self.rfcat_plugin.rfcat_is_found)