integrator.h
674cf4f45aebmaster

integrator.h
View Options

			#ifndef INTEGRATOR_H
			#define INTEGRATOR_H

			#include "casadi/casadi.hpp"
			#include "kite.h"

			/** Solve ODE of the form : xdot = f(x, u) */
			class ODESolver
			{
			public:
			ODESolver(const casadi::Function &rhs, const casadi::Dict &params = casadi::Dict());
			virtual ~ODESolver(){}

			//casadi::DM solve(const casadi::DM &x0, const casadi::DM &u, casadi::Dict = casadi::Dict()){return casadi::DM(1);}
			//casadi::DM solve(const casadi::DM &x0, const casadi::DM &u, const double &dt, casadi::Dict = casadi::Dict()){return casadi::DM(1);}

			private:
			/** right hand side of the ODE : f(x, u)*/
			casadi::Function RHS;
			casadi::Dict Parameters;

			casadi::DM InitCond;
			int NumColocationPoints;
			double dT;
			int Method;
			/** optional */
			casadi::DM XScaling;
			casadi::DM UScaling;
			int MaxIter;
			double Tolerance;
			/** reset line search parameter after each iteration */
			bool Restart;

			/** stats */
			double accuracy;
			int num_iterations;

			/** Chebychev parameters */
			casadi::DM X_ch, D;
			casadi::DM pseudospectral_solve();

			/** RK4 */
			casadi::DM rk4_solve(const casadi::DM &X0, const casadi::DM &U, const casadi::DM &dT);

			/** CVODES */
			casadi::DM cvodes_solve(const casadi::DM &X0, const casadi::DM &U, const casadi::DM &dT);
			casadi::Function cvodes_integrator;
			casadi::Function create_cvodes_integrator(const casadi::SX &x, const casadi::SX &u, const casadi::Dict);
			bool cvodes_initialized;

			};



			#endif // INTEGRATOR_H