From ef51ba35a7388d99b6580a2379de6dcd8b21be48 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Anders=20Chirico=20Indreb=C3=B8?= <ACIN@equinor.com>
Date: Thu, 22 Feb 2024 15:40:40 +0100
Subject: [PATCH] Add support for thermal images

---
 src/isar_exr/robotinterface.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/isar_exr/robotinterface.py b/src/isar_exr/robotinterface.py
index c315ee7..5de3300 100644
--- a/src/isar_exr/robotinterface.py
+++ b/src/isar_exr/robotinterface.py
@@ -398,7 +398,13 @@ def _create_and_add_poi(
                 )
             )
             add_point_of_interest_input["photoAction"] = photo_input
-
+        elif isinstance(step, TakeThermalImage):
+            photo_input: PointOfInterestActionPhotoInput = (
+                PointOfInterestActionPhotoInput(
+                    robotPose=photo_input_pose, sensor="back_cam_link"
+                )
+            )
+            add_point_of_interest_input["photoAction"] = photo_input
         elif isinstance(step, TakeVideo):
             video_input: PointOfInterestActionVideoInput = (
                 PointOfInterestActionVideoInput(