（H5 web CAD sdk）在线CAD如何实现图形识别功能

MxDraw · 发表于 2024-9-27 11:47:29

本帖最后由 MxDraw 于 2024-9-27 11:50 编辑

前言

CAD图形识别功能可帮助用户快速识别和提取CAD图纸中的各种图形，从而加速设计过程。可应用在识别与分类阶段，自动识别图纸中的不同元素，通过特定特征进行区分，减少了手动分类的工作量；也可应用在数量统计阶段，统计图纸中各种构件的数量用于预算；还可运用在图纸定位和应用阶段，快速定位图纸上的特定元素，便于快速查找和修改。

mxcad 为用户提供了图形识别功能和API，用户可根据自身需求对该功能进行拓展或二次开发，更多开发文档关注公众号：梦想云图网页CAD。

图形识别步骤

1. 打开mxcad在线示例demo：https://demo.mxdraw3d.com:3000/mxcad/。

2. 点击【工具(A)】菜单栏，找到图形识别功能，选择【图形识别】按钮，如下图：

3. 根据命令行提示，点击鼠标左键点选图元或拉框选择需要识别的图形（图形识别功能目前支持识别直线类、多段线类、弧线类、圆类，以及块这五类图元），如下图：

4. 点击右键结束选择后，在弹出的图形识别框内设置查找图形详情，设置图形名称可方便后续查找已识别图形；设置重新选择图形时，会覆盖当前选中图形；设置区域查找时，需要用户框选查找范围，若未框选范围则会默认为全图纸查找图形，如下图：

5. 在图形识别框内点击【开始识别】按钮，开始在目标范围内查找图形，并将查找结果展示在图形识别列表中。点击图形识别列表下的坐标对应的【查看】按钮会自动定位到对应的图形位置，并圈选出目标图形，如下图：

6. 当识别多个图形后，可点击目标图形统计表格对应的操作列中的【详情】按钮查看识别详情，图形坐标列表将转换为该目标图形的识别结果，如下图：

识别注意事项：

如果需要识别的图形较为复杂，为保证识别速度与精度，我们需尽可能查找图形中的特殊部分而不是选取整个图形（图形对象过多时可能会导致卡顿，影响用户使用效率）。
图形识别支持区域选择，用户可根据自身需求精确定位图形筛选区域。

功能开发

mxcad 图形识别功能中运用的核心思想是通过[McDbEntity]实体中每个实体的特征在图纸中进行查找，查找的实体中包括了[直线类]、[多段线类]、[弧线类]、[圆类]，以及[块]。

下列为基于mxcad 封装的图形识别类 SearchSamePattern 用户可根据该类的使用方法利用 mxcad 进行二次开发，代码如下：

