refactor: renaming of saxs_imaging_processor
This commit is contained in:
@@ -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}")
|
||||
Reference in New Issue
Block a user