import argparse
import numpy

from pybullet_envs.minitaur.envs import minitaur_logging

parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--log_file', help='path to protobuf file', default='')
args = parser.parse_args()
logging = minitaur_logging.MinitaurLogging()
episode = logging.restore_episode(args.log_file)
#print(dir (episode))
#print("episode=",episode)
fields = episode.ListFields()

recs = []

for rec in fields[0][1]:
  #print(rec.time)
  for motorState in rec.motor_states:
    #print("motorState.angle=",motorState.angle)
    #print("motorState.velocity=",motorState.velocity)
    #print("motorState.action=",motorState.action)
    #print("motorState.torque=",motorState.torque)
    recs.append([motorState.angle, motorState.velocity, motorState.action, motorState.torque])

a = numpy.array(recs)
numpy.savetxt("motorq_qdot_action_torque.csv", a, delimiter=",")