SystemParseManager.cs 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573
  1. /*-------------------------------------------------------------------------
  2. * 功能描述:SystemParseManager
  3. * 作者:xulisong
  4. * 创建时间: 2019/1/10 14:34:23
  5. * 版本号:v1.0
  6. * -------------------------------------------------------------------------*/
  7. using System;
  8. using System.Collections.Generic;
  9. using System.Linq;
  10. using System.Windows.Media.Media3D;
  11. using Autodesk.Revit.DB;
  12. using Autodesk.Revit.DB.Mechanical;
  13. using Autodesk.Revit.DB.Plumbing;
  14. using SAGA.Models;
  15. using SAGA.Models.GplotElement;
  16. using SAGA.Models.Graphs;
  17. using SAGA.RevitUtils;
  18. using SAGA.RevitUtils.Extends;
  19. using SAGA.RevitUtils.MEP;
  20. namespace SAGA.GplotRelationComputerManage
  21. {
  22. /// <summary>
  23. /// 系统解析管理
  24. /// </summary>
  25. public class SystemParseManager
  26. {
  27. #region 解析管路系统数据
  28. #region 通用基础方法
  29. /// <summary>
  30. /// 创建树节点
  31. /// </summary>
  32. /// <param name="startElement"></param>
  33. /// <param name="startConnector"></param>
  34. /// <param name="predicateEnd"></param>
  35. /// <returns></returns>
  36. public static ElementTreeNode CreateTreeNode(Element startElement, Connector startConnector,
  37. Predicate<Element> predicateEnd)
  38. {
  39. ElementTreeNode node = new ElementTreeNode();
  40. node.Current = startElement;
  41. List<ElementTreeNode> reference = new List<ElementTreeNode>();
  42. reference.Add(node);
  43. for (int i = 0; i < reference.Count; i++)
  44. {
  45. var baseNode = reference[i];
  46. List<Connector> connectors;
  47. if (i == 0 && startConnector != null)
  48. {
  49. connectors = new List<Connector> { startConnector };
  50. }
  51. else
  52. {
  53. connectors = baseNode.Current.GetAllConnectors().ToList();
  54. }
  55. foreach (var connector in connectors)
  56. {
  57. if (connector.Searchable())
  58. {
  59. var refConnectors = connector.GetReferenceConnectors()
  60. .Where(c => ConnectorType.Physical.HasFlag(c.ConnectorType)).ToList();
  61. foreach (var refConnector in refConnectors)
  62. {
  63. var refElement = refConnector.Owner;
  64. if (reference.Reverse<ElementTreeNode>().Any(e => e.Current.Id.IntegerValue == refElement.Id.IntegerValue))
  65. {
  66. continue;
  67. }
  68. ElementTreeNode refNode = new ElementTreeNode();
  69. refNode.Current = refElement;
  70. refNode.StartLocation = refConnector.Origin;
  71. baseNode.Nodes.Add(refNode);
  72. if (predicateEnd != null && predicateEnd(refElement))
  73. {
  74. continue;
  75. }
  76. //if (reference.All(e => e.Current.Id.IntegerValue != refElement.Id.IntegerValue))
  77. //{
  78. reference.Add(refNode);
  79. //}
  80. }
  81. }
  82. }
  83. }
  84. return node;
  85. }
  86. #endregion
  87. /// <summary>
  88. /// 构件系统连接节点
  89. /// </summary>
  90. /// <param name="etn">开始元素</param>
  91. /// <param name="domain"></param>
  92. /// <param name="endConditon">遍历中断条件,遇到指定条件的管道则停止遍历</param>
  93. /// <returns>树形节点中包好的Id信息</returns>
  94. public static List<int> BuildSystemNode(ElementTreeNode etn, Domain domain, Func<MEPCurve, bool> endConditon)
  95. {
  96. List<int> useIds = new List<int>();
  97. etn.GetAllNodes().ForEach(n =>
  98. {
  99. if (!n.IsLeaf&&n.Current != null)
  100. {
  101. useIds.Add(n.Current.Id.IntegerValue);
  102. }
  103. });
  104. var leaves = etn.GetLeaves();
  105. foreach (var elementTreeNode in leaves)
  106. {
  107. BuildSystemNode(elementTreeNode, domain, useIds, endConditon);
  108. }
  109. return useIds;
  110. }
  111. private static void BuildSystemNode(ElementTreeNode etn, Domain domain, List<int> useIds, Func<MEPCurve, bool> endConditon)
  112. {
  113. var useElement = etn.Current;
  114. #region 预判退出
  115. if (useElement == null)
  116. {
  117. return;
  118. }
  119. if (useIds.Any(id => id == useElement.Id.IntegerValue))
  120. {
  121. return;
  122. }
  123. #endregion
  124. useIds.Add(useElement.Id.IntegerValue);
  125. var connectors = useElement.GetConnectors(domain);
  126. foreach (var baseConnector in connectors)
  127. {
  128. if (!baseConnector.Searchable())
  129. continue;
  130. var refConnectors = baseConnector.GetReferenceConnectors()
  131. .Where(c => ConnectorType.Physical.HasFlag(c.ConnectorType)).ToList();
  132. foreach (var refConnector in refConnectors)
  133. {
  134. var refElement = refConnector.Owner;
  135. //逻辑变更,如果存在,不继续遍历。但要完成本次操作;移动到最后
  136. //if (useIds.Any(id => id == refElement.Id.IntegerValue))
  137. //{
  138. // continue;
  139. //}
  140. //初始化当前节点,未必一定加入集合,为了下面代码书写方便,先进性初始化
  141. ElementTreeNode currentNode = new ElementTreeNode()
  142. {
  143. Current = refElement,
  144. StartLocation = refConnector.Origin
  145. };
  146. if (refElement is MEPCurve mepCurve)
  147. {
  148. //优先判定系统,再进行立管相关逻辑的处理;如果系统不兼容则直接退出遍历
  149. //var systemName=mepCurve.GetSystemTypeName();
  150. if (endConditon != null && endConditon(mepCurve))
  151. {
  152. continue;
  153. }
  154. if (SystemCalcUtil.IsStart(mepCurve))
  155. {
  156. currentNode.ElementTypeName = AcType.MarkStandPipe;
  157. etn.AddChild(currentNode);
  158. useIds.Add(refElement.Id.IntegerValue);
  159. continue;
  160. }
  161. }
  162. else if (refElement is FamilyInstance fi)
  163. {
  164. //非管线,遇到结束标志,加入当前子节点,但不在进行递归;
  165. if (SystemCalcUtil.IsStart(fi))
  166. {
  167. currentNode.ElementTypeName = AcType.Valve;
  168. etn.AddChild(currentNode);
  169. useIds.Add(refElement.Id.IntegerValue);
  170. continue;
  171. }
  172. }
  173. //useIds的控制:跳出时,直接加入UseIds。继续遍历的,在遍历函数中加入
  174. etn.AddChild(currentNode);
  175. if (useIds.Any(id => id == refElement.Id.IntegerValue))
  176. {
  177. continue;
  178. }
  179. BuildSystemNode(currentNode, domain, useIds, endConditon);
  180. }
  181. }
  182. }
  183. #endregion
  184. #region 创建绘图数据相关
  185. /// <summary>
  186. /// 构件绘图数据
  187. /// </summary>
  188. /// <param name="etn"></param>
  189. /// <returns></returns>
  190. public static SystemDrawItems CreateDrawing(ElementTreeNode etn)
  191. {
  192. #region 描述
  193. /*
  194. *读取几何信息
  195. * 1、管道,管件直接按connector连线获取。
  196. * 2、设备,可能出现多个connector,确定有效的连线;
  197. * 最终:如果是管道,通过本身的connector生成线。如果是familyInstance通过定位点连接,父节点和子节点
  198. * 3、具体名字相关内容,后续操作中添加
  199. */
  200. #endregion
  201. List<SystemCurveItem> curveItems = new List<SystemCurveItem>();
  202. List<SystemPointItem> pointItems = new List<SystemPointItem>();
  203. Queue<ElementTreeNode> nodeQueue = new Queue<ElementTreeNode>() ;
  204. nodeQueue.Enqueue(etn);
  205. while (nodeQueue.Any())
  206. {
  207. #region 控制验证
  208. var currentNode = nodeQueue.Dequeue();
  209. if (currentNode == null)
  210. continue;
  211. var element = currentNode.Current;
  212. if (element == null)
  213. continue;
  214. #endregion
  215. if (element is MEPCurve mepCurve)
  216. {
  217. #region 联通线处理
  218. Curve curve = mepCurve.GetCurve();
  219. var xyzs = curve.Tessellate().Select(xyz => xyz.XYZToPoint());
  220. SystemCurveItem curveItem = new SystemCurveItem();
  221. curveItem.Points.AddRange(xyzs.ToList());
  222. curveItem.RefId = element.Id.IntegerValue;
  223. curveItems.Add(curveItem);
  224. #endregion
  225. #region 点信息处理
  226. if (SystemCalcUtil.IsStart(mepCurve))
  227. {
  228. SystemPointItem point = new SystemPointItem();
  229. point.Point = mepCurve.GetCurve().StartPoint().XYZToPoint();
  230. point.RefId = mepCurve.Id.IntegerValue;
  231. point.IsVirtual = true;
  232. point.EquipmentType = PointItemType.StandPipe.ToString();
  233. point.Name = AppSetting.StartFlag;
  234. pointItems.Add(point);
  235. }
  236. #endregion
  237. }
  238. else if (element is FamilyInstance fiEq)
  239. {
  240. #region 获取引用点
  241. List<XYZ> usePoints = new List<XYZ>();
  242. if (currentNode.StartLocation != null)
  243. {
  244. usePoints.Add(currentNode.StartLocation);
  245. }
  246. foreach (ElementTreeNode childNode in currentNode.Nodes)
  247. {
  248. var usePoint = childNode.StartLocation;
  249. if (usePoint != null)
  250. {
  251. usePoints.Add(usePoint);
  252. }
  253. }
  254. #endregion
  255. var location = element.GetBoxCenter().XYZToPoint();
  256. #region 根据定位点并创建线条
  257. foreach (var usePoint in usePoints)
  258. {
  259. SystemCurveItem curveItem = new SystemCurveItem();
  260. curveItem.Points.Add(location);
  261. curveItem.Points.Add(usePoint.XYZToPoint());
  262. curveItem.RefId = element.Id.IntegerValue;
  263. curveItems.Add(curveItem);
  264. }
  265. #endregion
  266. #region 判断是否是设备
  267. //先初始化不一定加入
  268. SystemPointItem point = new SystemPointItem();
  269. point.Point = location;
  270. point.RefId = fiEq.Id.IntegerValue;
  271. point.BimId = element.GetBimId();
  272. if (MBIInfoUtil.IsEquipment(fiEq))
  273. {
  274. point.Name = string.Empty;
  275. point.EquipmentType = PointItemType.Equipment.ToString();
  276. var equipExtend = currentNode.GetTag<NodeExtendData>();
  277. if (equipExtend != null)
  278. {
  279. point.Name = equipExtend.Name;
  280. point.EquipmentType = equipExtend.Category;
  281. }
  282. pointItems.Add(point);
  283. }
  284. else if (SystemCalcUtil.IsStart(fiEq))
  285. {
  286. point.Name = string.Empty;
  287. point.EquipmentType = PointItemType.StartValve.ToString(); ;
  288. point.IsVirtual = true;
  289. point.Name=AppSetting.StartFlag;
  290. pointItems.Add(point);
  291. }
  292. #endregion
  293. }
  294. else if (element is Space)
  295. {
  296. SystemPointItem point = new SystemPointItem();
  297. point.Point = element.GetLocationPoint().XYZToPoint();
  298. point.RefId = element.Id.IntegerValue;
  299. point.EquipmentType = PointItemType.Space.ToString();
  300. point.IsVirtual = true;
  301. pointItems.Add(point);
  302. }
  303. currentNode.Nodes.ForEach(e => nodeQueue.Enqueue(e));
  304. }
  305. SystemDrawItems drawItems = new SystemDrawItems();
  306. drawItems.Curves = curveItems;
  307. drawItems.Points = pointItems;
  308. return drawItems;
  309. }
  310. #endregion
  311. #region 创建关系数据相关
  312. public static List<BinaryRelationItem> CreateRelation(ElementTreeNode etn)
  313. {
  314. List<BinaryRelationItem> result = new List<BinaryRelationItem>();
  315. #region 描述
  316. /*
  317. * 1、将树形结构初步处理,修剪成存留设备之间的关系;
  318. * 2、处理成二维关系
  319. */
  320. #endregion
  321. var equipmentNodes= ConvertEquipmentNodes(etn);
  322. var loopNodes = new List<EquipmentNode>(equipmentNodes);
  323. for (int i = 0; i < loopNodes.Count; i++)
  324. {
  325. var currentNode = loopNodes[i];
  326. foreach (var childNode in currentNode.Nodes)
  327. {
  328. BinaryRelationItem relation = new BinaryRelationItem();
  329. relation.From = currentNode.CloneCurrentNode();
  330. relation.To = childNode.CloneCurrentNode();
  331. result.Add(relation);
  332. loopNodes.Add(childNode);
  333. }
  334. }
  335. return result;
  336. }
  337. /// <summary>
  338. /// 将elment树转换成设备节点类
  339. /// </summary>
  340. /// <param name="etn"></param>
  341. /// <returns></returns>
  342. private static List<EquipmentNode> ConvertEquipmentNodes(ElementTreeNode etn)
  343. {
  344. List<EquipmentNode> nodes = new List<EquipmentNode>();
  345. List<ElementTreeNode> loopNodes = new List<ElementTreeNode>(){ etn };
  346. for (int i = 0; i < loopNodes.Count; i++)
  347. {
  348. var currentNode = loopNodes[i];
  349. if (TryGetNode(currentNode.Current, out EquipmentNode outNode))
  350. {
  351. foreach (var currentNodeNode in currentNode.Nodes)
  352. {
  353. outNode.AddChildren(ConvertEquipmentNodes(currentNodeNode));
  354. }
  355. nodes.Add(outNode);
  356. }
  357. else
  358. {
  359. loopNodes.AddRange(currentNode.Nodes);
  360. }
  361. }
  362. return nodes;
  363. }
  364. /// <summary>
  365. /// 获取element对应节点
  366. /// </summary>
  367. /// <param name="element">输入element</param>
  368. /// <param name="outNode">成功获取时,node真实值</param>
  369. /// <returns>成功获取</returns>
  370. private static bool TryGetNode(Element element,out EquipmentNode outNode)
  371. {
  372. bool flag = false;
  373. EquipmentNode node = new EquipmentNode();
  374. if (element is Space space)
  375. {
  376. node.Category = PointItemType.Space.ToString();
  377. }
  378. else if (element is Pipe&& SystemCalcUtil.IsStart(element))
  379. {
  380. node.Category = PointItemType.StandPipe.ToString();
  381. }
  382. else if (element is FamilyInstance)
  383. {
  384. string tempCategory;
  385. if (MBIInfoUtil.TryGetEquipmentCategory(element,out tempCategory))
  386. {
  387. node.Category = tempCategory;
  388. node.IsRealEquipment = true;
  389. }
  390. else if (SystemCalcUtil.IsStart(element))
  391. {
  392. node.Category = PointItemType.StartValve.ToString();
  393. }
  394. }
  395. flag = !string.IsNullOrEmpty(node.Category);
  396. outNode = flag ? node : null;
  397. #region 确定返回之后进行后初始化处理
  398. if (outNode != null)
  399. {
  400. outNode.RevitId = element.Id.ToString();
  401. outNode.BimId = element.GetBimId();
  402. }
  403. #endregion
  404. return flag;
  405. }
  406. #endregion
  407. #region 创建立面绘图和关系数据
  408. /// <summary>
  409. /// 计算立面数据
  410. /// </summary>
  411. public static VerticalResult ComputerVerticalData(List<VerticalPipe> pipeDatas, List<LevelData> levelDatas)
  412. {
  413. List<VerticalDrawRecord> drawRecords = new List<VerticalDrawRecord>();
  414. List<VerticalRelationRecord> relationRecords = new List<VerticalRelationRecord>();
  415. #region 标高绘图数据整理
  416. List<VFloor> floors = new List<VFloor>();
  417. for (int i = 0; i < levelDatas.Count; i++)
  418. {
  419. var tempFloor = new VFloor();
  420. tempFloor.Name = levelDatas[i].Name;
  421. if (tempFloor.Name == "F1")
  422. {
  423. tempFloor.ValueDescription = "G";
  424. }
  425. tempFloor.Index = i;
  426. tempFloor.FloorId = i.ToString();
  427. tempFloor.LinkId = levelDatas[i].Id;
  428. floors.Add(tempFloor);
  429. }
  430. #endregion
  431. #region 立管或相当于立管的空间处理
  432. /*
  433. * 按系统分组,[一个空间会不会对应多个系统]
  434. */
  435. var comparer = new VpComparer();
  436. int groupId = 0;
  437. //按系统分组
  438. var goupSystems = pipeDatas.GroupBy(p => p.PipeSytemType);
  439. foreach (var groupSystem in goupSystems)
  440. {
  441. //按坐标分组
  442. foreach (var groupPosition in groupSystem.GroupBy(p=>p.DownPoint3D, comparer))
  443. {
  444. var currentGroupdId = ++groupId ;
  445. SetRelationItem set = new SetRelationItem();
  446. #region 处理绘图数据
  447. foreach (var groupFloor in groupPosition.GroupBy(p => p.LevelName))
  448. {
  449. var verData = groupFloor.FirstOrDefault();
  450. string floorId = floors.FirstOrDefault(c => c.Name == groupFloor.Key)?.FloorId;
  451. #region 生成存储核心数据
  452. VPipe pipe = new VPipe();
  453. pipe.LinkId = verData.Id;
  454. pipe.Name = string.Join(",", groupFloor.Select(v => v.Display));//verData.Display;
  455. pipe.Tip = verData.DisplayTip;
  456. pipe.PipeSystemType = verData.PipeSytemType;
  457. pipe.FloorId = floorId;
  458. #endregion
  459. VerticalDrawRecord drawRecord = new VerticalDrawRecord();
  460. drawRecord.GroupId = currentGroupdId.ToString();
  461. drawRecord.RefPipe = pipe;
  462. drawRecord.SystemName = groupSystem.Key;
  463. drawRecords.Add(drawRecord);
  464. set.AddRange(groupFloor.Select(v => string.Format("{0}:{1}", floorId, v.Id)));
  465. }
  466. #endregion
  467. #region 处理关系数据
  468. VerticalRelationRecord relationRecord = new VerticalRelationRecord();
  469. relationRecord.SystemName = groupSystem.Key;
  470. relationRecord.RelationItems = set;
  471. relationRecords.Add(relationRecord);
  472. #endregion
  473. }
  474. }
  475. #endregion
  476. VerticalResult result = new VerticalResult();
  477. result.DrawData = new VerticalDrawData() { LevelDatas = floors, DrawRecords = drawRecords };
  478. result.RelationData = relationRecords;
  479. return result;
  480. }
  481. #endregion
  482. #region 创建机房绘图数据
  483. /// <summary>
  484. /// 创建机房绘图数据
  485. /// </summary>
  486. /// <param name="document"></param>
  487. /// <returns></returns>
  488. public static SystemDrawItems CreateRoomDrawing(GplotDocument document)
  489. {
  490. List<SystemCurveItem> curveItems = new List<SystemCurveItem>();
  491. List<SystemPointItem> pointItems = new List<SystemPointItem>();
  492. foreach (var element in document.Elements)
  493. {
  494. if (element is TVertex vertex)
  495. {
  496. SystemPointItem point = new SystemPointItem();
  497. point.Point = vertex.Location;
  498. point.Name = vertex.Name;
  499. point.IsVirtual = vertex.VertexType != Models.GplotElement.VertexType.Solid;
  500. pointItems.Add(point);
  501. continue;
  502. }
  503. if (element is TLine line)
  504. {
  505. SystemCurveItem curveItem = new SystemCurveItem();
  506. curveItem.Points.Add(line.StartPoint);
  507. curveItem.Points.Add(line.EndPoint);
  508. curveItems.Add(curveItem);
  509. continue;
  510. }
  511. if (element is TText text)
  512. {
  513. continue;
  514. }
  515. }
  516. SystemDrawItems drawItems = new SystemDrawItems();
  517. drawItems.Curves = curveItems;
  518. drawItems.Points = pointItems;
  519. return drawItems;
  520. }
  521. #endregion
  522. }
  523. /// <summary>
  524. /// 点位置比较相关
  525. /// </summary>
  526. public class VpComparer : IEqualityComparer<Point3D>
  527. {
  528. /// <summary>
  529. /// 立管左右错开小于2Cm
  530. /// </summary>
  531. /// <param name="x"></param>
  532. /// <param name="y"></param>
  533. /// <returns></returns>
  534. public bool Equals(Point3D x, Point3D y)
  535. {//数据为英寸
  536. var diff = 2 * 10d / 304.8;
  537. return Math.Abs(x.X - y.X) <= diff && Math.Abs(x.Y - y.Y) <= diff;
  538. }
  539. public int GetHashCode(Point3D obj)
  540. {
  541. return 0;
  542. }
  543. }
  544. }