/*
 * 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.
 *
*/

#ifdef _WIN32
  // Ensure that Winsock2.h is included before Windows.h, which can get
  // pulled in by anybody (e.g., Boost).
  #include <Winsock2.h>
#endif

#include <tbb/parallel_for.h>
#include <tbb/blocked_range.h>

#include <sdf/sdf.hh>

#include <algorithm>
#include <map>
#include <string>
#include <utility>
#include <vector>

#include <ignition/math/Rand.hh>
#include <ignition/math/Vector3.hh>

#include "gazebo/util/Diagnostics.hh"
#include "gazebo/common/Assert.hh"
#include "gazebo/common/Console.hh"
#include "gazebo/common/Exception.hh"
#include "gazebo/common/Time.hh"
#include "gazebo/common/Timer.hh"

#include "gazebo/transport/Publisher.hh"

#include "gazebo/physics/PhysicsTypes.hh"
#include "gazebo/physics/PhysicsFactory.hh"
#include "gazebo/physics/World.hh"
#include "gazebo/physics/Entity.hh"
#include "gazebo/physics/Model.hh"
#include "gazebo/physics/SurfaceParams.hh"
#include "gazebo/physics/Collision.hh"
#include "gazebo/physics/MapShape.hh"
#include "gazebo/physics/ContactManager.hh"

#include "gazebo/physics/ode/ODECollision.hh"
#include "gazebo/physics/ode/ODELink.hh"
#include "gazebo/physics/ode/ODEScrewJoint.hh"
#include "gazebo/physics/ode/ODEHingeJoint.hh"
#include "gazebo/physics/ode/ODEGearboxJoint.hh"
#include "gazebo/physics/ode/ODEHinge2Joint.hh"
#include "gazebo/physics/ode/ODESliderJoint.hh"
#include "gazebo/physics/ode/ODEBallJoint.hh"
#include "gazebo/physics/ode/ODEUniversalJoint.hh"
#include "gazebo/physics/ode/ODEFixedJoint.hh"

#include "gazebo/physics/ode/ODERayShape.hh"
#include "gazebo/physics/ode/ODEBoxShape.hh"
#include "gazebo/physics/ode/ODESphereShape.hh"
#include "gazebo/physics/ode/ODECylinderShape.hh"
#include "gazebo/physics/ode/ODEPlaneShape.hh"
#include "gazebo/physics/ode/ODEMeshShape.hh"
#include "gazebo/physics/ode/ODEMultiRayShape.hh"
#include "gazebo/physics/ode/ODEHeightmapShape.hh"
#include "gazebo/physics/ode/ODEPolylineShape.hh"

#include "gazebo/physics/ode/ODEPhysics.hh"
#include "gazebo/physics/ode/ODESurfaceParams.hh"

#include "gazebo/physics/ode/ODEPhysicsPrivate.hh"

using namespace gazebo;
using namespace physics;

GZ_REGISTER_PHYSICS_ENGINE("ode", ODEPhysics)

/*
class ContactUpdate_TBB
{
  public: ContactUpdate_TBB(tbb::concurrent_vector<ContactFeedback> *_contacts)
          : contacts(contacts) {}
  public: void operator() (const tbb::blocked_range<size_t> &_r) const
  {
    for (size_t i = _r.begin(); i != _r.end(); i++)
    {
      this->engine->ProcessContactFeedback((*this->contacts)[i]);
    }
  }

  private: tbb::concurrent_vector<ContactFeedback> *contacts;
  private: ODEPhysics *engine;
};
*/

class Colliders_TBB
{
  public: Colliders_TBB(
              std::vector<std::pair<ODECollision*, ODECollision*> > *_colliders,
              ODEPhysics *_engine,
              dContactGeom* _contactCollisions) :
    colliders(_colliders),
              engine(_engine), contactCollisions(_contactCollisions)
  {
    // dAllocateODEDataForThread(dAllocateMaskAll);
  }

  public: void operator() (const tbb::blocked_range<size_t> &_r) const
  {
    for (size_t i = _r.begin(); i != _r.end(); i++)
    {
      ODECollision *collision1 = (*this->colliders)[i].first;
      ODECollision *collision2 = (*this->colliders)[i].second;
      this->engine->Collide(collision1, collision2, contactCollisions);
    }
  }

  private: std::vector< std::pair<ODECollision*, ODECollision*> > *colliders;
  private: ODEPhysics *engine;
  private: dContactGeom* contactCollisions;
};

//////////////////////////////////////////////////
extern "C" void dMessageQuiet(int, const char *, va_list)
{
}

//////////////////////////////////////////////////
ODEPhysics::ODEPhysics(WorldPtr _world)
    : PhysicsEngine(_world), dataPtr(new ODEPhysicsPrivate)
{
  this->dataPtr->physicsStepFunc = nullptr;
  this->dataPtr->maxContacts = 0;

  // Collision detection init
  dInitODE2(0);

  dAllocateODEDataForThread(dAllocateMaskAll);

  this->dataPtr->worldId = dWorldCreate();

  this->dataPtr->spaceId = dHashSpaceCreate(0);
  dHashSpaceSetLevels(this->dataPtr->spaceId, -2, 8);

  this->dataPtr->contactGroup = dJointGroupCreate(0);

  this->dataPtr->colliders.resize(100);

  // Set random seed for physics engine based on gazebo's random seed.
  // Note: this was moved from physics::PhysicsEngine constructor.
  this->SetSeed(ignition::math::Rand::Seed());
}

//////////////////////////////////////////////////
ODEPhysics::~ODEPhysics()
{
  this->Fini();

  delete this->dataPtr;
  this->dataPtr = nullptr;
}

//////////////////////////////////////////////////
void ODEPhysics::Load(sdf::ElementPtr _sdf)
{
  PhysicsEngine::Load(_sdf);

  this->dataPtr->maxContacts = _sdf->Get<unsigned int>("max_contacts");
  this->SetMaxContacts(this->dataPtr->maxContacts);

  sdf::ElementPtr odeElem = this->sdf->GetElement("ode");
  sdf::ElementPtr solverElem = odeElem->GetElement("solver");

  this->dataPtr->stepType = solverElem->Get<std::string>("type");
  if (solverElem->HasElement("use_dynamic_moi_rescaling"))
  {
    dWorldSetQuickStepInertiaRatioReduction(this->dataPtr->worldId,
      solverElem->Get<bool>("use_dynamic_moi_rescaling"));
  }
  else
  {
    dWorldSetQuickStepInertiaRatioReduction(this->dataPtr->worldId, true);
  }

  /// \TODO: defaultvelocity decay!? This is BAD if it's true.
  dWorldSetDamping(this->dataPtr->worldId, 0.0001, 0.0001);

  // Help prevent "popping of deeply embedded object
  dWorldSetContactMaxCorrectingVel(this->dataPtr->worldId,
      odeElem->GetElement("constraints")->Get<double>(
        "contact_max_correcting_vel"));

  // This helps prevent jittering problems.
  dWorldSetContactSurfaceLayer(this->dataPtr->worldId,
       odeElem->GetElement("constraints")->Get<double>(
        "contact_surface_layer"));

  // Enable auto-disable by default. Models with joints are excluded from
  // auto-disable
  dWorldSetAutoDisableFlag(this->dataPtr->worldId, 1);

  dWorldSetAutoDisableTime(this->dataPtr->worldId, 1);
  dWorldSetAutoDisableLinearThreshold(this->dataPtr->worldId, 0.1);
  dWorldSetAutoDisableAngularThreshold(this->dataPtr->worldId, 0.1);
  dWorldSetAutoDisableSteps(this->dataPtr->worldId, 5);

  auto g = this->world->Gravity();

  if (g == ignition::math::Vector3d::Zero)
    gzwarn << "Gravity vector is (0, 0, 0). Objects will float.\n";

  dWorldSetGravity(this->dataPtr->worldId, g.X(), g.Y(), g.Z());

  if (odeElem->HasElement("constraints"))
  {
    dWorldSetCFM(this->dataPtr->worldId,
        odeElem->GetElement("constraints")->Get<double>("cfm"));
    dWorldSetERP(this->dataPtr->worldId,
        odeElem->GetElement("constraints")->Get<double>("erp"));
  }
  else
    dWorldSetERP(this->dataPtr->worldId, 0.2);

  dWorldSetQuickStepNumIterations(this->dataPtr->worldId,
    this->GetSORPGSIters());
  dWorldSetQuickStepW(this->dataPtr->worldId, this->GetSORPGSW());

  // Set the physics update function
  this->SetStepType(this->dataPtr->stepType);
  if (this->dataPtr->physicsStepFunc == nullptr)
    gzthrow(std::string("Invalid step type[") + this->dataPtr->stepType);
}

