Allow overload inference for polymorphic operators

bonsai-rx · Mar 17, 2023 · 5ac597a · 5ac597a
1 parent 5c9c041
commit 5ac597a
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/Bonsai.Design/TypeMappingEditor.cs b/Bonsai.Design/TypeMappingEditor.cs
@@ -51,10 +51,16 @@ public override object EditValue(ITypeDescriptorContext context, IServiceProvide
                         var intersection = default(HashSet<Type>);
                         foreach (var successor in builderNode.Successors)
                         {
-                            var combinatorBuilder = ExpressionBuilder.Unwrap(successor.Target.Value) as CombinatorBuilder;
-                            if (combinatorBuilder == null || combinatorBuilder.Combinator == null) continue;
+                            var combinator = ExpressionBuilder.Unwrap(successor.Target.Value) switch
+                            {
+                                CombinatorBuilder combinatorBuilder => combinatorBuilder.Combinator,
+                                ICustomTypeDescriptor typeDescriptor =>
+                                    TypeDescriptor.GetDefaultProperty(typeDescriptor.GetType())?.GetValue(typeDescriptor),
+                                _ => null
+                            };
 
-                            var inputTypes = from method in GetProcessMethods(combinatorBuilder.Combinator.GetType())
+                            if (combinator == null) continue;
+                            var inputTypes = from method in GetProcessMethods(combinator.GetType())
                                              where !method.IsGenericMethod
                                              let parameters = method.GetParameters()
                                              where parameters.Length > successor.Label.Index &&