PreComputedMotions.hpp

Go to the documentation of this file.

#pragma once
 
#include "DiscreteTheta.hpp"
#include "Mobility.hpp"
#include <limits>
#include <stdexcept>
#include <vector>
#include <base/Pose.hpp>
#include <maps/grid/Index.hpp>
#include <sbpl_spline_primitives/SbplSplineMotionPrimitives.hpp>
 
namespace ugv_nav4d
{
 
struct PoseWithCell
{
    base::Pose2D pose;
    maps::grid::Index cell;
};
 
struct CellWithPoses 
{
    maps::grid::Index cell;
    std::vector<base::Pose2D> poses;
};
 
 
class Motion
{    
public:
    
    static double costScaleFactor;
    
    enum Type {
        MOV_FORWARD,
        MOV_BACKWARD,
        MOV_POINTTURN,
        MOV_LATERAL,
    };
 
    Motion(unsigned int numAngles = 0) : endTheta(0, numAngles),startTheta(0, numAngles), baseCost(0), id(std::numeric_limits<size_t>::max()) {};
    
    static int calculateCost(double translationalDist, double angularDist, double translationVelocity, double angularVelocity, double costMultiplier);
    
    int xDiff;
    int yDiff;
    DiscreteTheta endTheta;
    DiscreteTheta startTheta;
    
    Type type;
    
    std::vector<PoseWithCell> intermediateStepsTravMap;
    
    std::vector<CellWithPoses> fullSplineSamples;
    
    int baseCost; //time the robot needs to follow the primivite scaled by some factors
    int costMultiplier;//is used to scale the baseCost (can be used to punish certain motions)
    double translationlDist; //translational length of the motion
    double angularDist; //angular length of the motion
    
    size_t id;
    
};
 
class PreComputedMotions
{
    //indexed by discrete start theta
    std::vector<std::vector<Motion> > thetaToMotion;
    std::vector<Motion> idToMotion;
    sbpl_spline_primitives::SbplSplineMotionPrimitives primitives;
    Mobility mobilityConfig;
public:
    PreComputedMotions(const sbpl_spline_primitives::SplinePrimitivesConfig& primitiveConfig,
                       const Mobility& mobilityConfig);
    
    void readMotionPrimitives(const sbpl_spline_primitives::SbplSplineMotionPrimitives& primGen,
                              const Mobility& mobilityConfig, double travGridResolution);
    
    void computeMotions(double travGridResolution);
    
    void setMotionForTheta(const Motion &motion, const DiscreteTheta &theta);
    
    void preComputeCost(Motion &motion);
    
    const std::vector<Motion> &getMotionForStartTheta(const DiscreteTheta &theta) const;
    
    const Motion &getMotion(std::size_t id) const; 
    
    const sbpl_spline_primitives::SbplSplineMotionPrimitives& getPrimitives() const;
    
    static double calculateCurvatureFromRadius(const double r);
private:
    
    void sampleOnResolution(double gridResolution, base::geometry::Spline2 spline, std::vector< ugv_nav4d::PoseWithCell >& result, std::vector< ugv_nav4d::CellWithPoses >& fullResult);
    
    base::Pose2D getPointClosestToCellMiddle(const ugv_nav4d::CellWithPoses& cwp, const double gridResolution);
    
    
    void computeSplinePrimCost(const sbpl_spline_primitives::SplinePrimitive& prim,
                               const Mobility& mobilityConfig, Motion& outMotion) const;
    
};
 
 
}