From cb95b26000df592e50b092c9b54191361d86aec7 Mon Sep 17 00:00:00 2001
From: Jun <62534263+TanJunKiat@users.noreply.github.com>
Date: Fri, 20 Dec 2024 11:12:05 +0800
Subject: [PATCH] Add a conditional check for level transformation (#520)

Signed-off-by: tanjunkiat <tanjunkiat@outlook.sg>
---
 rmf_traffic_editor/gui/building.cpp | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/rmf_traffic_editor/gui/building.cpp b/rmf_traffic_editor/gui/building.cpp
index 1ffee5ce..d1729d59 100644
--- a/rmf_traffic_editor/gui/building.cpp
+++ b/rmf_traffic_editor/gui/building.cpp
@@ -524,6 +524,16 @@ Building::Transform Building::compute_transform(
 
   // calculate the rotation between each pair of fiducials
   vector<std::pair<double, double>> rotations;
+
+  if (fiducials.size() < 2)
+  {
+    printf(
+      "not enough fiducials to compute transform between levels %d and %d\n",
+      from_level_idx,
+      to_level_idx);
+    return Building::Transform();
+  }
+
   // we take the first fiducial as the reference point
   std::pair<Fiducial, Fiducial> ref_rotation = make_pair(fiducials[0].first,
       fiducials[0].second);