def main():
if len(sys.argv) < 3:
print 'Usage: %s input_gait_file output_position_file' % sys.argv[0]
sys.exit(1)
gaitFile = sys.argv[1]
posFile = sys.argv[2]
strategy = TimedFileStrategy(posFile = gaitFile)
motionFunction, logInfo = strategy.getNext()
#runman = RunManager()
#runman.do_many_runs(strategy, SineModel5.typicalRanges)
#timeScale = .3
#motionFunctionScaled = scaleTime(motionFunction, timeScale)
wiiTrack = WiiTrackFastClient("localhost", 8080)
time.sleep(.5)
position,age = wiiTrack.getPosAge()
if age is None:
raise Exception('Could not get position from wiiTrack.')
robot = Robot(loud = True)
bucket = []
def foo():
savePosition(wiiTrack, bucket)
robot.run(motionFunction, runSeconds = 10, resetFirst = True,
interpBegin = 2, interpEnd = 2, extraLogInfoFn = foo)
print 'Positions:'
print len(bucket)
relTimeBucket = []
for ii, line in enumerate(bucket):
delta = line[0] - bucket[0][0]
relTime = delta.seconds + delta.microseconds/1e6
relTimeBucket.append((relTime, line[1], line[2]))
ff = open (posFile, 'w')
ff.write('# time (junk junk)x9 pos.x pos.y 0 age\n')
for ii, timePosAge in enumerate(relTimeBucket):
timeOfPosition, position, age = timePosAge
line = '%.3f' % timeOfPosition
line += ' -1 -1' * 9
line += ' %.1f %.1f %.1f' % (position[0], position[1], 0)
line += ' %f' % age
ff.write(line + '\n')
ff.close()
print 'Wrote position file:', posFile
def main22():
if len(sys.argv) > 2 and sys.argv[1] == '-filt':
filtFile = sys.argv[2]
strategy = FileStrategy(filtFile = filtFile)
motionFunction, logInfo = strategy.getNext()
elif len(sys.argv) > 2 and sys.argv[1] == '-sine':
sineModel5Params = [eval(xx) for xx in sys.argv[2].split()]
print 'Using SineModel5 with params: ', sineModel5Params
motionFunction = lambda time: SineModel5().model(time,
parameters = sineModel5Params)
else:
#filtFile = '../results/hyperneatTo20gens_101/neat_110115_175446_00014_008_filt'
filtFile = '../results/hyperneatTo20gens_101/neat_110115_175446_00004_007_filt'
strategy = FileStrategy(filtFile = filtFile)
motionFunction, logInfo = strategy.getNext()
#runman = RunManager()
#runman.do_many_runs(strategy, SineModel5.typicalRanges)
#timeScale = .3
#motionFunctionScaled = scaleTime(motionFunction, timeScale)
wiiTrack = WiiTrackFastClient("localhost", 8080)
time.sleep(.5)
position,age = wiiTrack.getPosAge()
if age is None:
raise Exception('Could not get position from wiiTrack.')
robot = Robot(loud = True)
bucket = []
def foo():
savePosition(wiiTrack, bucket)
robot.run(motionFunction, runSeconds = 8, resetFirst = False,
interpBegin = 2, interpEnd = 2, extraLogInfoFn = foo)
print bucket
