[Unity] Add listVehicles API

microsoft · Oct 9, 2020 · 6d2de35 · 6d2de35
1 parent 032f6d7
commit 6d2de35
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/Unity/AirLibWrapper/AirsimWrapper/Source/WorldSimApi.cpp b/Unity/AirLibWrapper/AirsimWrapper/Source/WorldSimApi.cpp
@@ -183,6 +183,16 @@ bool WorldSimApi::isRecording() const
 void WorldSimApi::setWind(const Vector3r& wind) const
 {
     simmode_->setWind(wind);
-};
+}
+
+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
@@ -62,6 +62,7 @@ class WorldSimApi : public msr::airlib::WorldSimApiBase
     virtual bool isRecording() const override;
 
     virtual void setWind(const Vector3r& wind) const override;
+    virtual std::vector<std::string> listVehicles() const override;
 
 private:
 	SimModeBase * simmode_;