Fix for case when no ROOT files are found

villekf · Jul 18, 2024 · cd2138c · cd2138c
1 parent a638f85
commit cd2138c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/source/Python/omegatomo/io/loadData.py b/source/Python/omegatomo/io/loadData.py
@@ -136,6 +136,17 @@ def loadROOT(options, store_coordinates = False):
     else:
         files = glob.glob(os.path.join(options.fpath, '*.root'))
         nFiles = len(files)
+    if nFiles == 0:
+        print('No files found! Please select a ROOT file')
+        import tkinter as tk
+        from tkinter.filedialog import askopenfilename
+        root = tk.Tk()
+        root.withdraw()
+        filename = askopenfilename(title='Select first ROOT file',filetypes=([('ROOT Files','*.root')]))
+        if not filename:
+            raise ValueError('No file was selected')
+        files = glob.glob(os.path.join(os.path.split(filename)[0], '*.root'))
+        nFiles = len(files)