@glodon-aiot/dataset-annotation 3.2.2-beta.3 → 3.2.2-beta.4

package/dist/es/components/OcrAnnotation/MarkOCRBoxModal/OCRBoxMaker/style.less.mjs

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

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

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

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

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

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

@@ -0,0 +1,180 @@
+import { jsxs as a, jsx as s, Fragment as G } from "react/jsx-runtime";
+import { useRef as N, useState as n, useContext as H, useEffect as k } from "react";
+import { Modal as u, Button as d, Tabs as I, Tooltip as Q, Space as X, message as v } from "antd";
+import "./style.less.mjs";
+import Y from "react-json-view";
+import { useKeyPress as R } from "ahooks";
+import Z from "../../MarkDescriptions/index.mjs";
+import T from "../../Iconfont/index.mjs";
+import _ from "../../../DatasetContext/index.mjs";
+import $ from "../../OCRImageViewer/index.mjs";
+import S from "../../ScanTable/index.mjs";
+const {
+  TabPane: x
+} = I, me = (V) => {
+  var O, M;
+  const {
+    isOcrModalVisible: c,
+    setIsOcrModalVisible: C,
+    record: h,
+    onNextClick: p,
+    onPrevClick: b,
+    datasetId: i,
+    versionId: r,
+    ocrMarkTmpl: l,
+    handleMarkClick: j,
+    updateSampleList: w,
+    isPreLabel: m
+  } = V, z = N(null), D = N({}), [f, K] = n("annotation"), [e, P] = n(), [J, L] = n(), [E, q] = n([]), [A, g] = n(!1), {
+    services: {
+      cvforceDatahub: t
+    },
+    featuresConfig: {
+      datasetManager: y
+    } = {
+      datasetManager: !1
+    }
+  } = H(_);
+  R("rightarrow", () => {
+    p();
+  }), R("leftarrow", () => {
+    b();
+  }), k(() => {
+    i && (t == null || t.listCategory(i).then((o) => q(o || [])));
+  }, [i]), k(() => {
+    c && i && r && (t == null || t.getDatasetsIdVersionsVidSsamplesSid(i, r, h.id).then((o) => {
+      P(o);
+    }));
+  }, [c, h]);
+  const B = () => {
+    C(!1);
+  }, F = (o) => {
+    K(o);
+  }, U = (o) => {
+    L(o);
+  }, W = () => {
+    u.confirm({
+      title: "确认要删除此数据吗？",
+      content: "删除后不可恢复，请慎重操作",
+      onOk: () => {
+        !i || !r || !(e != null && e.id) || t == null || t.delDatasetsIdVersionsVidSamplesId(i, r, e.id).then(() => {
+          v.success("删除成功"), C(!1), w();
+        }).catch(() => v.error("删除失败"));
+      }
+    });
+  };
+  return /* @__PURE__ */ a(u, {
+    title: /* @__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]
+      })]
+    }),
+    width: "100%",
+    wrapClassName: "ocr-modal-wrap",
+    className: "ocr-modal",
+    open: c,
+    destroyOnClose: !0,
+    onCancel: () => B(),
+    maskClosable: !1,
+    centered: !0,
+    footer: null,
+    children: [/* @__PURE__ */ a("div", {
+      className: "ocr-detail",
+      children: [/* @__PURE__ */ a("div", {
+        className: "detail-content",
+        children: [/* @__PURE__ */ s(d, {
+          className: "select-prev",
+          icon: /* @__PURE__ */ s(T, {
+            type: "icon-zuojiantou"
+          }),
+          type: "text",
+          onClick: () => b()
+        }), /* @__PURE__ */ s("div", {
+          className: "select-main",
+          ref: D,
+          children: /* @__PURE__ */ s($, {
+            zoomMin: 0.1,
+            zoomMax: 10,
+            zoomStep: 0.1,
+            url: (e == null ? void 0 : e.fileUrl) || "",
+            ref: z,
+            labels: e == null ? void 0 : e.labels
+          })
+        }), /* @__PURE__ */ s(d, {
+          className: "select-next",
+          icon: /* @__PURE__ */ s(T, {
+            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(I, {
+            className: "header-tab",
+            onChange: F,
+            activeKey: f,
+            children: [/* @__PURE__ */ s(x, {
+              tab: "标注信息"
+            }, "annotation"), /* @__PURE__ */ s(x, {
+              tab: "JSON"
+            }, "json")]
+          })
+        }), /* @__PURE__ */ a("div", {
+          className: "slide-main",
+          children: [f === "annotation" && /* @__PURE__ */ s(G, {
+            children: l === "TableOCR" ? /* @__PURE__ */ s(S, {
+              currentInfer: e == null ? void 0 : e.labels,
+              onCellSelect: (o) => U(o),
+              currentCell: J
+            }) : (M = e == null ? void 0 : e.labels) != null && M.length ? /* @__PURE__ */ s(Z, {
+              labels: e == null ? void 0 : e.labels,
+              catagories: E
+            }) : "暂无标注信息"
+          }), f === "json" && // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+          // @ts-ignore
+          /* @__PURE__ */ s(Y, {
+            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) || {}
+          })]
+        })]
+      })]
+    }), /* @__PURE__ */ s("div", {
+      className: "ocr-mark-action",
+      style: l === "DetectionOCR" ? {
+        width: "100%"
+      } : void 0,
+      children: /* @__PURE__ */ s(Q, {
+        title: "预标注进行中，请任务完成后再试",
+        open: A,
+        children: /* @__PURE__ */ a(X, {
+          onMouseEnter: () => m && g(!0),
+          onMouseLeave: () => g(!1),
+          children: [/* @__PURE__ */ s(d, {
+            type: "primary",
+            disabled: l !== "StructKVCOCR" && l !== "DetectionOCR" || m,
+            onClick: () => j(e == null ? void 0 : e.id),
+            children: "标注"
+          }), y && y.delete && /* @__PURE__ */ s(d, {
+            danger: !0,
+            type: "ghost",
+            onClick: W,
+            disabled: m,
+            children: "删除数据"
+          })]
+        })
+      })
+    })]
+  });
+};
+export {
+  me as default
+};

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

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