tr_shader: factorize stage checking and move the code to a dedicated function

[illwieckz]

Yes another step in my effort to make that shader code less messy and easier to follow, here moving more things out of the lengthy FinishShader function and factorizing the code more to avoid copy-paste errors when modifying it.