@glodon-aiot/dataset-annotation 3.22.4 → 3.23.0-alpha.3

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

@@ -1,198 +1,227 @@
-var _ = Object.defineProperty, $ = Object.defineProperties;
-var I = Object.getOwnPropertyDescriptors;
-var P = Object.getOwnPropertySymbols;
-var L = Object.prototype.hasOwnProperty, ee = Object.prototype.propertyIsEnumerable;
-var B = (i, o, s) => o in i ? _(i, o, { enumerable: !0, configurable: !0, writable: !0, value: s }) : i[o] = s, N = (i, o) => {
-  for (var s in o || (o = {}))
-    L.call(o, s) && B(i, s, o[s]);
-  if (P)
-    for (var s of P(o))
-      ee.call(o, s) && B(i, s, o[s]);
+var L = Object.defineProperty, ee = Object.defineProperties;
+var le = Object.getOwnPropertyDescriptors;
+var F = Object.getOwnPropertySymbols;
+var te = Object.prototype.hasOwnProperty, ie = Object.prototype.propertyIsEnumerable;
+var U = (i, n, a) => n in i ? L(i, n, { enumerable: !0, configurable: !0, writable: !0, value: a }) : i[n] = a, g = (i, n) => {
+  for (var a in n || (n = {}))
+    te.call(n, a) && U(i, a, n[a]);
+  if (F)
+    for (var a of F(n))
+      ie.call(n, a) && U(i, a, n[a]);
   return i;
-}, g = (i, o) => $(i, I(o));
-import { jsxs as h, jsx as a } from "react/jsx-runtime";
-import { Row as U, Button as Z, Input as te, Space as ie, Empty as le, Modal as re } from "antd";
-import { forwardRef as oe, useState as u, useContext as ne, useEffect as y, useImperativeHandle as se } from "react";
+}, A = (i, n) => ee(i, le(n));
+import { jsxs as p, jsx as r, Fragment as re } from "react/jsx-runtime";
+import { Row as Z, Button as $, Input as ne, Space as oe, Empty as se, Modal as ae, Tooltip as ce } from "antd";
+import { forwardRef as de, useState as h, useContext as fe, useEffect as M, useImperativeHandle as me } from "react";
 import q from "../TagItem/index.mjs";
 import "./style.less.mjs";
