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.parameter_server_conf
as param_server_conf
10 from dynamic_graph
import plug
11 from dynamic_graph.sot.core
import (
15 MatrixHomoToPoseQuaternion,
18 from dynamic_graph.sot.core.matrix_util
import matrixToTuple
19 from dynamic_graph.sot.core.meta_tasks_kine
import (
24 from dynamic_graph.sot.dynamic_pinocchio
import DynamicPinocchio
25 from dynamic_graph.tracer_real_time
import TracerRealTime
28 cm_conf.CTRL_MAX = 10.0
30 robot.timeStep = robot.device.getTimeStep()
32 robot.device.setControlInputType(
"noInteg")
36 robot.dynamic.com.recompute(0)
37 robotDim = robot.dynamic.getDimension()
38 mass = robot.dynamic.data.mass[0]
39 h = robot.dynamic.com.value[2]
47 robot.dynamic.createOpPoint(
"LF", robot.OperationalPointsMap[
"left-ankle"])
48 robot.dynamic.createOpPoint(
"RF", robot.OperationalPointsMap[
"right-ankle"])
49 robot.dynamic.createOpPoint(
"WT", robot.OperationalPointsMap[
"waist"])
50 robot.dynamic.LF.recompute(0)
51 robot.dynamic.RF.recompute(0)
52 robot.dynamic.WT.recompute(0)
57 if test_folder
is not None:
58 if sot_talos_balance_folder:
59 from rospkg
import RosPack
63 folder = rospack.get_path(
"sot-talos-balance") +
"/data/" + test_folder
72 robot.triggerTrajGen = BooleanIdentity(
"triggerTrajGen")
73 robot.triggerTrajGen.sin.value = 0
77 robot.comTrajGen.x.recompute(0)
78 plug(robot.triggerTrajGen.sout, robot.comTrajGen.trigger)
82 robot.lfTrajGen.x.recompute(0)
84 robot.lfToMatrix = PoseRollPitchYawToMatrixHomo(
"lf2m")
85 plug(robot.lfTrajGen.x, robot.lfToMatrix.sin)
86 plug(robot.triggerTrajGen.sout, robot.lfTrajGen.trigger)
90 robot.rfTrajGen.x.recompute(0)
92 robot.rfToMatrix = PoseRollPitchYawToMatrixHomo(
"rf2m")
93 plug(robot.rfTrajGen.x, robot.rfToMatrix.sin)
94 plug(robot.triggerTrajGen.sout, robot.rfTrajGen.trigger)
98 robot.zmpTrajGen.x.recompute(0)
99 plug(robot.triggerTrajGen.sout, robot.zmpTrajGen.trigger)
103 robot.waistTrajGen.x.recompute(0)
105 robot.waistMix = Mix_of_vector(
"waistMix")
106 robot.waistMix.setSignalNumber(3)
107 robot.waistMix.addSelec(1, 0, 3)
108 robot.waistMix.addSelec(2, 3, 3)
109 robot.waistMix.default.value = [0.0] * 6
110 robot.waistMix.signal(
"sin1").value = [0.0] * 3
111 plug(robot.waistTrajGen.x, robot.waistMix.signal(
"sin2"))
113 robot.waistToMatrix = PoseRollPitchYawToMatrixHomo(
"w2m")
114 plug(robot.waistMix.sout, robot.waistToMatrix.sin)
115 plug(robot.triggerTrajGen.sout, robot.waistTrajGen.trigger)
118 if folder
is not None:
119 robot.comTrajGen.playTrajectoryFile(folder +
"CoM.dat")
120 robot.lfTrajGen.playTrajectoryFile(folder +
"LeftFoot.dat")
121 robot.rfTrajGen.playTrajectoryFile(folder +
"RightFoot.dat")
123 robot.waistTrajGen.playTrajectoryFile(folder +
"WaistOrientation.dat")
127 wp = DummyWalkingPatternGenerator(
"dummy_wp")
129 wp.omega.value = omega
130 plug(robot.waistToMatrix.sout, wp.waist)
131 plug(robot.lfToMatrix.sout, wp.footLeft)
132 plug(robot.rfToMatrix.sout, wp.footRight)
133 plug(robot.comTrajGen.x, wp.com)
134 plug(robot.comTrajGen.dx, wp.vcom)
135 plug(robot.comTrajGen.ddx, wp.acom)
142 robot.wp.comDes.recompute(0)
143 robot.wp.dcmDes.recompute(0)
144 robot.wp.zmpDes.recompute(0)
153 robot.m2qLF = MatrixHomoToPoseQuaternion(
"m2qLF")
154 plug(robot.dynamic.LF, robot.m2qLF.sin)
155 plug(robot.m2qLF.sout, robot.base_estimator.lf_ref_xyzquat)
156 robot.m2qRF = MatrixHomoToPoseQuaternion(
"m2qRF")
157 plug(robot.dynamic.RF, robot.m2qRF.sin)
158 plug(robot.m2qRF.sout, robot.base_estimator.rf_ref_xyzquat)
163 e2q = EulerToQuat(
"e2q")
164 plug(robot.base_estimator.q, e2q.euler)
168 robot.rdynamic = DynamicPinocchio(
"real_dynamics")
169 robot.rdynamic.setModel(robot.dynamic.model)
170 robot.rdynamic.setData(robot.rdynamic.model.createData())
171 plug(robot.base_estimator.q, robot.rdynamic.position)
172 robot.rdynamic.velocity.value = [0.0] * robotDim
173 robot.rdynamic.acceleration.value = [0.0] * robotDim
176 cdc_estimator = DcmEstimator(
"cdc_estimator")
177 cdc_estimator.init(dt, robot_name)
178 plug(robot.e2q.quaternion, cdc_estimator.q)
179 plug(robot.base_estimator.v, cdc_estimator.v)
180 robot.cdc_estimator = cdc_estimator
183 estimator = DummyDcmEstimator(
"dummy")
184 plug(robot.wp.omegaDes, estimator.omega)
185 estimator.mass.value = 1.0
186 plug(robot.cdc_estimator.c, estimator.com)
187 plug(robot.cdc_estimator.dc, estimator.momenta)
189 robot.estimator = estimator
195 zmp_estimator = SimpleZmpEstimator(
"zmpEst")
196 robot.rdynamic.createOpPoint(
"sole_LF",
"left_sole_link")
197 robot.rdynamic.createOpPoint(
"sole_RF",
"right_sole_link")
198 plug(robot.rdynamic.sole_LF, zmp_estimator.poseLeft)
199 plug(robot.rdynamic.sole_RF, zmp_estimator.poseRight)
200 plug(robot.ftc.left_foot_force_out, zmp_estimator.wrenchLeft)
201 plug(robot.ftc.right_foot_force_out, zmp_estimator.wrenchRight)
203 robot.zmp_estimator = zmp_estimator
209 Ki_dcm = [0.0, 0.0, 0.0]
213 dcm_controller = DcmController(
"dcmCtrl")
215 dcm_controller.Kp.value = Kp_dcm
216 dcm_controller.Ki.value = Ki_dcm
217 dcm_controller.Kz.value = Kz_dcm
218 dcm_controller.decayFactor.value = gamma_dcm
219 dcm_controller.mass.value = mass
220 plug(robot.wp.omegaDes, dcm_controller.omega)
222 plug(robot.cdc_estimator.c, dcm_controller.com)
223 plug(robot.estimator.dcm, dcm_controller.dcm)
225 plug(robot.wp.zmpDes, dcm_controller.zmpDes)
226 plug(robot.wp.dcmDes, dcm_controller.dcmDes)
228 plug(robot.zmp_estimator.zmp, dcm_controller.zmp)
230 dcm_controller.init(dt)
232 robot.dcm_control = dcm_controller
234 Ki_dcm = [1.0, 1.0, 1.0]
236 Kz_dcm = [0.0, 0.0, 0.0]
239 Kp_adm = [0.0, 0.0, 0.0]
241 com_admittance_control = ComAdmittanceController(
"comAdmCtrl")
242 com_admittance_control.Kp.value = Kp_adm
243 plug(robot.zmp_estimator.zmp, com_admittance_control.zmp)
244 com_admittance_control.zmpDes.value = (
245 robot.wp.zmpDes.value
247 plug(robot.wp.acomDes, com_admittance_control.ddcomDes)
249 com_admittance_control.init(dt)
250 com_admittance_control.setState(robot.wp.comDes.value, [0.0, 0.0, 0.0])
252 robot.com_admittance_control = com_admittance_control
254 Kp_adm = [15.0, 15.0, 0.0]
258 robot.cm.addCtrlMode(
"sot_input")
259 robot.cm.setCtrlMode(
"all",
"sot_input")
260 robot.cm.addEmergencyStopSIN(
"zmp")
265 robot.taskUpperBody = Task(
"task_upper_body")
266 robot.taskUpperBody.feature = FeaturePosture(
"feature_upper_body")
268 q = list(robot.dynamic.position.value)
269 robot.taskUpperBody.feature.state.value = q
270 robot.taskUpperBody.feature.posture.value = q
272 robotDim = robot.dynamic.getDimension()
273 for i
in range(18, robotDim):
274 robot.taskUpperBody.feature.selectDof(i,
True)
276 robot.taskUpperBody.controlGain.value = 100.0
277 robot.taskUpperBody.add(robot.taskUpperBody.feature.name)
278 plug(robot.dynamic.position, robot.taskUpperBody.feature.state)
282 robot.contactLF = MetaTaskKine6d(
283 "contactLF", robot.dynamic,
"LF", robot.OperationalPointsMap[
"left-ankle"]
285 robot.contactLF.feature.frame(
"desired")
286 robot.contactLF.gain.setConstant(300)
287 plug(robot.wp.footLeftDes, robot.contactLF.featureDes.position)
288 locals()[
"contactLF"] = robot.contactLF
290 robot.contactRF = MetaTaskKine6d(
291 "contactRF", robot.dynamic,
"RF", robot.OperationalPointsMap[
"right-ankle"]
293 robot.contactRF.feature.frame(
"desired")
294 robot.contactRF.gain.setConstant(300)
295 plug(robot.wp.footRightDes, robot.contactRF.featureDes.position)
296 locals()[
"contactRF"] = robot.contactRF
299 robot.taskComH = MetaTaskKineCom(robot.dynamic, name=
"comH")
300 plug(robot.wp.comDes, robot.taskComH.featureDes.errorIN)
301 robot.taskComH.task.controlGain.value = 100.0
302 robot.taskComH.feature.selec.value =
"100"
305 robot.taskCom = MetaTaskKineCom(robot.dynamic)
306 plug(robot.com_admittance_control.comRef, robot.taskCom.featureDes.errorIN)
307 plug(robot.com_admittance_control.dcomRef, robot.taskCom.featureDes.errordotIN)
308 robot.taskCom.task.controlGain.value = 100.0
309 robot.taskCom.task.setWithDerivative(
True)
310 robot.taskCom.feature.selec.value =
"011"
313 robot.keepWaist = MetaTaskKine6d(
314 "keepWaist", robot.dynamic,
"WT", robot.OperationalPointsMap[
"waist"]
316 robot.keepWaist.feature.frame(
"desired")
317 robot.keepWaist.gain.setConstant(300)
318 plug(robot.wp.waistDes, robot.keepWaist.featureDes.position)
319 robot.keepWaist.feature.selec.value =
"111000"
320 locals()[
"keepWaist"] = robot.keepWaist
323 robot.sot = SOT(
"sot")
324 robot.sot.setSize(robot.dynamic.getDimension())
327 robot.integrate = SimpleStateIntegrator(
"integrate")
328 robot.integrate.init(dt)
329 robot.integrate.setState(robot.device.state.value)
330 robot.integrate.setVelocity(robot.dynamic.getDimension() * [0.0])
333 plug(robot.sot.control, robot.cm.ctrl_sot_input)
334 plug(robot.cm.u_safe, robot.integrate.control)
335 plug(robot.integrate.state, robot.device.control)
337 robot.sot.push(robot.taskUpperBody.name)
338 robot.sot.push(robot.contactRF.task.name)
339 robot.sot.push(robot.contactLF.task.name)
340 robot.sot.push(robot.taskComH.task.name)
341 robot.sot.push(robot.taskCom.task.name)
342 robot.sot.push(robot.keepWaist.task.name)
348 robot.delay_vel.setMemory(robot.dynamic.getDimension() * [0.0])
349 robot.device.before.addSignal(robot.delay_vel.name +
".current")
350 plug(robot.cm.u_safe, robot.delay_vel.sin)
353 plug(robot.delay_vel.previous, robot.vselec.sin)
356 plug(robot.delay_vel.previous, robot.dynamic.velocity)
357 robot.dvdt = Derivator_of_Vector(
"dv_dt")
358 robot.dvdt.dt.value = dt
359 plug(robot.delay_vel.previous, robot.dvdt.sin)
360 plug(robot.dvdt.sout, robot.dynamic.acceleration)
367 create_topic(robot.publisher, robot.device,
"state", robot=robot, data_type=
"vector")
369 robot.publisher, robot.base_estimator,
"q", robot=robot, data_type=
"vector"
374 robot.publisher, robot.comTrajGen,
"x", robot=robot, data_type=
"vector"
377 robot.publisher, robot.comTrajGen,
"dx", robot=robot, data_type=
"vector"
380 robot.publisher, robot.comTrajGen,
"ddx", robot=robot, data_type=
"vector"
384 robot.publisher, robot.wp,
"comDes", robot=robot, data_type=
"vector"
388 robot.publisher, robot.cdc_estimator,
"c", robot=robot, data_type=
"vector"
391 robot.publisher, robot.cdc_estimator,
"dc", robot=robot, data_type=
"vector"
396 robot.com_admittance_control,
402 robot.publisher, robot.dynamic,
"com", robot=robot, data_type=
"vector"
406 robot.publisher, robot.dcm_control,
"dcmDes", robot=robot, data_type=
"vector"
409 robot.publisher, robot.estimator,
"dcm", robot=robot, data_type=
"vector"
413 robot.publisher, robot.zmpTrajGen,
"x", robot=robot, data_type=
"vector"
416 robot.publisher, robot.wp,
"zmpDes", robot=robot, data_type=
"vector"
419 robot.publisher, robot.dynamic,
"zmp", robot=robot, data_type=
"vector"
422 robot.publisher, robot.zmp_estimator,
"zmp", robot=robot, data_type=
"vector"
425 robot.publisher, robot.dcm_control,
"zmpRef", robot=robot, data_type=
"vector"
435 robot.publisher, robot.waistTrajGen,
"x", robot=robot, data_type=
"vector"
439 robot.publisher, robot.lfTrajGen,
"x", robot=robot, data_type=
"vector"
442 robot.publisher, robot.rfTrajGen,
"x", robot=robot, data_type=
"vector"
446 robot.publisher, robot.ftc,
"left_foot_force_out", robot=robot, data_type=
"vector"
449 robot.publisher, robot.ftc,
"right_foot_force_out", robot=robot, data_type=
"vector"
453 robot.publisher, robot.dynamic,
"LF", robot=robot, data_type=
"matrixHomo"
456 robot.publisher, robot.dynamic,
"RF", robot=robot, data_type=
"matrixHomo"
460 robot.tracer = TracerRealTime(
"com_tracer")
461 robot.tracer.setBufferSize(80 * (2**20))
462 robot.tracer.open(
"/tmp",
"dg_",
".dat")
463 robot.device.after.addSignal(
"{0}.triger".format(robot.tracer.name))
465 addTrace(robot.tracer, robot.wp,
"comDes")
467 addTrace(robot.tracer, robot.cdc_estimator,
"c")
470 addTrace(robot.tracer, robot.com_admittance_control,
"comRef")
471 addTrace(robot.tracer, robot.dynamic,
"com")
478 addTrace(robot.tracer, robot.zmp_estimator,
"zmp")
479 addTrace(robot.tracer, robot.dcm_control,
"zmpRef")