一些细节修改

xk6338062 · Apr 14, 2023 · 777c804 · 777c804
1 parent c547cb4
commit 777c804
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 9 deletions.
diff --git a/ChatRevit/AppLoad.cs b/ChatRevit/AppLoad.cs
@@ -10,10 +10,10 @@ public class AppLoad : ApplicationBase
     {
         public override void RegisterTypes()
         {
-            Services.AddScoped<IDockablePaneService, ChatRevitService>();
+            Services.AddSingleton<IDockablePaneService, ChatRevitService>();
             Services.AddSingleton<IApplicationUI, AppUIManager>();
-            Services.AddScoped(typeof(ChatRevitPanel));
-            Services.AddScoped(typeof(ChatRevitViewModel));
+            Services.AddSingleton(typeof(ChatRevitPanel));
+            Services.AddSingleton(typeof(ChatRevitViewModel));
             Ioc.Default.ConfigureServices(
                 Services
                 .BuildServiceProvider());

diff --git a/ChatRevit/AppUIManager.cs b/ChatRevit/AppUIManager.cs
@@ -17,6 +17,9 @@ public Result Initial()
             var provider = Ioc.Default.GetService<IDockablePaneService>();
             var paneId = provider.GetDockablePaneId();
             application.RegisterDockablePane(paneId, "ChatRevit", provider);
+            //var pane = application.GetDockablePane(paneId);
+            //if (!pane.IsShown())
+            //    pane.Show();
             return Result.Succeeded;
 
         }

diff --git a/ChatRevit/ChatRevitService.cs b/ChatRevit/ChatRevitService.cs
@@ -12,7 +12,7 @@ namespace ChatRevit
 {
     public class ChatRevitService : IDockablePaneService
     {
-        public static string PageGuid = "3C3F2CCB-CA0B-4427-AACE-2694B685A4B2";
+        public static string PageGuid = "A811F969-2114-4AC8-A289-F5A8982946CE";
 
         public ChatRevitService()
         {
@@ -21,6 +21,7 @@ public ChatRevitService()
         public FrameworkElement GetDockablePane()
         {
             var mainPage = Ioc.Default.GetService<ChatRevitPanel>();
+            mainPage.DataContext = Ioc.Default.GetService<ChatRevitViewModel>();
             return mainPage;
         }
 

diff --git a/ChatRevit/ChatRevitViewModel.cs b/ChatRevit/ChatRevitViewModel.cs
@@ -18,6 +18,7 @@
 using System.Windows;
 using Microsoft.CodeAnalysis.CSharp;
 using Microsoft.CodeAnalysis.CSharp.Syntax;
+using Autodesk.Revit.UI;
 
 namespace ChatRevit
 {
@@ -81,7 +82,7 @@ private async void Enter()
                     {
                         using var tran = new Transaction(document, "chat");
                         tran.Start();
-                        execute(document, response);
+                        execute(document, app.ActiveUIDocument, response);
                         tran.Commit();
                     }
                     catch (Exception e)
@@ -111,8 +112,8 @@ private string GetPureCode(string response)
 
         string systemPersonality => "I want you to act as a Revit Developer. Now Write a Revit C# script of a method.\n" +
                  " - This method provides its functionality as a public function named 'RevitFunction'.\n " +
-                 " - This method has only one parameter called 'document' which is passed in from the previous code " +
-                 "and the type of parameter 'document' is Document.\n " +
+                 " - This method has only two parameter called 'document' and 'uiDocument' which is passed in from the previous code " +
+                 "and the type of parameter 'document' is Document, the type of parameter 'uiDocument' is UIDocument.\n " +
                  " - This method only contains methods available in RevitAPI.\n" +
                  " - This method does not need Transaction.\n" +
                  " - This method immediately does the task when the function is invoked.\n" +
@@ -123,7 +124,7 @@ string WrapPrompt(string input)
                 =>
                  "The task is described as follows:\n" + input;
 
-        private void execute(Document doc, string codeOutput)
+        private void execute(Document doc, UIDocument uiDoc, string codeOutput)
         {
             var codeString = "using System;\n " +
                              "using System.IO;\n " +
@@ -173,7 +174,7 @@ private void execute(Document doc, string codeOutput)
             object obj = assembly.CreateInstance("Chat.ChatWithRevit");
 
             // 执行方法
-            obj.GetType().GetMethod("RevitFunction").Invoke(obj, new object[] { doc });
+            obj.GetType().GetMethod("RevitFunction").Invoke(obj, new object[] { doc, uiDoc });
         }
     }
 }