#!/usr/bin/python import time import sys import math import copy frames = [0,0,0,0,0,0,0,0,0,0] pframes = [0,0,0,0,0,0,0,0,0,0] zvalue = [[],[],[],[],[],[],[],[],[],[]] pframe = 0 lines = [[],[],[],[],[],[],[],[],[],[]] num = 1 for arg in sys.argv[1:]: sk_file = open(arg) for line in sk_file: run, frame, id, joint, state, x, y, z, dx, dy = line.strip().split(',') if frame != 'Frame': id = int(id) frame = int(frame) if frame != pframe: for i in range(len(frames)): if frames[i] > 0 and pframe > pframes[i]+1: #print str(frames[i]) #if frames[i] >= 50 and frames[i] <= 100: #qt = int(frames[i]/4) #if zvalue[i][frames[i]-qt] - zvalue[i][qt] < 0: zdelta = [] for j in range(1,len(zvalue[i])): zdelta += [zvalue[i][j]-zvalue[i][j-1]] #print zdelta zdelta.sort() median = int(math.floor((len(zdelta)-1)/2)) #print median,len(zdelta)-1 #zdelta[100000] = 5 qt = int(frames[i]/4) #if zdelta[median] < 0: if zdelta[frames[i]-qt]> 0 and zdelta[qt] > 0: for l in lines[i]: print l num += 1 frames[i] = 0 lines[i] = [] zvalue[i] = [] pframe = frame if pframes[id] != frame: frames[id] += 1 pframes[id] = frame if joint == 'Head': zvalue[id] += [float(z)] lines[id] += [line.strip()]