@labelbee/lb-annotation 1.14.0-alpha.2 → 1.14.0-alpha.3

package/es/core/toolOperation/checkOperation.js

@@ -1 +1,206 @@
-import{cloneDeep as L}from"lodash";import w from"../../utils/tool/CommonToolUtils.js";import R from"../../utils/tool/RectUtils.js";import D from"../../utils/tool/TagUtils.js";import{DEFAULT_TEXT_OFFSET as I}from"../../constant/annotation.js";import{EToolName as c}from"../../constant/tool.js";import S from"../../utils/tool/AttributeUtils.js";import f from"../../utils/tool/AxisUtils.js";import d from"../../utils/tool/DrawUtils.js";import O from"../../utils/tool/StyleUtils.js";import{BasicToolOperation as U}from"./basicToolOperation.js";var y=Object.defineProperty,P=Object.getOwnPropertySymbols,C=Object.prototype.hasOwnProperty,j=Object.prototype.propertyIsEnumerable,g=(a,i,t)=>i in a?y(a,i,{enumerable:!0,configurable:!0,writable:!0,value:t}):a[i]=t,_=(a,i)=>{for(var t in i||(i={}))C.call(i,t)&&g(a,t,i[t]);if(P)for(var t of P(i))j.call(i,t)&&g(a,t,i[t]);return a};const b={x:8,y:26},x=2;class E extends U{constructor(i){super(i);this.getHoverRectID=t=>{var o,e;const s=this.getCoordinateUnderZoom(t),l=(e=(o=this.resultList.find(r=>r.toolName===c.Rect))==null?void 0:o.result)!=null?e:[];if(l.length>0){const r=l.filter(n=>R.isInRect(s,n,x,this.zoom));if(r.length===0)return"";if(r.length===1)return r[0].id;if(r.length>1)return r.map(h=>({size:h.width*h.height,id:h.id})).sort((h,u)=>h.size-u.size)[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(!0),this.forbidOperation=!0}onMouseDown(i){var t,o;if(super.onMouseDown(i)||this.forbidMouseOperation||!this.imgInfo)return!0;const e=this.mouseHoverID,s=(o=(t=this.resultList.find(l=>l.toolName===c.Rect))==null?void 0:t.result)!=null?o:[];if(i.button===0){let l=[e],r=!0;e&&s.find(n=>n.id===e&&(n==null?void 0:n.isSelected))&&(r=!1),e||(l=s.map(n=>n.id),r=!1),this.emit("setSelectedID",l,r),this.render()}}updateRotate(){}onMouseMove(i){if(super.onMouseMove(i)||this.forbidMouseOperation||!this.imgInfo)return;const t=this.mouseHoverID,o=this.getHoverRectID(i);if(t!==o){this.mouseHoverID=o;let e=[o];o||(e=[]),this.emit("setHoverID",e),this.render()}}setResult(i){this.resultList=L(i),this.render()}drawPolygon(i,t){i==null||i.forEach(o=>{var e,s,l,r,n;const h=this.getColor(o.attribute,t),u=O.getStrokeAndFill(h,o.valid);let v=(s=(e=this.style)==null?void 0:e.width)!=null?s:2;this.hoverID.includes(o.id)?(v=4,d.drawPolygonWithFillAndLine(this.canvas,f.changePointListByZoom(o.pointList,this.zoom,this.currentPos),{fillColor:u.fill,strokeColor:u.stroke,pointColor:"white",thickness:v,lineCap:"round",isClose:!0,lineType:t==null?void 0:t.lineType})):d.drawPolygon(this.canvas,f.changePointListByZoom(o.pointList,this.zoom,this.currentPos),{color:u.fill,lineType:(l=this.config)==null?void 0:l.lineType,thickness:v,isClose:!0});let m=`${(n=S.getAttributeShowText(o.attribute,(r=t==null?void 0:t.attributeList)!=null?r:[]))!=null?n:""}`;(t==null?void 0:t.isShowOrder)&&(o==null?void 0:o.order)>0&&(m=`${o.order} ${m}`),d.drawText(this.canvas,f.changePointByZoom(o.pointList[0],this.zoom,this.currentPos),m,_({color:u.stroke},I));const p=f.changePointListByZoom(o.pointList||[],this.zoom,this.currentPos),T=p[p.length-1];d.drawText(this.canvas,{x:T.x+b.x,y:T.y+b.y},o==null?void 0:o.textAttribute,_({color:u.stroke},I))})}drawRect(i,t){i==null||i.forEach(o=>{let e=1;this.hoverID.includes(o.id)&&(e=3);const s=this.getColor(o.attribute,t),l=f.changeRectByZoom(o,this.zoom,this.currentPos);d.drawRect(this.canvas,l,{color:(o==null?void 0:o.valid)?s.valid.stroke:s.invalid.stroke,thickness:e}),this.fillID.includes(o.id)&&d.drawRectWithFill(this.canvas,l,{color:(o==null?void 0:o.valid)?s.valid.fill:s.invalid.fill})})}drawTag(i,t){const o=i.reduce((e,s)=>[...e,...(t==null?void 0:t.inputList)?D.getTagNameList(s.result,t.inputList):D.getTagnameListWithoutConfig(s.result)],[]);d.drawTag(this.canvas,o)}setHoverID(i){this.hoverID=i,this.render()}setFillID(i){this.fillID=i,this.render()}render(){var i;super.render(),(i=this.resultList)==null||i.forEach(t=>{switch(t==null?void 0:t.toolName){case c.Rect:this.drawRect(t.result,w.jsonParser(t.config));break;case c.Polygon:this.drawPolygon(t.result,w.jsonParser(t.config));break;case c.Tag:this.drawTag(t.result,w.jsonParser(t.config));break}})}exportData(){return[[],{}]}}export{E as default};
+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 };