/////////////////////////////////////////////////
void ODEPhysics::OnRequest(ConstRequestPtr &_msg)
{
  msgs::Response response;
  response.set_id(_msg->id());
  response.set_request(_msg->request());
  response.set_response("success");
  std::string *serializedData = response.mutable_serialized_data();

  if (_msg->request() == "physics_info")
  {
    msgs::Physics physicsMsg;
    physicsMsg.set_type(msgs::Physics::ODE);
    physicsMsg.set_solver_type(this->dataPtr->stepType);
    // min_step_size is defined but not yet used
    boost::any min_step_size;
    try
    {
      if (this->GetParam("min_step_size", min_step_size))
        physicsMsg.set_min_step_size(boost::any_cast<double>(min_step_size));
    }
    catch(boost::bad_any_cast &_e)
    {
      gzerr << "Failed boost::any_cast in ODEPhysics.cc: " << _e.what();
    }
    physicsMsg.set_precon_iters(this->GetSORPGSPreconIters());
    physicsMsg.set_iters(this->GetSORPGSIters());
    physicsMsg.set_enable_physics(this->world->PhysicsEnabled());
    physicsMsg.set_sor(this->GetSORPGSW());
    physicsMsg.set_cfm(this->GetWorldCFM());
    physicsMsg.set_erp(this->GetWorldERP());
    physicsMsg.set_contact_max_correcting_vel(
      this->GetContactMaxCorrectingVel());
    physicsMsg.set_contact_surface_layer(
      this->GetContactSurfaceLayer());
    physicsMsg.mutable_gravity()->CopyFrom(
      msgs::Convert(this->world->Gravity()));
    physicsMsg.mutable_magnetic_field()->CopyFrom(
      msgs::Convert(this->world->MagneticField()));
    physicsMsg.set_real_time_update_rate(this->realTimeUpdateRate);
    physicsMsg.set_real_time_factor(this->targetRealTimeFactor);
    physicsMsg.set_max_step_size(this->maxStepSize);

    response.set_type(physicsMsg.GetTypeName());
    physicsMsg.SerializeToString(serializedData);
    this->responsePub->Publish(response);
  }
}

/////////////////////////////////////////////////
void ODEPhysics::OnPhysicsMsg(ConstPhysicsPtr &_msg)
{
  // Parent class handles many generic parameters
  // This should be done first so that the profile settings
  // can be over-ridden by other message parameters.
  PhysicsEngine::OnPhysicsMsg(_msg);

  if (_msg->has_solver_type())
    this->SetStepType(_msg->solver_type());

  if (_msg->has_min_step_size())
    this->SetParam("min_step_size", _msg->min_step_size());

  if (_msg->has_precon_iters())
    this->SetSORPGSPreconIters(_msg->precon_iters());

  if (_msg->has_iters())
    this->SetSORPGSIters(_msg->iters());

  if (_msg->has_sor())
    this->SetSORPGSW(_msg->sor());

  if (_msg->has_cfm())
    this->SetWorldCFM(_msg->cfm());

  if (_msg->has_erp())
    this->SetWorldERP(_msg->erp());

  if (_msg->has_enable_physics())
    this->world->SetPhysicsEnabled(_msg->enable_physics());

  if (_msg->has_contact_max_correcting_vel())
    this->SetContactMaxCorrectingVel(_msg->contact_max_correcting_vel());

  if (_msg->has_contact_surface_layer())
    this->SetContactSurfaceLayer(_msg->contact_surface_layer());

  if (_msg->has_gravity())
    this->SetGravity(msgs::ConvertIgn(_msg->gravity()));

  if (_msg->has_real_time_factor())
    this->SetTargetRealTimeFactor(_msg->real_time_factor());

  if (_msg->has_real_time_update_rate())
  {
    this->SetRealTimeUpdateRate(_msg->real_time_update_rate());
  }

  if (_msg->has_max_step_size())
  {
    this->SetMaxStepSize(_msg->max_step_size());
  }

  /// Make sure all models get at least on update cycle.
  this->world->EnableAllModels();
}



//////////////////////////////////////////////////
void ODEPhysics::Init()
{
}

//////////////////////////////////////////////////
void ODEPhysics::InitForThread()
{
  dAllocateODEDataForThread(dAllocateMaskAll);
}

//////////////////////////////////////////////////
void ODEPhysics::UpdateCollision()
{
  DIAG_TIMER_START("ODEPhysics::UpdateCollision");

  boost::recursive_mutex::scoped_lock lock(*this->physicsUpdateMutex);
  dJointGroupEmpty(this->dataPtr->contactGroup);

  unsigned int i = 0;
  this->dataPtr->collidersCount = 0;
  this->dataPtr->trimeshCollidersCount = 0;
  this->dataPtr->jointFeedbackIndex = 0;

  // Reset the contact count
  this->contactManager->ResetCount();

  // Do collision detection; this will add contacts to the contact group
  dSpaceCollide(this->dataPtr->spaceId, this, CollisionCallback);
  DIAG_TIMER_LAP("ODEPhysics::UpdateCollision", "dSpaceCollide");

  // Generate non-trimesh collisions.
  for (i = 0; i < this->dataPtr->collidersCount; ++i)
  {
    this->Collide(this->dataPtr->colliders[i].first,
        this->dataPtr->colliders[i].second, this->dataPtr->contactCollisions);
  }
  DIAG_TIMER_LAP("ODEPhysics::UpdateCollision", "collideShapes");

  // Generate trimesh collision.
  // This must happen in this thread sequentially
  for (i = 0; i < this->dataPtr->trimeshCollidersCount; ++i)
  {
    ODECollision *collision1 = this->dataPtr->trimeshColliders[i].first;
    ODECollision *collision2 = this->dataPtr->trimeshColliders[i].second;
    this->Collide(collision1, collision2, this->dataPtr->contactCollisions);
  }
  DIAG_TIMER_LAP("ODEPhysics::UpdateCollision", "collideTrimeshes");

  DIAG_TIMER_STOP("ODEPhysics::UpdateCollision");
}