-import { PlusOutlined as ce } from "@ant-design/icons";
-import F from "../DisabledTooltip/index.mjs";
-import ae from "../../../../DatasetContext/index.mjs";
-import { tagColors as k } from "../../../../utils/index.mjs";
-import de from "../../../Iconfont/index.mjs";
-const fe = oe(({
+import { PlusOutlined as pe } from "@ant-design/icons";
+import G from "../DisabledTooltip/index.mjs";
+import ue from "../../../../DatasetContext/index.mjs";
+import { tagColors as T } from "../../../../utils/index.mjs";
+import { formatCount as he } from "../../../../utils/formatCount.mjs";
+import be from "../../../Iconfont/index.mjs";
+const xe = de(({
   datasetId: i,
-  data: o,
-  selected: s = [],
-  showZero: me = !0,
-  shortcut: M = !1,
-  selectMode: G = "multiple",
-  onItemClick: C,
-  onSelectedChange: T,
-  onListChange: m,
-  isPreLabel: v = !1,
-  settingHide: j = !1
-}, H) => {
-  const [n, A] = u([]), [ue, z] = u(!1), [r, p] = u([]), [E, R] = u(""), [x, S] = u(null), [w, J] = u(!1), {
+  data: n,
+  selected: a = [],
+  showZero: ve = !0,
+  shortcut: C = !1,
+  showStatColumns: D = !1,
+  selectMode: H = "multiple",
+  onItemClick: E,
+  onSelectedChange: j,
+  onListChange: u,
+  isPreLabel: x = !1,
+  settingHide: z = !1
+}, J) => {
+  const [s, S] = h([]), [Ne, R] = h(!1), [o, b] = h([]), [O, P] = h(""), [v, w] = h(null), [y, K] = h(!1), {
     services: {
-      cvforceDatahub: d
+      cvforceDatahub: f
     },
     featuresConfig: {
-      labelManager: f
+      labelManager: m
     } = {
       labelManager: !1
     }
-  } = ne(ae);
-  y(() => {
-    r.length && A(s.map((e) => r.find((t) => t.id === e.id)));
-  }, [s, r]), y(() => {
-    i && D();
-  }, [i]), y(() => {
-    !i && o && p(o.map((e) => g(N({}, e), {
+  } = fe(ue);
+  M(() => {
+    o.length && S(a.map((e) => o.find((l) => l.id === e.id)));
+  }, [a, o]), M(() => {
+    i && k();
+  }, [i]), M(() => {
+    !i && n && b(n.map((e) => A(g({}, e), {
       createdAt: 0,
       updatedAt: 0
     })));
-  }, [i, o]);
-  const D = () => {
-    !i || !d || (z(!0), d.listLabels(i, {
+  }, [i, n]);
+  const k = () => {
+    !i || !f || (R(!0), f.listLabels(i, {
       pageNo: 1,
       pageSize: 1e4
     }).then((e) => {
-      p(e.recordList.sort((t, l) => t.createdAt - l.createdAt > 0 ? -1 : t.createdAt - l.createdAt < 0 ? 1 : 0).map((t, l) => g(N({}, t), {
-        color: k[l % 40]
+      b(e.recordList.sort((l, t) => l.createdAt - t.createdAt > 0 ? -1 : l.createdAt - t.createdAt < 0 ? 1 : 0).map((l, t) => A(g({}, l), {
+        color: T[t % 40]
       })));
-    }).finally(() => z(!1)));
+    }).finally(() => R(!1)));
   };
-  se(H, () => ({
-    fetchLabels: D
+  me(J, () => ({
+    fetchLabels: k
   }));
-  const K = (e) => !i || !d ? Promise.resolve(!1) : d.postLabels(i, {
+  const Q = (e) => !i || !f ? Promise.resolve(!1) : f.postLabels(i, {
     labels: [{
       label: e
     }]
-  }).then((t) => {
-    p([...r, {
+  }).then((l) => {
+    b([...o, {
       label: e,
-      id: t[0],
-      color: k[r.length % 40],
+      id: l[0],
+      color: T[o.length % 40],
       createdAt: Date.now(),
       updatedAt: Date.now()
-    }]), S(null), m && m([...r]);
-  }).then(() => !0).catch(() => !1), Q = (e) => {
-    let t = [];
-    G === "multiple" ? n != null && n.some((l) => l.id === e.id) ? t = n == null ? void 0 : n.filter((l) => l.id !== e.id) : t = [...n, e] : n != null && n.some((l) => l.id === e.id) ? t = [] : t = [e], A(t), T && T(t);
-  }, V = (e) => {
-    R(e);
+    }]), w(null), u && u([...o]);
+  }).then(() => !0).catch(() => !1), V = (e) => {
+    let l = [];
+    H === "multiple" ? s != null && s.some((t) => t.id === e.id) ? l = s == null ? void 0 : s.filter((t) => t.id !== e.id) : l = [...s, e] : s != null && s.some((t) => t.id === e.id) ? l = [] : l = [e], S(l), j && j(l);
   }, W = (e) => {
-    if (!i || !d)
-      return Promise.resolve(!1);
-    const t = r.findIndex((l) => l.id === e.id);
-    return d.patchLabel(i, e.id, e).then(() => (t < 0 || (r[t] = e, p([...r]), m && m([...r])), !0)).catch(() => !1);
+    P(e);
   }, X = (e) => {
-    re.confirm({
+    if (!i || !f)
+      return Promise.resolve(!1);
+    const l = o.findIndex((t) => t.id === e.id);
+    return f.patchLabel(i, e.id, e).then(() => (l < 0 || (o[l] = e, b([...o]), u && u([...o])), !0)).catch(() => !1);
+  }, Y = (e) => {
+    ae.confirm({
       title: "确认要删除当前标签吗？",
       content: "删除后，将会删除当前数据集中所有与当前标签关联的数据，删除后无法恢复，请确认是否删除！",
       onOk: () => {
-        if (!i || !d)
+        if (!i || !f)
           return;
-        const t = r.findIndex((c) => c.id === e), l = n.findIndex((c) => c.id === e);
-        return d.deleteLabel(i, e).then(() => {
-          l > -1 && (n.splice(l, 1), A([...n])), r.splice(t, 1), p([...r]), m && m([...r]), D();
+        const l = o.findIndex((d) => d.id === e), t = s.findIndex((d) => d.id === e);
+        return f.deleteLabel(i, e).then(() => {
+          t > -1 && (s.splice(t, 1), S([...s])), o.splice(l, 1), b([...o]), u && u([...o]), k();
         });
       }
     });
-  }, b = (r == null ? void 0 : r.filter((e) => !E || e.label.includes(E))) || [];
-  return /* @__PURE__ */ h("div", {
-    className: "label-list-wrap",
-    children: [/* @__PURE__ */ h(U, {
+  }, N = (o == null ? void 0 : o.filter((e) => !O || e.label.includes(O))) || [], _ = (e) => {
+    var t, d;
+    const l = he((t = e == null ? void 0 : e.boxCount) != null ? t : 0);
+    return /* @__PURE__ */ p(re, {
+      children: [/* @__PURE__ */ r("span", {
+        className: "label-stat-col",
+        children: l.exact ? /* @__PURE__ */ r(ce, {
+          title: l.exact,
+          children: l.text
+        }) : l.text
+      }), /* @__PURE__ */ r("span", {
+        className: "label-stat-col",
+        children: (d = e == null ? void 0 : e.count) != null ? d : 0
+      })]
+    });
+  };
+  return /* @__PURE__ */ p("div", {
+    className: `label-list-wrap${D ? " label-list-wrap--stats" : ""}`,
+    children: [/* @__PURE__ */ p(Z, {
       justify: "space-between",
       className: "label-list-title",
-      children: [/* @__PURE__ */ a("span", {
+      children: [/* @__PURE__ */ r("span", {
         className: "label-list-title",
         children: "标签列表"
-      }), /* @__PURE__ */ a("span", {
+      }), /* @__PURE__ */ r("span", {
         className: "label-list-icon",
-        children: !j && f && f.visible && /* @__PURE__ */ a(F, {
-          isPreLabel: v,
+        children: !z && m && m.visible && /* @__PURE__ */ r(G, {
+          isPreLabel: x,
           placement: "topRight",
-          children: /* @__PURE__ */ a(Z, {
-            type: w ? "link" : "text",
+          children: /* @__PURE__ */ r($, {
+            type: y ? "link" : "text",
             size: "small",
-            onClick: () => J(!w),
-            icon: /* @__PURE__ */ a(de, {
+            onClick: () => K(!y),
+            icon: /* @__PURE__ */ r(be, {
               type: "icon-shezhi"
             }),
-            disabled: v
+            disabled: x
           })
         })
       })]
-    }), /* @__PURE__ */ h(U, {
+    }), /* @__PURE__ */ p(Z, {
       className: "label-list-opration",
-      children: [!j && f && f.add && /* @__PURE__ */ a(F, {
-        isPreLabel: v,
-        children: /* @__PURE__ */ a(Z, {
+      children: [!z && m && m.add && /* @__PURE__ */ r(G, {
+        isPreLabel: x,
+        children: /* @__PURE__ */ r($, {
           className: "label-list-create",
           type: "default",
-          icon: /* @__PURE__ */ a(ce, {}),
-          disabled: x !== null || v,
+          icon: /* @__PURE__ */ r(pe, {}),
+          disabled: v !== null || x,
           onClick: () => {
-            S({
-              color: k[r.length % 40],
+            w({
+              color: T[o.length % 40],
               label: ""
             });
           }
         })
-      }), /* @__PURE__ */ a("span", {
+      }), /* @__PURE__ */ r("span", {
         className: "label-list-search",
-        children: /* @__PURE__ */ a(te.Search, {
+        children: /* @__PURE__ */ r(ne.Search, {
           placeholder: "请输入标签名称搜索",
-          onSearch: V,
+          onSearch: W,
           onChange: (e) => {
-            R(e.target.value);
+            P(e.target.value);
           }
         })
       })]
-    }), /* @__PURE__ */ h(ie, {
+    }), D && /* @__PURE__ */ p("div", {
+      className: "label-list-header",
+      children: [/* @__PURE__ */ r("span", {
+        className: "label-list-header-name",
+        children: "标签名称"
+      }), /* @__PURE__ */ r("span", {
+        className: "label-stat-col",
+        children: "标注框数"
+      }), /* @__PURE__ */ r("span", {
+        className: "label-stat-col",
+        children: "图片数"
+      })]
+    }), /* @__PURE__ */ p(oe, {
       direction: "vertical",
       className: "label-list",
-      children: [x && /* @__PURE__ */ a(q, {
-        onSave: K,
+      children: [v && /* @__PURE__ */ r(q, {
+        onSave: Q,
         onCancel: () => {
-          S(null);
+          w(null);
         },
-        label: x.label || "",
-        color: x.color || "",
+        label: v.label || "",
+        color: v.color || "",
         editMode: !0,
         isActive: !1,
         onCheckDuplication: (e) => {
-          const t = b.findIndex((l) => l.label === e);
-          return Promise.resolve(t > -1);
+          const l = N.findIndex((t) => t.label === e);
+          return Promise.resolve(l > -1);
         }
-      }), b.length ? b.reverse().map((e, t) => {
-        var l;
-        return /* @__PURE__ */ a(q, {
-          isActive: n.some((c) => (c == null ? void 0 : c.id) === e.id),
-          editable: w,
-          shortcut: M && t < 10 ? t.toString() : void 0,
-          extra: M ? t < 10 ? /* @__PURE__ */ h("span", {
+      }), N.length ? N.reverse().map((e, l) => {
+        var t, d;
+        return /* @__PURE__ */ r(q, {
+          isActive: s.some((c) => (c == null ? void 0 : c.id) === e.id),
+          editable: y,
+          shortcut: C && l < 10 ? l.toString() : void 0,
+          extra: C ? l < 10 ? /* @__PURE__ */ p("span", {
             className: "shortcut",
-            children: ["快捷键", t]
-          }) : null : ((l = o.find((c) => c.label === e.label)) == null ? void 0 : l.count) || 0,
+            children: ["快捷键", l]
+          }) : null : D ? _(n.find((c) => c.label === e.label)) : (d = (t = n.find((c) => c.label === e.label)) == null ? void 0 : t.count) != null ? d : 0,
           label: e.label,
           color: e.color,
           onClick: () => {
-            Q(e), C && C(e);
+            V(e), E && E(e);
           },
-          onSave: f && f.edit ? (c) => W(g(N({}, e), {
+          onSave: m && m.edit ? (c) => X(A(g({}, e), {
             label: c
           })) : void 0,
-          onDelete: f && f.delete ? () => X(e.id) : void 0,
+          onDelete: m && m.delete ? () => Y(e.id) : void 0,
           onCheckDuplication: (c) => {
-            const Y = b.findIndex((O) => O.label === c && e.id !== O.id);
-            return Promise.resolve(Y > -1);
+            const I = N.findIndex((B) => B.label === c && e.id !== B.id);
+            return Promise.resolve(I > -1);
           }
         }, e.id);
-      }) : /* @__PURE__ */ a(le, {
+      }) : /* @__PURE__ */ r(se, {
         className: "label-list-empty",
         image: !1,
         description: "请先添加标签，然后在图中框选标注"
@@ -200,7 +229,7 @@ const fe = oe(({
     })]
   });
 });
-fe.displayName = "Labels";
+xe.displayName = "Labels";
 export {
-  fe as default
+  xe as default
 };

package/dist/es/components/DetectionAnnotation/hooks/useSamplePreload.mjs

@@ -0,0 +1,40 @@
+import { useRef as D, useEffect as a } from "react";
+import { PRELOAD_FORWARD as E, PRELOAD_BACKWARD as P, ImagePreloader as d } from "../utils/imagePreloader.mjs";
+const v = ({
+  samples: f,
+  currentIndex: o,
+  active: i,
+  loadedImageUrl: s
+}) => {
+  var c;
+  const t = D(void 0), u = o >= 0 ? (c = f[o]) == null ? void 0 : c.fileUrl : void 0;
+  a(() => {
+    var p;
+    if (!i) {
+      (p = t.current) == null || p.dispose(), t.current = void 0;
+      return;
+    }
+    if (!u || s !== u)
+      return;
+    const e = [], n = (r) => {
+      var A;
+      if (r < 0 || r >= f.length)
+        return;
+      const R = (A = f[r]) == null ? void 0 : A.fileUrl;
+      R && e.push(R);
+    };
+    for (let r = 1; r <= E; r++)
+      n(o + r);
+    for (let r = 1; r <= P; r++)
+      n(o - r);
+    t.current || (t.current = new d());
+    const l = t.current;
+    l.pruneExcept([u, ...e]), l.setTargets(e);
+  }, [i, s, u, o, f]), a(() => () => {
+    var e;
+    (e = t.current) == null || e.dispose(), t.current = void 0;
+  }, []);
+};
+export {
+  v as default
+};