基于Qt的OpenGL可编程管线学习（4）- 使用Subroutine绘制不同光照的模型

使用Subroutine在shader中封装不同的函数，在CPU端选择调用那个函数

效果如下图所示

左侧：环境光

中间：环境光 + 漫反射

右侧：环境光 + 漫反射 + 高光

1、Subroutine 在shader中的内容

subroutine vec4 SurfaceColor(); subroutine uniform SurfaceColor U_SurfaceColor;

定义SurfaceColor()函数类型

环境光函数

// Ambient subroutine (SurfaceColor) vec4 Ambient() {         // Ambient         vec4 M_AmbientLightColor = vec4(0.2, 0.2, 0.2, 1.0);         vec4 M_AmbientMaterial = vec4(0.2, 0.2, 0.2, 1.0);         vec4 ambientColor = M_AmbientLightColor * M_AmbientMaterial;         return ambientColor; }

环境光+漫反射函数

subroutine (SurfaceColor) vec4 Diffuse() {         // Ambient         vec4 M_AmbientLightColor = vec4(0.2, 0.2, 0.2, 1.0);         vec4 M_AmbientMaterial = vec4(0.2, 0.2, 0.2, 1.0);         vec4 ambientColor = M_AmbientLightColor * M_AmbientMaterial;         // Diffuse         vec3 M_LightPos = vec3(10.0, 10.0, 0.0);         vec3 LightNormal = normalize(M_LightPos);       // 指向光源的单位向量         vec3 NormalNormal = normalize(M_normal);      //  法线的单位向量         // 点乘获取光照强度         vec4 M_DiffuseLightColor = vec4(1.0, 1.0, 1.0, 1.0);         vec4 M_DiffuseMaterial = vec4(0.8, 0.8, 0.8, 1.0);         vec4 diffuseColor = M_DiffuseLightColor * M_DiffuseMaterial * max(0.0, dot(NormalNormal, LightNormal));         return ambientColor + texture2D(U_MainTexture, M_coord) * diffuseColor; }

镜面反射函数

subroutine (SurfaceColor) vec4 Specular() {         // Ambient         vec4 M_AmbientLightColor = vec4(0.2, 0.2, 0.2, 1.0);         vec4 M_AmbientMaterial = vec4(0.2, 0.2, 0.2, 1.0);         vec4 ambientColor = M_AmbientLightColor * M_AmbientMaterial;         // Diffuse         vec3 M_LightPos = vec3(10.0, 10.0, 0.0);         vec3 LightNormal = normalize(M_LightPos);       // 指向光源的单位向量         vec3 NormalNormal = normalize(M_normal);      //  法线的单位向量         // 点乘获取光照强度         vec4 M_DiffuseLightColor = vec4(1.0, 1.0, 1.0, 1.0);         vec4 M_DiffuseMaterial = vec4(0.8, 0.8, 0.8, 1.0);         vec4 diffuseColor = M_DiffuseLightColor * M_DiffuseMaterial * max(0.0, dot(NormalNormal, LightNormal));         // 镜面反射         vec4 specularLightColor = vec4(1.0, 1.0, 1.0, 1.0);         vec4 specularMaterial = vec4(0.4, 0.4, 0.4, 1.0);         vec3 reflerDir = normalize(reflect(-LightNormal, NormalNormal));         vec3 eyeDir = normalize(vec3(0.0) - M_WordPos.xyz);         vec4 specularColor = specularLightColor * specularMaterial * pow(max(0.0, dot(reflerDir, eyeDir)), 180);         return ambientColor + texture2D(U_MainTexture, M_coord) * diffuseColor + specularColor; }

main函数

void main() {         gl_FragColor = U_SurfaceColor(); }

2、CPU端设置调用subrountine

获取subrountine的Location

m_SubRoutineLocation = OpenGLCore->glGetSubroutineUniformLocation(programId, \                        GL_FRAGMENT_SHADER, "U_SurfaceColor"); // 获取 SubRoutine Index m_AmbientIndex=OpenGLCore->glGetSubroutineIndex(programId,GL_FRAGMENT_SHADER,"Ambient"); m_DiffuseIndex=OpenGLCore->glGetSubroutineIndex(programId,GL_FRAGMENT_SHADER,"Diffuse"); m_SpecularIndex=OpenGLCore->glGetSubroutineIndex(programId,GL_FRAGMENT_SHADER,"Specular");

为U_SurfaceColor设置值

OpenGLCore->glUniformSubroutinesuiv(GL_FRAGMENT_SHADER, 1, &m_AmbientIndex); OpenGLCore->glUniformSubroutinesuiv(GL_FRAGMENT_SHADER, 1, &m_DiffuseIndex); OpenGLCore->glUniformSubroutinesuiv(GL_FRAGMENT_SHADER, 1, &m_SpecularIndex);