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

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

@@ -0,0 +1,382 @@
+var xe = Object.defineProperty, ye = Object.defineProperties;
+var Le = Object.getOwnPropertyDescriptors;
+var re = Object.getOwnPropertySymbols;
+var Ve = Object.prototype.hasOwnProperty, Pe = Object.prototype.propertyIsEnumerable;
+var oe = (f, n, r) => n in f ? xe(f, n, { enumerable: !0, configurable: !0, writable: !0, value: r }) : f[n] = r, h = (f, n) => {
+  for (var r in n || (n = {}))
+    Ve.call(n, r) && oe(f, r, n[r]);
+  if (re)
+    for (var r of re(n))
+      Pe.call(n, r) && oe(f, r, n[r]);
+  return f;
+}, C = (f, n) => ye(f, Le(n));
+import { jsx as c, jsxs as D } from "react/jsx-runtime";
+import { useState as a, useEffect as m } from "react";
+import { message as $, ConfigProvider as Re, Space as x, Select as Ne, Tabs as Oe, Tooltip as we, Button as ze, Row as Be, Col as Ke, Pagination as Ae } from "antd";
+import Ee from "antd/es/locale/zh_CN";
+import "./style.less.mjs";
+import Fe from "./OcrModal/index.mjs";
+import Ue from "./MarkKVOCRModal/index.mjs";
+import Ze from "./MarkOCRBoxModal/index.mjs";
+import $e from "./PreLabelOCR/index.mjs";
+import je from "lodash/merge";
+import { UploadZipsStatus as ie, PreLabelTaskStatus as We } from "@glodon-aiot/apis";
+import _e from "../../Common/Page/index.mjs";
+import He from "../Thumbnail/index.mjs";
+import Je from "../DataEmpty/index.mjs";
+import Qe from "../../DatasetContext/index.mjs";
+import { DEFAULT_FILTER_SWITCH as qe } from "../../constant.mjs";
+import Ge from "../PageNav/index.mjs";
+const gt = (f) => {
+  const {
+    services: n,
+    datasetId: r,
+    versionId: S,
+    backTo: ne,
+    labelManager: ce = !1,
+    datasetManager: T = !1,
+    annotation: le,
+    header: d = {
+      versionSelect: !1
+    },
+    exampleSrcRoot: ae
+  } = f, [i, R] = a(), [t, I] = a(), [y, j] = a([]), [W, L] = a(!1), [N, O] = a(!1), [g, _] = a(0), [u, H] = a(), [w, b] = a(), [se, J] = a(), [Q, M] = a(-1), [l, k] = a({
+    current: 1,
+    pageSize: 50
+  }), [V, q] = a(0), [v, de] = a(""), [G, z] = a(!1), [X, Y] = a(), [B, ee] = a(!1), [P, K] = a(null), [ue, te] = a(!1), [fe, pe] = a(!1), [A, me] = a(""), [he, ge] = a(), E = h({}, qe);
+  je(E, f.filterSwitch);
+  const be = () => {
+    J(void 0), z(!0);
+  }, ke = (e, o) => {
+    k({
+      current: e,
+      pageSize: o
+    });
+  }, Ce = (e) => {
+    if (!u)
+      return;
+    M(e);
+    const o = u[e];
+    b(o), O(!0);
+  };
+  m(() => {
+    n.cvforceDatahub && n.cvforceDatahub.getOptionsImgproxy().then((e) => ge(e));
+  }, [n.cvforceDatahub]), m(() => {
+    !r || t || (L(!0), Promise.all([
+      // 数据集
+      n.cvforceDatahub.getDatasetsId(r).then((e) => {
+        R(e);
+      }),
+      // 版本列表
+      n.cvforceDatahub.getDatasetsIdVersions(r, 1, 1e3).then((e) => {
+        j(e.recordList), I(S ? e.recordList.find((o) => o.id === S) : e.recordList[0]);
+      })
+    ]).finally(() => L(!1)));
+  }, [r, t == null ? void 0 : t.id]), m(() => {
+    r && (L(!0), Promise.all([
+      // 数据集
+      n.cvforceDatahub.getDatasetsId(r).then((e) => {
+        R(e);
+      }),
+      // 版本列表
+      n.cvforceDatahub.getDatasetsIdVersions(r, 1, 1e3).then((e) => {
+        j(e.recordList), I(S ? e.recordList.find((o) => o.id === S) : e.recordList[0]);
+      })
+    ]).finally(() => L(!1)));
+  }, [r, S]), m(() => {
+    r && n.cvforceDatahub.getDatasetsId(r).then((e) => {
+      R(e);
+    });
+  }, []);
+  const Se = () => {
+    !r || !t || n.cvforceDatahub.getDatasetsIdVersionsVid(r, t.id).then((e) => {
+      me(e.versionAsyncTask.state);
+    });
+  };
+  m(() => {
+    Se();
+  }, [r, t == null ? void 0 : t.id]), m(() => {
+    A !== ie.Pending && A !== ie.Started || $.warning("文件导入中，请刷新页面再试");
+  }, [A]);
+  const Te = () => {
+    !r || !t || n.cvforceDatahub.getDatasetsIdVersionsVidLabelcount(r, t.id).then((e) => {
+      const o = e.find((p) => p.id === "allcodeid"), s = e.find((p) => p.id === "nocodeid");
+      _(o ? o.count : 0), q(s ? s.count : 0);
+    });
+  }, F = () => {
+    const e = l.current || 1, o = l.pageSize || 50;
+    !r || !t || (n.cvforceDatahub.getDatasetsIdVersionsVidSamples(r, t.id, {
+      labels: v || void 0,
+      pageNo: e,
+      pageSize: o
+    }).then((s) => {
+      const p = s.recordList || [];
+      if (!p.length && s.totalResults > 0) {
+        k(C(h({}, l), {
+          current: Math.ceil(s.totalResults / o)
+        })), Math.ceil(s.totalResults / o) === e && H([]);
+        return;
+      }
+      if (H(p), k(C(h({}, l), {
+        total: s.totalResults
+      })), l.type === "next" && p.length) {
+        M(0);
+        const Z = s.recordList[0];
+        b(Z);
+      }
+      if (l.type === "prev" && p.length) {
+        M(l.pageSize - 1);
+        const Z = s.recordList[l.pageSize - 1];
+        b(Z);
+      }
+    }), Te());
+  };
+  m(F, [l.current, l.pageSize, v, X, t == null ? void 0 : t.id, r]);
+  const Ie = () => {
+    if (!u)
+      return;
+    const e = Q + 1;
+    if (e >= u.length) {
+      let o = g;
+      v === "hascode" && (o = g - V), v === "nocode" && (o = V), (l.current - 1) * l.pageSize + e >= o ? $.error("已经是最后一个了") : k({
+        current: l.current + 1,
+        pageSize: l.pageSize,
+        type: "next"
+      });
+    } else {
+      M(e);
+      const o = u[e];
+      b(o);
+    }
+  }, Me = () => {
+    if (!u)
+      return;
+    const e = Q - 1;
+    if (e < 0)
+      (l.current - 1) * l.pageSize + e < 0 ? $.error("已经是第一个了") : k({
+        current: l.current - 1,
+        pageSize: l.pageSize,
+        type: "prev"
+      });
+    else {
+      M(e);
+      const o = u[e];
+      b(o);
+    }
+  };
+  m(() => {
+    if (!r) {
+      P && (clearTimeout(P), K(null)), Y(void 0);
+      return;
+    }
+    U();
+  }, [t == null ? void 0 : t.id]), m(() => {
+    U();
+  }, [fe]);
+  const U = () => {
+    !r || !t || n.cvforceDatahub.getPreLabelTask(r, t.id).then((e) => {
+      const o = e[0];
+      if (Y(o), P && (clearTimeout(P), K(null)), !(o != null && o.state) || o.state !== We.Started) {
+        ee(!1);
+        return;
+      } else
+        ee(!0);
+      K(setTimeout(() => {
+        U();
+      }, 1e4));
+    });
+  }, ve = [{
+    key: "",
+    label: `全部（${g}）`,
+    children: null
+  }, {
+    key: "nocode",
+    label: `未标注（${V}）`,
+    children: null
+  }, {
+    key: "hascode",
+    label: `已标注（${g - V}）`,
+    children: null
+  }];
+  m(() => {
+    N || b(void 0);
+  }, [N]), m(() => {
+    i && (i.markType !== "ocr" || (i == null ? void 0 : i.ocrMarkTmpl) !== "DetectionOCR") && console.error(`当前数据集不是DetectionOCR(ocr检测)类型，请检查数据集类型。
+请检查markType是否为ocr并且ocrMarkTmpl为DetectionOCR。`);
+  }, [i]);
+  const De = (i == null ? void 0 : i.ocrMarkTmpl) === "StructKVCOCR" ? Ue : Ze;
+  return /* @__PURE__ */ c(Re, {
+    locale: Ee,
+    children: /* @__PURE__ */ c(Qe.Provider, {
+      value: {
+        datasetId: r,
+        versionId: (t == null ? void 0 : t.id) || "",
+        no: (t == null ? void 0 : t.no) || "",
+        loading: W,
+        version: t,
+        versions: y,
+        samples: u,
+        pagination: l,
+        services: n,
+        featuresConfig: {
+          labelManager: ce,
+          datasetManager: T,
+          annotation: le,
+          filterSwitch: E,
+          header: d,
+          exampleSrcRoot: ae
+        }
+      },
+      children: i ? i.markType === "ocr" ? /* @__PURE__ */ D("div", {
+        className: "ocr-version-detail",
+        children: [/* @__PURE__ */ D(_e, {
+          className: "product-details-page",
+          header: d === !1 ? void 0 : /* @__PURE__ */ c(Ge, {
+            backTo: ne,
+            onBackClick: d != null && d.onBackClick ? () => {
+              d.onBackClick && d.onBackClick();
+            } : void 0,
+            left: !W && /* @__PURE__ */ D(x, {
+              children: [/* @__PURE__ */ c("span", {
+                style: {
+                  fontWeight: "bold"
+                },
+                children: i == null ? void 0 : i.name
+              }), /* @__PURE__ */ c(Ne, {
+                disabled: (d == null ? void 0 : d.versionSelect) === !1,
+                value: t == null ? void 0 : t.id,
+                style: {
+                  width: 70
+                },
+                options: y.map((e) => ({
+                  label: e.no,
+                  value: e.id
+                })),
+                onChange: (e) => {
+                  const o = y.find((p) => p.id === e);
+                  if (!i || !o)
+                    return;
+                  const s = y.find((p) => p.id === o.id);
+                  I(s), d != null && d.onVersionChange && d.onVersionChange(s);
+                }
+              })]
+            }),
+            right: /* @__PURE__ */ c(x, {
+              children: !w && T && T.preLabel && (t != null && t.id) ? /* @__PURE__ */ c($e, {
+                datasetId: r,
+                versionId: t.id,
+                disabled: !g || (i == null ? void 0 : i.ocrMarkTmpl) !== "DetectionOCR",
+                setIsTaskChange: pe,
+                preLabelTask: X,
+                datasetType: "ocr"
+              }) : null
+            })
+          }),
+          children: [/* @__PURE__ */ D("div", {
+            className: "ocr-header",
+            children: [/* @__PURE__ */ c(x, {
+              children: E.tabs && /* @__PURE__ */ c(Oe, {
+                className: "header-tab",
+                defaultActiveKey: "",
+                onChange: (e) => {
+                  k(C(h({}, l), {
+                    current: 1,
+                    total: 0
+                  })), de(e);
+                },
+                items: ve
+              })
+            }), /* @__PURE__ */ c(x, {
+              children: T && T.label && /* @__PURE__ */ c(we, {
+                title: "预标注进行中，请任务完成后再试",
+                open: ue,
+                children: /* @__PURE__ */ c(ze, {
+                  type: "primary",
+                  disabled: !g || (i == null ? void 0 : i.ocrMarkTmpl) !== "StructKVCOCR" && (i == null ? void 0 : i.ocrMarkTmpl) !== "DetectionOCR" || B,
+                  onClick: be,
+                  onMouseEnter: () => B && te(!0),
+                  onMouseLeave: () => te(!1),
+                  style: {
+                    marginRight: "10px"
+                  },
+                  children: "单张标注"
+                })
+              })
+            })]
+          }), /* @__PURE__ */ c("div", {
+            className: "d-row",
+            children: u != null && u.length ? /* @__PURE__ */ D(x, {
+              direction: "vertical",
+              className: "d-right",
+              children: [/* @__PURE__ */ c(Be, {
+                gutter: [16, 16],
+                wrap: !0,
+                className: "d-right-row",
+                children: u == null ? void 0 : u.map((e, o) => /* @__PURE__ */ c(Ke, {
+                  className: "d-right-col",
+                  onClick: () => Ce(o),
+                  children: /* @__PURE__ */ c("div", {
+                    className: "right-col-wrapper",
+                    children: /* @__PURE__ */ c("div", {
+                      className: "right-col-img",
+                      children: /* @__PURE__ */ c(He, {
+                        fileKey: e.fileKey,
+                        width: 124,
+                        imgProxy: he
+                      })
+                    })
+                  })
+                }, e.id))
+              }), /* @__PURE__ */ c(Ae, h({
+                onChange: ke,
+                className: "d-right-pageination",
+                showQuickJumper: !0,
+                showSizeChanger: !0
+              }, l))]
+            }) : /* @__PURE__ */ c(Je, {})
+          })]
+        }), w && /* @__PURE__ */ c(Fe, {
+          isOcrModalVisible: N,
+          setIsOcrModalVisible: O,
+          record: w,
+          onNextClick: () => Ie(),
+          onPrevClick: () => Me(),
+          ocrMarkTmpl: i == null ? void 0 : i.ocrMarkTmpl,
+          datasetId: r,
+          versionId: t == null ? void 0 : t.id,
+          handleMarkClick: (e) => {
+            J(e), O(!1), z(!0);
+          },
+          updateSampleList: F,
+          isPreLabel: B
+        }), G && (t == null ? void 0 : t.id) && /* @__PURE__ */ c(De, {
+          visible: G,
+          selectSampleId: se,
+          ocrMarkTmpl: i == null ? void 0 : i.ocrMarkTmpl,
+          datasetId: r,
+          versionId: t.id,
+          onClose: () => z(!1),
+          onSave: (e) => {
+            var o;
+            if (!r || !t)
+              throw "datasetId or version is not exist.";
+            return n.cvforceDatahub.updateSamples(r, t.id, [C(h({}, e), {
+              labels: (o = e.labels) == null ? void 0 : o.map((s) => C(h({}, s), {
+                box: s.box.length > 4 ? s.box.splice(0, 4) : s.box
+              }))
+            })]).then(() => (F(), !0));
+          }
+        })]
+      }) : /* @__PURE__ */ c("div", {
+        style: {
+          color: "red",
+          textAlign: "center"
+        },
+        children: "当前数据集不是ocr(ocr检测)类型，请检查数据集类型。"
+      }) : null
+    })
+  });
+};
+export {
+  gt as default
+};

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

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

