Define a template for an operator that does all-in-one loading of base schemas.

.gitignore

@@ -0,0 +1 @@
+tmp/

copier.yml

@@ -52,13 +52,12 @@ dotnet_name:
   when: false
 
 dotnet_full_name:
-  # Eg: Aind.Behavior.NewProject
-  default: '{{ (prefix | replace("-", " ") | title | replace(" ", ".")) }}.{{ (dotnet_name) }}'
+  # Eg: UclOpenNewProject
+  default: '{{ (prefix | replace("-", " ") | title | replace(" ", "")) }}{{ (dotnet_name) }}'
   help: Internal .NET module name (auto-generated)
   when: false
 
 python_class_prefix:
-  # Eg: AindBehaviorNewProject
   default: '{{ dotnet_full_name | replace(".", "") }}'
   help: Internal Python class prefix (auto-generated)
   type: str

template/examples/session.py.jinja

@@ -1,7 +1,7 @@
 import datetime
 import os
 
-from swc.aeon.schema import Experiment
+from ucl_open.rigs.experiment import Experiment
 
 # TODO - autofill experiment fields
 session = Experiment(

template/scripts/deploy.ps1 → template/scripts/deploy.ps1.jinja

@@ -17,7 +17,13 @@ if (Test-Path -Path ./.venv) {
 &uv venv
 .\.venv\Scripts\Activate.ps1
 Write-Output "Synchronizing environment..."
-&uv sync 
+&uv sync
+
+Write-Output "Creating sgen output folder"
+mkdir src\DataSchemas
+
+Write-Output "Generate placeholder extension classes..."
+&uv run .\src\{{ python_folder_name }}\regenerate.py
 
 Write-Output "Creating a Bonsai environment and installing packages..."
 if (Test-Path -Path "bonsai") {
@@ -32,7 +38,4 @@ if (Test-Path -Path "bonsai") {
 Set-Location ..
 
 Write-Output "Creating bonsai extensions folder..."
-mkdir src\Extensions
-
-Write-Output "Creating sgen output folder"
-mkdir src\DataSchemas
+mkdir src\Extensions

template/src/Extensions/LoadSchemas.cs.jinja

@@ -0,0 +1,37 @@
+using System;
+using System.ComponentModel;
+using System.IO;
+using System.Reactive.Linq;
+using Bonsai;
+using {{ dotnet_full_name }}DataSchema;
+
+public class LoadSchemas : Source<Tuple<Experiment, {{ dotnet_full_name }}Rig, {{ dotnet_full_name }}TaskLogic>>
+{
+    [Description("The relative or absolute path of the session file to open for reading.")]
+    [Editor("Bonsai.Design.OpenFileNameEditor, Bonsai.Design", DesignTypes.UITypeEditor)]
+    public string SessionSettingsPath {get; set;}
+
+    [Description("The relative or absolute path of the rig file to open for reading.")]
+    [Editor("Bonsai.Design.OpenFileNameEditor, Bonsai.Design", DesignTypes.UITypeEditor)]
+    public string RigSettingsPath {get; set;}
+
+    [Description("The relative or absolute path of the task file to open for reading.")]
+    [Editor("Bonsai.Design.OpenFileNameEditor, Bonsai.Design", DesignTypes.UITypeEditor)]
+    public string TaskLogicSettingsPath {get; set;}
+
+    public override IObservable<Tuple<Experiment, {{ dotnet_full_name }}Rig, {{ dotnet_full_name }}TaskLogic>> Generate()
+    {
+        string sessionSettingsRaw = File.ReadAllText(SessionSettingsPath);
+        string rigSettingsRaw = File.ReadAllText(RigSettingsPath);
+        string taskLogicSettingsRaw = File.ReadAllText(TaskLogicSettingsPath);
+
+        var sessionSource = Observable.Return(Newtonsoft.Json.JsonConvert.DeserializeObject<Experiment>(sessionSettingsRaw));
+        var rigSource = Observable.Return(Newtonsoft.Json.JsonConvert.DeserializeObject<{{ dotnet_full_name }}Rig>(rigSettingsRaw));
+        var taskLogicSource = Observable.Return(Newtonsoft.Json.JsonConvert.DeserializeObject<{{ dotnet_full_name }}TaskLogic>(taskLogicSettingsRaw));
+
+        return sessionSource.Zip(rigSource, taskLogicSource, (s1, s2, s3) =>
+        {
+            return Tuple.Create(s1, s2, s3);
+        });
+    }
+}

template/src/{{ python_folder_name }}/regenerate.py.jinja

@@ -17,6 +17,7 @@ def main():
     models = [
         {{ python_folder_name }}.task.{{ python_class_prefix }}TaskLogic,
         {{ python_folder_name }}.rig.{{ python_class_prefix }}Rig,
+        Experiment
     ]
     model = pydantic.RootModel[Union[tuple(models)]]