prepare_trajectory_data function

Prepares trajectory data for an ENA plot.