@glodon-aiot/dataset-annotation 3.18.0-alpha.11 → 3.18.0-alpha.13

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

@@ -0,0 +1,310 @@
+import { jsxs as u, jsx as n } from "react/jsx-runtime";
+import { useContext as ze, useState as s, useEffect as k, useRef as Z } from "react";
+import { Button as C, Space as fe, Tooltip as q, Empty as me, message as ge } from "antd";
+import { ZoomInOutlined as He, ZoomOutOutlined as Te, FullscreenExitOutlined as De, FullscreenOutlined as Fe } from "@ant-design/icons";
+import { useFullscreen as $e } from "ahooks";
+import je from "../../icon_linear.svg.mjs";
+import "./style.less.mjs";
+import Re from "../../../../../DatasetContext/index.mjs";
+import { tagColors as Be } from "../../../../../utils/index.mjs";
+import he from "../../../../Iconfont/index.mjs";
+const ot = (pe) => {
+  const {
+    serviceId: v,
+    confidence: P,
+    datasetId: E,
+    versionId: z,
+    isVisible: _
+  } = pe, {
+    services: {
+      cvforceDatahub: H,
+      aiShop: T
+    }
+  } = ze(Re), [d, D] = s(), [h, xe] = s([]), [F, G] = s(!1), [I, J] = s([]), [f, ve] = s([]), [y, Ie] = s([]);
+  k(() => {
+    !H || !E || !z || !_ || H.getDatasetsIdVersionsVidSamples(E, z, {
+      pageNo: 1,
+      pageSize: 10,
+      labels: "nocode",
+      confirmStatus: 0,
+      extractPercent: 10,
+      extractMax: 10
+    }).then((e) => {
+      const t = e.recordList || [], o = t == null ? void 0 : t[0];
+      o != null && o.fileUrl && D(o == null ? void 0 : o.fileUrl), xe(t);
+    }).catch(() => !1);
+  }, [_, E, z, H]), k(() => {
+    !d || !v || (G(!0), J([]), T == null || T.detectionLayout(v, {
+      pages: [{
+        image: d
+      }]
+    }).then((e) => {
+      var o;
+      const t = (o = e == null ? void 0 : e.prediction) != null ? o : [];
+      J(Array.isArray(t) ? t : []);
+    }).finally(() => {
+      G(!1);
+    }));
+  }, [d, v]), k(() => {
+    var t;
+    const e = (t = I == null ? void 0 : I.filter((o) => {
+      var r;
+      return ((r = o.score) != null ? r : 1) >= (P != null ? P : 0);
+    })) != null ? t : [];
+    ve(e);
+  }, [I, P]), k(() => {
+    if (!f)
+      return;
+    const e = I.map((t) => t.label).reduce((t, o) => t.includes(o) ? t : [...t, o], []);
+    Ie(e.map((t, o) => ({
+      label: t,
+      count: f.filter((r) => r.label === t).length,
+      color: Be[o],
+      id: `${o}_${t}`
+    })));
+  }, [f, I]);
+  const ye = () => {
+    const e = h.findIndex((t) => t.fileUrl === d);
+    if (e <= 0) {
+      ge.warning("已经是第一张了");
+      return;
+    }
+    D(h[e - 1].fileUrl);
+  }, Me = () => {
+    const e = h.findIndex((t) => t.fileUrl === d);
+    if (e >= h.length - 1) {
+      ge.warning("已经是最后一张了");
+      return;
+    }
+    D(h[e + 1].fileUrl);
+  }, w = Z(null), $ = Z({}), K = Z(null), [Q, {
+    toggleFullscreen: Ne
+  }] = $e(K), [ee, Ce] = s(0), [te, Pe] = s(0), [c, j] = s(1), [m, p] = s(0), [g, x] = s(0), [we, Se] = s(!1), [oe, ne] = s(-1), [l, R] = s(0), [a, B] = s(0), [se, S] = s(!1), [ce, Ye] = s(), [Y, b] = s({
+    x: 0,
+    y: 0
+  }), re = 50, ie = 50, le = 50, ae = 50, X = 0.05, O = 3, W = 0.2, be = () => {
+    if (!w.current)
+      return;
+    const e = $.current.clientWidth, t = $.current.clientHeight, o = w.current.naturalWidth, r = w.current.naturalHeight, M = e / o, N = t / r, i = M < N ? M : N, ke = (e - o * i) / 2, Ee = (t - r * i) / 2;
+    Ce(e), Pe(t), j(i), p(ke), x(Ee), R(o * i), B(r * i), Se(!0);
+  }, L = (e) => ne(e), Xe = () => ne(-1), U = () => {
+    m < -l + ae && p(-l + ae), g < -a + re && x(-a + re), m > ee - ie && p(ee - ie), g > te - le && x(te - le);
+  }, V = () => {
+    if (c >= O)
+      return;
+    let e = c * (1 + X), t = X;
+    e >= O && (e = O, t = O / c - 1), p(m - l * (t / 2)), x(g - a * (t / 2)), R(l + l * t), B(a + a * t), j(e), U();
+  }, A = () => {
+    if (c <= W)
+      return;
+    let e = c * (1 - X), t = X;
+    e <= W && (e = W, t = 1 - W / c), p(m - l * (-t / 2)), x(g - a * (-t / 2)), R(l + l * -t), B(a + a * -t), j(e), U();
+  };
+  window.onmouseup = () => S(!1);
+  const Oe = (e) => (e.deltaY < 0 ? V() : A(), !1), We = (e) => {
+    S(!0), b({
+      x: e.pageX,
+      y: e.pageY
+    });
+  }, Le = (e) => {
+    if (!se)
+      return;
+    const t = e.pageX - Y.x, o = e.pageY - Y.y;
+    b({
+      x: e.pageX,
+      y: e.pageY
+    }), p(m + t), x(g + o), U();
+  }, ue = () => S(!1), Ue = (e) => {
+    const {
+      touches: t
+    } = e;
+    S(!0), t.length < 2 ? b({
+      x: t[0].clientX,
+      y: t[0].clientY
+    }) : Ye(t);
+  }, de = (e, t) => {
+    const o = t.pageX - e.pageX, r = t.pageY - e.pageY;
+    return Math.sqrt(o * o + r * r);
+  };
+  return /* @__PURE__ */ u("div", {
+    className: "pre-label-content",
+    children: [v && d && /* @__PURE__ */ u("div", {
+      className: "pre-label-content-preview",
+      ref: K,
+      children: [/* @__PURE__ */ n(C, {
+        className: "select-prev",
+        icon: /* @__PURE__ */ n(he, {
+          type: "icon-zuojiantou"
+        }),
+        disabled: F,
+        type: "text",
+        onClick: () => ye(),
+        tabIndex: 0
+      }), /* @__PURE__ */ u("div", {
+        className: "select-main",
+        children: [/* @__PURE__ */ n("div", {
+          className: "marker-action",
+          children: /* @__PURE__ */ n("div", {
+            className: "create-mode-select",
+            children: /* @__PURE__ */ u(fe, {
+              direction: "vertical",
+              size: 16,
+              children: [/* @__PURE__ */ n(q, {
+                title: "放大-鼠标滚轮前滑",
+                placement: "right",
+                children: /* @__PURE__ */ n(C, {
+                  icon: /* @__PURE__ */ n(He, {}),
+                  onClick: V
+                })
+              }), /* @__PURE__ */ n(q, {
+                title: "缩小-鼠标滚轮后滑",
+                placement: "right",
+                children: /* @__PURE__ */ n(C, {
+                  icon: /* @__PURE__ */ n(Te, {}),
+                  onClick: A
+                })
+              }), /* @__PURE__ */ n(q, {
+                title: Q ? "退出全屏" : "全屏",
+                placement: "right",
+                children: /* @__PURE__ */ n(C, {
+                  icon: Q ? /* @__PURE__ */ n(De, {}) : /* @__PURE__ */ n(Fe, {}),
+                  onClick: Ne
+                })
+              })]
+            })
+          })
+        }), /* @__PURE__ */ n("div", {
+          className: "right-col-wrapper",
+          children: /* @__PURE__ */ u("div", {
+            className: "right-col-img",
+            ref: $,
+            onMouseDown: We,
+            onMouseMove: Le,
+            onMouseUp: ue,
+            onWheel: Oe,
+            onTouchStart: Ue,
+            onTouchMove: (e) => {
+              if (!(!se || !e.touches))
+                if (e.touches.length < 2) {
+                  const t = e.touches[0].clientX - Y.x, o = e.touches[0].clientY - Y.y;
+                  b({
+                    x: e.touches[0].clientX,
+                    y: e.touches[0].clientY
+                  }), p(m + t), x(g + o), U();
+                } else {
+                  const t = de(e.touches[0], e.touches[1]), o = de(ce[0], ce[1]);
+                  t > o ? V() : A();
+                }
+            },
+            onTouchEnd: ue,
+            children: [/* @__PURE__ */ n("img", {
+              src: d,
+              style: {
+                position: "absolute",
+                left: m,
+                top: g,
+                width: l,
+                height: a
+              },
+              ref: w,
+              onLoad: be,
+              alt: ""
+            }), we && /* @__PURE__ */ n("svg", {
+              style: {
+                position: "absolute",
+                left: m,
+                top: g,
+                width: l,
+                height: a
+              },
+              children: (f == null ? void 0 : f.length) && (f == null ? void 0 : f.map((e, t) => {
+                var o, r, M, N;
+                return /* @__PURE__ */ u("g", {
+                  children: [/* @__PURE__ */ n("g", {
+                    style: {
+                      cursor: "pointer"
+                    },
+                    children: /* @__PURE__ */ n("polygon", {
+                      points: (o = e == null ? void 0 : e.box) == null ? void 0 : o.map((i) => `${i[0] * c} ${i[1] * c}`).join(","),
+                      onMouseOver: () => L(t),
+                      onMouseEnter: () => L(t),
+                      onMouseLeave: () => Xe(),
+                      style: {
+                        fill: t === oe ? `${(r = y == null ? void 0 : y.find((i) => i.label === e.label)) == null ? void 0 : r.color}59` : "transparent",
+                        stroke: ((M = y == null ? void 0 : y.find((i) => i.label === e.label)) == null ? void 0 : M.color) || "",
+                        strokeWidth: 2,
+                        cursor: "pointer"
+                      }
+                    }, e.label)
+                  }), t === oe && /* @__PURE__ */ u("g", {
+                    onMouseOver: () => L(t),
+                    onMouseEnter: () => L(t),
+                    style: {
+                      position: "absolute",
+                      left: "100px",
+                      height: "100px",
+                      cursor: "pointer"
+                    },
+                    fill: "#fff",
+                    children: [/* @__PURE__ */ n("text", {
+                      x: e.box[0][0] * c + 4,
+                      y: e.box[0][1] * c + 16,
+                      children: e.label
+                    }), /* @__PURE__ */ u("text", {
+                      x: e.box[0][0] * c + 4,
+                      y: e.box[0][1] * c + 32,
+                      children: [(e.box[2][0] - e.box[0][0]).toFixed(0), "x", (e.box[2][1] - e.box[0][1]).toFixed(0)]
+                    }), /* @__PURE__ */ u("text", {
+                      x: e.box[0][0] * c + 4,
+                      y: e.box[0][1] * c + 48,
+                      children: ["Score: ", (N = e.score) != null ? N : "-"]
+                    })]
+                  }, t) || null]
+                }, t);
+              }))
+            }), F && /* @__PURE__ */ n("img", {
+              className: "pic-curr-linear",
+              src: je,
+              alt: ""
+            })]
+          })
+        }), /* @__PURE__ */ n("div", {
+          className: "page-info",
+          children: (() => {
+            const e = h.findIndex((o) => o.fileUrl === d);
+            return `${e >= 0 ? e + 1 : 0}/${h.length}`;
+          })()
+        })]
+      }), /* @__PURE__ */ n(C, {
+        className: "select-next",
+        disabled: F,
+        icon: /* @__PURE__ */ n(he, {
+          type: "icon-youjiantou"
+        }),
+        type: "text",
+        onClick: () => Me()
+      })]
+    }), v && !d && /* @__PURE__ */ n("div", {
+      className: "pre-label-content-nodata",
+      children: /* @__PURE__ */ n(me, {
+        description: "暂无未标注的图片"
+      })
+    }), !v && /* @__PURE__ */ n("div", {
+      className: "pre-label-content-nodata",
+      children: /* @__PURE__ */ n(me, {
+        description: /* @__PURE__ */ u(fe, {
+          direction: "vertical",
+          size: 8,
+          children: [/* @__PURE__ */ n("div", {
+            children: "选择服务，查看预览测试效果"
+          }), /* @__PURE__ */ n("div", {
+            children: "随机抽取占比不超过未标注总的 10%，最少1张最高不超过 10 张"
+          })]
+        })
+      })
+    })]
+  });
+};
+export {
+  ot as default
+};

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

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

