添加项目文件。

Sai.Toolkit.Revit/Helpers/Filters.cs

@@ -0,0 +1,131 @@
+using Autodesk.Revit.DB;
+using Autodesk.Revit.UI.Selection;
+
+namespace Sai.Toolkit.Revit.Helpers;
+
+/// <summary>
+/// 任意选择过滤器
+/// </summary>
+public class FuncFilter(Func<Element, bool> allowElement, Func<Reference, XYZ, bool> allowReference = null)
+    : ISelectionFilter
+{
+    private readonly Func<Element, bool> elementFunc = allowElement;
+    private readonly Func<Reference, XYZ, bool> referenceFunc = allowReference;
+
+    public bool AllowElement(Element elem)
+    {
+        return elementFunc(elem);
+    }
+
+    public bool AllowReference(Reference reference, XYZ position)
+    {
+        return referenceFunc == null || referenceFunc(reference, position);
+    }
+}
+
+/// <summary>
+///     类型过滤
+/// </summary>
+/// <typeparam name="T"></typeparam>
+public class GenericFilter<T> : ISelectionFilter
+{
+    public bool AllowElement(Element elem)
+    {
+        return elem is T;
+    }
+
+    public bool AllowReference(Reference reference, XYZ position)
+    {
+        return true;
+    }
+}
+
+/// <summary>
+///     获取当前模型或链接模型的实体元素
+/// </summary>
+public class ElementInLinkOrCurrentDocument : ISelectionFilter
+{
+    public ElementInLinkOrCurrentDocument(Document doc)
+    {
+        this.doc = doc;
+    }
+
+    private readonly Document doc;
+
+    public bool LastCheckedWasFromLink => null != LinkedDocument;
+
+    public Document LinkedDocument { get; private set; }
+
+    public bool AllowElement(Element e)
+    {
+        return true;
+    }
+
+    public bool AllowReference(Reference r, XYZ p)
+    {
+        LinkedDocument = null;
+
+        var e = doc.GetElement(r);
+
+        if (e is RevitLinkInstance li)
+        {
+            LinkedDocument = li.GetLinkDocument();
+
+            e = LinkedDocument.GetElement(r.LinkedElementId);
+        }
+
+        return e != null
+            && e.CanHaveTypeAssigned()
+            && e.HasPhases()
+            && e.get_BoundingBox(null) != null
+            && e.Category is { Parent: null }
+            && e is not Panel;
+    }
+}
+
+public class DwgBlockSelection : ISelectionFilter
+{
+    private Element e;
+
+    public bool AllowElement(Element elem)
+    {
+        e = elem;
+        return e.Document.GetElement(e.GetTypeId()) is CADLinkType;
+    }
+
+    public bool AllowReference(Reference reference, XYZ position)
+    {
+        //块
+        var instance = e.GetGeometryObjectFromReference(reference) as GeometryInstance;
+        if (instance == null)
+        {
+            return false;
+        }
+        ////dwg
+        //foreach (var dwgIns in e.get_Geometry(new Options()))
+        //{
+        //    if (dwgIns is not GeometryInstance item) continue;
+        //    //遍历dwg包含的所有内容
+        //    foreach (var obj in item.SymbolGeometry)
+        //    {
+        //        return obj is GeometryInstance ins && !ins.SymbolGeometry.Any();
+        //    }
+        //}
+
+        //if (selectBlock == null)
+        //{
+        //    return false;
+        //}
+        //块是否包含子块
+        //foreach (var item in instance.SymbolGeometry)
+        //{
+        //    if (item is GeometryInstance)
+        //    {
+        //        return false;
+        //    }
+        //}
+        //var isNest = selectBlock.SymbolGeometry.OfType<GeometryInstance>().Any();
+        return /*!isNest &&*/ reference.ElementReferenceType == ElementReferenceType.REFERENCE_TYPE_INSTANCE
+            && instance.GraphicsStyleId != ElementId.InvalidElementId;
+    }
+}