//////////////////////////////////////////////////
void ODEPhysics::UpdatePhysics()
{
  DIAG_TIMER_START("ODEPhysics::UpdatePhysics");

  // need to lock, otherwise might conflict with world resetting
  {
    boost::recursive_mutex::scoped_lock lock(*this->physicsUpdateMutex);

    // Update the dynamical model
    (*(this->dataPtr->physicsStepFunc))
      (this->dataPtr->worldId, this->maxStepSize);

    ignition::math::Vector3d f1, f2, t1, t2;

    // Set the joint contact feedback for each contact.
    for (unsigned int i = 0; i < this->dataPtr->jointFeedbackIndex; ++i)
    {
      Contact *contactFeedback = this->dataPtr->jointFeedbacks[i]->contact;
      Collision *col1 = contactFeedback->collision1;
      Collision *col2 = contactFeedback->collision2;

      GZ_ASSERT(col1 != nullptr, "Collision 1 is null");
      GZ_ASSERT(col2 != nullptr, "Collision 2 is null");

      for (int j = 0; j < this->dataPtr->jointFeedbacks[i]->count; ++j)
      {
        dJointFeedback fb = this->dataPtr->jointFeedbacks[i]->feedbacks[j];
        f1.Set(fb.f1[0], fb.f1[1], fb.f1[2]);
        f2.Set(fb.f2[0], fb.f2[1], fb.f2[2]);
        t1.Set(fb.t1[0], fb.t1[1], fb.t1[2]);
        t2.Set(fb.t2[0], fb.t2[1], fb.t2[2]);

        // set force torque in link frame
        this->dataPtr->jointFeedbacks[i]->contact->wrench[j].body1Force =
             col1->GetLink()->WorldPose().Rot().RotateVectorReverse(f1);
        this->dataPtr->jointFeedbacks[i]->contact->wrench[j].body2Force =
             col2->GetLink()->WorldPose().Rot().RotateVectorReverse(f2);
        this->dataPtr->jointFeedbacks[i]->contact->wrench[j].body1Torque =
             col1->GetLink()->WorldPose().Rot().RotateVectorReverse(t1);
        this->dataPtr->jointFeedbacks[i]->contact->wrench[j].body2Torque =
             col2->GetLink()->WorldPose().Rot().RotateVectorReverse(t2);
      }
    }
  }

  DIAG_TIMER_STOP("ODEPhysics::UpdatePhysics");
}

//////////////////////////////////////////////////
void ODEPhysics::Fini()
{
  dCloseODE();

  if (this->dataPtr->contactGroup)
    dJointGroupDestroy(this->dataPtr->contactGroup);
  this->dataPtr->contactGroup = nullptr;

  // Delete all the joint feedbacks.
  for (auto iter = this->dataPtr->jointFeedbacks.begin();
      iter != this->dataPtr->jointFeedbacks.end(); ++iter)
  {
    delete *iter;
  }
  this->dataPtr->jointFeedbacks.clear();

  if (this->dataPtr->spaceId)
  {
    dSpaceSetCleanup(this->dataPtr->spaceId, 0);
    dSpaceDestroy(this->dataPtr->spaceId);
  }

  if (this->dataPtr->worldId)
    dWorldDestroy(this->dataPtr->worldId);
  this->dataPtr->worldId = nullptr;

  this->dataPtr->spaceId = nullptr;

  PhysicsEngine::Fini();
}

//////////////////////////////////////////////////
void ODEPhysics::Reset()
{
  boost::recursive_mutex::scoped_lock lock(*this->physicsUpdateMutex);
  // Very important to clear out the contact group
  dJointGroupEmpty(this->dataPtr->contactGroup);
}

//////////////////////////////////////////////////
LinkPtr ODEPhysics::CreateLink(ModelPtr _parent)
{
  if (_parent == nullptr)
    gzthrow("Link must have a parent\n");

  std::map<std::string, dSpaceID>::iterator iter;
  iter = this->dataPtr->spaces.find(_parent->GetName());

  if (iter == this->dataPtr->spaces.end())
    this->dataPtr->spaces[_parent->GetName()] =
      dSimpleSpaceCreate(this->dataPtr->spaceId);

  ODELinkPtr link(new ODELink(_parent));

  link->SetSpaceId(this->dataPtr->spaces[_parent->GetName()]);
  link->SetWorld(_parent->GetWorld());

  return link;
}

//////////////////////////////////////////////////
CollisionPtr ODEPhysics::CreateCollision(const std::string &_type,
                                         LinkPtr _body)
{
  ODECollisionPtr collision(new ODECollision(_body));
  ShapePtr shape = this->CreateShape(_type, collision);
  collision->SetShape(shape);
  shape->SetWorld(_body->GetWorld());
  return collision;
}

//////////////////////////////////////////////////
ShapePtr ODEPhysics::CreateShape(const std::string &_type,
                                 CollisionPtr _collision)
{
  ShapePtr shape;
  ODECollisionPtr collision =
    boost::dynamic_pointer_cast<ODECollision>(_collision);

  if (_type == "sphere")
    shape.reset(new ODESphereShape(collision));
  else if (_type == "plane")
    shape.reset(new ODEPlaneShape(collision));
  else if (_type == "box")
    shape.reset(new ODEBoxShape(collision));
  else if (_type == "cylinder")
    shape.reset(new ODECylinderShape(collision));
  else if (_type == "polyline")
    shape.reset(new ODEPolylineShape(collision));
  else if (_type == "multiray")
  {
    if (_collision)
      shape.reset(new ODEMultiRayShape(collision));
    else
      shape.reset(new ODEMultiRayShape(this->world->Physics()));
  }
  else if (_type == "mesh" || _type == "trimesh")
    shape.reset(new ODEMeshShape(collision));
  else if (_type == "heightmap")
    shape.reset(new ODEHeightmapShape(collision));
  else if (_type == "map" || _type == "image")
    shape.reset(new MapShape(collision));
  else if (_type == "ray")
  {
    if (_collision)
      shape.reset(new ODERayShape(collision));
    else
      shape.reset(new ODERayShape(this->world->Physics()));
  }
  else
    gzerr << "Unable to create collision of type[" << _type << "]\n";

  return shape;
}

//////////////////////////////////////////////////
dWorldID ODEPhysics::GetWorldId()
{
  return this->dataPtr->worldId;
}

//////////////////////////////////////////////////
void ODEPhysics::ConvertMass(InertialPtr _inertial, void *_engineMass)
{
  dMass *odeMass = static_cast<dMass*>(_engineMass);

  _inertial->SetMass(odeMass->mass);
  _inertial->SetCoG(odeMass->c[0], odeMass->c[1], odeMass->c[2]);
  _inertial->SetInertiaMatrix(odeMass->I[0*4+0], odeMass->I[1*4+1],
      odeMass->I[2*4+2], odeMass->I[0*4+1],
      odeMass->I[0*4+2], odeMass->I[1*4+2]);
}

//////////////////////////////////////////////////
Friction_Model ODEPhysics::ConvertFrictionModel(const std::string &_fricModel)
{
  Friction_Model result = pyramid_friction;
  if (_fricModel.compare("pyramid_model") == 0)
      result = pyramid_friction;
  else if (_fricModel.compare("cone_model") == 0)
      result = cone_friction;
  else if (_fricModel.compare("box_model") == 0)
      result = box_friction;
  else
    gzerr << "Unrecognized friction model ["
          << _fricModel
          << "], returning pyramid friction"
          << std::endl;
  return result;
}

//////////////////////////////////////////////////
std::string ODEPhysics::ConvertFrictionModel(const Friction_Model _fricModel)
{
  std::string result;
  switch (_fricModel)
  {
    case pyramid_friction:
    {
      result = "pyramid_model";
      break;
    }
    case cone_friction:
    {
      result = "cone_model";
      break;
    }
    case box_friction:
    {
      result = "box_model";
      break;
    }
    default:
    {
      result = "unknown";
      gzerr << "Unrecognized friction model [" << _fricModel << "]"
            << std::endl;
    }
  }
  return result;
}

