RevitManager.cs 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148
  1. using Microsoft.Win32;
  2. using System;
  3. using System.Collections.Generic;
  4. using System.IO;
  5. using System.Linq;
  6. using System.Text;
  7. using System.Text.RegularExpressions;
  8. using System.Threading.Tasks;
  9. namespace LRH
  10. {
  11. /// <summary>
  12. /// revit相关管理
  13. /// </summary>
  14. public class RevitManager
  15. {
  16. #region 私有设置
  17. private const string RevitNameMatch = @"^20\d{2}$";
  18. private const string ExeName = "Revit.exe";
  19. private static Dictionary<string, RevitItem> m_DicRevitItems = new Dictionary<string, RevitItem>();
  20. static RevitManager()
  21. {
  22. LoadRevitItems();
  23. }
  24. /// <summary>
  25. /// 初始化所有安装revit项目
  26. /// </summary>
  27. private static void LoadRevitItems()
  28. {
  29. List<RevitItem> revitItems = new List<RevitItem>();
  30. string baseKey = string.Format("SOFTWARE\\Autodesk\\Revit");
  31. //LocalMachine key值
  32. var useRegistryView = Environment.Is64BitOperatingSystem ? RegistryView.Registry64 : RegistryView.Registry32;
  33. var localMachineKey = RegistryKey.OpenBaseKey(RegistryHive.LocalMachine, useRegistryView);
  34. #region 查找注册表安装路径
  35. using (RegistryKey registryKey = localMachineKey.OpenSubKey(baseKey))
  36. {
  37. if (registryKey != null)
  38. {
  39. //找到形如(2018,2019)的键值
  40. var revitVersions = registryKey.GetSubKeyNames().Where(n => CheckRevitName(n)).ToList();
  41. foreach (var version in revitVersions)
  42. {
  43. var versionKey = registryKey.OpenSubKey(version);
  44. //找到包含REVIT名称的键值
  45. var subKeyName = versionKey.GetSubKeyNames().FirstOrDefault(n => n.ToUpper().StartsWith("REVIT"));
  46. if (string.IsNullOrWhiteSpace(subKeyName))
  47. {
  48. continue;
  49. }
  50. var tempKey = versionKey.OpenSubKey(subKeyName);
  51. if (tempKey != null)
  52. {
  53. #region 初始化RevitItem
  54. var installationLocation = tempKey.GetValue("InstallationLocation") as string ?? string.Empty;
  55. var revitFile = Path.Combine(installationLocation, ExeName);
  56. if (File.Exists(revitFile))
  57. {
  58. RevitItem item = new RevitItem(version, revitFile);
  59. revitItems.Add(item);
  60. }
  61. #endregion
  62. }
  63. }
  64. registryKey.Close();
  65. }
  66. }
  67. #endregion
  68. revitItems.ForEach(revit => m_DicRevitItems[revit.Name]=revit);
  69. }
  70. #endregion
  71. /// <summary>
  72. /// 获取所有Revit项目
  73. /// </summary>
  74. /// <returns></returns>
  75. public static List<RevitItem> GetRevitItems()
  76. {
  77. return m_DicRevitItems.Values.ToList();
  78. }
  79. /// <summary>
  80. /// 获取指定名称的Revit项目
  81. /// </summary>
  82. /// <param name="revitName"></param>
  83. /// <returns></returns>
  84. public static RevitItem GetRevitItem(string revitName)
  85. {
  86. m_DicRevitItems.TryGetValue(revitName, out RevitItem revitItem);
  87. return revitItem;
  88. }
  89. /// <summary>
  90. /// 检查Revit名称格式是否正确
  91. /// </summary>
  92. /// <param name="revitName"></param>
  93. /// <returns></returns>
  94. public static bool CheckRevitName(string revitName)
  95. {
  96. return Regex.IsMatch(revitName, RevitNameMatch);
  97. }
  98. /// <summary>
  99. /// 获取插件放置公共目录
  100. /// </summary>
  101. /// <param name="revitItemName"></param>
  102. /// <returns></returns>
  103. public static string GetAllUsersAddInFolder(string revitItemName)
  104. {
  105. return Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.CommonApplicationData), @"Autodesk\Revit\AddIns", revitItemName);
  106. }
  107. /// <summary>
  108. /// 获取插件放置当前用户目录
  109. /// </summary>
  110. /// <param name="revitItemName"></param>
  111. /// <returns></returns>
  112. public static string GetCurrentUserAddInFolder(string revitItemName)
  113. {
  114. return Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData), @"Autodesk\Revit\AddIns", revitItemName);
  115. }
  116. }
  117. /// <summary>
  118. /// Revit安装信息
  119. /// </summary>
  120. public class RevitItem
  121. {
  122. internal RevitItem(string name,string revitFile)
  123. {
  124. Name = name;
  125. RevitFile = revitFile;
  126. AllUsersAddInFolder = RevitManager.GetAllUsersAddInFolder(name);
  127. CurrentUserAddInFolder= RevitManager.GetCurrentUserAddInFolder(name);
  128. }
  129. /// <summary>
  130. /// Revit名称(2018,2019)
  131. /// </summary>
  132. public string Name { get; private set; }
  133. /// <summary>
  134. /// revit.exe所在位置
  135. /// </summary>
  136. public string RevitFile { get;private set; }
  137. /// <summary>
  138. /// 所有用户插件目录
  139. /// </summary>
  140. public string AllUsersAddInFolder { get; private set; }
  141. /// <summary>
  142. /// 当前用户插件目录
  143. /// </summary>
  144. public string CurrentUserAddInFolder { get; private set; }
  145. }
  146. }