zhangyuekai
2024-08-08 cd7d3ceff46090b1df76b96b788c2ffedb6f99f5
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
using Autodesk.Revit.DB;
using HStation.Model;
using HStation.RevitDev.Model.AttributeClass;
using HStation.RevitDev.Model.Common;
using HStation.RevitDev.Model.ModelEnum;
using HStation.RevitDev.RevitDataExport.Common;
using Newtonsoft.Json.Linq;
using System;
using System.Reflection;
 
namespace HStation.RevitDev.RevitDataExport.Utility
{
    public class ExportModelHelper
    {
        Document m_doc;
 
        public ExportModelHelper(Document doc)
        {
            m_doc = doc;
        }
 
        public string Export()
        {
            var modelMap = GlobalResource.RevitModels;
            JArray retArray = new JArray();
            foreach (var modelPair in modelMap)
            {
                var revitType = modelPair.Key;
                var jArray = ExportRevitModel(revitType);
                retArray.Merge(jArray);
            }
            return retArray.ToString();
        }
 
        private string GetRevitModelPropertyName(RevitType revitType)
        {
            Type revitModelType = typeof(RevitModel);
            ExportType expType = ElementTypeMapping.GetExportType(revitType);
            var props = revitModelType.GetProperties();
            foreach (var prop in props)
            {
                var attr = prop.GetCustomAttribute<ExportTypeAttribute>();
                if (attr == null) { continue; }
 
                var attrExpType = attr.m_exportType;
                if (expType == attrExpType)
                {
                    return prop.Name;
                }
            }
            return null;
        }
 
        private JArray ExportRevitModel(RevitType revitType)
        {
            var revitModels = GlobalResource.RevitModels[revitType];
            var outElems = new JArray();
 
            foreach (var model in revitModels)
            {
                ElementId id = null;
                var elemId = model.Id;
                if (int.TryParse(elemId, out int elementId))
                {
                    id = new ElementId(elementId);
                }
                Element rvtElem = m_doc.GetElement(id);
                if (rvtElem == null) { continue; }
 
                //JObject outElem = rvtElem.Export(revitType);
                JObject outElem = rvtElem.Export2();
                outElem["构件类型"] = revitType.GetDescription();
                outElems.Add(outElem);
            }
            return outElems;
        }
    }
}