|
31 | 31 | fref = np.zeros(force_size) |
32 | 32 | fref[2] = -model_handler.getMass() / nk * gravity[2] |
33 | 33 | u0 = np.concatenate((fref, fref, fref, fref, np.zeros(model_handler.getModel().nv - 6))) |
34 | | - |
| 34 | +dt = 0.01 |
35 | 35 |
|
36 | 36 | w_basepos = [0, 0, 100, 10, 10, 0] |
37 | 37 | w_legpos = [1, 1, 1] |
|
60 | 60 | w_centder = np.diag(np.concatenate((w_centder_lin, w_centder_ang))) |
61 | 61 |
|
62 | 62 | problem_conf = dict( |
63 | | - timestep=0.01, |
| 63 | + timestep=dt, |
64 | 64 | w_x=w_x, |
65 | 65 | w_u=w_u, |
66 | 66 | w_cent=w_cent, |
|
85 | 85 | T_ss = 30 |
86 | 86 |
|
87 | 87 | mpc_conf = dict( |
88 | | - ddpIteration=1, |
89 | 88 | support_force=-model_handler.getMass() * gravity[2], |
90 | 89 | TOL=1e-4, |
91 | 90 | mu_init=1e-8, |
|
94 | 93 | swing_apex=0.15, |
95 | 94 | T_fly=T_ss, |
96 | 95 | T_contact=T_ds, |
97 | | - timestep=0.01, |
| 96 | + timestep=dt, |
98 | 97 | ) |
99 | 98 |
|
100 | 99 | mpc = MPC(mpc_conf, dynproblem) |
|
149 | 148 | qp = IDSolver(id_conf, model_handler.getModel()) |
150 | 149 |
|
151 | 150 | """ Interpolation """ |
152 | | -interpolator = Interpolator(nq + nv, nv, nu, nf, 0.01) |
| 151 | +interpolator = Interpolator(model_handler.getModel()) |
153 | 152 |
|
154 | 153 | """ Initialize simulation""" |
155 | 154 | device = BulletRobot( |
|
252 | 251 |
|
253 | 252 | for j in range(N_simu): |
254 | 253 | # time.sleep(0.01) |
255 | | - interpolator.interpolate(j / float(N_simu), xss, uss, ddqs, forces) |
| 254 | + delay = j / float(N_simu) * dt |
| 255 | + |
| 256 | + acc_interp = interpolator.interpolateLinear(delay, dt, ddqs) |
| 257 | + force_interp = interpolator.interpolateLinear(delay, dt, forces) |
256 | 258 |
|
257 | 259 | q_meas, v_meas = device.measureState() |
258 | 260 | x_measured = np.concatenate([q_meas, v_meas]) |
|
263 | 265 | mpc.getDataHandler().getData(), |
264 | 266 | contact_states, |
265 | 267 | x_measured[nq:], |
266 | | - interpolator.a_interpolated, |
| 268 | + acc_interp, |
267 | 269 | np.zeros(12), |
268 | | - interpolator.forces_interpolated, |
| 270 | + force_interp, |
269 | 271 | mpc.getDataHandler().getData().M, |
270 | 272 | ) |
271 | 273 |
|
|
0 commit comments