Update zigzag shaders for new shader spec

[lachlanmcdonald]

Update the zigzag2, zigzag3 and zigzag_range shaders to match the same approach as the new diagonal shader.