[Unity] Add listVehicles API

microsoft · Aug 16, 2020 · e5f229b · e5f229b
1 parent 3a90233
commit e5f229b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/Unity/AirLibWrapper/AirsimWrapper/Source/WorldSimApi.cpp b/Unity/AirLibWrapper/AirsimWrapper/Source/WorldSimApi.cpp
@@ -173,4 +173,15 @@ bool WorldSimApi::isRecording() const
     return false;
 }
 
+std::vector<std::string> WorldSimApi::listVehicles() const
+{
+    auto vehicle_names = (simmode_->getApiProvider()->getVehicleSimApis()).keys();
+    // Remove '' from the list, representing default vehicle
+    auto position = std::find(vehicle_names.begin(), vehicle_names.end(), "");
+    if (position != vehicle_names.end())
+        vehicle_names.erase(position);
+    return vehicle_names;
+}
+
+
 #pragma endregion
diff --git a/Unity/AirLibWrapper/AirsimWrapper/Source/WorldSimApi.h b/Unity/AirLibWrapper/AirsimWrapper/Source/WorldSimApi.h
@@ -59,6 +59,8 @@ class WorldSimApi : public msr::airlib::WorldSimApiBase
     virtual void stopRecording() override;
     virtual bool isRecording() const override;
 
+    virtual std::vector<std::string> listVehicles() const override;
+
 private:
 	SimModeBase * simmode_;
 	std::string vehicle_name_;