//////////////////////////////////////////////////
World_Solver_Type
ODEPhysics::ConvertWorldStepSolverType(const std::string &_solverType)
{
  World_Solver_Type result = ODE_DEFAULT;
  if (_solverType.compare("ODE_DANTZIG") == 0)
    result = ODE_DEFAULT;
  else if (_solverType.compare("DART_PGS") == 0)
    result = DART_PGS;
  else if (_solverType.compare("BULLET_LEMKE") == 0)
    result = BULLET_LEMKE;
  else if (_solverType.compare("BULLET_PGS") == 0)
    result = BULLET_PGS;
  else
  {
    gzerr << "Unrecognized world step solver ["
          << _solverType
          << "], returning ODE_DANTZIG"
          << std::endl;
  }
  return result;
}

//////////////////////////////////////////////////
std::string
ODEPhysics::ConvertWorldStepSolverType(const World_Solver_Type _solverType)
{
  std::string result;
  switch (_solverType)
  {
    case ODE_DEFAULT:
    {
      result = "ODE_DANTZIG";
      break;
    }
    case DART_PGS:
    {
      result = "DART_PGS";
      break;
    }
    case BULLET_LEMKE:
    {
      result = "BULLET_LEMKE";
      break;
    }
    case BULLET_PGS:
    {
      result = "BULLET_PGS";
      break;
    }
    default:
    {
      result = "unknown";
      gzerr << "Unrecognized world step solver [" << _solverType << "]"
            << std::endl;
    }
  }
  return result;
}

//////////////////////////////////////////////////
void ODEPhysics::SetSORPGSPreconIters(unsigned int _iters)
{
  this->sdf->GetElement("ode")->GetElement("solver")->
    GetElement("precon_iters")->Set(_iters);

  dWorldSetQuickStepPreconIterations(this->dataPtr->worldId, _iters);
}

//////////////////////////////////////////////////
void ODEPhysics::SetSORPGSIters(unsigned int _iters)
{
  this->sdf->GetElement("ode")->GetElement(
      "solver")->GetElement("iters")->Set(_iters);
  dWorldSetQuickStepNumIterations(this->dataPtr->worldId, _iters);
}

//////////////////////////////////////////////////
void ODEPhysics::SetSORPGSW(double _w)
{
  this->sdf->GetElement("ode")->GetElement(
      "solver")->GetElement("sor")->Set(_w);
  dWorldSetQuickStepW(this->dataPtr->worldId, _w);
}

//////////////////////////////////////////////////
void ODEPhysics::SetFrictionModel(const std::string &_fricModel)
{
  /// Uncomment this until sdformat changes (sdformat repo issue #96)
  ///
  /// this->sdf->GetElement("ode")->GetElement(
  ///   "solver")->GetElement("friction_model")->Set(_fricModel);
  dWorldSetQuickStepFrictionModel(this->dataPtr->worldId,
    ConvertFrictionModel(_fricModel));
}

//////////////////////////////////////////////////
void ODEPhysics::SetWorldCFM(double _cfm)
{
  sdf::ElementPtr elem = this->sdf->GetElement("ode");
  elem = elem->GetElement("constraints");
  elem->GetElement("cfm")->Set(_cfm);

  dWorldSetCFM(this->dataPtr->worldId, _cfm);
}

//////////////////////////////////////////////////
void ODEPhysics::SetWorldERP(double _erp)
{
  sdf::ElementPtr elem = this->sdf->GetElement("ode");
  elem = elem->GetElement("constraints");
  elem->GetElement("erp")->Set(_erp);
  dWorldSetERP(this->dataPtr->worldId, _erp);
}

//////////////////////////////////////////////////
void ODEPhysics::SetContactMaxCorrectingVel(double _vel)
{
  this->sdf->GetElement("ode")->GetElement(
      "constraints")->GetElement(
        "contact_max_correcting_vel")->Set(_vel);
  dWorldSetContactMaxCorrectingVel(this->dataPtr->worldId, _vel);
}

//////////////////////////////////////////////////
void ODEPhysics::SetContactSurfaceLayer(double _depth)
{
  this->sdf->GetElement("ode")->GetElement(
      "constraints")->GetElement("contact_surface_layer")->Set(_depth);
  dWorldSetContactSurfaceLayer(this->dataPtr->worldId, _depth);
}

//////////////////////////////////////////////////
void ODEPhysics::SetMaxContacts(unsigned int _maxContacts)
{
  this->dataPtr->maxContacts = _maxContacts;
  this->sdf->GetElement("max_contacts")->GetValue()->Set(_maxContacts);
}

//////////////////////////////////////////////////
void ODEPhysics::SetWorldStepSolverType(const std::string &_worldSolverType)
{
    dWorldSetWorldStepSolverType(this->dataPtr->worldId,
    ConvertWorldStepSolverType(_worldSolverType));
}

//////////////////////////////////////////////////
int ODEPhysics::GetSORPGSPreconIters()
{
  return this->sdf->GetElement("ode")->GetElement(
      "solver")->Get<int>("precon_iters");
}
//////////////////////////////////////////////////
int ODEPhysics::GetSORPGSIters()
{
  return this->sdf->GetElement("ode")->GetElement(
      "solver")->Get<int>("iters");
}

//////////////////////////////////////////////////
double ODEPhysics::GetSORPGSW()
{
  return this->sdf->GetElement("ode")->GetElement(
      "solver")->Get<double>("sor");
}

//////////////////////////////////////////////////
std::string ODEPhysics::GetFrictionModel() const
{
  return ConvertFrictionModel(
    dWorldGetQuickStepFrictionModel(this->dataPtr->worldId));
}

//////////////////////////////////////////////////
std::string ODEPhysics::GetWorldStepSolverType() const
{
  return ConvertWorldStepSolverType(
    dWorldGetWorldStepSolverType(this->dataPtr->worldId));
}

//////////////////////////////////////////////////
double ODEPhysics::GetWorldCFM()
{
  sdf::ElementPtr elem = this->sdf->GetElement("ode");
  elem = elem->GetElement("constraints");
  return elem->Get<double>("cfm");
}

//////////////////////////////////////////////////
double ODEPhysics::GetWorldERP()
{
  sdf::ElementPtr elem = this->sdf->GetElement("ode");
  elem = elem->GetElement("constraints");
  return elem->Get<double>("erp");
}

//////////////////////////////////////////////////
double ODEPhysics::GetContactMaxCorrectingVel()
{
  return this->sdf->GetElement("ode")->GetElement(
      "constraints")->Get<double>("contact_max_correcting_vel");
}

//////////////////////////////////////////////////
double ODEPhysics::GetContactSurfaceLayer()
{
  return this->sdf->GetElement("ode")->GetElement(
      "constraints")->Get<double>("contact_surface_layer");
}

//////////////////////////////////////////////////
unsigned int ODEPhysics::GetMaxContacts()
{
  return this->dataPtr->maxContacts;
}

//////////////////////////////////////////////////
void ODEPhysics::ConvertMass(void *_engineMass, InertialPtr _inertial)
{
  dMass *odeMass = static_cast<dMass*>(_engineMass);

  odeMass->mass = _inertial->Mass();
  odeMass->c[0] = _inertial->CoG()[0];
  odeMass->c[1] = _inertial->CoG()[1];
  odeMass->c[2] = _inertial->CoG()[2];

  odeMass->I[0*4+0] = _inertial->PrincipalMoments()[0];
  odeMass->I[1*4+1] = _inertial->PrincipalMoments()[1];
  odeMass->I[2*4+2] = _inertial->PrincipalMoments()[2];

  odeMass->I[0*4+1] = _inertial->ProductsOfInertia()[0];
  odeMass->I[1*4+0] = _inertial->ProductsOfInertia()[0];

  odeMass->I[0*4+2] = _inertial->ProductsOfInertia()[1];
  odeMass->I[1*4+0] = _inertial->ProductsOfInertia()[1];

  odeMass->I[1*4+2] = _inertial->ProductsOfInertia()[2];
  odeMass->I[2*4+1] = _inertial->ProductsOfInertia()[2];
}

