world.cpp

Go to the documentation of this file.

/*********************************************************************
 * Software License Agreement (BSD License)
 *
 *  Copyright (c) 2013, Willow Garage, Inc.
 *  All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions
 *  are met:
 *
 *   * Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *   * Redistributions in binary form must reproduce the above
 *     copyright notice, this list of conditions and the following
 *     disclaimer in the documentation and/or other materials provided
 *     with the distribution.
 *   * Neither the name of the Willow Garage nor the names of its
 *     contributors may be used to endorse or promote products derived
 *     from this software without specific prior written permission.
 *
 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 *  FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 *  COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 *  INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 *  BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 *  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 *  CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 *  LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
 *  ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 *  POSSIBILITY OF SUCH DAMAGE.
 *********************************************************************/

/* Author: Acorn Pooley, Ioan Sucan */

#include <moveit/collision_detection/world.h>
#include <console_bridge/console.h>

collision_detection::World::World()
{ }

collision_detection::World::World(const World &other)
{
  objects_ = other.objects_;
}

collision_detection::World::~World()
{
  while (!observers_.empty())
    removeObserver(observers_.front());
}

inline void collision_detection::World::addToObjectInternal(const ObjectPtr &obj,
                                                            const shapes::ShapeConstPtr &shape,
                                                            const Eigen::Affine3d &pose)
{
  obj->shapes_.push_back(shape);
  obj->shape_poses_.push_back(pose);
}

void collision_detection::World::addToObject(const std::string &id,
                                             const std::vector<shapes::ShapeConstPtr> &shapes,
                                             const EigenSTL::vector_Affine3d &poses)
{
  if (shapes.size() != poses.size())
  {
    logError("Number of shapes and number of poses do not match. Not adding this object to collision world.");
    return;
  }

  if (!shapes.size())
    return;

  int action = ADD_SHAPE;

  ObjectPtr& obj = objects_[id];
  if (!obj) {
    obj.reset(new Object(id));
    action |= CREATE;
  }

  ensureUnique(obj);

  for (std::size_t i = 0 ; i < shapes.size() ; ++i)
    addToObjectInternal(obj, shapes[i], poses[i]);

  notify(obj, Action(action));
}

void collision_detection::World::addToObject(const std::string &id,
                                             const shapes::ShapeConstPtr &shape,
                                             const Eigen::Affine3d &pose)
{
  int action = ADD_SHAPE;

  ObjectPtr& obj = objects_[id];
  if (!obj) {
    obj.reset(new Object(id));
    action |= CREATE;
  }

  ensureUnique(obj);
  addToObjectInternal(obj, shape, pose);

  notify(obj, Action(action));
}

std::vector<std::string> collision_detection::World::getObjectIds() const
{
  std::vector<std::string> id;
  for (std::map<std::string, ObjectPtr>::const_iterator it = objects_.begin() ; it != objects_.end() ; ++it)
    id.push_back(it->first);
  return id;
}

collision_detection::World::ObjectConstPtr collision_detection::World::getObject(const std::string &id) const
{
  std::map<std::string, ObjectPtr>::const_iterator it = objects_.find(id);
  if (it == objects_.end())
    return ObjectConstPtr();
  else
    return it->second;
}

void collision_detection::World::ensureUnique(ObjectPtr &obj)
{
  if (obj && !obj.unique())
    obj.reset(new Object(*obj));
}

bool collision_detection::World::hasObject(const std::string &id) const
{
  return objects_.find(id) != objects_.end();
}

bool collision_detection::World::moveShapeInObject(const std::string &id,
                                                   const shapes::ShapeConstPtr &shape,
                                                   const Eigen::Affine3d &pose)
{
  std::map<std::string, ObjectPtr>::iterator it = objects_.find(id);
  if (it != objects_.end())
  {
    unsigned int n = it->second->shapes_.size();
    for (unsigned int i = 0 ; i < n ; ++i)
      if (it->second->shapes_[i] == shape)
      {
        ensureUnique(it->second);
        it->second->shape_poses_[i] = pose;

        notify(it->second, MOVE_SHAPE);
        return true;
      }
  }
  return false;
}

bool collision_detection::World::removeShapeFromObject(const std::string &id,
                                                       const shapes::ShapeConstPtr &shape)
{
  std::map<std::string, ObjectPtr>::iterator it = objects_.find(id);
  if (it != objects_.end())
  {
    unsigned int n = it->second->shapes_.size();
    for (unsigned int i = 0 ; i < n ; ++i)
      if (it->second->shapes_[i] == shape)
      {
        ensureUnique(it->second);
        it->second->shapes_.erase(it->second->shapes_.begin() + i);
        it->second->shape_poses_.erase(it->second->shape_poses_.begin() + i);

        if (it->second->shapes_.empty())
        {
          notify(it->second, DESTROY);
          objects_.erase(it);
        }
        else
        {
          notify(it->second, REMOVE_SHAPE);
        }
        return true;
      }
  }
  return false;
}

bool collision_detection::World::removeObject(const std::string &id)
{
  std::map<std::string, ObjectPtr>::iterator it = objects_.find(id);
  if (it != objects_.end())
  {
    notify(it->second, DESTROY);
    objects_.erase(it);
    return true;
  }
  return false;
}

void collision_detection::World::clearObjects()
{
  notifyAll(DESTROY);
  objects_.clear();
}

collision_detection::World::ObserverHandle collision_detection::World::addObserver(const ObserverCallbackFn &callback)
{
  Observer *o = new Observer(callback);
  observers_.push_back(o);
  return ObserverHandle(o);
}

void collision_detection::World::removeObserver(ObserverHandle observer_handle)
{
  for (std::vector<Observer*>::iterator obs = observers_.begin() ; obs != observers_.end() ; ++obs)
  {
    if (*obs == observer_handle.observer_)
    {
      delete *obs;
      observers_.erase(obs);
      return;
    }
  }
}

void collision_detection::World::notifyAll(Action action)
{
  for (std::map<std::string, ObjectPtr>::const_iterator it = objects_.begin() ; it != objects_.end() ; ++it)
    notify(it->second, action);
}

void collision_detection::World::notify(const ObjectConstPtr& obj, Action action)
{
  for (std::vector<Observer*>::const_iterator obs = observers_.begin() ; obs != observers_.end() ; ++obs)
    (*obs)->callback_(obj, action);
}

void collision_detection::World::notifyObserverAllObjects(const ObserverHandle observer_handle, Action action) const
{
  for (std::vector<Observer*>::const_iterator obs = observers_.begin() ; obs != observers_.end() ; ++obs)
  {
    if (*obs == observer_handle.observer_)
    {
      // call the callback for each object
      for (std::map<std::string, ObjectPtr>::const_iterator obj = objects_.begin() ; obj != objects_.end() ; ++obj)
        (*obs)->callback_(obj->second, action);
      break;
    }
  }
}