Pass mla index arg to apply mask that is specific to the microlens index

[gschlafly]

Description

The microlens index should be able to be successfully passed into the numpy and pytorch functions. Only the mask for the lenslet (0,0) is being used. However, when passing the mla_index argument, some of the tests fail. https://github.com/PolarizedLightFieldMicroscopy/GeoBirT/blob/f478893ad949a3b3f6e36fc86639c644dfbb0d62/VolumeRaytraceLFM/birefringence_implementations.py#L1389-L1397

Files

• VolumeRaytraceLFM/birefringence_implementations.py
• tests/test_all.py

To Reproduce

Have the function ret_and_azim_images() return self.ret_and_azim_images_torch(volume_in, microlens_offset, mla_index=mla_index) for the pytorch backend. Then, run the pytest test_forward_projection_different_volumes for single_voxel.

Error message from the pytest:

Tasks

• [x] Inspect the value of the mla_index variable passed into the following function. https://github.com/PolarizedLightFieldMicroscopy/GeoBirT/blob/f478893ad949a3b3f6e36fc86639c644dfbb0d62/VolumeRaytraceLFM/birefringence_implementations.py#L1546-L1569
• [x] Write a test to verify that the mla_index is passed through the functions appropriately.