//////////////////////////////////////////////////
JointPtr ODEPhysics::CreateJoint(const std::string &_type, ModelPtr _parent)
{
  JointPtr joint;

  if (_type == "prismatic")
    joint.reset(new ODESliderJoint(this->dataPtr->worldId, _parent));
  else if (_type == "screw")
    joint.reset(new ODEScrewJoint(this->dataPtr->worldId, _parent));
  else if (_type == "revolute")
    joint.reset(new ODEHingeJoint(this->dataPtr->worldId, _parent));
  else if (_type == "gearbox")
    joint.reset(new ODEGearboxJoint(this->dataPtr->worldId, _parent));
  else if (_type == "revolute2")
    joint.reset(new ODEHinge2Joint(this->dataPtr->worldId, _parent));
  else if (_type == "ball")
    joint.reset(new ODEBallJoint(this->dataPtr->worldId, _parent));
  else if (_type == "universal")
    joint.reset(new ODEUniversalJoint(this->dataPtr->worldId, _parent));
  else if (_type == "fixed")
    joint.reset(new ODEFixedJoint(this->dataPtr->worldId, _parent));
  else
    gzthrow("Unable to create joint of type[" << _type << "]");

  return joint;
}

//////////////////////////////////////////////////
dSpaceID ODEPhysics::GetSpaceId() const
{
  return this->dataPtr->spaceId;
}

//////////////////////////////////////////////////
std::string ODEPhysics::GetStepType() const
{
  sdf::ElementPtr elem = this->sdf->GetElement("ode")->GetElement("solver");
  return elem->Get<std::string>("type");
}

//////////////////////////////////////////////////
void ODEPhysics::SetStepType(const std::string &_type)
{
  sdf::ElementPtr elem = this->sdf->GetElement("ode")->GetElement("solver");
  elem->GetElement("type")->Set(_type);
  this->dataPtr->stepType = _type;

  // Set the physics update function
  if (this->dataPtr->stepType == "quick")
    this->dataPtr->physicsStepFunc = &dWorldQuickStep;
  else if (this->dataPtr->stepType == "world")
    this->dataPtr->physicsStepFunc = &dWorldStep;
  else
    gzerr << "Invalid step type[" << this->dataPtr->stepType
          << "]" << std::endl;
}

//////////////////////////////////////////////////
void ODEPhysics::SetGravity(const ignition::math::Vector3d &_gravity)
{
  this->world->SetGravitySDF(_gravity);
  dWorldSetGravity(this->dataPtr->worldId, _gravity.X(), _gravity.Y(),
      _gravity.Z());
}

//////////////////////////////////////////////////
void ODEPhysics::CollisionCallback(void *_data, dGeomID _o1, dGeomID _o2)
{
  dBodyID b1 = dGeomGetBody(_o1);
  dBodyID b2 = dGeomGetBody(_o2);

  // exit without doing anything if the two bodies are connected by a joint
  if (b1 && b2 && dAreConnectedExcluding(b1, b2, dJointTypeContact))
    return;

  // Get a pointer to the physics engine
  ODEPhysics *self = static_cast<ODEPhysics*>(_data);

  // Check if either are spaces
  if (dGeomIsSpace(_o1) || dGeomIsSpace(_o2))
  {
    dSpaceCollide2(_o1, _o2, self, &CollisionCallback);
  }
  else
  {
    ODECollision *collision1 = nullptr;
    ODECollision *collision2 = nullptr;

    // Exit if both bodies are not enabled
    if (dGeomGetCategoryBits(_o1) != GZ_SENSOR_COLLIDE &&
        dGeomGetCategoryBits(_o2) != GZ_SENSOR_COLLIDE &&
        ((b1 && b2 && !dBodyIsEnabled(b1) && !dBodyIsEnabled(b2)) ||
        (!b2 && b1 && !dBodyIsEnabled(b1)) ||
        (!b1 && b2 && !dBodyIsEnabled(b2))))
    {
      return;
    }

    // Get pointers to the underlying collisions
    if (dGeomGetClass(_o1) == dGeomTransformClass)
      collision1 =
        static_cast<ODECollision*>(dGeomGetData(dGeomTransformGetGeom(_o1)));
    else
      collision1 = static_cast<ODECollision*>(dGeomGetData(_o1));

    if (dGeomGetClass(_o2) == dGeomTransformClass)
      collision2 =
        static_cast<ODECollision*>(dGeomGetData(dGeomTransformGetGeom(_o2)));
    else
      collision2 = static_cast<ODECollision*>(dGeomGetData(_o2));

    // Make sure both collision pointers are valid.
    if (collision1 && collision2)
    {
      // Add either a tri-mesh collider or a regular collider.
      if (collision1->HasType(Base::MESH_SHAPE) ||
          collision2->HasType(Base::MESH_SHAPE))
        self->AddTrimeshCollider(collision1, collision2);
      else
      {
        self->AddCollider(collision1, collision2);
      }
    }
  }
}


