@glodon-aiot/dataset-annotation 3.23.0-alpha.2 → 3.23.0-alpha.4

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

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