devices: hailo: Add a helper to determine the Hailo device architecture

This will be needed in a future commit to auto-select the correct model
HEF files for the detect and pose examples.

Signed-off-by: Naushir Patuck <naush@raspberrypi.com>

Author: naushir

picamera2/devices/__init__.py

@@ -1,6 +1,6 @@
 try:
     # Hailo requires hailo_platform package, which may not be installed on non-Hailo platforms.
-    from .hailo import Hailo
+    from .hailo import Hailo, hailo_architecture
 except ModuleNotFoundError:
     pass
 from .imx500 import IMX500

picamera2/devices/hailo/__init__.py

@@ -1 +1 @@
-from .hailo import Hailo
+from .hailo import Hailo, hailo_architecture

picamera2/devices/hailo/hailo.py

@@ -1,10 +1,31 @@
+import subprocess
 from concurrent.futures import Future
 from functools import partial
 
 import numpy as np
 from hailo_platform import HEF, FormatType, HailoSchedulingAlgorithm, VDevice
 
 
+def hailo_architecture():
+    """
+    Executes the hailortcli command and extracts the Device Architecture.
+
+    Returns the architecture string if found, otherwise returns None.
+    """
+    command = ["hailortcli", "fw-control", "identify"]
+
+    try:
+        result = subprocess.run(command, capture_output=True, text=True, check=True).stdout
+        for line in result.splitlines():
+            if "Device Architecture:" in line:
+                return line.split(":")[-1].strip()
+    except Exception as e:
+        print(f"Error identifying Hailo device: {e}")
+        return None
+
+    return None
+
+
 class Hailo:
     TARGET = None
     TARGET_REF_COUNT = 0