//////////////////////////////////////////////////
void ODEPhysics::Collide(ODECollision *_collision1, ODECollision *_collision2,
                         dContactGeom *_contactCollisions)
{
  // Filter collisions based on collide bitmask.
  if ((_collision1->GetSurface()->collideBitmask &
        _collision2->GetSurface()->collideBitmask) == 0)
    return;

  // Filter collisions based on contact bitmask if collide_without_contact is
  // on.The bitmask is set mainly for speed improvements otherwise a collision
  // with collide_without_contact may potentially generate a large number of
  // contacts.
  if (_collision1->GetSurface()->collideWithoutContact ||
      _collision2->GetSurface()->collideWithoutContact)
  {
    if ((_collision1->GetSurface()->collideWithoutContactBitmask &
         _collision2->GetSurface()->collideWithoutContactBitmask) == 0)
    {
      return;
    }
  }

  /*
  if (_collision1->GetCollisionId() && _collision2->GetCollisionId())
  {
    const dVector3 *pos1 =
      (const dVector3*)dGeomGetPosition(_collision1->GetCollisionId());
    const dVector3 *pos2 =
      (const dVector3*)dGeomGetPosition(_collision2->GetCollisionId());
    std::cout << "1[" << (*pos1)[0]<< " " << (*pos1)[1] << " "
              << (*pos1)[2] << "] "
      << "2[" << (*pos2)[0]<< " " << (*pos2)[1] << " " << (*pos2)[2] << "]\n";
  }*/

  unsigned int numc = 0;
  dContact contact;

  // maxCollide must less than the size of this->dataPtr->indices
  // Check the header
  unsigned int maxCollide = MAX_CONTACT_JOINTS;

  // max_contacts specified globally
  if (this->GetMaxContacts() > 0 && this->GetMaxContacts() < MAX_CONTACT_JOINTS)
    maxCollide = this->GetMaxContacts();

  // over-ride with minimum of max_contacts from both collisions
  if (_collision1->GetMaxContacts() < maxCollide)
    maxCollide = _collision1->GetMaxContacts();

  if (_collision2->GetMaxContacts() < maxCollide)
    maxCollide = _collision2->GetMaxContacts();

  // Generate the contacts
  numc = dCollide(_collision1->GetCollisionId(), _collision2->GetCollisionId(),
      MAX_COLLIDE_RETURNS, _contactCollisions, sizeof(_contactCollisions[0]));

  // Return if no contacts.
  if (numc == 0)
    return;

  // Store the indices of the contacts.
  for (int i = 0; i < MAX_CONTACT_JOINTS; i++)
    this->dataPtr->indices[i] = i;

  // Choose only the best contacts if too many were generated.
  if (maxCollide > 0 && numc > maxCollide)
  {
    double max = _contactCollisions[maxCollide-1].depth;
    for (unsigned int i = maxCollide; i < numc; ++i)
    {
      if (_contactCollisions[i].depth > max)
      {
        max = _contactCollisions[i].depth;
        this->dataPtr->indices[maxCollide-1] = i;
      }
    }

    // Make sure numc has the valid number of contacts.
    numc = maxCollide;
  }

  // Set the contact surface parameter flags.
  contact.surface.mode = dContactBounce |
                         dContactMu2 |
                         dContactSoftERP |
                         dContactSoftCFM |
                         dContactApprox1 |
                         dContactApprox3 |
                         dContactSlip1 |
                         dContactSlip2;

  ODESurfaceParamsPtr surf1 = _collision1->GetODESurface();
  ODESurfaceParamsPtr surf2 = _collision2->GetODESurface();

  // Compute the CFM and ERP by assuming the two bodies form a
  // spring-damper system.
  double kp = 1.0 / (1.0 / surf1->kp + 1.0 / surf2->kp);
  double kd = surf1->kd + surf2->kd;

  contact.surface.soft_erp = (this->maxStepSize * kp) /
                             (this->maxStepSize * kp + kd);

  contact.surface.soft_cfm = 1.0 / (this->maxStepSize * kp + kd);

  // contact.surface.soft_erp = 0.5*(_collision1->surface->softERP +
  //                                _collision2->surface->softERP);
  // contact.surface.soft_cfm = 0.5*(_collision1->surface->softCFM +
  //                                _collision2->surface->softCFM);

  // assign fdir1 if not set as 0
  ignition::math::Vector3d fd = surf1->FrictionPyramid()->direction1;
  if (fd != ignition::math::Vector3d::Zero)
  {
    // fdir1 is in body local frame, rotate it into world frame
    fd = _collision1->WorldPose().Rot().RotateVector(fd);
  }

  /// \TODO: Better treatment when both surfaces have fdir1 specified.
  /// Ideally, we want to use fdir1 specified by surface with
  /// a smaller friction coefficient, but it's not clear how
  /// that can be determined with friction pyramid approximations.
  /// As a hack, we'll simply compare mu1 from
  /// both surfaces for now, and use fdir1 specified by
  /// surface with smaller mu1.
  ignition::math::Vector3d fd2 = surf2->FrictionPyramid()->direction1;
  if (fd2 != ignition::math::Vector3d::Zero &&
      (fd == ignition::math::Vector3d::Zero ||
        surf1->FrictionPyramid()->MuPrimary() >
        surf2->FrictionPyramid()->MuPrimary()))
  {
    // fdir1 is in body local frame, rotate it into world frame
    fd2 = _collision2->WorldPose().Rot().RotateVector(fd2);

    /// \TODO: uncomment gzlog below once we confirm it does not affect
    /// performance
    /// if (fd2 != ignition::math::Vector3d::Zero &&
    ///       fd != ignition::math::Vector3d::Zero &&
    ///       _collision1->surface->mu1 > _collision2->surface->mu1)
    ///   gzlog << "both contact surfaces have non-zero fdir1, comparing"
    ///         << " comparing mu1 from both surfaces, and use fdir1"
    ///         << " from surface with smaller mu1\n";
  }

  if (fd != ignition::math::Vector3d::Zero)
  {
    contact.surface.mode |= dContactFDir1;
    contact.fdir1[0] = fd.X();
    contact.fdir1[1] = fd.Y();
    contact.fdir1[2] = fd.Z();
  }

  // Set the friction coefficients.
  contact.surface.mu = std::min(surf1->FrictionPyramid()->MuPrimary(),
                                surf2->FrictionPyramid()->MuPrimary());
  contact.surface.mu2 = std::min(surf1->FrictionPyramid()->MuSecondary(),
                                 surf2->FrictionPyramid()->MuSecondary());
  contact.surface.mu3 = std::min(surf1->FrictionPyramid()->MuTorsion(),
                                 surf2->FrictionPyramid()->MuTorsion());

  // Combine the slip values
  // The slip is equivalent to the inverse of a viscous damping term
  // To combine dampers in series, the inverse of damping is summed
  // So the sum of slip parameters is used to combine them
  contact.surface.slip1 = surf1->slip1 + surf2->slip1;
  contact.surface.slip2 = surf1->slip2 + surf2->slip2;
  contact.surface.slip3 = surf1->slipTorsion + surf2->slipTorsion;

  // Combine torsional friction patch radius values
  contact.surface.patch_radius =
      std::max(surf1->FrictionPyramid()->PatchRadius(),
               surf2->FrictionPyramid()->PatchRadius());

  // For torsional friction, the curvature is combined using
  //   1/R = 1/R1 + 1/R2
  // we can consider doing the same for the patch radius
  double curv1 = 0;
  if (surf1->FrictionPyramid()->SurfaceRadius() > 0)
    curv1 = 1 / surf1->FrictionPyramid()->SurfaceRadius();

  double curv2 = 0;
  if (surf2->FrictionPyramid()->SurfaceRadius() > 0)
    curv2 = 1 / surf2->FrictionPyramid()->SurfaceRadius();

  double curvSum = curv1 + curv2;
  contact.surface.surface_radius = 0;
  if (curvSum > 0)
    contact.surface.surface_radius = 1 / curvSum;

  /// \todo Not sure how to combine these logic flags
  /// If user wanted to use patch radius, but got settings
  /// overwritten by the logic combination, how do we make sure the
  /// the surface radius is specified or makes sense?
  contact.surface.use_patch_radius =
      surf1->FrictionPyramid()->UsePatchRadius() &&
      surf2->FrictionPyramid()->UsePatchRadius();

  if (contact.surface.mu3 > 0)
  {
    // Patch radius
    if ((contact.surface.use_patch_radius &&
        contact.surface.patch_radius > 0) ||
    // Surface radius
        (!contact.surface.use_patch_radius &&
        contact.surface.surface_radius > 0))
    {
      contact.surface.mode |= dContactMu3;

      if (contact.surface.slip3 > 0)
      {
        contact.surface.mode |= dContactSlip3;
      }
    }
  }

  // Set the elastic modulus
  // Using Hertzian contact
  // equation 5.26 from Contact Mechanics and Friction by Popov
  double nu1 = surf1->FrictionPyramid()->PoissonsRatio();
  double nu2 = surf2->FrictionPyramid()->PoissonsRatio();
  double e1 = surf1->FrictionPyramid()->ElasticModulus();
  double e2 = surf2->FrictionPyramid()->ElasticModulus();
  if (e1 > 0 && e2 > 0)
  {
    contact.surface.elastic_modulus = 1.0 /
      ((1.0 - nu1*nu1)/e1 + (1.0 - nu2*nu2)/e2);

    // Turn on Contact Elastic Modulus model if elastic modulus > 0
    if (contact.surface.elastic_modulus > 0.0)
    {
      contact.surface.mode |= dContactEM;
    }
  }

  // Set the bounce values
  contact.surface.bounce = std::min(surf1->bounce,
                                    surf2->bounce);
  contact.surface.bounce_vel =
    std::min(surf1->bounceThreshold,
             surf2->bounceThreshold);

  // Get the ODE body IDs
  dBodyID b1 = dGeomGetBody(_collision1->GetCollisionId());
  dBodyID b2 = dGeomGetBody(_collision2->GetCollisionId());

  // Add a new contact to the manager. This will return nullptr if no one is
  // listening for contact information.
  Contact *contactFeedback = this->contactManager->NewContact(_collision1,
      _collision2, this->world->SimTime());

  ODEJointFeedback *jointFeedback = nullptr;

  // Create a joint feedback mechanism
  if (contactFeedback)
  {
    if (this->dataPtr->jointFeedbackIndex <
        this->dataPtr->jointFeedbacks.size())
      jointFeedback =
          this->dataPtr->jointFeedbacks[this->dataPtr->jointFeedbackIndex];
    else
    {
      jointFeedback = new ODEJointFeedback();
      this->dataPtr->jointFeedbacks.push_back(jointFeedback);
    }

    this->dataPtr->jointFeedbackIndex++;
    jointFeedback->count = 0;
    jointFeedback->contact = contactFeedback;
  }

  // Create a joint for each contact
  for (unsigned int j = 0; j < numc; ++j)
  {
    contact.geom = _contactCollisions[this->dataPtr->indices[j]];

    // Create the contact joint. This introduces the contact constraint to
    // ODE
    dJointID contactJoint = dJointCreateContact(this->dataPtr->worldId,
      this->dataPtr->contactGroup, &contact);

    // Store contact information.
    if (contactFeedback && jointFeedback)
    {
      // Store the contact depth
      contactFeedback->depths[j] =
        _contactCollisions[this->dataPtr->indices[j]].depth;

      // Store the contact position
      contactFeedback->positions[j].Set(
          _contactCollisions[this->dataPtr->indices[j]].pos[0],
          _contactCollisions[this->dataPtr->indices[j]].pos[1],
          _contactCollisions[this->dataPtr->indices[j]].pos[2]);

      // Store the contact normal
      contactFeedback->normals[j].Set(
          _contactCollisions[this->dataPtr->indices[j]].normal[0],
          _contactCollisions[this->dataPtr->indices[j]].normal[1],
          _contactCollisions[this->dataPtr->indices[j]].normal[2]);

      // Set the joint feedback.
      dJointSetFeedback(contactJoint, &(jointFeedback->feedbacks[j]));

      // Increase the counters
      contactFeedback->count++;
      jointFeedback->count++;
    }

    // Attach the contact joint if collideWithoutContact flags aren't set.
    if (!_collision1->GetSurface()->collideWithoutContact &&
        !_collision2->GetSurface()->collideWithoutContact)
      dJointAttach(contactJoint, b1, b2);
  }
}

