def translate(coords_to_translate, translation_vector, active=False):
r"""Translates a 3 x Y array of the form np.array((x, y, z)) by a given
displacement vector of the same form. Transforms can be active (alibi)
or passive (alias), but are passive by default.
"""
translation_dims = np.shape(coords_to_translate)
for _ in translation_dims[1::]:
translation_vector = np.expand_dims(translation_vector, axis=-1)
if active:
translation_vector = -translation_vector
return coords_to_translate - translation_vector
def _calc_device_open(blocked_by_device):
device_open = {}
for device, value in blocked_by_device.items():
device_sum = np.sum(
np.concatenate(
[np.expand_dims(blocked, axis=0) for _, blocked in value.items()],
axis=0,
),
axis=0,
)
device_open[device] = 1 - device_sum
return device_open