wouter/Artemis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implement band category labels

Browse Source
This commit is contained in:
alessandro90
2018-10-06 12:04:46 +02:00
parent 2bde799c66
commit ed9c061b0a
2 changed files with 643 additions and 572 deletions
Download Patch File Download Diff File Expand all files Collapse all files

156
main.py
View File

@@ -41,6 +41,8 @@ class MyApp(QMainWindow, Ui_MainWindow):
self.db_version = None self.db_version = None
self.db = None self.db = None
self.current_signal_name = '' self.current_signal_name = ''
self.undefined_freq = False
self.undefined_band = False
self.signal_names = [] self.signal_names = []
self.category_labels = [self.cat_mil, self.category_labels = [self.cat_mil,
self.cat_rad, self.cat_rad,
@@ -80,21 +82,34 @@ class MyApp(QMainWindow, Ui_MainWindow):
self.volume, self.volume,
self.audio_progress) self.audio_progress)
BandLabel = namedtuple("BandLabel", ["left", "center", "right"])
self.band_labels = [ self.band_labels = [
[self.elf, self.elf_s1, self.elf_s2], BandLabel(self.elf_left, self.elf, self.elf_right),
[self.slf, self.slf_s1, self.slf_s2], BandLabel(self.slf_left, self.slf, self.slf_right),
[self.ulf, self.ulf_s1, self.ulf_s2], BandLabel(self.ulf_left, self.ulf, self.ulf_right),
[self.vlf, self.vlf_s1, self.vlf_s2], BandLabel(self.vlf_left, self.vlf, self.vlf_right),
[self.lf , self.lf_s1 , self.lf_s2], BandLabel(self.lf_left, self.lf, self.lf_right),
[self.mf , self.mf_s1 , self.mf_s2], BandLabel(self.mf_left, self.mf, self.mf_right),
[self.hf , self.hf_s1 , self.hf_s2], BandLabel(self.hf_left, self.hf, self.hf_right),
[self.vhf, self.vhf_s1, self.vhf_s2], BandLabel(self.vhf_left, self.vhf, self.vhf_right),
[self.uhf, self.uhf_s1, self.uhf_s2], BandLabel(self.uhf_left, self.uhf, self.uhf_right),
[self.shf, self.shf_s1, self.shf_s2], BandLabel(self.shf_left, self.shf, self.shf_right),
[self.ehf, None, None], BandLabel(self.ehf_left, self.ehf, self.ehf_right),
] ]
def load_db(self): def load_db(self):
names = ["name",
"inf_freq",
"sup_freq",
"mode",
"inf_band",
"sup_band",
"location",
"url",
"description",
"modulation",
"category_code",
"acf",]
try: try:
self.db = read_csv(os.path.join('Data', 'db.csv'), self.db = read_csv(os.path.join('Data', 'db.csv'),
sep = '*', sep = '*',
@@ -106,20 +121,8 @@ class MyApp(QMainWindow, Ui_MainWindow):
'inf_band': str, 'inf_band': str,
'sup_band': str, 'sup_band': str,
'category_code': str,}, 'category_code': str,},
names = ["name", names = names,
"inf_freq",
"sup_freq",
"mode",
"inf_band",
"sup_band",
"location",
"url",
"description",
"modulation",
"category_code",
"acf",],
) )
self.db.fillna("N/A", inplace = True)
except FileNotFoundError: except FileNotFoundError:
self.search_bar.setDisabled(True) self.search_bar.setDisabled(True)
box = QMessageBox(self) box = QMessageBox(self)
@@ -129,7 +132,8 @@ class MyApp(QMainWindow, Ui_MainWindow):
box.show() box.show()
else: else:
self.signal_names = self.db.index self.signal_names = self.db.index
self.db.fillna("N/A", inplace = True)
self.db["url_clicked"] = False
try: try:
with open(os.path.join('Data', 'verdb.ini'), 'r') as dbver: with open(os.path.join('Data', 'verdb.ini'), 'r') as dbver:
self.db_version = int(dbver.read()) self.db_version = int(dbver.read())
@@ -144,7 +148,6 @@ class MyApp(QMainWindow, Ui_MainWindow):
else: else:
self.setStatusTip(f"Database version: {self.db_version}") self.setStatusTip(f"Database version: {self.db_version}")
def display_signals(self): def display_signals(self):
self.result_list.clear() self.result_list.clear()
for signal in self.signal_names: for signal in self.signal_names:
@@ -155,8 +158,6 @@ class MyApp(QMainWindow, Ui_MainWindow):
self.display_spectrogram() self.display_spectrogram()
item = self.result_list.currentItem() item = self.result_list.currentItem()
if item: if item:
self.url_button.setEnabled(True)
self.url_button.setStyleSheet("color: #4c75ff;")
self.current_signal_name = item.text() self.current_signal_name = item.text()
words = self.current_signal_name.split(' ') words = self.current_signal_name.split(' ')
if len(words) > 3: if len(words) > 3:
@@ -167,16 +168,25 @@ class MyApp(QMainWindow, Ui_MainWindow):
words = self.current_signal_name words = self.current_signal_name
self.name_lab.setText(words) self.name_lab.setText(words)
current_signal = self.db.loc[self.current_signal_name] 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"] category_code = current_signal.loc["category_code"]
self.freq_lab.setText(self.format_numbers( self.find_if_undefined(current_signal)
current_signal.loc["inf_freq"], if not self.undefined_freq:
current_signal.loc["sup_freq"]) self.freq_lab.setText(self.format_numbers(
) current_signal.loc["inf_freq"],
self.band_lab.setText(self.format_numbers( current_signal.loc["sup_freq"])
current_signal.loc["inf_band"], )
current_signal.loc["sup_band"]) 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.mode_lab.setText(current_signal.loc["mode"])
self.modul_lab.setText(current_signal.loc["modulation"]) self.modul_lab.setText(current_signal.loc["modulation"])
self.loc_lab.setText(current_signal.loc["location"]) self.loc_lab.setText(current_signal.loc["location"])
@@ -200,18 +210,33 @@ class MyApp(QMainWindow, Ui_MainWindow):
self.set_band_range() self.set_band_range()
self.audio_widget.set_audio_player() 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 @classmethod
def format_numbers(cls, lower, upper): def format_numbers(cls, lower, upper):
units = {1: 'Hz', 1000: 'kHz', 10**6: 'MHz', 10**9: 'GHz'} units = {1: 'Hz', 1000: 'kHz', 10**6: 'MHz', 10**9: 'GHz'}
lower_factor = cls.change_unit(lower) lower_factor = cls.change_unit(lower)
upper_factor = cls.change_unit(upper) upper_factor = cls.change_unit(upper)
if lower != upper: pre_lower = lower
lower = int(lower) / lower_factor pre_upper = upper
upper = int(upper) / upper_factor lower = int(lower) / lower_factor
return f"{lower} {units[lower_factor]} - {upper} {units[upper_factor]}" upper = int(upper) / upper_factor
if pre_lower != pre_upper:
return f"{lower:,} {units[lower_factor]} - {upper:,} {units[upper_factor]}"
else: else:
lower = int(lower) / lower_factor return f"{lower:,} {units[lower_factor]}"
return f"{lower} {units[lower_factor]}"
@staticmethod @staticmethod
def change_unit(num): def change_unit(num):
@@ -237,33 +262,38 @@ class MyApp(QMainWindow, Ui_MainWindow):
path_spectr = default_pic path_spectr = default_pic
self.spectrogram.setPixmap(QPixmap(path_spectr)) 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): def set_band_range(self, current_signal = None):
# How to deal with one-frequency signals? if current_signal is not None and not self.undefined_freq:
if current_signal is not None: lower_freq = int(current_signal.loc["inf_freq"])
inf_band = int(current_signal.loc["inf_freq"]) upper_freq = int(current_signal.loc["sup_freq"])
sup_band = int(current_signal.loc["sup_freq"]) zipped = zip(self.bands, self.band_labels)
for band, band_label in zip(self.bands, self.band_labels): for i, w in enumerate(zipped):
delta = (band.upper - band.lower) // 2 + band.lower band, band_label = w
if inf_band <= band.lower and sup_band > band.lower: if lower_freq >= band.lower and lower_freq < band.upper:
band_label[0].setStyleSheet("color: #39eaff;") 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: else:
band_label[0].setStyleSheet("color: #9f9f9f;") self.activate_band_category(band_label, False)
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;")
else: 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): def go_to_web_page_signal(self):
if self.current_signal_name: if self.current_signal_name:
self.url_button.setStyleSheet("color: #942ccc;")
webbrowser.open(self.db.loc[self.current_signal_name].loc["url"]) webbrowser.open(self.db.loc[self.current_signal_name].loc["url"])
# self.db.loc[self.current_signal_name].loc["url_clicked"] = True

1059
main_window.ui
View File

File diff suppressed because it is too large Load Diff