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

package/dist/core/toolOperation/checkOperation.js

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