Update ShaderSourceProcessor_inline.h

Author: aaaapai

MobileGL/MG_Util/ShaderTranspiler/ShaderSourceProcessor_inline.h

@@ -1137,6 +1137,113 @@ vec4 GI_TemporalFilter() {
     }
 
 
+// ==================== 3. 注入光照安全补丁 ====================
+    
+    // 补丁1: 防止除零（SHADING_STRENGTH）
+    const char* str_rcp_shading = "rcp(SHADING_STRENGTH)";
+    if (source.find(str_rcp_shading) != String::npos) {
+        SizeT pos = source.find(str_rcp_shading);
+        while (pos != String::npos) {
+            MGLOG_D("Applying safety patch 1: Safe SHADING_STRENGTH division");
+            std::string replacement = "(SHADING_STRENGTH > 0.001 ? 1.0 / SHADING_STRENGTH : 1.0)";
+            source.replace(pos, strlen(str_rcp_shading), replacement);
+            pos = source.find(str_rcp_shading, pos + replacement.length());
+        }
+    }
+    
+    // 补丁2: 防止pow函数接收负值
+    const char* str_pow8 = "pow8(blocklight)";
+    if (source.find(str_pow8) != String::npos) {
+        SizeT pos = source.find(str_pow8);
+        while (pos != String::npos) {
+            MGLOG_D("Applying safety patch 2: Safe pow8");
+            std::string replacement = "pow(max(blocklight, 0.0), 8.0)";
+            source.replace(pos, strlen(str_pow8), replacement);
+            pos = source.find(str_pow8, pos + replacement.length());
+        }
+    }
+    
+    // 补丁3: 浮点相等性检查
+    const char* str_float_eq = "if (distance_weight_sum == 0.0)";
+    if (source.find(str_float_eq) != String::npos) {
+        SizeT pos = source.find(str_float_eq);
+        while (pos != String::npos) {
+            MGLOG_D("Applying safety patch 3: Float equality check");
+            std::string replacement = "if (distance_weight_sum <= 0.0001)";
+            source.replace(pos, strlen(str_float_eq), replacement);
+            pos = source.find(str_float_eq, pos + replacement.length());
+        }
+    }
+    
+    // 补丁4: 防止光照方向y分量为负导致的异常
+    const char* str_lightdir_div = "clamp01(rcp(0.02) * light_dir.y)";
+    if (source.find(str_lightdir_div) != String::npos) {
+        SizeT pos = source.find(str_lightdir_div);
+        while (pos != String::npos) {
+            MGLOG_D("Applying safety patch 4: Safe light_dir.y division");
+            std::string replacement = "clamp01(max(light_dir.y, 0.0) * 50.0)";
+            source.replace(pos, strlen(str_lightdir_div), replacement);
+            pos = source.find(str_lightdir_div, pos + replacement.length());
+        }
+    }
+    
+    // 补丁5: 防止NaN传播
+    const char* str_return_lighting = "return max0(lighting) * material.albedo * rcp_pi";
+    if (source.find(str_return_lighting) != String::npos) {
+        SizeT pos = source.find(str_return_lighting);
+        while (pos != String::npos) {
+            MGLOG_D("Applying safety patch 5: NaN protection");
+            
+            // 构建替换代码
+            std::string replacement = R"(    vec3 result = max0(lighting) * material.albedo * rcp_pi;
+    // 防止NaN和无穷大
+    result = clamp(result, vec3(0.0), vec3(100.0));
+    result = isnan(result.x) ? vec3(0.0) : result;
+    return result)";
+            
+            source.replace(pos, strlen(str_return_lighting), replacement);
+            pos = source.find(str_return_lighting, pos + replacement.length());
+        }
+    }
+    
+    // 补丁6: SSS系数安全计算
+    const char* str_sss_coeff = "vec3 coeff = albedo * inversesqrt(dot(albedo, luminance_weights) + eps);";
+    if (source.find(str_sss_coeff) != String::npos) {
+        SizeT pos = source.find(str_sss_coeff);
+        while (pos != String::npos) {
+            MGLOG_D("Applying safety patch 6: Safe SSS coefficient");
+            std::string replacement = "vec3 coeff = albedo * inversesqrt(max(dot(albedo, luminance_weights), 0.001) + eps);";
+            source.replace(pos, strlen(str_sss_coeff), replacement);
+            pos = source.find(str_sss_coeff, pos + replacement.length());
+        }
+    }
+    
+    // 补丁7: 最小光照保护
+    const char* str_diffuse_shadows = "lighting += diffuse * shadows";
+    if (source.find(str_diffuse_shadows) != String::npos) {
+        SizeT pos = source.find(str_diffuse_shadows);
+        while (pos != String::npos) {
+            MGLOG_D("Applying safety patch 7: Minimum lighting");
+            std::string replacement = "lighting += max(diffuse, vec3(0.01)) * shadows";
+            source.replace(pos, strlen(str_diffuse_shadows), replacement);
+            pos = source.find(str_diffuse_shadows, pos + replacement.length());
+        }
+    }
+    
+    // 补丁8: 安全的lift函数调用
+    const char* str_lift_nol = "lift(max0(NoL), 0.25 * rcp(SHADING_STRENGTH))";
+    if (source.find(str_lift_nol) != String::npos) {
+        SizeT pos = source.find(str_lift_nol);
+        while (pos != String::npos) {
+            MGLOG_D("Applying safety patch 8: Safe lift function");
+            std::string replacement = "max(lift(max0(NoL), 0.25 * (SHADING_STRENGTH > 0.001 ? 1.0 / SHADING_STRENGTH : 4.0)), 0.01)";
+            source.replace(pos, strlen(str_lift_nol), replacement);
+            pos = source.find(str_lift_nol, pos + replacement.length());
+        }
+}
+
+
+
 }