bindings/python/collision/parallel/geometry.cpp

Go to the documentation of this file.

//
// Copyright (c) 2021-2022 INRIA
//
 
#include "pinocchio/bindings/python/algorithm/algorithms.hpp"
#include "pinocchio/collision/parallel/geometry.hpp"
#include "pinocchio/collision/parallel/broadphase.hpp"
 
#include <eigenpy/eigen-from-python.hpp>
 
namespace pinocchio
{
  namespace python
  {
 
    using namespace Eigen;
 
    typedef Eigen::Matrix<bool, Eigen::Dynamic, 1> VectorXb;
 
    static bool computeCollisionsInParallel_proxy(
      const size_t num_threads,
      const GeometryModel & geom_model,
      GeometryData & geom_data,
      const bool stopAtFirstCollision = false)
    {
      return computeCollisionsInParallel(num_threads, geom_model, geom_data, stopAtFirstCollision);
    }
 
    static bool computeCollisionsInParallel_full_proxy(
      const size_t num_threads,
      const Model & model,
      Data & data,
      const GeometryModel & geom_model,
      GeometryData & geom_data,
      const Eigen::VectorXd & q,
      const bool stopAtFirstCollision = false)
    {
      return computeCollisionsInParallel(
        num_threads, model, data, geom_model, geom_data, q, stopAtFirstCollision);
    }
 
    static VectorXb computeCollisionsInParallel_pool_proxy(
      const int num_thread,
      GeometryPool & pool,
      const Eigen::MatrixXd & q,
      bool stopAtFirstCollisionInConfiguration = false,
      bool stopAtFirstCollisionInBatch = false)
    {
      VectorXb res(q.cols());
      computeCollisionsInParallel(
        num_thread, pool, q, res, stopAtFirstCollisionInConfiguration, stopAtFirstCollisionInBatch);
      return res;
    }
 
    void exposeParallelGeometry()
    {
      namespace bp = boost::python;
 
      using namespace Eigen;
 
      bp::def(
        "computeCollisionsInParallel", computeCollisionsInParallel_proxy,
        (bp::arg("num_thread"), bp::arg("geometry_model"), bp::arg("geometry_data"),
         bp::arg("stop_at_first_collision") = false),
        "Evaluates in parallel the collisions for a single data and returns the result.\n\n"
        "Parameters:\n"
        "\tnum_thread: number of threads used for the computation\n"
        "\tgeometry_model: the geometry model\n"
        "\tgeometry_data: the geometry data\n"
        "\tstop_at_first_collision: if set to true, stops when encountering the first "
        "collision.\n");
 
      bp::def(
        "computeCollisionsInParallel", computeCollisionsInParallel_full_proxy,
        (bp::arg("num_thread"), bp::arg("model"), bp::arg("data"), bp::arg("geometry_model"),
         bp::arg("geometry_data"), bp::arg("q"), bp::arg("stop_at_first_collision") = false),
        "Evaluates in parallel the collisions for a single data and returns the result.\n\n"
        "Parameters:\n"
        "\tnum_thread: number of threads used for the computation\n"
        "\tmodel: the kinematic model\n"
        "\tdata: the data associated to the model\n"
        "\tgeometry_model: the geometry model\n"
        "\tgeometry_data: the geometry data associated to the tgeometry_model\n"
        "\tq: the joint configuration vector (size model.nq)\n"
        "\tstop_at_first_collision: if set to true, stops when encountering the first "
        "collision.\n");
 
      bp::def(
        "computeCollisionsInParallel", computeCollisionsInParallel_pool_proxy,
        (bp::arg("num_thread"), bp::arg("pool"), bp::arg("q"),
         bp::arg("stop_at_first_collision_in_configuration") = false,
         bp::arg("stop_at_first_collision_in_batch") = false),
        "Evaluates in parallel the collisions and returns the result.\n\n"
        "Parameters:\n"
        "\tnum_thread: number of threads used for the computation\n"
        "\tpool: pool of geometry model/ geometry data\n"
        "\tq: the joint configuration vector (size model.nq x batch_size)\n"
        "\tstop_at_first_collision_in_configuration: if set to true, stops when encountering "
        "the first collision in a configuration\n"
        "\tstop_at_first_collision_in_batch: if set to true, stops when encountering the "
        "first collision in a batch.\n");
    }
 
  } // namespace python
} // namespace pinocchio