isar 1.19.1__py3-none-any.whl → 1.20.1__py3-none-any.whl

isar/__init__.py

@@ -1,6 +1,3 @@
-from importlib.metadata import PackageNotFoundError, distribution
+from importlib.metadata import version
 
-try:
-    __version__ = distribution(__name__).version
-except PackageNotFoundError:
-    pass  # package is not installed
+__version__ = version(__package__ or __name__)

isar/apis/models/start_mission_definition.py

@@ -2,7 +2,7 @@ import time
 from enum import Enum
 from typing import Any, Dict, List, Optional, Union
 
-from alitra import Position
+from alitra import Position, Pose, Orientation, Frame
 from pydantic import BaseModel, Field
 
 from isar.apis.models.models import InputPose, InputPosition
@@ -61,6 +61,7 @@ class StartMissionDefinition(BaseModel):
     tasks: List[StartMissionTaskDefinition]
     id: Optional[str] = None
     name: Optional[str] = None
+    start_pose: Optional[InputPose] = None
 
 
 def to_isar_mission(mission_definition: StartMissionDefinition) -> Mission:
@@ -92,6 +93,26 @@ def to_isar_mission(mission_definition: StartMissionDefinition) -> Mission:
     if mission_definition.id:
         isar_mission.id = mission_definition.id
 
+    if mission_definition.start_pose:
+        input_pose: InputPose = mission_definition.start_pose
+        input_frame: Frame = Frame(name=input_pose.frame_name)
+        input_position: Position = Position(
+            input_pose.position.x,
+            input_pose.position.y,
+            input_pose.position.z,
+            input_frame,
+        )
+        input_orientation: Orientation = Orientation(
+            input_pose.orientation.x,
+            input_pose.orientation.y,
+            input_pose.orientation.z,
+            input_pose.orientation.w,
+            input_frame,
+        )
+        isar_mission.start_pose = Pose(
+            position=input_position, orientation=input_orientation, frame=input_frame
+        )
+
     return isar_mission
 
 

isar/script.py

@@ -0,0 +1,154 @@
+import logging
+import time
+from logging import Logger
+from threading import Thread
+from typing import Any, List
+
+from injector import Injector
+
+import isar
+from isar.apis.api import API
+from isar.config.keyvault.keyvault_service import Keyvault
+from isar.config.log import setup_loggers
+from isar.config.settings import settings
+from isar.models.communication.queues.queues import Queues
+from isar.modules import get_injector
+from isar.services.service_connections.mqtt.mqtt_client import MqttClient
+from isar.services.service_connections.mqtt.robot_heartbeat_publisher import (
+    RobotHeartbeatPublisher,
+)
+from isar.services.service_connections.mqtt.robot_info_publisher import (
+    RobotInfoPublisher,
+)
+from isar.state_machine.state_machine import StateMachine, main
+from isar.storage.uploader import Uploader
+from robot_interface.robot_interface import RobotInterface
+
+
+def print_setting(
+    setting: str = "", value: Any = "", fillchar: str = " ", width: int = 48
+):
+    separator = ": " if value != "" else ""
+    text = setting.ljust(22, fillchar) + separator + str(value)
+    print("*", text.ljust(width - 4, fillchar), "*")
+
+
+def print_startup_info():
+    print(
+        """
+         __   ________   ___        ________
+        / /  / ______/  /   |      / ____  /
+       / /  / /_____   / /| |     / /___/ /
+      / /  /_____  /  / __  |    / __  __/
+     / /  ______/ /  / /  | |   / /  | |
+    /_/  /_______/  /_/   |_|  /_/   |_|
+
+"""
+    )
+
+    WIDTH = 48
+
+    def print_setting(setting: str = "", value: Any = "", fillchar: str = " "):
+        separator = ": " if value != "" else ""
+        text = setting.ljust(22, fillchar) + separator + str(value)
+        print("*", text.ljust(WIDTH - 4, fillchar), "*")
+
+    print("Integration and Supervisory control".center(WIDTH, " "))
+    print("of Autonomous Robots".center(WIDTH, " "))
+    print()
+    print(f"Version: {isar.__version__}\n".center(WIDTH, " "))
+
+    print_setting(fillchar="*")
+    print_setting("ISAR settings")
+    print_setting(fillchar="-")
+    print_setting("Robot package", settings.ROBOT_PACKAGE)
+    print_setting("Robot name", settings.ROBOT_NAME)
+    print_setting("Run mission stepwise", settings.RUN_MISSION_STEPWISE)
+    print_setting("Running on port", settings.API_PORT)
+    print_setting("Mission planner", settings.MISSION_PLANNER)
+    print_setting("Using local storage", settings.STORAGE_LOCAL_ENABLED)
+    print_setting("Using blob storage", settings.STORAGE_BLOB_ENABLED)
+    print_setting("Using SLIMM storage", settings.STORAGE_SLIMM_ENABLED)
+    print_setting("Plant code", settings.PLANT_CODE)
+    print_setting("Plant name", settings.PLANT_NAME)
+    print_setting("Plant shortname", settings.PLANT_SHORT_NAME)
+    print_setting(fillchar="*")
+    print()
+
+
+def start():
+    injector: Injector = get_injector()
+
+    keyvault_client = injector.get(Keyvault)
+    setup_loggers(keyvault=keyvault_client)
+    logger: Logger = logging.getLogger("main")
+
+    print_startup_info()
+
+    state_machine: StateMachine = injector.get(StateMachine)
+    uploader: Uploader = injector.get(Uploader)
+    robot: RobotInterface = injector.get(RobotInterface)
+    queues: Queues = injector.get(Queues)
+
+    threads: List[Thread] = []
+
+    state_machine_thread: Thread = Thread(
+        target=main, name="ISAR State Machine", args=[state_machine], daemon=True
+    )
+    threads.append(state_machine_thread)
+
+    uploader_thread: Thread = Thread(
+        target=uploader.run, name="ISAR Uploader", daemon=True
+    )
+    threads.append(uploader_thread)
+    if settings.MQTT_ENABLED:
+        mqtt_client: MqttClient = MqttClient(mqtt_queue=queues.mqtt_queue)
+
+        mqtt_thread: Thread = Thread(
+            target=mqtt_client.run, name="ISAR MQTT Client", daemon=True
+        )
+        threads.append(mqtt_thread)
+
+        robot_info_publisher: RobotInfoPublisher = RobotInfoPublisher(
+            mqtt_queue=queues.mqtt_queue
+        )
+        robot_info_thread: Thread = Thread(
+            target=robot_info_publisher.run,
+            name="ISAR Robot Info Publisher",
+            daemon=True,
+        )
+        threads.append(robot_info_thread)
+
+        robot_heartbeat_publisher: RobotHeartbeatPublisher = RobotHeartbeatPublisher(
+            mqtt_queue=queues.mqtt_queue
+        )
+
+        robot_heartbeat_thread: Thread = Thread(
+            target=robot_heartbeat_publisher.run,
+            name="ISAR Robot Heartbeat Publisher",
+            daemon=True,
+        )
+        threads.append(robot_heartbeat_thread)
+
+        publishers: List[Thread] = robot.get_telemetry_publishers(
+            queue=queues.mqtt_queue,
+            robot_name=settings.ROBOT_NAME,
+            isar_id=settings.ISAR_ID,
+        )
+        if publishers:
+            threads.extend(publishers)
+
+    api: API = injector.get(API)
+    api_thread: Thread = Thread(target=api.run_app, name="ISAR API", daemon=True)
+    threads.append(api_thread)
+
+    for thread in threads:
+        thread.start()
+        logger.info(f"Started thread: {thread.name}")
+
+    while True:
+        for thread in threads:
+            if not thread.is_alive():
+                logger.critical("Thread '%s' failed - ISAR shutting down", thread.name)
+                exit(1)
+        time.sleep(state_machine.sleep_time)

