@glodon-aiot/dataset-annotation 3.16.0-alpha.37 → 3.16.0-alpha.39

package/dist/es/components/DetectionAnnotation/components/LabelMaker/index.mjs

@@ -2,15 +2,15 @@ var $ = Math.pow;
 import { jsx as f, jsxs as V } from "react/jsx-runtime";
 import { useRef as ee, useState as D, useEffect as h } from "react";
 import { fabric as X } from "fabric";
-import { useKeyPress as T } from "ahooks";
+import { useKeyPress as R } from "ahooks";
 import "./style.less.mjs";
 import le from "react-dom";
-import { Spin as fe, Space as de, Tooltip as L, Button as S } from "antd";
+import { Spin as fe, Space as de, Tooltip as A, Button as L } from "antd";
 import te from "./RectLabel.mjs";
 import { drawable as ue } from "./Drawable.mjs";
 import { ZoomInOutlined as me, ZoomOutOutlined as ge, DragOutlined as he, ClearOutlined as pe } from "@ant-design/icons";
 import be from "classnames";
-import { approxeq as A } from "../../../../utils/index.mjs";
+import { approxeq as S } from "../../../../utils/index.mjs";
 const oe = (a) => {
   a.off("mouse:down"), a.off("mouse:up");
 }, O = (a) => {
@@ -28,9 +28,9 @@ const oe = (a) => {
     id: a.config.id,
     label: a.config.name,
     color: a.config.color,
-    box: (d == null ? void 0 : d.map((y) => [A(y.x - m.x, 0) ? 0 : A(y.x - m.x, p.width) ? p.width : y.x - m.x, A(y.y - m.y, 0) ? 0 : A(y.y - m.y, p.height) ? p.height : y.y - m.y])) || []
+    box: (d == null ? void 0 : d.map((y) => [S(y.x - m.x, 0) ? 0 : S(y.x - m.x, p.width) ? p.width : y.x - m.x, S(y.y - m.y, 0) ? 0 : S(y.y - m.y, p.height) ? p.height : y.y - m.y])) || []
   };
-}, Le = (a) => {
+}, Ae = (a) => {
   const d = ee(null), {
     labels: m,
     imageUrl: p,
@@ -38,12 +38,12 @@ const oe = (a) => {
     onImageLoad: E,
     onChange: l,
     isPreLabel: y,
-    disabled: C,
+    disabled: k,
     renderLabelItem: N
-  } = a, [z, B] = D(!0), [e, Z] = D(), [ye, ne] = D({}), [k, K] = D(), [W, re] = D({
+  } = a, [z, B] = D(!0), [e, Z] = D(), [ye, ne] = D({}), [j, K] = D(), [W, re] = D({
     left: 0,
     top: 0
-  }), [ie, U] = D([1, 1]), [P, H] = D(), [x, v] = D([]), j = ee(/* @__PURE__ */ new Set()), [b, R] = D(C || !1), Y = (o, t) => {
+  }), [ie, U] = D([1, 1]), [M, H] = D(), [x, v] = D([]), w = ee(/* @__PURE__ */ new Set()), [b, P] = D(k || !1), Y = (o, t) => {
     if (!e)
       return;
     t || (t = {
@@ -74,8 +74,8 @@ const oe = (a) => {
         return i && i === t;
       })), u >= 0 && r[u]) {
         const s = r[u];
-        j.current.add(s.config.id), s.destroy(), r.splice(u, 1), l && l(r.map(O)), setTimeout(() => {
-          j.current.delete(s.config.id);
+        w.current.add(s.config.id), s.destroy(), r.splice(u, 1), l && l(r.map(O)), setTimeout(() => {
+          w.current.delete(s.config.id);
         }, 100);
       } else if (t && e) {
         const s = r.findIndex((i) => {
@@ -84,8 +84,8 @@ const oe = (a) => {
         });
         if (s >= 0 && r[s]) {
           const i = r[s];
-          j.current.add(i.config.id), i.destroy(), r.splice(s, 1), l && l(r.map(O)), setTimeout(() => {
-            j.current.delete(i.config.id);
+          w.current.add(i.config.id), i.destroy(), r.splice(s, 1), l && l(r.map(O)), setTimeout(() => {
+            w.current.delete(i.config.id);
           }, 100);
         } else
           e.remove(t), e.discardActiveObject(), e.renderAll();
@@ -93,21 +93,21 @@ const oe = (a) => {
       return [...r];
     });
   };
-  T("Delete", q), T("Backspace", q), T("ctrl", () => {
-    R(!0);
+  R("Delete", q), R("Backspace", q), R("ctrl", () => {
+    P(!0);
   }, {
     events: ["keydown"]
-  }), T("ctrl", () => {
-    C || R(!1);
+  }), R("ctrl", () => {
+    k || P(!1);
   }, {
     events: ["keyup"]
   }), h(() => {
-    R(C || !1);
-  }, [C]);
+    P(k || !1);
+  }, [k]);
   const F = (o) => {
     e && oe(e), v((t) => {
       const n = t.findIndex((r) => r === o);
-      return n > -1 && (t.splice(n, 1), l && l(t.map(O)), M()), [...t];
+      return n > -1 && (t.splice(n, 1), l && l(t.map(O)), T()), [...t];
     });
   }, G = (o) => {
     v((t) => {
@@ -139,7 +139,7 @@ const oe = (a) => {
       });
     };
   }, [e, p]);
-  const M = () => {
+  const T = () => {
     e && ue(e, (o) => {
       const t = new te({
         id: (c == null ? void 0 : c.id) || o.id,
@@ -154,7 +154,7 @@ const oe = (a) => {
       t.active = !0, v((n) => {
         const r = [...n, t];
         return l && l(r.map(O)), setTimeout(() => {
-          e && !b && M();
+          e && !b && T();
         }, 0), r;
       });
     }, y);
@@ -188,8 +188,8 @@ const oe = (a) => {
         return;
       const I = e.getPointer(t.e);
       if (I.x < i.tl.x || I.x > i.br.x || I.y < i.tl.y || I.y > i.br.y) {
-        const w = n == null ? void 0 : n._onMouseUp;
-        typeof w == "function" && w(t.e);
+        const C = n == null ? void 0 : n._onMouseUp;
+        typeof C == "function" && C(t.e);
       }
     };
     return e.on("mouse:move", o), () => {
@@ -220,7 +220,7 @@ const oe = (a) => {
       }), o))) : (e.defaultCursor = "default", e.off("mouse:down:before"), e.off("mouse:up"), e.setCursor("default"), v((o) => (o.forEach((t) => {
         var n;
         return (n = t.object) == null ? void 0 : n.set("evented", !0);
-      }), o)), M()), () => {
+      }), o)), T()), () => {
         ae();
       };
   }, [e, b]), h(() => {
@@ -233,7 +233,7 @@ const oe = (a) => {
         n.renderLabel();
       });
     }), !b)
