Thanks for your answer Structure, the code works fine as long as the light parameter Zrot is 0, in cases where Z rotatation is different from zero the results are wrong.

There should be some way to translate any light position to one where the Zrot parameter is 0 and then run the script, but I haven't figured out how to do it.