epraya.oneori

oneori(nx, ny, nz)

Wrap function to use JPadaptarray and JNresina with the JAX.vmap implementation. Vmap vectorizes this two functions, making them work with block of data instead of individual process in a loop. The blocks are divided in parts of csize size, and process with the function Processvmap, that calculates the total values of resonant fields, intenitys and transitions.

Parameters:

• nx (jax.np.array) – Vectors (points in the grid) to consider in the calculation of the spectrum in the x direction.

• ny (jax.np.array) – Vectors (points in the grid) to consider in the calculation of the spectrum in the y direction.

• nz (jax.np.array) – Vectors (points in the grid) to consider in the calculation of the spectrum in the z direction.

Returns:

• resfield (jax.np.array) – List of the resonant fields of the system.

• intensy (jax.np.array) – List of the intensity of the spectrum, evaluated in the resonant fields.

• ntrans (jax.np.array) – Zero dimensional array with the number of possible transitions related to the resonant fields.