Visualizing the force around a trap

This example show how to easily visualize the radiation force in the vicinity of an object.

import numpy as np
import levitate

pos = np.array([0, 0, 60e-3])
array = levitate.arrays.RectangularArray(16)
state = levitate.complex(array.focus_phases(pos) + array.signature(stype='twin'))

radii = [1e-3, 2e-3, 4e-3, 8e-3, 16e-3]
for radius in radii:
    array.force_diagram.append([pos, {'radius': radius, 'name': '{} mm'.format(radius * 1e3)}])
array.force_diagram(state).show()