功能更新

ShrlAlgoToolkit.RevitCore/Base/BaseApplication.cs

@@ -10,11 +10,14 @@ namespace ShrlAlgoToolkit.RevitCore.Assists
 {
     public abstract class BaseApplication : IExternalApplication
     {
-        public UIControlledApplication UiApplication { get; private set; }
-
+        public UIApplication UiApplication { get; private set; }
+       
         public Result OnStartup(UIControlledApplication application)
         {
-            UiApplication = application;
+            var flag = System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance | System.Reflection.BindingFlags.InvokeMethod;
+
+            UiApplication = (Autodesk.Revit.UI.UIApplication)application.GetType().InvokeMember("getUIApplication", flag, Type.DefaultBinder, application, null);
+
 
             // 1. 注册程序集依赖解析事件 (解决第三方库找不到的问题)
             AppDomain.CurrentDomain.AssemblyResolve += OnAssemblyResolve;