jan/masque
1
1
Fork 0
Code Issues Pull requests Releases Wiki Activity

[repetition.Grid] disallow b_vector=None (except when initializing)

Browse source
This commit is contained in:
jan 2026-03-08 22:43:58 -07:00
commit 5eb460ecb7
1 changed files with 1 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
masque/repetition.py
View file

@ -64,7 +64,7 @@ class Grid(Repetition):
_a_count: int _a_count: int
""" Number of instances along the direction specified by the `a_vector` """ """ Number of instances along the direction specified by the `a_vector` """
_b_vector: NDArray[numpy.float64] | None _b_vector: NDArray[numpy.float64]
""" Vector `[x, y]` specifying a second lattice vector for the grid. """ Vector `[x, y]` specifying a second lattice vector for the grid.
Specifies center-to-center spacing between adjacent elements. Specifies center-to-center spacing between adjacent elements.
Can be `None` for a 1D array. Can be `None` for a 1D array.
@ -199,9 +199,6 @@ class Grid(Repetition):
@property @property
def displacements(self) -> NDArray[numpy.float64]: def displacements(self) -> NDArray[numpy.float64]:
if self.b_vector is None:
return numpy.arange(self.a_count)[:, None] * self.a_vector[None, :]
aa, bb = numpy.meshgrid(numpy.arange(self.a_count), numpy.arange(self.b_count), indexing='ij') aa, bb = numpy.meshgrid(numpy.arange(self.a_count), numpy.arange(self.b_count), indexing='ij')
return (aa.flatten()[:, None] * self.a_vector[None, :] return (aa.flatten()[:, None] * self.a_vector[None, :]
+ bb.flatten()[:, None] * self.b_vector[None, :]) # noqa + bb.flatten()[:, None] * self.b_vector[None, :]) # noqa