detect tng-api zoomins (#191)

* detect tng-api zoomins
cbyrohl · Jan 20, 2025 · a2eb331 · a2eb331
1 parent 3e7e3ed
commit a2eb331
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 6 deletions.
diff --git a/src/scida/customs/gadgetstyle/dataset.py b/src/scida/customs/gadgetstyle/dataset.py
@@ -123,13 +123,14 @@ def validate_path(
             metadata_raw = load_metadata(path, **kwargs)
             # need some silly combination of attributes to be sure
             if all([k in metadata_raw for k in ["/Header"]]):
+                # tng-project.org api snapshots do not have "NumPart_Total" in header
+                is_zoom = "CutoutID" in metadata_raw["/Header"]
+                headerattrs = ["NumPart_ThisFile"]
+                if not is_zoom:
+                    headerattrs.append("NumPart_Total")
+
                 # identifying snapshot or group catalog
-                is_snap = all(
-                    [
-                        k in metadata_raw["/Header"]
-                        for k in ["NumPart_ThisFile", "NumPart_Total"]
-                    ]
-                )
+                is_snap = all([k in metadata_raw["/Header"] for k in headerattrs])
                 is_grp = all(
                     [
                         k in metadata_raw["/Header"]

diff --git a/tests/testdata.yaml b/tests/testdata.yaml
@@ -208,6 +208,18 @@ testdata:
       - areposnapshot_withcatalog|F|1|2
     fn: minimal_TNG50-1_group_z3.hdf5
 
+  TNG100-3_api_cutout_z0:
+    types:
+      - interface
+      - areposnapshot
+    fn: tngapi_tng100-3_snap99_cutout_sid42.hdf5
+
+  TNG100-1_api_cutout_z0:
+    types:
+      - interface
+      - areposnapshot
+    fn: tngapi_tng100-1_snap99_cutout_hid97993.hdf5
+
   TNG50-1_group_z0_minimal:
     types:
       - interface