-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathDDPHelpers.cuh
52 lines (45 loc) · 1.5 KB
/
DDPHelpers.cuh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
/*****************************************************************
* DDP Helper Functions
* (currently only supports iLQR - UDP in future release)
*
* Combines the following files (note they have cross dependencies
* which defines the order of imports):
* 1: Backward Pass Helpers
* 2: Forward Pass Helpers
* 3: Next Iteration Setup and Init Helpers
* 4: DDP Algorithm Wrappers
* 5: MPC Helpers
*
*****************************************************************/
// include util functions and config parameters file
#include "config.h"
// include the correct set of cost and dynamics functions
#if PLANT == 1
#include "plants/cost_pend.cuh"
#include "plants/dynamics_pend.cuh"
#elif PLANT == 2
#include "plants/cost_cart.cuh"
#include "plants/dynamics_cart.cuh"
#elif PLANT == 3
#include "plants/cost_quad.cuh"
#include "plants/dynamics_quad.cuh"
#elif PLANT == 4
#include "plants/cost_arm.cuh"
#include "plants/dynamics_arm.cuh"
#endif
// include integrators for those dynamics
#include "utils/integrators.cuh"
// 1: Backward Pass Helpers
#include "DDPHelpers/bpHelpers.cuh"
// 2: Forward Pass Helpers
#include "DDPHelpers/fpHelpers.cuh"
// 3: Next Iteration Setup and Init Helpers
#include "DDPHelpers/nisInitHelpers.cuh"
// 4: DDP Algorithm Wrappers
#if !defined(MPC_MODE) || MPC_MODE == 0
#include "DDPHelpers/DDPWrappers.cuh"
#endif
// 5: MPC Helpers and Wrappers//
#if defined(MPC_MODE) && MPC_MODE == 1
#include "DDPHelpers/MPCHelpers.cuh"
#endif