package/dist/es/components/OcrAnnotation/PreLabelOCR/icon_linear.svg.mjs

@@ -0,0 +1,4 @@
+const I = "data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMTkyMCIgaGVpZ2h0PSI2NCIgdmlld0JveD0iMCAwIDE5MjAgNjQiIGZpbGw9Im5vbmUiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyI+CjxnIGNsaXAtcGF0aD0idXJsKCNjbGlwMF8zNDQyXzkxNzkpIj4KPHBhdGggZD0iTTE5MjAgMEgwVjY0SDE5MjBWMFoiIGZpbGw9InVybCgjcGFpbnQwX2xpbmVhcl8zNDQyXzkxNzkpIi8+CjwvZz4KPGRlZnM+CjxsaW5lYXJHcmFkaWVudCBpZD0icGFpbnQwX2xpbmVhcl8zNDQyXzkxNzkiIHgxPSI5NjAiIHkxPSIwIiB4Mj0iOTYwIiB5Mj0iNjQiIGdyYWRpZW50VW5pdHM9InVzZXJTcGFjZU9uVXNlIj4KPHN0b3Agc3RvcC1jb2xvcj0iIzBGNjRGRiIgc3RvcC1vcGFjaXR5PSIwIi8+CjxzdG9wIG9mZnNldD0iMSIgc3RvcC1jb2xvcj0iIzNBODZGRiIgc3RvcC1vcGFjaXR5PSIwLjUiLz4KPC9saW5lYXJHcmFkaWVudD4KPGNsaXBQYXRoIGlkPSJjbGlwMF8zNDQyXzkxNzkiPgo8cmVjdCB3aWR0aD0iMTkyMCIgaGVpZ2h0PSI2NCIgZmlsbD0id2hpdGUiLz4KPC9jbGlwUGF0aD4KPC9kZWZzPgo8L3N2Zz4K";
+export {
+  I as default
+};