def create_example_solve_scene():
"""
Very basic single frame solver set up.
This function does not execute the solve, execution must be done manually.
:return: API Collection object.
"""
# Camera
cam_tfm = maya.cmds.createNode('transform',
name='cam_tfm')
cam_shp = maya.cmds.createNode('camera',
name='cam_shp',
parent=cam_tfm)
maya.cmds.setAttr(cam_tfm + '.tx', -1.0)
maya.cmds.setAttr(cam_tfm + '.ty', 1.0)
maya.cmds.setAttr(cam_tfm + '.tz', -5.0)
cam = mmapi.Camera(shape=cam_shp)
# Bundle
bnd = mmapi.Bundle().create_node()
bundle_tfm = bnd.get_node()
maya.cmds.setAttr(bundle_tfm + '.tx', 5.5)
maya.cmds.setAttr(bundle_tfm + '.ty', 6.4)
maya.cmds.setAttr(bundle_tfm + '.tz', -25.0)
assert mmapi.get_object_type(bundle_tfm) == 'bundle'
# Marker
mkr = mmapi.Marker().create_node(cam=cam, bnd=bnd)
marker_tfm = mkr.get_node()
assert mmapi.get_object_type(marker_tfm) == 'marker'
maya.cmds.setAttr(marker_tfm + '.tx', 0.0)
maya.cmds.setAttr(marker_tfm + '.ty', 0.0)
# Attributes
attr_tx = mmapi.Attribute(bundle_tfm + '.tx')
attr_ty = mmapi.Attribute(bundle_tfm + '.ty')
# Frames
frm_list = [
mmapi.Frame(1, primary=True)
]
# Solver
sol = mmapi.Solver()
sol.set_max_iterations(10)
sol.set_verbose(True)
sol.set_frame_list(frm_list)
# Collection
col = mmapi.Collection()
col.create_node('mySolveCollection')
col.add_solver(sol)
col.add_marker(mkr)
col.add_attribute(attr_tx)
col.add_attribute(attr_ty)
return col
def get_selected_node_default_attributes():
"""
Get the attributes on the selected nodes.
"""
attr_list = []
sel = maya.cmds.ls(selection=True, long=True) or []
if len(sel) == 0:
return attr_list
for node in sel:
node_type = maya.cmds.nodeType(node)
obj_type = mmapi.get_object_type(node)
attr_names = []
if obj_type == mmapi.OBJECT_TYPE_BUNDLE:
# Default bundle attributes.
attr_names += [
'translateX',
'translateY',
'translateZ',
]
elif obj_type == mmapi.OBJECT_TYPE_CAMERA:
# Camera default attributes, for both transform and
# camera nodes.
if node_type == 'transform':
attr_names += [
'translateX',
'translateY',
'translateZ',
'rotateX',
'rotateY',
'rotateZ',
]
elif node_type == 'camera':
attr_names += [
'focalLength',
]
else:
# Fallback - get all logical attributes.
attrs = maya.cmds.listAttr(
node,
keyable=True,
settable=True,
scalar=True,
shortNames=False,
) or []
attr_types = [
'doubleLinear',
'doubleAngle',
'double',
'float',
]
attr_names += [
n for n in attrs if maya.cmds.attributeQuery(
n, node=node, attributeType=True) in attr_types
]
for attr_name in attr_names:
attr_obj = mmapi.Attribute(node=node, attr=attr_name)
attr_list.append(attr_obj)
return attr_list
def collect_bundles(node_categories):
unknown_node_found = False
list_to_delete = list()
for node in node_categories['bundle']:
if cmds.objectType(node) != 'transform':
continue
list_to_delete.append(node)
children = cmds.listRelatives(node,
children=True,
type='transform',
fullPath=True)
if children:
for child in children:
if mmapi.get_object_type(child) != 'bundle':
unknown_node_found = True
return list_to_delete, unknown_node_found
def get_nodes(nodes):
"""
Return nodes sorted into mmSolver-specific object type categories.
Supported categories are:
- 'camera'
- 'marker'
- 'markergroup'
- 'bundle'
- 'attribute'
- 'collection'
- 'other'
:param nodes: Maya nodes to categorise.
:type nodes: list of str
:returns: Dictionary with lists for each object type.
:rtype: dict
"""
assert isinstance(nodes, (list, tuple))
result = {
'camera': [],
'marker': [],
'markergroup': [],
'bundle': [],
'attribute': [],
'collection': [],
'other': []
}
for node in nodes:
obj_type = mmapi.get_object_type(node)
if obj_type == mmapi.OBJECT_TYPE_MARKER:
result['marker'].append(node)
elif obj_type == mmapi.OBJECT_TYPE_MARKER_GROUP:
result['markergroup'].append(node)
elif obj_type == mmapi.OBJECT_TYPE_BUNDLE:
result['bundle'].append(node)
elif obj_type == mmapi.OBJECT_TYPE_CAMERA:
result['camera'].append(node)
elif obj_type == mmapi.OBJECT_TYPE_ATTRIBUTE:
result['attribute'].append(node)
elif obj_type == mmapi.OBJECT_TYPE_COLLECTION:
result['collection'].append(node)
else:
result['other'].append(node)
return result
def input_attributes_filter(attr_list):
"""
Apply logic to remove any non-input attributes from the given list.
:param attr_list: Attribute list to filter.
:type attr_list: [Attribute, ..]
:returns: List of attributes that are filtered.
:rtype: [Attribute, ..]
"""
result = []
for attr_obj in attr_list:
node = attr_obj.get_node()
obj_type = mmapi.get_object_type(node)
if obj_type in const.ATTR_INVALID_OBJECT_TYPES:
continue
result.append(attr_obj)
return result
def get_marker_groups(cam):
"""
Get all MarkerGroups for the given camera.
:rtype: list of mmSolver.api.MarkerGroup
"""
mkr_grp_list = []
if cam is None:
return mkr_grp_list
assert isinstance(cam, mmapi.Camera)
if cam.is_valid() is False:
return mkr_grp_list
cam_tfm = cam.get_transform_node()
below_nodes = maya.cmds.ls(
cam_tfm, dag=True, long=True,
type='mmMarkerGroupTransform') or []
mkr_grp_list = [mmapi.MarkerGroup(node=n) for n in below_nodes
if mmapi.get_object_type(n) == mmapi.OBJECT_TYPE_MARKER_GROUP]
return mkr_grp_list
def test_init(self):
"""
Single Frame solve.
"""
# Camera
cam_tfm = maya.cmds.createNode('transform',
name='cam_tfm')
cam_shp = maya.cmds.createNode('camera',
name='cam_shp',
parent=cam_tfm)
maya.cmds.setAttr(cam_tfm + '.tx', -1.0)
maya.cmds.setAttr(cam_tfm + '.ty', 1.0)
maya.cmds.setAttr(cam_tfm + '.tz', -5.0)
cam = mmapi.Camera(shape=cam_shp)
# Bundle
bnd = mmapi.Bundle().create_node()
bundle_tfm = bnd.get_node()
maya.cmds.setAttr(bundle_tfm + '.tx', 5.5)
maya.cmds.setAttr(bundle_tfm + '.ty', 6.4)
maya.cmds.setAttr(bundle_tfm + '.tz', -25.0)
assert mmapi.get_object_type(bundle_tfm) == 'bundle'
# Marker
mkr = mmapi.Marker().create_node(cam=cam, bnd=bnd)
marker_tfm = mkr.get_node()
assert mmapi.get_object_type(marker_tfm) == 'marker'
maya.cmds.setAttr(marker_tfm + '.tx', 0.0)
maya.cmds.setAttr(marker_tfm + '.ty', 0.0)
# Attributes
attr_tx = mmapi.Attribute(bundle_tfm + '.tx')
attr_ty = mmapi.Attribute(bundle_tfm + '.ty')
# Frames
frm_list = [
mmapi.Frame(1, primary=True)
]
# Solver
sol = mmapi.Solver()
sol.set_max_iterations(10)
sol.set_solver_type(mmapi.SOLVER_TYPE_DEFAULT)
sol.set_verbose(True)
sol.set_frame_list(frm_list)
# Collection
col = mmapi.Collection()
col.create_node('mySolveCollection')
col.add_solver(sol)
col.add_marker(mkr)
col.add_attribute(attr_tx)
col.add_attribute(attr_ty)
# save the output
path = self.get_data_path('test_solve_init_before.ma')
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
# Run solver!
results = col.execute()
# Set Deviation
mmapi.update_deviation_on_markers([mkr], results)
mmapi.update_deviation_on_collection(col, results)
# save the output
path = self.get_data_path('test_solve_init_after.ma')
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
# Ensure the values are correct
self.checkSolveResults(results)
# assert self.approx_equal(maya.cmds.getAttr(bundle_tfm+'.tx'), -6.0)
# assert self.approx_equal(maya.cmds.getAttr(bundle_tfm+'.ty'), 3.6)
return
def test_stA_refine_good_solve(self):
"""
Test file based on 3DEqualizer 'stA' image sequence.
The Maya file loaded contains a good 3DEqualizer solve. This
test tests the solver to ensure it produces good results,
given an already good solution.
The 'stA' image sequence has a frame range of 0 to 94.
"""
start_frame = 0
end_frame = 94
path = self.get_data_path('scenes', 'stA', 'stA.ma')
ok = maya.cmds.file(path, open=True, ignoreVersion=True, force=True)
assert isinstance(ok, (str, unicode))
# Camera
cam_name = 'stA_1_1Shape1'
cam = mmapi.Camera(shape=cam_name)
cam_tfm_node = cam.get_transform_node()
cam_shp_node = cam.get_shape_node()
# Marker Group
mkr_grp_name = 'markerGroup1'
mkr_grp = mmapi.MarkerGroup(node=mkr_grp_name)
mkr_grp_node = mkr_grp.get_node()
# Markers
mkr_list = []
bnd_list = []
mkr_nodes = maya.cmds.listRelatives(
mkr_grp_node,
children=True,
shapes=False) or []
for node in mkr_nodes:
if node.endswith('_MKR') is False:
continue
assert mmapi.get_object_type(node) == 'marker'
mkr = mmapi.Marker(node=node)
bnd = mkr.get_bundle()
mkr_list.append(mkr)
bnd_list.append(bnd)
assert len(mkr_list) > 0
assert len(bnd_list) > 0
# Attributes
attr_list = []
for bnd in bnd_list:
bnd_node = bnd.get_node()
attr_tx = mmapi.Attribute(node=bnd_node, attr='tx')
attr_ty = mmapi.Attribute(node=bnd_node, attr='ty')
attr_tz = mmapi.Attribute(node=bnd_node, attr='tz')
attr_list.append(attr_tx)
attr_list.append(attr_ty)
attr_list.append(attr_tz)
attr_tx = mmapi.Attribute(node=cam_tfm_node, attr='tx')
attr_ty = mmapi.Attribute(node=cam_tfm_node, attr='ty')
attr_tz = mmapi.Attribute(node=cam_tfm_node, attr='tz')
attr_rx = mmapi.Attribute(node=cam_tfm_node, attr='rx')
attr_ry = mmapi.Attribute(node=cam_tfm_node, attr='ry')
attr_rz = mmapi.Attribute(node=cam_tfm_node, attr='rz')
attr_fl = mmapi.Attribute(node=cam_shp_node, attr='focalLength')
attr_list.append(attr_tx)
attr_list.append(attr_ty)
attr_list.append(attr_tz)
attr_list.append(attr_rx)
attr_list.append(attr_ry)
attr_list.append(attr_rz)
attr_list.append(attr_fl)
# Frames
frm_list = []
root_frm_list = []
not_root_frm_list = []
all_frames = range(start_frame, end_frame + 1, 1)
for f in all_frames:
prim = ((float(f) % 20.0) == 0) \
or (f == start_frame) \
or (f == end_frame)
sec = prim is not True
frm = mmapi.Frame(f, primary=prim, secondary=sec)
if prim is True:
root_frm_list.append(frm)
else:
not_root_frm_list.append(frm)
frm_list.append(frm)
# Solvers
sol_list = []
print_stats = False
use_solver_steps = False
if print_stats is True:
# Print statistics
stats_sol = mmapi.SolverStep()
stats_sol.set_verbose(False)
stats_sol.set_frame_list(frm_list)
stats_sol.set_print_statistics_inputs(True)
stats_sol.set_print_statistics_affects(True)
stats_sol.set_print_statistics_deviation(True)
sol_list.append(stats_sol)
if use_solver_steps is True:
# Global solve with every 10th frame (and start/end frames)
sol = mmapi.SolverStep()
sol.set_verbose(False)
sol.set_max_iterations(10)
sol.set_frames_use_tags(['primary'])
sol.set_attributes_use_static(True)
sol.set_attributes_use_animated(True)
sol.set_frame_list(frm_list)
sol_list.append(sol)
# Per-frame solvers
for frm in frm_list:
sol = mmapi.SolverStep()
sol.set_verbose(False)
sol.set_max_iterations(10)
sol.set_frames_use_tags(['primary', 'secondary'])
sol.set_attributes_use_static(False)
sol.set_attributes_use_animated(True)
sol.set_frame_list([frm])
sol_list.append(sol)
else:
# Solver
sol = mmapi.SolverStandard()
sol.set_root_frame_list(root_frm_list)
sol.set_frame_list(not_root_frm_list)
sol.set_only_root_frames(False)
sol.set_global_solve(False)
sol._triangulate_bundles = False
sol._auto_attr_blocks = False
sol.set_use_single_frame(False)
sol_list.append(sol)
# Collection
col = mmapi.Collection()
col.create_node('mySolveCollection')
col.set_solver_list(sol_list)
col.add_marker_list(mkr_list)
col.add_attribute_list(attr_list)
# save the output
path = self.get_data_path('test_solve_stA_refine_before.ma')
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
# Run solver!
LOG.warning('Running Solver Test... (it may take some time to finish).')
results = col.execute()
# Set Deviation
mmapi.update_deviation_on_markers(mkr_list, results)
mmapi.update_deviation_on_collection(col, results)
# save the output
path = self.get_data_path('test_solve_stA_refine_after.ma')
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
self.checkSolveResults(results)
return
def test_per_frame(self):
"""
Solve animated values, per-frame.
"""
# Camera
cam_tfm = maya.cmds.createNode('transform',
name='cam_tfm')
cam_shp = maya.cmds.createNode('camera',
name='cam_shp',
parent=cam_tfm)
maya.cmds.setAttr(cam_tfm + '.tx', -1.0)
maya.cmds.setAttr(cam_tfm + '.ty', 1.0)
maya.cmds.setAttr(cam_tfm + '.tz', -5.0)
cam = mmapi.Camera(shape=cam_shp)
# Bundle
bnd = mmapi.Bundle().create_node()
bundle_tfm = bnd.get_node()
maya.cmds.setAttr(bundle_tfm + '.tx', 5.5)
maya.cmds.setAttr(bundle_tfm + '.ty', 6.4)
maya.cmds.setAttr(bundle_tfm + '.tz', -25.0)
assert mmapi.get_object_type(bundle_tfm) == 'bundle'
maya.cmds.setKeyframe(bundle_tfm,
attribute='translateX',
time=1, value=5.5,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(bundle_tfm,
attribute='translateY',
time=1, value=6.4,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(bundle_tfm,
attribute='translateZ',
time=1, value=-25.0,
inTangentType='linear',
outTangentType='linear')
# Marker
mkr = mmapi.Marker().create_node(cam=cam, bnd=bnd)
marker_tfm = mkr.get_node()
assert mmapi.get_object_type(marker_tfm) == 'marker'
# maya.cmds.setAttr(marker_tfm + '.tx', 0.0)
# maya.cmds.setAttr(marker_tfm + '.ty', 0.0)
maya.cmds.setKeyframe(marker_tfm,
attribute='translateX',
time=1, value=-0.5,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(marker_tfm,
attribute='translateX',
time=5, value=0.5,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(marker_tfm,
attribute='translateY',
time=1, value=-0.5,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(marker_tfm,
attribute='translateY',
time=5, value=0.5,
inTangentType='linear',
outTangentType='linear')
# Attributes
attr_tx = mmapi.Attribute(bundle_tfm + '.tx')
attr_ty = mmapi.Attribute(bundle_tfm + '.ty')
# Frames
frm_list = [
mmapi.Frame(1),
mmapi.Frame(2),
mmapi.Frame(3),
mmapi.Frame(4),
mmapi.Frame(5),
]
# Solver
sol_list = []
for frm in frm_list:
sol = mmapi.Solver()
sol.set_max_iterations(10)
sol.set_verbose(True)
sol.set_frame_list([frm])
sol_list.append(sol)
# Collection
col = mmapi.Collection()
col.create_node('mySolveCollection')
col.add_solver_list(sol_list)
col.add_marker(mkr)
col.add_attribute(attr_tx)
col.add_attribute(attr_ty)
# save the output
path = self.get_data_path('test_solve_per_frame_before.ma')
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
# Run solver!
results = col.execute()
# Set Deviation
mmapi.update_deviation_on_markers([mkr], results)
mmapi.update_deviation_on_collection(col, results)
# save the output
path = self.get_data_path('test_solve_per_frame_after.ma')
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
self.checkSolveResults(results)
return
def test_marker_enable(self):
start = 1
end = 5
# Set Time Range
maya.cmds.playbackOptions(
animationStartTime=start,
minTime=start,
animationEndTime=end,
maxTime=end
)
# Camera
cam_tfm = maya.cmds.createNode('transform',
name='cam_tfm')
cam_shp = maya.cmds.createNode('camera',
name='cam_shp',
parent=cam_tfm)
maya.cmds.setAttr(cam_tfm + '.tx', -1.0)
maya.cmds.setAttr(cam_tfm + '.ty', 1.0)
maya.cmds.setAttr(cam_tfm + '.tz', -5.0)
cam = mmapi.Camera(shape=cam_shp)
# Bundle
bnd = mmapi.Bundle().create_node()
bundle_tfm = bnd.get_node()
maya.cmds.setAttr(bundle_tfm + '.tx', 5.5)
maya.cmds.setAttr(bundle_tfm + '.ty', 6.4)
maya.cmds.setAttr(bundle_tfm + '.tz', -25.0)
assert mmapi.get_object_type(bundle_tfm) == 'bundle'
# calculate angle of view (AOV)
f = maya.cmds.getAttr(cam_shp + '.focalLength')
fbw = maya.cmds.getAttr(cam_shp + '.horizontalFilmAperture') * 25.4
fbh = maya.cmds.getAttr(cam_shp + '.verticalFilmAperture') * 25.4
aov = math.degrees(2.0 * math.atan(fbw * (0.5 / f)))
# Set Camera Anim
maya.cmds.setKeyframe(cam_tfm, attribute='rotateY', time=start, value=-(aov/2),
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(cam_tfm, attribute='rotateY', time=end, value=(aov/2),
inTangentType='linear',
outTangentType='linear')
# Marker
mkr = mmapi.Marker().create_node(cam=cam, bnd=bnd)
marker_tfm = mkr.get_node()
assert mmapi.get_object_type(marker_tfm) == 'marker'
mid_value = 0.23534346
maya.cmds.setKeyframe(marker_tfm, attribute='translateX', time=start, value=-0.5,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(marker_tfm, attribute='translateX', time=start+1,
value=-mid_value,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(marker_tfm, attribute='translateX', time=end-1,
value=mid_value,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(marker_tfm, attribute='translateX', time=end, value=0.5,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setAttr(marker_tfm + '.ty', 0.0)
maya.cmds.setKeyframe(marker_tfm, attribute='enable', time=1, value=1,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(marker_tfm, attribute='enable', time=2, value=1,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(marker_tfm, attribute='enable', time=3, value=0,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(marker_tfm, attribute='enable', time=4, value=1,
inTangentType='linear',
outTangentType='linear')
maya.cmds.setKeyframe(marker_tfm, attribute='enable', time=5, value=1,
inTangentType='linear',
outTangentType='linear')
# Create Sphere
sph_tfm, shp_node = maya.cmds.polySphere()
maya.cmds.setAttr(sph_tfm + '.tx', -1.0)
maya.cmds.setAttr(sph_tfm + '.ty', 1.0)
maya.cmds.setAttr(sph_tfm + '.tz', -25.0)
# Attributes
attr_tx = mmapi.Attribute(bundle_tfm + '.tx')
attr_ty = mmapi.Attribute(bundle_tfm + '.ty')
# Frames
frm_list = [
mmapi.Frame(1, primary=True),
mmapi.Frame(2, primary=True),
mmapi.Frame(3, primary=True),
mmapi.Frame(4, primary=True),
mmapi.Frame(5, primary=True)
]
# Solver
sol = mmapi.Solver()
sol.set_max_iterations(1000)
sol.set_solver_type(mmapi.SOLVER_TYPE_DEFAULT)
sol.set_verbose(True)
sol.set_frame_list(frm_list)
# Collection
col = mmapi.Collection()
col.create_node('mySolveCollection')
col.add_solver(sol)
col.add_marker(mkr)
col.add_attribute(attr_tx)
col.add_attribute(attr_ty)
# save the output
path = self.get_data_path('test_solve_marker_enabled_before.ma')
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
# Run solver!
results = col.execute()
# Ensure the values are correct
for res in results:
success = res.get_success()
err = res.get_final_error()
print('error stats: ' + pprint.pformat(res.get_error_stats()))
print('timer stats: ' + pprint.pformat(res.get_timer_stats()))
print('solver stats: ' + pprint.pformat(res.get_solver_stats()))
print('frame error list: ' + pprint.pformat(dict(res.get_frame_error_list())))
print('marker error list: ' + pprint.pformat(dict(res.get_marker_error_list())))
self.assertTrue(success)
# self.assertGreater(0.001, err)
# assert self.approx_equal(maya.cmds.getAttr(bundle_tfm+'.tx'), -6.0)
# assert self.approx_equal(maya.cmds.getAttr(bundle_tfm+'.ty'), 3.6)
# Set Deviation
mmapi.update_deviation_on_markers([mkr], results)
mmapi.update_deviation_on_collection(col, results)
# save the output
path = self.get_data_path('test_solve_marker_enabled_after.ma')
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
self.checkSolveResults(results)
return
def get_node_default_attributes(nodes):
"""
Get the default attributes for solving on the given nodes.
:param nodes: List of nodes to be considered.
:type nodes: [str, ..]
:returns: List of mmSolver API Attribute objects.
:rtype: [Attribute, ..]
"""
attr_list = []
for node in nodes:
node_type = maya.cmds.nodeType(node)
obj_type = mmapi.get_object_type(node)
attr_names = []
if obj_type == mmapi.OBJECT_TYPE_BUNDLE:
# Default bundle attributes.
attr_names += [
'translateX',
'translateY',
'translateZ',
]
elif obj_type == mmapi.OBJECT_TYPE_CAMERA:
# Camera default attributes, for both transform and
# camera nodes.
if node_type == 'transform':
attr_names += [
'translateX',
'translateY',
'translateZ',
'rotateX',
'rotateY',
'rotateZ',
]
elif node_type == 'camera':
attr_names += [
'focalLength',
]
else:
# Fallback - get all logical attributes.
attrs = maya.cmds.listAttr(
node,
keyable=True,
settable=True,
scalar=True,
shortNames=False,
) or []
attr_types = [
'doubleLinear',
'doubleAngle',
'double',
'float',
]
attr_names += [
n for n in attrs if maya.cmds.attributeQuery(
n, node=node, attributeType=True) in attr_types
]
for attr_name in attr_names:
attr_obj = mmapi.Attribute(node=node, attr=attr_name)
attr_list.append(attr_obj)
return attr_list
def test_stA_refine_good_solve(self):
"""
Test file based on 3DEqualizer 'stA' image sequence.
The Maya file loaded contains a good 3DEqualizer solve. This
test tests the solver to ensure it produces good results,
given an already good solution.
The 'stA' image sequence has a frame range of 0 to 94.
"""
start_frame = 0
end_frame = 94
path = self.get_data_path('scenes', 'stA', 'stA.ma')
ok = maya.cmds.file(path, open=True, ignoreVersion=True, force=True)
assert isinstance(ok, (str, unicode))
# Camera
cam_name = 'stA_1_1Shape1'
cam = mmapi.Camera(shape=cam_name)
cam_tfm_node = cam.get_transform_node()
cam_shp_node = cam.get_shape_node()
# Marker Group
mkr_grp_name = 'markerGroup1'
mkr_grp = mmapi.MarkerGroup(node=mkr_grp_name)
mkr_grp_node = mkr_grp.get_node()
# Markers
mkr_list = []
bnd_list = []
mkr_nodes = maya.cmds.listRelatives(
mkr_grp_node,
children=True,
shapes=False) or []
for node in mkr_nodes:
if node.endswith('_MKR') is False:
continue
assert mmapi.get_object_type(node) == 'marker'
mkr = mmapi.Marker(node=node)
bnd = mkr.get_bundle()
mkr_list.append(mkr)
bnd_list.append(bnd)
assert len(mkr_list) > 0
assert len(bnd_list) > 0
# Attributes
attr_list = []
for bnd in bnd_list:
bnd_node = bnd.get_node()
attr_tx = mmapi.Attribute(node=bnd_node, attr='tx')
attr_ty = mmapi.Attribute(node=bnd_node, attr='ty')
attr_tz = mmapi.Attribute(node=bnd_node, attr='tz')
attr_list.append(attr_tx)
attr_list.append(attr_ty)
attr_list.append(attr_tz)
attr_tx = mmapi.Attribute(node=cam_tfm_node, attr='tx')
attr_ty = mmapi.Attribute(node=cam_tfm_node, attr='ty')
attr_tz = mmapi.Attribute(node=cam_tfm_node, attr='tz')
attr_rx = mmapi.Attribute(node=cam_tfm_node, attr='rx')
attr_ry = mmapi.Attribute(node=cam_tfm_node, attr='ry')
attr_rz = mmapi.Attribute(node=cam_tfm_node, attr='rz')
attr_fl = mmapi.Attribute(node=cam_shp_node, attr='focalLength')
attr_list.append(attr_tx)
attr_list.append(attr_ty)
attr_list.append(attr_tz)
attr_list.append(attr_rx)
attr_list.append(attr_ry)
attr_list.append(attr_rz)
attr_list.append(attr_fl)
# Frames
#
# Root Frames are automatically calculated from the markers.
root_frm_list = []
not_root_frm_list = []
min_frames_per_marker = 2
frame_nums = mmapi.get_root_frames_from_markers(
mkr_list, min_frames_per_marker, start_frame, end_frame)
for f in frame_nums:
frm = mmapi.Frame(f)
root_frm_list.append(frm)
for f in range(start_frame, end_frame + 1):
frm = mmapi.Frame(f)
not_root_frm_list.append(frm)
# Solvers
sol_list = []
sol = mmapi.SolverStandard()
sol.set_root_frame_list(root_frm_list)
sol.set_frame_list(not_root_frm_list)
sol.set_only_root_frames(False)
sol.set_global_solve(False)
sol.set_use_single_frame(False)
sol_list.append(sol)
# Collection
col = mmapi.Collection()
col.create_node('mySolveCollection')
col.set_solver_list(sol_list)
col.add_marker_list(mkr_list)
col.add_attribute_list(attr_list)
# save the output
path = self.get_data_path('test_solve_stA_refine_before.ma')
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
# Run solver!
LOG.warning('Running Solver Test... (it may take some time to finish).')
results = mmapi.execute(col)
# Set Deviation
mmapi.update_deviation_on_markers(mkr_list, results)
mmapi.update_deviation_on_collection(col, results)
# save the output
path = self.get_data_path('test_solve_stA_refine_after.ma')
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
self.checkSolveResults(results)
return
