From b1a7f9a626c64a18414b78bfaa8b0e7c99aed313 Mon Sep 17 00:00:00 2001 From: Jan-Lukas Wynen Date: Fri, 1 Nov 2024 14:18:40 +0100 Subject: [PATCH] Merge dnd builder methods --- src/sqomega/_build.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/src/sqomega/_build.py b/src/sqomega/_build.py index 7e4f3f6..0c4136e 100644 --- a/src/sqomega/_build.py +++ b/src/sqomega/_build.py @@ -173,13 +173,17 @@ def add_empty_detector_params(self) -> SqwBuilder: ) return self - def add_dnd_metadata(self, block: SqwDndMetadata) -> SqwBuilder: + def _add_dnd_metadata(self, block: SqwDndMetadata) -> SqwBuilder: self._data_blocks[("data", "metadata")] = block return self - def add_empty_dnd_data(self) -> SqwBuilder: - self._dnd_placeholder = _DndPlaceholder(shape=(50, 50, 50, 50)) - return self + def add_empty_dnd_data(self, block: SqwDndMetadata) -> SqwBuilder: + # The file must always contain a DND block + builder = self._add_dnd_metadata(block) + builder._dnd_placeholder = _DndPlaceholder( + shape=tuple(map(int, block.axes.n_bins_all_dims)) + ) + return builder def add_default_instrument(self, instrument: SqwIXNullInstrument) -> SqwBuilder: self._instrument = instrument