Tool for determining calibration constants from energy spectra

dosview/__init__.py

@@ -26,6 +26,7 @@
 from .version import __version__
 from pyqtgraph import ImageView
 
+from .calibration_widget import CalibrationTab
 from .parsers import (
     BaseLogParser,
     Airdos04CLogParser,
@@ -1078,6 +1079,10 @@ def solve_startup_args(self):
             print("Oteviram zalozku s labdosem")
             self.openLabdosTab()
 
+        if self.args.calibration:
+            print("Oteviram zalozku s kalibraci")
+            self.openCalibrationTab()
+
     def updateStackedWidget(self):
         print("Updating stacked widget")
         print(self.tab_widget.count())
@@ -1086,6 +1091,14 @@ def updateStackedWidget(self):
         else:
             self.stacked_container.setCurrentIndex(0)
 
+    def close_tab(self, index):
+        widget = self.tab_widget.widget(index)
+        if widget is None:
+            return
+        self.tab_widget.removeTab(index)
+        widget.deleteLater()
+        self.updateStackedWidget()
+
     def openPlotTab(self, file_path = None):
         plot_tab = PlotTab()
         if not file_path:
@@ -1112,6 +1125,12 @@ def openLabdosTab(self):
         self.tab_widget.setCurrentIndex(self.tab_widget.count() - 1)
         self.updateStackedWidget()
 
+    def openCalibrationTab(self):
+        calibration_tab = CalibrationTab()
+        self.tab_widget.addTab(calibration_tab, "Calibration")
+        self.tab_widget.setCurrentIndex(self.tab_widget.count() - 1)
+        self.updateStackedWidget()
+
     def blank_page(self):
         # This is widget for blank page
         # When no tab is opened
@@ -1134,6 +1153,7 @@ def initUI(self):
 
         self.tab_widget.setCurrentIndex(0)
         self.tab_widget.setTabsClosable(True)
+        self.tab_widget.tabCloseRequested.connect(self.close_tab)
 
         bar = self.menuBar()
         file = bar.addMenu("&File")
@@ -1159,6 +1179,10 @@ def initUI(self):
         tools_labdosctrl.triggered.connect(self.action_switch_labdoscontrol)
         tools.addAction(tools_labdosctrl)
 
+        tool_calibration = QAction("Calibration", self)
+        tool_calibration.triggered.connect(self.action_switch_calibration)
+        tools.addAction(tool_calibration)
+
 
         help = bar.addMenu("&Help")
         doc = QAction("Documentation", self)
@@ -1192,6 +1216,9 @@ def action_switch_airdoscontrol(self):
     def action_switch_labdoscontrol(self):
         self.openLabdosTab()
 
+    def action_switch_calibration(self):
+        self.openCalibrationTab()
+
     import sys
     import datetime
     from PyQt5.QtCore import QT_VERSION_STR
@@ -1257,6 +1284,7 @@ def main():
     parser.add_argument('file_path', type=str, help='Path to the input file', default=False, nargs='?')
     parser.add_argument('--airdos', action='store_true', help='Enable airdos control tab')
     parser.add_argument('--labdos', action='store_true', help='Enable labdos control tab')
+    parser.add_argument('--calibration', action='store_true', help='Enable calibration tab')
     parser.add_argument('--no_gui', action='store_true', help='Disable GUI and run in headless mode')
     parser.add_argument('--version', action='store_true', help='Print version and exit')
     parser.add_argument('--new-window', action='store_true', help="Open file in new window")