@labelbee/lb-annotation 1.8.0 → 1.8.1-alpha.0

package/es/core/toolOperation/checkOperation.js

@@ -0,0 +1,206 @@
+import { cloneDeep } from 'lodash';
+import CommonToolUtils from '../../utils/tool/CommonToolUtils.js';
+import RectUtils from '../../utils/tool/RectUtils.js';
+import TagUtil from '../../utils/tool/TagUtils.js';
+import { DEFAULT_TEXT_OFFSET } from '../../constant/annotation.js';
+import { EToolName } from '../../constant/tool.js';
+import AttributeUtils from '../../utils/tool/AttributeUtils.js';
+import AxisUtils from '../../utils/tool/AxisUtils.js';
+import DrawUtils from '../../utils/tool/DrawUtils.js';
+import StyleUtils from '../../utils/tool/StyleUtils.js';
+import { BasicToolOperation } from './basicToolOperation.js';
+
+var __defProp = Object.defineProperty;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, {enumerable: true, configurable: true, writable: true, value}) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+const TEXT_ATTRIBUTE_OFFSET = {
+  x: 8,
+  y: 26
+};
+const newScope = 2;
+class CheckOperation extends BasicToolOperation {
+  constructor(props) {
+    super(props);
+    this.getHoverRectID = (e) => {
+      var _a, _b;
+      const coordinate = this.getCoordinateUnderZoom(e);
+      const currentShowList = (_b = (_a = this.resultList.find((v) => v.toolName === EToolName.Rect)) == null ? void 0 : _a.result) != null ? _b : [];
+      if (currentShowList.length > 0) {
+        const hoverList = currentShowList.filter((rect) => RectUtils.isInRect(coordinate, rect, newScope, this.zoom));
+        if (hoverList.length === 0) {
+          return "";
+        }
+        if (hoverList.length === 1) {
+          return hoverList[0].id;
+        }
+        if (hoverList.length > 1) {
+          const rectSizeList = hoverList.map((rect) => ({size: rect.width * rect.height, id: rect.id})).sort((a, b) => a.size - b.size);
+          return rectSizeList[0].id;
+        }
+      }
+      return "";
+    };
+    this.resultList = [];
+    this.hoverID = [];
+    this.fillID = [];
+    this.render = this.render.bind(this);
+    this.drawPolygon = this.drawPolygon.bind(this);
+    this.drawRect = this.drawRect.bind(this);
+    this.drawTag = this.drawTag.bind(this);
+    this.setShowDefaultCursor(true);
+    this.forbidOperation = true;
+  }
+  onMouseDown(e) {
+    var _a, _b;
+    if (super.onMouseDown(e) || this.forbidMouseOperation || !this.imgInfo) {
+      return true;
+    }
+    const newMouseSelectedID = this.mouseHoverID;
+    const currentShowList = (_b = (_a = this.resultList.find((v) => v.toolName === EToolName.Rect)) == null ? void 0 : _a.result) != null ? _b : [];
+    if (e.button === 0) {
+      let selectedID = [newMouseSelectedID];
+      let isShow = true;
+      if (newMouseSelectedID && currentShowList.find((rect) => rect.id === newMouseSelectedID && (rect == null ? void 0 : rect.isSelected))) {
+        isShow = false;
+      }
+      if (!newMouseSelectedID) {
+        selectedID = currentShowList.map((rect) => rect.id);
+        isShow = false;
+      }
+      this.emit("setSelectedID", selectedID, isShow);
+      this.render();
+    }
+  }
+  updateRotate() {
+  }
+  onMouseMove(e) {
+    if (super.onMouseMove(e) || this.forbidMouseOperation || !this.imgInfo) {
+      return;
+    }
+    const oldMouseHoverID = this.mouseHoverID;
+    const newMouseHoverID = this.getHoverRectID(e);
+    if (oldMouseHoverID !== newMouseHoverID) {
+      this.mouseHoverID = newMouseHoverID;
+      let hoverID = [newMouseHoverID];
+      if (!newMouseHoverID) {
+        hoverID = [];
+      }
+      this.emit("setHoverID", hoverID);
+      this.render();
+    }
+  }
+  setResult(result) {
+    this.resultList = cloneDeep(result);
+    this.render();
+  }
+  drawPolygon(polygonList, config) {
+    polygonList == null ? void 0 : polygonList.forEach((polygon) => {
+      var _a, _b, _c, _d, _e;
+      const toolColor = this.getColor(polygon.attribute, config);
+      const toolData = StyleUtils.getStrokeAndFill(toolColor, polygon.valid);
+      let thickness = (_b = (_a = this.style) == null ? void 0 : _a.width) != null ? _b : 2;
+      if (this.hoverID.includes(polygon.id)) {
+        thickness = 4;
+        DrawUtils.drawPolygonWithFillAndLine(this.canvas, AxisUtils.changePointListByZoom(polygon.pointList, this.zoom, this.currentPos), {
+          fillColor: toolData.fill,
+          strokeColor: toolData.stroke,
+          pointColor: "white",
+          thickness,
+          lineCap: "round",
+          isClose: true,
+          lineType: config == null ? void 0 : config.lineType
+        });
+      } else {
+        DrawUtils.drawPolygon(this.canvas, AxisUtils.changePointListByZoom(polygon.pointList, this.zoom, this.currentPos), {
+          color: toolData.fill,
+          lineType: (_c = this.config) == null ? void 0 : _c.lineType,
+          thickness,
+          isClose: true
+        });
+      }
+      let showText = `${(_e = AttributeUtils.getAttributeShowText(polygon.attribute, (_d = config == null ? void 0 : config.attributeList) != null ? _d : [])) != null ? _e : ""}`;
+      if ((config == null ? void 0 : config.isShowOrder) && (polygon == null ? void 0 : polygon.order) > 0) {
+        showText = `${polygon.order} ${showText}`;
+      }
+      DrawUtils.drawText(this.canvas, AxisUtils.changePointByZoom(polygon.pointList[0], this.zoom, this.currentPos), showText, __spreadValues({
+        color: toolData.stroke
+      }, DEFAULT_TEXT_OFFSET));
+      const transformPointList = AxisUtils.changePointListByZoom(polygon.pointList || [], this.zoom, this.currentPos);
+      const endPoint = transformPointList[transformPointList.length - 1];
+      DrawUtils.drawText(this.canvas, {x: endPoint.x + TEXT_ATTRIBUTE_OFFSET.x, y: endPoint.y + TEXT_ATTRIBUTE_OFFSET.y}, polygon == null ? void 0 : polygon.textAttribute, __spreadValues({
+        color: toolData.stroke
+      }, DEFAULT_TEXT_OFFSET));
+    });
+  }
+  drawRect(rectList, config) {
+    rectList == null ? void 0 : rectList.forEach((rect) => {
+      let thickness = 1;
+      if (this.hoverID.includes(rect.id)) {
+        thickness = 3;
+      }
+      const toolColor = this.getColor(rect.attribute, config);
+      const renderRect = AxisUtils.changeRectByZoom(rect, this.zoom, this.currentPos);
+      DrawUtils.drawRect(this.canvas, renderRect, {
+        color: (rect == null ? void 0 : rect.valid) ? toolColor.valid.stroke : toolColor.invalid.stroke,
+        thickness
+      });
+      if (this.fillID.includes(rect.id)) {
+        DrawUtils.drawRectWithFill(this.canvas, renderRect, {
+          color: (rect == null ? void 0 : rect.valid) ? toolColor.valid.fill : toolColor.invalid.fill
+        });
+      }
+    });
+  }
+  drawTag(tagList, config) {
+    const tagInfoList = tagList.reduce((acc, cur) => {
+      return [
+        ...acc,
+        ...(config == null ? void 0 : config.inputList) ? TagUtil.getTagNameList(cur.result, config.inputList) : TagUtil.getTagnameListWithoutConfig(cur.result)
+      ];
+    }, []);
+    DrawUtils.drawTag(this.canvas, tagInfoList);
+  }
+  setHoverID(hoverID) {
+    this.hoverID = hoverID;
+    this.render();
+  }
+  setFillID(fillID) {
+    this.fillID = fillID;
+    this.render();
+  }
+  render() {
+    var _a;
+    super.render();
+    (_a = this.resultList) == null ? void 0 : _a.forEach((item) => {
+      switch (item == null ? void 0 : item.toolName) {
+        case EToolName.Rect:
+          this.drawRect(item.result, CommonToolUtils.jsonParser(item.config));
+          break;
+        case EToolName.Polygon:
+          this.drawPolygon(item.result, CommonToolUtils.jsonParser(item.config));
+          break;
+        case EToolName.Tag:
+          this.drawTag(item.result, CommonToolUtils.jsonParser(item.config));
+          break;
+      }
+    });
+  }
+  exportData() {
+    return [[], {}];
+  }
+}
+
+export { CheckOperation as default };