import {
DxfCode, McCmColor, McDbArc, McDbBlockReference, McDbCircle, McDbLine, McDbPolyline, McDbWipeout, McGePoint3d,
McGePoint3dArray, McGeVector3d, MxCADResbuf, MxCADSelectionSet, MxCADUtility, MxCpp
} from "mxcad";
import { MxFun } from "mxdraw";
export class SearchSamePattern {
/** 识别图元数 */
private objectCount: number = 0;
/** 选择图形集合 Map<className,[Ent]>*/
private selectObjects: any;
private dTol = 0.000001;
/**
* pt1:左下角
* pt2:右上角
* pt3:左上角
* pt4:右下角
*/
private pt1: McGePoint3d;
private pt2: McGePoint3d;
private pt3: McGePoint3d;
private pt4: McGePoint3d;//包围盒大小
private pl: McDbPolyline;
/** 识别图形类名集合 [string] */
private classNames: any
/** 识别图形图片地址 */
private patternImgUrl: String
/** 识别对象详情 Map<ClassName,[{}]>*/
private objectsDetails: any = {};
// 下列为识别结果数据
/** 识别后的图形位置数组集合 */;
private graphBoxsPos: any[] = [];
private formPt: McGePoint3d;
private toPt: McGePoint3d;
/** 点精度值 */
private dTolPt: any;
/** 识别后图形列表 */
private patternList: any = {};
/** 选择的图形集合 */
public setSelectset(ss: MxCADSelectionSet): Promise<String> {
if (!ss?.count()) return;
const mxcad = MxCpp.getCurrentMxCAD();
// 获取当前图纸的空间块表记录包围盒
const { maxPt, minPt } = mxcad.getDatabase().currentSpace.getBoundingBox();
const currentDrawOrder = mxcad.getDatabase().currentSpace.getMinMaxDrawOrder()
const n = MxFun.screenCoordLong2Doc(Math.abs(maxPt.x - minPt.x));
minPt.x -= n; minPt.y -= n; maxPt.x += n; maxPt.y += n;
let points = new McGePoint3dArray();
points.append(minPt);
points.append(new McGePoint3d(minPt.x, maxPt.y, 0));
points.append(new McGePoint3d(maxPt.x, maxPt.y, 0));
points.append(new McGePoint3d(maxPt.x, minPt.y, 0));
this.objectCount = ss.count();
// 获取遮罩层drawOrder
let wipeout = new McDbWipeout
wipeout.setVertices(points);
const wipeoutId = mxcad.drawEntity(wipeout);
const e = wipeoutId.getMcDbEntity();
e.drawOrder = currentDrawOrder.maxDrawOrder + 1;
this.selectObjects = {};
const oldDrawOder = [];
ss.forEach(id => {
const ent = id.getMcDbEntity();
/** 记录识别图形的信息 */
if (!this.selectObjects[ent.objectName]) this.selectObjects[ent.objectName] = [];
this.selectObjects[ent.objectName].push(ent);
if (ent.objectName === 'McDbLine') {
if (!this.objectsDetails['McDbLine']) this.objectsDetails['McDbLine'] = [];
this.objectsDetails['McDbLine'].push({ entity: ent, length: (ent as McDbLine).getLength().val });
} else if (ent.objectName === 'McDbArc') {
if (!this.objectsDetails['McDbArc']) this.objectsDetails['McDbArc'] = [];
const length = (ent as McDbArc).getLength().val;
const { bugle } = this.getBugle((ent as McDbArc), length)
this.objectsDetails['McDbArc'].push({ entity: ent, length, bugle });
} else if (ent.objectName === 'McDbCircle') {
if (!this.objectsDetails['McDbCircle']) this.objectsDetails['McDbCircle'] = [];
const radius = (ent as McDbCircle).radius;
this.objectsDetails['McDbCircle'].push({ entity: ent, radius });
} else if (ent.objectName === 'McDbBlockReference') {
if (!this.objectsDetails['McDbBlockReference']) this.objectsDetails['McDbBlockReference'] = [];
const blkName = (ent as McDbBlockReference).blockName;
this.objectsDetails['McDbBlockReference'].push({ entity: ent, blkName });
} else if (ent.objectName === 'McDbPolyline') {
if (!this.objectsDetails['McDbPolyline']) this.objectsDetails['McDbPolyline'] = [];
const length = (ent as McDbPolyline).getLength().val;
this.objectsDetails['McDbPolyline'].push({ entity: ent, length });
}
/** 渲染顺序 */
oldDrawOder.push({ obj: ent, drawOrder: ent.drawOrder });
ent.drawOrder = e.drawOrder + 1;
const { maxPt, minPt } = ent.getBoundingBox();
if (!this.pt1 || !this.pt2) {
this.pt1 = minPt;
this.pt2 = maxPt;
} else {
if (minPt.x < this.pt1.x) this.pt1.x = minPt.x;
if (minPt.y < this.pt1.y) this.pt1.y = minPt.y;
if (maxPt.x > this.pt2.x) this.pt2.x = maxPt.x;
if (maxPt.y > this.pt2.y) this.pt2.y = maxPt.y;
}
});
this.classNames = Object.keys(this.selectObjects);
/**
* 计算包围盒其他两个角点
*/
this.pt3 = new McGePoint3d(this.pt1.x, this.pt2.y);
this.pt4 = new McGePoint3d(this.pt2.x, this.pt1.y);
/** 绘制识别图形包围盒 */
this.pl = new McDbPolyline()
this.pl.addVertexAt(this.pt1);
this.pl.addVertexAt(this.pt3);
this.pl.addVertexAt(this.pt2);
this.pl.addVertexAt(this.pt4);
this.pl.isClosed = true;
const num = MxFun.screenCoordLong2Doc(10);
this.pt1.x -= num; this.pt1.y -= num; this.pt2.x += num; this.pt2.y += num;
// 生成预览图
let w = Math.abs(this.pt1.x - this.pt2.x);
let h = Math.abs(this.pt1.y - this.pt2.y);
if (w < 1 || h < 1) return;
let jpg_width = 300;
let jpg_height = 240;
return new Promise<String>((resolve, reject) => {
MxFun.getCurrentDraw().createCanvasImageData(
(imageData: String) => {
this.patternImgUrl = imageData;
// 恢复绘制顺序
oldDrawOder.forEach(item => {
item.obj.drawOrder = item.drawOrder;
})
wipeoutId.erase();
resolve(imageData)
},
{
width: jpg_width, // 图片宽
height: jpg_height, // 图片高
range_pt1: this.pt1.toVector3(), // 截图范围角点1
range_pt2: this.pt2.toVector3(), // 截图范围角点2
}
);
});
};
// 开始识别图形
/**
* 1.查找初始包围盒
* 2.筛选符合条件的包围盒
* 3.比较包围盒内的每一个图元与需要识别的图元是否匹配
*/
public identifyGraphics(corner1?: McGePoint3d, corner2?: McGePoint3d): any {
/** 记录图形缩略图 */
this.patternList.patternImgUrl = this.patternImgUrl;
/**初始化图形位置列表 */
this.graphBoxsPos.length = 0;
const mxcad = MxCpp.getCurrentMxCAD();
this.dTolPt = mxcad.mxdraw.viewCoordLong2Cad(0.5);// 设置精度值
if (!this.classNames) return;
const array = this.classNames.map(name => this.selectObjects[name].length);
const minIndex = array.findIndex(element => element === Math.min(...array));
const str = this.classNames[minIndex];
/** 识别块 */
if (str === 'McDbBlockReference') {
const refBlk = this.selectObjects['McDbBlockReference'][0].clone() as McDbBlockReference;
const blkName = refBlk.blockName; // 图块名
const { position } = this.getBlkMidPt(refBlk);
this.formPt = position;
const rotate = refBlk.rotation;
// 设置图块过滤器
const filter = new MxCADResbuf([DxfCode.kEntityType, "INSERT"])
const ss = new MxCADSelectionSet()
if (corner1 && corner2) {
// 局部选择
ss.crossingSelect(corner1.x, corner1.y, corner2.x, corner2.y, filter);
} else {
// 全局选择
ss.allSelect(filter);
};
// 筛选出符合条件的块
const boxs: any[] = [];
ss.forEach(id => {
const _blk = id.getMcDbEntity().clone() as McDbBlockReference;
/** 块名相等=》计算旋转角度=》计算包围盒大小 */
if (blkName === _blk.blockName) {
const _rotate = _blk.rotation;
const angle = rotate >= 0 ? _rotate - rotate : rotate - _rotate;
const { position: _position, minPt, maxPt } = this.getBlkMidPt(_blk);
if (this.classNames.length === 1) {
this.toPt = _position;
this.identificationBBox(minPt, maxPt, 0)
} else {
// 平移旋转初始包围盒。计算出最新包围盒的大小
const _pl = this.pl.clone() as McDbPolyline;
_pl.move(this.formPt, _position);
_pl.rotate(_position, angle);
let x = [], y = [];
for (let i = 0; i < _pl.numVerts(); i++) {
const pt = _pl.getPointAt(i).val;
x.push(pt.x);
y.push(pt.y);
}
const _minPt = new McGePoint3d(Math.min(...x) - 1, Math.min(...y) - 1);
const _maxPt = new McGePoint3d(Math.max(...x) + 1, Math.max(...y) + 1);
const res = boxs.filter(item => {
let ret1 = _minPt.distanceTo(item.pt1) < this.dTolPt && _maxPt.distanceTo(item.pt2) < this.dTolPt
let ret2 = _minPt.distanceTo(item.pt2) < this.dTolPt && _maxPt.distanceTo(item.pt1) < this.dTolPt
return ret1 || ret2
})
if (!res.length) {
boxs.push({ pt1: _minPt, pt2: _maxPt });
this.toPt = _position;
// MxCpp.getCurrentMxCAD().drawLine(_maxPt.x, _maxPt.y, _minPt.x, _minPt.y);
this.identificationBBox(_minPt, _maxPt, angle);
}
}
}
});
}
/** 识别弧线 */
else if (str === 'McDbArc') {
// 识别直线
const refArc = this.selectObjects['McDbArc'][0].clone() as McDbArc;
const length = refArc.getLength().val; // 弧线线长度
// 圆弧凸度
const { bugle, midPt } = this.getBugle(refArc, length);
const center = refArc.center;
const vec = center.sub(midPt);
// 设置弧线过滤器
const filter = new MxCADResbuf([DxfCode.kEntityType, "ARC"])
const ss = new MxCADSelectionSet()
if (corner1 && corner2) {
// 局部选择
ss.crossingSelect(corner1.x, corner1.y, corner2.x, corner2.y, filter);
} else {
// 全局选择
ss.allSelect(filter);
};
// 筛选出符合条件的弧线
const boxs: any[] = [];
ss.forEach(id => {
const _arc = id.getMcDbEntity().clone() as McDbArc;
const _length = _arc.getLength().val;
const { bugle: _bugle, midPt: _midPt } = this.getBugle(_arc, _length);
/** 筛选出圆弧凸度与长度相同的弧线*/
if (Math.abs(_length - length) < this.dTol && Math.abs(_bugle - bugle) < this.dTol) {
/** 分两种情况
* 1.直接旋转平移得到
* 2.镜像得到=》移动到目标位置+旋转角度+镜像翻转=》比较起始点位置
*/
// 平移+旋转
const _center = _arc.center;
const _vec = _center.sub(_midPt);
const angle = _vec.angleTo2(vec, McGeVector3d.kNegateZAxis);
// 旋转平移初始包围盒。计算出最新包围盒的大小
const _pl = this.pl.clone() as McDbPolyline;
_pl.move(midPt, _midPt);
_pl.rotate(_midPt, angle);
let x = [], y = [];
for (let i = 0; i < _pl.numVerts(); i++) {
const pt = _pl.getPointAt(i).val;
x.push(pt.x);
y.push(pt.y);
}
const _minPt = new McGePoint3d(Math.min(...x) - 1, Math.min(...y) - 1);
const _maxPt = new McGePoint3d(Math.max(...x) + 1, Math.max(...y) + 1);
const res = boxs.filter(item => {
let ret1 = _minPt.distanceTo(item.pt1) < this.dTolPt && _maxPt.distanceTo(item.pt2) < this.dTolPt
let ret2 = _minPt.distanceTo(item.pt2) < this.dTolPt && _maxPt.distanceTo(item.pt1) < this.dTolPt
let ret3 = _midPt.distanceTo(item.midPt) < this.dTolPt
return (ret1 || ret2) && ret3
})
if (!res.length) {
boxs.push({ pt1: _minPt, pt2: _maxPt, midPt: _midPt });
this.toPt = _midPt;
this.formPt = midPt;
this.identificationBBox(_minPt, _maxPt, angle);
}
}
});
}
/** 识别直线 */
else if (str === 'McDbLine') {
// 识别直线
const refLine = this.selectObjects['McDbLine'][0].clone() as McDbLine;
const length = refLine.getLength().val; // 直线长度
const vec = refLine.endPoint.sub(refLine.startPoint); // 直线角度
// 设置直线过滤器
const filter = new MxCADResbuf([DxfCode.kEntityType, "LINE"])
const ss = new MxCADSelectionSet()
if (corner1 && corner2) {
// 局部选择
ss.crossingSelect(corner1.x, corner1.y, corner2.x, corner2.y, filter);
} else {
// 全局选择
ss.allSelect(filter);
};
// 筛选出符合条件的直线
const lineArr: McDbLine[] = [];
ss.forEach(id => {
const line = id.getMcDbEntity().clone() as McDbLine;
const _length = line.getLength().val;
/**
* 1.未旋转
* 2.旋转过
*/
if (Math.abs(_length - length) < this.dTol) {
lineArr.push(line);
}
});
// 筛选出符合条件的直线后，根据包围盒范围再次筛选出符合条件的直线组合
const boxs: any[] = [];
if (lineArr.length > 0) {
lineArr.forEach((item) => {
/**
* 将要识别的图形进行平移+旋转到目标位置
*/
const ptArr = [item.startPoint, item.endPoint];
ptArr.forEach((point, index) => {
const _line = refLine.clone() as McDbLine;
_line.move(refLine.startPoint, point);
const _vec = ptArr[1 - index].sub(point);
const angle = _vec.angleTo2(vec, McGeVector3d.kNegateZAxis);
_line.rotate(point, angle)
const pt = _line.endPoint;
const _pl = this.pl.clone() as McDbPolyline;
_pl.move(refLine.startPoint, point);
_pl.rotate(point, angle);
const _refBoxPt1 = _pl.getPointAt(0).val;
const _refBoxPt2 = _pl.getPointAt(1).val;
const refVec1 = _refBoxPt1.sub(_line.startPoint);
const refVec2 = _refBoxPt2.sub(_line.startPoint);
if (pt.distanceTo(ptArr[1 - index]) < this.dTolPt && refVec1.isEqualTo(_refBoxPt1.sub(point)) && refVec2.isEqualTo(_refBoxPt2.sub(point))) {
let x = [], y = [];
for (let i = 0; i < _pl.numVerts(); i++) {
const pt = _pl.getPointAt(i).val;
x.push(pt.x);
y.push(pt.y);
}
const _minPt = new McGePoint3d(Math.min(...x) - 1, Math.min(...y) - 1);
const _maxPt = new McGePoint3d(Math.max(...x) + 1, Math.max(...y) + 1);
const res = boxs.filter(item => {
let r1 = _minPt.distanceTo(item.pt1) < this.dTolPt && _maxPt.distanceTo(item.pt2) < this.dTolPt
let r2 = _minPt.distanceTo(item.pt2) < this.dTolPt && _maxPt.distanceTo(item.pt1) < this.dTolPt
return r1 || r2
})
if (!res.length) {
boxs.push({ pt1: _minPt, pt2: _maxPt });
this.toPt = point;
this.formPt = refLine.startPoint;
// MxCpp.getCurrentMxCAD().drawLine(_maxPt.x, _maxPt.y, _minPt.x, _minPt.y);
this.identificationBBox(_minPt, _maxPt, angle);
}
}
})
})
};
}
/** 识别圆 */
else if (str === 'McDbCircle') {
const refCricle = this.selectObjects['McDbCircle'][0].clone() as McDbCircle;
const radius = refCricle.radius; // 圆半径
// 设置圆过滤器
const filter = new MxCADResbuf([DxfCode.kEntityType, "CIRCLE"])
const ss = new MxCADSelectionSet()
if (corner1 && corner2) {
// 局部选择
ss.crossingSelect(corner1.x, corner1.y, corner2.x, corner2.y, filter);
} else {
// 全局选择
ss.allSelect(filter);
};
// 筛选出符合条件的圆
ss.forEach(id => {
const _circle = id.getMcDbEntity().clone() as McDbCircle;
const _radius = _circle.radius;
/** 筛选出圆半径相同的圆*/
if (Math.abs(_radius - radius) < this.dTol) {
const _center = _circle.center;
this.toPt = _center;
this.formPt = refCricle.center;
// 平移初始包围盒。计算出最新包围盒的大小
const _pl = this.pl.clone() as McDbPolyline;
_pl.move(this.formPt, this.toPt);
let x = [], y = [];
for (let i = 0; i < _pl.numVerts(); i++) {
const pt = _pl.getPointAt(i).val;
x.push(pt.x);
y.push(pt.y);
}
const _minPt = new McGePoint3d(Math.min(...x) - 1, Math.min(...y) - 1);
const _maxPt = new McGePoint3d(Math.max(...x) + 1, Math.max(...y) + 1);
// MxCpp.getCurrentMxCAD().drawLine(_maxPt.x, _maxPt.y, _minPt.x, _minPt.y);
this.identificationBBox(_minPt, _maxPt, 0);
}
});
}
/** 识别多段线 */
else if (str === 'McDbPolyline') {
/** 长度+顶点数+顶点对应的凸度相等=》旋转+平移后的顶点相等+有凸度线段的中点相等*/
const refPoly = this.selectObjects['McDbPolyline'][0].clone() as McDbPolyline;
const length = refPoly.getLength().val;
const { lengthArr, bugleArr, vertNum, vec, point } = this.getPlDetails(refPoly);
// 设置多段线过滤器
const filter = new MxCADResbuf([DxfCode.kEntityType, "LWPOLYLINE"])
const ss = new MxCADSelectionSet()
if (corner1 && corner2) {
// 局部选择
ss.crossingSelect(corner1.x, corner1.y, corner2.x, corner2.y, filter);
} else {
// 全局选择
ss.allSelect(filter);
};
// 筛选出符合条件的多段线
const boxs: any[] = [];
ss.forEach(id => {
const _poly = id.getMcDbEntity().clone() as McDbPolyline;
const _length = _poly.getLength().val;
/** 筛选长度+顶点数+顶点对应的凸度相同的多段线*/
if (Math.abs(_length - length) < this.dTol && _poly.isClosed === refPoly.isClosed) {
const { lengthArr: _lengthArr, bugleArr: _bugleArr, vertNum: _vertNum, vec: _vec, point: _point } = this.getPlDetails(_poly);
const revBugleArr = _bugleArr.map(item => item === 0 ? item : -item);
// 全等
const r1 = lengthArr.toString() === _lengthArr.toString() && bugleArr.toString() === _bugleArr.toString();
let r2 = false;
if (!r1) {
// 反向
r2 = lengthArr.toString() === [..._lengthArr].reverse().toString() && bugleArr.toString() === [...revBugleArr].reverse().toString();
}
const r3 = vertNum === _vertNum;
// 镜像
const r4 = lengthArr.toString() === _lengthArr.toString() && bugleArr.toString() === [...revBugleArr].reverse().toString();
if ((r1 || r2 || r4) && r3) {
const poly = refPoly.clone() as McDbPolyline;
const angle1 = vec.angleTo2(McGeVector3d.kXAxis, McGeVector3d.kNegateZAxis);
// 如果是反向则方向取反
const angle2 = (r2 ? _vec.negate() : _vec).angleTo2(McGeVector3d.kXAxis, McGeVector3d.kNegateZAxis);
const angle = angle2 - angle1;
poly.move(point, _point);
poly.rotate(_point, angle);
poly.trueColor = new McCmColor(255, 0, 0);
let num = 0;
for (let i = 0; i < vertNum; i++) {
const pt = poly.getPointAt(i).val;
const width = poly.getWidthsAt(i);
const _pt = r2 ? _poly.getPointAt(vertNum - 1 - i).val : _poly.getPointAt(i).val;
const _width = r2 ? _poly.getWidthsAt(vertNum - 1 - i) : _poly.getWidthsAt(i);
if (pt.distanceTo(_pt) < this.dTolPt && width.val1 === _width.val1 && width.val2 === _width.val2) {
num += 1
} else {
// 两个点且反向的情况
if (vertNum === 2 && i === 0) {
const _pt1 = _poly.getPointAt(0).val;
const _pt2 = _poly.getPointAt(1).val;
const pt2 = poly.getPointAt(1).val;
const _width = _poly.getWidthsAt(0);
if (pt2.distanceTo(_pt2) < this.dTolPt && pt.distanceTo(_pt1) < this.dTolPt && width.val1 === _width.val2 && width.val2 === _width.val1) {
num += 2;
}
}
}
};
if (num === vertNum) {
// 旋转平移初始包围盒。计算出最新包围盒的大小
const _pl = this.pl.clone() as McDbPolyline;
_pl.move(point, _point);
_pl.rotate(_point, angle);
let x = [], y = [];
for (let i = 0; i < _pl.numVerts(); i++) {
const pt = _pl.getPointAt(i).val;
x.push(pt.x);
y.push(pt.y);
}
const _minPt = new McGePoint3d(Math.min(...x) - 1, Math.min(...y) - 1);
const _maxPt = new McGePoint3d(Math.max(...x) + 1, Math.max(...y) + 1);
const res = boxs.filter(item => {
let ret1 = _minPt.distanceTo(item.pt1) < this.dTolPt && _maxPt.distanceTo(item.pt2) < this.dTolPt
let ret2 = _minPt.distanceTo(item.pt2) < this.dTolPt && _maxPt.distanceTo(item.pt1) < this.dTolPt
return ret1 || ret2
})
if (!res.length) {
boxs.push({ pt1: _minPt, pt2: _maxPt });
this.toPt = _point;
this.formPt = point;
// MxCpp.getCurrentMxCAD().drawLine(_maxPt.x, _maxPt.y, _minPt.x, _minPt.y);
this.identificationBBox(_minPt, _maxPt, angle);
}
}
}
}
});
}
/** 返回识别列表 */
return this.patternList;
}
// 计算块中心点
private getBlkMidPt = (blkRef: McDbBlockReference) => {
const { minPt, maxPt } = blkRef.getBoundingBox();
const pt = minPt.clone().addvec(maxPt.sub(minPt).normalize().mult(maxPt.distanceTo(minPt) / 2))
return { position: pt, minPt, maxPt }
}
// 计算凸度
private getBugle = (arc: McDbArc, length: number) => {
const midPt = arc.getPointAtDist(length / 2).val;
const bugle = MxCADUtility.calcBulge(arc.getPointAtDist(0).val, midPt, arc.getPointAtDist(length).val).val;
return { bugle, midPt }
}
// 计算多段线的每一段线的长度和凸度
private getPlDetails = (pl: McDbPolyline) => {
const vertNum = pl.numVerts();
const lengthArr: number[] = [];
const bugleArr: number[] = [];
let point1: McGePoint3d, point2: McGePoint3d;
for (let i = 0; i < vertNum; i++) {
if (i < vertNum - 1) {
const p1 = pl.getPointAt(i).val;
const p2 = pl.getPointAt(i + 1).val;
const length = pl.getDistAtPoint(p2).val - pl.getDistAtPoint(p1).val;
const bugle = pl.getBulgeAt(i);
lengthArr.push(Number(length.toFixed(6)));
bugleArr.push(Number(bugle.toFixed(6)));
if (i === 0) {
point1 = p1;
}
if (i === vertNum - 2) {
point2 = p2;
}
}
};
const vec = point2.sub(point1)// 多段线角度
const point = point1.clone().addvec(point2.sub(point1).normalize().mult(point2.distanceTo(point1) / 2))
return { lengthArr, bugleArr, vertNum, vec, point }
}
// 将找到的实体数组与需要识别的实体数组比对
private comparisonPhysical = (objects: any, maxPt: McGePoint3d, minPt: McGePoint3d, angle: number): any => {
const calssNames = Object.keys(this.selectObjects);
if (!calssNames) return;
let num = 0;
if (calssNames.includes('McDbLine')) {
let ents = objects['McDbLine'];
const lineObjects = this.objectsDetails['McDbLine'];
for (let i = 0; i < lineObjects.length; i++) {
const element = lineObjects[i];
let _line = (element.entity as McDbLine).clone() as McDbLine;
_line.move(this.formPt, this.toPt);
_line.rotate(this.toPt, angle);
const indexArr: number[] = [];
const mLineArr = ents.filter((item: McDbLine, index: number) => {
if (Math.abs(element.length - item.getLength().val) < this.dTol) {
let r1 = item.startPoint.distanceTo(_line.startPoint) < this.dTolPt && item.endPoint.distanceTo(_line.endPoint) < this.dTolPt;
let r2 = item.startPoint.distanceTo(_line.endPoint) < this.dTolPt && item.endPoint.distanceTo(_line.startPoint) < this.dTolPt;
if (r1 || r2) indexArr.push(index);
return r1 || r2;
}
});
if (mLineArr.length > 0) {
num += 1;
ents.splice(indexArr[0], 1);
ents = [...ents];
} else {
break;
}
}
};
if (calssNames.includes('McDbArc')) {
let ents = objects['McDbArc'];
const arcObjects = this.objectsDetails['McDbArc'];
for (let i = 0; i < arcObjects.length; i++) {
const element = arcObjects[i];
let _arc = (element.entity as McDbArc).clone() as McDbArc;
_arc.move(this.formPt, this.toPt);
_arc.rotate(this.toPt, angle);
const startPoint = _arc.getPointAtDist(0).val;
const endPoint = _arc.getPointAtDist(element.length).val;
const indexArr: number[] = [];
const mArcArr = ents.filter((item: McDbArc, index: number) => {
const _length = item.getLength().val;
const { bugle: _bugle } = this.getBugle(item, _length)
if (Math.abs(element.length - item.getLength().val) < this.dTol && Math.abs(_bugle - element.bugle) < this.dTol) {
const _startPoint = item.getPointAtDist(0).val;
const _endPoint = item.getPointAtDist(_length).val;
let r1 = startPoint.distanceTo(_startPoint) < this.dTolPt && endPoint.distanceTo(_endPoint) < this.dTolPt;
let r2 = startPoint.distanceTo(_endPoint) < this.dTolPt && endPoint.distanceTo(_startPoint) < this.dTolPt;
if (r1 || r2) {
indexArr.push(index);
}
return r1 || r2;
}
});
if (mArcArr.length > 0) {
num += 1;
ents.splice(indexArr[0], 1);
ents = [...ents];
} else {
break;
}
}
};
if (calssNames.includes('McDbCircle')) {
let ents = objects['McDbCircle'];
const circleObjects = this.objectsDetails['McDbCircle'];
for (let i = 0; i < circleObjects.length; i++) {
const element = circleObjects[i];
let _circle = (element.entity as McDbCircle).clone() as McDbCircle;
_circle.move(this.formPt, this.toPt);
_circle.rotate(this.toPt, angle);
const center = _circle.center;
const radius = _circle.radius;
const indexArr: number[] = [];
const mCiecleArr = ents.filter((item: McDbCircle, index: number) => {
const _center = item.center;
const _radius = item.radius;
const r = Math.abs(_radius - radius) < this.dTol && center.distanceTo(_center) < this.dTolPt;
if (r) {
indexArr.push(index)
}
return r;
});
if (mCiecleArr.length > 0) {
num += 1;
ents.splice(indexArr[0], 1);
ents = [...ents];
} else {
break;
}
}
};
if (calssNames.includes('McDbBlockReference')) {
/** 平移+旋转=>比对包围盒大小+块名字 */
let ents = objects['McDbBlockReference'];
const blkObjects = this.objectsDetails['McDbBlockReference'];
for (let i = 0; i < blkObjects.length; i++) {
const element = blkObjects[i];
let _blk = (element.entity as McDbBlockReference).clone() as McDbBlockReference;
_blk.move(this.formPt, this.toPt);
_blk.rotate(this.toPt, angle);
const blkName = _blk.blockName;
const { minPt: blkMinPt, maxPt: blkMaxPt } = this.getBlkMidPt(_blk);
const indexArr: number[] = [];
const mArcArr = ents.filter((item: McDbBlockReference, index: number) => {
const _blkName = item.blockName;
if (_blkName === blkName) {
if (this.classNames.length === 1) {
indexArr.push(index);
return true
} else {
const { minPt: _blkMinPt, maxPt: _blkMaxPt } = this.getBlkMidPt(item);
let r1 = blkMinPt.distanceTo(_blkMinPt) < this.dTolPt && blkMaxPt.distanceTo(_blkMaxPt) < this.dTolPt;
if (r1) {
indexArr.push(index);
}
return r1
}
}
});
if (mArcArr.length > 0) {
num += 1;
ents.splice(indexArr[0], 1);
ents = [...ents];
} else {
return;
}
}
}
if (calssNames.includes('McDbPolyline')) {
/** 平移+旋转=>比对凸度+点位置 */
let ents = objects['McDbPolyline'];
const plObjects = this.objectsDetails['McDbPolyline'];
for (let i = 0; i < plObjects.length; i++) {
const element = plObjects[i];
let _poly = (element.entity as McDbPolyline).clone() as McDbPolyline;
_poly.move(this.formPt, this.toPt);
_poly.rotate(this.toPt, angle);
const { lengthArr, bugleArr, vertNum } = this.getPlDetails(_poly);
const indexArr: number[] = [];
const mPlArr = ents.filter((item: McDbPolyline, index: number) => {
const _length = item.getLength().val;
if (Math.abs(_length - element.length) < this.dTol && _poly.isClosed === _poly.isClosed) {
const { lengthArr: _lengthArr, bugleArr: _bugleArr, vertNum: _vertNum } = this.getPlDetails(item);
const revBugleArr = _bugleArr.map(item => item === 0 ? item : -item);
const r1 = lengthArr.toString() === _lengthArr.toString() && bugleArr.toString() === _bugleArr.toString();
let r2 = false;
if (!r1) {
r2 = lengthArr.toString() === [..._lengthArr].reverse().toString() && bugleArr.toString() === [...revBugleArr].reverse().toString();
}
const r3 = vertNum === _vertNum;
if ((r1 || r2) && r3) {
let _num = 0;
for (let i = 0; i < vertNum; i++) {
const pt = _poly.getPointAt(i).val;
const width = _poly.getWidthsAt(i);
const _pt = r2 ? item.getPointAt(vertNum - 1 - i).val : item.getPointAt(i).val;
const _width = r2 ? item.getWidthsAt(vertNum - 1 - i) : item.getWidthsAt(i);
if (pt.distanceTo(_pt) < this.dTolPt && width.val1 === _width.val1 && width.val2 === _width.val2) {
_num += 1
} else {
// 两个点且反向的情况
if (vertNum === 2 && i === 0) {
const _pt1 = item.getPointAt(0).val;
const _pt2 = item.getPointAt(1).val;
const _width = item.getWidthsAt(0);
const pt2 = _poly.getPointAt(1).val;
if (pt2.distanceTo(_pt2) < this.dTolPt && pt.distanceTo(_pt1) < this.dTolPt && width.val1 === _width.val2 && width.val2 === _width.val1) {
_num += 2;
}
}
}
};
if (_num === vertNum) {
indexArr.push(index);
}
return _num === vertNum;
}
}
});
if (mPlArr.length > 0) {
num += 1;
ents.splice(indexArr[0], 1);
ents = [...ents];
} else {
break;
}
}
}
if (num === this.objectCount) {
// 匹配成功
const pt = minPt.clone().addvec(maxPt.clone().sub(minPt).normalize().mult(minPt.distanceTo(maxPt) / 2));
const res = this.graphBoxsPos.filter(point => pt.distanceTo(point) < this.dTolPt);
if (!res.length) {
this.graphBoxsPos.push(pt)
}
};
this.patternList.graphBoxsPos = [...this.graphBoxsPos];
}
// 识别包围盒是否符合要求
private identificationBBox = (minPt: McGePoint3d, maxPt: McGePoint3d, angle: number): any => {
const _ss = new MxCADSelectionSet()
_ss.crossingSelect(minPt.x, minPt.y, maxPt.x, maxPt.y);
const boxEntity = {};
_ss.forEach(id => {
const ent = id.getMcDbEntity();
if (!boxEntity[ent.objectName]) boxEntity[ent.objectName] = [];
boxEntity[ent.objectName].push(ent);
});
let r = this.classNames.filter(name => boxEntity[name] && this.selectObjects[name].length <= boxEntity[name].length);
if (r.length === this.classNames.length) {
this.comparisonPhysical(boxEntity, maxPt, minPt, angle);
}
}
};

