This commit is contained in:
anapnoe
2023-05-29 16:17:20 +03:00
154 changed files with 4535 additions and 3408 deletions
+93 -101
View File
@@ -1,120 +1,112 @@
let currentWidth = null;
let currentHeight = null;
let arFrameTimeout = setTimeout(function () {}, 0);
let arFrameTimeout = setTimeout(function() {}, 0);
function dimensionChange(e, is_width, is_height) {
if (is_width) {
currentWidth = e.target.value * 1.0;
}
if (is_height) {
currentHeight = e.target.value * 1.0;
}
var inImg2img =
gradioApp().querySelector("#tab_img2img").style.display == "block";
if (!inImg2img) {
return;
}
var targetElement = null;
var tabIndex = get_tab_index("mode_img2img");
if (tabIndex == 0) {
// img2img
targetElement = gradioApp().querySelector(
"#img2img_image div[data-testid=image] img"
);
} else if (tabIndex == 1) {
//Sketch
targetElement = gradioApp().querySelector(
"#img2img_sketch div[data-testid=image] img"
);
} else if (tabIndex == 2) {
// Inpaint
targetElement = gradioApp().querySelector(
"#img2maskimg div[data-testid=image] img"
);
} else if (tabIndex == 3) {
// Inpaint sketch
targetElement = gradioApp().querySelector(
"#inpaint_sketch div[data-testid=image] img"
);
}
if (targetElement) {
var arPreviewRect = gradioApp().querySelector("#imageARPreview");
if (!arPreviewRect) {
arPreviewRect = document.createElement("div");
arPreviewRect.id = "imageARPreview";
gradioApp().appendChild(arPreviewRect);
if (is_width) {
currentWidth = e.target.value * 1.0;
}
if (is_height) {
currentHeight = e.target.value * 1.0;
}
var viewportOffset = targetElement.getBoundingClientRect();
var inImg2img = gradioApp().querySelector("#tab_img2img").style.display == "block";
var viewportscale = Math.min(
targetElement.clientWidth / targetElement.naturalWidth,
targetElement.clientHeight / targetElement.naturalHeight
);
if (!inImg2img) {
return;
}
var scaledx = targetElement.naturalWidth * viewportscale;
var scaledy = targetElement.naturalHeight * viewportscale;
var targetElement = null;
var cleintRectTop = viewportOffset.top + window.scrollY;
var cleintRectLeft = viewportOffset.left + window.scrollX;
var cleintRectCentreY = cleintRectTop + targetElement.clientHeight / 2;
var cleintRectCentreX = cleintRectLeft + targetElement.clientWidth / 2;
var tabIndex = get_tab_index('mode_img2img');
if (tabIndex == 0) { // img2img
targetElement = gradioApp().querySelector('#img2img_image div[data-testid=image] img');
} else if (tabIndex == 1) { //Sketch
targetElement = gradioApp().querySelector('#img2img_sketch div[data-testid=image] img');
} else if (tabIndex == 2) { // Inpaint
targetElement = gradioApp().querySelector('#img2maskimg div[data-testid=image] img');
} else if (tabIndex == 3) { // Inpaint sketch
targetElement = gradioApp().querySelector('#inpaint_sketch div[data-testid=image] img');
}
var arscale = Math.min(scaledx / currentWidth, scaledy / currentHeight);
var arscaledx = currentWidth * arscale;
var arscaledy = currentHeight * arscale;
var arRectTop = cleintRectCentreY - arscaledy / 2;
var arRectLeft = cleintRectCentreX - arscaledx / 2;
var arRectWidth = arscaledx;
var arRectHeight = arscaledy;
if (targetElement) {
arPreviewRect.style.top = arRectTop + "px";
arPreviewRect.style.left = arRectLeft + "px";
arPreviewRect.style.width = arRectWidth + "px";
arPreviewRect.style.height = arRectHeight + "px";
var arPreviewRect = gradioApp().querySelector('#imageARPreview');
if (!arPreviewRect) {
arPreviewRect = document.createElement('div');
arPreviewRect.id = "imageARPreview";
gradioApp().appendChild(arPreviewRect);
}
clearTimeout(arFrameTimeout);
arFrameTimeout = setTimeout(function () {
arPreviewRect.style.display = "none";
}, 2000);
arPreviewRect.style.display = "block";
}
var viewportOffset = targetElement.getBoundingClientRect();
var viewportscale = Math.min(targetElement.clientWidth / targetElement.naturalWidth, targetElement.clientHeight / targetElement.naturalHeight);
var scaledx = targetElement.naturalWidth * viewportscale;
var scaledy = targetElement.naturalHeight * viewportscale;
var cleintRectTop = (viewportOffset.top + window.scrollY);
var cleintRectLeft = (viewportOffset.left + window.scrollX);
var cleintRectCentreY = cleintRectTop + (targetElement.clientHeight / 2);
var cleintRectCentreX = cleintRectLeft + (targetElement.clientWidth / 2);
var arscale = Math.min(scaledx / currentWidth, scaledy / currentHeight);
var arscaledx = currentWidth * arscale;
var arscaledy = currentHeight * arscale;
var arRectTop = cleintRectCentreY - (arscaledy / 2);
var arRectLeft = cleintRectCentreX - (arscaledx / 2);
var arRectWidth = arscaledx;
var arRectHeight = arscaledy;
arPreviewRect.style.top = arRectTop + 'px';
arPreviewRect.style.left = arRectLeft + 'px';
arPreviewRect.style.width = arRectWidth + 'px';
arPreviewRect.style.height = arRectHeight + 'px';
clearTimeout(arFrameTimeout);
arFrameTimeout = setTimeout(function() {
arPreviewRect.style.display = 'none';
}, 2000);
arPreviewRect.style.display = 'block';
}
}
onUiUpdate(function () {
var arPreviewRect = gradioApp().querySelector("#imageARPreview");
if (arPreviewRect) {
arPreviewRect.style.display = "none";
}
var tabImg2img = gradioApp().querySelector("#tab_img2img");
if (tabImg2img) {
var inImg2img = tabImg2img.style.display == "block";
if (inImg2img) {
let inputs = gradioApp().querySelectorAll("input");
inputs.forEach(function (e) {
var is_width = e.parentElement.id == "img2img_width";
var is_height = e.parentElement.id == "img2img_height";
if ((is_width || is_height) && !e.classList.contains("scrollwatch")) {
e.addEventListener("input", function (e) {
dimensionChange(e, is_width, is_height);
});
e.classList.add("scrollwatch");
}
if (is_width) {
currentWidth = e.value * 1.0;
}
if (is_height) {
currentHeight = e.value * 1.0;
}
});
onUiUpdate(function() {
var arPreviewRect = gradioApp().querySelector('#imageARPreview');
if (arPreviewRect) {
arPreviewRect.style.display = 'none';
}
}
});
var tabImg2img = gradioApp().querySelector("#tab_img2img");
if (tabImg2img) {
var inImg2img = tabImg2img.style.display == "block";
if (inImg2img) {
let inputs = gradioApp().querySelectorAll('input');
inputs.forEach(function(e) {
var is_width = e.parentElement.id == "img2img_width";
var is_height = e.parentElement.id == "img2img_height";
if ((is_width || is_height) && !e.classList.contains('scrollwatch')) {
e.addEventListener('input', function(e) {
dimensionChange(e, is_width, is_height);
});
e.classList.add('scrollwatch');
}
if (is_width) {
currentWidth = e.value * 1.0;
}
if (is_height) {
currentHeight = e.value * 1.0;
}
});
}
}
});