package/es/core/toolOperation/eventListener.js

@@ -0,0 +1,35 @@
+class EventListener {
+  constructor() {
+    this._events = new Map();
+  }
+  on(eventName, callback) {
+    const existEvents = this._events.get(eventName) || [];
+    if (!existEvents.some((fn) => fn === callback)) {
+      this._events.set(eventName, existEvents.concat(callback));
+    }
+  }
+  singleOn(eventName, callback) {
+    this._events.set(eventName, [callback]);
+  }
+  emit(eventName, ...args) {
+    const listener = this._events.get(eventName);
+    if (listener) {
+      listener.forEach((fn) => {
+        if (fn) {
+          fn(...args);
+        }
+      });
+    }
+  }
+  unbind(eventName, callback) {
+    const existEvents = this._events.get(eventName);
+    if (existEvents) {
+      this._events.set(eventName, existEvents.filter((fn) => fn !== callback));
+    }
+  }
+  unbindAll(eventName) {
+    this._events.delete(eventName);
+  }
+}
+
+export { EventListener as default };

package/es/core/toolOperation/measureOperation.js

@@ -0,0 +1,39 @@
+import { RectOperation } from './rectOperation.js';
+
+var __defProp = Object.defineProperty;
+var __defProps = Object.defineProperties;
+var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, {enumerable: true, configurable: true, writable: true, value}) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
+const config = {
+  textConfigurable: false,
+  attributeConfigurable: true,
+  attributeList: []
+};
+class MeasureOperation extends RectOperation {
+  constructor(props) {
+    super(__spreadProps(__spreadValues({}, props), {config: JSON.stringify(config)}));
+  }
+  setSelectedIdAfterAddingDrawingRect() {
+    if (!this.drawingRect) {
+      return;
+    }
+    this.setSelectedRectID(this.drawingRect.id);
+  }
+}
+
+export { MeasureOperation as default };

