update testing script

dont_fret/tmp.py

@@ -1,3 +1,4 @@
+# %%
 import uuid
 from dataclasses import replace
 from functools import reduce
@@ -19,7 +20,8 @@
 from dont_fret.web.components import RegexSelectDialog
 from dont_fret.web.home import HomePage
 from dont_fret.web.main import Page as MainPage
-from dont_fret.web.models import BurstNode, ListStore, PhotonNode
+from dont_fret.web.methods import make_burst_nodes
+from dont_fret.web.models import BurstNode, FRETNode, ListStore, PhotonNode
 from dont_fret.web.trace import TracePage
 from dont_fret.web.utils import (
     find_index,
@@ -29,7 +31,14 @@
     wrap_callback,
 )
 
-data = yaml.safe_load(Path("default_testing.yaml").read_text())
+# %%
+
+ROOT = Path(__file__).parent.parent
+ROOT
+# %%
+
+
+data = yaml.safe_load((ROOT / "default_testing.yaml").read_text())
 cfg.update(data)
 
 style = """
@@ -39,15 +48,37 @@
 }
 """
 
-my_selection = ListStore([])
+pth = ROOT / "tests" / "test_data" / "input" / "ds2"
+
+# %%
+pth.exists()
+# %%
+
+
+def on_start():
+    photon_nodes = [PhotonNode(file_path=ptu_pth) for ptu_pth in pth.glob("*.ptu")]
+    burst_nodes = make_burst_nodes(photon_nodes, cfg.burst_search)
 
-values = ["a", "b1", "c1", "d3"]
+    node_1 = FRETNode(
+        name=solara.Reactive("my_node"),
+        photons=ListStore(photon_nodes),
+        bursts=ListStore(burst_nodes),
+    )
 
-state.disable_burst_page.set(False)
-state.disable_trace_page.set(False)
+    state.fret_nodes.set([node_1])
+    state.disable_burst_page.set(False)
+    state.disable_trace_page.set(False)
+
+
+import solara.lifecycle
+
+solara.lifecycle.on_kernel_start(on_start)
 
 
 @solara.component
 def Page():
     solara.Style(style)
     MainPage()
+
+
+# %%