整理代码

ShrlAlgoToolkit.RevitAddins/Converters/EnumDescriptionExtension.cs

@@ -0,0 +1,32 @@
+using System;
+using System.ComponentModel;
+using System.Linq;
+using System.Windows.Markup;
+
+using ShrlAlgoToolkit.RevitAddins.Converters;
+using ShrlAlgoToolkit;
+using ShrlAlgoToolkit.Mvvm;
+
+namespace ShrlAlgoToolkit.RevitAddins.Converters
+{
+    public class EnumDescriptionExtension : MarkupExtension
+    {
+        private readonly object _enumValue;
+
+        public EnumDescriptionExtension(object enumValue)
+        {
+            _enumValue = enumValue;
+        }
+
+        public override object ProvideValue(IServiceProvider serviceProvider)
+        {
+            if (_enumValue == null) return string.Empty;
+
+            var field = _enumValue.GetType().GetField(_enumValue.ToString());
+            var attribute = field?.GetCustomAttributes(typeof(DescriptionAttribute), false)
+                            .FirstOrDefault() as DescriptionAttribute;
+
+            return attribute?.Description ?? _enumValue.ToString();
+        }
+    }
+}