revit二次开发——如何选取元素（revit2016）

revit二次开发——如何选取元素（revit2016）

这是我在网上和书中比较写出的代码，可能不是太成熟，希望大家多多指教

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Autodesk.Revit.UI;
using Autodesk.Revit.DB;
using Autodesk.Revit.UI.Selection;

namespace 如何选取Revit元素
{
        [Autodesk.Revit.Attributes.Transaction(Autodesk.Revit.Attributes.TransactionMode.Automatic)]
        public class Class1 : IExternalCommand
        {
            public Autodesk.Revit.UI.Result Execute(ExternalCommandData revit, ref string message, ElementSet elements)
            {
                UIDocument uidoc = revit.Application.ActiveUIDocument;
                Document doc = uidoc.Document;
                Selection selection = uidoc.Selection;


                try
                {
                    //共有两种选择方式
                    //1，获取Revit已选中的元素
                    String info = "所选元素类型为: ";
                    foreach (var elementId in selection.GetElementIds())
                    {
                        var elem = doc.GetElement(elementId);
                        if (elem != null)
                        {
                            info += "\n\t" + elem.GetType().ToString();
                        }
                    }
                    TaskDialog.Show("Revit已选中的元素为：", info);


                    //2，用户自行选择元素
                    var referenceCollection = selection.PickObjects(ObjectType.Element, "请选择模型构件");
                    if (0 == referenceCollection.Count)
                    {
                        TaskDialog.Show("Revit", "你没有选任何元素.");
                    }
                    else
                    {
                        info = "所选元素类型为: ";
                        foreach (var reference in referenceCollection)
                        {
                            var elem = doc.GetElement(reference);
                            if (elem != null)
                            {
                                info += "\n\t" + elem.GetType().ToString();
                            }
                        }
                        TaskDialog.Show("你所选择的元素为", info);
                    }
                }
                catch (Exception e)
                {
                    message = e.Message;
                    return Autodesk.Revit.UI.Result.Failed;
                }
                return Autodesk.Revit.UI.Result.Succeeded;
            }
        }
}