feat(hardware-testing): enable multi sensor processing in liquid probe (#14883)

<!--
Thanks for taking the time to open a pull request! Please make sure
you've read the "Opening Pull Requests" section of our Contributing
Guide:


https://github.com/Opentrons/opentrons/blob/edge/CONTRIBUTING.md#opening-pull-requests

To ensure your code is reviewed quickly and thoroughly, please fill out
the sections below to the best of your ability!
-->

# Overview
This allows the liquid level detection script to tell the pipette to
buffer the data from both pipettes and fetch them afterwards,
it will now spit out seprate CSVs for each sensor. post processing not
yet updated so the final report just grabs one from each trial, will
implement in EXEC-268

<!--
Use this section to describe your pull-request at a high level. If the
PR addresses any open issues, please tag the issues here.
-->

# Test Plan

<!--
Use this section to describe the steps that you took to test your Pull
Request.
If you did not perform any testing provide justification why.

OT-3 Developers: You should default to testing on actual physical
hardware.
Once again, if you did not perform testing against hardware, justify
why.

Note: It can be helpful to write a test plan before doing development

Example Test Plan (HTTP API Change)

- Verified that new optional argument `dance-party` causes the robot to
flash its lights, move the pipettes,
then home.
- Verified that when you omit the `dance-party` option the robot homes
normally
- Added protocol that uses `dance-party` argument to G-Code Testing
Suite
- Ran protocol that did not use `dance-party` argument and everything
was successful
- Added unit tests to validate that changes to pydantic model are
correct

-->

# Changelog

<!--
List out the changes to the code in this PR. Please try your best to
categorize your changes and describe what has changed and why.

Example changelog:
- Fixed app crash when trying to calibrate an illegal pipette
- Added state to API to track pipette usage
- Updated API docs to mention only two pipettes are supported

IMPORTANT: MAKE SURE ANY BREAKING CHANGES ARE PROPERLY COMMUNICATED
-->

# Review requests

<!--
Describe any requests for your reviewers here.
-->

# Risk assessment

<!--
Carefully go over your pull request and look at the other parts of the
codebase it may affect. Look for the possibility, even if you think it's
small, that your change may affect some other part of the system - for
instance, changing return tip behavior in protocol may also change the
behavior of labware calibration.

Identify the other parts of the system your codebase may affect, so that
in addition to your own review and testing, other people who may not
have the system internalized as much as you can focus their attention
and testing there.
-->

Author: ryanthecoder

api/src/opentrons/config/defaults_ot3.py

@@ -1,8 +1,8 @@
-from typing import Any, Dict, cast, List, Iterable, Tuple
+from typing import Any, Dict, cast, List, Iterable, Tuple, Optional
 from typing_extensions import Final
 from dataclasses import asdict
 
-from opentrons.hardware_control.types import OT3AxisKind
+from opentrons.hardware_control.types import OT3AxisKind, InstrumentProbeType
 from .types import (
     OT3Config,
     ByGantryLoad,
@@ -34,7 +34,7 @@
     aspirate_while_sensing=False,
     auto_zero_sensor=True,
     num_baseline_reads=10,
-    data_file="/var/pressure_sensor_data.csv",
+    data_files={InstrumentProbeType.PRIMARY: "/data/pressure_sensor_data.csv"},
 )
 
 DEFAULT_CALIBRATION_SETTINGS: Final[OT3CalibrationSettings] = OT3CalibrationSettings(
@@ -194,6 +194,49 @@
 )
 
 
+def _build_output_option_with_default(
+    from_conf: Any, default: OutputOptions
+) -> OutputOptions:
+    if from_conf is None:
+        return default
+    else:
+        if isinstance(from_conf, OutputOptions):
+            return from_conf
+        else:
+            try:
+                enumval = OutputOptions[from_conf]
+            except KeyError:  # not an enum entry
+                return default
+            else:
+                return enumval
+
+
+def _build_log_files_with_default(
+    from_conf: Any,
+    default: Optional[Dict[InstrumentProbeType, str]],
+) -> Optional[Dict[InstrumentProbeType, str]]:
+    print(f"from_conf {from_conf} default {default}")
+    if not isinstance(from_conf, dict):
+        if default is None:
+            return None
+        else:
+            return {k: v for k, v in default.items()}
+    else:
+        validated: Dict[InstrumentProbeType, str] = {}
+        for k, v in from_conf.items():
+            if isinstance(k, InstrumentProbeType):
+                validated[k] = v
+            else:
+                try:
+                    enumval = InstrumentProbeType[k]
+                except KeyError:  # not an enum entry
+                    pass
+                else:
+                    validated[enumval] = v
+        print(f"result {validated}")
+        return validated
+
+
 def _build_dict_with_default(
     from_conf: Any,
     default: Dict[OT3AxisKind, float],
@@ -278,6 +321,17 @@ def _build_default_cap_pass(
 def _build_default_liquid_probe(
     from_conf: Any, default: LiquidProbeSettings
 ) -> LiquidProbeSettings:
+    output_option = _build_output_option_with_default(
+        from_conf.get("output_option", None), default.output_option
+    )
+    data_files: Optional[Dict[InstrumentProbeType, str]] = None
+    if (
+        output_option is OutputOptions.sync_buffer_to_csv
+        or output_option is OutputOptions.stream_to_csv
+    ):
+        data_files = _build_log_files_with_default(
+            from_conf.get("data_files", {}), default.data_files
+        )
     return LiquidProbeSettings(
         starting_mount_height=from_conf.get(
             "starting_mount_height", default.starting_mount_height
@@ -302,7 +356,7 @@ def _build_default_liquid_probe(
         num_baseline_reads=from_conf.get(
             "num_baseline_reads", default.num_baseline_reads
         ),
-        data_file=from_conf.get("data_file", default.data_file),
+        data_files=data_files,
     )
 
 
@@ -412,7 +466,7 @@ def build_with_defaults(robot_settings: Dict[str, Any]) -> OT3Config:
 def serialize(config: OT3Config) -> Dict[str, Any]:
     def _build_dict(pairs: Iterable[Tuple[Any, Any]]) -> Dict[str, Any]:
         def _normalize_key(key: Any) -> Any:
-            if isinstance(key, OT3AxisKind):
+            if isinstance(key, OT3AxisKind) or isinstance(key, InstrumentProbeType):
                 return key.name
             return key
 

api/src/opentrons/config/types.py

@@ -2,7 +2,7 @@
 from dataclasses import dataclass, asdict, fields
 from typing import Dict, Tuple, TypeVar, Generic, List, cast, Optional
 from typing_extensions import TypedDict, Literal
-from opentrons.hardware_control.types import OT3AxisKind
+from opentrons.hardware_control.types import OT3AxisKind, InstrumentProbeType
 
 
 class AxisDict(TypedDict):
@@ -139,7 +139,7 @@ class LiquidProbeSettings:
     aspirate_while_sensing: bool
     auto_zero_sensor: bool
     num_baseline_reads: int
-    data_file: Optional[str]
+    data_files: Optional[Dict[InstrumentProbeType, str]]
 
 
 @dataclass(frozen=True)

api/src/opentrons/hardware_control/backends/flex_protocol.py

@@ -147,7 +147,7 @@ async def liquid_probe(
         plunger_speed: float,
         threshold_pascals: float,
         output_format: OutputOptions = OutputOptions.can_bus_only,
-        data_file: Optional[str] = None,
+        data_files: Optional[Dict[InstrumentProbeType, str]] = None,
         auto_zero_sensor: bool = True,
         num_baseline_reads: int = 10,
         probe: InstrumentProbeType = InstrumentProbeType.PRIMARY,

api/src/opentrons/hardware_control/backends/ot3controller.py

@@ -1351,7 +1351,7 @@ async def liquid_probe(
         plunger_speed: float,
         threshold_pascals: float,
         output_option: OutputOptions = OutputOptions.can_bus_only,
-        data_file: Optional[str] = None,
+        data_files: Optional[Dict[InstrumentProbeType, str]] = None,
         auto_zero_sensor: bool = True,
         num_baseline_reads: int = 10,
         probe: InstrumentProbeType = InstrumentProbeType.PRIMARY,
@@ -1372,6 +1372,14 @@ async def liquid_probe(
         can_bus_only_output = bool(
             output_option.value & OutputOptions.can_bus_only.value
         )
+        data_files_transposed = (
+            None
+            if data_files is None
+            else {
+                sensor_id_for_instrument(probe): data_files[probe]
+                for probe in data_files.keys()
+            }
+        )
         positions = await liquid_probe(
             messenger=self._messenger,
             tool=tool,
@@ -1383,7 +1391,7 @@ async def liquid_probe(
             csv_output=csv_output,
             sync_buffer_output=sync_buffer_output,
             can_bus_only_output=can_bus_only_output,
-            data_file=data_file,
+            data_files=data_files_transposed,
             auto_zero_sensor=auto_zero_sensor,
             num_baseline_reads=num_baseline_reads,
             sensor_id=sensor_id_for_instrument(probe),

api/src/opentrons/hardware_control/backends/ot3simulator.py

@@ -346,7 +346,7 @@ async def liquid_probe(
         plunger_speed: float,
         threshold_pascals: float,
         output_format: OutputOptions = OutputOptions.can_bus_only,
-        data_file: Optional[str] = None,
+        data_files: Optional[Dict[InstrumentProbeType, str]] = None,
         auto_zero_sensor: bool = True,
         num_baseline_reads: int = 10,
         probe: InstrumentProbeType = InstrumentProbeType.PRIMARY,

api/src/opentrons/hardware_control/backends/ot3utils.py

@@ -544,6 +544,7 @@ def sensor_node_for_pipette(mount: OT3Mount) -> PipetteProbeTarget:
 _instr_sensor_id_lookup: Dict[InstrumentProbeType, SensorId] = {
     InstrumentProbeType.PRIMARY: SensorId.S0,
     InstrumentProbeType.SECONDARY: SensorId.S1,
+    InstrumentProbeType.BOTH: SensorId.BOTH,
 }
 
 

api/src/opentrons/hardware_control/ot3api.py

@@ -2601,7 +2601,7 @@ async def liquid_probe(
             (probe_settings.plunger_speed * plunger_direction),
             probe_settings.sensor_threshold_pascals,
             probe_settings.output_option,
-            probe_settings.data_file,
+            probe_settings.data_files,
             probe_settings.auto_zero_sensor,
             probe_settings.num_baseline_reads,
             probe=probe if probe else InstrumentProbeType.PRIMARY,

api/src/opentrons/hardware_control/types.py

@@ -624,6 +624,7 @@ class GripperJawState(enum.Enum):
 class InstrumentProbeType(enum.Enum):
     PRIMARY = enum.auto()
     SECONDARY = enum.auto()
+    BOTH = enum.auto()
 
 
 class GripperProbe(enum.Enum):

api/tests/opentrons/config/ot3_settings.py

@@ -129,7 +129,7 @@
         "aspirate_while_sensing": False,
         "auto_zero_sensor": True,
         "num_baseline_reads": 10,
-        "data_file": "/var/pressure_sensor_data.csv",
+        "data_files": {"PRIMARY": "/data/pressure_sensor_data.csv"},
     },
     "calibration": {
         "z_offset": {

api/tests/opentrons/hardware_control/backends/test_ot3_controller.py

@@ -61,6 +61,7 @@
     UpdateState,
     EstopState,
     CurrentConfig,
+    InstrumentProbeType,
 )
 from opentrons.hardware_control.errors import (
     InvalidPipetteName,
@@ -185,7 +186,7 @@ def fake_liquid_settings() -> LiquidProbeSettings:
         aspirate_while_sensing=False,
         auto_zero_sensor=False,
         num_baseline_reads=8,
-        data_file="fake_data_file",
+        data_files={InstrumentProbeType.PRIMARY: "fake_file_name"},
     )