Better shader generator function support