package/dist/es/components/OcrImageAnnotation/index.mjs

@@ -0,0 +1,40 @@
+import { jsx as i } from "react/jsx-runtime";
+import { useRef as b, useState as h, useEffect as v } from "react";
+import "./style.less.mjs";
+import g from "../OcrAnnotation/MarkOCRBoxModal/ImageDetail/index.mjs";
+const S = (l) => {
+  const {
+    value: e,
+    onChange: n,
+    readonly: o,
+    onNextClick: s,
+    onPrevClick: c,
+    showLabelList: C,
+    categories: d,
+    renderLabelItem: m
+  } = l, u = b(null), [r, f] = h();
+  return v(() => {
+    e && f(e);
+  }, [e]), /* @__PURE__ */ i("div", {
+    className: "ocr-version-detail",
+    children: /* @__PURE__ */ i("div", {
+      className: "d-right",
+      ref: u,
+      children: r && /* @__PURE__ */ i(g, {
+        readonly: o,
+        currentSample: r,
+        onChange: n,
+        onNextClick: s,
+        onPrevClick: c,
+        categories: d || Object.values(r.labels.reduce((t, a) => (t.find((p) => p.id === a.id) || t.push({
+          id: a.id,
+          name: a.label
+        }), t), [])),
+        renderLabelItem: m
+      })
+    })
+  });
+};
+export {
+  S as default
+};

package/dist/es/components/OcrImageAnnotation/style.less.mjs

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

package/dist/es/components/PageNav/index.mjs

@@ -0,0 +1,30 @@
+import { jsxs as e, jsx as i } from "react/jsx-runtime";
+import { Space as r } from "antd";
+import "./style.less.mjs";
+import s from "../Iconfont/index.mjs";
+const d = (c) => {
+  const {
+    left: n,
+    right: o,
+    onBackClick: a
+  } = c;
+  return /* @__PURE__ */ e("div", {
+    className: "page-nav",
+    children: [/* @__PURE__ */ e(r, {
+      size: 12,
+      children: [a && /* @__PURE__ */ e(r, {
+        size: 4,
+        className: "header-back",
+        onClick: a,
+        children: [/* @__PURE__ */ i(s, {
+          type: "icon-fanhui1"
+        }), /* @__PURE__ */ i("span", {
+          children: "返回"
+        })]
+      }), n]
+    }), o]
+  });
+};
+export {
+  d as default
+};

package/dist/es/components/PageNav/style.less.mjs

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