From e0aecb1dc195eb2c5c7a29cb67fbad052d5182f3 Mon Sep 17 00:00:00 2001
From: Abdullah Alfaraj <abdullah.a.f2014@gmail.com>
Date: Thu, 23 Mar 2023 15:12:37 +0300
Subject: [PATCH] methods to visualize both the mask and the mask after
 expansion

---
 index.js           |  1 +
 utility/session.js |  1 +
 utility/tab/sd.js  | 26 +++++++++++++++++++++++++-
 3 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/index.js b/index.js
index 655163a..4c389cd 100644
--- a/index.js
+++ b/index.js
@@ -1897,6 +1897,7 @@ async function getSettings() {
                     iterations
                 )
                 if (mask) {
+                    g_generation_session.base64maskExpansionImage = mask
                     payload['mask'] = mask
                 }
             }
diff --git a/utility/session.js b/utility/session.js
index 5ab42f1..ac767b2 100644
--- a/utility/session.js
+++ b/utility/session.js
@@ -31,6 +31,7 @@ class GenerationSession {
         this.base64OutputImages = {} //image_id/path => base64_image
         this.base64initImages = {} //init_image_path => base64
         this.base64maskImage = []
+        this.base64maskExpansionImage
         this.activeBase64InitImage
         this.activeBase64MaskImage
         this.image_paths_to_layers = {}
diff --git a/utility/tab/sd.js b/utility/tab/sd.js
index 6e03456..7a0ade1 100644
--- a/utility/tab/sd.js
+++ b/utility/tab/sd.js
@@ -60,8 +60,32 @@ async function updateClickEventHandler(current_version) {
     }
 }
 
+function viewMaskExpansion() {
+    if (g_generation_session.base64maskExpansionImage) {
+        const mask_src = general.base64ToBase64Url(
+            g_generation_session.base64maskExpansionImage
+        )
+        html_manip.setInitImageMaskSrc(mask_src)
+    } else {
+        console.log(
+            'the mask has not been expanded, g_generation_session.base64maskExpansionImage is empty'
+        )
+    }
+}
+function viewDrawnMask() {
+    //this is the generated mask or user drawn mask, but it's not the mask after expansion
+    if (g_generation_session.activeBase64MaskImage) {
+        const mask_src = general.base64ToBase64Url(
+            g_generation_session.activeBase64MaskImage
+        )
+        html_manip.setInitImageMaskSrc(mask_src)
+    } else {
+        console.log('no mask is available')
+    }
+}
+
 document.getElementById('btnUpdate').addEventListener('click', async () => {
     await updateClickEventHandler(g_version)
 })
 
-module.exports = { updateClickEventHandler }
+module.exports = { updateClickEventHandler, viewMaskExpansion, viewDrawnMask }