271 lines
11 KiB
C#
271 lines
11 KiB
C#
using System;
|
||
using System.Linq;
|
||
using System.Text;
|
||
using System.Linq.Expressions;
|
||
using System.Collections;
|
||
using System.Collections.Generic;
|
||
using Autodesk.Revit.Attributes;
|
||
using Autodesk.Revit.ApplicationServices;
|
||
using Autodesk.Revit.Creation;
|
||
using Autodesk.Revit.DB;
|
||
using Autodesk.Revit.DB.Architecture;
|
||
using Autodesk.Revit.DB.Plumbing;
|
||
using Autodesk.Revit.UI;
|
||
using Autodesk.Revit.UI.Selection;
|
||
using Document = Autodesk.Revit.DB.Document;
|
||
using Newtonsoft.Json;
|
||
using System.IO;
|
||
using Autodesk.Revit.DB.DirectContext3D;
|
||
using System.Runtime.CompilerServices;
|
||
|
||
namespace GeoTest
|
||
{
|
||
[TransactionAttribute(TransactionMode.Manual)]
|
||
[RegenerationAttribute(RegenerationOption.Manual)]
|
||
|
||
public class TestCommand : IExternalCommand
|
||
{
|
||
public Result Execute(ExternalCommandData commandData, ref string message, ElementSet elements)
|
||
{
|
||
var applicationApp = commandData.Application.Application;
|
||
var app = commandData.Application;
|
||
var uidoc = app.ActiveUIDocument;
|
||
var doc = uidoc.Document;
|
||
var inf=GetData(doc, applicationApp);
|
||
OutPut(inf);
|
||
return Result.Succeeded;
|
||
}
|
||
|
||
private Information GetData(Document doc, Autodesk.Revit.ApplicationServices.Application app)
|
||
{
|
||
var instances = new List<Instance>();
|
||
var materials = new List<Material>();
|
||
var meshes = new List<Mesh>();
|
||
int instanceID = 0;
|
||
|
||
var option = GetGeometryOption(app);
|
||
// Find all Wall instances in the document by using category filter
|
||
ElementCategoryFilter filter = new ElementCategoryFilter(BuiltInCategory.OST_StructuralColumns);
|
||
|
||
// Apply the filter to the elements in the active document,
|
||
// Use shortcut WhereElementIsNotElementType() to find wall instances only
|
||
FilteredElementCollector collector = new FilteredElementCollector(doc);
|
||
IList<Element> walls = collector.WherePasses(filter).WhereElementIsNotElementType().ToElements();
|
||
foreach (var element in walls)
|
||
{
|
||
|
||
var materialIds = element.GetMaterialIds(false).ToList();
|
||
var thisMaterials = materialIds.Select(x => doc.GetElement(x) as Autodesk.Revit.DB.Material).ToList();
|
||
var transformList = new List<XYZ>();
|
||
foreach (var mat in thisMaterials)
|
||
{
|
||
var color = mat.Color;
|
||
var newMat = new Material(new List<int>() { color.Red,color.Green,color.Blue}, mat.Transparency);
|
||
materials.Add(newMat);
|
||
}
|
||
if (element is FamilyInstance)
|
||
{
|
||
var transform = (element as FamilyInstance).GetTotalTransform();
|
||
transformList.Add(transform.BasisX);
|
||
transformList.Add(transform.BasisY);
|
||
transformList.Add(transform.BasisZ);
|
||
}
|
||
else
|
||
{
|
||
var transform = Transform.Identity;
|
||
transformList.Add(transform.BasisX);
|
||
transformList.Add(transform.BasisY);
|
||
transformList.Add(transform.BasisZ);
|
||
}
|
||
var geoElement = element.get_Geometry(option);
|
||
|
||
foreach (var geoObject in geoElement)
|
||
{
|
||
Solid solid = geoObject as Solid;
|
||
if (null == solid || 0 == solid.Faces.Size || 0 == solid.Edges.Size)
|
||
{
|
||
continue;
|
||
}
|
||
|
||
|
||
var vertices = new List<XYZ>();
|
||
var faceVertices = new List<int>();//按顺序将每个面用3个顶点表示
|
||
var triInfo = new Dictionary<List<int>, XYZ>();
|
||
var normals = new List<XYZ>();
|
||
|
||
// Get the faces and edges from solid, and transform the formed points
|
||
foreach (Face face in solid.Faces)
|
||
{
|
||
Autodesk.Revit.DB.Mesh mesh = face.Triangulate();
|
||
var count = mesh.NumTriangles;
|
||
for(int i=0;i<count;i++)
|
||
{
|
||
var vertexIndecies = new List<int>();
|
||
var triangle = mesh.get_Triangle(i);
|
||
//normal
|
||
//(x3 - x1).CrossProduct(x2 - x1)
|
||
var thisNormal = ((triangle.get_Vertex(2) - triangle.get_Vertex(0)).CrossProduct((triangle.get_Vertex(1) - triangle.get_Vertex(0)))).Normalize();
|
||
|
||
for (int j=0;j<3;j++)
|
||
{
|
||
var vertex = triangle.get_Vertex(j);
|
||
var sameIndex=vertices.Where(x => x.DistanceTo(vertex) < 0.00001);
|
||
if (!sameIndex.Any())
|
||
{
|
||
vertices.Add(vertex);
|
||
vertexIndecies.Add(vertices.Count() - 1);
|
||
}
|
||
else
|
||
{
|
||
vertexIndecies.Add(vertices.IndexOf(sameIndex.First()));
|
||
}
|
||
}
|
||
triInfo.Add(vertexIndecies, thisNormal);
|
||
faceVertices.AddRange(vertexIndecies);
|
||
}
|
||
|
||
}
|
||
|
||
//对每个顶点寻找邻接面
|
||
//如果两面夹角小于90度，用几何均值合并normal
|
||
//如果大于90度，使用各自的normal
|
||
for (int i = 0; i < vertices.Count; i++)
|
||
{
|
||
XYZ vertex = vertices[i];
|
||
var adjacentFaces = triInfo.Where(x => x.Key.Contains(i)).ToList();
|
||
var jc = adjacentFaces.Count;
|
||
if (!adjacentFaces.Any()) continue;
|
||
for (int j = 0; j < jc; j++)
|
||
{
|
||
for (int k = j + 1; k < jc; k++)
|
||
{
|
||
if (Math.Cos(adjacentFaces[j].Value.AngleTo(adjacentFaces[k].Value)) > 0)//即<pi/2
|
||
{
|
||
//合并
|
||
var changedNormal = (adjacentFaces[j].Value + adjacentFaces[k].Value).Normalize();
|
||
var tempPair = new KeyValuePair<List<int>, XYZ>(adjacentFaces[k].Key, changedNormal);
|
||
adjacentFaces[k] = tempPair;
|
||
//回到本层开头重新循环
|
||
adjacentFaces.RemoveAt(j);
|
||
jc--;
|
||
j--;
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
normals.AddRange(adjacentFaces.Select(x => x.Value));
|
||
}
|
||
var newMesh = new Mesh(vertices, faceVertices, normals);
|
||
meshes.Add(newMesh);
|
||
var newInstance = new Instance(instanceID, transformList, solid.Visibility, meshes.Count() - 1);
|
||
instances.Add(newInstance);
|
||
instanceID++;
|
||
}
|
||
|
||
|
||
|
||
|
||
}
|
||
|
||
var information = new Information(instances, meshes, materials);
|
||
return information;
|
||
}
|
||
|
||
|
||
Options GetGeometryOption(Autodesk.Revit.ApplicationServices.Application app)
|
||
{
|
||
Autodesk.Revit.DB.Options option = app.Create.NewGeometryOptions();
|
||
option.ComputeReferences = true; //打开计算几何引用
|
||
option.DetailLevel = ViewDetailLevel.Fine; //视图详细程度为最好
|
||
return option;
|
||
|
||
}
|
||
|
||
private void OutPut(Information information)
|
||
{
|
||
string fp = System.Windows.Forms.Application.StartupPath + "\\geometryInfo.json";
|
||
var path = @"D:\geometryInfo.json";
|
||
//FileStream fs1 = new FileStream(fp, FileMode.Create, FileAccess.ReadWrite);
|
||
var instanceString = JsonConvert.SerializeObject(information);
|
||
|
||
File.WriteAllText(path, instanceString);
|
||
return;
|
||
}
|
||
|
||
|
||
|
||
}
|
||
|
||
public class Information
|
||
{
|
||
public List<Instance> instances { get; set; }
|
||
public List<Mesh> meshes { get; set; }
|
||
public List<Material> materials { get; set; }
|
||
public Information(List<Instance> instances, List<Mesh> meshes, List<Material> materials)
|
||
{
|
||
this.instances = instances;
|
||
this.meshes = meshes;
|
||
this.materials = materials;
|
||
}
|
||
}
|
||
|
||
|
||
public class Instance
|
||
{
|
||
public int id { get; set; }
|
||
public List<List<double>> transform { get; set; }
|
||
public int visible { get; set; }
|
||
public int meshId { get; set; }//meshID
|
||
public string info { get; set; }
|
||
public Instance(int id, List<XYZ> transform, Visibility visible, int meshId)
|
||
{
|
||
this.id = id;
|
||
this.transform = new List<List<double>>();
|
||
foreach(var trans in transform)
|
||
{
|
||
var tempTrans = new List<double>() { trans.X, trans.Y, trans.Z };
|
||
this.transform.Add(tempTrans);
|
||
}
|
||
this.visible = (int)visible;
|
||
this.meshId = meshId;
|
||
}
|
||
}
|
||
|
||
public class Mesh
|
||
{
|
||
public List<List<double>> vertices { get; set; }
|
||
public List<int> vertexIndices { get; set; }
|
||
public List<List<double>> normals { get; set; }//点法向量
|
||
public Mesh(List<XYZ> vertices, List<int> vertexIndices, List<XYZ> normals)
|
||
{
|
||
this.vertexIndices = vertexIndices;
|
||
this.vertices = new List<List<double>>();
|
||
foreach(var vertex in vertices)
|
||
{
|
||
var tempV = new List<double>() { vertex.X, vertex.Y, vertex.Z };
|
||
this.vertices.Add(tempV);
|
||
|
||
}
|
||
this.normals = new List<List<double>>();
|
||
foreach (var n in normals)
|
||
{
|
||
var tempN = new List<double>() { n.X, n.Y, n.Z };
|
||
this.normals.Add(tempN);
|
||
|
||
}
|
||
}
|
||
}
|
||
|
||
public class Material
|
||
{
|
||
public List<int> color { get; set; }
|
||
public double transparent { get; set; }//透明度
|
||
public Material(List<int> color, double transparent)
|
||
{
|
||
this.color = color;
|
||
this.transparent = transparent;
|
||
}
|
||
}
|
||
|
||
} |