Save log reading configs

sohstationviewer/conf/config_processor.py

@@ -13,6 +13,7 @@ default_config = f'''
 from_data_card = False
 data = False
 sdata = False
+log = False
 
 [ColorMode]
 black = True
@@ -48,6 +49,7 @@ to_date = {QtCore.QDate.currentDate().toString("yyyy-MM-dd")}
 mp_color_mode = regular
 tps_color_mode = High
 date_mode = YYYY-MM-DD
+add_mass_pos_to_soh = False
 '''
 
 
@@ -62,7 +64,7 @@ class ConfigProcessor:
     def __init__(self):
         self.config = configparser.ConfigParser()
         self.expected_keys = {
-            'from_data_card', 'data', 'sdata',
+            'from_data_card', 'data', 'sdata', 'log',
             'black', 'white',
             'min_gap_length',
             'mp123zne', 'mp456uvw',
@@ -71,7 +73,8 @@ class ConfigProcessor:
             'plot_tps', 'plot_raw',
             'all_soh',
             'from_date', 'to_date',
-            'mp_color_mode', 'tps_color_mode', 'date_mode'
+            'mp_color_mode', 'tps_color_mode', 'date_mode',
+                'add_mass_pos_to_soh'
         }
 
     def load_config(self):
@@ -109,10 +112,16 @@ class ConfigProcessor:
                 f'missing keys are: {", ".join(missing_keys)}'
             )
 
+        from_data_card_checked = self.config.getboolean('FileRead',
+                                                        'from_data_card')
         data_checked = self.config.getboolean('FileRead', 'data')
         sdata_checked = self.config.getboolean('FileRead', 'sdata')
+        log_checked = self.config.getboolean('FileRead', 'log')
         if data_checked and sdata_checked:
             raise BadConfigError('data and sdata cannot both be chosen.')
+        if from_data_card_checked and log_checked:
+            raise BadConfigError('Cannot read from data card and log file at '
+                                 'the same time.')
 
         black_color_mode = self.config.getboolean('ColorMode', 'black')
         white_color_mode = self.config.getboolean('ColorMode', 'white')
@@ -187,6 +196,7 @@ class ConfigProcessor:
         )
         window.data_radio_button.setChecked(get_bool('FileRead', 'data'))
         window.sdata_radio_button.setChecked(get_bool('FileRead', 'sdata'))
+        window.log_checkbox.setChecked(get_bool('FileRead', 'log'))
 
         window.background_black_radio_button.setChecked(
             get_bool('ColorMode', 'black')
@@ -253,6 +263,10 @@ class ConfigProcessor:
         elif date_mode == 'YYYY:DOY':
             window.yyyy_doy_action.trigger()
 
+        window.add_masspos_to_rt130_soh.setChecked(
+            get_bool('MiscOptions', 'add_mass_pos_to_soh')
+        )
+
     def reset(self):
         self.config = configparser.ConfigParser()
         self.config.read_string(default_config)

sohstationviewer/view/main_window.py

@@ -1112,6 +1112,8 @@ class MainWindow(QtWidgets.QMainWindow, UIMainWindow):
                         str(self.data_radio_button.isChecked()))
         self.config.set('FileRead', 'sdata',
                         str(self.sdata_radio_button.isChecked()))
+        self.config.set('FileRead', 'log',
+                        str(self.log_checkbox.isChecked()))
         self.config.set('ColorMode', 'black',
                         str(self.background_black_radio_button.isChecked()))
         self.config.set('ColorMode', 'white',
@@ -1163,6 +1165,8 @@ class MainWindow(QtWidgets.QMainWindow, UIMainWindow):
             raise Exception('Something is very wrong. No date mode is chosen.'
                             'Please contact the software group.')
         self.config.set('MiscOptions', 'date_mode', date_mode)
+        self.config.set('MiscOptions', 'add_mass_pos_to_soh',
+                        str(self.add_masspos_to_rt130_soh.isChecked()))
         with open('sohstationviewer/conf/read_settings.ini', 'w') as file:
             self.config.write(file)