annotate src/light9/blender/time_sync/time_from_graph.py @ 2433:ffa69645e9dc

refactor
author drewp@bigasterisk.com
date Tue, 28 May 2024 15:41:53 -0700
parents b8a408caf115
children 9b101d8bd7ea
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
2432
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
1 import threading
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
2 import time
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
3
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
4 import bpy # type: ignore
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
5 from bpy.app.handlers import persistent # type: ignore
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
6 from light9.typedgraph import typedValue
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
7 from rdfdb.syncedgraph.syncedgraph import SyncedGraph
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
8 from rdflib import URIRef
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
9
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
10 from light9 import networking
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
11 from light9.blender.asyncio_thread import startLoopInThread
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
12 from light9.namespaces import L9
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
13 from light9.run_local import log
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
14
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
15
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
16 def clamp(lo, hi, x):
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
17 return max(lo, min(hi, x))
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
18
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
19
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
20 UPDATE_PERIOD = 1 / 20
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
21
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
22
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
23 class Sync:
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
24 lock = threading.Lock()
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
25 duration: float = 1
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
26 wallStartTime: float | None = None
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
27 pausedSongTime: float | None = None
2433
ffa69645e9dc refactor
drewp@bigasterisk.com
parents: 2432
diff changeset
28 playing = False
2432
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
29 latestTime: float
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
30
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
31 def __init__(self):
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
32 # main thread
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
33 self.lastSetFrame = -1
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
34 self.lastGraphFrame = -1
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
35 startLoopInThread(self.task())
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
36 bpy.app.timers.register(self.update)
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
37 bpy.app.handlers.frame_change_post.append(self.on_frame_change_post)
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
38
2433
ffa69645e9dc refactor
drewp@bigasterisk.com
parents: 2432
diff changeset
39 ## updates from graph -> self
2432
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
40
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
41 async def task(self):
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
42 # bg thread with asyncio loop
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
43 self.graph = SyncedGraph(networking.rdfdb.url, "time_sync")
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
44 self.graph.addHandler(self.syncFromGraph)
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
45
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
46 def syncFromGraph(self):
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
47 # bg thread
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
48 with self.lock:
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
49 asco = L9['ascoltami']
2433
ffa69645e9dc refactor
drewp@bigasterisk.com
parents: 2432
diff changeset
50 self.wallStartTime = typedValue(float | None, self.graph, asco, L9['wallStartTime'])
ffa69645e9dc refactor
drewp@bigasterisk.com
parents: 2432
diff changeset
51 self.pausedSongTime = typedValue(float | None, self.graph, asco, L9['pausedSongTime'])
ffa69645e9dc refactor
drewp@bigasterisk.com
parents: 2432
diff changeset
52 self.duration = typedValue(float | None, self.graph, asco, L9['duration']) or 1.0
2432
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
53 self.playing = typedValue(bool, self.graph, asco, L9['playing'])
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
54
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
55 def currentTime(self) -> float | None:
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
56 if self.wallStartTime is not None:
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
57 return time.time() - self.wallStartTime
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
58 if self.pausedSongTime is not None:
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
59 return self.pausedSongTime
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
60 log.warn('no time data')
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
61 return None
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
62
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
63 ## graph time -> blender time
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
64
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
65 @persistent
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
66 def update(self):
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
67 # main thread? wherever blender runs timers
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
68 if self.playing:
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
69 with self.lock:
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
70 t = self.currentTime()
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
71 if t is not None:
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
72 self.setBlenderTime(t, self.duration)
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
73 return UPDATE_PERIOD
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
74
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
75 def setBlenderTime(self, t: float, duration: float):
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
76 scene = bpy.context.scene
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
77 fps = scene.render.fps
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
78 scene.frame_start = 0
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
79 scene.frame_end = int(duration * fps)
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
80 fr = int(clamp(t, 0, duration) * fps)
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
81 self.lastSetFrame = fr
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
82 scene.frame_set(fr)
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
83
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
84 ## blender time -> graph time
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
85
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
86 @persistent
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
87 def on_frame_change_post(self, scene, deps):
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
88 if scene.frame_current != self.lastSetFrame:
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
89 # self.setGraphTime(scene.frame_current / scene.render.fps, self.duration)
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
90 self.lastSetFrame = scene.frame_current
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
91 t = scene.frame_current / scene.render.fps
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
92 self.setInGraph(t)
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
93
b8a408caf115 start blender sync
drewp@bigasterisk.com
parents:
diff changeset
94 def setInGraph(self, t: float):
2433
ffa69645e9dc refactor
drewp@bigasterisk.com
parents: 2432
diff changeset
95 log.warning(f'todo: set graph to {t}')