isar/state_machine/states/initialize.py

@@ -1,6 +1,6 @@
 import logging
 import time
-from typing import Callable, Optional, TYPE_CHECKING
+from typing import TYPE_CHECKING, Callable, Optional
 
 from injector import inject
 from transitions import State

isar/state_machine/states/initiate.py

@@ -1,6 +1,6 @@
 import logging
 import time
-from typing import Any, Callable, Optional, TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, Callable, Optional
 
 from transitions import State
 

isar/state_machine/states/monitor.py

@@ -6,8 +6,8 @@ from typing import TYPE_CHECKING, Callable, Optional, Sequence, Tuple, Union
 from injector import inject
 from transitions import State
 
-from isar.mission_planner.task_selector_interface import TaskSelectorStop
 from isar.config.settings import settings
+from isar.mission_planner.task_selector_interface import TaskSelectorStop
 from isar.services.utilities.threaded_request import (
     ThreadedRequest,
     ThreadedRequestNotFinishedError,
@@ -156,6 +156,10 @@ class Monitor(State):
             else:
                 if isinstance(status, StepStatus):
                     if self._step_finished(self.state_machine.current_step):
+                        self.state_machine.update_current_task()
+                        if self.state_machine.current_task == None:
+                            transition = self.state_machine.full_mission_finished  # type: ignore
+                            break
                         self.state_machine.update_current_step()
                         self.state_machine.current_task.update_task_status()
                     else:  # If not all steps are done

isar/state_machine/states/paused.py

@@ -1,6 +1,6 @@
 import logging
 import time
-from typing import Callable, TYPE_CHECKING
+from typing import TYPE_CHECKING, Callable
 
 from transitions import State
 

isar/state_machine/states/stop.py

@@ -1,6 +1,6 @@
 import logging
 import time
-from typing import Callable, Optional, TYPE_CHECKING
+from typing import TYPE_CHECKING, Callable, Optional
 
 from transitions import State
 

isar/storage/local_storage.py

@@ -2,10 +2,10 @@ import logging
 from pathlib import Path
 
 from isar.config.settings import settings
-from robot_interface.models.mission.mission import Mission
 from isar.storage.storage_interface import StorageException, StorageInterface
 from isar.storage.utilities import construct_metadata_file, construct_paths
 from robot_interface.models.inspection.inspection import Inspection
+from robot_interface.models.mission.mission import Mission
 
 
 class LocalStorage(StorageInterface):
@@ -27,9 +27,10 @@ class LocalStorage(StorageInterface):
             inspection=inspection, mission=mission, filename=local_path.name
         )
         try:
-            with open(absolute_path, "wb") as file, open(
-                absolute_metadata_path, "wb"
-            ) as metadata_file:
+            with (
+                open(absolute_path, "wb") as file,
+                open(absolute_metadata_path, "wb") as metadata_file,
+            ):
                 file.write(inspection.data)
                 metadata_file.write(metadata_bytes)
         except IOError as e: