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

Add category filters

Browse Source
This commit is contained in:
alessandro90
2018-11-01 12:29:31 +01:00
parent db6d6a7914
commit 10893b2937
2 changed files with 1690 additions and 1196 deletions
Download Patch File Download Diff File Expand all files Collapse all files

168
main.py
View File

@@ -155,6 +155,14 @@ class MyApp(QMainWindow, Ui_MainWindow):
self.lower_freq_filter_unit, self.lower_freq_filter_unit,
self.lower_freq_confidence) self.lower_freq_confidence)
) )
self.activate_low_freq_filter_btn.toggled.connect(
partial(self.set_min_value_upper_limit,
self.lower_freq_filter_unit,
self.lower_freq_spinbox,
self.upper_freq_filter_unit,
self.upper_freq_spinbox)
)
self.activate_low_freq_filter_btn.clicked.connect( self.activate_low_freq_filter_btn.clicked.connect(
partial(self.set_band_filter_label, partial(self.set_band_filter_label,
self.activate_low_freq_filter_btn, self.activate_low_freq_filter_btn,
@@ -326,6 +334,13 @@ class MyApp(QMainWindow, Ui_MainWindow):
self.lower_band_filter_unit, self.lower_band_filter_unit,
self.lower_band_confidence) self.lower_band_confidence)
) )
self.activate_low_band_filter_btn.toggled.connect(
partial(self.set_min_value_upper_limit,
self.lower_band_filter_unit,
self.lower_band_spinbox,
self.upper_band_filter_unit,
self.upper_band_spinbox)
)
self.activate_low_band_filter_btn.clicked.connect( self.activate_low_band_filter_btn.clicked.connect(
partial(self.set_band_filter_label, partial(self.set_band_filter_label,
self.activate_low_band_filter_btn, self.activate_low_band_filter_btn,
@@ -407,7 +422,36 @@ class MyApp(QMainWindow, Ui_MainWindow):
self.apply_remove_band_filter_btn.clicked.connect(self.display_signals) self.apply_remove_band_filter_btn.clicked.connect(self.display_signals)
self.reset_band_filters_btn.clicked.connect(partial(self.reset_fb_filters, 'band')) self.reset_band_filters_btn.clicked.connect(partial(self.reset_fb_filters, 'band'))
# Manage category filters
# Order matters!
self.cat_filter_btns = [self.military_btn,
self.radar_btn,
self.active_btn,
self.inactive_btn,
self.ham_btn,
self.commercial_btn,
self.aviation_btn,
self.marine_btn,
self.analogue_btn,
self.digital_btn,
self.trunked_btn,
self.utility_btn,
self.sat_btn,
self.navigation_btn,
self.interfering_btn,
self.number_stations_btn,
self.time_signal_btn,]
self.apply_remove_cat_filter_btn.set_texts('Apply', 'Remove')
self.apply_remove_cat_filter_btn.set_slave_filters([*self.cat_filter_btns,
self.cat_at_least_one,
self.cat_all])
self.apply_remove_cat_filter_btn.clicked.connect(self.display_signals)
self.reset_cat_filters_btn.clicked.connect(self.reset_cat_filters)
# ####################################################################################### # #######################################################################################
self.reset_filters_btn.clicked.connect(self.reset_all_filters) self.reset_filters_btn.clicked.connect(self.reset_all_filters)
UrlColors = namedtuple("UrlColors", ["inactive", "active", "clicked"]) UrlColors = namedtuple("UrlColors", ["inactive", "active", "clicked"])
@@ -559,33 +603,34 @@ class MyApp(QMainWindow, Ui_MainWindow):
lower_spin_box, lower_spin_box,
upper_combo_box, upper_combo_box,
upper_spin_box): upper_spin_box):
unit_conversion = {'Hz' : ['kHz', 'MHz', 'GHz'], if lower_spin_box.isEnabled():
'kHz': ['MHz', 'GHz'], unit_conversion = {'Hz' : ['kHz', 'MHz', 'GHz'],
'MHz': ['GHz'] 'kHz': ['MHz', 'GHz'],
} 'MHz': ['GHz']
lower_units = lower_combo_box.currentText() }
upper_units = upper_combo_box.currentText() lower_units = lower_combo_box.currentText()
lower_value = lower_spin_box.value() upper_units = upper_combo_box.currentText()
upper_value = upper_spin_box.value() lower_value = lower_spin_box.value()
inf_limit = (lower_value * self.conversion_factors[lower_units]) \ upper_value = upper_spin_box.value()
// self.conversion_factors[upper_units] inf_limit = (lower_value * self.conversion_factors[lower_units]) \
counter = 0 // self.conversion_factors[upper_units]
while inf_limit > upper_spin_box.maximum(): counter = 0
counter += 1 while inf_limit > upper_spin_box.maximum():
inf_limit //= 1000 counter += 1
if upper_spin_box.minimum() != inf_limit: inf_limit //= 1000
upper_spin_box.setMinimum(inf_limit) if upper_spin_box.minimum() != inf_limit:
if counter > 0: upper_spin_box.setMinimum(inf_limit)
new_unit = unit_conversion[upper_units][counter - 1] if counter > 0:
upper_combo_box.disconnect() new_unit = unit_conversion[upper_units][counter - 1]
upper_combo_box.setCurrentText(new_unit) upper_combo_box.disconnect()
upper_combo_box.currentTextChanged.connect( upper_combo_box.setCurrentText(new_unit)
partial(self.set_min_value_upper_limit, upper_combo_box.currentTextChanged.connect(
lower_combo_box, partial(self.set_min_value_upper_limit,
lower_spin_box, lower_combo_box,
upper_combo_box, lower_spin_box,
upper_spin_box) upper_combo_box,
) upper_spin_box)
)
@pyqtSlot() @pyqtSlot()
def set_band_filter_label(self, def set_band_filter_label(self,
@@ -627,7 +672,7 @@ class MyApp(QMainWindow, Ui_MainWindow):
elif not activate_low and activate_high: elif not activate_low and activate_high:
title = 'High-pass\n\n' title = 'High-pass\n\n'
else: else:
title = "Frequency range:\n\n" title = "Selected range:\n\n"
to_display = "Inactive" to_display = "Inactive"
to_display = title + to_display to_display = title + to_display
range_lbl.setText(to_display) range_lbl.setText(to_display)
@@ -645,7 +690,10 @@ class MyApp(QMainWindow, Ui_MainWindow):
text = self.search_bar.text() text = self.search_bar.text()
available_signals = 0 available_signals = 0
for signal in self.signal_names: for signal in self.signal_names:
if text.lower() in signal.lower() and self.frequency_filters_ok(signal) and self.band_filters_ok(signal): if text.lower() in signal.lower() and \
self.frequency_filters_ok(signal) and \
self.band_filters_ok(signal) and \
self.category_filters_ok(signal):
self.result_list.addItem(signal) self.result_list.addItem(signal)
available_signals += 1 available_signals += 1
self.update_status_tip(available_signals) self.update_status_tip(available_signals)
@@ -661,17 +709,21 @@ class MyApp(QMainWindow, Ui_MainWindow):
def reset_fb_filters(self, ftype): def reset_fb_filters(self, ftype):
if ftype != 'freq' and ftype != 'band': if ftype != 'freq' and ftype != 'band':
raise ValueError("Wrong ftype in function 'reset_fb_filters'") raise ValueError("Wrong ftype in function 'reset_fb_filters'")
apply_remove_btn = getattr(self, 'apply_remove_' + ftype + '_filter_btn') apply_remove_btn = getattr(self, 'apply_remove_' + ftype + '_filter_btn')
include_undef_btn = getattr(self, 'include_undef_' + ftype + 's') include_undef_btn = getattr(self, 'include_undef_' + ftype + 's')
activate_low = getattr(self, 'activate_low_' + ftype + '_filter_btn') activate_low = getattr(self, 'activate_low_' + ftype + '_filter_btn')
activate_up = getattr(self, 'activate_up_' + ftype + '_filter_btn') activate_up = getattr(self, 'activate_up_' + ftype + '_filter_btn')
lower_unit = getattr(self, 'lower_' + ftype + '_filter_unit') lower_unit = getattr(self, 'lower_' + ftype + '_filter_unit')
upper_unit = getattr(self, 'upper_' + ftype + '_filter_unit') upper_unit = getattr(self, 'upper_' + ftype + '_filter_unit')
lower_spinbox = getattr(self, 'lower_' + ftype + '_spinbox') lower_spinbox = getattr(self, 'lower_' + ftype + '_spinbox')
upper_spinbox = getattr(self, 'upper_' + ftype + '_spinbox') upper_spinbox = getattr(self, 'upper_' + ftype + '_spinbox')
lower_confidence = getattr(self, 'lower_' + ftype + '_confidence') lower_confidence = getattr(self, 'lower_' + ftype + '_confidence')
upper_confidence = getattr(self, 'lower_' + ftype + '_confidence') upper_confidence = getattr(self, 'lower_' + ftype + '_confidence')
default_val = 1 if ftype == 'freq' else 5000 default_val = 1 if ftype == 'freq' else 5000
if ftype == 'freq':
for f in self.frequency_filters_btns:
if f.isChecked():
f.setChecked(False)
if apply_remove_btn.isChecked(): if apply_remove_btn.isChecked():
apply_remove_btn.setChecked(False) apply_remove_btn.setChecked(False)
apply_remove_btn.clicked.emit() apply_remove_btn.clicked.emit()
@@ -685,11 +737,20 @@ class MyApp(QMainWindow, Ui_MainWindow):
activate_up.clicked.emit() activate_up.clicked.emit()
lower_unit.setCurrentText("MHz") lower_unit.setCurrentText("MHz")
upper_unit.setCurrentText("MHz") upper_unit.setCurrentText("MHz")
upper_spinbox.setValue(default_val) lower_spinbox.setValue(default_val)
upper_spinbox.setMinimum(1)
upper_spinbox.setValue(default_val) upper_spinbox.setValue(default_val)
lower_confidence.setValue(0) lower_confidence.setValue(0)
upper_confidence.setValue(0) upper_confidence.setValue(0)
def reset_cat_filters(self):
if self.apply_remove_cat_filter_btn.isChecked():
self.apply_remove_cat_filter_btn.setChecked(False)
self.apply_remove_cat_filter_btn.clicked.emit()
for f in self.cat_filter_btns:
f.setChecked(False) if f.isChecked() else None
self.cat_at_least_one.setChecked(True)
def frequency_filters_ok(self, signal_name): def frequency_filters_ok(self, signal_name):
if not self.apply_remove_freq_filter_btn.isChecked(): if not self.apply_remove_freq_filter_btn.isChecked():
return True return True
@@ -754,6 +815,22 @@ class MyApp(QMainWindow, Ui_MainWindow):
upper_limit_ok = False upper_limit_ok = False
return lower_limit_ok and upper_limit_ok return lower_limit_ok and upper_limit_ok
def category_filters_ok(self, signal_name):
if not self.apply_remove_cat_filter_btn.isChecked():
return True
cat_code = self.db.at[signal_name, 'category_code']
cat_checked = 0
positive_cases = 0
for index, cat in enumerate(self.cat_filter_btns):
if cat.isChecked():
cat_checked += 1
if cat_code[index] == '1':
positive_cases += 1
if self.cat_at_least_one.isChecked():
return positive_cases > 0
else:
return cat_checked == positive_cases and cat_checked > 0
@classmethod @classmethod
def filters_ok(cls, spinbox, filter_unit, confidence, sign = 1): def filters_ok(cls, spinbox, filter_unit, confidence, sign = 1):
band_filter = spinbox.value() * cls.conversion_factors[filter_unit.currentText()] band_filter = spinbox.value() * cls.conversion_factors[filter_unit.currentText()]
@@ -776,16 +853,16 @@ class MyApp(QMainWindow, Ui_MainWindow):
undef_freq, undef_band = self.find_if_undefined(current_signal) undef_freq, undef_band = self.find_if_undefined(current_signal)
if not undef_freq: if not undef_freq:
self.freq_lab.setText(self.format_numbers( self.freq_lab.setText(self.format_numbers(
current_signal.at["inf_freq"], current_signal.at["inf_freq"],
current_signal.at["sup_freq"]) current_signal.at["sup_freq"])
) )
else: else:
self.freq_lab.setText("Undefined") self.freq_lab.setText("Undefined")
if not undef_band: if not undef_band:
self.band_lab.setText(self.format_numbers( self.band_lab.setText(self.format_numbers(
current_signal.at["inf_band"], current_signal.at["inf_band"],
current_signal.at["sup_band"]) current_signal.at["sup_band"])
) )
else: else:
self.band_lab.setText("Undefined") self.band_lab.setText("Undefined")
@@ -903,6 +980,7 @@ class MyApp(QMainWindow, Ui_MainWindow):
def reset_all_filters(self): def reset_all_filters(self):
self.reset_frequency_filters_btn.clicked.emit() self.reset_frequency_filters_btn.clicked.emit()
self.reset_band_filters_btn.clicked.emit() self.reset_band_filters_btn.clicked.emit()
self.reset_cat_filters_btn.clicked.emit()
@pyqtSlot() @pyqtSlot()
def go_to_web_page_signal(self): def go_to_web_page_signal(self):

2718
main_window.ui
View File

File diff suppressed because it is too large Load Diff