именно
сейчас код рассчитан только на самый элементарный кейс - когда два шара двигаются навстречу друг другу, с одинаковыми скоростями, и их центры едут по одной и той же прямой (центральное соударение).
для общего случая надо будет упороться с формулами, используя закон сохранения импульса. А если делать совсем как в жизни, то придется ещё и моменты импульсов добавить - шары при нецентральном соударении могут менять скорость вращения. Там полнейший ад получается