osiris-utils 1.2.1__py3-none-any.whl → 1.2.2__py3-none-any.whl

osiris_utils/data/data.py

@@ -186,7 +186,7 @@ class OsirisGridFile(OsirisData):
         Field label/name (LaTeX formatted, e.g., r'$E_x$')
     """
 
-    def __init__(self, filename, data_slice: slice | None = None):
+    def __init__(self, filename, data_slice: slice | None = None, load_data: bool = True):
         super().__init__(filename)
 
         variable_key = self._get_variable_key(self._file)
@@ -195,12 +195,14 @@ class OsirisGridFile(OsirisData):
         self._label = self._file.attrs["LABEL"][0].decode("utf-8")
         self._FFTdata = None
 
-        data = np.array(self._file[variable_key][:]) if data_slice is None else np.array(self._file[variable_key][data_slice])
+        # Use dataset.shape to obtain sizes without loading full data when possible
+        dset = self._file[variable_key]
 
         axis = list(self._file["AXIS"].keys())
         if len(axis) == 1:
             self._grid = self._file["AXIS/" + axis[0]][()]
-            self._nx = len(data)
+            # nx for 1D is dataset length along its only axis
+            self._nx = dset.shape[0]
             self._dx = (self.grid[1] - self.grid[0]) / self.nx
             self._x = np.arange(self.grid[0], self.grid[1], self.dx)
         else:
@@ -208,15 +210,14 @@ class OsirisGridFile(OsirisData):
             for ax in axis:
                 grid.append(self._file["AXIS/" + ax][()])
             self._grid = np.array(grid)
-            self._nx = self._file[variable_key][()].transpose().shape
+            # use dataset.shape to avoid reading the whole dataset
+            # dataset stored may need transpose to match expected ordering; preserve shape only
+            self._nx = tuple(dset.shape[::-1]) if len(dset.shape) > 1 else (dset.shape[0],)
             self._dx = (self.grid[:, 1] - self.grid[:, 0]) / self.nx
 
-            # There's an issue when the dimension is 3 and we want to plot a 2D phasespace. I believe this
-            # is a problem for all cases where the dim != dim_of_phasespace
             try:
                 self._x = [np.arange(self.grid[i, 0], self.grid[i, 1], self.dx[i]) for i in range(self.dim)]
-            except Exception as e:
-                print(f"Error occurred while creating spatial coordinates: {e}")
+            except Exception:
                 self._x = [np.arange(self.grid[i, 0], self.grid[i, 1], self.dx[i]) for i in range(1)]
 
         self._axis = []
@@ -231,7 +232,13 @@ class OsirisGridFile(OsirisData):
             }
             self._axis.append(axis_data)
 
-        self._data = np.ascontiguousarray(data.T)
+        # Only load data if explicitly requested. Otherwise keep a placeholder so metadata-only
+        # initializations are cheap for large files.
+        if load_data:
+            data = np.array(dset[:]) if data_slice is None else np.array(dset[data_slice])
+            self._data = np.ascontiguousarray(data.T)
+        else:
+            self._data = None
 
         self._close_file()
 

osiris_utils/data/diagnostic.py

@@ -398,7 +398,7 @@ class Diagnostic:
             for file_num in range(1, self._maxiter + 1):
                 path_file = os.path.join(file_template + f"{file_num:06d}.h5")
                 if os.path.exists(path_file):
-                    dump = OsirisGridFile(path_file)
+                    dump = OsirisGridFile(path_file, load_data=False)
                     self._dx = dump.dx
                     self._nx = dump.nx
                     self._x = dump.x

{osiris_utils-1.2.1.dist-info → osiris_utils-1.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: osiris_utils
-Version: 1.2.1
+Version: 1.2.2
 Summary: Utilities to manipulate and visualise OSIRIS plasma PIC output data
 Author-email: João Pedro Ferreira Biu <joaopedrofbiu@tecnico.ulisboa.pt>
 License: MIT

{osiris_utils-1.2.1.dist-info → osiris_utils-1.2.2.dist-info}/RECORD

@@ -1056,8 +1056,8 @@ osiris_utils/cli/info.py,sha256=E2JPwGpO3T6OJFWDOuLIRD6gJkbYxNoyDztyHLo22JQ,5177
 osiris_utils/cli/plot.py,sha256=HV69rxcouDQTA3MNFx62HUWaEKMcVSoKQ1NUR4yoSvI,5063
 osiris_utils/cli/validate.py,sha256=MdOzsVzgQPwVNHmoAICpcRn-POhpz9yN2I9bb8TYRro,7253
 osiris_utils/data/__init__.py,sha256=Yc7SyMDBO-HHE8VNqug1yf_gxyl2uawVd80E0q3FDAE,394
-osiris_utils/data/data.py,sha256=VtoMDoTj8nXZlGZeCjIgIZUICtwkST9TxBFShzYsnAI,22059
-osiris_utils/data/diagnostic.py,sha256=sy-e4hz4G8Am9wjtkeTr3w64elMLXGizsfGcA9xfHAw,39181
+osiris_utils/data/data.py,sha256=zk7CyW2TnXVkd8P94Qop_VPGa47sogAELggCqVUW0RY,22372
+osiris_utils/data/diagnostic.py,sha256=v9RaRF1UfWA6PxwwWJthYHyevOy0yqsgcpwxNsQU4kM,39198
 osiris_utils/data/simulation.py,sha256=bn5-Uv894dm38PNhUtMl9TZCmEBQlVqN4DuteM6ui-Y,7256
 osiris_utils/decks/__init__.py,sha256=m6lNzGk5DSHs9wpRmW6d3vdir4Wg28ImWkeg5fzQx3k,114
 osiris_utils/decks/decks.py,sha256=HW9iAGP2xjtJ5VpHceLhXYK2tnB4rcYdsfDquRQVv3o,12194
@@ -1072,9 +1072,9 @@ osiris_utils/postprocessing/postprocess.py,sha256=-_zydCuK7RsH2kTT5vGpIZd5dQco9i
 osiris_utils/postprocessing/pressure_correction.py,sha256=N0UwUmW7eEcr1OdgzyIwY-rFxUKXomfsQvTUobKXBlw,7223
 osiris_utils/vis/__init__.py,sha256=Vr0o8wtRFtPOJdSK5cRIa_bDvbylAe2sHTk8c7qO_tM,51
 osiris_utils/vis/plot3d.py,sha256=ywP9UGI4i3SRgUdsz-ILHxc7W_StCSN1D1U4Tmpd_6o,4658
-osiris_utils-1.2.1.dist-info/licenses/LICENSE.txt,sha256=z1lVzLA_QsfcnLffhfWMrFw5cR_nuwpvbOGCOFUN7iY,1066
-osiris_utils-1.2.1.dist-info/METADATA,sha256=p7q91bTXXIo2cZzX2gXsDno84ItWr4_x9oI3kKG4uM0,5582
-osiris_utils-1.2.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-osiris_utils-1.2.1.dist-info/entry_points.txt,sha256=s3WXJWsWF5AWEFhJ8rXscjGttQCOXvViOiNmX45AJjo,49
-osiris_utils-1.2.1.dist-info/top_level.txt,sha256=tSs2XcOs-JtYsqVepo5WVSNcilf6ALHfKyYuPaVZLco,38
-osiris_utils-1.2.1.dist-info/RECORD,,
+osiris_utils-1.2.2.dist-info/licenses/LICENSE.txt,sha256=z1lVzLA_QsfcnLffhfWMrFw5cR_nuwpvbOGCOFUN7iY,1066
+osiris_utils-1.2.2.dist-info/METADATA,sha256=EUi5pM1-7_vd0eneUXPL-TZF2qxPwdzalO7flX_wndw,5582
+osiris_utils-1.2.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+osiris_utils-1.2.2.dist-info/entry_points.txt,sha256=s3WXJWsWF5AWEFhJ8rXscjGttQCOXvViOiNmX45AJjo,49
+osiris_utils-1.2.2.dist-info/top_level.txt,sha256=tSs2XcOs-JtYsqVepo5WVSNcilf6ALHfKyYuPaVZLco,38
+osiris_utils-1.2.2.dist-info/RECORD,,