bec/bec_widgets
0
0
Fork 0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-14 11:41:49 +02:00
Code Activity

refactor: register scan_segment callback directly

Browse Source 
* this allows to skip client.start() setup
This commit is contained in:
usov_i
2023-08-10 17:18:20 +02:00
parent 6a3df34cdf
commit dfce55b675
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
bec_widgets/bec_dispatcher.py
View File

@ -57,7 +57,6 @@ class _BECDispatcher(QObject):
bec_config = "bec_config.yaml" bec_config = "bec_config.yaml"
self.client.initialize(config=ServiceConfig(config_path=bec_config)) self.client.initialize(config=ServiceConfig(config_path=bec_config))
self.client.start()
self._slot_signal_map = { self._slot_signal_map = {
"on_scan_segment": self.scan_segment, "on_scan_segment": self.scan_segment,
@ -71,16 +70,22 @@ class _BECDispatcher(QObject):
# self.new_projection_id.connect(self.new_projection_data) # self.new_projection_id.connect(self.new_projection_data)
def _scan_segment_cb(scan_segment, metadata): def _scan_segment_cb(msg):
msg = BECMessage.ScanMessage.loads(msg.value)[0]
with scan_lock: with scan_lock:
# TODO: use ScanStatusMessage instead? # TODO: use ScanStatusMessage instead?
scan_id = metadata["scanID"] scan_id = msg.content["scanID"]
if self._scan_id != scan_id: if self._scan_id != scan_id:
self._scan_id = scan_id self._scan_id = scan_id
self.new_scan.emit(scan_segment, metadata) self.new_scan.emit(msg.content, msg.metadata)
self.scan_segment.emit(scan_segment, metadata) self.scan_segment.emit(msg.content, msg.metadata)
self.client.callbacks.register("scan_segment", _scan_segment_cb, sync=False) scan_segment_topic = MessageEndpoints.scan_segment()
self._scan_segment_thread = self.client.connector.consumer(
topics=scan_segment_topic,
cb=_scan_segment_cb,
)
self._scan_segment_thread.start()
def connect(self, widget): def connect(self, widget):
for slot_name, signal in self._slot_signal_map.items(): for slot_name, signal in self._slot_signal_map.items():