Add Anisotropy

google · Jul 11, 2023 · 78c10ae · 78c10ae
1 parent 22c0997
commit 78c10ae
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 0 deletions.
diff --git a/packages/model-viewer/src/features/scene-graph/api.ts b/packages/model-viewer/src/features/scene-graph/api.ts
@@ -147,6 +147,9 @@ export declare interface Material {
   readonly iridescenceThicknessMinimum: number;
   readonly iridescenceThicknessMaximum: number;
   readonly iridescenceThicknessTexture: TextureInfo;
+  readonly anisotropyStrength: number;
+  readonly anisotropyRotation: number;
+  readonly anisotropyTexture: TextureInfo;
 
   setEmissiveStrength(emissiveStrength: number): void;
   setClearcoatFactor(clearcoatFactor: number): void;
@@ -165,6 +168,8 @@ export declare interface Material {
   setIridescenceIor(ior: number): void;
   setIridescenceThicknessMinimum(thicknessMin: number): void;
   setIridescenceThicknessMaximum(thicknessMax: number): void;
+  setAnisotropyStrength(strength: number): void;
+  setAnisotropyRotation(rotation: number): void;
 
   /**
    * The PBRMetallicRoughness configuration of the material.

diff --git a/packages/model-viewer/src/features/scene-graph/material.ts b/packages/model-viewer/src/features/scene-graph/material.ts
@@ -142,6 +142,9 @@ export class Material extends ThreeDOMElement implements MaterialInterface {
     createTextureInfo(TextureUsage.Thickness);
     createTextureInfo(TextureUsage.Specular);
     createTextureInfo(TextureUsage.SpecularColor);
+    createTextureInfo(TextureUsage.Iridescence);
+    createTextureInfo(TextureUsage.IridescenceThickness);
+    createTextureInfo(TextureUsage.Anisotropy);
   }
 
   async[$getLoadedMaterial](): Promise<MeshPhysicalMaterial> {
@@ -659,4 +662,36 @@ export class Material extends ThreeDOMElement implements MaterialInterface {
   }
 
   // KHR_materials_anisotropy
+  get anisotropyStrength(): number {
+    this[$ensureMaterialIsLoaded]();
+    return (this[$backingThreeMaterial] as any).anisotropy;
+  }
+
+  get anisotropyRotation(): number {
+    this[$ensureMaterialIsLoaded]();
+    return (this[$backingThreeMaterial] as any).anisotropyRotation;
+  }
+
+  get anisotropyTexture(): TextureInfo {
+    this[$ensureMaterialIsLoaded]();
+    return this[$pbrTextures].get(TextureUsage.Anisotropy)!;
+  }
+
+  setAnisotropyStrength(strength: number) {
+    this[$ensureMaterialIsLoaded]();
+    for (const material of this[$correlatedObjects] as
+         Set<MeshPhysicalMaterial>) {
+      (material as any).anisotropy = strength;
+    }
+    this[$onUpdate]();
+  }
+
+  setAnisotropyRotation(rotation: number) {
+    this[$ensureMaterialIsLoaded]();
+    for (const material of this[$correlatedObjects] as
+         Set<MeshPhysicalMaterial>) {
+      (material as any).anisotropyRotation = rotation;
+    }
+    this[$onUpdate]();
+  }
 }
diff --git a/packages/model-viewer/src/features/scene-graph/texture-info.ts b/packages/model-viewer/src/features/scene-graph/texture-info.ts
@@ -44,6 +44,7 @@ export enum TextureUsage {
   SpecularColor,
   Iridescence,
   IridescenceThickness,
+  Anisotropy,
 }
 
 interface TextureTransform {
@@ -189,6 +190,9 @@ export class TextureInfo implements TextureInfoInterface {
           case TextureUsage.IridescenceThickness:
             material.iridescenceThicknessMap = threeTexture;
             break;
+          case TextureUsage.Anisotropy:
+            (material as any).anisotropyMap = threeTexture;
+            break;
           default:
         }
         material.needsUpdate = true;