add invert logic to grayscale mask composite

Author: dunkeroni

invokeai/frontend/web/src/features/controlLayers/konva/CanvasCompositorModule.ts

@@ -442,12 +442,13 @@ export class CanvasCompositorModule extends CanvasModuleBase {
     adapters: CanvasEntityAdapterInpaintMask[],
     rect: Rect,
     attribute: 'noiseLevel' | 'denoiseLimit' = 'noiseLevel',
+    invertMask: boolean = false,
     uploadOptions: SetOptional<Omit<UploadImageArg, 'file'>, 'image_category'> = { is_intermediate: true },
     forceUpload?: boolean
   ): Promise<ImageDTO> => {
     assert(rect.width > 0 && rect.height > 0, 'Unable to rasterize empty rect');
     // Use a unique hash that includes the attribute name for caching
-    const hash = this.getCompositeHash(adapters, { rect, attribute, grayscale: true });
+    const hash = this.getCompositeHash(adapters, { rect, attribute, invertMask, grayscale: true });
     const cachedImageName = forceUpload ? undefined : this.manager.cache.imageNameCache.get(hash);
 
     let imageDTO: ImageDTO | null = null;
@@ -507,7 +508,7 @@ export class CanvasCompositorModule extends CanvasModuleBase {
           // input has transparency
           // Calculate grayscale value: white (255) for no mask, darker for stronger mask
           let grayValue = 255; // Default to white for unmasked areas
-          if ((data[i + 3] ?? 0) > 127) {
+          if (invertMask ? (data[i + 3] ?? 0) < 128 : (data[i + 3] ?? 0) > 127) {
             grayValue = Math.max(0, Math.min(255, 255 - Math.round(255 * attributeValue)));
           }
 

invokeai/frontend/web/src/features/nodes/util/graph/generation/addInpaint.ts

@@ -69,17 +69,24 @@ export const addInpaint = async ({
   // Create a composite noise mask if we have any adapters with noise settings
   let noiseMaskImage: ImageDTO | null = null;
   if (noiseMaskAdapters.length > 0) {
-    noiseMaskImage = await manager.compositor.getGrayscaleMaskCompositeImageDTO(noiseMaskAdapters, rect, 'noiseLevel', {
-      is_intermediate: true,
-      silent: true,
-    });
+    noiseMaskImage = await manager.compositor.getGrayscaleMaskCompositeImageDTO(
+      noiseMaskAdapters,
+      rect,
+      'noiseLevel',
+      canvasSettings.preserveMask,
+      {
+        is_intermediate: true,
+        silent: true,
+      }
+    );
   }
 
   // Create a composite denoise limit mask
   const maskImage = await manager.compositor.getGrayscaleMaskCompositeImageDTO(
     inpaintMaskAdapters, // denoise limit defaults to 1 for masks that don't have it
     rect,
     'denoiseLimit',
+    canvasSettings.preserveMask,
     {
       is_intermediate: true,
       silent: true,

invokeai/frontend/web/src/features/nodes/util/graph/generation/addOutpaint.ts

@@ -72,17 +72,24 @@ export const addOutpaint = async ({
   // Create a composite noise mask if we have any adapters with noise settings
   let noiseMaskImage: ImageDTO | null = null;
   if (noiseMaskAdapters.length > 0) {
-    noiseMaskImage = await manager.compositor.getGrayscaleMaskCompositeImageDTO(noiseMaskAdapters, rect, 'noiseLevel', {
-      is_intermediate: true,
-      silent: true,
-    });
+    noiseMaskImage = await manager.compositor.getGrayscaleMaskCompositeImageDTO(
+      noiseMaskAdapters,
+      rect,
+      'noiseLevel',
+      canvasSettings.preserveMask,
+      {
+        is_intermediate: true,
+        silent: true,
+      }
+    );
   }
 
   // Create a composite denoise limit mask
   const maskImage = await manager.compositor.getGrayscaleMaskCompositeImageDTO(
     inpaintMaskAdapters, // denoise limit defaults to 1 for masks that don't have it
     rect,
     'denoiseLimit',
+    canvasSettings.preserveMask,
     {
       is_intermediate: true,
       silent: true,