@glodon-aiot/dataset-annotation 3.16.0-alpha.38 → 3.16.0-alpha.40

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
+};

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

@@ -1,178 +1,211 @@
-import { jsxs as a, jsx as s, Fragment as H } from "react/jsx-runtime";
-import { useRef as N, useState as n, useContext as Q, useEffect as k } from "react";
-import { Modal as u, Button as d, Tabs as V, Tooltip as X, Space as Y, message as R } from "antd";
+import { jsx as t, jsxs as a, Fragment as k } from "react/jsx-runtime";
+import { useRef as u, useState as r, useContext as oe, useEffect as R } from "react";
+import { Modal as I, Button as c, Tabs as z, Tooltip as se, Space as ie, message as w } from "antd";
 import "./style.less.mjs";
-import v from "react-json-view";
-import { useKeyPress as T } from "ahooks";
-import Z from "../../MarkDescriptions/index.mjs";
-import x from "../../Iconfont/index.mjs";
-import _ from "../../../DatasetContext/index.mjs";
-import $ from "../../OCRImageViewer/index.mjs";
-import S from "../../ScanTable/index.mjs";
-const ee = v.default || v, {
-  TabPane: I
-} = V, fe = (j) => {
-  var O, M;
+import T from "react-json-view";
+import { useKeyPress as j } from "ahooks";
+import ne from "../../MarkDescriptions/index.mjs";
+import P from "../../Iconfont/index.mjs";
+import le from "../../../DatasetContext/index.mjs";
+import re from "../../OCRImageViewer/index.mjs";
+import de from "../../ScanTable/index.mjs";
+import ce from "../OCRBoxDetail/index.mjs";
+const me = T.default || T, {
+  TabPane: V
+} = z, ke = (D) => {
+  var x, M, O;
   const {
-    isOcrModalVisible: c,
-    setIsOcrModalVisible: C,
-    record: h,
-    onNextClick: p,
+    mode: m = "preview",
+    visible: d,
+    onClose: C,
+    record: f,
+    onNextClick: g,
     onPrevClick: b,
     datasetId: i,
-    versionId: r,
-    ocrMarkTmpl: l,
-    handleMarkClick: w,
-    updateSampleList: z,
-    isPreLabel: m
-  } = j, D = N(null), J = N({}), [f, K] = n("annotation"), [e, P] = n(), [L, E] = n(), [q, A] = n([]), [B, g] = n(!1), {
+    versionId: l,
+    ocrMarkTmpl: n,
+    handleMarkClick: F,
+    updateSampleList: J,
+    isPreLabel: p,
+    selectSampleId: K,
+    onSave: L,
+    filterMode: B,
+    pagination: E,
+    currentIndex: q,
+    onFilterModeChange: A,
+    onPaginationChange: U,
+    samples: W
+  } = D, G = u(null), H = u({}), [h, Q] = r("annotation"), [e, X] = r(), [Y, Z] = r(), [y, _] = r([]), [$, v] = r(!1), {
     services: {
-      cvforceDatahub: t
+      cvforceDatahub: o
     },
     featuresConfig: {
-      datasetManager: y
+      datasetManager: N
     } = {
       datasetManager: !1
     }
-  } = Q(_);
-  T("rightarrow", () => {
-    p();
-  }), T("leftarrow", () => {
+  } = oe(le);
+  j("rightarrow", () => {
+    g();
+  }), j("leftarrow", () => {
     b();
-  }), k(() => {
-    i && (t == null || t.listCategory(i).then((o) => A(o || [])));
-  }, [i]), k(() => {
-    c && i && r && (t == null || t.getDatasetsIdVersionsVidSsamplesSid(i, r, h.id).then((o) => {
-      P(o);
+  }), R(() => {
+    i && (o == null || o.listCategory(i).then((s) => _(s || [])));
+  }, [i]), R(() => {
+    d && i && l && f && m === "preview" && (o == null || o.getDatasetsIdVersionsVidSsamplesSid(i, l, f.id).then((s) => {
+      X(s);
     }));
-  }, [c, h]);
-  const F = () => {
-    C(!1);
-  }, U = (o) => {
-    K(o);
-  }, W = (o) => {
-    E(o);
-  }, G = () => {
-    u.confirm({
+  }, [d, f]);
+  const S = () => {
+    C();
+  }, ee = (s) => {
+    Q(s);
+  }, te = (s) => {
+    Z(s);
+  }, ae = () => {
+    I.confirm({
       title: "确认要删除此数据吗？",
       content: "删除后不可恢复，请慎重操作",
       onOk: () => {
-        !i || !r || !(e != null && e.id) || t == null || t.delDatasetsIdVersionsVidSamplesId(i, r, e.id).then(() => {
-          R.success("删除成功"), C(!1), z();
-        }).catch(() => R.error("删除失败"));
+        !i || !l || !(e != null && e.id) || o == null || o.delDatasetsIdVersionsVidSamplesId(i, l, e.id).then(() => {
+          w.success("删除成功"), C(), J();
+        }).catch(() => w.error("删除失败"));
       }
     });
   };
-  return /* @__PURE__ */ a(u, {
-    title: /* @__PURE__ */ a("div", {
+  return /* @__PURE__ */ t(I, {
+    title: m === "preview" ? /* @__PURE__ */ a("div", {
       children: ["详情", /* @__PURE__ */ a("span", {
         className: "ocr-modal-label-count",
-        children: ["标注框数量：", (O = e == null ? void 0 : e.labels) == null ? void 0 : O.length]
+        children: ["标注框数量：", (x = e == null ? void 0 : e.labels) == null ? void 0 : x.length]
       })]
-    }),
+    }) : "标注",
     width: "100%",
     wrapClassName: "ocr-modal-wrap",
     className: "ocr-modal",
-    open: c,
+    style: {
+      padding: 12
+    },
+    open: d,
     destroyOnClose: !0,
-    onCancel: () => F(),
+    onCancel: () => S(),
     maskClosable: !1,
     centered: !0,
     footer: null,
-    children: [/* @__PURE__ */ a("div", {
-      className: "ocr-detail",
+    children: m === "preview" ? /* @__PURE__ */ a(k, {
       children: [/* @__PURE__ */ a("div", {
-        className: "detail-content",
-        children: [/* @__PURE__ */ s(d, {
-          className: "select-prev",
-          icon: /* @__PURE__ */ s(x, {
-            type: "icon-zuojiantou"
-          }),
-          type: "text",
-          onClick: () => b()
-        }), /* @__PURE__ */ s("div", {
-          className: "select-main",
-          ref: J,
-          children: /* @__PURE__ */ s($, {
-            zoomMin: 0.1,
-            zoomMax: 10,
-            zoomStep: 0.1,
-            url: (e == null ? void 0 : e.fileUrl) || "",
-            ref: D,
-            labels: e == null ? void 0 : e.labels
-          })
-        }), /* @__PURE__ */ s(d, {
-          className: "select-next",
-          icon: /* @__PURE__ */ s(x, {
-            type: "icon-youjiantou"
-          }),
-          type: "text",
-          onClick: () => p()
-        })]
-      }), l !== "DetectionOCR" && /* @__PURE__ */ a("div", {
-        className: "detail-slide",
-        children: [/* @__PURE__ */ s("div", {
-          className: "slide-header",
-          children: /* @__PURE__ */ a(V, {
-            className: "header-tab",
-            onChange: U,
-            activeKey: f,
-            children: [/* @__PURE__ */ s(I, {
-              tab: "标注信息"
-            }, "annotation"), /* @__PURE__ */ s(I, {
-              tab: "JSON"
-            }, "json")]
-          })
-        }), /* @__PURE__ */ a("div", {
-          className: "slide-main",
-          children: [f === "annotation" && /* @__PURE__ */ s(H, {
-            children: l === "TableOCR" ? /* @__PURE__ */ s(S, {
-              currentInfer: e == null ? void 0 : e.labels,
-              onCellSelect: (o) => W(o),
-              currentCell: L
-            }) : (M = e == null ? void 0 : e.labels) != null && M.length ? /* @__PURE__ */ s(Z, {
-              labels: e == null ? void 0 : e.labels,
-              catagories: q
-            }) : "暂无标注信息"
-          }), f === "json" && /* @__PURE__ */ s(ee, {
-            name: !1,
-            iconStyle: "square",
-            indentWidth: 2,
-            displayDataTypes: !1,
-            displayObjectSize: !1,
-            enableClipboard: !1,
-            src: l === "TableOCR" ? e == null ? void 0 : e.labels : (e == null ? void 0 : e.ocrJson) || {}
+        className: "ocr-detail",
+        children: [/* @__PURE__ */ a("div", {
+          className: "detail-content",
+          children: [/* @__PURE__ */ t(c, {
+            className: "select-prev",
+            icon: /* @__PURE__ */ t(P, {
+              type: "icon-zuojiantou"
+            }),
+            type: "text",
+            onClick: () => b()
+          }), /* @__PURE__ */ a("div", {
+            className: "select-main",
+            ref: H,
+            children: [((M = e == null ? void 0 : e.sampleFileName) == null ? void 0 : M.name) && /* @__PURE__ */ t("div", {
+              className: "img-detail-name-overlay",
+              children: e.sampleFileName.name
+            }), /* @__PURE__ */ t(re, {
+              zoomMin: 0.1,
+              zoomMax: 10,
+              zoomStep: 0.1,
+              url: (e == null ? void 0 : e.fileUrl) || "",
+              ref: G,
+              labels: e == null ? void 0 : e.labels
+            })]
+          }), /* @__PURE__ */ t(c, {
+            className: "select-next",
+            icon: /* @__PURE__ */ t(P, {
+              type: "icon-youjiantou"
+            }),
+            type: "text",
+            onClick: () => g()
           })]
-        })]
-      })]
-    }), /* @__PURE__ */ s("div", {
-      className: "ocr-mark-action",
-      style: l === "DetectionOCR" ? {
-        width: "100%"
-      } : void 0,
-      children: /* @__PURE__ */ s(X, {
-        title: "预标注进行中，请任务完成后再试",
-        open: B,
-        children: /* @__PURE__ */ a(Y, {
-          onMouseEnter: () => m && g(!0),
-          onMouseLeave: () => g(!1),
-          children: [/* @__PURE__ */ s(d, {
-            type: "primary",
-            disabled: l !== "StructKVCOCR" && l !== "DetectionOCR" || m,
-            onClick: () => w(e == null ? void 0 : e.id),
-            children: "标注"
-          }), y && y.delete && /* @__PURE__ */ s(d, {
-            danger: !0,
-            type: "ghost",
-            onClick: G,
-            disabled: m,
-            children: "删除数据"
+        }), n !== "DetectionOCR" && /* @__PURE__ */ a("div", {
+          className: "detail-slide",
+          children: [/* @__PURE__ */ t("div", {
+            className: "slide-header",
+            children: /* @__PURE__ */ a(z, {
+              className: "header-tab",
+              onChange: ee,
+              activeKey: h,
+              children: [/* @__PURE__ */ t(V, {
+                tab: "标注信息"
+              }, "annotation"), /* @__PURE__ */ t(V, {
+                tab: "JSON"
+              }, "json")]
+            })
+          }), /* @__PURE__ */ a("div", {
+            className: "slide-main",
+            children: [h === "annotation" && /* @__PURE__ */ t(k, {
+              children: n === "TableOCR" ? /* @__PURE__ */ t(de, {
+                currentInfer: e == null ? void 0 : e.labels,
+                onCellSelect: (s) => te(s),
+                currentCell: Y
+              }) : (O = e == null ? void 0 : e.labels) != null && O.length ? /* @__PURE__ */ t(ne, {
+                labels: e == null ? void 0 : e.labels,
+                catagories: y
+              }) : "暂无标注信息"
+            }), h === "json" && /* @__PURE__ */ t(me, {
+              name: !1,
+              iconStyle: "square",
+              indentWidth: 2,
+              displayDataTypes: !1,
+              displayObjectSize: !1,
+              enableClipboard: !1,
+              src: n === "TableOCR" ? e == null ? void 0 : e.labels : (e == null ? void 0 : e.ocrJson) || {}
+            })]
           })]
+        })]
+      }), /* @__PURE__ */ t("div", {
+        className: "ocr-mark-action",
+        style: n === "DetectionOCR" ? {
+          width: "100%"
+        } : void 0,
+        children: /* @__PURE__ */ t(se, {
+          title: "预标注进行中，请任务完成后再试",
+          open: $,
+          children: /* @__PURE__ */ a(ie, {
+            onMouseEnter: () => p && v(!0),
+            onMouseLeave: () => v(!1),
+            children: [/* @__PURE__ */ t(c, {
+              type: "primary",
+              disabled: n !== "StructKVCOCR" && n !== "DetectionOCR" || p,
+              onClick: () => F(e == null ? void 0 : e.id),
+              children: "标注"
+            }), N && N.delete && /* @__PURE__ */ t(c, {
+              danger: !0,
+              type: "ghost",
+              onClick: ae,
+              disabled: p,
+              children: "删除数据"
+            })]
+          })
         })
-      })
-    })]
+      })]
+    }) : /* @__PURE__ */ t(ce, {
+      visible: d,
+      selectSampleId: K,
+      ocrMarkTmpl: n,
+      datasetId: i,
+      versionId: l,
+      onClose: C,
+      onSave: L,
+      categories: y,
+      externalFilterMode: B,
+      onFilterModeChange: A,
+      onPaginationChange: U,
+      externalSamples: W,
+      externalPagination: E,
+      externalCurrentIndex: q
+    })
   });
 };
 export {
-  fe as default
+  ke as default
 };