Return style function from shader