fix: fixed logic in data subscription to redis

This commit is contained in:
2023-08-09 21:40:21 +02:00
parent b88346182e
commit 37705f76b0
2 changed files with 13 additions and 5 deletions

View File

@@ -60,13 +60,20 @@ class StreamProcessorPx(StreamProcessor):
while self.queue.empty():
start = time.time()
data_msgs = self._get_data(proj_nr)
print(f"Processing took {time.time() - start}")
data.extend([msg.content["signals"]["data"] for msg in data_msgs if msg is not None])
# if len(data) > :
#if len(data) > 80:
# out = np.asarray(data)
# result = ({0: {"z": np.sum(out, axis=(-1, -2))}}, {1: {}})
#else:
# continue
print(f"Loading took {time.time() - start}")
result = self.process(data, metadata)
print(f"Processing took {time.time() - start}")
if not result:
continue
msg = BECMessage.ProcessedDataMessage(data=result[0][0], metadata=result[1]).dumps()
print(f"Length of data is {result[0][0]['z'].shape}")
msg = BECMessage.ProcessedDataMessage(
data=result[0][0], metadata=result[1]).dumps()
print("Publishing result")
self._publish_result(msg)
@@ -86,6 +93,7 @@ class StreamProcessorPx(StreamProcessor):
azint_data = np.asarray(data)
norm_sum = metadata["norm_sum"]
q = metadata["q"]
out = []
#####################################
# Pick contrast 0:f1amp, 1:f2amp, 2:f2phase