diff --git a/io_anim_c3d/__init__.py b/io_anim_c3d/__init__.py
index 66d1419be3b49a2db512a4f49064d2fe8fb80ff1..2c9cd0d6320e50e7e88367e6f6f87696fb9f91c0 100644
--- a/io_anim_c3d/__init__.py
+++ b/io_anim_c3d/__init__.py
@@ -25,8 +25,8 @@
bl_info = {
"name": "C3D Graphics Lab Motion Capture file (.c3d)",
"author": "Daniel Monteiro Basso <daniel@basso.inf.br>",
- "version": (2013, 12, 10, 1),
- "blender": (2, 69, 5),
+ "version": (2015, 5, 5, 1),
+ "blender": (2, 74, 1),
"location": "File > Import",
"description": "Imports C3D Graphics Lab Motion Capture files",
"wiki_url": "http://wiki.blender.org/index.php/Extensions:2.6/Py/"
@@ -37,14 +37,14 @@ bl_info = {
import bpy
from bpy.props import (
- StringProperty,
- BoolProperty,
- FloatProperty,
- IntProperty,
- )
+ StringProperty,
+ BoolProperty,
+ FloatProperty,
+ IntProperty,
+)
+import os
import math
-import time
from mathutils import Vector
from . import import_c3d
@@ -56,6 +56,7 @@ class C3DAnimateCloud(bpy.types.Operator):
bl_idname = "import_anim.c3danim"
bl_label = "Animate C3D"
+ is_armature = False
markerset = None
uname = None
curframe = 0
@@ -64,25 +65,46 @@ class C3DAnimateCloud(bpy.types.Operator):
timer = None
Y_up = False
+ def update_empty(self, fno, ml, m):
+ name = self.unames[self.prefix + ml]
+ o = bpy.context.scene.objects[name]
+ p = Vector(m.position) * self.scale
+ o.location = Vector((p[0], -p[2], p[1])) if self.Y_up else p
+ o.keyframe_insert('location', frame=fno)
+
+ def update_bone(self, fno, ml, m, bones):
+ name = self.prefix + ml
+ if name not in bones:
+ return
+ b = bones[name]
+ p = Vector(m.position) * self.scale
+ b.matrix.translation = Vector((p[0], -p[2], p[1])) if self.Y_up else p
+ b.keyframe_insert('location', -1, fno, name)
+
+ def update_frame(self):
+ fno = self.curframe
+ if not self.use_frame_no:
+ fno = (self.curframe - self.markerset.startFrame) / self.fskip
+ for i in range(self.fskip):
+ self.markerset.readNextFrameData()
+ if self.is_armature:
+ bones = bpy.context.active_object.pose.bones
+ for ml in self.markerset.markerLabels:
+ m = self.markerset.getMarker(ml, self.curframe)
+ if m.confidence < self.confidence:
+ continue
+ if self.is_armature:
+ self.update_bone(fno, ml, m, bones)
+ else:
+ self.update_empty(fno, ml, m)
+
def modal(self, context, event):
if event.type == 'ESC':
return self.cancel(context)
if event.type == 'TIMER':
if self.curframe > self.markerset.endFrame:
return self.cancel(context)
- fno = self.curframe
- if not self.use_frame_no:
- fno = (self.curframe - self.markerset.startFrame) / self.fskip
- for i in range(self.fskip):
- self.markerset.readNextFrameData()
- for ml in self.markerset.markerLabels:
- name = self.unames[self.prefix + ml]
- o = bpy.context.scene.objects[name]
- m = self.markerset.getMarker(ml, self.curframe)
- p = Vector(m.position) * self.scale
- o.location = Vector((p[0], -p[2], p[1])) if self.Y_up else p
- if m.confidence >= self.confidence:
- o.keyframe_insert('location', frame=fno)
+ self.update_frame()
self.curframe += self.fskip
return {'PASS_THROUGH'}
@@ -106,91 +128,98 @@ class C3DImporter(bpy.types.Operator):
bl_label = "Import C3D"
filepath = StringProperty(
- subtype='FILE_PATH',
- )
+ subtype='FILE_PATH',
+ )
Y_up = BoolProperty(
- name="Up vector is Y axis",
- default=False,
- description="Check when the data uses Y-up, uncheck when it uses Z-up",
- )
+ name="Up vector is Y axis",
+ default=False,
+ description="Check when the data uses Y-up, uncheck when it uses Z-up",
+ )
from_inches = BoolProperty(
- name="Convert from inches to meters",
- default=False,
- description="Scale by 2.54/100",
- )
+ name="Convert from inches to meters",
+ default=False,
+ description="Scale by 2.54/100",
+ )
scale = FloatProperty(
- name="Scale",
- default=1.0,
- description="Scale the positions by this value",
- min=0.0000001, max=1000000.0,
- soft_min=0.001, soft_max=100.0,
- )
+ name="Scale",
+ default=1.0,
+ description="Scale the positions by this value",
+ min=0.0000001, max=1000000.0,
+ soft_min=0.001, soft_max=100.0,
+ )
auto_scale = BoolProperty(
- name="Adjust scale automatically",
- default=False,
- description="Guess correct scale factor",
- )
+ name="Adjust scale automatically",
+ default=False,
+ description="Guess correct scale factor",
+ )
auto_magnitude = BoolProperty(
- name="Adjust scale magnitude",
- default=True,
- description="Automatically adjust scale magnitude",
- )
+ name="Adjust scale magnitude",
+ default=True,
+ description="Automatically adjust scale magnitude",
+ )
+ create_armature = BoolProperty(
+ name="Create an armature",
+ default=True,
+ description="Import the markers as bones instead of empties",
+ )
size = FloatProperty(
- name="Empty Size",
- default=.03,
- description="The size of each empty",
- min=0.0001, max=1000000.0,
- soft_min=0.001, soft_max=100.0,
- )
+ name="Empty or bone size",
+ default=.03,
+ description="The size of each empty or bone",
+ min=0.0001, max=1000000.0,
+ soft_min=0.001, soft_max=100.0,
+ )
x_ray = BoolProperty(
- name="Use X-Ray",
- default=True,
- description="Show the empties over other objects",
- )
+ name="Use X-Ray",
+ default=True,
+ description="Show the empties or armature over other objects",
+ )
frame_skip = IntProperty(
- name="Fps divisor",
- default=4,
- # usually the sample rate is 120, so the default 4 gives you 30fps
- description="Frame supersampling factor",
- min=1,
- )
+ name="Fps divisor",
+ default=1,
+ description="Frame supersampling factor",
+ min=1,
+ )
use_frame_no = BoolProperty(
- name="Use frame numbers",
- default=False,
- description="Offset start of animation according to the source",
- )
+ name="Use frame numbers",
+ default=False,
+ description="Offset start of animation according to the source",
+ )
show_names = BoolProperty(
- name="Show Names", default=False,
- description="Show the markers' name",
- )
+ name="Show Names", default=False,
+ description="Show the markers' name",
+ )
prefix = StringProperty(
- name="Name Prefix", maxlen=32,
- description="Prefix object names with this",
- )
+ name="Name Prefix", maxlen=32,
+ description="Prefix object names with this",
+ )
use_existing = BoolProperty(
- name="Use existing empties",
- default=False,
- description="Use previously created homonymous empties",
- )
+ name="Use existing empties or armature",
+ default=False,
+ description="Use previously created homonymous empties or bones",
+ )
confidence = FloatProperty(
- name="Minimum Confidence Level", default=0,
- description="Only consider markers with at least "
- "this confidence level",
- min=-1., max=1000000.0,
- soft_min=-1., soft_max=100.0,
- )
+ name="Minimum Confidence Level", default=0,
+ description="Only consider markers with at least "
+ "this confidence level",
+ min=-1., max=1000000.0,
+ soft_min=-1., soft_max=100.0,
+ )
filter_glob = StringProperty(default="*.c3d;*.csv", options={'HIDDEN'})
def find_height(self, ms):
"""
Heuristic to find the height of the subject in the markerset
- (only works for standing poses)
+ (only works for standing poses and you must have correct data
+ on the first frame)
"""
- zmin = None
+ zmin, zmax = None, None
hidx = 1 if self.properties.Y_up else 2
for ml in ms.markerLabels:
+ # check if LTOE is a substring of this marker label
if 'LTOE' in ml:
+ # substitute the substring to get the head marker
hd = ml.replace('LTOE', 'LFHD')
if hd not in ms.markerLabels:
break
@@ -198,6 +227,7 @@ class C3DImporter(bpy.types.Operator):
pmax_idx = ms.markerLabels.index(hd)
zmin = ms.frames[0][pmin_idx].position[hidx]
zmax = ms.frames[0][pmax_idx].position[hidx]
+ break
if zmin is None: # could not find named markers, get extremes
allz = [m.position[hidx] for m in ms.frames[0]]
zmin, zmax = min(allz), max(allz)
@@ -205,43 +235,32 @@ class C3DImporter(bpy.types.Operator):
def adjust_scale_magnitude(self, height, scale):
mag = math.log10(height * scale)
- #print('mag',mag, 'scale',scale)
return scale * math.pow(10, -int(mag))
def adjust_scale(self, height, scale):
+ """
+ Try to find the correct scale for some common configurations
+ found in CMU's c3d files.
+ """
factor = height * scale / 1.75 # normalize
if factor < 0.5:
scale /= 10.0
factor *= 10.0
cmu_factors = [(1.0, 1.0), (1.1, 1.45), (1.6, 1.6), (2.54, 2.54)]
- sqerr, fix = min(((cf[0] - factor) ** 2.0, 1.0 / cf[1])
- for cf in cmu_factors)
- #print('height * scale: {:.2f}'.format(height * scale))
- #print(factor, fix)
+ sqerr, fix = min(
+ ((cf[0] - factor) ** 2.0, 1.0 / cf[1])
+ for cf in cmu_factors
+ )
return scale * fix
- def execute(self, context):
- s = self.properties.size
- empty_size = (s, s, s)
- ms = import_c3d.read(self.properties.filepath, onlyHeader=True)
- ms.readNextFrameData()
- #print(ms.fileName)
-
- # determine the final scale
- height = self.find_height(ms)
- #print('h', height)
- scale = 1.0 if not self.properties.from_inches else 0.0254
- scale *= ms.scale
- if self.properties.auto_magnitude:
- scale = self.adjust_scale_magnitude(height, scale)
- #print('scale',scale)
- if self.properties.auto_scale:
- scale = self.adjust_scale(height, scale)
- scale *= self.properties.scale
-
- # create the empties and get their collision-free names
+ def create_empties(self, ms):
+ """
+ Create the empties and get their collision-free names
+ """
unames = {}
use_existing = self.properties.use_existing
+ s = self.properties.size
+ empty_size = (s, s, s)
for ml in ms.markerLabels:
name = self.properties.prefix + ml
if use_existing and name in bpy.context.scene.objects:
@@ -256,10 +275,62 @@ class C3DImporter(bpy.types.Operator):
o.show_x_ray = self.properties.x_ray
for name in unames.values():
bpy.context.scene.objects[name].select = True
+ return unames
+
+ def create_armature_obj(self, ms, scale):
+ """
+ Create or use existing armature, return a bone dict,
+ leave the armature in POSE mode
+ """
+ head_dir = Vector((0, 0, self.properties.size))
+ ao = bpy.context.active_object
+ # when using an existing armature we restrict importing
+ # the markers only for existing bones
+ if not self.properties.use_existing or not ao or ao.type != 'ARMATURE':
+ bpy.ops.object.add(type='ARMATURE', enter_editmode=True)
+ arm = bpy.context.active_object
+ arm.name = os.path.basename(self.properties.filepath)
+ arm.data.show_names = self.properties.show_names
+ arm.show_x_ray = self.properties.x_ray
+ for idx, ml in enumerate(ms.markerLabels):
+ name = self.properties.prefix + ml
+ bpy.ops.armature.select_all(action='DESELECT')
+ bpy.ops.armature.bone_primitive_add(name=name)
+ pos = Vector(ms.frames[0][idx].position) * scale
+ if self.properties.Y_up:
+ pos = Vector((pos[0], -pos[2], pos[1]))
+ b = arm.data.edit_bones[name]
+ b.head = pos + head_dir
+ b.tail = pos
+ bpy.ops.object.mode_set(mode='POSE')
+ bpy.ops.pose.select_all(action='SELECT')
+
+ def execute(self, context):
+ ms = import_c3d.read(self.properties.filepath, onlyHeader=True)
+ ms.readNextFrameData()
+
+ # determine the final scale
+ height = self.find_height(ms)
+ scale = 1.0 if not self.properties.from_inches else 0.0254
+ scale *= ms.scale
+ if self.properties.auto_magnitude:
+ scale = self.adjust_scale_magnitude(height, scale)
+ if self.properties.auto_scale:
+ scale = self.adjust_scale(height, scale)
+ scale *= self.properties.scale
+
+ if bpy.context.mode != 'OBJECT':
+ bpy.ops.object.mode_set(mode='OBJECT')
+ if self.properties.create_armature:
+ self.create_armature_obj(ms, scale)
+ else:
+ unames = self.create_empties(ms)
# start animating the empties
C3DAnimateCloud.markerset = ms
- C3DAnimateCloud.unames = unames
+ C3DAnimateCloud.is_armature = self.properties.create_armature
+ if not C3DAnimateCloud.is_armature:
+ C3DAnimateCloud.unames = unames
C3DAnimateCloud.scale = scale
C3DAnimateCloud.Y_up = self.properties.Y_up
C3DAnimateCloud.fskip = self.properties.frame_skip