import { 
    DxfCode, McCmColor, McDbArc, McDbBlockReference, McDbCircle, McDbLine, McDbPolyline, McDbWipeout, McGePoint3d, 
    McGePoint3dArray, McGeVector3d, MxCADResbuf, MxCADSelectionSet, MxCADUtility, MxCpp 
} from "mxcad"; 
import { MxFun } from "mxdraw"; 
export class SearchSamePattern { 
    /** 识别图元数 */ 
    private objectCount: number = 0; 
    /** 选择图形集合 Map<className,[Ent]>*/ 
    private selectObjects: any; 
    private dTol = 0.000001; 
    /** 
     * pt1:左下角 
     * pt2:右上角 
     * pt3:左上角 
     * pt4:右下角 
     */ 
    private pt1: McGePoint3d; 
    private pt2: McGePoint3d; 
    private pt3: McGePoint3d; 
    private pt4: McGePoint3d;//包围盒大小 
    private pl: McDbPolyline; 
    /** 识别图形类名集合 [string] */ 
    private classNames: any 
    /** 识别图形图片地址 */ 
    private patternImgUrl: String 
    /** 识别对象详情  Map<ClassName,[{}]>*/ 
    private objectsDetails: any = {}; 
    // 下列为识别结果数据 
    /** 识别后的图形位置数组集合 */; 
    private graphBoxsPos: any[] = []; 
    private formPt: McGePoint3d; 
    private toPt: McGePoint3d; 
    /** 点精度值 */ 
    private dTolPt: any; 
    /** 识别后图形列表 */ 
    private patternList: any = {}; 
    /** 选择的图形集合 */ 
    public setSelectset(ss: MxCADSelectionSet): Promise<String> { 
        if (!ss?.count()) return; 
        const mxcad = MxCpp.getCurrentMxCAD(); 
        // 获取当前图纸的空间块表记录包围盒 
        const { maxPt, minPt } = mxcad.getDatabase().currentSpace.getBoundingBox(); 
        const currentDrawOrder = mxcad.getDatabase().currentSpace.getMinMaxDrawOrder() 
        const n = MxFun.screenCoordLong2Doc(Math.abs(maxPt.x - minPt.x)); 
        minPt.x -= n; minPt.y -= n; maxPt.x += n; maxPt.y += n; 
        let points = new McGePoint3dArray(); 
        points.append(minPt); 
        points.append(new McGePoint3d(minPt.x, maxPt.y, 0)); 
        points.append(new McGePoint3d(maxPt.x, maxPt.y, 0)); 
        points.append(new McGePoint3d(maxPt.x, minPt.y, 0)); 
        this.objectCount = ss.count(); 
        // 获取遮罩层drawOrder 
        let wipeout = new McDbWipeout 
        wipeout.setVertices(points); 
        const wipeoutId = mxcad.drawEntity(wipeout); 
        const e = wipeoutId.getMcDbEntity(); 
        e.drawOrder = currentDrawOrder.maxDrawOrder + 1; 
        this.selectObjects = {}; 
        const oldDrawOder = []; 
        ss.forEach(id => { 
            const ent = id.getMcDbEntity(); 
            /** 记录识别图形的信息 */ 
            if (!this.selectObjects[ent.objectName]) this.selectObjects[ent.objectName] = []; 
            this.selectObjects[ent.objectName].push(ent); 
            if (ent.objectName === 'McDbLine') { 
                if (!this.objectsDetails['McDbLine']) this.objectsDetails['McDbLine'] = []; 
                this.objectsDetails['McDbLine'].push({ entity: ent, length: (ent as McDbLine).getLength().val }); 
            } else if (ent.objectName === 'McDbArc') { 
                if (!this.objectsDetails['McDbArc']) this.objectsDetails['McDbArc'] = []; 
                const length = (ent as McDbArc).getLength().val; 
                const { bugle } = this.getBugle((ent as McDbArc), length) 
                this.objectsDetails['McDbArc'].push({ entity: ent, length, bugle }); 
            } else if (ent.objectName === 'McDbCircle') { 
                if (!this.objectsDetails['McDbCircle']) this.objectsDetails['McDbCircle'] = []; 
                const radius = (ent as McDbCircle).radius; 
                this.objectsDetails['McDbCircle'].push({ entity: ent, radius }); 
            } else if (ent.objectName === 'McDbBlockReference') { 
                if (!this.objectsDetails['McDbBlockReference']) this.objectsDetails['McDbBlockReference'] = []; 
                const blkName = (ent as McDbBlockReference).blockName; 
                this.objectsDetails['McDbBlockReference'].push({ entity: ent, blkName }); 
            } else if (ent.objectName === 'McDbPolyline') { 
                if (!this.objectsDetails['McDbPolyline']) this.objectsDetails['McDbPolyline'] = []; 
                const length = (ent as McDbPolyline).getLength().val; 
                this.objectsDetails['McDbPolyline'].push({ entity: ent, length }); 
            } 
            /** 渲染顺序 */ 
            oldDrawOder.push({ obj: ent, drawOrder: ent.drawOrder }); 
            ent.drawOrder = e.drawOrder + 1; 
            const { maxPt, minPt } = ent.getBoundingBox(); 
            if (!this.pt1 || !this.pt2) { 
                this.pt1 = minPt; 
                this.pt2 = maxPt; 
            } else { 
                if (minPt.x < this.pt1.x) this.pt1.x = minPt.x; 
                if (minPt.y < this.pt1.y) this.pt1.y = minPt.y; 
                if (maxPt.x > this.pt2.x) this.pt2.x = maxPt.x; 
                if (maxPt.y > this.pt2.y) this.pt2.y = maxPt.y; 
            } 
        }); 
        this.classNames = Object.keys(this.selectObjects); 
        /** 
         * 计算包围盒其他两个角点 
         */ 
        this.pt3 = new McGePoint3d(this.pt1.x, this.pt2.y); 
        this.pt4 = new McGePoint3d(this.pt2.x, this.pt1.y); 
        /** 绘制识别图形包围盒 */ 
        this.pl = new McDbPolyline() 
        this.pl.addVertexAt(this.pt1); 
        this.pl.addVertexAt(this.pt3); 
        this.pl.addVertexAt(this.pt2); 
        this.pl.addVertexAt(this.pt4); 
        this.pl.isClosed = true; 
        const num = MxFun.screenCoordLong2Doc(10); 
        this.pt1.x -= num; this.pt1.y -= num; this.pt2.x += num; this.pt2.y += num; 
        // 生成预览图 
        let w = Math.abs(this.pt1.x - this.pt2.x); 
        let h = Math.abs(this.pt1.y - this.pt2.y); 
        if (w < 1 || h < 1) return; 
        let jpg_width = 300; 
        let jpg_height = 240; 
        return new Promise<String>((resolve, reject) => { 
            MxFun.getCurrentDraw().createCanvasImageData( 
                (imageData: String) => { 
                    this.patternImgUrl = imageData; 
                    // 恢复绘制顺序 
                    oldDrawOder.forEach(item => { 
                        item.obj.drawOrder = item.drawOrder; 
                    }) 
                    wipeoutId.erase(); 
                    resolve(imageData) 
                }, 
                { 
                    width: jpg_width, // 图片宽 
                    height: jpg_height, // 图片高 
                    range_pt1: this.pt1.toVector3(), // 截图范围角点1 
                    range_pt2: this.pt2.toVector3(), // 截图范围角点2 
                } 
            ); 
        }); 
    }; 
    // 开始识别图形 
    /** 
     * 1.查找初始包围盒 
     * 2.筛选符合条件的包围盒 
     * 3.比较包围盒内的每一个图元与需要识别的图元是否匹配 
     */ 
    public identifyGraphics(corner1?: McGePoint3d, corner2?: McGePoint3d): any { 
        /** 记录图形缩略图 */ 
        this.patternList.patternImgUrl = this.patternImgUrl; 
        /**初始化图形位置列表 */ 
        this.graphBoxsPos.length = 0; 
        const mxcad = MxCpp.getCurrentMxCAD(); 
        this.dTolPt = mxcad.mxdraw.viewCoordLong2Cad(0.5);// 设置精度值 
        if (!this.classNames) return; 
        const array = this.classNames.map(name => this.selectObjects[name].length); 
        const minIndex = array.findIndex(element => element === Math.min(...array)); 
        const str = this.classNames[minIndex]; 
        /** 识别块 */ 
        if (str === 'McDbBlockReference') { 
            const refBlk = this.selectObjects['McDbBlockReference'][0].clone() as McDbBlockReference; 
            const blkName = refBlk.blockName; // 图块名 
            const { position } = this.getBlkMidPt(refBlk); 
            this.formPt = position; 
            const rotate = refBlk.rotation; 
            // 设置图块过滤器 
            const filter = new MxCADResbuf([DxfCode.kEntityType, "INSERT"]) 
            const ss = new MxCADSelectionSet() 
            if (corner1 && corner2) { 
                // 局部选择 
                ss.crossingSelect(corner1.x, corner1.y, corner2.x, corner2.y, filter); 
            } else { 
                // 全局选择 
                ss.allSelect(filter); 
            }; 
            // 筛选出符合条件的块 
            const boxs: any[] = []; 
            ss.forEach(id => { 
                const _blk = id.getMcDbEntity().clone() as McDbBlockReference; 
                /** 块名相等=》计算旋转角度=》计算包围盒大小 */ 
                if (blkName === _blk.blockName) { 
                    const _rotate = _blk.rotation; 
                    const angle = rotate >= 0 ? _rotate - rotate : rotate - _rotate; 
                    const { position: _position, minPt, maxPt } = this.getBlkMidPt(_blk); 
                    if (this.classNames.length === 1) { 
                        this.toPt = _position; 
                        this.identificationBBox(minPt, maxPt, 0) 
                    } else { 
                        // 平移旋转初始包围盒。计算出最新包围盒的大小 
                        const _pl = this.pl.clone() as McDbPolyline; 
                        _pl.move(this.formPt, _position); 
                        _pl.rotate(_position, angle); 
                        let x = [], y = []; 
                        for (let i = 0; i < _pl.numVerts(); i++) { 
                            const pt = _pl.getPointAt(i).val; 
                            x.push(pt.x); 
                            y.push(pt.y); 
                        } 
                        const _minPt = new McGePoint3d(Math.min(...x) - 1, Math.min(...y) - 1); 
                        const _maxPt = new McGePoint3d(Math.max(...x) + 1, Math.max(...y) + 1); 
                        const res = boxs.filter(item => { 
                            let ret1 = _minPt.distanceTo(item.pt1) < this.dTolPt && _maxPt.distanceTo(item.pt2) < this.dTolPt 
                            let ret2 = _minPt.distanceTo(item.pt2) < this.dTolPt && _maxPt.distanceTo(item.pt1) < this.dTolPt 
                            return ret1 || ret2 
                        }) 
                        if (!res.length) { 
                            boxs.push({ pt1: _minPt, pt2: _maxPt }); 
                            this.toPt = _position; 
                            // MxCpp.getCurrentMxCAD().drawLine(_maxPt.x, _maxPt.y, _minPt.x, _minPt.y); 
                            this.identificationBBox(_minPt, _maxPt, angle); 
                        } 
                    } 
                } 
            }); 
        } 
        /** 识别弧线 */ 
        else if (str === 'McDbArc') { 
            // 识别直线 
            const refArc = this.selectObjects['McDbArc'][0].clone() as McDbArc; 
            const length = refArc.getLength().val; // 弧线线长度 
            // 圆弧凸度 
            const { bugle, midPt } = this.getBugle(refArc, length); 
            const center = refArc.center; 
            const vec = center.sub(midPt); 
            // 设置弧线过滤器 
            const filter = new MxCADResbuf([DxfCode.kEntityType, "ARC"]) 
            const ss = new MxCADSelectionSet() 
            if (corner1 && corner2) { 
                // 局部选择 
                ss.crossingSelect(corner1.x, corner1.y, corner2.x, corner2.y, filter); 
            } else { 
                // 全局选择 
                ss.allSelect(filter); 
            }; 
            // 筛选出符合条件的弧线 
            const boxs: any[] = []; 
            ss.forEach(id => { 
                const _arc = id.getMcDbEntity().clone() as McDbArc; 
                const _length = _arc.getLength().val; 
                const { bugle: _bugle, midPt: _midPt } = this.getBugle(_arc, _length); 
                /** 筛选出圆弧凸度与长度相同的弧线*/ 
                if (Math.abs(_length - length) < this.dTol && Math.abs(_bugle - bugle) < this.dTol) { 
                    /** 分两种情况 
                     * 1.直接旋转平移得到 
                     * 2.镜像得到=》移动到目标位置+旋转角度+镜像翻转=》比较起始点位置 
                     */ 
                    // 平移+旋转 
                    const _center = _arc.center; 
                    const _vec = _center.sub(_midPt); 
                    const angle = _vec.angleTo2(vec, McGeVector3d.kNegateZAxis); 
                    // 旋转平移初始包围盒。计算出最新包围盒的大小 
                    const _pl = this.pl.clone() as McDbPolyline; 
                    _pl.move(midPt, _midPt); 
                    _pl.rotate(_midPt, angle); 
                    let x = [], y = []; 
                    for (let i = 0; i < _pl.numVerts(); i++) { 
                        const pt = _pl.getPointAt(i).val; 
                        x.push(pt.x); 
                        y.push(pt.y); 
                    } 
                    const _minPt = new McGePoint3d(Math.min(...x) - 1, Math.min(...y) - 1); 
                    const _maxPt = new McGePoint3d(Math.max(...x) + 1, Math.max(...y) + 1); 
                    const res = boxs.filter(item => { 
                        let ret1 = _minPt.distanceTo(item.pt1) < this.dTolPt && _maxPt.distanceTo(item.pt2) < this.dTolPt 
                        let ret2 = _minPt.distanceTo(item.pt2) < this.dTolPt && _maxPt.distanceTo(item.pt1) < this.dTolPt 
                        let ret3 = _midPt.distanceTo(item.midPt) < this.dTolPt 
                        return (ret1 || ret2) && ret3 
                    }) 
                    if (!res.length) { 
                        boxs.push({ pt1: _minPt, pt2: _maxPt, midPt: _midPt }); 
                        this.toPt = _midPt; 
                        this.formPt = midPt; 
                        this.identificationBBox(_minPt, _maxPt, angle); 
                    } 
                } 
            }); 
        } 
        /** 识别直线 */ 
        else if (str === 'McDbLine') { 
            // 识别直线 
            const refLine = this.selectObjects['McDbLine'][0].clone() as McDbLine; 
            const length = refLine.getLength().val; // 直线长度 
            const vec = refLine.endPoint.sub(refLine.startPoint); // 直线角度 
            // 设置直线过滤器 
            const filter = new MxCADResbuf([DxfCode.kEntityType, "LINE"]) 
            const ss = new MxCADSelectionSet() 
            if (corner1 && corner2) { 
                // 局部选择 
                ss.crossingSelect(corner1.x, corner1.y, corner2.x, corner2.y, filter); 
            } else { 
                // 全局选择 
                ss.allSelect(filter); 
            }; 
            // 筛选出符合条件的直线 
            const lineArr: McDbLine[] = []; 
            ss.forEach(id => { 
                const line = id.getMcDbEntity().clone() as McDbLine; 
                const _length = line.getLength().val; 
                /** 
                 * 1.未旋转 
                 * 2.旋转过 
                 */ 
                if (Math.abs(_length - length) < this.dTol) { 
                    lineArr.push(line); 
                } 
            }); 
            // 筛选出符合条件的直线后，根据包围盒范围再次筛选出符合条件的直线组合 
            const boxs: any[] = []; 
            if (lineArr.length > 0) { 
                lineArr.forEach((item) => { 
                    /** 
                     * 将要识别的图形进行平移+旋转到目标位置 
                     */ 
                    const ptArr = [item.startPoint, item.endPoint]; 
 
                    ptArr.forEach((point, index) => { 
                        const _line = refLine.clone() as McDbLine; 
                        _line.move(refLine.startPoint, point); 
                        const _vec = ptArr[1 - index].sub(point); 
                        const angle = _vec.angleTo2(vec, McGeVector3d.kNegateZAxis); 
                        _line.rotate(point, angle) 
                        const pt = _line.endPoint; 
                        const _pl = this.pl.clone() as McDbPolyline; 
                        _pl.move(refLine.startPoint, point); 
                        _pl.rotate(point, angle); 
                        const _refBoxPt1 = _pl.getPointAt(0).val; 
                        const _refBoxPt2 = _pl.getPointAt(1).val; 
                        const refVec1 = _refBoxPt1.sub(_line.startPoint); 
                        const refVec2 = _refBoxPt2.sub(_line.startPoint); 
                        if (pt.distanceTo(ptArr[1 - index]) < this.dTolPt && refVec1.isEqualTo(_refBoxPt1.sub(point)) && refVec2.isEqualTo(_refBoxPt2.sub(point))) { 
                            let x = [], y = []; 
                            for (let i = 0; i < _pl.numVerts(); i++) { 
                                const pt = _pl.getPointAt(i).val; 
                                x.push(pt.x); 
                                y.push(pt.y); 
                            } 
                            const _minPt = new McGePoint3d(Math.min(...x) - 1, Math.min(...y) - 1); 
                            const _maxPt = new McGePoint3d(Math.max(...x) + 1, Math.max(...y) + 1); 
                            const res = boxs.filter(item => { 
                                let r1 = _minPt.distanceTo(item.pt1) < this.dTolPt && _maxPt.distanceTo(item.pt2) < this.dTolPt 
                                let r2 = _minPt.distanceTo(item.pt2) < this.dTolPt && _maxPt.distanceTo(item.pt1) < this.dTolPt 
                                return r1 || r2 
                            }) 
                            if (!res.length) { 
                                boxs.push({ pt1: _minPt, pt2: _maxPt }); 
                                this.toPt = point; 
                                this.formPt = refLine.startPoint; 
                                // MxCpp.getCurrentMxCAD().drawLine(_maxPt.x, _maxPt.y, _minPt.x, _minPt.y); 
                                this.identificationBBox(_minPt, _maxPt, angle); 
                            } 
                        } 
                    }) 
                }) 
            }; 
        } 
        /** 识别圆 */ 
        else if (str === 'McDbCircle') { 
            const refCricle = this.selectObjects['McDbCircle'][0].clone() as McDbCircle; 
            const radius = refCricle.radius; // 圆半径 
            // 设置圆过滤器 
            const filter = new MxCADResbuf([DxfCode.kEntityType, "CIRCLE"]) 
            const ss = new MxCADSelectionSet() 
            if (corner1 && corner2) { 
                // 局部选择 
                ss.crossingSelect(corner1.x, corner1.y, corner2.x, corner2.y, filter); 
            } else { 
                // 全局选择 
                ss.allSelect(filter); 
            }; 
            // 筛选出符合条件的圆 
            ss.forEach(id => { 
                const _circle = id.getMcDbEntity().clone() as McDbCircle; 
                const _radius = _circle.radius; 
                /** 筛选出圆半径相同的圆*/ 
                if (Math.abs(_radius - radius) < this.dTol) { 
                    const _center = _circle.center; 
                    this.toPt = _center; 
                    this.formPt = refCricle.center; 
                    // 平移初始包围盒。计算出最新包围盒的大小 
                    const _pl = this.pl.clone() as McDbPolyline; 
                    _pl.move(this.formPt, this.toPt); 
                    let x = [], y = []; 
                    for (let i = 0; i < _pl.numVerts(); i++) { 
                        const pt = _pl.getPointAt(i).val; 
                        x.push(pt.x); 
                        y.push(pt.y); 
                    } 
                    const _minPt = new McGePoint3d(Math.min(...x) - 1, Math.min(...y) - 1); 
                    const _maxPt = new McGePoint3d(Math.max(...x) + 1, Math.max(...y) + 1); 
                    // MxCpp.getCurrentMxCAD().drawLine(_maxPt.x, _maxPt.y, _minPt.x, _minPt.y); 
                    this.identificationBBox(_minPt, _maxPt, 0); 
                } 
            }); 
        } 
        /** 识别多段线 */ 
        else if (str === 'McDbPolyline') { 
            /** 长度+顶点数+顶点对应的凸度相等=》旋转+平移后的顶点相等+有凸度线段的中点相等*/ 
            const refPoly = this.selectObjects['McDbPolyline'][0].clone() as McDbPolyline; 
            const length = refPoly.getLength().val; 
            const { lengthArr, bugleArr, vertNum, vec, point } = this.getPlDetails(refPoly); 
            // 设置多段线过滤器 
            const filter = new MxCADResbuf([DxfCode.kEntityType, "LWPOLYLINE"]) 
            const ss = new MxCADSelectionSet() 
            if (corner1 && corner2) { 
                // 局部选择 
                ss.crossingSelect(corner1.x, corner1.y, corner2.x, corner2.y, filter); 
            } else { 
                // 全局选择 
                ss.allSelect(filter); 
            }; 
            // 筛选出符合条件的多段线 
            const boxs: any[] = []; 
            ss.forEach(id => { 
                const _poly = id.getMcDbEntity().clone() as McDbPolyline; 
                const _length = _poly.getLength().val; 
                /** 筛选 长度+顶点数+顶点对应的凸度 相同的多段线*/ 
                if (Math.abs(_length - length) < this.dTol && _poly.isClosed === refPoly.isClosed) { 
                    const { lengthArr: _lengthArr, bugleArr: _bugleArr, vertNum: _vertNum, vec: _vec, point: _point } = this.getPlDetails(_poly); 
                    const revBugleArr = _bugleArr.map(item => item === 0 ? item : -item); 
                    // 全等 
                    const r1 = lengthArr.toString() === _lengthArr.toString() && bugleArr.toString() === _bugleArr.toString(); 
                    let r2 = false; 
                    if (!r1) { 
                        // 反向 
                        r2 = lengthArr.toString() === [..._lengthArr].reverse().toString() && bugleArr.toString() === [...revBugleArr].reverse().toString(); 
                    } 
                    const r3 = vertNum === _vertNum; 
                    // 镜像 
                    const r4 = lengthArr.toString() === _lengthArr.toString() && bugleArr.toString() === [...revBugleArr].reverse().toString(); 
                    if ((r1 || r2 || r4) && r3) { 
                        const poly = refPoly.clone() as McDbPolyline; 
                        const angle1 = vec.angleTo2(McGeVector3d.kXAxis, McGeVector3d.kNegateZAxis); 
                        // 如果是反向则方向取反 
                        const angle2 = (r2 ? _vec.negate() : _vec).angleTo2(McGeVector3d.kXAxis, McGeVector3d.kNegateZAxis); 
                        const angle = angle2 - angle1; 
                        poly.move(point, _point); 
                        poly.rotate(_point, angle); 
                        poly.trueColor = new McCmColor(255, 0, 0); 
                        let num = 0; 
                        for (let i = 0; i < vertNum; i++) { 
                            const pt = poly.getPointAt(i).val; 
                            const width = poly.getWidthsAt(i); 
                            const _pt = r2 ? _poly.getPointAt(vertNum - 1 - i).val : _poly.getPointAt(i).val; 
                            const _width = r2 ? _poly.getWidthsAt(vertNum - 1 - i) : _poly.getWidthsAt(i); 
                            if (pt.distanceTo(_pt) < this.dTolPt && width.val1 === _width.val1 && width.val2 === _width.val2) { 
                                num += 1 
                            } else { 
                                // 两个点且反向的情况 
                                if (vertNum === 2 && i === 0) { 
                                    const _pt1 = _poly.getPointAt(0).val; 
                                    const _pt2 = _poly.getPointAt(1).val; 
                                    const pt2 = poly.getPointAt(1).val; 
                                    const _width = _poly.getWidthsAt(0); 
                                    if (pt2.distanceTo(_pt2) < this.dTolPt && pt.distanceTo(_pt1) < this.dTolPt && width.val1 === _width.val2 && width.val2 === _width.val1) { 
                                        num += 2; 
                                    } 
                                } 
                            } 
 
                        }; 
                        if (num === vertNum) { 
                            // 旋转平移初始包围盒。计算出最新包围盒的大小 
                            const _pl = this.pl.clone() as McDbPolyline; 
                            _pl.move(point, _point); 
                            _pl.rotate(_point, angle); 
                            let x = [], y = []; 
                            for (let i = 0; i < _pl.numVerts(); i++) { 
                                const pt = _pl.getPointAt(i).val; 
                                x.push(pt.x); 
                                y.push(pt.y); 
                            } 
                            const _minPt = new McGePoint3d(Math.min(...x) - 1, Math.min(...y) - 1); 
                            const _maxPt = new McGePoint3d(Math.max(...x) + 1, Math.max(...y) + 1); 
                            const res = boxs.filter(item => { 
                                let ret1 = _minPt.distanceTo(item.pt1) < this.dTolPt && _maxPt.distanceTo(item.pt2) < this.dTolPt 
                                let ret2 = _minPt.distanceTo(item.pt2) < this.dTolPt && _maxPt.distanceTo(item.pt1) < this.dTolPt 
                                return ret1 || ret2 
                            }) 
                            if (!res.length) { 
                                boxs.push({ pt1: _minPt, pt2: _maxPt }); 
                                this.toPt = _point; 
                                this.formPt = point; 
                                // MxCpp.getCurrentMxCAD().drawLine(_maxPt.x, _maxPt.y, _minPt.x, _minPt.y); 
                                this.identificationBBox(_minPt, _maxPt, angle); 
                            } 
                        } 
                    } 
                } 
            }); 
        } 
        /** 返回识别列表 */ 
        return this.patternList; 
    } 
    // 计算块中心点 
    private getBlkMidPt = (blkRef: McDbBlockReference) => { 
        const { minPt, maxPt } = blkRef.getBoundingBox(); 
        const pt = minPt.clone().addvec(maxPt.sub(minPt).normalize().mult(maxPt.distanceTo(minPt) / 2)) 
        return { position: pt, minPt, maxPt } 
    } 
    // 计算凸度 
    private getBugle = (arc: McDbArc, length: number) => { 
        const midPt = arc.getPointAtDist(length / 2).val; 
        const bugle = MxCADUtility.calcBulge(arc.getPointAtDist(0).val, midPt, arc.getPointAtDist(length).val).val; 
        return { bugle, midPt } 
    } 
    // 计算多段线的每一段线的长度和凸度 
    private getPlDetails = (pl: McDbPolyline) => { 
        const vertNum = pl.numVerts(); 
        const lengthArr: number[] = []; 
        const bugleArr: number[] = []; 
        let point1: McGePoint3d, point2: McGePoint3d; 
        for (let i = 0; i < vertNum; i++) { 
            if (i < vertNum - 1) { 
                const p1 = pl.getPointAt(i).val; 
                const p2 = pl.getPointAt(i + 1).val; 
                const length = pl.getDistAtPoint(p2).val - pl.getDistAtPoint(p1).val; 
                const bugle = pl.getBulgeAt(i); 
                lengthArr.push(Number(length.toFixed(6))); 
                bugleArr.push(Number(bugle.toFixed(6))); 
                if (i === 0) { 
                    point1 = p1; 
                } 
                if (i === vertNum - 2) { 
                    point2 = p2; 
                } 
            } 
        }; 
        const vec = point2.sub(point1)// 多段线角度 
        const point = point1.clone().addvec(point2.sub(point1).normalize().mult(point2.distanceTo(point1) / 2)) 
        return { lengthArr, bugleArr, vertNum, vec, point } 
    } 
    // 将找到的实体数组与需要识别的实体数组比对 
    private comparisonPhysical = (objects: any, maxPt: McGePoint3d, minPt: McGePoint3d, angle: number): any => { 
        const calssNames = Object.keys(this.selectObjects); 
        if (!calssNames) return; 
        let num = 0; 
        if (calssNames.includes('McDbLine')) { 
            let ents = objects['McDbLine']; 
            const lineObjects = this.objectsDetails['McDbLine']; 
            for (let i = 0; i < lineObjects.length; i++) { 
                const element = lineObjects[i]; 
                let _line = (element.entity as McDbLine).clone() as McDbLine; 
                _line.move(this.formPt, this.toPt); 
                _line.rotate(this.toPt, angle); 
                const indexArr: number[] = []; 
                const mLineArr = ents.filter((item: McDbLine, index: number) => { 
                    if (Math.abs(element.length - item.getLength().val) < this.dTol) { 
                        let r1 = item.startPoint.distanceTo(_line.startPoint) < this.dTolPt && item.endPoint.distanceTo(_line.endPoint) < this.dTolPt; 
                        let r2 = item.startPoint.distanceTo(_line.endPoint) < this.dTolPt && item.endPoint.distanceTo(_line.startPoint) < this.dTolPt; 
                        if (r1 || r2) indexArr.push(index); 
                        return r1 || r2; 
                    } 
                }); 
                if (mLineArr.length > 0) { 
                    num += 1; 
                    ents.splice(indexArr[0], 1); 
                    ents = [...ents]; 
                } else { 
                    break; 
                } 
            } 
        }; 
        if (calssNames.includes('McDbArc')) { 
            let ents = objects['McDbArc']; 
            const arcObjects = this.objectsDetails['McDbArc']; 
            for (let i = 0; i < arcObjects.length; i++) { 
                const element = arcObjects[i]; 
                let _arc = (element.entity as McDbArc).clone() as McDbArc; 
                _arc.move(this.formPt, this.toPt); 
                _arc.rotate(this.toPt, angle); 
                const startPoint = _arc.getPointAtDist(0).val; 
                const endPoint = _arc.getPointAtDist(element.length).val; 
                const indexArr: number[] = []; 
                const mArcArr = ents.filter((item: McDbArc, index: number) => { 
                    const _length = item.getLength().val; 
                    const { bugle: _bugle } = this.getBugle(item, _length) 
                    if (Math.abs(element.length - item.getLength().val) < this.dTol && Math.abs(_bugle - element.bugle) < this.dTol) { 
                        const _startPoint = item.getPointAtDist(0).val; 
                        const _endPoint = item.getPointAtDist(_length).val; 
                        let r1 = startPoint.distanceTo(_startPoint) < this.dTolPt && endPoint.distanceTo(_endPoint) < this.dTolPt; 
                        let r2 = startPoint.distanceTo(_endPoint) < this.dTolPt && endPoint.distanceTo(_startPoint) < this.dTolPt; 
                        if (r1 || r2) { 
                            indexArr.push(index); 
                        } 
                        return r1 || r2; 
                    } 
                }); 
                if (mArcArr.length > 0) { 
                    num += 1; 
                    ents.splice(indexArr[0], 1); 
                    ents = [...ents]; 
                } else { 
                    break; 
                } 
            } 
        }; 
        if (calssNames.includes('McDbCircle')) { 
            let ents = objects['McDbCircle']; 
            const circleObjects = this.objectsDetails['McDbCircle']; 
            for (let i = 0; i < circleObjects.length; i++) { 
                const element = circleObjects[i]; 
                let _circle = (element.entity as McDbCircle).clone() as McDbCircle; 
                _circle.move(this.formPt, this.toPt); 
                _circle.rotate(this.toPt, angle); 
                const center = _circle.center; 
                const radius = _circle.radius; 
                const indexArr: number[] = []; 
                const mCiecleArr = ents.filter((item: McDbCircle, index: number) => { 
                    const _center = item.center; 
                    const _radius = item.radius; 
                    const r = Math.abs(_radius - radius) < this.dTol && center.distanceTo(_center) < this.dTolPt; 
                    if (r) { 
                        indexArr.push(index) 
                    } 
                    return r; 
                }); 
                if (mCiecleArr.length > 0) { 
                    num += 1; 
                    ents.splice(indexArr[0], 1); 
                    ents = [...ents]; 
                } else { 
                    break; 
                } 
            } 
        }; 
        if (calssNames.includes('McDbBlockReference')) { 
            /** 平移+旋转=>比对包围盒大小+块名字 */ 
            let ents = objects['McDbBlockReference']; 
            const blkObjects = this.objectsDetails['McDbBlockReference']; 
            for (let i = 0; i < blkObjects.length; i++) { 
                const element = blkObjects[i]; 
                let _blk = (element.entity as McDbBlockReference).clone() as McDbBlockReference; 
                _blk.move(this.formPt, this.toPt); 
                _blk.rotate(this.toPt, angle); 
                const blkName = _blk.blockName; 
                const { minPt: blkMinPt, maxPt: blkMaxPt } = this.getBlkMidPt(_blk); 
                const indexArr: number[] = []; 
               const mArcArr = ents.filter((item: McDbBlockReference, index: number) => { 
                    const _blkName = item.blockName; 
                    if (_blkName === blkName) { 
                        if (this.classNames.length === 1) { 
                            indexArr.push(index); 
                            return true 
                        } else { 
                            const { minPt: _blkMinPt, maxPt: _blkMaxPt } = this.getBlkMidPt(item); 
                            let r1 = blkMinPt.distanceTo(_blkMinPt) < this.dTolPt && blkMaxPt.distanceTo(_blkMaxPt) < this.dTolPt; 
                            if (r1) { 
                                indexArr.push(index); 
                            } 
                            return r1 
                        } 
                    } 
                }); 
                if (mArcArr.length > 0) { 
                    num += 1; 
                    ents.splice(indexArr[0], 1); 
                    ents = [...ents]; 
                } else { 
                    return; 
                } 
            } 
        } 
        if (calssNames.includes('McDbPolyline')) { 
            /** 平移+旋转=>比对凸度+点位置 */ 
            let ents = objects['McDbPolyline']; 
            const plObjects = this.objectsDetails['McDbPolyline']; 
            for (let i = 0; i < plObjects.length; i++) { 
                const element = plObjects[i]; 
                let _poly = (element.entity as McDbPolyline).clone() as McDbPolyline; 
                _poly.move(this.formPt, this.toPt); 
                _poly.rotate(this.toPt, angle); 
                const { lengthArr, bugleArr, vertNum } = this.getPlDetails(_poly); 
                const indexArr: number[] = []; 
                const mPlArr = ents.filter((item: McDbPolyline, index: number) => { 
                    const _length = item.getLength().val; 
                    if (Math.abs(_length - element.length) < this.dTol && _poly.isClosed === _poly.isClosed) { 
                        const { lengthArr: _lengthArr, bugleArr: _bugleArr, vertNum: _vertNum } = this.getPlDetails(item); 
                        const revBugleArr = _bugleArr.map(item => item === 0 ? item : -item); 
                        const r1 = lengthArr.toString() === _lengthArr.toString() && bugleArr.toString() === _bugleArr.toString(); 
                        let r2 = false; 
                        if (!r1) { 
                            r2 = lengthArr.toString() === [..._lengthArr].reverse().toString() && bugleArr.toString() === [...revBugleArr].reverse().toString(); 
                        } 
                        const r3 = vertNum === _vertNum; 
                        if ((r1 || r2) && r3) { 
                            let _num = 0; 
                            for (let i = 0; i < vertNum; i++) { 
                                const pt = _poly.getPointAt(i).val; 
                                const width = _poly.getWidthsAt(i); 
                                const _pt = r2 ? item.getPointAt(vertNum - 1 - i).val : item.getPointAt(i).val; 
                                const _width = r2 ? item.getWidthsAt(vertNum - 1 - i) : item.getWidthsAt(i); 
                                if (pt.distanceTo(_pt) < this.dTolPt && width.val1 === _width.val1 && width.val2 === _width.val2) { 
                                    _num += 1 
                                } else { 
                                    // 两个点且反向的情况 
                                    if (vertNum === 2 && i === 0) { 
                                        const _pt1 = item.getPointAt(0).val; 
                                        const _pt2 = item.getPointAt(1).val; 
                                        const _width = item.getWidthsAt(0); 
                                        const pt2 = _poly.getPointAt(1).val; 
                                        if (pt2.distanceTo(_pt2) < this.dTolPt && pt.distanceTo(_pt1) < this.dTolPt && width.val1 === _width.val2 && width.val2 === _width.val1) { 
                                           _num += 2; 
                                        } 
                                    } 
                                } 
                            }; 
                            if (_num === vertNum) { 
                                indexArr.push(index); 
                            } 
                            return _num === vertNum; 
                        } 
                    } 
                }); 
                if (mPlArr.length > 0) { 
                    num += 1; 
                    ents.splice(indexArr[0], 1); 
                    ents = [...ents]; 
                } else { 
                    break; 
                } 
            } 
        } 
        if (num === this.objectCount) { 
            // 匹配成功 
            const pt = minPt.clone().addvec(maxPt.clone().sub(minPt).normalize().mult(minPt.distanceTo(maxPt) / 2)); 
            const res = this.graphBoxsPos.filter(point => pt.distanceTo(point) < this.dTolPt); 
            if (!res.length) { 
                this.graphBoxsPos.push(pt) 
            } 
        }; 
        this.patternList.graphBoxsPos = [...this.graphBoxsPos]; 
    } 
    // 识别包围盒是否符合要求 
    private identificationBBox = (minPt: McGePoint3d, maxPt: McGePoint3d, angle: number): any => { 
        const _ss = new MxCADSelectionSet() 
        _ss.crossingSelect(minPt.x, minPt.y, maxPt.x, maxPt.y); 
        const boxEntity = {}; 
        _ss.forEach(id => { 
            const ent = id.getMcDbEntity(); 
            if (!boxEntity[ent.objectName]) boxEntity[ent.objectName] = []; 
            boxEntity[ent.objectName].push(ent); 
        }); 
        let r = this.classNames.filter(name => boxEntity[name] && this.selectObjects[name].length <= boxEntity[name].length); 
        if (r.length === this.classNames.length) { 
            this.comparisonPhysical(boxEntity, maxPt, minPt, angle); 
        } 
    } 
};

