6 import dynamic_graph.sot_talos_balance.talos.base_estimator_conf
as base_estimator_conf
7 import dynamic_graph.sot_talos_balance.talos.control_manager_conf
as cm_conf
8 import dynamic_graph.sot_talos_balance.talos.ft_calibration_conf
as ft_conf
9 import dynamic_graph.sot_talos_balance.talos.hip_flexibility_compensation_conf
as hipFlexCompConfig
10 import dynamic_graph.sot_talos_balance.talos.parameter_server_conf
as param_server_conf
11 from dynamic_graph
import plug
12 from dynamic_graph.sot.core
import (
16 MatrixHomoToPoseQuaternion,
19 from dynamic_graph.sot.core.matrix_util
import matrixToTuple
20 from dynamic_graph.sot.core.meta_tasks_kine
import (
25 from dynamic_graph.sot.dynamic_pinocchio
import DynamicPinocchio
26 from dynamic_graph.tracer_real_time
import TracerRealTime
29 cm_conf.CTRL_MAX = 10.0
31 robot.timeStep = robot.device.getTimeStep()
33 robot.device.setControlInputType(
"noInteg")
37 robot.dynamic.com.recompute(0)
38 robotDim = robot.dynamic.getDimension()
39 mass = robot.dynamic.data.mass[0]
40 h = robot.dynamic.com.value[2]
48 robot.dynamic.createOpPoint(
"LF", robot.OperationalPointsMap[
"left-ankle"])
49 robot.dynamic.createOpPoint(
"RF", robot.OperationalPointsMap[
"right-ankle"])
50 robot.dynamic.createOpPoint(
"WT", robot.OperationalPointsMap[
"waist"])
51 robot.dynamic.LF.recompute(0)
52 robot.dynamic.RF.recompute(0)
53 robot.dynamic.WT.recompute(0)
58 if test_folder
is not None:
59 if sot_talos_balance_folder:
60 from rospkg
import RosPack
64 folder = rospack.get_path(
"sot-talos-balance") +
"/data/" + test_folder
73 robot.triggerTrajGen = BooleanIdentity(
"triggerTrajGen")
78 robot.comTrajGen.x.recompute(0)
79 plug(robot.triggerTrajGen.sout, robot.comTrajGen.trigger)
83 robot.lfTrajGen.x.recompute(0)
85 robot.lfToMatrix = PoseRollPitchYawToMatrixHomo(
"lf2m")
86 plug(robot.lfTrajGen.x, robot.lfToMatrix.sin)
87 plug(robot.triggerTrajGen.sout, robot.lfTrajGen.trigger)
91 robot.rfTrajGen.x.recompute(0)
93 robot.rfToMatrix = PoseRollPitchYawToMatrixHomo(
"rf2m")
94 plug(robot.rfTrajGen.x, robot.rfToMatrix.sin)
95 plug(robot.triggerTrajGen.sout, robot.rfTrajGen.trigger)
99 robot.zmpTrajGen.x.recompute(0)
100 plug(robot.triggerTrajGen.sout, robot.zmpTrajGen.trigger)
104 robot.waistTrajGen.x.recompute(0)
106 robot.waistMix = Mix_of_vector(
"waistMix")
107 robot.waistMix.setSignalNumber(3)
108 robot.waistMix.addSelec(1, 0, 3)
109 robot.waistMix.addSelec(2, 3, 3)
110 robot.waistMix.default.value = [0.0] * 6
111 robot.waistMix.signal(
"sin1").value = [0.0] * 3
112 plug(robot.waistTrajGen.x, robot.waistMix.signal(
"sin2"))
114 robot.waistToMatrix = PoseRollPitchYawToMatrixHomo(
"w2m")
115 plug(robot.waistMix.sout, robot.waistToMatrix.sin)
116 plug(robot.triggerTrajGen.sout, robot.waistTrajGen.trigger)
120 robot.phaseTrajGen.x.recompute(0)
121 robot.phaseScalar = Component_of_vector(
"phase_scalar")
122 robot.phaseScalar.setIndex(0)
123 plug(robot.phaseTrajGen.x, robot.phaseScalar.sin)
124 robot.phaseInt = RoundDoubleToInt(
"phase_int")
125 plug(robot.phaseScalar.sout, robot.phaseInt.sin)
126 plug(robot.triggerTrajGen.sout, robot.phaseTrajGen.trigger)
133 wp = DummyWalkingPatternGenerator(
"dummy_wp")
135 wp.omega.value = omega
136 plug(robot.waistToMatrix.sout, wp.waist)
137 plug(robot.lfToMatrix.sout, wp.footLeft)
138 plug(robot.rfToMatrix.sout, wp.footRight)
139 plug(robot.comTrajGen.x, wp.com)
140 plug(robot.comTrajGen.dx, wp.vcom)
141 plug(robot.comTrajGen.ddx, wp.acom)
148 robot.wp.comDes.recompute(0)
149 robot.wp.dcmDes.recompute(0)
150 robot.wp.zmpDes.recompute(0)
159 robot.m2qLF = MatrixHomoToPoseQuaternion(
"m2qLF")
160 plug(robot.dynamic.LF, robot.m2qLF.sin)
161 plug(robot.m2qLF.sout, robot.base_estimator.lf_ref_xyzquat)
162 robot.m2qRF = MatrixHomoToPoseQuaternion(
"m2qRF")
163 plug(robot.dynamic.RF, robot.m2qRF.sin)
164 plug(robot.m2qRF.sout, robot.base_estimator.rf_ref_xyzquat)
169 e2q = EulerToQuat(
"e2q")
170 plug(robot.base_estimator.q, e2q.euler)
174 robot.rdynamic = DynamicPinocchio(
"real_dynamics")
175 robot.rdynamic.setModel(robot.dynamic.model)
176 robot.rdynamic.setData(robot.rdynamic.model.createData())
177 plug(robot.base_estimator.q, robot.rdynamic.position)
178 robot.rdynamic.velocity.value = [0.0] * robotDim
179 robot.rdynamic.acceleration.value = [0.0] * robotDim
182 cdc_estimator = DcmEstimator(
"cdc_estimator")
183 cdc_estimator.init(dt, robot_name)
184 plug(robot.e2q.quaternion, cdc_estimator.q)
185 plug(robot.base_estimator.v, cdc_estimator.v)
186 robot.cdc_estimator = cdc_estimator
189 estimator = DummyDcmEstimator(
"dummy")
190 plug(robot.wp.omegaDes, estimator.omega)
191 estimator.mass.value = 1.0
192 plug(robot.cdc_estimator.c, estimator.com)
193 plug(robot.cdc_estimator.dc, estimator.momenta)
195 robot.estimator = estimator
201 zmp_estimator = SimpleZmpEstimator(
"zmpEst")
202 robot.rdynamic.createOpPoint(
"sole_LF",
"left_sole_link")
203 robot.rdynamic.createOpPoint(
"sole_RF",
"right_sole_link")
204 plug(robot.rdynamic.sole_LF, zmp_estimator.poseLeft)
205 plug(robot.rdynamic.sole_RF, zmp_estimator.poseRight)
206 plug(robot.ftc.left_foot_force_out, zmp_estimator.wrenchLeft)
207 plug(robot.ftc.right_foot_force_out, zmp_estimator.wrenchRight)
209 robot.zmp_estimator = zmp_estimator
215 Ki_dcm = [0.0, 0.0, 0.0]
219 dcm_controller = DcmController(
"dcmCtrl")
221 dcm_controller.Kp.value = Kp_dcm
222 dcm_controller.Ki.value = Ki_dcm
223 dcm_controller.Kz.value = Kz_dcm
224 dcm_controller.decayFactor.value = gamma_dcm
225 dcm_controller.mass.value = mass
226 plug(robot.wp.omegaDes, dcm_controller.omega)
228 plug(robot.cdc_estimator.c, dcm_controller.com)
229 plug(robot.estimator.dcm, dcm_controller.dcm)
231 plug(robot.wp.zmpDes, dcm_controller.zmpDes)
232 plug(robot.wp.dcmDes, dcm_controller.dcmDes)
234 plug(robot.zmp_estimator.zmp, dcm_controller.zmp)
236 dcm_controller.init(dt)
238 robot.dcm_control = dcm_controller
240 Ki_dcm = [1.0, 1.0, 1.0]
242 Kz_dcm = [0.0, 0.0, 0.0]
245 Kp_adm = [0.0, 0.0, 0.0]
247 com_admittance_control = ComAdmittanceController(
"comAdmCtrl")
248 com_admittance_control.Kp.value = Kp_adm
249 plug(robot.zmp_estimator.zmp, com_admittance_control.zmp)
250 com_admittance_control.zmpDes.value = (
251 robot.wp.zmpDes.value
253 plug(robot.wp.acomDes, com_admittance_control.ddcomDes)
255 com_admittance_control.init(dt)
256 com_admittance_control.setState(robot.wp.comDes.value, [0.0, 0.0, 0.0])
258 robot.com_admittance_control = com_admittance_control
260 Kp_adm = [15.0, 15.0, 0.0]
264 robot.cm.addCtrlMode(
"sot_input")
265 robot.cm.setCtrlMode(
"all",
"sot_input")
266 robot.cm.addEmergencyStopSIN(
"zmp")
271 robot.taskUpperBody = Task(
"task_upper_body")
272 robot.taskUpperBody.feature = FeaturePosture(
"feature_upper_body")
274 q = list(robot.dynamic.position.value)
275 robot.taskUpperBody.feature.state.value = q
276 robot.taskUpperBody.feature.posture.value = q
278 robotDim = robot.dynamic.getDimension()
279 for i
in range(18, robotDim):
280 robot.taskUpperBody.feature.selectDof(i,
True)
282 robot.taskUpperBody.controlGain.value = 100.0
283 robot.taskUpperBody.add(robot.taskUpperBody.feature.name)
284 plug(robot.dynamic.position, robot.taskUpperBody.feature.state)
288 robot.contactLF = MetaTaskKine6d(
289 "contactLF", robot.dynamic,
"LF", robot.OperationalPointsMap[
"left-ankle"]
291 robot.contactLF.feature.frame(
"desired")
292 robot.contactLF.gain.setConstant(300)
293 plug(robot.wp.footLeftDes, robot.contactLF.featureDes.position)
294 locals()[
"contactLF"] = robot.contactLF
296 robot.contactRF = MetaTaskKine6d(
297 "contactRF", robot.dynamic,
"RF", robot.OperationalPointsMap[
"right-ankle"]
299 robot.contactRF.feature.frame(
"desired")
300 robot.contactRF.gain.setConstant(300)
301 plug(robot.wp.footRightDes, robot.contactRF.featureDes.position)
302 locals()[
"contactRF"] = robot.contactRF
305 robot.taskComH = MetaTaskKineCom(robot.dynamic, name=
"comH")
306 plug(robot.wp.comDes, robot.taskComH.featureDes.errorIN)
307 robot.taskComH.task.controlGain.value = 100.0
308 robot.taskComH.feature.selec.value =
"100"
311 robot.taskCom = MetaTaskKineCom(robot.dynamic)
312 plug(robot.com_admittance_control.comRef, robot.taskCom.featureDes.errorIN)
313 plug(robot.com_admittance_control.dcomRef, robot.taskCom.featureDes.errordotIN)
314 robot.taskCom.task.controlGain.value = 100.0
315 robot.taskCom.task.setWithDerivative(
True)
316 robot.taskCom.feature.selec.value =
"011"
319 robot.keepWaist = MetaTaskKine6d(
320 "keepWaist", robot.dynamic,
"WT", robot.OperationalPointsMap[
"waist"]
322 robot.keepWaist.feature.frame(
"desired")
323 robot.keepWaist.gain.setConstant(300)
324 plug(robot.wp.waistDes, robot.keepWaist.featureDes.position)
325 robot.keepWaist.feature.selec.value =
"111000"
326 locals()[
"keepWaist"] = robot.keepWaist
329 robot.sot = SOT(
"sot")
330 robot.sot.setSize(robot.dynamic.getDimension())
333 robot.integrate = SimpleStateIntegrator(
"integrate")
334 robot.integrate.init(dt)
335 robot.integrate.setState(robot.device.state.value)
336 robot.integrate.setVelocity(robot.dynamic.getDimension() * [0.0])
341 robot, hipFlexCompConfig, robot_name
343 plug(robot.phaseInt.sout, robot.hipComp.phase)
345 robot.hipComp.K_l.value = float(
"inf")
346 robot.hipComp.K_r.value = float(
"inf")
349 plug(robot.sot.control, robot.cm.ctrl_sot_input)
350 plug(robot.cm.u_safe, robot.integrate.control)
351 plug(robot.integrate.state, robot.hipComp.q_des)
352 plug(robot.hipComp.q_cmd, robot.device.control)
354 robot.sot.push(robot.taskUpperBody.name)
355 robot.sot.push(robot.contactRF.task.name)
356 robot.sot.push(robot.contactLF.task.name)
357 robot.sot.push(robot.taskComH.task.name)
358 robot.sot.push(robot.taskCom.task.name)
359 robot.sot.push(robot.keepWaist.task.name)
365 robot.delay_pos.setMemory(robot.device.state.value)
366 robot.device.before.addSignal(robot.delay_pos.name +
".current")
367 plug(robot.integrate.state, robot.delay_pos.sin)
370 robot.delay_vel.setMemory(robot.dynamic.getDimension() * [0.0])
371 robot.device.before.addSignal(robot.delay_vel.name +
".current")
372 plug(robot.cm.u_safe, robot.delay_vel.sin)
375 plug(robot.delay_pos.previous, robot.pselec.sin)
376 plug(robot.pselec.sout, robot.base_estimator.joint_positions)
377 plug(robot.delay_vel.previous, robot.vselec.sin)
380 plug(robot.delay_pos.previous, robot.dynamic.position)
381 plug(robot.delay_vel.previous, robot.dynamic.velocity)
382 robot.dvdt = Derivator_of_Vector(
"dv_dt")
383 robot.dvdt.dt.value = dt
384 plug(robot.delay_vel.previous, robot.dvdt.sin)
385 plug(robot.dvdt.sout, robot.dynamic.acceleration)
392 create_topic(robot.publisher, robot.device,
"state", robot=robot, data_type=
"vector")
394 robot.publisher, robot.base_estimator,
"q", robot=robot, data_type=
"vector"
399 robot.publisher, robot.comTrajGen,
"x", robot=robot, data_type=
"vector"
402 robot.publisher, robot.comTrajGen,
"dx", robot=robot, data_type=
"vector"
405 robot.publisher, robot.comTrajGen,
"ddx", robot=robot, data_type=
"vector"
409 robot.publisher, robot.wp,
"comDes", robot=robot, data_type=
"vector"
413 robot.publisher, robot.cdc_estimator,
"c", robot=robot, data_type=
"vector"
416 robot.publisher, robot.cdc_estimator,
"dc", robot=robot, data_type=
"vector"
421 robot.com_admittance_control,
427 robot.publisher, robot.dynamic,
"com", robot=robot, data_type=
"vector"
431 robot.publisher, robot.dcm_control,
"dcmDes", robot=robot, data_type=
"vector"
434 robot.publisher, robot.estimator,
"dcm", robot=robot, data_type=
"vector"
438 robot.publisher, robot.zmpTrajGen,
"x", robot=robot, data_type=
"vector"
441 robot.publisher, robot.wp,
"zmpDes", robot=robot, data_type=
"vector"
444 robot.publisher, robot.dynamic,
"zmp", robot=robot, data_type=
"vector"
447 robot.publisher, robot.zmp_estimator,
"zmp", robot=robot, data_type=
"vector"
450 robot.publisher, robot.dcm_control,
"zmpRef", robot=robot, data_type=
"vector"
460 robot.publisher, robot.waistTrajGen,
"x", robot=robot, data_type=
"vector"
464 robot.publisher, robot.lfTrajGen,
"x", robot=robot, data_type=
"vector"
467 robot.publisher, robot.rfTrajGen,
"x", robot=robot, data_type=
"vector"
471 robot.publisher, robot.ftc,
"left_foot_force_out", robot=robot, data_type=
"vector"
474 robot.publisher, robot.ftc,
"right_foot_force_out", robot=robot, data_type=
"vector"
478 robot.publisher, robot.dynamic,
"LF", robot=robot, data_type=
"matrixHomo"
481 robot.publisher, robot.dynamic,
"RF", robot=robot, data_type=
"matrixHomo"
484 create_topic(robot.publisher, robot.hipComp,
"delta_q", robot=robot, data_type=
"vector")
485 create_topic(robot.publisher, robot.hipComp,
"q_cmd", robot=robot, data_type=
"vector")
486 create_topic(robot.publisher, robot.hipComp,
"q_des", robot=robot, data_type=
"vector")
487 create_topic(robot.publisher, robot.hipComp,
"tau", robot=robot, data_type=
"vector")
489 robot.publisher, robot.hipComp,
"tau_filt", robot=robot, data_type=
"vector"
493 robot.tracer = TracerRealTime(
"com_tracer")
494 robot.tracer.setBufferSize(80 * (2**20))
495 robot.tracer.open(
"/tmp",
"dg_",
".dat")
496 robot.device.after.addSignal(
"{0}.triger".format(robot.tracer.name))
498 addTrace(robot.tracer, robot.wp,
"comDes")
500 addTrace(robot.tracer, robot.cdc_estimator,
"c")
503 addTrace(robot.tracer, robot.com_admittance_control,
"comRef")
504 addTrace(robot.tracer, robot.dynamic,
"com")
511 addTrace(robot.tracer, robot.zmp_estimator,
"zmp")
512 addTrace(robot.tracer, robot.dcm_control,
"zmpRef")