Skip to content

Commit a18ae5d

Browse files
committed
mv _header to Step
1 parent 54bd478 commit a18ae5d

File tree

1 file changed

+14
-16
lines changed

1 file changed

+14
-16
lines changed

src/stagpy/step.py

Lines changed: 14 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ class Geometry:
4242

4343
@cached_property
4444
def _header(self) -> Mapping[str, Any]:
45-
hdr = self.step.fields._header
45+
hdr = self.step._header
4646
if hdr is None:
4747
raise error.NoGeomError(self.step)
4848
return hdr
@@ -356,7 +356,7 @@ def _get_raw_data(self, name: str) -> tuple[list[str], Any]:
356356
sdat = self.step.sdat
357357
if filestem in phyvars.SFIELD_FILES_H5:
358358
xmff = sdat._botxmf if name.endswith("bot") else sdat._topxmf
359-
header = self._header
359+
header = self.step._header
360360
else:
361361
xmff = sdat._dataxmf
362362
header = None
@@ -367,20 +367,6 @@ def _get_raw_data(self, name: str) -> tuple[list[str], Any]:
367367
break
368368
return list_fvar, parsed_data
369369

370-
@cached_property
371-
def _header(self) -> dict[str, Any] | None:
372-
if self.step.isnap is None:
373-
return None
374-
binfiles = self.step.sdat._binfiles_set(self.step.isnap)
375-
header = None
376-
if binfiles:
377-
header = stagyyparsers.field_header(binfiles.pop())
378-
elif self.step.sdat.hdf5:
379-
header = stagyyparsers.read_geom_h5(
380-
self.step.sdat._dataxmf, self.step.isnap
381-
)
382-
return header if header else None
383-
384370
@cached_property
385371
def geom(self) -> Geometry:
386372
"""Geometry information.
@@ -636,6 +622,18 @@ def __repr__(self) -> str:
636622
else:
637623
return f"{self.sdat!r}.steps[{self.istep}]"
638624

625+
@cached_property
626+
def _header(self) -> dict[str, Any] | None:
627+
if self.isnap is None:
628+
return None
629+
binfiles = self.sdat._binfiles_set(self.isnap)
630+
header = None
631+
if binfiles:
632+
header = stagyyparsers.field_header(binfiles.pop())
633+
elif self.sdat.hdf5:
634+
header = stagyyparsers.read_geom_h5(self.sdat._dataxmf, self.isnap)
635+
return header if header else None
636+
639637
@property
640638
def geom(self) -> Geometry:
641639
"""Geometry information.

0 commit comments

Comments
 (0)