调用 SearchSamePattern 类，实现图形识别功能，代码如下：

/** 获取需要识别的图形 */
const getIdentificationPattern = async () => {
// 获取需要识别的图形
const ss = new MxCADSelectionSet();// 选择集
const filter = new MxCADResbuf([DxfCode.kEntityType, "LINE,ARC,CIRCLE,LWPOLYLINE,INSERT"])
if (!await ss.userSelect("请点击左键或拉框选择需要识别的图形:", filter)) return;
if (ss.count() == 0) return;
// 搜索图形
const searchPattern = new SearchSamePattern();
// 搜索图形简图地址
const patternImgUrl = await searchPattern.setSelectset(ss);
if (!patternImgUrl.value) return;
// 获取识别结果
const searchList = searchPattern.identifyGraphics();
console.log(searchList);
}

复制代码

你有种再说一遍 · 发表于 2024-9-27 23:53:50

感觉大数据这个功能还是很慢呀.
不知道怎么加速好.

MxDraw · 发表于 2024-10-9 11:52:03

你有种再说一遍发表于 2024-9-27 23:53
感觉大数据这个功能还是很慢呀.
不知道怎么加速好.

这个我们要看看怎么优化

账号		自动登录	找回密码
密码			注册

（H5 web CAD sdk）在线CAD如何实现图形识别功能

本帖子中包含更多资源