adding data+code to git repo

1 files changed, 65 insertions, 0 deletions

diff --git a/data/skeleton.py b/data/skeleton.py
new file mode 100755
index 0000000..c5641c0
--- /dev/null
+++ b/data/skeleton.py
@@ -0,0 +1,65 @@
+#!/usr/bin/python
+
+import time
+import sys
+import math
+import matplotlib.pyplot as plt
+
+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
+
+
+#while 1:
+#    where = sk_file.tell()
+#    line = sk_file.readline()
+#    if not line:
+#        time.sleep(1)
+#        file.seek(where)
+#    else:
+print sys.argv
+splt = fig.add_subplot(111)
+for arg in sys.argv[1:]:
+    sk_file = open(arg)
+    for line in sk_file.readlines():
+        try:
+            frame, id, joint, state, x, y, z, dx, dy = line.split(',')
+        except:
+            run, frame, id, joint, state, x, y, z, dx, dy = line.split(',')
+        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 = []
+    
+plt.draw()
+plt.waitforbuttonpress()
+