Support Capsule and Ellipsoid in Scene3D

Signed-off-by: ahcorde <ahcorde@gmail.com>
gazebosim · Apr 28, 2021 · 84f46d0 · 84f46d0
1 parent 4df7622
commit 84f46d0
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/plugins/scene3d/Scene3D.cc b/src/plugins/scene3d/Scene3D.cc
@@ -681,6 +681,20 @@ rendering::GeometryPtr SceneManager::LoadGeometry(const msgs::Geometry &_msg,
     scale.Y() = scale.X();
     scale.Z() = _msg.cylinder().length();
   }
+  else if (_msg.has_capsule())
+  {
+    auto capsule = this->scene->CreateCapsule();
+    capsule->SetRadius(_msg.capsule().radius());
+    capsule->SetLength(_msg.capsule().length());
+    geom = capsule;
+  }
+  else if (_msg.has_ellipsoid())
+  {
+    geom = this->scene->CreateSphere();
+    scale.X() = _msg.ellipsoid().radii().x() * 2;
+    scale.Y() = _msg.ellipsoid().radii().y() * 2;
+    scale.Z() = _msg.ellipsoid().radii().z() * 2;
+  }
   else if (_msg.has_plane())
   {
     geom = this->scene->CreatePlane();