def __init__(self,robot,sequenceFile,trunkStabilize=False, hands=False, posture=False, forceSeqplay=True):
Application.__init__(self,robot)
self.hands =hands
self.posture = posture
self.trunkStabilize = trunkStabilize
self.robot = robot
plug(self.robot.dynamic.com,self.robot.device.zmp)
self.sot = self.solver.sot
self.seq = Seqplay ('seqplay')
self.seq.load (sequenceFile)
self.forceSeqplay = forceSeqplay
if self.forceSeqplay:
self.zmpRef = ZmpFromForces('zmpRef')
else:
self.zmpRef = self.seq
self.createTasks()
self.initTasks()
self.initTaskGains()
self.initialStack()
appli.robot.addTrace( robot.device.name, 'forceLLEG')
appli.robot.addTrace( robot.device.name, 'forceRLEG')
appli.robot.addTrace( robot.device.name, 'accelerometer')
appli.robot.addTrace( robot.device.name, 'gyrometer')
appli.startTracer()
appli.gains['trunk'].setConstant(5)
sequenceFile = '/home/mbenalle/devel/ros/install/resources/seqplay/walkfwd-resampled'
#sequenceFile = '/home/mbenalle/devel/ros/install/resources/seqplay/stand-on-left-foot'
seq = Seqplay ('seqplay')
seq.load (sequenceFile)
plug (seq.leftAnkle, appli.leftAnkle.reference)
plug (seq.rightAnkle, appli.rightAnkle.reference)
plug (seq.com, comRef)
plug (seq.comdot, appli.comdot)
plug (seq.leftAnkleVel, appli.leftAnkle.velocity)
plug (seq.rightAnkleVel, appli.rightAnkle.velocity)
plug (seq.posture, appli.featurePostureDes.errorIN)
zmpRef = ZmpFromForces('zmpRef')
plug (seq.forceLeftFoot , zmpRef.force_0)
plug (seq.forceRightFoot, zmpRef.force_1)
plug (robot.frames['leftFootForceSensor'].position , zmpRef.sensorPosition_0)
plug (robot.frames['rightFootForceSensor'].position, zmpRef.sensorPosition_1)
plug (zmpRef.zmp , robot.device.zmp)
class SeqPlayLqrTwoDofCoupledStabilizer(Application):
threePhaseScrew = True
tracesRealTime = True
def __init__(self,robot,sequenceFile,trunkStabilize=False, hands=False, posture=False, forceSeqplay=True):
Application.__init__(self,robot)
self.hands =hands
self.posture = posture
self.trunkStabilize = trunkStabilize
self.robot = robot
plug(self.robot.dynamic.com,self.robot.device.zmp)
self.sot = self.solver.sot
self.seq = Seqplay ('seqplay')
self.seq.load (sequenceFile)
self.forceSeqplay = forceSeqplay
if self.forceSeqplay:
self.zmpRef = ZmpFromForces('zmpRef')
else:
self.zmpRef = self.seq
self.createTasks()
self.initTasks()
self.initTaskGains()
self.initialStack()
# --- TASKS --------------------------------------------------------------------
# --- TASKS --------------------------------------------------------------------
# --- TASKS --------------------------------------------------------------------
def createTasks(self):
(self.tasks['trunk'],self.gains['trunk'])= createTrunkTask (self.robot, self, 'Tasktrunk')
(self.tasks['ankles'],self.gains['ankles'])= createAnklesTask (self.robot, self, 'TaskAnkles')
self.taskRF = self.tasks['left-ankle']
self.taskLF = self.tasks['right-ankle']
self.taskCom = self.tasks['com']
self.taskRH = self.tasks['right-wrist']
self.taskLH = self.tasks['left-wrist']
self.taskPosture = self.tasks['posture']
self.taskTrunk = self.tasks['trunk']
self.taskAnkles = self.tasks['ankles']
self.taskHalfStitting = MetaTaskPosture(self.robot.dynamic,'halfsitting')
(self.tasks['com-stabilized'],self.gains['com-stabilized']) = self.createStabilizedCoMTask()
self.taskCoMStabilized = self.tasks['com-stabilized']
#initialization is separated from the creation of the tasks because if we want to switch
#to second order controlm the initialization will remain while the creation is
#changed
def initTasks(self):
self.initTaskBalance()
self.initTaskPosture()
self.initTaskStabilize()
self.initSeqplay()
def initTaskBalance(self):
# --- BALANCE ---
self.features['chest'].frame('desired')
self.features['waist'].frame('desired')
self.features['gaze'].frame('desired')
#self.taskChest.feature.selec.value = '111111'
self.features['chest'].selec.value = '111000'
self.features['waist'].selec.value = '111000'
self.features['gaze'].selec.value = '111000'
self.featureCom.selec.value = '111'
def initTaskGains(self, setup = "medium"):
if setup == "medium":
self.gains['balance'].setConstant(10)
self.gains['trunk'].setConstant(10)
self.gains['com-stabilized'].setConstant(10)
self.gains['ankles'].setConstant(10)
self.gains['right-wrist'].setByPoint(4,0.2,0.01,0.8)
self.gains['left-wrist'].setByPoint(4,0.2,0.01,0.8)
self.taskHalfStitting.gain.setByPoint(2,0.2,0.01,0.8)
def createStabilizedCoMTask (self):
raise Exception("createStabilizedCoMTask is not overloaded")
def initSeqplay(self):
if self.forceSeqplay:
plug (self.seq.forceLeftFoot , self.zmpRef.force_0)
plug (self.seq.forceRightFoot, self.zmpRef.force_1)
plug (self.robot.frames['leftFootForceSensor'].position , self.zmpRef.sensorPosition_0)
plug (self.robot.frames['rightFootForceSensor'].position, self.zmpRef.sensorPosition_1)
plug (self.zmpRef.zmp , self.robot.device.zmp)
plug (self.zmpRef.zmp , self.tasks['com-stabilized'].zmpRef)
# --- SOLVER ----------------------------------------------------------------
def push(self,task,feature=None,keep=False):
if isinstance(task,str): taskName=task
elif "task" in task.__dict__: taskName=task.task.name
else: taskName=task.name
if taskName not in toList(self.sot):
self.sot.push(taskName)
if taskName!="posture" and "posture" in toList(self.sot):
self.sot.down("posture")
if keep: feature.keep()
def rm(self,task):
if isinstance(task,str): taskName=task
elif "task" in task.__dict__: taskName=task.task.name
else: taskName=task.name
if taskName in toList(self.sot): self.sot.remove(taskName)
# --- DISPLAY -------------------------------------------------------------
def initDisplay(self):
self.robot.device.viewer.updateElementConfig('red',[0,0,-1,0,0,0])
self.robot.device.viewer.updateElementConfig('yellow',[0,0,-1,0,0,0])
def updateDisplay(self):
'''Display the various objects corresponding to the calcul graph. '''
None
# --- TRACES -----------------------------------------------------------
def withTraces(self):
if self.tracesRealTime:
self.robot.tracerSize = 2**26
self.robot.initializeTracer()
else:
self.robot.tracer = Tracer('trace')
self.robot.device.after.addSignal('{0}.triger'.format(self.robot.tracer.name))
self.robot.tracer.open('/tmp/','','.dat')
#self.robot.tracer.add( self.taskRH.task.name+'.error','erh' )
def stopTracer(self):
self.robot.stopTracer()
def dumpTracer(self):
self.robot.tracer.dump()
def startTracer(self):
self.robot.startTracer()
# --- RUN --------------------------------------------------------------
def initialStack(self):
self.sot.clear()
self.push(self.tasks['balance'])
self.push(self.taskTrunk)
if self.hands:
self.push(self.taskRH)
self.push(self.taskLH)
if self.posture:
self.push(self.taskPosture)
#self.push(self.taskPosture)
def liftHands(self):
'''Lift both hanfs.'''
change6dPositionReference(self.taskRH,self.features['right-wrist'],\
self.gains['right-wrist'],\
(0.3,-0.25+self.comRef.value[1],1.1,0,-pi/2,0),'111111')
self.push(self.taskRH)
change6dPositionReference(self.taskLH,self.features['left-wrist'],\
self.gains['left-wrist'],\
(0.3,0.25+self.comRef.value[1],1.1,0,-pi/2,0),'111111')
self.push(self.taskLH)
None
def goHalfSitting(self):
'''End of application, go to final pose.'''
self.featurePostureDes.errorIN.value = self.robot.halfSitting
self.sot.clear()
self.comRef = self.tasks['com-stabilized'].comRef
self.push(self.tasks['balance'])
self.push(self.taskPosture)
# --- SEQUENCER ---
seqstep = 0
def nextStep(self,step=None):
if step!=None: self.seqstep = step
if self.seqstep==0:
self.prepareSeqplay()
print ('Seqplay prepared')
elif self.seqstep==1:
self.stabilize()
print ('Stabilizer on')
elif self.seqstep==2:
self.runSeqplay()
print ('Seqplay run')
elif self.seqstep==3:
self.liftHands()
print ('Hands lifted')
elif self.seqstep==4:
self.goHalfSitting()
print ('Half-Sitting the robot can be lifted')
self.seqstep += 1
def __add__(self,i):
self.nextStep()
# Stabilization ######################################
def initTaskStabilize(self):
from dynamic_graph.sot.application.state_observation import MovingFrameTransformation
if self.trunkStabilize:
### waist
self.transformerWaist = MovingFrameTransformation('tranformation_waist')
plug(self.ccMc,self.transformerWaist.gMl) # inverted flexibility
self.cMwaist = self.transformerWaist.lM0 # reference position in the world control frame
# You need to set up the inverted flexibility : plug( ..., self.ccMc)
# You need to set up a reference value here: plug( ... ,self.cMhref)
plug(self.ccVc,self.transformerWaist.gVl) # inverted flexibility velocity
self.cVwaist = self.transformerWaist.lV0 # reference velocity in the world control frame
# You need to set up the inverted flexibility velocity : plug( ..., self.ccVc)
# You need to set up a reference velocity value here: plug( ... ,self.cVhref)
self.ccMwaist = self.transformerWaist.gM0 # reference matrix h**o in the control frame
self.ccVwaist = self.transformerWaist.gV0
self.transformerWaist.setYawRemoved(True)
### chest
self.transformerChest = MovingFrameTransformation('tranformation_chest')
plug(self.ccMc,self.transformerChest.gMl) # inverted flexibility
self.cMchest = self.transformerChest.lM0 # reference position in the world control frame
# You need to set up the inverted flexibility : plug( ..., self.ccMc)
# You need to set up a reference value here: plug( ... ,self.cMhref)
plug(self.ccVc,self.transformerChest.gVl) # inverted flexibility velocity
self.cVchest = self.transformerChest.lV0 # reference velocity in the world control frame
# You need to set up the inverted flexibility velocity : plug( ..., self.ccVc)
# You need to set up a reference velocity value here: plug( ... ,self.cVhref)
self.ccMchest = self.transformerChest.gM0 # reference matrix h**o in the control frame
self.ccVchest = self.transformerChest.gV0
self.transformerChest.setYawRemoved(True)
### gaze
self.transformerGaze = MovingFrameTransformation('tranformation_gaze')
plug(self.ccMc,self.transformerGaze.gMl) # inverted flexibility
self.cMgaze = self.transformerGaze.lM0 # reference position in the world control frame
# You need to set up the inverted flexibility : plug( ..., self.ccMc)
# You need to set up a reference value here: plug( ... ,self.cMhref)
plug(self.ccVc,self.transformerGaze.gVl) # inverted flexibility velocity
self.cVgaze = self.transformerGaze.lV0 # reference velocity in the world control frame
# You need to set up the inverted flexibility velocity : plug( ..., self.ccVc)
# You need to set up a reference velocity value here: plug( ... ,self.cVhref)
self.ccMgaze = self.transformerGaze.gM0 # reference matrix h**o in the control frame
self.ccVgaze = self.transformerGaze.gV0
self.transformerGaze.setYawRemoved(True)
def stabilize(self):
'''Start to stabilize'''
self.sot.clear()
#self.tasks['com-stabilized'].comRef.value = (0.0, 0.0, self.tasks['com-stabilized'].comRef.value[2]) #self.tasks['com-stabilized'].com.value
self.comRef = self.tasks['com-stabilized'].comRef
self.push(self.tasks['ankles'])
self.push(self.tasks['com-stabilized'])
self.push(self.taskTrunk)
if self.hands:
self.push(self.taskRH)
self.push(self.taskLH)
#self.taskCoMStabilized.start()
if self.trunkStabilize:
#waist
self.cMwaist.value = self.robot.dynamic.signal('waist').value
self.cVwaist.value = (0.0 , 0.0 , 0.0 , 0.0 , 0.0 , 0.0)
plug(self.ccMwaist,self.features['waist'].reference)
plug(self.ccVwaist,self.features['waist'].velocity)
self.tasks['waist'].setWithDerivative (True)
### chest
self.cMchest.value = self.robot.dynamic.signal('chest').value
self.cVchest.value = (0.0 , 0.0 , 0.0 , 0.0 , 0.0 , 0.0)
plug(self.ccMchest,self.features['chest'].reference)
plug(self.ccVchest,self.features['chest'].velocity)
self.tasks['chest'].setWithDerivative (True)
### value
self.cMgaze.value = self.robot.dynamic.signal('gaze').value
self.cVgaze.value = (0.0 , 0.0 , 0.0 , 0.0 , 0.0 , 0.0)
plug(self.ccMgaze,self.features['gaze'].reference)
plug(self.ccVgaze,self.features['gaze'].velocity)
self.tasks['gaze'].setWithDerivative (True)
if self.posture:
self.push(self.taskPosture)
#######
#self.cMlhref.value = self.robot.dynamic.lh.value
#self.cVlhref.value = (0.0 , 0.0 , 0.0 , 0.0 , 0.0 , 0.0)
#print matrix(self.cMlhref.value)
######
# Seqplay ######################################
def prepareSeqplay(self):
self.seq.leftAnkle.recompute(2)
self.seq.rightAnkle.recompute(2)
self.seq.com.recompute(2)
self.seq.comdot.recompute(2)
self.seq.leftAnkleVel.recompute(2)
self.seq.rightAnkleVel.recompute(2)
self.seq.posture.recompute(2)
plug (self.seq.leftAnkle, self.leftAnkle.reference)
plug (self.seq.rightAnkle, self.rightAnkle.reference)
plug (self.seq.leftAnkleVel, self.leftAnkle.velocity)
plug (self.seq.rightAnkleVel, self.rightAnkle.velocity)
plug (self.seq.com, self.comRef)
plug (self.seq.comdot, self.comdot)
plug (self.seq.com, self.featureComDes.errorIN)
plug (self.seq.comdot, self.featureComDes.errordotIN)
plug (self.seq.com, self.tasks['com-stabilized'].comRef)
#plug (self.seq.comdot, self.tasks['com-stabilized'].comdotRef)
#plug (self.seq.comddot, self.tasks['com-stabilized'].comddotRef)
plug (self.seq.posture, self.featurePostureDes.errorIN)
def runSeqplay(self):
self.seq.start ()