Added ability to plat spectral data

Author: sebasj13

topasgraphsim/src/classes/paramframe.py

@@ -34,6 +34,28 @@ def __init__(self, parent, tgs_graph, lang):
 
                 self.beamquality.grid(row=1, column=1, sticky="nsew", padx=2, pady=2)
                 self.zmax.grid(row=2, column=1, sticky="nsew", padx=2, pady=2)
+                
+            elif self.graph.direction == "s":
+                
+                self.binlabel = ctk.CTkLabel(self, text=self.text.bins[self.lang], font=self.font, height=14)
+                self.binsizelabel = ctk.CTkLabel(self, text=self.text.binsize[self.lang], font=self.font, height=14)
+                self.minlabel = ctk.CTkLabel(self, text=self.text.min[self.lang], font=self.font, height=14)
+                self.maxlabel = ctk.CTkLabel(self, text=self.text.max[self.lang], font=self.font, height=14)
+                
+                self.bins = ctk.CTkLabel(self, text=str(self.parameters[0]), font=self.font, height=14)
+                self.binsize = ctk.CTkLabel(self, text=str(self.parameters[1]), font=self.font, height=14)
+                self.min = ctk.CTkLabel(self, text=str(self.parameters[2]), font=self.font, height=14)
+                self.max = ctk.CTkLabel(self, text=str(self.parameters[3]), font=self.font, height=14)
+                
+                self.namelabel.grid(row=0, column=0, columnspan=2, sticky="nsew", padx=2, pady=2)
+                self.binlabel.grid(row=1, column=0, sticky="w", padx=2, pady=2)
+                self.bins.grid(row=1, column=1, sticky="nsew", padx=2, pady=2)
+                self.binsizelabel.grid(row=2, column=0, sticky="w", padx=2, pady=2)
+                self.binsize.grid(row=2, column=1, sticky="nsew", padx=2, pady=2)
+                self.minlabel.grid(row=3, column=0, sticky="w", padx=2, pady=2)
+                self.min.grid(row=3, column=1, sticky="nsew", padx=2, pady=2)
+                self.maxlabel.grid(row=4, column=0, sticky="w", padx=2, pady=2)
+                self.max.grid(row=4, column=1, sticky="nsew", padx=2, pady=2)
 
             else:
                 self.halfwidthlabel = ctk.CTkLabel(self, text=self.text.fwhm[self.lang], font=self.font, height=14)

topasgraphsim/src/classes/sim_import.py

@@ -1,5 +1,6 @@
 from tkinter import simpledialog as sd
 
+import re
 import numpy as np
 import topas2numpy
 
@@ -17,6 +18,17 @@ def __init__(self, filepath):
         else:
             self.filename = self.filepath.split("\\")[-1][:-4]
 
+        with open(self.filepath, "r") as f:
+            self.lines = f.readlines()
+            for line in self.lines:
+                if "Binned" in line:
+                    self.direction = "s"
+                    self.dose = np.array(self.lines[-1].split(","), dtype=np.float64)[1:-2]   
+                    self.bins, self.binsize, self.min, self.max = re.findall(r"[-+]?(?:\d*\.*\d+)", self.lines[-3])
+                    self.axis = np.linspace(float(self.min), float(self.max), int(self.bins))
+                    self.std_dev = np.zeros(len(self.dose))
+                    return
+                     
         self.data = topas2numpy.BinnedResult(self.filepath)
         bins = [dim.n_bins for dim in self.data.dimensions]
         if bins.count(1) != 2:
@@ -75,6 +87,8 @@ def params(self):
                 self.dose / max(self.dose),
                 self.std_dev / max(self.dose),
             )
+        elif self.direction == "s":
+            return self.bins, self.binsize, self.min, self.max
         else:
             params = dp.calculate_parameters(
                 self.axis, self.dose / max(self.dose)

topasgraphsim/src/classes/tgs_graph.py

@@ -50,6 +50,7 @@ def data(self):
         axis = np.add(self.dataObject.axis,self.axshift)
         dose = self.dataObject.dose.copy()
         error = self.dataObject.std_dev.copy()
+        
         if self.normalize:
             if self.normalization == "maximum":
                 error /= np.max(dose)
@@ -78,7 +79,7 @@ def plot(self, ax):
 
         axis, dose, error = self.data()
         if self.caxcorrection:
-            if self.direction != "Z":
+            if self.direction != "Z" and self.direction != "s":
                 axis = np.add(axis, self.dataObject.params()[1])
 
         if self.error:

topasgraphsim/src/resources/language.py

@@ -20,6 +20,10 @@ def __init__(self):
             "de": "Parameter konnten nicht berechnen werden!",
             "en": "Parameters could not be calculated!",
         }
+        self.bins = {"de": "Klassenanzahl", "en": "Number of bins"}
+        self.binsize = {"de": "Klassenbreite", "en": "Bin size"}
+        self.min = {"de": "Minimum", "en": "Minimum"}
+        self.max = {"de": "Maximum", "en": "Maximum"}
         self.criterion = {"de": "Kriterium", "en": "Criterion"}
         self.calcparams = {"de": "Parameter berechnen", "en": "Calculate parameters"}
         self.calculate = {"de": "Berechnen", "en": "Calculate"}

topasgraphsim/src/resources/profile.json

@@ -1,5 +1,5 @@
 {
-    "geometry": "960x540+482+208",
+    "geometry": "960x540+208+208",
     "state": "normal",
     "draganddrop": true,
     "color_scheme": "light",