diff --git a/pmd_beamphysics/wavefront.py b/pmd_beamphysics/wavefront.py index 8ee25cc..6c89b7b 100644 --- a/pmd_beamphysics/wavefront.py +++ b/pmd_beamphysics/wavefront.py @@ -1075,11 +1075,6 @@ def grid(self) -> tuple[int, ...]: """The rmesh shape, without padding.""" return self._grid - @property - def rmesh_shape(self) -> tuple[int, ...]: - """The rmesh shape, without padding.""" - return self._grid - @property def pad(self) -> tuple[int, ...]: """ @@ -1121,10 +1116,6 @@ def kspace_domain(self) -> tuple[np.ndarray, ...]: shifted=True, ) - @property - def _k_center_indices(self) -> tuple[int, ...]: - return tuple(grid // 2 + pad for grid, pad in zip(self.rmesh.shape, self.pad)) - @property def rmesh(self) -> np.ndarray: """Real-space (cartesian space) wavefront field data.""" @@ -1246,7 +1237,7 @@ def grid_spacing(self) -> tuple[float, ...]: @property def ranges(self): """ - Get the physical range the entire mesh represents in (low, high) pairs + Get the physical range the entire rmesh represents in (low, high) pairs for each dimension. Returns @@ -2173,7 +2164,7 @@ def drift(wf: Wavefront, distance: float) -> Wavefront: # for idx, label in enumerate(wf.axis_labels) # if label != wf.longitudinal_axis # ] - indices = [0, 1] + indices = (0, 1) transverse_kspace_grid = nd_kspace_mesh( coeffs=(1.0,) * len(wf.grid), sizes=[wf.rmesh.shape[idx] for idx in indices],