To be honest, I don’t know what a lot of the movement functions are doing but if you can identify where the Z based gravity was being used and what the math calculation was doing with it you can try to turn rewrite that little bit so that the calculation will work with a varying gravity direction.
Unfortunately I’m really busy at work so I’ve not looked at this for ages and won’t get to any time soon.