Exemple #1
0
def getBVHFeatures(body_part, filename_path):
    """
    Extracts X,Y,Z coordinate data per frame for a given body part
    from a time-series BVH file. BVH file must be recursively parsed
    prior to the extraction to locate the relevant node to begin processing.

    Args:
        body_part (string): body part whose Cartesian coordinate is
        to be extracted filename_path (string): path of BVH file

    Returns:
        list: list of lists of X,Y,Z coordiantes. The number of lists
        must equal to number of frames in the BVH
    """
    myskeleton = process_bvhfile(filename_path, DEBUG=0)
    frameTime = myskeleton.dt

    for t in range(1, myskeleton.frames+1):
        myskeleton.create_edges_onet(t)

    nodeFound = findTargetNode(myskeleton.hips, body_part, [])[0]
    XYZ_displacement = getFrameLevelDisplacements(
        nodeFound,
        1,
        myskeleton.frames +
        1)
    return XYZ_displacement
Exemple #2
0
def getBVHFeatures(body_part, filename_path):
    """
    Extracts X,Y,Z coordinate data per frame for a given body part
    from a time-series BVH file. BVH file must be recursively parsed
    prior to the extraction to locate the relevant node to begin processing.

    Args:
        body_part (string): body part whose Cartesian coordinate is
        to be extracted filename_path (string): path of BVH file

    Returns:
        list: list of lists of X,Y,Z coordiantes. The number of lists
        must equal to number of frames in the BVH
    """
    myskeleton = process_bvhfile(filename_path, DEBUG=0)
    frameTime = myskeleton.dt

    for t in range(1, myskeleton.frames + 1):
        myskeleton.create_edges_onet(t)

    nodeFound = findTargetNode(myskeleton.hips, body_part, [])[0]
    XYZ_displacement = getFrameLevelDisplacements(nodeFound, 1,
                                                  myskeleton.frames + 1)
    return XYZ_displacement
def open_file():  # read_file file_read load_file
    global root  # Root window
    global myskeleton
    global skelscreenedges
    global gridedges
    global slidert
    global myviewport
    global mycamera
    global mymenu
    global mytransport
    global redraw_grid
    global redraw_axes
    global file_prefix

    # No, you aren't allowed to try to load a new BVH in the middle of playing
    # back the current BVH... nice try.
    mytransport.playing = 0

    mycanvas = myviewport.canvas
    if file_prefix == 'NONE':
        filename = askopenfilename(title = 'Open BVH file', parent=root, \
                   filetypes =[ ('BVH files', '*.bvh'), ('All files', '*')]     )
    else:
        filename = askopenfilename(title = 'Open BVH file', parent=root, \
                   initialdir = file_prefix,                             \
                   filetypes =[ ('BVH files', '*.bvh'), ('All files', '*')]     )

    print "filename = ", filename  # Remove this line later
    index = filename.rfind('/')  # Unix
    index2 = filename.rfind('\\')  # Windows
    if index != -1:
        file_prefix = filename[0:index + 1]
        print "File prefix is ", file_prefix
    elif index2 != -1:
        file_prefix = filename[0:index2 + 1]
        print "File prefix is ", file_prefix

    # askopenfilename also allows: initialdir = ''
    # "filename" will have length 0 if user cancels the open.
    if len(filename) > 0:
        try:
            if filename[-3:] == 'bvh':
                myskeleton2 = process_bvhfile(filename, DEBUG=0)
            else:
                myskeleton2 = process_bvhfile(filename, DEBUG=0)
                print 'wangqi'
    ## myskeleton2 = profile.run('process_bvhfile(FILE,DEBUG=0)')
        except IOError:
            string = "IO Error while attempting to read file."
            print string
            error_popup(string)
            return
        except SyntaxError:
            string = "Parse error in BVH file."
            print string
            error_popup(string)
            return
        except:
            string = "Unknown error while attempting to read file."
            print string
            error_popup(string)
            return

    # If we make it here then we've successfully read and parsed the BVH file,
    # and created myskeleton2

    # Undraw all grid edges
        for edge in gridedges:
            edge.undraw(mycanvas)
    # Undraw all skeleton edges
        for edge in skelscreenedges:
            edge.undraw(mycanvas)
    # Don't need to undraw axes - leave as-is

    # Reset slider to 1
        slidert.set(1)

        # Say farewell to previous skeleton
        myskeleton = myskeleton2

        # Set camera position to something that will hopefully put the skeleton
        # in view.  For Z we want something large so that we're set back from
        # the skeleton.
        mycamera.t[0] = int((myskeleton.minx + myskeleton.maxx) / 2)
        mycamera.t[1] = int((myskeleton.miny + myskeleton.maxy) / 2)
        if (mycamera.t[1] < 10): mycamera.t[1] = 10
        mycamera.t[2] = myskeleton.maxz + 100
        mycamera.yrot = 0
        mycamera.Recompute()
        if mymenu.readout:
            myviewport.draw_readout(mycamera)

