#!/usr/bin/python

import time
import sys
import math
import matplotlib.pyplot as plt
import pickle

plt.ion()
fig = plt.figure(1)

#sk_file = open(sys.argv[1])
#j1 = "Spine"
#j2 = "HipCenter"
#vals = {}
#pframe = -1
X = []
Y = []
colors = ['r', 'k', 'y', 'g', 'b', 'm']
i = 0
l = 4
#person = 'Jean'
prun = 0


#while 1:
#    where = sk_file.tell()
#    line = sk_file.readline()
#    if not line:
#        time.sleep(1)
#        file.seek(where)
#    else:
splt = fig.add_subplot(111)
people = pickle.load(open(sys.argv[2]))
for person in people:
    plt.cla()
    print person
    sk_file = open(sys.argv[1])
    for line in sk_file.readlines():
        sl = line.split(',') #frame, id, joint, state, x, y, z, dx, dy = line.split(',')
        name = sl[0]
        if name == person and sl[l] != '?':
            run = int(sl[1])
            if prun != 0 and run != prun:
                c = "%c+" % (colors[i],)
                splt.plot(X,Y,c)
                i = (i+1)%(len(colors)) 
                X = []
                Y = []


            dist = float(sl[3])
            limb = float(sl[l])
            X += [dist]
            Y += [limb]
            prun = run
    plt.draw()
    plt.waitforbuttonpress()

#        if frame != pframe and vals != {}:
#            if j1 in vals and j2 in vals:
#                dx = vals[j1][0] - vals[j2][0]
#                dy = vals[j1][1] - vals[j2][1]
#                dz = vals[j1][2] - vals[j2][2]
#                limb = math.sqrt(dx*dx + dy*dy + dz*dz)
#                dist = math.fabs((dz + 2*vals[j2][2])/2)
#                X += [dist]
#                Y += [limb]
#                #print str(dist) + " " + str(limb)
#                #print str(dist) + "\t" + str(limb)
#            vals = {}
#        if frame != 'Frame':
#            vals[joint] = [float(x), float(y), float(z)]
#            pframe = frame


    #plt.figure(1)
    #c = "%c+" % (colors[i],)
    #splt.plot(X,Y,c)
    #i = (i+1)%(len(colors)) 
    #print c
    #X = []
    #Y = []