优化代码

zhangyuekai

2024-08-08 1a0f5a002c0c513c8d48be3b382c1e01242642b3

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
using Autodesk.Revit.DB; 
using Autodesk.Revit.DB.Plumbing; 
using Autodesk.Revit.UI; 
using HStation.RevitDev.RevitDataExport.Common; 
using System.Collections.Generic; 
using System.Linq; 
 
namespace HStation.RevitDev.RevitDataExport.Utility 
{ 
    public class SystemCheckUtils 
    { 
        public static void SystemCheck(ExternalCommandData data) 
        { 
            UIDocument uidoc = data.Application.ActiveUIDocument; 
            Document doc = uidoc.Document; 
            List<ElementId> ids = new List<ElementId>(); 
            foreach (var id in GlobalResource.RevitModels.GetIds()) 
            { 
                var element = doc.GetElement(new ElementId(int.Parse(id))); 
                if (element == null) { continue; } 
 
                bool checkResult = CheckElement(element); 
 
                if (!checkResult) 
                { 
                    ids.Add(new ElementId(int.Parse(id))); 
                } 
            } 
            uidoc.Selection.SetElementIds(ids); 
        } 
 
        private static bool CheckElement(Element element) 
        { 
            //管道两端都必须有节点构件连接 
            if (element is Pipe pipe) 
            { 
                var connectors = pipe.ConnectorManager.Connectors; 
                foreach (Connector connector in connectors) 
                { 
                    var connectElems = connector.GetConnectorElements(); 
                    if (connectElems.Count < 1) 
                    { 
                        return false; 
                    } 
 
                    var fi = connectElems[0] as FamilyInstance; 
                    if (fi == null) { return false; } 
 
                    foreach (var elem in connectElems) 
                    { 
                        var id = elem.Id.IntegerValue.ToString(); 
                        if (!GlobalResource.RevitModels.Contains(id)) 
                        { 
                            return false; 
                        } 
                    } 
                } 
                return true; 
            } 
 
            //节点不能连接节点 
            else if (element is FamilyInstance fi) 
            { 
                var mepModel = fi.MEPModel; 
                if (mepModel == null) { return true; } 
 
                ConnectorSet connectors = mepModel.ConnectorManager?.Connectors; 
                if (connectors == null) { return true; } 
 
                foreach(Connector connector in connectors) 
                { 
                    var connectElems = connector.GetConnectorElements(); 
                    foreach (var connectElem in connectElems) 
                    { 
                        if (connectElem is FamilyInstance) 
                        { 
                            return false; 
                        } 
                    } 
                } 
                return true; 
            } 
            else 
            { 
                return false; 
            } 
        } 
    } 
}