using Autodesk.Revit.DB;
|
using Autodesk.Revit.DB.Events;
|
using Autodesk.Revit.DB.Plumbing;
|
using Autodesk.Revit.UI;
|
using Autodesk.Revit.UI.Events;
|
using DevExpress.Utils.Extensions;
|
using HStation.RevitDev.Model.ModelEnum;
|
using HStation.RevitDev.RevitDataExport.Common;
|
using HStation.RevitDev.RevitDataExport.Entity;
|
using HStation.RevitDev.RevitDataExport.Service;
|
using System;
|
using System.Collections.Generic;
|
using System.Collections.ObjectModel;
|
using System.Linq;
|
|
namespace HStation.RevitDev.RevitDataExport.Utility
|
{
|
public static class DocumentUtil
|
{
|
public static void RegistDocumentEvent(ExternalCommandData commandData)
|
{
|
commandData.Application.Application.DocumentChanged -= App_DocumentChanged;
|
commandData.Application.Application.DocumentChanged += App_DocumentChanged;
|
|
GlobalResource.CurrentUIDocument = commandData.Application.ActiveUIDocument;
|
RegistViewEvent(commandData);
|
}
|
|
private static void App_DocumentChanged(object sender, DocumentChangedEventArgs e)
|
{
|
var doc = e.GetDocument();
|
try
|
{
|
var addedIds = e.GetAddedElementIds();
|
var removeIds = e.GetDeletedElementIds();
|
var modifiedIds = e.GetModifiedElementIds();
|
|
AddElements(doc, addedIds);
|
RemoveElements(removeIds);
|
ModifyElements(modifiedIds);
|
|
AddUnhandledElements();
|
GlobalResource.InstancePanel.UpdateForm();
|
}
|
catch (Exception ex)
|
{
|
TaskDialog.Show("错误", ex.Message);
|
}
|
}
|
|
private static void UpdateLinkerInfo(Document doc)
|
{
|
if (!doc.IsValidObject) { return; }
|
|
foreach (var pair in GlobalResource.RevitModels)
|
{
|
foreach (var model in pair.Value)
|
{
|
var id = model.Id;
|
var element = doc.GetElement(new ElementId(int.Parse(id)));
|
var linkedElems = element.GetConnectElements();
|
var linkedIds = linkedElems.Select(x => x.Id.IntegerValue.ToString());
|
model.LinkIds = string.Join(",", linkedIds);
|
}
|
}
|
}
|
|
private static void ModifyElements(ICollection<ElementId> modifiedIds)
|
{
|
foreach (var id in modifiedIds)
|
{
|
|
}
|
}
|
|
private static void RemoveElements(ICollection<ElementId> removeIds)
|
{
|
foreach (var id in removeIds)
|
{
|
var strId = id.IntegerValue.ToString();
|
if (GlobalResource.RevitModels.Contains(strId))
|
{
|
GlobalResource.RevitModels.Remove(id.IntegerValue.ToString());
|
foreach (var pair in GlobalResource.RevitModels)
|
{
|
var values = pair.Value;
|
var model = values.Where(x => x.Id == strId)?.ToList();
|
if (model == null || model.Count() == 0)
|
{
|
continue;
|
}
|
else
|
{
|
values.Remove(model[0]);
|
}
|
}
|
}
|
}
|
|
return;
|
}
|
|
private static void AddElements(Document doc, ICollection<ElementId> addedIds)
|
{
|
if (GlobalResource.PlacingType == RevitType.RFT_Pipe)
|
{
|
foreach (var id in addedIds)
|
{
|
var elem = doc.GetElement(id);
|
if (elem is FamilyInstance fi)
|
{
|
AddAutoCreated(fi);
|
}
|
if (elem is Pipe)
|
{
|
GlobalResource.RevitModels.Add(elem, RevitType.RFT_Pipe);
|
}
|
}
|
}
|
else if (GlobalResource.PlacingType != RevitType.RFT_Unknown)
|
{
|
foreach (var id in addedIds)
|
{
|
var elem = doc.GetElement(id);
|
if (elem is FamilyInstance)
|
{
|
GlobalResource.RevitModels.Add(elem, GlobalResource.PlacingType);
|
}
|
}
|
}
|
else
|
{
|
bool isPumpSystemElement = false;
|
foreach (var id in addedIds)
|
{
|
if (IsConnectedWithPumpSystem(doc, id))
|
{
|
isPumpSystemElement = true;
|
break;
|
}
|
}
|
|
if (isPumpSystemElement)
|
{
|
foreach (var id in addedIds)
|
{
|
AddToPumpSystem(doc, id);
|
}
|
}
|
}
|
return;
|
}
|
|
private static bool IsConnectedWithPumpSystem(Document doc, ElementId id)
|
{
|
var elem = doc.GetElement(id);
|
if (elem is Pipe pipe) //直接点击水泵的连接件创建管道
|
{
|
if (pipe.ConnectWithPumpSystem())
|
{
|
return true;
|
}
|
}
|
else if (elem is FamilyInstance fi)
|
{
|
if (fi.ConnectWithPumpSystem()) //直接点击管道连接点创建管道时，Revit会自动创建弯头，都需要加入到缓存
|
{
|
return true;
|
}
|
}
|
return false;
|
}
|
|
private static void AddToPumpSystem(Document doc, ElementId id)
|
{
|
var elem = doc.GetElement(id);
|
if (elem is Pipe)
|
{
|
GlobalResource.RevitModels.Add(elem, RevitType.RFT_Pipe);
|
}
|
else if (elem is FamilyInstance fi)
|
{
|
AddAutoCreated(fi);
|
}
|
return;
|
}
|
|
private static void AddAutoCreated(FamilyInstance fi)
|
{
|
if (fi.IsWanTou())
|
{
|
GlobalResource.RevitModels.Add(fi, RevitType.RFT_Elbow);
|
}
|
else if (fi.IsConverter())
|
{
|
GlobalResource.RevitModels.Add(fi, RevitType.RFT_Converter);
|
}
|
else if (fi.IsSanTong())
|
{
|
GlobalResource.RevitModels.Add(fi, RevitType.RFT_ThreeJoint);
|
}
|
else if (fi.IsSiTong())
|
{
|
GlobalResource.RevitModels.Add(fi, RevitType.RFT_FourJoint);
|
}
|
else
|
{
|
TaskDialog.Show("警告", $"未知类型被添加！id:{fi.Id.IntegerValue}");
|
//报错
|
}
|
}
|
|
public static void RegistViewEvent(ExternalCommandData commandData)
|
{
|
commandData.Application.ViewActivated -= Application_ViewActivated;
|
commandData.Application.ViewActivated += Application_ViewActivated;
|
}
|
|
private static void Application_ViewActivated(object sender, ViewActivatedEventArgs e)
|
{
|
GlobalResource.CurrentDocument = e.CurrentActiveView.Document;
|
}
|
|
public static List<Element> GetModelElements(this Document doc)
|
{
|
if (!doc.IsValidObject)
|
{
|
return new List<Element>();
|
}
|
|
var collector1 = new FilteredElementCollector(doc).
|
OfCategory(BuiltInCategory.OST_PipeCurves).
|
WhereElementIsNotElementType();
|
|
var collector2 = new FilteredElementCollector(doc).
|
OfClass(typeof(FamilyInstance)).
|
WhereElementIsNotElementType();
|
|
var elements1 = collector1.ToElements()?.ToList();
|
var elements2 = collector2.ToElements()?.ToList();
|
elements1.AddRange(elements2);
|
|
elements1 = elements1.Where(x => x.Category.CategoryType == CategoryType.Model).ToList();
|
elements1 = elements1.Where(x => x.get_BoundingBox(null) != null).ToList();
|
|
return elements1;
|
}
|
|
public static void AddUnhandledElements()
|
{
|
var elements = GlobalResource.CurrentDocument.GetModelElements();
|
foreach (var element in elements)
|
{
|
var id = element.Id.IntegerValue.ToString();
|
if (GlobalResource.RevitModels.Contains(id))
|
{
|
continue;
|
}
|
RevitMepCategoryService.AddToCache(element);
|
}
|
UpdateLinkerInfo(GlobalResource.CurrentDocument);
|
}
|
|
public static void UpdataDocumentCache(this Document doc)
|
{
|
if (!doc.IsValidObject)
|
{
|
return;
|
}
|
if (!GlobalResource.CurrentDocument.IsValidObject || doc.GetHashCode() != GlobalResource.CurrentDocument.GetHashCode())
|
{
|
GlobalResource.CurrentDocument = doc;
|
GlobalResource.RevitModels = new Dictionary<RevitType, ObservableCollection<ElementModel>>();
|
}
|
AddUnhandledElements();
|
}
|
}
|
}
|