package/es/core/toolOperation/pointCloud2dOperation.js

@@ -0,0 +1 @@
+import{ESortDirection as v}from"../../constant/annotation.js";import{EPolygonPattern as u}from"../../constant/tool.js";import c from"../../utils/tool/AxisUtils.js";import m from"../../utils/tool/CommonToolUtils.js";import d from"../../utils/tool/DrawUtils.js";import g from"../../utils/tool/PolygonUtils.js";import p from"../../utils/tool/StyleUtils.js";import I from"./polygonOperation.js";var _=Object.defineProperty,S=Object.defineProperties,L=Object.getOwnPropertyDescriptors,f=Object.getOwnPropertySymbols,w=Object.prototype.hasOwnProperty,C=Object.prototype.propertyIsEnumerable,P=(s,t,e)=>t in s?_(s,t,{enumerable:!0,configurable:!0,writable:!0,value:e}):s[t]=e,y=(s,t)=>{for(var e in t||(t={}))w.call(t,e)&&P(s,e,t[e]);if(f)for(var e of f(t))C.call(t,e)&&P(s,e,t[e]);return s},D=(s,t)=>S(s,L(t));class b extends I{constructor(t){super(t);this.selectedIDs=[],this.rightMouseUp=i=>{if(this.drawingPointList.length>0){this.addDrawingPointToPolygonList();return}i.ctrlKey&&this.hoverID?this.emit("addSelectedIDs",this.hoverID):this.emit("setSelectedIDs",this.hoverID)},this.onKeyDown=()=>{},this.onKeyUp=()=>{},this.renderSingleSelectedPolygon=i=>{var r;if(this.selectedPolygons){const n=this.getColor(i.attribute),a=p.getStrokeAndFill(n,i.valid,{isSelected:!0}),l=c.changePointListByZoom(i.pointList,this.zoom,this.currentPos);d.drawSelectedPolygonWithFillAndLine(this.canvas,l,{fillColor:"transparent",strokeColor:a.stroke,pointColor:"white",thickness:2,lineCap:"round",isClose:!0,lineType:(r=this.config)==null?void 0:r.lineType}),i.isRect===!0&&this.showDirectionLine===!0&&this.renderRectPolygonDirection(l)}};var e,o;this.showDirectionLine=(e=t.showDirectionLine)!=null?e:!0,this.forbidAddNew=(o=t.forbidAddNew)!=null?o:!1}get getSelectedIDs(){return this.selectedIDs}setSelectedIDs(t){this.selectedIDs=t,this.setSelectedID(this.selectedIDs.length===1?this.selectedIDs[0]:""),this.render()}deleteSelectedID(){super.deleteSelectedID(),this.selectedIDs=[],this.emit("deleteSelectedIDs")}get selectedPolygons(){return g.getPolygonByIDs(this.polygonList,this.selectedIDs)}getLineColor(){return"rgba(0, 255, 255, 0.5)"}renderStaticPolygon(){var t;this.isHidden===!1&&((t=this.polygonList)==null||t.forEach(e=>{var o,i,r;if([...this.selectedIDs,this.editPolygonID].includes(e.id))return;const{attribute:n}=e,a=this.getColor(n),l=p.getStrokeAndFill(a,e.valid),h=c.changePointListByZoom(e.pointList||[],this.zoom,this.currentPos);d.drawPolygonWithFillAndLine(this.canvas,h,{fillColor:"transparent",strokeColor:l.stroke,pointColor:"white",thickness:(i=(o=this.style)==null?void 0:o.width)!=null?i:2,lineCap:"round",isClose:!0,lineType:(r=this.config)==null?void 0:r.lineType}),e.isRect===!0&&this.showDirectionLine===!0&&this.renderRectPolygonDirection(h)}))}renderSelectedPolygon(){var t;(t=this.selectedPolygons)==null||t.forEach(e=>{this.renderSingleSelectedPolygon(e)})}renderRectPolygonDirection(t){t.length<2||d.drawLine(this.canvas,t[0],t[1],{color:"red",thickness:3})}get currentPolygonListByPattern(){return this.polygonList.filter(t=>this.pattern===u.Rect?t.isRect===!0:this.pattern===u.Normal?t.isRect!==!0:!0)}getHoverID(t){var e;const o=this.getCoordinateUnderZoom(t),r=this.currentPolygonListByPattern.map(n=>D(y({},n),{pointList:c.changePointListByZoom(n.pointList,this.zoom)}));return g.getHoverPolygonID(o,r,10,(e=this.config)==null?void 0:e.lineType)}switchToNextPolygon(t=v.ascend){if(this.drawingPointList.length>0)return;const e=this.currentPolygonListByPattern.map(i=>{var r,n,a,l;return D(y({},i),{x:(n=(r=i.pointList[0])==null?void 0:r.x)!=null?n:0,y:(l=(a=i.pointList[0])==null?void 0:a.y)!=null?l:0})}),o=m.getNextSelectedRectID(e,t,this.selectedID);if(o)return this.setSelectedIDs([o.id]),this.render(),[o.id]}setSelectedIdAfterAddingDrawing(t){this.drawingPointList.length!==0&&this.setSelectedID(t)}setSelectedID(t){var e,o;const i=this.selectedID;t!==i&&i&&((e=this._textAttributInstance)==null||e.changeSelected()),t||(o=this._textAttributInstance)==null||o.clearTextAttribute(),this.selectedID=t,this.render()}addPointInDrawing(t){this.forbidAddNew||super.addPointInDrawing(t)}setCanvasSize(t){var e,o;const i=this.pixelRatio;this.size=t,this.setImgInfo(t),this.updateCanvasBasicStyle(this.basicCanvas,t,0),this.updateCanvasBasicStyle(this.canvas,t,10),(e=this.ctx)==null||e.scale(i,i),(o=this.basicCtx)==null||o.scale(i,i),this.initImgPos(),this.renderBasicCanvas(),this.render()}}export{b as default};