Allow constructing values from any source sequence

bonsai-rx · Dec 27, 2023 · 86468dd · 86468dd
1 parent 499ae29
commit 86468dd
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 1 deletion.
diff --git a/Bonsai.Sgen/Bonsai.Sgen.csproj b/Bonsai.Sgen/Bonsai.Sgen.csproj
@@ -4,7 +4,7 @@
     <OutputType>Exe</OutputType>
     <TargetFramework>net6.0</TargetFramework>
     <ImplicitUsings>enable</ImplicitUsings>
-    <VersionPrefix>0.1.0</VersionPrefix>
+    <VersionPrefix>0.2.0</VersionPrefix>
     <Nullable>enable</Nullable>
   </PropertyGroup>
 

diff --git a/Bonsai.Sgen/CSharpClassTemplate.cs b/Bonsai.Sgen/CSharpClassTemplate.cs
@@ -153,7 +153,37 @@ public string Render()
             {{
 {propertyAssignments}
             }})"))));
+
+                var genericTypeParameter = new CodeTypeParameter("TSource");
+                var genericSourceParameter = new CodeParameterDeclarationExpression(new CodeTypeReference(typeof(IObservable<>))
+                {
+                    TypeArguments = { new CodeTypeReference(genericTypeParameter) }
+                }, "source");
+                var genericProcessMethod = new CodeMemberMethod
+                {
+                    Name = "Process",
+                    Attributes = MemberAttributes.Public | MemberAttributes.Final,
+                    TypeParameters = { genericTypeParameter },
+                    Parameters = { genericSourceParameter },
+                    ReturnType = new CodeTypeReference(typeof(IObservable<>))
+                    {
+                        TypeArguments = { new CodeTypeReference(Model.ClassName) }
+                    }
+                };
+                genericProcessMethod.Statements.Add(new CodeMethodReturnStatement(
+                    new CodeMethodInvokeExpression(
+                        new CodeMethodReferenceExpression(
+                            new CodeTypeReferenceExpression("System.Reactive.Linq.Observable"),
+                            "Select"),
+                        new CodeVariableReferenceExpression(genericSourceParameter.Name),
+                        new CodeSnippetExpression(
+                            @$"_ => new {Model.ClassName}
+            {{
+{propertyAssignments}
+            }}"))));
+
                 type.Members.Add(processMethod);
+                type.Members.Add(genericProcessMethod);
                 type.CustomAttributes.Add(new CodeAttributeDeclaration(
                     new CodeTypeReference("Bonsai.CombinatorAttribute")));
                 type.CustomAttributes.Add(new CodeAttributeDeclaration(