discretization.h

#pragma once
 
#include "math/eigen_interface.h"
#include <Eigen/unsupported/Eigen/MatrixFunctions>
 
#include <cmath>
#include <tuple>

template <int STATES, int INPUTS>
std::tuple<EMat<STATES, STATES>, EMat<STATES, INPUTS>> discretize_AB(EMat<STATES, STATES> Ac, EMat<STATES, INPUTS> Bc, const double &dt) {
    // Form the intermediate matrix M
    //
    //       [A B]
    //   M = [0 0]
    //
    EMat<STATES + INPUTS, STATES + INPUTS> M;
    M.template block<STATES, STATES>(0, 0) = Ac;
    M.template block<STATES, INPUTS>(0, STATES) = Bc;
    M.template block<INPUTS, STATES + INPUTS>(STATES, 0).setZero();
 
    //
    //  M * T   [Ad Bd]
    // e      = [0   I]
    //
    EMat<STATES + INPUTS, STATES + INPUTS> phi = (M * dt).exp();
 
    // Extract Ad and Bd from phi and put them in a tuple
    return std::make_tuple(phi.template block<STATES, STATES>(0, 0), phi.template block<STATES, INPUTS>(0, STATES));
}