Merge 570be9e into 4a5cd1a

visgl · Mar 21, 2024 · e0b9a6b · e0b9a6b
2 parents 4a5cd1a + 570be9e
commit e0b9a6b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/modules/core/src/adapter/types/parameters.ts b/modules/core/src/adapter/types/parameters.ts
@@ -180,6 +180,9 @@ export type RenderPassParameters = {
   blendConstant?: number[]; // GPUColor
   /** Stencil operation "replace" sets the value to stencilReference */
   stencilReference?: number; // GPUStencilValue
+
+  /** Bitmask controlling which channels are are written to when drawing/clearing. defaulting to 0xF */
+  colorMask?: number;
 };
 
 export type RenderPipelineParameters = RasterizationParameters &

diff --git a/modules/webgl/src/adapter/resources/webgl-render-pass.ts b/modules/webgl/src/adapter/resources/webgl-render-pass.ts
@@ -17,6 +17,8 @@ const GL_STENCIL_BUFFER_BIT = 0x00000400;
 const GL_COLOR_BUFFER_BIT = 0x00004000;
 
 const GL_COLOR = 0x1800;
+const {RED, GREEN, BLUE, ALPHA} = GPUColorWrite;
+const COLOR_CHANNELS = [RED, GREEN, BLUE, ALPHA];
 
 export class WEBGLRenderPass extends RenderPass {
   readonly device: WebGLDevice;
@@ -96,6 +98,12 @@ export class WEBGLRenderPass extends RenderPass {
       parameters[GL.STENCIL_REF] = parameters.stencilReference;
     }
 
+    if (parameters.colorMask) {
+      glParameters.colorMask = COLOR_CHANNELS.map(channel =>
+        Boolean(channel & parameters.colorMask)
+      );
+    }
+
     this.glParameters = glParameters;
 
     setGLParameters(this.device.gl, glParameters);