mpc_python_learn/mpc_pybullet_demo/utils.py

import numpy as np
from scipy.interpolate import interp1d

from mpc_config import Params
P=Params()

def compute_path_from_wp(start_xp, start_yp, step = 0.1):
    """
    Computes a reference path given a set of waypoints
    """
    
    final_xp=[]
    final_yp=[]
    delta = step #[m]

    for idx in range(len(start_xp)-1):
        section_len = np.sum(np.sqrt(np.power(np.diff(start_xp[idx:idx+2]),2)+np.power(np.diff(start_yp[idx:idx+2]),2)))

        interp_range = np.linspace(0,1,np.floor(section_len/delta).astype(int))
        
        fx=interp1d(np.linspace(0,1,2),start_xp[idx:idx+2],kind=1)
        fy=interp1d(np.linspace(0,1,2),start_yp[idx:idx+2],kind=1)
        
        final_xp=np.append(final_xp,fx(interp_range))
        final_yp=np.append(final_yp,fy(interp_range))
    
    dx = np.append(0, np.diff(final_xp))
    dy = np.append(0, np.diff(final_yp))
    theta = np.arctan2(dy, dx)

    return np.vstack((final_xp,final_yp,theta))


def get_nn_idx(state,path):
    """
    Computes the index of the waypoint closest to vehicle
    """

    dx = state[0]-path[0,:]
    dy = state[1]-path[1,:]
    dist = np.hypot(dx,dy)
    nn_idx = np.argmin(dist)

    try:
        v = [path[0,nn_idx+1] - path[0,nn_idx],
             path[1,nn_idx+1] - path[1,nn_idx]]   
        v /= np.linalg.norm(v)

        d = [path[0,nn_idx] - state[0],
             path[1,nn_idx] - state[1]]

        if np.dot(d,v) > 0:
            target_idx = nn_idx
        else:
            target_idx = nn_idx+1

    except IndexError as e:
        target_idx = nn_idx

    return target_idx

def get_ref_trajectory(state, path, target_v):
    """
    """
    xref = np.zeros((P.N, P.T + 1))
    dref = np.zeros((1, P.T + 1))
    
    #sp = np.ones((1,T +1))*target_v #speed profile
    
    ncourse = path.shape[1]

    ind = get_nn_idx(state, path)

    xref[0, 0] = path[0,ind] #X
    xref[1, 0] = path[1,ind] #Y
    xref[2, 0] = target_v    #sp[ind] #V
    xref[3, 0] = path[2,ind] #Theta
    dref[0, 0] = 0.0         # steer operational point should be 0
    
    dl = P.path_tick # Waypoints spacing [m]
    travel = 0.0

    for i in range(P.T + 1):
        travel += abs(target_v) * P.dt #current V or target V?
        dind = int(round(travel / dl))

        if (ind + dind) < ncourse:
            xref[0, i] = path[0,ind + dind]
            xref[1, i] = path[1,ind + dind]
            xref[2, i] = target_v #sp[ind + dind]
            xref[3, i] = path[2,ind + dind]
            dref[0, i] = 0.0
        else:
            xref[0, i] = path[0,ncourse - 1]
            xref[1, i] = path[1,ncourse - 1]
            xref[2, i] = 0.0 #stop? #sp[ncourse - 1]
            xref[3, i] = path[2,ncourse - 1]
            dref[0, i] = 0.0

    return xref, dref