# Create skelscreenedges[] with sufficient space to handle one
# screenedge per skeleton edge.
        skelscreenedges = myskeleton.make_skelscreenedges(DEBUG=0,
                                                          arrow='none',
                                                          circle=1)
        # Give us some screen edges to display.
        myskeleton.populate_skelscreenedges(skelscreenedges, slidert.get(),  \
                                                DEBUG=0)
        # print "skelscreenedges after population is:"
        # print skelscreenedges

        myplaybar.resetscale(myskeleton.frames)
        gridedges = grid_setup(myskeleton.minx, myskeleton.minz, \
                     myskeleton.maxx, myskeleton.maxz, DEBUG=0)
        redraw_grid = 1
        redraw_axes = 1
        redraw()
Exemple #4
0
 def __init__(self, bvh_filename, camera_x, camera_y, camera_z):
     self._skeleton = process_bvhfile(bvh_filename)
     self._camera = Camera(camera_x, camera_y, camera_z, cfx=20, ppdist=30)
     self._skelscreenedges = self._skeleton.make_skelscreenedges()
     self._floor_y = None
Exemple #5
0
from skeleton import process_bvhfile

import glob
import os
import json
import numpy as np
from collections import OrderedDict

last_desc = ""
files = glob.glob("Data/ACCAD/Transform_Male1_bvh/Short_data/*.bvh")
print("Number of data : {}".format(len(files)))

for fi in files:
    try:
        print("Processing %s" % fi)
        skel = process_bvhfile(fi)
    except ValueError:
        print("Skipping %s" % fi)
        continue

    body = OrderedDict()

    for frame in range(skel.frames):

        skel.create_edges_onet(frame)

        for edge in skel.edges[frame]:
            for vert in (edge.wv1, edge.wv2):
                if vert.descr is not last_desc:
                    if vert.descr not in body:
                        body[vert.descr] = []
def open_file():      # read_file file_read load_file
  global root  # Root window
  global myskeleton
  global skelscreenedges
  global gridedges
  global slidert
  global myviewport
  global mycamera
  global mymenu
  global mytransport
  global redraw_grid
  global redraw_axes
  global file_prefix

# No, you aren't allowed to try to load a new BVH in the middle of playing
# back the current BVH... nice try.
  mytransport.playing = 0

  mycanvas = myviewport.canvas
  if file_prefix == 'NONE':
    filename = askopenfilename(title = 'Open BVH file', parent=root, \
               filetypes =[ ('BVH files', '*.bvh'), ('All files', '*')]     )
  else:
    filename = askopenfilename(title = 'Open BVH file', parent=root, \
               initialdir = file_prefix,                             \
               filetypes =[ ('BVH files', '*.bvh'), ('All files', '*')]     )

  print "filename = ",filename  # Remove this line later
  index = filename.rfind('/')  # Unix
  index2 = filename.rfind('\\') # Windows
  if index != -1:
    file_prefix = filename[0:index+1]
    print "File prefix is ", file_prefix
  elif index2 != -1:      
    file_prefix = filename[0:index2+1]
    print "File prefix is ", file_prefix      

  # askopenfilename also allows: initialdir = ''
  # "filename" will have length 0 if user cancels the open.
  if len(filename) > 0:
      try:
        myskeleton2 = process_bvhfile(filename,DEBUG=0)
     ## myskeleton2 = profile.run('process_bvhfile(FILE,DEBUG=0)')
      except IOError:
        string = "IO Error while attempting to read file."
        print string
        error_popup(string)
        return
      except SyntaxError:
        string = "Parse error in BVH file."
        print string
        error_popup(string)
        return
      except:
        string = "Unknown error while attempting to read file."
        print string
        error_popup(string)
        return

  # If we make it here then we've successfully read and parsed the BVH file,
  # and created myskeleton2

  # Undraw all grid edges
      for edge in gridedges:
          edge.undraw(mycanvas)
  # Undraw all skeleton edges
      for edge in skelscreenedges:
          edge.undraw(mycanvas)
  # Don't need to undraw axes - leave as-is

  # Reset slider to 1
      slidert.set(1)

# Say farewell to previous skeleton
      myskeleton = myskeleton2  

# Set camera position to something that will hopefully put the skeleton
# in view.  For Z we want something large so that we're set back from
# the skeleton.
      mycamera.t[0] = int((myskeleton.minx + myskeleton.maxx)/2)
      mycamera.t[1] = int((myskeleton.miny + myskeleton.maxy)/2)
      if (mycamera.t[1] < 10): mycamera.t[1] = 10
      mycamera.t[2] = myskeleton.maxz + 100
      mycamera.yrot = 0
      mycamera.Recompute()
      if mymenu.readout:
          myviewport.draw_readout(mycamera)

# Create skelscreenedges[] with sufficient space to handle one
# screenedge per skeleton edge.
      skelscreenedges = myskeleton.make_skelscreenedges(DEBUG=0, arrow='none',
                                                  circle=1)
# Give us some screen edges to display.
      myskeleton.populate_skelscreenedges(skelscreenedges, slidert.get(),  \
                                              DEBUG=0)
# print "skelscreenedges after population is:"
# print skelscreenedges

      myplaybar.resetscale(myskeleton.frames)
      gridedges = grid_setup(myskeleton.minx, myskeleton.minz, \
                   myskeleton.maxx, myskeleton.maxz, DEBUG=0)
      redraw_grid = 1
      redraw_axes = 1
      redraw()