This commit is contained in:
flandre 2025-04-16 18:11:03 +08:00
parent 442b6a672d
commit 7753541fe7
3 changed files with 45 additions and 31 deletions

View File

@ -120,12 +120,7 @@ class Muxer(Node):
self.playback_rf_msg = msg
elif isinstance(msg, ImageArgMsg):
self.arg = msg
elif isinstance(msg, SetSeqMetaMsg):
match msg.target:
case 'playback':
self.seq_meta_playback = RfSequenceMeta.from_name(msg.name)
elif isinstance(msg, SeqMetaMsg):
logger.info(f'use live {msg}')
match msg.target:
case 'live':
self.seq_meta_live = RfSequenceMeta.from_name(msg.name)

View File

@ -131,7 +131,7 @@ class RfMat:
seq_meta = frame.seq_meta
else:
raise NotImplementedError()
m = np.frombuffer(frame.__bytes__(), dtype=np.int16).reshape(seq_meta.shape)
m = np.frombuffer(frame.__bytes__(), dtype=np.int16).reshape(seq_meta.shape).copy()
if device == 'gpu':
m = cp.asarray(m)
return RfMat(m, frame.meta, seq_meta)
@ -377,6 +377,11 @@ class RfMat:
self.m = self.m.astype(np.int64)
return self
def jupyter(self):
from matplotlib import pyplot as plt
plt.figure(figsize=(40, 20))
plt.imshow(self.m, cmap='grey')
if __name__ == '__main__':
cp.zeros((1, 2, 3)) + 1

File diff suppressed because one or more lines are too long