Export Entity Type bindings

Emerge-Lab · May 1, 2024 · 98393cd · 98393cd
1 parent d09a632
commit 98393cd
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/bindings.cpp b/src/bindings.cpp
@@ -54,6 +54,19 @@ namespace gpudrive
             .value("AgentRemoved", CollisionBehaviour::AgentRemoved)
             .value("Ignore", CollisionBehaviour::Ignore);
 
+        nb::enum_<EntityType>(m, "EntityType")
+            .value("_None", EntityType::None) 
+            .value("RoadEdge", EntityType::RoadEdge) 
+            .value("RoadLine", EntityType::RoadLine) 
+            .value("RoadLane", EntityType::RoadLane) 
+            .value("CrossWalk", EntityType::CrossWalk) 
+            .value("SpeedBump", EntityType::SpeedBump) 
+            .value("StopSign", EntityType::StopSign) 
+            .value("Vehicle", EntityType::Vehicle) 
+            .value("Pedestrian", EntityType::Pedestrian) 
+            .value("Cyclist", EntityType::Cyclist) 
+            .value("Padding", EntityType::Padding) 
+            .value("NumTypes", EntityType::NumTypes);
 
         // Bindings for Manager class
         nb::class_<Manager>(m, "SimManager")