/////////////////////////////////////////////////
void ODEPhysics::AddTrimeshCollider(ODECollision *_collision1,
                                    ODECollision *_collision2)
{
  if (this->dataPtr->trimeshCollidersCount >=
      this->dataPtr->trimeshColliders.size())
    this->dataPtr->trimeshColliders.resize(
      this->dataPtr->trimeshColliders.size() + 100);

  this->dataPtr->trimeshColliders[this->dataPtr->trimeshCollidersCount].first  =
    _collision1;
  this->dataPtr->trimeshColliders[this->dataPtr->trimeshCollidersCount].second =
    _collision2;
  this->dataPtr->trimeshCollidersCount++;
}

/////////////////////////////////////////////////
void ODEPhysics::AddCollider(ODECollision *_collision1,
                             ODECollision *_collision2)
{
  if (this->dataPtr->collidersCount >= this->dataPtr->colliders.size())
    this->dataPtr->colliders.resize(this->dataPtr->colliders.size() + 100);

  this->dataPtr->colliders[this->dataPtr->collidersCount].first  = _collision1;
  this->dataPtr->colliders[this->dataPtr->collidersCount].second = _collision2;
  this->dataPtr->collidersCount++;
}

/////////////////////////////////////////////////
void ODEPhysics::DebugPrint() const
{
  dBodyID b;
  std::cout << "Debug Print[" <<
    dWorldGetBodyCount(this->dataPtr->worldId) << "]\n";
  for (int i = 0; i < dWorldGetBodyCount(this->dataPtr->worldId); ++i)
  {
    b = dWorldGetBody(this->dataPtr->worldId, i);
    ODELink *link = static_cast<ODELink*>(dBodyGetData(b));
    ignition::math::Pose3d pose = link->WorldPose();
    const dReal *pos = dBodyGetPosition(b);
    const dReal *rot = dBodyGetRotation(b);
    ignition::math::Vector3d dpos(pos[0], pos[1], pos[2]);
    ignition::math::Quaterniond drot(rot[0], rot[1], rot[2], rot[3]);

    std::cout << "Body[" << link->GetScopedName() << "]\n";
    std::cout << "  World: Pos[" << dpos << "] Rot[" << drot << "]\n";
    if (pose.Pos() != dpos)
      std::cout << "    Incorrect world pos[" << pose.Pos() << "]\n";
    if (pose.Rot() != drot)
      std::cout << "    Incorrect world rot[" << pose.Rot() << "]\n";

    dMass mass;
    dBodyGetMass(b, &mass);
    std::cout << "  Mass[" << mass.mass << "] COG[" << mass.c[0]
              << " " << mass.c[1] << " " << mass.c[2] << "]\n";

    dGeomID g = dBodyGetFirstGeom(b);
    while (g)
    {
      ODECollision *coll = static_cast<ODECollision*>(dGeomGetData(g));

      pose = coll->WorldPose();
      const dReal *gpos = dGeomGetPosition(g);
      const dReal *grot = dGeomGetRotation(g);
      dpos.Set(gpos[0], gpos[1], gpos[2]);
      drot.Set(grot[0], grot[1], grot[2], grot[3]);

      std::cout << "    Geom[" << coll->GetScopedName() << "]\n";
      std::cout << "      World: Pos[" << dpos << "] Rot[" << drot << "]\n";

      if (pose.Pos() != dpos)
        std::cout << "      Incorrect world pos[" << pose.Pos() << "]\n";
      if (pose.Rot() != drot)
        std::cout << "      Incorrect world rot[" << pose.Rot() << "]\n";

      g = dBodyGetNextGeom(g);
    }
  }
}

/////////////////////////////////////////////////
void ODEPhysics::SetSeed(uint32_t _seed)
{
  dRandSetSeed(_seed);
}

