Usé el siguiente código basado enest

ballA.vx = (u1x * (m1 - m2) + 2 * m2 * u2x) / (m1 + m2);
ballA.vy = (u1y * (m1 - m2) + 2 * m2 * u2y) / (m1 + m2);

ballB.vx = (u2x * (m2 - m1) + 2 * m1 * u1x) / (m1 + m2);
ballB.vy = (u2y * (m2 - m1) + 2 * m1 * u1y) / (m1 + m2);

pero obviamente no funciona bien ya que la fórmula está diseñada para colisiones unidimensionales.

Así que traté de usar la siguiente fórmula deesta secció.

Pero el problema es que no sé cuál es el ángulo de desviación y cómo calcularlo. Además, ¿cómo tener en cuenta el coeficiente de rebote en esta fórmula?

Edita: Puede que no haya sido claro. El código anteriorhac work, aunque puede no ser el comportamiento esperado, ya que la fórmula original está diseñada para colisiones 1D. Los problemas que estoy tratando son:

Cuál es el 2D equivalente? Cómo tomar el coeficiente de rebote ¿en cuenta Cómo calcular eldirecció (que se expresa convx yvy) de las dos bolas después de la colisión?