Add RecordAudio inspection step

src/isar/apis/models/start_mission_definition.py

@@ -10,6 +10,7 @@
 from robot_interface.models.mission.step import (
     STEPS,
     DriveToPose,
+    RecordAudio,
     TakeImage,
     TakeThermalImage,
     TakeThermalVideo,
@@ -22,6 +23,7 @@ class InspectionTypes(str, Enum):
     thermal_image = "ThermalImage"
     video = "Video"
     thermal_video = "ThermalVideo"
+    audio = "Audio"
 
 
 class StartMissionInspectionDefinition(BaseModel):
@@ -119,6 +121,8 @@ def create_inspection_step(
         inspection_step = TakeThermalImage(target=target)
     elif inspection_type == InspectionTypes.thermal_video.value:
         inspection_step = TakeThermalVideo(target=target, duration=duration)
+    elif inspection_type == InspectionTypes.audio.value:
+        inspection_step = RecordAudio(target=target, duration=duration)
     else:
         raise ValueError(f"Inspection type '{inspection_type}' not supported")
 

src/robot_interface/models/inspection/__init__.py

@@ -1,4 +1,5 @@
 from .inspection import (
+    Audio,
     Image,
     ImageMetadata,
     Inspection,

src/robot_interface/models/inspection/inspection.py

@@ -38,6 +38,11 @@ class ThermalVideoMetadata(InspectionMetadata):
     duration: Optional[float] = field(default=None)
 
 
+@dataclass
+class AudioMetadata(InspectionMetadata):
+    duration: Optional[float] = field(default=None)
+
+
 @dataclass
 class Inspection:
     id: str = field(default_factory=uuid4_string, init=False)
@@ -83,3 +88,12 @@ class ThermalVideo(Inspection):
     @staticmethod
     def get_metadata_type() -> Type[InspectionMetadata]:
         return ThermalVideoMetadata
+
+
+@dataclass
+class Audio(Inspection):
+    metadata: AudioMetadata
+
+    @staticmethod
+    def get_metadata_type() -> Type[InspectionMetadata]:
+        return AudioMetadata

src/robot_interface/models/mission/step.py

@@ -4,8 +4,14 @@
 from alitra import Pose, Position
 
 from isar.services.utilities.uuid_string_factory import uuid4_string
-from robot_interface.models.inspection import Image, ThermalImage, ThermalVideo, Video
-from robot_interface.models.inspection.inspection import Inspection
+from robot_interface.models.inspection import (
+    Audio,
+    Image,
+    Inspection,
+    ThermalImage,
+    ThermalVideo,
+    Video,
+)
 from robot_interface.models.mission.status import StepStatus
 
 
@@ -166,11 +172,29 @@ def get_inspection_type() -> Type[Inspection]:
         return ThermalVideo
 
 
+@dataclass
+class RecordAudio(InspectionStep):
+    """
+    Step which causes the robot to record a video at its position, facing the target.
+
+    Duration of audio is given in seconds.
+    """
+
+    target: Position
+    duration: float
+    type: Literal["record_audio"] = "record_audio"
+
+    @staticmethod
+    def get_inspection_type() -> Type[Inspection]:
+        return Audio
+
+
 STEPS = Union[
     DriveToPose,
     DockingProcedure,
     TakeImage,
     TakeThermalImage,
     TakeVideo,
     TakeThermalVideo,
+    RecordAudio,
 ]