//////////////////////////////////////////////////
bool ODEPhysics::SetParam(const std::string &_key, const boost::any &_value)
{
  sdf::ElementPtr odeElem = this->sdf->GetElement("ode");
  GZ_ASSERT(odeElem != nullptr, "ODE SDF element does not exist");

  try
  {
    if (_key == "solver_type")
    {
      this->SetStepType(boost::any_cast<std::string>(_value));
    }
    else if (_key == "cfm")
    {
      double value = boost::any_cast<double>(_value);
      odeElem->GetElement("constraints")->GetElement("cfm")->Set(value);
      dWorldSetCFM(this->dataPtr->worldId, value);
    }
    else if (_key == "erp")
    {
      double value = boost::any_cast<double>(_value);
      odeElem->GetElement("constraints")->GetElement("erp")->Set(value);
      dWorldSetERP(this->dataPtr->worldId, value);
    }
    else if (_key == "precon_iters")
    {
      int value = boost::any_cast<int>(_value);
      odeElem->GetElement("solver")->GetElement("precon_iters")->Set(value);
      dWorldSetQuickStepPreconIterations(this->dataPtr->worldId, value);
    }
    else if (_key == "iters")
    {
      int value = boost::any_cast<int>(_value);
      odeElem->GetElement("solver")->GetElement("iters")->Set(value);
      dWorldSetQuickStepNumIterations(this->dataPtr->worldId, value);
    }
    else if (_key == "sor")
    {
      double value = boost::any_cast<double>(_value);
      odeElem->GetElement("solver")->GetElement("sor")->Set(value);
      dWorldSetQuickStepW(this->dataPtr->worldId, value);
    }
    else if (_key == "friction_model")
      this->SetFrictionModel(boost::any_cast<std::string>(_value));
    else if (_key == "world_step_solver")
      this->SetWorldStepSolverType(boost::any_cast<std::string>(_value));
    else if (_key == "contact_max_correcting_vel")
    {
      double value = boost::any_cast<double>(_value);
      odeElem->GetElement("constraints")->GetElement(
          "contact_max_correcting_vel")->Set(value);
      dWorldSetContactMaxCorrectingVel(this->dataPtr->worldId, value);
    }
    else if (_key == "contact_surface_layer")
    {
      double value = boost::any_cast<double>(_value);
      odeElem->GetElement("constraints")->GetElement(
          "contact_surface_layer")->Set(value);
      dWorldSetContactSurfaceLayer(this->dataPtr->worldId, value);
    }
    else if (_key == "max_contacts")
    {
      int value = boost::any_cast<int>(_value);
      this->sdf->GetElement("max_contacts")->GetValue()->Set(value);
    }
    else if (_key == "min_step_size")
    {
      /// TODO: Implement min step size param
      double value = boost::any_cast<double>(_value);
      odeElem->GetElement("solver")->GetElement("min_step_size")->Set(value);
    }
    else if (_key == "sor_lcp_tolerance")
    {
      dWorldSetQuickStepTolerance(this->dataPtr->worldId,
          boost::any_cast<double>(_value));
    }
    else if (_key == "rms_error_tolerance")
    {
      gzwarn << "please use sor_lcp_tolerance in the future.\n";
      dWorldSetQuickStepTolerance(this->dataPtr->worldId,
          boost::any_cast<double>(_value));
    }
    else if (_key == "inertia_ratio_reduction" ||
             _key == "use_dynamic_moi_rescaling")
    {
      bool value = boost::any_cast<bool>(_value);
      dWorldSetQuickStepInertiaRatioReduction(this->dataPtr->worldId, value);
      if (odeElem->GetElement("solver")->HasElement(
            "use_dynamic_moi_rescaling"))
      {
        odeElem->GetElement("solver")->GetElement(
            "use_dynamic_moi_rescaling")->Set(value);
      }
    }
    else if (_key == "contact_residual_smoothing")
    {
      dWorldSetQuickStepContactResidualSmoothing(this->dataPtr->worldId,
        boost::any_cast<double>(_value));
    }
    else if (_key == "contact_sor_scale")
    {
      dWorldSetQuickStepContactSORScalingFactor(this->dataPtr->worldId,
        boost::any_cast<double>(_value));
    }
    else if (_key == "thread_position_correction")
    {
      dWorldSetQuickStepThreadPositionCorrection(this->dataPtr->worldId,
        boost::any_cast<bool>(_value));
    }
    else if (_key == "experimental_row_reordering")
    {
      dWorldSetQuickStepExperimentalRowReordering(this->dataPtr->worldId,
        boost::any_cast<bool>(_value));
    }
    else if (_key == "warm_start_factor")
    {
      dWorldSetQuickStepWarmStartFactor(this->dataPtr->worldId,
        boost::any_cast<double>(_value));
    }
    else if (_key == "extra_friction_iterations")
    {
      dWorldSetQuickStepExtraFrictionIterations(this->dataPtr->worldId,
        boost::any_cast<int>(_value));
    }
    else if (_key == "island_threads")
    {
      int value;
      try
      {
        value = boost::any_cast<int>(_value);
      }
      catch(const boost::bad_any_cast &e)
      {
        gzerr << "boost any_cast error:" << e.what() << "\n";
        return false;
      }
      dWorldSetIslandThreads(this->dataPtr->worldId, value);
    }
    else if (_key == "ode_quiet")
    {
      bool odeQuiet = boost::any_cast<bool>(_value);
      if (odeQuiet)
      {
        dSetMessageHandler(&dMessageQuiet);
      }
      else
      {
        dSetMessageHandler(0);
      }
    }
    else
    {
      return PhysicsEngine::SetParam(_key, _value);
    }
  }
  catch(boost::bad_any_cast &e)
  {
    gzerr << "ODEPhysics::SetParam(" << _key << ") boost::any_cast error: "
          << e.what() << std::endl;
    return false;
  }
  return true;
}

//////////////////////////////////////////////////
boost::any ODEPhysics::GetParam(const std::string &_key) const
{
  boost::any value;
  this->GetParam(_key, value);
  return value;
}

//////////////////////////////////////////////////
bool ODEPhysics::GetParam(const std::string &_key, boost::any &_value) const
{
  sdf::ElementPtr odeElem = this->sdf->GetElement("ode");
  GZ_ASSERT(odeElem != nullptr, "ODE SDF element does not exist");

  if (_key == "solver_type")
  {
    _value = odeElem->GetElement("solver")->Get<std::string>("type");
  }
  else if (_key == "cfm")
  {
    _value = odeElem->GetElement("constraints")->Get<double>("cfm");
  }
  else if (_key == "erp")
    _value = odeElem->GetElement("constraints")->Get<double>("erp");
  else if (_key == "precon_iters")
    _value = odeElem->GetElement("solver")->Get<int>("precon_iters");
  else if (_key == "iters")
    _value = odeElem->GetElement("solver")->Get<int>("iters");
  else if (_key == "sor")
    _value = odeElem->GetElement("solver")->Get<double>("sor");
  else if (_key == "contact_max_correcting_vel")
    _value = odeElem->GetElement("constraints")->Get<double>(
        "contact_max_correcting_vel");
  else if (_key == "contact_surface_layer")
    _value = odeElem->GetElement("constraints")->Get<double>(
        "contact_surface_layer");
  else if (_key == "max_contacts")
    _value = this->sdf->Get<int>("max_contacts");
  else if (_key == "min_step_size")
    _value = odeElem->GetElement("solver")->Get<double>("min_step_size");
  else if (_key == "sor_lcp_tolerance")
    _value = dWorldGetQuickStepTolerance(this->dataPtr->worldId);
  else if (_key == "rms_error_tolerance")
  {
    gzwarn << "please use sor_lcp_tolerance in the future.\n";
    _value = dWorldGetQuickStepTolerance(this->dataPtr->worldId);
  }
  else if (_key == "rms_error")
    _value = dWorldGetQuickStepRMSDeltaLambda(this->dataPtr->worldId);
  else if (_key == "constraint_residual")
    _value = dWorldGetQuickStepRMSConstraintResidual(this->dataPtr->worldId);
  else if (_key == "num_contacts")
    _value = dWorldGetQuickStepNumContacts(this->dataPtr->worldId);
  else if (_key == "inertia_ratio_reduction" ||
           _key == "use_dynamic_moi_rescaling")
    _value = dWorldGetQuickStepInertiaRatioReduction(this->dataPtr->worldId);
  else if (_key == "contact_residual_smoothing")
    _value = dWorldGetQuickStepContactResidualSmoothing(this->dataPtr->worldId);
  else if (_key == "contact_sor_scale")
    _value = dWorldGetQuickStepContactSORScalingFactor(this->dataPtr->worldId);
  else if (_key == "thread_position_correction")
    _value = dWorldGetQuickStepThreadPositionCorrection(this->dataPtr->worldId);
  else if (_key == "experimental_row_reordering")
  {
    _value = dWorldGetQuickStepExperimentalRowReordering
        (this->dataPtr->worldId);
  }
  else if (_key == "warm_start_factor")
    _value = dWorldGetQuickStepWarmStartFactor(this->dataPtr->worldId);
  else if (_key == "extra_friction_iterations")
    _value = dWorldGetQuickStepExtraFrictionIterations(this->dataPtr->worldId);
  else if (_key == "friction_model")
    _value = this->GetFrictionModel();
  else if (_key == "island_threads")
    _value = dWorldGetIslandThreads(this->dataPtr->worldId);
  else if (_key == "ode_quiet")
    _value = dGetMessageHandler() != 0;
  else if (_key == "world_step_solver")
    _value = this->GetWorldStepSolverType();
  else
  {
    return PhysicsEngine::GetParam(_key, _value);
  }
  return true;
}
