glsl: Rework assignments with write_masks to have LHS chan count match RHS.