Allow multiple reconstruction recipes for sxt

Author: stephen-riggs

src/murfey/client/contexts/sxt.py

@@ -127,6 +127,7 @@ def register_sxt_data_collection(
 
             recipes_to_assign_pjids = [
                 "sxt-aretomo",
+                "sxt-imod-patch",
             ]
             for recipe in recipes_to_assign_pjids:
                 capture_post(

src/murfey/workflows/sxt/process_sxt_tilt_series.py

@@ -59,6 +59,7 @@ def process_sxt_tilt_series_workflow(
         murfey_db.add(tilt_series)
         murfey_db.commit()
 
+    # Find all processing jobs registered for this tilt series
     collected_ids = murfey_db.exec(
         select(DataCollectionGroup, DataCollection, ProcessingJob, AutoProcProgram)
         .where(DataCollectionGroup.session_id == session_id)
@@ -67,8 +68,7 @@ def process_sxt_tilt_series_workflow(
         .where(DataCollection.dcg_id == DataCollectionGroup.id)
         .where(ProcessingJob.dc_id == DataCollection.id)
         .where(AutoProcProgram.pj_id == ProcessingJob.id)
-        .where(ProcessingJob.recipe == "sxt-aretomo")
-    ).one()
+    ).all()
     instrument_name = (
         murfey_db.exec(select(Session).where(Session.id == session_id))
         .one()
@@ -95,29 +95,34 @@ def process_sxt_tilt_series_workflow(
         / f"{tilt_series.tag}_stack.mrc"
     )
     stack_file.parent.mkdir(parents=True, exist_ok=True)
-    zocalo_message = {
-        "recipes": ["sxt-aretomo"],
-        "parameters": {
-            "txrm_file": tilt_series_info.txrm,
-            "xrm_reference": tilt_series_info.xrm_reference or "",
-            "dcid": collected_ids[1].id,
-            "appid": collected_ids[3].id,
-            "stack_file": str(stack_file),
-            "tilt_axis": 0,
-            "pixel_size": tilt_series_info.pixel_size,
-            "manual_tilt_offset": -tilt_series_info.tilt_offset,
-            "node_creator_queue": machine_config.node_creator_queue,
-        },
-    }
-    if _transport_object:
-        logger.info(
-            f"Sending Zocalo message for processing: {sanitise(str(zocalo_message))}"
-        )
-        _transport_object.send("processing_recipe", zocalo_message, new_connection=True)
-    else:
-        logger.info(
-            f"No transport object found. Zocalo message would be {sanitise(str(zocalo_message))}"
-        )
+
+    for recipe_ids in collected_ids:
+        # Loop over all processing jobs, and send the alignment recipe for it
+        zocalo_message = {
+            "recipes": recipe_ids[2].recipe,
+            "parameters": {
+                "txrm_file": tilt_series_info.txrm,
+                "xrm_reference": tilt_series_info.xrm_reference or "",
+                "dcid": recipe_ids[1].id,
+                "appid": recipe_ids[3].id,
+                "stack_file": str(stack_file),
+                "tilt_axis": 0,
+                "pixel_size": tilt_series_info.pixel_size,
+                "manual_tilt_offset": -tilt_series_info.tilt_offset,
+                "node_creator_queue": machine_config.node_creator_queue,
+            },
+        }
+        if _transport_object:
+            logger.info(
+                f"Sending Zocalo message for processing: {sanitise(str(zocalo_message))}"
+            )
+            _transport_object.send(
+                "processing_recipe", zocalo_message, new_connection=True
+            )
+        else:
+            logger.info(
+                f"No transport object found. Zocalo message would be {sanitise(str(zocalo_message))}"
+            )
     tilt_series.processing_requested = True
     murfey_db.add(tilt_series)
     murfey_db.commit()