Skip to content

Commit

Permalink
minor changes in repr str depiction
Browse files Browse the repository at this point in the history
  • Loading branch information
Stef-Martin committed Mar 5, 2025
1 parent b982a1f commit e3ebc43
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 7 deletions.
5 changes: 3 additions & 2 deletions src/mrpro/operators/FastFourierOp.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,6 +153,7 @@ def adjoint(self, y: torch.Tensor) -> tuple[torch.Tensor,]:
def __repr__(self) -> str:
"""Representation method for FastFourierOperator."""
k2k1k0 = ['k2', 'k1', 'k0']
dimension_210 = [k2k1k0[i] for i in self._dim if i in range(-3, 0)]
out = f'Dimension(s) along which FFT is applied: {list(dimension_210)!s}'
dimension_210 = tuple(k2k1k0[i] for i in self._dim if i in range(-3, 0))
dimension_210_str = ', '.join(dimension_210)
out = f'Dimension(s) along which FFT is applied: ({dimension_210_str})'
return out
12 changes: 7 additions & 5 deletions src/mrpro/operators/NonUniformFastFourierOp.py
Original file line number Diff line number Diff line change
Expand Up @@ -271,13 +271,15 @@ def __repr__(self) -> str:
device = self._omega.device if self._omega is not None else 'none'
zyx = ['z', 'y', 'x']
k2k1k0 = ['k2', 'k1', 'k0']
direction_zyx = [zyx[i] for i in self._direction_zyx if i in range(-3, 0)]
dimension_210 = [k2k1k0[i] for i in self._dimension_210 if i in range(-3, 0)]

direction_zyx = tuple(zyx[i] for i in self._direction_zyx if i in range(-3, 0))
dimension_210 = tuple(k2k1k0[i] for i in self._dimension_210 if i in range(-3, 0))
recon_size_str = ', '.join(f'{dim}={size}' for dim, size in zip(direction_zyx, self._im_size, strict=False))
direction_zyx_str = ', '.join(direction_zyx)
dimension_210_str = ', '.join(dimension_210)
out = (
f'{type(self).__name__} on device: {device}\n'
f'Dimension(s) along which NUFFT is applied: {list(direction_zyx)!s} / {dimension_210}\n'
f'Reconstructed image size in dimension {direction_zyx}: {self._im_size}'
f'Dimension(s) along which NUFFT is applied: ({direction_zyx_str}) / ({dimension_210_str})\n'
f'Reconstructed image size {recon_size_str}'
)
return out

Expand Down

0 comments on commit e3ebc43

Please sign in to comment.