Modelling ballast via a non-linear lattice to assess its compaction behaviour at railway transition zones