@glodon-aiot/dataset-annotation 3.13.0-alpha.29 → 3.14.0-alpha.30

package/dist/es/components/VLMAnnotation/VLMLabelMaker/hooks/useOCRRecognition.mjs

@@ -1,68 +1,101 @@
-var N = (i, l, s) => new Promise((f, o) => {
-  var c = (t) => {
+var F = (f, m, i) => new Promise((s, n) => {
+  var S = (r) => {
     try {
-      d(s.next(t));
-    } catch (n) {
-      o(n);
+      c(i.next(r));
+    } catch (u) {
+      n(u);
     }
-  }, m = (t) => {
+  }, y = (r) => {
     try {
-      d(s.throw(t));
-    } catch (n) {
-      o(n);
+      c(i.throw(r));
+    } catch (u) {
+      n(u);
     }
-  }, d = (t) => t.done ? f(t.value) : Promise.resolve(t.value).then(c, m);
-  d((s = s.apply(i, l)).next());
+  }, c = (r) => r.done ? s(r.value) : Promise.resolve(r.value).then(S, y);
+  c((i = i.apply(f, m)).next());
 });
-import { useRef as k, useCallback as R } from "react";
-const G = ({
-  ocrRecognize: i,
-  onChange: l,
-  updateTextLayers: s,
-  canvas: f,
-  updateOCRLoadingIcon: o,
-  rectMapRef: c,
-  textLayersMapRef: m,
-  textLayoutContentMapRef: d
+import { useRef as k, useCallback as N } from "react";
+const H = ({
+  ocrRecognize: f,
+  onChange: m,
+  updateTextLayers: i,
+  canvas: s,
+  rectMapRef: n,
+  textLayersMapRef: S,
+  textLayoutContentMapRef: y
 }) => {
-  const t = k(/* @__PURE__ */ new Set()), n = k(/* @__PURE__ */ new Map());
-  return {
-    performOCRIfNeeded: R((e, a) => N(void 0, null, function* () {
-      var x;
-      if (a !== "create" && a !== "update" || !i || !e.image)
+  const c = k(/* @__PURE__ */ new Set()), r = k(/* @__PURE__ */ new Set()), u = k(/* @__PURE__ */ new Map()), v = N((e, o) => F(void 0, null, function* () {
+    var t;
+    if (o !== "create" && o !== "update" || !f || !e.image)
+      return;
+    const l = (t = Array.from(n.current.entries()).find(([, d]) => d.id === e.id)) == null ? void 0 : t[0];
+    if (!l)
+      return;
+    const A = S.current.get(l);
+    A && s && (A.forEach((d) => {
+      d.off("mousedown"), d.off("mouseover"), d.off("mouseout"), y.current.delete(d), s.remove(d);
+    }), S.current.delete(l), s.renderAll());
+    const q = n.current.get(l);
+    q && q.textLayouts && delete q.textLayouts;
+    const a = (u.current.get(e.id) || 0) + 1;
+    u.current.set(e.id, a), r.current.delete(e.id), c.current.add(e.id);
+    try {
+      const d = yield f(e.image);
+      if (u.current.get(e.id) !== a)
+        return;
+      c.current.delete(e.id), r.current.delete(e.id);
+      const E = n.current.get(l);
+      if (E && u.current.get(e.id) === a && (E.textLayouts = d, i(l, E), m)) {
+        const B = Array.from(n.current.values());
+        m(B);
+      }
+    } catch (d) {
+      if (u.current.get(e.id) !== a)
         return;
-      const u = (x = Array.from(c.current.entries()).find(([, r]) => r.id === e.id)) == null ? void 0 : x[0];
-      if (!u)
+      c.current.delete(e.id), r.current.add(e.id), console.error("Failed to perform OCR recognition:", d);
+    }
+  }), [f, m, i, s, n, S, y]), x = N((e) => F(void 0, null, function* () {
+    var a;
+    if (!f || !e.image)
+      return;
+    const o = (a = Array.from(n.current.entries()).find(([, t]) => t.id === e.id)) == null ? void 0 : a[0];
+    if (!o)
+      return;
+    const l = S.current.get(o);
+    l && s && (l.forEach((t) => {
+      t.off("mousedown"), t.off("mouseover"), t.off("mouseout"), y.current.delete(t), s.remove(t);
+    }), S.current.delete(o), s.renderAll());
+    const A = n.current.get(o);
+    A && A.textLayouts && delete A.textLayouts;
+    const q = (u.current.get(e.id) || 0) + 1;
+    u.current.set(e.id, q), r.current.delete(e.id), c.current.add(e.id);
+    try {
+      const t = yield f(e.image);
+      if (u.current.get(e.id) !== q)
         return;
-      const w = m.current.get(u);
-      w && f && (w.forEach((r) => {
-        r.off("mousedown"), r.off("mouseover"), r.off("mouseout"), d.current.delete(r), f.remove(r);
-      }), m.current.delete(u), f.renderAll());
-      const S = c.current.get(u);
-      S && S.textLayouts && delete S.textLayouts;
-      const q = (n.current.get(e.id) || 0) + 1;
-      n.current.set(e.id, q), t.current.add(e.id);
-      const A = c.current.get(u);
-      A && o(u, A, !0);
-      try {
-        const r = yield i(e.image);
-        if (n.current.get(e.id) !== q)
-          return;
-        t.current.delete(e.id), o(u, e, !1);
-        const y = c.current.get(u);
-        if (y && n.current.get(e.id) === q && (y.textLayouts = r, s(u, y), l)) {
-          const F = Array.from(c.current.values());
-          l(F);
-        }
-      } catch (r) {
-        if (n.current.get(e.id) !== q)
-          return;
-        t.current.delete(e.id), o(u, e, !1), console.error("Failed to perform OCR recognition:", r);
+      if (!t || !Array.isArray(t)) {
+        c.current.delete(e.id), r.current.add(e.id), console.error("OCR returned invalid result:", t);
+        return;
+      }
+      c.current.delete(e.id), r.current.delete(e.id);
+      const w = n.current.get(o);
+      if (w && u.current.get(e.id) === q && (w.textLayouts = t, i(o, w), m)) {
+        const I = Array.from(n.current.values());
+        m(I);
       }
-    }), [i, l, s, f, o, c, m, d]),
-    ocrRecognizingRef: t
+    } catch (t) {
+      if (u.current.get(e.id) !== q)
+        return;
+      c.current.delete(e.id), r.current.add(e.id), console.error("Failed to perform OCR recognition:", t);
+    }
+  }), [f, m, i, s, n, S, y]);
+  return {
+    performOCRIfNeeded: v,
+    performOCR: x,
+    ocrRecognizingRef: c,
+    ocrErrorRef: r
   };
 };
 export {
-  G as useOCRRecognition
+  H as useOCRRecognition
 };

package/dist/es/components/VLMAnnotation/VLMLabelMaker/hooks/useRegionBoundaryCheck.mjs

@@ -0,0 +1,103 @@
+import { useCallback as q, useEffect as D } from "react";
+const G = ({
+  canvas: n,
+  image: y,
+  canvasToImageCoords: S,
+  imageToCanvasCoords: b,
+  bgImagePositionRef: k,
+  bgImageSizeRef: A,
+  regionNameTextMapRef: Y
+}) => {
+  const O = q((h) => {
+    if (!n || !y)
+      return;
+    const r = y.getElement();
+    if (!(r != null && r.width) || !(r != null && r.height))
+      return;
+    const i = r.width, u = r.height, d = (w) => {
+      const t = h, g = t.left || 0, p = t.top || 0, a = t.getScaledWidth(), m = t.getScaledHeight(), l = S(g, p);
+      let e = l.x, o = l.y, c = a, s = m;
+      const W = e, H = o;
+      if (e < 0 && (e = 0), o < 0 && (o = 0), e + c > i && (e = i - c), o + s > u && (o = u - s), c <= 0 || s <= 0)
+        return;
+      const L = b(e, o);
+      (e !== W || o !== H) && t.set({
+        left: L.x,
+        top: L.y
+      });
+      const Z = Y.current.get(t);
+      if (Z) {
+        const f = n.getZoom(), x = t.left || 0, X = t.top || 0;
+        Z.set({
+          left: x,
+          top: X,
+          scaleX: 1 / f,
+          scaleY: 1 / f
+        }), n.renderAll();
+      }
+    }, M = (w) => {
+      const t = h, g = t.left || 0, p = t.top || 0, a = t.getScaledWidth(), m = t.getScaledHeight(), l = S(g, p);
+      let e = l.x, o = l.y, c = a, s = m;
+      e < 0 && (c = c + e, e = 0), o < 0 && (s = s + o, o = 0), e + c > i && (c = i - e), o + s > u && (s = u - o), c = Math.max(1, c), s = Math.max(1, s);
+      const W = Math.abs(c - a) > 1e-3, H = Math.abs(s - m) > 1e-3, L = Math.abs(e - l.x) > 1e-3, Z = Math.abs(o - l.y) > 1e-3;
+      if (W || H || L || Z) {
+        const f = b(e, o), x = t.width || 1, X = t.height || 1, B = W ? Math.max(0.01, c / x) : t.scaleX || 1, U = H ? Math.max(0.01, s / X) : t.scaleY || 1;
+        t.set({
+          left: f.x,
+          top: f.y,
+          scaleX: B,
+          scaleY: U
+        });
+        const C = Y.current.get(t);
+        if (C) {
+          const T = n.getZoom(), _ = t.left || 0, j = t.top || 0;
+          C.set({
+            left: _,
+            top: j,
+            scaleX: 1 / T,
+            scaleY: 1 / T
+          }), n.renderAll();
+        }
+      } else {
+        const f = Y.current.get(t);
+        if (f) {
+          const x = n.getZoom(), X = t.left || 0, B = t.top || 0;
+          f.set({
+            left: X,
+            top: B,
+            scaleX: 1 / x,
+            scaleY: 1 / x
+          }), n.renderAll();
+        }
+      }
+    };
+    return h.on("moving", d), h.on("scaling", M), () => {
+      h.off("moving", d), h.off("scaling", M);
+    };
+  }, [n, y, S, b, Y]);
+  return D(() => {
+    if (!n || !y)
+      return;
+    const h = (r) => {
+      const i = n, u = i == null ? void 0 : i._currentTransform;
+      if (!u)
+        return;
+      const d = u.action;
+      if (!d || d !== "move" && !d.startsWith("scale"))
+        return;
+      const M = n.getPointer(r.e), w = M.x, t = M.y, g = k.current.left, p = k.current.top, a = g + A.current.width, m = p + A.current.height;
+      if (w < g || w > a || t < p || t > m) {
+        const e = i == null ? void 0 : i._onMouseUp;
+        typeof e == "function" && e(r.e);
+      }
+    };
+    return n.on("mouse:move", h), () => {
+      n.off("mouse:move", h);
+    };
+  }, [n, y, k, A]), {
+    setupBoundaryCheck: O
+  };
+};
+export {
+  G as useRegionBoundaryCheck
+};

package/dist/es/components/VLMAnnotation/VLMLabelMaker/hooks/useRegionLoader.mjs

@@ -1,61 +1,65 @@
-var X = Object.defineProperty;
-var F = Object.getOwnPropertySymbols;
-var b = Object.prototype.hasOwnProperty, j = Object.prototype.propertyIsEnumerable;
-var K = (e, t, s) => t in e ? X(e, t, { enumerable: !0, configurable: !0, writable: !0, value: s }) : e[t] = s, Z = (e, t) => {
-  for (var s in t || (t = {}))
-    b.call(t, s) && K(e, s, t[s]);
-  if (F)
-    for (var s of F(t))
-      j.call(t, s) && K(e, s, t[s]);
-  return e;
+var B = Object.defineProperty;
+var H = Object.getOwnPropertySymbols;
+var C = Object.prototype.hasOwnProperty, D = Object.prototype.propertyIsEnumerable;
+var X = (o, t, n) => t in o ? B(o, t, { enumerable: !0, configurable: !0, writable: !0, value: n }) : o[t] = n, P = (o, t) => {
+  for (var n in t || (t = {}))
+    C.call(t, n) && X(o, n, t[n]);
+  if (H)
+    for (var n of H(t))
+      D.call(t, n) && X(o, n, t[n]);
+  return o;
 };
-import { useEffect as q } from "react";
-import { fabric as N } from "fabric";
-import { TRANSPARENT_FILL as P, RECT_STROKE_COLOR as B, BASE_STROKE_WIDTH as G, REGION_NAME_PREFIX as H } from "../../constants.mjs";
-const z = ({
-  canvas: e,
+import { useEffect as O } from "react";
+import { fabric as j } from "fabric";
+import { TRANSPARENT_FILL as q, RECT_STROKE_COLOR as J, BASE_STROKE_WIDTH as Q, REGION_NAME_PREFIX as V } from "../../constants.mjs";
+const a = ({
+  canvas: o,
   sample: t,
-  image: s,
-  bgImagePosition: O,
-  canvasInitializedRef: $,
-  bgImagePositionRef: w,
+  image: n,
+  bgImagePosition: Y,
+  canvasInitializedRef: v,
+  bgImagePositionRef: m,
   sampleRef: y,
-  labelsLoadedRef: d,
-  rectMapRef: f,
-  textLayersMapRef: S,
-  textLayoutContentMapRef: A,
-  regionNameTextMapRef: C,
-  ocrLoadingIconMapRef: _,
-  imageToCanvasCoords: x,
-  updateRegionShapeAndImage: E,
-  createDeleteControl: m,
-  updateRegionNameText: W,
-  updateTextLayers: U
+  labelsLoadedRef: E,
+  rectMapRef: u,
+  textLayersMapRef: _,
+  textLayoutContentMapRef: G,
+  regionNameTextMapRef: x,
+  ocrLoadingIconMapRef: F,
+  imageToCanvasCoords: A,
+  updateRegionShapeAndImage: k,
+  createDeleteControl: W,
+  createOCRControl: U,
+  updateRegionNameText: Z,
+  updateTextLayers: $,
+  setupBoundaryCheck: K,
+  cropRegionImage: N,
+  onChange: w
 }) => {
-  q(() => {
-    if (!e || !(t != null && t.labels) || !s || !$.current || w.current.left === 0 && w.current.top === 0)
+  O(() => {
+    if (!o || !(t != null && t.labels) || !n || !v.current || m.current.left === 0 && m.current.top === 0)
       return;
-    y.current !== t ? (y.current = t, d.current = !1, f.current.forEach((o, n) => {
-      const l = S.current.get(n);
-      l && (l.forEach((r) => {
-        r.off("mousedown"), r.off("mouseover"), r.off("mouseout"), A.current.delete(r), e.remove(r);
-      }), S.current.delete(n));
-      const u = C.current.get(n);
-      u && (n.off("moved"), n.off("modified"), e.remove(u), C.current.delete(n));
-      const i = _.current.get(n);
-      i && (e.remove(i), _.current.delete(n)), e.remove(n);
-    }), f.current.clear(), t.labels.forEach((o, n) => {
-      if (o.shape.name !== "rect")
+    y.current !== t ? (y.current = t, E.current = !1, u.current.forEach((i, s) => {
+      const f = _.current.get(s);
+      f && (f.forEach((e) => {
+        e.off("mousedown"), e.off("mouseover"), e.off("mouseout"), G.current.delete(e), o.remove(e);
+      }), _.current.delete(s));
+      const h = x.current.get(s);
+      h && (s.off("moved"), s.off("modified"), o.remove(h), x.current.delete(s));
+      const d = F.current.get(s);
+      d && (o.remove(d), F.current.delete(s)), o.remove(s);
+    }), u.current.clear(), t.labels.forEach((i, s) => {
+      if (i.shape.name !== "rect")
         return;
-      const l = x(o.shape.x, o.shape.y), u = e.getZoom(), i = G / u, r = new N.Rect({
-        left: l.x,
-        top: l.y,
-        width: o.shape.width,
-        height: o.shape.height,
-        fill: P,
+      const f = A(i.shape.x, i.shape.y), h = o.getZoom(), d = Q / h, e = new j.Rect({
+        left: f.x,
+        top: f.y,
+        width: i.shape.width,
+        height: i.shape.height,
+        fill: q,
         // 无填充色
-        stroke: B,
-        strokeWidth: i,
+        stroke: J,
+        strokeWidth: d,
         strokeUniform: !0,
         // 边框宽度不受缩放影响
         selectable: !0,
@@ -66,30 +70,42 @@ const z = ({
         // 锁定旋转
         lockUniScaling: !1,
         // 允许非等比缩放，控制点跟随鼠标
+        lockScalingFlip: !0,
+        // 禁止缩放时翻转
         objectCaching: !1,
         // 禁用缓存，避免缩放时边框变虚
         statefullCache: !1,
         noScaleCache: !1
       });
-      r.on("modified", () => {
-        E(r);
-      }), r.on("moved", () => {
-        E(r);
-      }), e.add(r), r.controls.deleteControl = m();
-      const h = Z({}, o);
-      h.name || (h.name = `${H}${n + 1}`), f.current.set(r, h), W(r, h), h.textLayouts && h.textLayouts.length > 0 && U(r, h);
-    }), d.current = !0, e.renderAll()) : !d.current && t.labels.length > 0 && (t.labels.forEach((o, n) => {
-      if (o.shape.name !== "rect" || Array.from(f.current.values()).find((k) => k.shape.x === o.shape.x && k.shape.y === o.shape.y && k.shape.width === o.shape.width && k.shape.height === o.shape.height))
+      e.on("modified", () => {
+        k(e);
+      }), e.on("moved", () => {
+        k(e);
+      }), o.add(e), K(e), e.controls.deleteControl = W(), e.controls.ocrControl = U();
+      const r = P({}, i);
+      r.name || (r.name = `${V}${s + 1}`), u.current.set(e, r), Z(e, r), r.textLayouts && r.textLayouts.length > 0 && $(e, r), r.image || N(r).then((c) => {
+        if (c) {
+          const l = u.current.get(e);
+          if (l && (l.image = c, w)) {
+            const S = Array.from(u.current.values());
+            w(S);
+          }
+        }
+      }).catch((c) => {
+        console.error("Failed to crop region image:", c);
+      });
+    }), E.current = !0, o.renderAll()) : !E.current && t.labels.length > 0 && (t.labels.forEach((i, s) => {
+      if (i.shape.name !== "rect" || Array.from(u.current.values()).find((l) => l.shape.x === i.shape.x && l.shape.y === i.shape.y && l.shape.width === i.shape.width && l.shape.height === i.shape.height))
         return;
-      const u = x(o.shape.x, o.shape.y), i = e.getZoom(), r = G / i, h = new N.Rect({
-        left: u.x,
-        top: u.y,
-        width: o.shape.width,
-        height: o.shape.height,
-        fill: P,
+      const h = A(i.shape.x, i.shape.y), d = o.getZoom(), e = Q / d, r = new j.Rect({
+        left: h.x,
+        top: h.y,
+        width: i.shape.width,
+        height: i.shape.height,
+        fill: q,
         // 无填充色
-        stroke: B,
-        strokeWidth: r,
+        stroke: J,
+        strokeWidth: e,
         strokeUniform: !0,
         // 边框宽度不受缩放影响
         selectable: !0,
@@ -100,42 +116,58 @@ const z = ({
         // 锁定旋转
         lockUniScaling: !1,
         // 允许非等比缩放，控制点跟随鼠标
+        lockScalingFlip: !0,
+        // 禁止缩放时翻转
         objectCaching: !1,
         // 禁用缓存，避免缩放时边框变虚
         statefullCache: !1,
         noScaleCache: !1
       });
-      h.on("modified", () => {
-        E(h);
-      }), h.on("moved", () => {
-        E(h);
-      }), e.add(h), h.controls.deleteControl = m();
-      const c = Z({}, o);
-      c.name || (c.name = `${H}${n + 1}`), f.current.set(h, c), W(h, c), c.textLayouts && c.textLayouts.length > 0 && U(h, c);
-    }), d.current = !0, e.renderAll());
+      r.on("modified", () => {
+        k(r);
+      }), r.on("moved", () => {
+        k(r);
+      }), o.add(r), K(r), r.controls.deleteControl = W(), r.controls.ocrControl = U();
+      const c = P({}, i);
+      c.name || (c.name = `${V}${s + 1}`), u.current.set(r, c), Z(r, c), c.textLayouts && c.textLayouts.length > 0 && $(r, c), c.image || N(c).then((l) => {
+        if (l) {
+          const S = u.current.get(r);
+          if (S && (S.image = l, w)) {
+            const z = Array.from(u.current.values());
+            w(z);
+          }
+        }
+      }).catch((l) => {
+        console.error("Failed to crop region image:", l);
+      });
+    }), E.current = !0, o.renderAll());
   }, [
-    e,
+    o,
     t,
     t == null ? void 0 : t.labels,
-    s,
-    O,
+    n,
+    Y,
+    N,
+    w,
     // 添加 bgImagePosition 作为依赖，确保在位置设置后触发加载
-    x,
-    E,
-    U,
+    A,
+    k,
+    $,
+    Z,
     W,
+    U,
+    v,
     m,
-    $,
-    w,
     y,
-    d,
-    f,
-    S,
-    A,
-    C,
-    _
+    E,
+    u,
+    _,
+    G,
+    x,
+    F,
+    K
   ]);
 };
 export {
-  z as useRegionLoader
+  a as useRegionLoader
 };