Add Family Size Table support to Family Document

jeremytammik · May 18, 2024 · 31aa432 · 31aa432
1 parent a962dc9
commit 31aa432
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/source/RevitLookup/Core/ComponentModel/Descriptors/DocumentDescriptor.cs b/source/RevitLookup/Core/ComponentModel/Descriptors/DocumentDescriptor.cs
@@ -24,7 +24,7 @@
 
 namespace RevitLookup.Core.ComponentModel.Descriptors;
 
-public sealed class DocumentDescriptor : Descriptor, IDescriptorResolver
+public sealed class DocumentDescriptor : Descriptor, IDescriptorResolver, IDescriptorExtension
 {
     private readonly Document _document;
 
@@ -71,4 +71,14 @@ IVariants ResolveGetAllUnusedElements()
         }
 #endif
     }
+
+    public void RegisterExtensions(IExtensionManager manager)
+    {
+        if (!_document.IsFamilyDocument) return;
+        manager.Register(nameof(FamilySizeTableManager.GetFamilySizeTableManager), context =>
+        {
+            var result = FamilySizeTableManager.GetFamilySizeTableManager(_document, new ElementId(BuiltInParameter.RBS_LOOKUP_TABLE_NAME));
+            return Variants.Single(result);
+        });
+    }
 }