pxmlw6n2f/Gazebo_Distributed_TCP/gazebo/physics/bullet/BulletSphereShape.hh

118 lines
4.1 KiB
C++

/*
* Copyright (C) 2012 Open Source Robotics Foundation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
/* Desc: Sphere collisionetry
* Author: Nate Koenig
* Date: 21 May 2009
*/
#ifndef _BULLETSPHERESHAPE_HH_
#define _BULLETSPHERESHAPE_HH_
#include "gazebo/physics/bullet/BulletPhysics.hh"
#include "gazebo/physics/bullet/BulletCollision.hh"
#include "gazebo/physics/bullet/BulletLink.hh"
#include "gazebo/physics/World.hh"
#include "gazebo/physics/SphereShape.hh"
#include "gazebo/util/system.hh"
namespace gazebo
{
namespace physics
{
/// \ingroup gazebo_physics
/// \addtogroup gazebo_physics_bullet Bullet Physics
/// \{
/// \brief Bullet sphere collision
class GZ_PHYSICS_VISIBLE BulletSphereShape : public SphereShape
{
/// \brief Constructor
public: BulletSphereShape(CollisionPtr _parent) : SphereShape(_parent) {}
/// \brief Destructor
public: virtual ~BulletSphereShape() {}
/// \brief Set the radius
/// \param[in] _radius Sphere radius
public: void SetRadius(double _radius)
{
if (_radius < 0)
{
gzerr << "Sphere shape does not support negative radius\n";
return;
}
if (math::equal(_radius, 0.0))
{
// Warn user, but still create shape with very small value
// otherwise later resize operations using setLocalScaling
// will not be possible
gzwarn << "Setting sphere shape's radius to zero \n";
_radius = 1e-4;
}
SphereShape::SetRadius(_radius);
BulletCollisionPtr bParent;
bParent = boost::dynamic_pointer_cast<BulletCollision>(
this->collisionParent);
btCollisionShape *shape = bParent->GetCollisionShape();
if (!shape)
{
this->initialSize = math::Vector3(_radius, _radius, _radius);
bParent->SetCollisionShape(new btSphereShape(_radius));
}
else
{
btVector3 sphereScale;
sphereScale.setX(_radius / this->initialSize.x);
sphereScale.setY(_radius / this->initialSize.y);
sphereScale.setZ(_radius / this->initialSize.z);
shape->setLocalScaling(sphereScale);
// clear bullet cache and re-add the collision shape
// otherwise collisions won't work properly after scaling
BulletLinkPtr bLink =
boost::dynamic_pointer_cast<BulletLink>(
bParent->GetLink());
bLink->ClearCollisionCache();
// remove and add the shape again
if (bLink->GetBulletLink()->getCollisionShape()->isCompound())
{
btCompoundShape *compoundShape =
dynamic_cast<btCompoundShape *>(
bLink->GetBulletLink()->getCollisionShape());
compoundShape->removeChildShape(shape);
auto relativePose =
this->collisionParent->GetRelativePose().Ign()
- bLink->GetInertial()->GetPose().Ign();
compoundShape->addChildShape(
BulletTypes::ConvertPose(relativePose), shape);
}
}
}
/// \brief Initial size of sphere.
private: math::Vector3 initialSize;
};
/// \}
}
}
#endif