jan/fdfd_tools
2
2
Fork 1
Code Issues Pull Requests Releases Wiki Activity

Ignore size-1 axes when shifting

Browse Source
This commit is contained in:
Jan Petykiewicz 2019-11-22 00:56:03 -08:00
parent 63ad67800a
commit 6f2faca7dc
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
meanas/fdfd/operators.py
View File

@ -531,6 +531,8 @@ def e_boundary_source(mask: vfield_t,
shift = lambda axis, polarity: shift_with_mirror(axis=axis, shape=shape, shift_distance=polarity) shift = lambda axis, polarity: shift_with_mirror(axis=axis, shape=shape, shift_distance=polarity)
for axis in (0, 1, 2): for axis in (0, 1, 2):
if shape[axis] == 1:
continue
for polarity in (-1, +1): for polarity in (-1, +1):
r = shift(axis, polarity) - sparse.eye(numpy.prod(shape)) # shifted minus original r = shift(axis, polarity) - sparse.eye(numpy.prod(shape)) # shifted minus original
r3 = sparse.block_diag((r, r, r)) r3 = sparse.block_diag((r, r, r))