-      return M(), () => {
+      return T(), () => {
         e && (oe(e), e.off("mouse:wheel"));
       };
   }, [e, c, x, b]);
@@ -250,7 +250,7 @@ const oe = (a) => {
       crossOrigin: "anonymous"
     });
   }, [p]), h(() => {
-    if (!k && e) {
+    if (!j && e) {
       try {
         e.clear();
       } catch (_) {
@@ -259,23 +259,23 @@ const oe = (a) => {
     }
     if (!d.current || !d.current.parentElement)
       return;
-    const o = k == null ? void 0 : k.getElement();
+    const o = j == null ? void 0 : j.getElement();
     if (!(o != null && o.width) || !(o != null && o.height))
       return;
     const t = o.width / o.height, n = d.current.parentElement.offsetWidth, r = d.current.parentElement.offsetHeight, u = n / o.width, s = r / o.height;
     let i = [n, r];
     u > s ? i = [r * t, r] : i = [n, n / t];
-    const I = i[0] / o.width, J = i[1] / o.height, w = Math.min(I, J);
-    U([w, w]);
+    const I = i[0] / o.width, J = i[1] / o.height, C = Math.min(I, J);
+    U([C, C]);
     const g = new X.Canvas(d.current, {
       containerClass: "marking-canvas",
       width: n,
       height: r
     });
-    if (g.uniScaleKey = "", g.uniformScaling = !1, g == null || g.setZoom(w), g == null || g.renderAll(), g && k) {
-      const _ = (n - i[0]) / w / 2, Q = (r - i[1]) / w / 2;
+    if (g.uniScaleKey = "", g.uniformScaling = !1, g == null || g.setZoom(C), g == null || g.renderAll(), g && j) {
+      const _ = (n - i[0]) / C / 2, Q = (r - i[1]) / C / 2;
       g.setBackgroundImage(
-        k,
+        j,
         () => {
           g.renderAll(), re({
             left: _,
@@ -291,12 +291,12 @@ const oe = (a) => {
         }
       );
     }
-  }, [k]), h(() => {
+  }, [j]), h(() => {
     if (x.length > m.length) {
-      if (j.current.size > 0)
+      if (w.current.size > 0)
         return;
       x.forEach((t) => {
-        const n = j.current.has(t.config.id), r = t._isDestroying;
+        const n = w.current.has(t.config.id), r = t._isDestroying;
         !m.find((s) => s.id === t.config.id) && !r && !n && t.destroy();
       });
     }
@@ -311,12 +311,12 @@ const oe = (a) => {
     }));
   }, [c, e, b]), h(() => {
     var o;
-    if (z && d.current && !P) {
+    if (z && d.current && !M) {
       const t = document.createElement("div");
       t.setAttribute("style", "width: 100%; height: 100%; display: flex; align-items: center; justify-content: center; position: absolute;top: 0;"), (o = d.current.parentElement) == null || o.appendChild(t), le.render(/* @__PURE__ */ f(fe, {}), t), H(t);
     } else
-      !z && P && (P.remove(), H(null));
-  }, [z, d.current, P]), /* @__PURE__ */ V("div", {
+      !z && M && (M.remove(), H(null));
+  }, [z, d.current, M]), /* @__PURE__ */ V("div", {
     style: {
       width: "100%",
       height: "100%",
@@ -329,46 +329,54 @@ const oe = (a) => {
         children: /* @__PURE__ */ V(de, {
           direction: "vertical",
           size: 16,
-          children: [/* @__PURE__ */ f(L, {
+          children: [/* @__PURE__ */ f(A, {
             title: "放大-鼠标滚轮前滑",
             placement: "right",
-            children: /* @__PURE__ */ f(S, {
+            children: /* @__PURE__ */ f(L, {
               icon: /* @__PURE__ */ f(me, {}),
               onClick: () => {
                 Y(-100);
               }
             })
-          }), /* @__PURE__ */ f(L, {
+          }), /* @__PURE__ */ f(A, {
             title: "缩小-鼠标滚轮后滑",
             placement: "right",
-            children: /* @__PURE__ */ f(S, {
+            children: /* @__PURE__ */ f(L, {
               icon: /* @__PURE__ */ f(ge, {}),
               onClick: () => {
                 Y(100);
               }
             })
-          }), !C && /* @__PURE__ */ f(L, {
+          }), !k && /* @__PURE__ */ f(A, {
             title: "拖动-按住ctrl",
             placement: "right",
-            children: /* @__PURE__ */ f(S, {
-              disabled: C,
+            children: /* @__PURE__ */ f(L, {
+              disabled: k,
               icon: /* @__PURE__ */ f(he, {}),
               className: b ? "active" : "",
               onClick: () => {
-                R(!b);
+                P(!b);
               }
             })
-          }), !C && /* @__PURE__ */ f(L, {
+          }), !k && /* @__PURE__ */ f(A, {
             title: "一键清空全部标注",
             placement: "right",
-            children: /* @__PURE__ */ f(S, {
+            children: /* @__PURE__ */ f(L, {
               icon: /* @__PURE__ */ f(pe, {}),
               disabled: !(x != null && x.length),
               onClick: () => {
-                v((o) => (o.forEach((t) => {
-                  t.destroy();
-                }), [])), l && l([]), setTimeout(() => {
-                  e && !b && M();
+                if (!e)
+                  return;
+                v((t) => (t.forEach((n) => {
+                  w.current.add(n.config.id);
+                }), t.forEach((n) => {
+                  n.destroy();
+                }), setTimeout(() => {
+                  w.current.clear();
+                }, 100), [])), e.discardActiveObject(), e.getObjects().forEach((t) => {
+                  t !== e.backgroundImage && e.remove(t);
+                }), e.renderAll(), l && l([]), setTimeout(() => {
+                  e && !b && T();
                 }, 0);
               }
             })
@@ -391,5 +399,5 @@ const oe = (a) => {
   });
 };
 export {
-  Le as default
+  Ae as default
 };

package/dist/es/components/OcrAnnotation/OCRBoxDetail/index.mjs

@@ -0,0 +1,239 @@
+var de = Object.defineProperty, ae = Object.defineProperties;
+var fe = Object.getOwnPropertyDescriptors;
+var ne = Object.getOwnPropertySymbols;
+var ue = Object.prototype.hasOwnProperty, ge = Object.prototype.propertyIsEnumerable;
+var re = (f, c, n) => c in f ? de(f, c, { enumerable: !0, configurable: !0, writable: !0, value: n }) : f[c] = n, w = (f, c) => {
+  for (var n in c || (c = {}))
+    ue.call(c, n) && re(f, n, c[n]);
+  if (ne)
+    for (var n of ne(c))
+      ge.call(c, n) && re(f, n, c[n]);
+  return f;
+}, K = (f, c) => ae(f, fe(c));
+import { jsxs as T, Fragment as he, jsx as C } from "react/jsx-runtime";
+import { Tabs as oe, Button as O, message as y } from "antd";
+import { useContext as pe, useRef as ie, useState as x, useEffect as W } from "react";
+import "./style.less.mjs";
+import xe from "../../DataEmpty/index.mjs";
+import Ie from "../../../DatasetContext/index.mjs";
+import me from "../MarkOCRBoxModal/ImageDetail/index.mjs";
+const {
+  TabPane: q
+} = oe, De = (f) => {
+  var te;
+  const {
+    visible: c,
+    selectSampleId: n,
+    datasetId: m,
+    versionId: S,
+    ocrMarkTmpl: Se,
+    onClose: be,
+    onSave: E,
+    categories: se,
+    externalSamples: I,
+    externalPagination: L,
+    externalCurrentIndex: F,
+    externalFilterMode: A,
+    onFilterModeChange: M,
+    onPaginationChange: $
+  } = f, {
+    services: {
+      cvforceDatahub: h
+    }
+  } = pe(Ie), z = ie(null), [r, a] = x(), [Ce, v] = x(0), [p, D] = x(() => n && (I != null && I.length) ? I : []), [j, b] = x(() => n && F !== void 0 ? F : -1), [i, P] = x(() => n && L ? w({}, L) : !n && L ? {
+    current: 1,
+    pageSize: L.pageSize
+  } : {
+    current: 1,
+    pageSize: 50
+  }), [G, J] = x(0), [Q, U] = x(0), [u, le] = x(() => n && A ? A : n ? "" : "nocode"), X = ie(!1), [B, R] = x(!1), [H, Y] = x(!1);
+  W(() => {
+    if (!m || !S)
+      return;
+    const e = i.current || 1, l = i.pageSize || 50;
+    if (!X.current && n && (I != null && I.length) && L && F !== void 0) {
+      const o = I.find((t) => t.id === n);
+      if (o) {
+        X.current = !0, a(o), v(I.length), k();
+        return;
+      }
+    }
+    h == null || h.getDatasetsIdVersionsVidSamples(m, S, {
+      labels: u ? [u] : void 0,
+      pageNo: e,
+      pageSize: l
+    }).then((o) => {
+      const t = o.recordList || [];
+      D(t);
+      const g = K(w({}, i), {
+        current: e,
+        pageSize: l,
+        total: o.totalResults
+      });
+      if (P(g), $ == null || $(g), i.type === "next" && t.length) {
+        const d = n ? t.find((s) => s.id === n) : null;
+        if (d) {
+          const s = t.findIndex((V) => V.id === n);
+          b(s), a(d), v(t.length);
+          return;
+        } else {
+          b(0);
+          const s = o.recordList[0];
+          a(s), v(t.length);
+          return;
+        }
+      }
+      if (i.type === "prev" && t.length) {
+        const d = n ? t.find((s) => s.id === n) : null;
+        if (d) {
+          const s = t.findIndex((V) => V.id === n);
+          b(s), a(d), v(t.length);
+          return;
+        } else {
+          b(i.pageSize - 1);
+          const s = o.recordList[i.pageSize - 1];
+          a(s), v(t.length);
+          return;
+        }
+      }
+      t.find((d) => d.id === n) ? (a(n ? t.find((d) => d.id === n) : t[0]), b(0), v(t.length)) : n ? h.getDatasetsIdVersionsVidSsamplesSid(m, S, n).then((d) => {
+        a(d);
+      }) : (a(t[0]), b(0), v(t.length));
+    }), k();
+  }, [i.current, u, m, S]);
+  const k = () => {
+    !m || !S || h == null || h.getDatasetsIdVersionsVidLabelcount(m, S).then((e) => {
+      const l = e.find((t) => t.id === "allcodeid"), o = e.find((t) => t.id === "nocodeid");
+      J(l ? l.count : 0), U(o ? o.count : 0);
+    });
+  }, Z = () => {
+    p && D((e) => {
+      var t, g, N, d;
+      let l;
+      const o = !r || (t = r.labels) != null && t.length && u === "hascode" || !((g = r.labels) != null && g.length) && u === "nocode" || u === "" ? j + 1 : j;
+      if (r && ((N = r.labels) != null && N.length && u === "nocode" || !((d = r.labels) != null && d.length) && u === "hascode")) {
+        const s = e.findIndex((V) => V.id === r.id);
+        s > -1 && (e.splice(s, 1), l = [...e], i.total && (i.total = i.total - 1));
+      }
+      if (o >= e.length) {
+        const s = (i.current - 1) * i.pageSize + o;
+        i.total && s >= i.total ? y.error("已经是最后一个了") : P({
+          current: i.current + 1,
+          pageSize: i.pageSize,
+          type: "next"
+        });
+      } else
+        setTimeout(() => {
+          b(o);
+          const s = e[o];
+          a(s);
+        }, 200);
+      return l || e;
+    });
+  }, ce = () => {
+    var o, t;
+    if (!p)
+      return;
+    let e;
+    if (r && p && ((o = r.labels) != null && o.length && u === "nocode" || !((t = r.labels) != null && t.length) && u === "hascode")) {
+      const g = p.findIndex((N) => N.id === r.id);
+      g > -1 && (p.splice(g, 1), e = [...p], i.total && (i.total = i.total - 1));
+    }
+    const l = j - 1;
+    if (l < 0)
+      (i.current - 1) * i.pageSize + l < 0 ? y.error("已经是第一个了") : P({
+        current: i.current - 1,
+        pageSize: i.pageSize,
+        type: "prev"
+      });
+    else {
+      b(l);
+      const g = p[l];
+      a(g);
+    }
+    e && D(e);
+  };
+  W(() => {
+    c && m && S && r && (h == null || h.getDatasetsIdVersionsVidSsamplesSid(m, S, r.id).then((e) => {
+      a(e);
+    }));
+  }, [r == null ? void 0 : r.id]), W(() => {
+    z.current && (z.current.width = z.current.offsetWidth, z.current.height = z.current.offsetHeight);
+  }, [z.current]);
+  const _ = () => {
+    r && (R(!0), E(r).then((e) => {
+      e ? (ce(), k()) : y.error("保存失败");
+    }).finally(() => {
+      R(!1);
+    }));
+  }, ee = () => {
+    r && (R(!0), E(r).then((e) => {
+      e ? (Z(), k()) : y.error("保存失败");
+    }).finally(() => {
+      R(!1);
+    }));
+  };
+  return /* @__PURE__ */ T(he, {
+    children: [/* @__PURE__ */ T("div", {
+      className: "ocr-box-detail-records-content",
+      children: [/* @__PURE__ */ T(oe, {
+        className: "ocr-box-detail-records-content-tabs",
+        activeKey: u,
+        onChange: (e) => {
+          P(K(w({}, i), {
+            current: 1
+          })), le(e), M == null || M(e);
+        },
+        children: [/* @__PURE__ */ C(q, {
+          tab: `全部（${G}）`
+        }, ""), /* @__PURE__ */ C(q, {
+          tab: `未标注（${Q}）`
+        }, "nocode"), /* @__PURE__ */ C(q, {
+          tab: `已标注（${G - Q}）`
+        }, "hascode")]
+      }), p != null && p.length ? /* @__PURE__ */ C(me, {
+        currentSample: r,
+        categories: se,
+        onChange: (e) => {
+          a(e), D((l) => {
+            const o = l.findIndex((t) => t.id === e.id);
+            return l[o] = e, [...l];
+          });
+        },
+        onNextClick: ee,
+        onPrevClick: _
+      }) : /* @__PURE__ */ C(xe, {})]
+    }), /* @__PURE__ */ T("div", {
+      className: "ocr-box-detail-footer",
+      children: [/* @__PURE__ */ T("span", {
+        className: "ocr-modal-label-count",
+        children: ["当前图片标注框数量：", ((te = r == null ? void 0 : r.labels) == null ? void 0 : te.length) || 0]
+      }), /* @__PURE__ */ C(O, {
+        disabled: !r || B,
+        type: "primary",
+        loading: H,
+        onClick: () => {
+          r && (Y(!0), E(r).then((e) => {
+            e ? (y.success("保存成功"), Z(), k()) : y.error("保存失败");
+          }).finally(() => {
+            Y(!1);
+          }));
+        },
+        children: "保存标注"
+      }, "save"), /* @__PURE__ */ C(O, {
+        type: "default",
+        onClick: _,
+        disabled: B || H,
+        children: "上一张"
+      }, "prev"), /* @__PURE__ */ C(O, {
+        type: "default",
+        onClick: ee,
+        disabled: B || H,
+        children: "下一张"
+      }, "next")]
+    })]
+  });
+};
+export {
+  De as default
+};

package/dist/es/components/OcrAnnotation/OCRBoxDetail/style.less.mjs

@@ -0,0 +1,4 @@
+const t = "";
+export {
+  t as default
+};