Fix: fix loadimage function to robustly load image

news/load-image-fix.rst

@@ -0,0 +1,23 @@
+**Added:**
+
+* <news item>
+
+**Changed:**
+
+* <news item>
+
+**Deprecated:**
+
+* <news item>
+
+**Removed:**
+
+* <news item>
+
+**Fixed:**
+
+* Fixed `loadimage` function in any directory.
+
+**Security:**
+
+* <news item>

src/diffpy/srxplanar/loadimage.py

@@ -16,6 +16,7 @@
 import fnmatch
 import os
 import time
+from pathlib import Path
 
 import numpy as np
 
@@ -66,31 +67,40 @@ def flipImage(self, pic):
         return pic
 
     def loadImage(self, filename):
-        """Load image file, if failed (for example loading an incomplete
-        file), then it will keep trying loading file for 5s.
+        """Load image file using pathlib. If loading fails (e.g.
+        incomplete file), retry for 5 seconds (10×0.5s).
 
-        :param filename: str, image file name
-        :return: 2d ndarray, 2d image array (flipped)
+        :param filename: str or Path, image file name or path
+        :return: 2D ndarray, flipped image array
         """
-        if os.path.exists(filename):
+        filename = Path(
+            filename
+        ).expanduser()  # handle "~", make it a Path object
+        if filename.exists():
             filenamefull = filename
         else:
-            filenamefull = os.path.join(self.opendirectory, filename)
-        image = np.zeros(10000).reshape(100, 100)
-        if os.path.exists(filenamefull):
-            i = 0
-            while i < 10:
-                try:
-                    if os.path.splitext(filenamefull)[-1] == ".npy":
-                        image = np.load(filenamefull)
-                    else:
-                        image = openImage(filenamefull)
-                    i = 10
-                except FileNotFoundError:
-                    i = i + 1
-                    time.sleep(0.5)
-            image = self.flipImage(image)
-            image[image < 0] = 0
+            found_files = list(Path.home().rglob(filename.name))
+            filenamefull = found_files[0] if found_files else None
+
+        if filenamefull is None or not filenamefull.exists():
+            print(f"Warning: file not found: {filename}")
+            return np.zeros((100, 100))
+
+        image = np.zeros((100, 100))
+        for _ in range(10):  # retry 10 times (5 seconds total)
+            try:
+                if filenamefull.suffix == ".npy":
+                    image = np.load(filenamefull)
+                else:
+                    image = openImage(
+                        str(filenamefull)
+                    )  # openImage expects str
+                break
+            except FileNotFoundError:
+                time.sleep(0.5)
+
+        image = self.flipImage(image)
+        image[image < 0] = 0
         return image
 
     def genFileList(