refactor: renaming of saxs_imaging_processor

This commit is contained in:
2023-08-10 15:03:02 +02:00
parent a9d9c7d440
commit 59ad0fd7e4

View File

@@ -1,17 +1,15 @@
from __future__ import annotations
import numpy as np
from string import Template
from queue import Queue
import time
# import bec_lib
from data_processing.stream_processor import StreamProcessor
from bec_lib.core import BECMessage, MessageEndpoints
from bec_lib.core.redis_connector import MessageObject, RedisConnector
from queue import Queue
from typing import Optional, Tuple
from data_processing.stream_processor import StreamProcessor
from bec_lib.core import BECMessage
from bec_lib.core.redis_connector import MessageObject, RedisConnector
class StreamProcessorPx(StreamProcessor):
class SaxsImagingProcessor(StreamProcessor):
def __init__(self, connector: RedisConnector, config: dict) -> None:
""""""
super().__init__(connector, config)
@@ -141,15 +139,14 @@ class StreamProcessorPx(StreamProcessor):
data = []
while self.queue.empty():
# TODO debug code for timing
#
data_msgs = self._get_data(proj_nr)
if data_msgs is None:
continue
data.extend([msg.content["signals"]["data"] for msg in data_msgs if msg is not None])
# print(f"Loading took {time.time() - start}")
start = time.time()
# start = time.time()
result = self.process(data, metadata)
print(f"Processing took {time.time() - start}")
# print(f"Processing took {time.time() - start}")
if result is None:
continue
print(f"Length of data is {result[0][0]['z'].shape}")