added __str__ methods for cube types (pytrip#636)

ljelen · Oct 30, 2021 · 64535a3 · 64535a3
1 parent f519773
commit 64535a3
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/pytrip/cube.py b/pytrip/cube.py
@@ -145,6 +145,9 @@ def __init__(self, cube=None):
 
             self.z_table = False  # positions are stored in self.slice_pos (list of slice#,pos(mm),thickness(mm),tilt)
 
+    def __str__(self):
+        return "Cube: "+self.basename
+
     def __add__(self, other):
         """ Overload + operator
         """

diff --git a/pytrip/dos.py b/pytrip/dos.py
@@ -71,6 +71,11 @@ def __init__(self, cube=None):
         self._plan_dicom_series_instance_uid = uid.generate_uid(prefix=None)
         self._dose_dicom_series_instance_uid = uid.generate_uid(prefix=None)
 
+    def __str__(self):
+        if self.type == "DOS":
+            return "Dose: " + self.basename
+        return "Dose (type " + self.type + "): " + self.basename
+
     def read_dicom(self, dcm):
         """ Imports the dose distribution from DICOM object.
 

diff --git a/pytrip/let.py b/pytrip/let.py
@@ -53,6 +53,11 @@ def __init__(self, cube=None):
         self.type = "LET"
         self.let_type = None
 
+    def __str__(self):
+        if self.type == "LET":
+            return "LET: " + self.basename
+        return "LET (type " + self.type + "): " + self.basename
+
     def get_max(self):
         """ Returns the largest value in the LETCube.