draw: Make it more clear that *_jit_context points to pipe_viewport_state structures.