stable-diffusion-webui

1

2
let currentWidth = null;
3
let currentHeight = null;
4
let arFrameTimeout = setTimeout(function() {}, 0);
5

6
function dimensionChange(e, is_width, is_height) {
7

8
    if (is_width) {
9
        currentWidth = e.target.value * 1.0;
10
    }
11
    if (is_height) {
12
        currentHeight = e.target.value * 1.0;
13
    }
14

15
    var inImg2img = gradioApp().querySelector("#tab_img2img").style.display == "block";
16

17
    if (!inImg2img) {
18
        return;
19
    }
20

21
    var targetElement = null;
22

23
    var tabIndex = get_tab_index('mode_img2img');
24
    if (tabIndex == 0) { // img2img
25
        targetElement = gradioApp().querySelector('#img2img_image div[data-testid=image] img');
26
    } else if (tabIndex == 1) { //Sketch
27
        targetElement = gradioApp().querySelector('#img2img_sketch div[data-testid=image] img');
28
    } else if (tabIndex == 2) { // Inpaint
29
        targetElement = gradioApp().querySelector('#img2maskimg div[data-testid=image] img');
30
    } else if (tabIndex == 3) { // Inpaint sketch
31
        targetElement = gradioApp().querySelector('#inpaint_sketch div[data-testid=image] img');
32
    }
33

34

35
    if (targetElement) {
36

37
        var arPreviewRect = gradioApp().querySelector('#imageARPreview');
38
        if (!arPreviewRect) {
39
            arPreviewRect = document.createElement('div');
40
            arPreviewRect.id = "imageARPreview";
41
            gradioApp().appendChild(arPreviewRect);
42
        }
43

44

45

46
        var viewportOffset = targetElement.getBoundingClientRect();
47

48
        var viewportscale = Math.min(targetElement.clientWidth / targetElement.naturalWidth, targetElement.clientHeight / targetElement.naturalHeight);
49

50
        var scaledx = targetElement.naturalWidth * viewportscale;
51
        var scaledy = targetElement.naturalHeight * viewportscale;
52

53
        var cleintRectTop = (viewportOffset.top + window.scrollY);
54
        var cleintRectLeft = (viewportOffset.left + window.scrollX);
55
        var cleintRectCentreY = cleintRectTop + (targetElement.clientHeight / 2);
56
        var cleintRectCentreX = cleintRectLeft + (targetElement.clientWidth / 2);
57

58
        var arscale = Math.min(scaledx / currentWidth, scaledy / currentHeight);
59
        var arscaledx = currentWidth * arscale;
60
        var arscaledy = currentHeight * arscale;
61

62
        var arRectTop = cleintRectCentreY - (arscaledy / 2);
63
        var arRectLeft = cleintRectCentreX - (arscaledx / 2);
64
        var arRectWidth = arscaledx;
65
        var arRectHeight = arscaledy;
66

67
        arPreviewRect.style.top = arRectTop + 'px';
68
        arPreviewRect.style.left = arRectLeft + 'px';
69
        arPreviewRect.style.width = arRectWidth + 'px';
70
        arPreviewRect.style.height = arRectHeight + 'px';
71

72
        clearTimeout(arFrameTimeout);
73
        arFrameTimeout = setTimeout(function() {
74
            arPreviewRect.style.display = 'none';
75
        }, 2000);
76

77
        arPreviewRect.style.display = 'block';
78

79
    }
80

81
}
82

83

84
onAfterUiUpdate(function() {
85
    var arPreviewRect = gradioApp().querySelector('#imageARPreview');
86
    if (arPreviewRect) {
87
        arPreviewRect.style.display = 'none';
88
    }
89
    var tabImg2img = gradioApp().querySelector("#tab_img2img");
90
    if (tabImg2img) {
91
        var inImg2img = tabImg2img.style.display == "block";
92
        if (inImg2img) {
93
            let inputs = gradioApp().querySelectorAll('input');
94
            inputs.forEach(function(e) {
95
                var is_width = e.parentElement.id == "img2img_width";
96
                var is_height = e.parentElement.id == "img2img_height";
97

98
                if ((is_width || is_height) && !e.classList.contains('scrollwatch')) {
99
                    e.addEventListener('input', function(e) {
100
                        dimensionChange(e, is_width, is_height);
101
                    });
102
                    e.classList.add('scrollwatch');
103
                }
104
                if (is_width) {
105
                    currentWidth = e.value * 1.0;
106
                }
107
                if (is_height) {
108
                    currentHeight = e.value * 1.0;
109
                }
110
            });
111
        }
112
    }
113
});
114