From 04dcc006700d5d5e26fb2aba35317a9ee4c63edd Mon Sep 17 00:00:00 2001
From: shadowcz007 <chizhiwei007@163.com>
Date: Fri, 22 Dec 2023 20:20:03 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=8F=AF=E8=A7=86=E5=8C=96?=
 =?UTF-8?q?=E9=80=89=E5=8C=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 nodes/ImageNode.py             | 14 +-----
 web/javascript/layer_mixlab.js | 86 ++++++++++++++++++++++------------
 2 files changed, 57 insertions(+), 43 deletions(-)

diff --git a/nodes/ImageNode.py b/nodes/ImageNode.py
index d79b24c..0d36334 100644
--- a/nodes/ImageNode.py
+++ b/nodes/ImageNode.py
@@ -1186,18 +1186,7 @@ class NewLayer:
             mask=pil2tensor(mask)
         else:
             mask=mask[0]
-
-        canvas_h=0
-        canvas_w=0
-        canvas_base64=""
-        if canvas!=None: 
-            c=tensor2pil(canvas[0])
-            w, h = c.size
-            canvas_h=h
-            canvas_w=w
-            canvas_base64 = base64.b64encode(c.tobytes()).decode('utf-8')
         
-
         layer_n=[{
             "x":x[0],
             "y":y[0],
@@ -1212,8 +1201,7 @@ class NewLayer:
         if layers!=None:
             layer_n=layer_n+layers
 
-        return {"ui":{"canvas": [canvas_w,canvas_h,canvas_base64]},"result": (layer_n,)}
-
+        return (layer_n,)
 
 class ShowLayer:
     @classmethod
diff --git a/web/javascript/layer_mixlab.js b/web/javascript/layer_mixlab.js
index 694bcc4..bdf0f16 100644
--- a/web/javascript/layer_mixlab.js
+++ b/web/javascript/layer_mixlab.js
@@ -156,7 +156,7 @@ const parseSvg = async svgContent => {
   return { data, image: base64, svgElement }
 }
 
-async function setArea (cw, ch, fn) {
+async function setArea (cw, ch, base64, data, fn) {
   let displayHeight = Math.round(window.screen.availHeight * 0.6)
   let div = document.createElement('div')
   div.innerHTML = `
@@ -176,9 +176,9 @@ async function setArea (cw, ch, fn) {
   // document.body.querySelector('#ml_overlay')
   document.body.appendChild(div)
 
-  let canvas = document.createElement('canvas')
-  canvas.width = cw
-  canvas.height = ch
+  // let canvas = document.createElement('canvas')
+  // canvas.width = cw
+  // canvas.height = ch
 
   let img = div.querySelector('#ml_video')
   let overlay = div.querySelector('#ml_overlay')
@@ -186,7 +186,8 @@ async function setArea (cw, ch, fn) {
   let startX, startY, endX, endY
   let start = false
   // Set video source
-  img.src = canvas.toDataURL()
+  img.src = base64
+  // canvas.toDataURL();
 
   // init area
   // const data = getSetAreaData()
@@ -195,6 +196,17 @@ async function setArea (cw, ch, fn) {
     width = (cw * displayHeight) / ch,
     height = displayHeight
 
+  let imgWidth = cw
+  let imgHeight = ch
+
+  if (data && data.width > 0 && data.height > 0  ) {
+    // 相同尺寸窗口，恢复选区
+    x = (width * data.x) / imgWidth
+    y = (height * data.y) / imgHeight
+    width = (width * data.width) / imgWidth
+    height = (height * data.height) / imgHeight
+  }
+
   selection.style.left = x + 'px'
   selection.style.top = y + 'px'
   selection.style.width = width + 'px'
@@ -475,11 +487,14 @@ app.registerExtension({
   name: 'Mixlab.layer.NewLayer',
   async beforeRegisterNodeDef (nodeType, nodeData, app) {
     if (nodeData.name === 'NewLayer') {
-      function populate (canvas) {
-        console.log(canvas, this.widgets)
-        let b=this.widgets.filter(w=>w.type==='button')[0]
-        const [w, h] = canvas
-        if (w > 0 && h > 0&&!b) {
+      const orig_nodeCreated = nodeType.prototype.onNodeCreated
+      nodeType.prototype.onNodeCreated = async function () {
+        orig_nodeCreated?.apply(this, arguments)
+
+        let b = this.widgets.filter(w => w.type === 'button')[0]
+        // const [w, h, base64] = canvas
+
+        if (!b) {
           const updateValue = (x1, y1, w1, h1) => {
             if (this.widgets) {
               for (const widget of this.widgets) {
@@ -500,30 +515,41 @@ app.registerExtension({
           }
 
           this.addWidget('button', 'Set Area', '', () => {
-            // console.log('click')
-            setArea(w, h, updateValue)
+            let data = {}
+            for (const widget of this.widgets) {
+              if (widget.name === 'x') {
+                data.x = widget.value
+              }
+              if (widget.name === 'y') {
+                data.y = widget.value
+              }
+              if (widget.name === 'width') {
+                data.width = widget.value
+              }
+              if (widget.name === 'height') {
+                data.height = widget.value
+              }
+            }
+
+            let linkId = this.inputs[3].link
+            let nodeId = app.graph.links[linkId].origin_id
+            // console.log(linkId,this.inputs)
+            let im = app.graph.getNodeById(nodeId).imgs[0]
+            let src = im.src
+            setArea(im.naturalWidth, im.naturalHeight, src, data, updateValue)
           })
         }
-
-        // console.log('ShowTextForGPT',this.widgets.length)
-        requestAnimationFrame(() => {
-          const sz = this.computeSize()
-          if (sz[0] < this.size[0]) {
-            sz[0] = this.size[0]
-          }
-          if (sz[1] < this.size[1]) {
-            sz[1] = this.size[1]
-          }
-          this.onResize?.(sz)
-          app.graph.setDirtyCanvas(true, false)
-        })
       }
 
-      // When the node is executed we will be sent the input text, display this in the widget
-      const onExecuted = nodeType.prototype.onExecuted
-      nodeType.prototype.onExecuted = function (message) {
-        onExecuted?.apply(this, arguments)
-        populate.call(this, message.canvas)
+      const onRemoved = this.onRemoved
+      this.onRemoved = () => {
+        // let b = this.widgets.filter(w => w.type === 'button')[0];
+
+        return onRemoved?.()
+      }
+
+      if (this.onResize) {
+        this.onResize(this.size)
       }
 
       this.serialize_widgets = true //需要保存参数