From e92976d1589c7cbcacef1cb55193eee207c8f91d Mon Sep 17 00:00:00 2001
From: tim h <zero01101@gmail.com>
Date: Thu, 24 Nov 2022 22:44:38 -0600
Subject: [PATCH] hacky re-addition of painted mask auto disappearance after
 image generation

Former-commit-id: f34d1d077f0be69fce3f3b339eac596863d5846a
---
 index.html  |  6 +++---
 js/index.js | 16 ++++++++++++++++
 2 files changed, 19 insertions(+), 3 deletions(-)

diff --git a/index.html b/index.html
index 43e743a..df01f2e 100644
--- a/index.html
+++ b/index.html
@@ -35,7 +35,7 @@
 					<label for="prompt">Prompt:</label>
 					<br />
 					<textarea id="prompt">
-oceanographic study, underwater wildlife, award winning</textarea
+ocean floor scientific expedition, underwater wildlife</textarea
 					>
 					<br />
 					<label for="negPrompt">Negative prompt:</label>
@@ -44,7 +44,7 @@ oceanographic study, underwater wildlife, award winning</textarea
 people, person, humans, human, divers, diver, glitch, error, text, watermark, bad quality, blurry</textarea
 					>
 					<br />
-					<hr />
+					<!-- <hr /> -->
 				</div>
 				<!-- SD section -->
 				<button type="button" class="collapsible">
@@ -139,7 +139,7 @@ people, person, humans, human, divers, diver, glitch, error, text, watermark, ba
 						</a>
 					</span>
 					<br />
-					<hr />
+					<!-- <hr /> -->
 				</div>
 				<div style="display: flex; align-items: center">
 					<div
diff --git a/js/index.js b/js/index.js
index c81d30c..9124b13 100644
--- a/js/index.js
+++ b/js/index.js
@@ -214,6 +214,7 @@ function imageAcceptReject(x, y, data, extra = null) {
 	document.getElementById("progressDiv").remove();
 	const img = new Image();
 	img.onload = function () {
+		backupAndClearMask(x, y, img.width, img.height);
 		tempCtx.drawImage(img, x, y); //imgCtx for actual image, tmp for... holding?
 		var div = document.createElement("div");
 		div.id = "veryTempDiv";
@@ -300,6 +301,21 @@ function removeChoiceButtons(evt) {
 	tempCtx.clearRect(0, 0, tempCanvas.width, tempCanvas.height);
 }
 
+function backupAndClearMask(x, y, w, h) {
+	var clearArea = maskPaintCtx.createImageData(w, h);
+	backupMaskChunk = maskPaintCtx.getImageData(x, y, w, h);
+	backupMaskX = x;
+	backupMaskY = y;
+	var clearD = clearArea.data;
+	for (i = 0; i < clearD.length; i += 4) {
+		clearD[i] = 0;
+		clearD[i + 1] = 0;
+		clearD[i + 2] = 0;
+		clearD[i + 3] = 0;
+	}
+	maskPaintCtx.putImageData(clearArea, x, y);
+}
+
 function restoreBackupMask() {
 	// reapply mask if exists
 	if (backupMaskChunk != null && backupMaskX != null && backupMaskY != null) {