From 03968799e4b5caaaa6ad2eef75ef40c1348cc0c3 Mon Sep 17 00:00:00 2001
From: Pawel Ziecina <1385347+pawel-ziecina@users.noreply.github.com>
Date: Wed, 27 May 2020 07:46:40 +0200
Subject: [PATCH] Add optional debug string to RealTrafficCar

---
 carla_real_traffic_scenarios/utils/carla.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/carla_real_traffic_scenarios/utils/carla.py b/carla_real_traffic_scenarios/utils/carla.py
index c06573d..60ef981 100644
--- a/carla_real_traffic_scenarios/utils/carla.py
+++ b/carla_real_traffic_scenarios/utils/carla.py
@@ -6,7 +6,7 @@
 import carla
 from carla_real_traffic_scenarios.utils.collections import smallest_by
 from carla_real_traffic_scenarios.utils.geometry import jaccard_rectangles
-from carla_real_traffic_scenarios.utils.transforms import Transform
+from carla_real_traffic_scenarios.utils.transforms import Transform, Vector3
 from carla_real_traffic_scenarios.vehicles import VehicleModel, VEHICLES
 
 LOGGER = logging.getLogger(__name__)
@@ -19,6 +19,7 @@ class RealTrafficVehicle(NamedTuple):
     length_m: float
     transform: Transform
     speed: float
+    debug: Optional[str]
 
 
 class RealTrafficVehiclesInCarla:
@@ -61,6 +62,10 @@ def step(self, vehicles):
                 )
                 self._vehicle_by_vehicle_id[real_vehicle.id] = carla_vehicle
 
+            if real_vehicle.debug:
+                self._world.debug.draw_string((target_transform.position + Vector3(2, 0, 4)).as_carla_location(),
+                                              str(real_vehicle.debug))
+
             now_vehicle_ids = {v.id for v in vehicles}
             previous_vehicles_ids = set(self._vehicle_by_vehicle_id.keys())