diff --git a/bec_plugins/data_processing/px_example.py b/bec_plugins/data_processing/saxs_imaging_processor.py similarity index 97% rename from bec_plugins/data_processing/px_example.py rename to bec_plugins/data_processing/saxs_imaging_processor.py index bcf3b76..831e3ae 100644 --- a/bec_plugins/data_processing/px_example.py +++ b/bec_plugins/data_processing/saxs_imaging_processor.py @@ -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}")