diff --git a/main.py b/main.py index ad3806b..5c4fe5b 100644 --- a/main.py +++ b/main.py @@ -41,6 +41,8 @@ class MyApp(QMainWindow, Ui_MainWindow): self.db_version = None self.db = None self.current_signal_name = '' + self.undefined_freq = False + self.undefined_band = False self.signal_names = [] self.category_labels = [self.cat_mil, self.cat_rad, @@ -80,21 +82,34 @@ class MyApp(QMainWindow, Ui_MainWindow): self.volume, self.audio_progress) + BandLabel = namedtuple("BandLabel", ["left", "center", "right"]) self.band_labels = [ - [self.elf, self.elf_s1, self.elf_s2], - [self.slf, self.slf_s1, self.slf_s2], - [self.ulf, self.ulf_s1, self.ulf_s2], - [self.vlf, self.vlf_s1, self.vlf_s2], - [self.lf , self.lf_s1 , self.lf_s2], - [self.mf , self.mf_s1 , self.mf_s2], - [self.hf , self.hf_s1 , self.hf_s2], - [self.vhf, self.vhf_s1, self.vhf_s2], - [self.uhf, self.uhf_s1, self.uhf_s2], - [self.shf, self.shf_s1, self.shf_s2], - [self.ehf, None, None], + BandLabel(self.elf_left, self.elf, self.elf_right), + BandLabel(self.slf_left, self.slf, self.slf_right), + BandLabel(self.ulf_left, self.ulf, self.ulf_right), + BandLabel(self.vlf_left, self.vlf, self.vlf_right), + BandLabel(self.lf_left, self.lf, self.lf_right), + BandLabel(self.mf_left, self.mf, self.mf_right), + BandLabel(self.hf_left, self.hf, self.hf_right), + BandLabel(self.vhf_left, self.vhf, self.vhf_right), + BandLabel(self.uhf_left, self.uhf, self.uhf_right), + BandLabel(self.shf_left, self.shf, self.shf_right), + BandLabel(self.ehf_left, self.ehf, self.ehf_right), ] def load_db(self): + names = ["name", + "inf_freq", + "sup_freq", + "mode", + "inf_band", + "sup_band", + "location", + "url", + "description", + "modulation", + "category_code", + "acf",] try: self.db = read_csv(os.path.join('Data', 'db.csv'), sep = '*', @@ -106,20 +121,8 @@ class MyApp(QMainWindow, Ui_MainWindow): 'inf_band': str, 'sup_band': str, 'category_code': str,}, - names = ["name", - "inf_freq", - "sup_freq", - "mode", - "inf_band", - "sup_band", - "location", - "url", - "description", - "modulation", - "category_code", - "acf",], + names = names, ) - self.db.fillna("N/A", inplace = True) except FileNotFoundError: self.search_bar.setDisabled(True) box = QMessageBox(self) @@ -129,7 +132,8 @@ class MyApp(QMainWindow, Ui_MainWindow): box.show() else: self.signal_names = self.db.index - + self.db.fillna("N/A", inplace = True) + self.db["url_clicked"] = False try: with open(os.path.join('Data', 'verdb.ini'), 'r') as dbver: self.db_version = int(dbver.read()) @@ -144,7 +148,6 @@ class MyApp(QMainWindow, Ui_MainWindow): else: self.setStatusTip(f"Database version: {self.db_version}") - def display_signals(self): self.result_list.clear() for signal in self.signal_names: @@ -155,8 +158,6 @@ class MyApp(QMainWindow, Ui_MainWindow): self.display_spectrogram() item = self.result_list.currentItem() if item: - self.url_button.setEnabled(True) - self.url_button.setStyleSheet("color: #4c75ff;") self.current_signal_name = item.text() words = self.current_signal_name.split(' ') if len(words) > 3: @@ -167,16 +168,25 @@ class MyApp(QMainWindow, Ui_MainWindow): words = self.current_signal_name self.name_lab.setText(words) current_signal = self.db.loc[self.current_signal_name] - print(current_signal.loc["inf_band"], current_signal.loc["sup_band"]) + self.url_button.setEnabled(True) + self.url_button.setStyleSheet("color: #4c75ff;") category_code = current_signal.loc["category_code"] - self.freq_lab.setText(self.format_numbers( - current_signal.loc["inf_freq"], - current_signal.loc["sup_freq"]) - ) - self.band_lab.setText(self.format_numbers( - current_signal.loc["inf_band"], - current_signal.loc["sup_band"]) - ) + self.find_if_undefined(current_signal) + if not self.undefined_freq: + self.freq_lab.setText(self.format_numbers( + current_signal.loc["inf_freq"], + current_signal.loc["sup_freq"]) + ) + else: + self.freq_lab.setText("Undefined") + if not self.undefined_band: + self.band_lab.setText(self.format_numbers( + current_signal.loc["inf_band"], + current_signal.loc["sup_band"]) + ) + else: + self.band_lab.setText("Undefined") + self.mode_lab.setText(current_signal.loc["mode"]) self.modul_lab.setText(current_signal.loc["modulation"]) self.loc_lab.setText(current_signal.loc["location"]) @@ -200,18 +210,33 @@ class MyApp(QMainWindow, Ui_MainWindow): self.set_band_range() self.audio_widget.set_audio_player() + def find_if_undefined(self, current_signal): + lower_freq = current_signal.loc["inf_freq"] + lower_band = current_signal.loc["inf_band"] + upper_freq = current_signal.loc["sup_freq"] + upper_band = current_signal.loc["sup_band"] + if lower_freq == '0' and upper_freq == "100000000000": + self.undefined_freq = True + else: + self.undefined_freq = False + if lower_band == '0' and upper_band == '100000000': + self.undefined_band = True + else: + self.undefined_band = False + @classmethod def format_numbers(cls, lower, upper): units = {1: 'Hz', 1000: 'kHz', 10**6: 'MHz', 10**9: 'GHz'} lower_factor = cls.change_unit(lower) upper_factor = cls.change_unit(upper) - if lower != upper: - lower = int(lower) / lower_factor - upper = int(upper) / upper_factor - return f"{lower} {units[lower_factor]} - {upper} {units[upper_factor]}" + pre_lower = lower + pre_upper = upper + lower = int(lower) / lower_factor + upper = int(upper) / upper_factor + if pre_lower != pre_upper: + return f"{lower:,} {units[lower_factor]} - {upper:,} {units[upper_factor]}" else: - lower = int(lower) / lower_factor - return f"{lower} {units[lower_factor]}" + return f"{lower:,} {units[lower_factor]}" @staticmethod def change_unit(num): @@ -237,33 +262,38 @@ class MyApp(QMainWindow, Ui_MainWindow): path_spectr = default_pic self.spectrogram.setPixmap(QPixmap(path_spectr)) + @staticmethod + def activate_band_category(band_label, activate = True): + color = "#39eaff;" if activate else "#9f9f9f" + for label in band_label: + label.setStyleSheet(f"color: {color}") + def set_band_range(self, current_signal = None): - # How to deal with one-frequency signals? - if current_signal is not None: - inf_band = int(current_signal.loc["inf_freq"]) - sup_band = int(current_signal.loc["sup_freq"]) - for band, band_label in zip(self.bands, self.band_labels): - delta = (band.upper - band.lower) // 2 + band.lower - if inf_band <= band.lower and sup_band > band.lower: - band_label[0].setStyleSheet("color: #39eaff;") + if current_signal is not None and not self.undefined_freq: + lower_freq = int(current_signal.loc["inf_freq"]) + upper_freq = int(current_signal.loc["sup_freq"]) + zipped = zip(self.bands, self.band_labels) + for i, w in enumerate(zipped): + band, band_label = w + if lower_freq >= band.lower and lower_freq < band.upper: + self.activate_band_category(band_label) + for uband, uband_label in list(zipped)[i + 1:]: + if upper_freq > uband.lower: + self.activate_band_category(uband_label) + else: + self.activate_band_category(uband_label, False) + break else: - band_label[0].setStyleSheet("color: #9f9f9f;") - if band_label[1]: - if inf_band <= delta and sup_band >= delta: - band_label[1].setStyleSheet("color: #39eaff;") - else: - band_label[1].setStyleSheet("color: #9f9f9f;") - if band_label[2]: - if inf_band <= band.upper and sup_band > band.upper: - band_label[2].setStyleSheet("color: #39eaff;") - else: - band_label[2].setStyleSheet("color: #9f9f9f;") + self.activate_band_category(band_label, False) else: - [label.setStyleSheet("color: #9f9f9f;") for labels in self.band_labels for label in labels if label] + for band_label in self.band_labels: + self.activate_band_category(band_label, False) def go_to_web_page_signal(self): if self.current_signal_name: + self.url_button.setStyleSheet("color: #942ccc;") webbrowser.open(self.db.loc[self.current_signal_name].loc["url"]) + # self.db.loc[self.current_signal_name].loc["url_clicked"] = True diff --git a/main_window.ui b/main_window.ui index da6fde1..25c0050 100644 --- a/main_window.ui +++ b/main_window.ui @@ -1248,315 +1248,11 @@ QTabBar::tab:!selected { color: #9f9f9f; - - - 15 - - - - - - 9 - 75 - true - - - - ELF - - - Qt::AlignCenter - - - - - - - - - - 12 - 75 - true - - - - color: #9f9f9f; - - - █ - - - Qt::AlignCenter - - - - - - - - 12 - 75 - true - - - - color: #9f9f9f; - - - █ - - - Qt::AlignCenter - - - - - - - - - - 9 - 75 - true - - - - color: #9f9f9f; - - - SLF - - - Qt::AlignCenter - - - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - - - 9 - 75 - true - - - - ULF - - - Qt::AlignCenter - - - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - - - 9 - 75 - true - - - - VLF - - - Qt::AlignCenter - - - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - - - 9 - 75 - true - - - - LF - - - Qt::AlignCenter - - - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - - - 9 - 75 - true - - - - MF - - - Qt::AlignCenter - - - + - + 12 @@ -1564,6 +1260,9 @@ QTabBar::tab:!selected { true + + color: #9f9f9f; + █ @@ -1573,7 +1272,24 @@ QTabBar::tab:!selected { - + + + + 9 + 75 + true + + + + ELF + + + Qt::AlignCenter + + + + + 12 @@ -1581,6 +1297,9 @@ QTabBar::tab:!selected { true + + color: #9f9f9f; + █ @@ -1591,195 +1310,10 @@ QTabBar::tab:!selected { - - - - - 9 - 75 - true - - - - color: #9f9f9f; - - - HF - - - Qt::AlignCenter - - - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - - - 9 - 75 - true - - - - VHF - - - Qt::AlignCenter - - - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - - - 9 - 75 - true - - - - UHF - - - Qt::AlignCenter - - - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - 12 - 75 - true - - - - █ - - - Qt::AlignCenter - - - - - - - - - - 9 - 75 - true - - - - SHF - - - Qt::AlignCenter - - - - + 12 @@ -1799,7 +1333,27 @@ QTabBar::tab:!selected { - + + + + 9 + 75 + true + + + + color: #9f9f9f; + + + SLF + + + Qt::AlignCenter + + + + + 12 @@ -1818,21 +1372,508 @@ QTabBar::tab:!selected { - - - - 9 - 75 - true - - - - EHF - - - Qt::AlignCenter - - + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + 9 + 75 + true + + + + ULF + + + Qt::AlignCenter + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + 9 + 75 + true + + + + VLF + + + Qt::AlignCenter + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + 9 + 75 + true + + + + LF + + + Qt::AlignCenter + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + 9 + 75 + true + + + + MF + + + Qt::AlignCenter + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + 9 + 75 + true + + + + color: #9f9f9f; + + + HF + + + Qt::AlignCenter + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + 9 + 75 + true + + + + VHF + + + Qt::AlignCenter + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + 9 + 75 + true + + + + UHF + + + Qt::AlignCenter + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + 9 + 75 + true + + + + SHF + + + Qt::AlignCenter + + + + + + + + 12 + 75 + true + + + + color: #9f9f9f; + + + █ + + + Qt::AlignCenter + + + + + + + + + + + + 12 + 75 + true + + + + █ + + + Qt::AlignCenter + + + + + + + + 9 + 75 + true + + + + EHF + + + Qt::AlignCenter + + + + + + + + 12 + 75 + true + + + + color: #9f9f9f; + + + █ + + + Qt::AlignCenter + + + +