114 lines
3.5 KiB
JavaScript
114 lines
3.5 KiB
JavaScript
|
|
let currentWidth = null;
|
|
let currentHeight = null;
|
|
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 = Boolean(gradioApp().querySelector("button.rounded-t-lg.border-gray-200"))
|
|
|
|
if(!inImg2img){
|
|
return;
|
|
}
|
|
|
|
var targetElement = null;
|
|
|
|
var tabIndex = get_tab_index('mode_img2img')
|
|
if(tabIndex == 0){ // img2img
|
|
targetElement = gradioApp().querySelector('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().getRootNode().appendChild(arPreviewRect)
|
|
}
|
|
|
|
|
|
|
|
var viewportOffset = targetElement.getBoundingClientRect();
|
|
|
|
viewportscale = Math.min( targetElement.clientWidth/targetElement.naturalWidth, targetElement.clientHeight/targetElement.naturalHeight )
|
|
|
|
scaledx = targetElement.naturalWidth*viewportscale
|
|
scaledy = targetElement.naturalHeight*viewportscale
|
|
|
|
cleintRectTop = (viewportOffset.top+window.scrollY)
|
|
cleintRectLeft = (viewportOffset.left+window.scrollX)
|
|
cleintRectCentreY = cleintRectTop + (targetElement.clientHeight/2)
|
|
cleintRectCentreX = cleintRectLeft + (targetElement.clientWidth/2)
|
|
|
|
viewRectTop = cleintRectCentreY-(scaledy/2)
|
|
viewRectLeft = cleintRectCentreX-(scaledx/2)
|
|
arRectWidth = scaledx
|
|
arRectHeight = scaledy
|
|
|
|
arscale = Math.min( arRectWidth/currentWidth, arRectHeight/currentHeight )
|
|
arscaledx = currentWidth*arscale
|
|
arscaledy = currentHeight*arscale
|
|
|
|
arRectTop = cleintRectCentreY-(arscaledy/2)
|
|
arRectLeft = cleintRectCentreX-(arscaledx/2)
|
|
arRectWidth = arscaledx
|
|
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 inImg2img = Boolean(gradioApp().querySelector("button.rounded-t-lg.border-gray-200"))
|
|
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
|
|
}
|
|
})
|
|
}
|
|
});
|