fix: refraction mode enable

galacean · Dec 27, 2024 · aa434e4 · aa434e4
1 parent e062e18
commit aa434e4
Showing 1 changed file with 1 addition and 15 deletions.
diff --git a/packages/core/src/material/PBRMaterial.ts b/packages/core/src/material/PBRMaterial.ts
@@ -317,7 +317,7 @@ export class PBRMaterial extends PBRBaseMaterial {
   set refractionMode(value: RefractionMode) {
     if (value !== this._refractionMode) {
       this._refractionMode = value;
-      this._setRefractionMode(value);
+      this.shaderData.enableMacro("REFRACTION_MODE", value.toString());
     }
   }
 
@@ -524,18 +524,4 @@ export class PBRMaterial extends PBRBaseMaterial {
       }
     }
   }
-
-  private _setRefractionMode(refractionMode: RefractionMode): void {
-    switch (refractionMode) {
-      case RefractionMode.Sphere:
-        this.shaderData.enableMacro("REFRACTION_MODE", "0");
-        break;
-      case RefractionMode.Plane:
-        this.shaderData.enableMacro("REFRACTION_MODE", "1");
-        break;
-      case RefractionMode.Thin:
-        this.shaderData.enableMacro("REFRACTION_MODE", "2");
-        break;
-    }
-  }
 }