Improve integration of trajectory modules for analytical simulations

Currently the trajectories have to be precalculated for analytical simulations, this results in very large memory footprints especially if trajectories/ r-vectors per sampling points are streamed. I assume this can be improved if the trajectories are evaluated for each batch on demand.

Suggested solutions:

• Integrate the trajectory-module call into the tf.data.Dataset pipeline