Are there altered forms of this ray equation for nonlinear cases, like bending light through a lens, for example?

I think in cases like that it might be easy to decompose the ray into multiple segments that meet where the ray interacts with other surfaces.