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 reset_attr_used_hints(col_name, node_attr_list):
import mmSolver.api as mmapi
col = mmapi.Collection(col_name)
for node_attr in node_attr_list:
attr = mmapi.Attribute(name=node_attr)
col.set_attribute_used_hint(attr,
const.ATTRIBUTE_USED_HINT_UNKNOWN_VALUE)
return
def get_selected_maya_attributes():
"""
Get the currently selected attributes from the Channel Box.
This function uses Maya ChannelBox logic to get the objects, which
is based on the selection, but may be doing more complex logic.
:return: List of Attribute objects for all nodes in the Channel Box.
:rtype: list of Attribute
"""
name = _get_channel_box_ui_name()
# Main Nodes and Attribute
main_nodes = maya.cmds.channelBox(name, query=True,
mainObjectList=True) or []
main_attrs = maya.cmds.channelBox(
name, query=True, selectedMainAttributes=True) or []
# Shape Nodes and Attribute
shape_nodes = maya.cmds.channelBox(name, query=True,
shapeObjectList=True) or []
shape_attrs = maya.cmds.channelBox(
name, query=True, selectedShapeAttributes=True) or []
# History Nodes and Attribute
history_nodes = maya.cmds.channelBox(
name, query=True, historyObjectList=True) or []
history_attrs = maya.cmds.channelBox(
name, query=True, selectedHistoryAttributes=True) or []
# Output Nodes and Attribute
output_nodes = maya.cmds.channelBox(
name, query=True, outputObjectList=True) or []
output_attrs = maya.cmds.channelBox(
name, query=True, selectedOutputAttributes=True) or []
attr_list = []
nodes_and_attrs = [
(main_nodes, main_attrs),
(shape_nodes, shape_attrs),
(history_nodes, history_attrs),
(output_nodes, output_attrs),
]
for nodes, attrs in nodes_and_attrs:
for n in nodes:
for a in attrs:
possible_attrs = maya.cmds.listAttr(n, shortNames=False) or []
possible_attrs += maya.cmds.listAttr(n, shortNames=True) or []
if a not in possible_attrs:
continue
attr = mmapi.Attribute(node=n, attr=a)
if attr.get_name() is not None:
attr_list.append(attr)
return attr_list
def _runAndSetUsedSolveObjects(col_name, *args, **kwargs):
"""
Call mmSolver, find (un)used objects, then set values for UIs to look up.
:param col_name:
:param args:
:param kwargs:
:return:
"""
# Generate mmSolver command.
import maya.cmds
solver_args = args
solver_kwargs = kwargs.copy()
del solver_kwargs['mode']
solver_kwargs['frame'] = [1] # 'mmSolver' must have a frame value
solver_kwargs['printStatistics'] = ['usedSolveObjects']
data = maya.cmds.mmSolver(*solver_args, **solver_kwargs)
markers_used = _parse_usage_list('markers_used=', data)
markers_unused = _parse_usage_list('markers_unused=', data)
attributes_used = _parse_usage_list('attributes_used=', data)
attributes_unused = _parse_usage_list('attributes_unused=', data)
# Set usage value attributes on Markers and Attributes.
import mmSolver.api as mmapi
for mkr_node in markers_used:
mkr = mmapi.Marker(node=mkr_node)
mkr.set_used_hint(const.MARKER_USED_HINT_USED_VALUE)
for mkr_node in markers_unused:
mkr = mmapi.Marker(node=mkr_node)
mkr.set_used_hint(const.MARKER_USED_HINT_NOT_USED_VALUE)
col = mmapi.Collection(col_name)
for node_attr in attributes_used:
attr = mmapi.Attribute(name=node_attr)
col.set_attribute_used_hint(attr, const.ATTRIBUTE_USED_HINT_USED_VALUE)
for node_attr in attributes_unused:
attr = mmapi.Attribute(name=node_attr)
col.set_attribute_used_hint(attr,
const.ATTRIBUTE_USED_HINT_NOT_USED_VALUE)
return
def test_opera_house(self):
"""
Load pre-tracked Markers for the 'opera house' image sequence,
then solve it.
http://danielwedge.com/fmatrix/operahouse.html
"""
start = 0
end = 41
# 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 + '.rotateOrder', 2) # zxy
maya.cmds.setAttr(cam_shp + '.focalLength', 14)
maya.cmds.setAttr(cam_shp + '.horizontalFilmAperture', 5.4187 / 25.4)
maya.cmds.setAttr(cam_shp + '.verticalFilmAperture', 4.0640 / 25.4)
cam = mmapi.Camera(shape=cam_shp)
# Set Camera Keyframes
cam_data = {
'0': (-0.19889791581420663, 0.5591321634949238, 7.258789219735233, -1.9999507874015703, -0.3999999999999992, 0.0),
'22': (-4.840404384215566, 0.7543627646977502, 6.3465857678271425, -3.0709513272069815, -36.91024116734281, 0.0),
'41': (-8.584368967987194, 0.6990718939718145, 5.508167213044364, -1.4738793091011815, -54.30997787050599, 0.0)
}
for key in sorted(cam_data.keys()):
frame = int(key)
for i, attr in enumerate(['tx', 'ty', 'tz', 'rx', 'ry', 'rz']):
value = cam_data[key][i]
maya.cmds.setKeyframe(cam_tfm, attribute=attr, time=frame, value=value)
maya.cmds.setKeyframe(cam_shp, attribute='focalLength', time=start, value=14.0)
maya.cmds.setKeyframe(cam_shp, attribute='focalLength', time=end, value=14.0)
# Create image plane
path = self.get_data_path('operahouse', 'frame00.jpg')
imgpl = maya.cmds.imagePlane(
camera=cam_shp,
fileName=path
)
maya.cmds.setAttr(imgpl[1] + '.useFrameExtension', 1)
maya.cmds.setAttr(imgpl[1] + '.depth', 2000)
maya.cmds.setAttr(imgpl[1] + '.frameCache', 0)
maya.cmds.setAttr(imgpl[1] + '.coverageX', 3072)
maya.cmds.setAttr(imgpl[1] + '.coverageY', 2304)
# Create Horizon Line
cir = maya.cmds.circle(name='horizon', nrx=0, nry=1, nrz=0)
maya.cmds.setAttr(cir[1] + ".radius", 3000)
# Create Cube for Opera House
cube = maya.cmds.polyCube()
maya.cmds.setAttr(cube[0] + ".ty", 0.5)
maya.cmds.setAttr(cube[0] + ".sx", 2.68)
maya.cmds.setAttr(cube[0] + ".sy", 0.91625416)
maya.cmds.setAttr(cube[0] + ".sz", 1.68658365)
# Marker Group
mkr_grp = mmapi.MarkerGroup().create_node(cam=cam)
mkr_grp_node = mkr_grp.get_node()
# Bundle Group
bnd_grp = maya.cmds.createNode('transform', name='bundleGroup')
bnd_fg_grp = maya.cmds.createNode(
'transform',
name='bundles_fg',
parent=bnd_grp)
bnd_bg_grp = maya.cmds.createNode(
'transform',
name='bundles_bg',
parent=bnd_grp)
# Load Markers
fg_points = [
'Track_01',
'Track_02',
'Track_05',
'Track_06',
'Track_08',
'Track_09',
'Track_10',
'Track_11',
'Track_19',
'Track_20',
'Track_21',
'Track_22',
'Track_23',
'Track_23',
'Track_24',
'Track_25',
'Track_26',
'Track_27',
]
bg_points = [
'Track_03',
'Track_04',
'Track_07',
'Track_12',
'Track_13',
'Track_14',
'Track_15',
'Track_16',
'Track_17',
'Track_18',
'Track_28',
'Track_29',
'Track_30',
'Track_31',
]
bnd_positions = {
'Track_23': (-0.7669678476654883, 0.704741253611808, 0.11480582185051777),
'Track_14': (-6.096859889443822, 2.0552736121532478, -64.25806442305448),
'Track_12': (45.11056705173852, 2.602519222901666, -43.16772737415769),
'Track_13': (-11.331222134074189, -0.9161249928992397, -63.60343691220178),
'Track_28': (12.97847320083373, 0.4908757961951475, -6.558878377403925),
'Track_24': (-0.9577362080844809, 0.11947272894636578, -0.29860515939718035),
'Track_25': (-0.3816240705349317, 0.09511793539283707, 0.5968218516602972),
'Track_05': (-0.5497538933513093, 0.9121450956455763, 0.0689419211208016),
'Track_06': (0.6442115545215732, 0.09146863102772763, 0.2698159600733472),
'Track_02': (-1.1928085448379213, 0.06849164070024401, 0.741609523996595),
'Track_17': (4.101733117764308, 0.4416977194116366, -20.775735845844235),
'Track_16': (10.499779696104385, 2.4959245952203037, -61.65315035391216),
'Track_21': (0.4422885021421483, 0.15594114410956195, -0.4586671394741284),
'Track_18': (13.426726902476766, 2.208127581689255, -62.440721369338476),
'Track_27': (-1.203371663768503, 0.07727436882970459, -0.34432924439358475),
'Track_07': (24.82344439444535, 3.8981611004590917, -62.57148439047777),
'Track_26': (-1.036542158437551, 0.1301250303434169, 0.6183349238312523),
'Track_11': (-1.2868698932117608, 0.07508027422294668, -0.6923287330737453),
'Track_09': (-1.1210978513200678, -0.0009538700668097195, -0.7481409812887209),
'Track_20': (0.5370453995103619, 0.32144750391315535, 0.10037404391850258),
'Track_08': (-0.35711469535141427, 0.8134673956410489, -0.8873816770491396),
'Track_19': (-1.0708190128497155, 0.5849715587489718, 0.22909459498373133),
'Track_10': (-0.8256010837265352, 0.04548785302325305, -0.6865934949556973),
'Track_30': (12.219883964568602, 1.6676763053004873, -63.511794156133575),
'Track_22': (-0.42435005852350927, 0.6386843510112235, -1.0271747982989685),
'Track_31': (14.4768210901898, 1.5761955139450978, -40.10088917167338),
'Track_15': (-0.17540615158899264, 2.5048877383268424, -64.10912011449136),
'Track_29': (15.264518808431728, 1.8337698745022983, -62.076762425418536),
'Track_03': (311.42375656555913, 16.402469194090923, -179.38329132993437),
'Track_01': (-1.0890118590423876, 0.5109764471108498, -0.707187214616633),
'Track_04': (209.73939576288353, 12.878819985707446, -150.30617721944793)
}
mkr_fg_grp = maya.cmds.createNode('transform',
name='fg',
parent=mkr_grp_node)
mkr_bg_grp = maya.cmds.createNode('transform',
name='bg',
parent=mkr_grp_node)
path = self.get_data_path('match_mover', 'loadmarker.rz2')
_, mkr_data_list = marker_read.read(path)
print 'mkr_data_list', mkr_data_list
mkr_list = marker_read.create_nodes(
mkr_data_list,
cam=cam,
mkr_grp=mkr_grp
)
mkr_fg_list = []
mkr_bg_list = []
for mkr in mkr_list:
mkr_node = mkr.get_node()
mgrp = mkr_grp_node
bgrp = bnd_grp
pos = None
for name in fg_points:
if name in mkr_node:
if name in bnd_positions:
pos = bnd_positions[name]
mgrp = mkr_fg_grp
bgrp = bnd_fg_grp
mkr_fg_list.append(mkr)
break
for name in bg_points:
if name in mkr_node:
if name in bnd_positions:
pos = bnd_positions[name]
mgrp = mkr_bg_grp
bgrp = bnd_bg_grp
mkr_bg_list.append(mkr)
break
maya.cmds.parent(mkr_node, mgrp, relative=True)
bnd = mkr.get_bundle()
bnd_node = bnd.get_node()
plug_tx = bnd_node + '.tx'
plug_ty = bnd_node + '.ty'
plug_tz = bnd_node + '.tz'
maya.cmds.setAttr(plug_tx, pos[0])
maya.cmds.setAttr(plug_ty, pos[1])
maya.cmds.setAttr(plug_tz, pos[2])
maya.cmds.parent(bnd_node, bgrp, relative=True)
# bnd_node = bnd.get_node()
# plug_tx = bnd_node + '.tx'
# plug_ty = bnd_node + '.ty'
# plug_tz = bnd_node + '.tz'
# maya.cmds.setAttr(plug_tx, lock=True)
# maya.cmds.setAttr(plug_ty, lock=True)
# maya.cmds.setAttr(plug_tz, lock=True)
# Frames
# prim = [0, 22, 41]
# sec = [3, 8, 12, 27, 33, 38]
# prim = [0, 3, 8, 12, 22, 27, 33, 38, 41]
frm_list = []
frm = mmapi.Frame(0, tags=['primary', '1', 'single001'])
frm_list.append(frm)
frm = mmapi.Frame(3, tags=['primary', '1', '2', 'single002'])
frm_list.append(frm)
frm = mmapi.Frame(8, tags=['primary', '2', '3', 'single003'])
frm_list.append(frm)
frm = mmapi.Frame(12, tags=['primary', '3', '4', 'single004'])
frm_list.append(frm)
frm = mmapi.Frame(22, tags=['primary', '4', '5', 'single005'])
frm_list.append(frm)
frm = mmapi.Frame(27, tags=['primary', '5', '6', 'single006'])
frm_list.append(frm)
frm = mmapi.Frame(33, tags=['primary', '6', '7', 'single007'])
frm_list.append(frm)
frm = mmapi.Frame(38, tags=['primary', '7', '8', 'single008'])
frm_list.append(frm)
frm = mmapi.Frame(41, tags=['primary', '8', 'single009'])
frm_list.append(frm)
root_frm_list = []
not_root_frm_list = []
for f in [0, 3, 8, 12, 16, 22, 27, 33, 38, 41]:
frm = mmapi.Frame(f)
root_frm_list.append(frm)
for f in range(0, 41):
frm = mmapi.Frame(f)
not_root_frm_list.append(frm)
sol_list = []
sol = mmapi.SolverStandard()
sol.set_single_frame(False)
sol.set_root_frame_list(root_frm_list)
sol.set_frame_list(not_root_frm_list)
sol.set_root_frame_strategy(mmapi.ROOT_FRAME_STRATEGY_GLOBAL_VALUE)
sol.set_only_root_frames(False)
sol.set_global_solve(False)
sol._robust_loss_type = 1
sol._robust_loss_scale = 10.0
sol._auto_attr_blocks = True
sol._triangulate_bundles = False
sol_list.append(sol)
# Collection
col = mmapi.Collection()
col.create_node('mySolverCollection')
col.add_solver_list(sol_list)
# Add markers
col.add_marker_list(mkr_fg_list)
col.add_marker_list(mkr_bg_list)
# Attributes
attr_cam_tx = mmapi.Attribute(cam_tfm + '.tx')
attr_cam_ty = mmapi.Attribute(cam_tfm + '.ty')
attr_cam_tz = mmapi.Attribute(cam_tfm + '.tz')
attr_cam_rx = mmapi.Attribute(cam_tfm + '.rx')
attr_cam_ry = mmapi.Attribute(cam_tfm + '.ry')
attr_cam_rz = mmapi.Attribute(cam_tfm + '.rz')
attr_cam_focal = mmapi.Attribute(cam_shp + '.focalLength')
col.add_attribute(attr_cam_tx)
col.add_attribute(attr_cam_ty)
col.add_attribute(attr_cam_tz)
col.add_attribute(attr_cam_rx)
col.add_attribute(attr_cam_ry)
col.add_attribute(attr_cam_rz)
mkr_list = col.get_marker_list()
for mkr in mkr_list:
bnd = mkr.get_bundle()
bnd_node = bnd.get_node()
attr_tx = mmapi.Attribute(bnd_node + '.tx')
attr_ty = mmapi.Attribute(bnd_node + '.ty')
attr_tz = mmapi.Attribute(bnd_node + '.tz')
col.add_attribute(attr_tx)
col.add_attribute(attr_ty)
col.add_attribute(attr_tz)
# save the output
path = self.get_data_path('test_solve_opera_house_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 'err', err, 'success', success
# 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_opera_house_after.ma')
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
self.checkSolveResults(results)
return
def test_solveAllFramesCausesStaticAnimCurves(self):
"""
Solving with the scene file 'mmSolverBasicSolveB_before.ma', was
reported to solve as static values, the same as the initial
values. The DG did not evaluate some how and the solve was
therefore useless.
GitHub Issue #53.
"""
s = time.time()
# Open the Maya file
file_name = 'mmSolverBasicSolveB_before.ma'
path = self.get_data_path('scenes', file_name)
maya.cmds.file(path, open=True, force=True, ignoreVersion=True)
# NOTE: We leave these nodes alone, since these are already in
# the 'correct' position, we are treating these as surveyed.
# When we have less than 3 points as survey the solve goes
# crazy.
dont_touch_these_nodes = [
'|bundle_12_BND',
'|bundle_13_BND',
'|bundle_14_BND']
# Triangulate all 3D points.
nodes = maya.cmds.ls(type='transform') or []
bnd_nodes = mmapi.filter_bundle_nodes(nodes)
bnd_list = [mmapi.Bundle(node=n) for n in bnd_nodes]
for bnd in bnd_list:
bnd_node = bnd.get_node()
if bnd_node in dont_touch_these_nodes:
continue
attrs = ['translateX', 'translateY', 'translateZ']
for attr_name in attrs:
plug = bnd_node + '.' + attr_name
maya.cmds.setAttr(plug, lock=False)
maya.cmds.setAttr(plug, 0.0)
# Get Bundle attributes to compute.
bnd_attr_list = []
for bnd in bnd_list:
node = bnd.get_node()
attrs = ['translateX', 'translateY', 'translateZ']
for attr_name in attrs:
attr = mmapi.Attribute(node=node, attr=attr_name)
bnd_attr_list.append(attr)
# Camera attributes
cam_tfm = 'stA_1_1'
cam = mmapi.Camera(cam_tfm)
cam_shp = cam.get_shape_node()
cam_attr_list = []
attrs = ['translateX', 'translateY', 'translateZ',
'rotateX', 'rotateY', 'rotateZ']
for attr_name in attrs:
attr = mmapi.Attribute(node=cam_tfm, attr=attr_name)
cam_attr_list.append(attr)
attr = mmapi.Attribute(node=cam_shp, attr='focalLength')
cam_attr_list.append(attr)
# Frame List
root_frm_list = []
not_root_frm_list = []
f_list = [14, 35, 50, 85]
for f in f_list:
frm = mmapi.Frame(f)
root_frm_list.append(frm)
for f in range(0, 94):
frm = mmapi.Frame(f)
not_root_frm_list.append(frm)
# Run solver!
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._auto_attr_blocks = True
sol._triangulate_bundles = False
sol.set_single_frame(False)
sol.set_root_frame_strategy(mmapi.ROOT_FRAME_STRATEGY_GLOBAL_VALUE)
# sol._robust_loss_type = mmapi.ROBUST_LOSS_TYPE_TRIVIAL_VALUE
# sol._robust_loss_scale = 1.0
sol_list.append(sol)
col = mmapi.Collection(node='collection1')
col.set_attribute_list(cam_attr_list + bnd_attr_list)
col.set_solver_list(sol_list)
e = time.time()
print 'pre=solve time:', e - s
s = time.time()
solres_list = col.execute()
e = time.time()
print 'total time:', e - s
# Set Deviation
mkr_list = col.get_marker_list()
mmapi.update_deviation_on_markers(mkr_list, solres_list)
mmapi.update_deviation_on_collection(col, solres_list)
# save the output
name = 'test_solve_solveAllFramesCausesStaticAnimCurves_after.ma'
path = self.get_data_path(name)
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
self.checkSolveResults(solres_list)
return
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 test_find_marker_attr_mapping(self):
# top level transform
root = maya.cmds.createNode('transform', name='top1')
# Camera A, don't parent under the root.
cam_tfm_a = maya.cmds.createNode('transform', name='camA_tfm')
cam_shp_a = maya.cmds.createNode('camera',
name='camA_shp',
parent=cam_tfm_a)
maya.cmds.setAttr(cam_tfm_a + '.tx', -1.0)
maya.cmds.setAttr(cam_tfm_a + '.ty', 1.0)
maya.cmds.setAttr(cam_tfm_a + '.tz', -5.0)
cam_a = mmapi.Camera(shape=cam_shp_a)
# Camera B, parent under the root
cam_tfm_b = maya.cmds.createNode('transform',
name='camB_tfm',
parent=root)
cam_shp_b = maya.cmds.createNode('camera',
name='camB_shp',
parent=cam_tfm_b)
maya.cmds.setAttr(cam_tfm_b + '.tx', 1.0)
maya.cmds.setAttr(cam_tfm_b + '.ty', 1.0)
maya.cmds.setAttr(cam_tfm_b + '.tz', -5.0)
cam_b = mmapi.Camera(shape=cam_shp_b)
# Hierarchy structure
dummy = maya.cmds.createNode('transform', name='dummy')
dummy = maya.cmds.ls(dummy, long=True)[0]
nothing = maya.cmds.createNode('transform', name='nothing')
nothing = maya.cmds.ls(nothing, long=True)[0]
top = maya.cmds.createNode('transform', name='top2')
top = maya.cmds.ls(top, long=True)[0]
multDivide = maya.cmds.createNode('multiplyDivide', name='multDiv')
multDivide2 = maya.cmds.createNode('multiplyDivide',
name='multDivNotValid')
child1 = maya.cmds.createNode('transform', name='child1', parent=top)
child2 = maya.cmds.createNode('transform',
name='child2',
parent=child1)
child1 = maya.cmds.ls(child1, long=True)[0]
child2 = maya.cmds.ls(child2, long=True)[0]
# Set up animation and connections
maya.cmds.setKeyframe(dummy, at='tx', t=1, v=1)
maya.cmds.setKeyframe(dummy, at='tx', t=10, v=10)
maya.cmds.connectAttr(dummy + '.tx', child2 + '.ty')
maya.cmds.connectAttr(dummy + '.tx', child1 + '.tz')
maya.cmds.connectAttr(dummy + '.ty', top + '.tx')
maya.cmds.connectAttr(dummy + '.ry', child2 + '.ry')
maya.cmds.connectAttr(multDivide + '.outputX', dummy + '.ty')
maya.cmds.connectAttr(multDivide2 + '.outputX', dummy + '.tz')
maya.cmds.setKeyframe(child2, at='tz', t=1, v=1)
maya.cmds.setKeyframe(child2, at='tz', t=10, v=10)
maya.cmds.setKeyframe(top, at='ty', t=1, v=1)
maya.cmds.setKeyframe(top, at='ty', t=10, v=10)
maya.cmds.setKeyframe(top, at='tz', t=1, v=1)
maya.cmds.setKeyframe(top, at='tz', t=10, v=10)
maya.cmds.setKeyframe(top, at='ry', t=1, v=-90)
maya.cmds.setKeyframe(top, at='ry', t=10, v=45)
# Bundle
bnd = mmapi.Bundle().create_node()
bundle_tfm = bnd.get_node()
maya.cmds.setKeyframe(bundle_tfm, at='tx', t=1, v=-1)
maya.cmds.setKeyframe(bundle_tfm, at='tx', t=10, v=1)
maya.cmds.parent(bundle_tfm, child2)
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)
# Marker A
mkr_a = mmapi.Marker().create_node(cam=cam_a, bnd=bnd)
marker_tfm = mkr_a.get_node()
maya.cmds.setAttr(marker_tfm + '.tx', 0.0)
maya.cmds.setAttr(marker_tfm + '.ty', 0.0)
# Marker B
mkr_b = mmapi.Marker().create_node(cam=cam_b, bnd=bnd)
marker_tfm = mkr_b.get_node()
maya.cmds.setAttr(marker_tfm + '.tx', 0.0)
maya.cmds.setAttr(marker_tfm + '.ty', 0.0)
mkr_list = [mkr_a, mkr_b]
# Attributes
attr_tx = mmapi.Attribute(bundle_tfm + '.tx')
attr_ty = mmapi.Attribute(bundle_tfm + '.ty')
attr_top_tz = mmapi.Attribute(top + '.tz')
attr_top_rx = mmapi.Attribute(top + '.rx')
attr_child1_ty = mmapi.Attribute(child1 + '.ty')
attr_child1_rx = mmapi.Attribute(child1 + '.rx')
attr_root_ty = mmapi.Attribute(root + '.ty')
attr_cam_a_ty = mmapi.Attribute(cam_tfm_a + '.ty')
attr_cam_a_focal = mmapi.Attribute(cam_shp_a + '.focalLength')
attr_nothing_tx = mmapi.Attribute(nothing + '.tx')
attr_list = [
attr_tx,
attr_ty,
attr_top_tz,
attr_top_rx,
attr_child1_ty,
attr_child1_rx,
attr_root_ty,
attr_cam_a_ty,
attr_cam_a_focal,
attr_nothing_tx,
]
# Test getting affected plugs from a base transform node.
bnd_node = bnd.get_node()
ret = affects_utils.find_plugs_affecting_transform(bnd_node, None)
assert (multDivide + '.input1X') in ret
assert (dummy + '.translateX') in ret
assert (dummy + '.rotateY') in ret
assert (dummy + '.translateY') not in ret
for r in ret:
assert multDivide2 not in r
print 'len(ret):', len(ret)
assert len(ret) == 103
# Test getting the affect mapping between markers and attrs.
ret = mmapi.find_marker_attr_mapping(mkr_list, attr_list)
expected = [[
True, True, True, True, True, True, False, True, True, False
], [True, True, True, True, True, True, True, False, False, False]]
assert ret == expected
# Save the output
path = self.get_data_path('find_marker_attr_mapping_after.ma')
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
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_solveAllFramesCausesStaticAnimCurves(self):
"""
Solving with the scene file 'mmSolverBasicSolveB_before.ma', was
reported to solve as static values, the same as the initial
values. The DG did not evaluate some how and the solve was
therefore useless.
GitHub Issue #53.
"""
# Open the Maya file
file_name = 'mmSolverBasicSolveB_before.ma'
path = self.get_data_path('scenes', file_name)
maya.cmds.file(path, open=True, force=True, ignoreVersion=True)
# NOTE: We leave these nodes along, since these are already in
# the 'correct' position, we are treating these as surveyed.
# When we have less than 3 points as survey the solve goes
# crazy.
dont_touch_these_nodes = [
'|bundle_12_BND', '|bundle_13_BND', '|bundle_14_BND'
]
# Triangulate all 3D points.
nodes = maya.cmds.ls(type='transform') or []
bnd_nodes = mmapi.filter_bundle_nodes(nodes)
bnd_list = [mmapi.Bundle(node=n) for n in bnd_nodes]
for bnd in bnd_list:
bnd_node = bnd.get_node()
if bnd_node in dont_touch_these_nodes:
continue
attrs = ['translateX', 'translateY', 'translateZ']
for attr_name in attrs:
plug = bnd_node + '.' + attr_name
maya.cmds.setAttr(plug, lock=False)
lib_triangulate.triangulate_bundle(bnd)
# Get Bundle attributes to compute.
bnd_attr_list = []
for bnd in bnd_list:
node = bnd.get_node()
attrs = ['translateX', 'translateY', 'translateZ']
for attr_name in attrs:
attr = mmapi.Attribute(node=node, attr=attr_name)
bnd_attr_list.append(attr)
# Camera attributes
cam_tfm = 'stA_1_1'
cam_attr_list = []
attrs = [
'translateX', 'translateY', 'translateZ', 'rotateX', 'rotateY',
'rotateZ'
]
for attr_name in attrs:
attr = mmapi.Attribute(node=cam_tfm, attr=attr_name)
cam_attr_list.append(attr)
# Run solver!
s = time.time()
# Solve camera transform based on triangulated bundle
# positions.
col = mmapi.Collection(node='collection1')
col.set_attribute_list(cam_attr_list)
lib_col.compile_collection(col)
solres_list = col.execute()
print 'time (solve #1):', time.time() - s
# Refine the bundle positions only
s2 = time.time()
col.set_attribute_list(bnd_attr_list)
lib_col.compile_collection(col)
solres_list = col.execute()
print 'time (solve #2):', time.time() - s2
# # Solve both camera transform and bundle positions together.
# s3 = time.time()
# col.set_attribute_list(cam_attr_list + bnd_attr_list)
# lib_col.compile_collection(col)
# solres_list = col.execute()
e = time.time()
# print 'time (solve #3):', e - s3
print 'total time:', e - s
# save the output
name = 'test_solve_solveAllFramesCausesStaticAnimCurves_after.ma'
path = self.get_data_path(name)
maya.cmds.file(rename=path)
maya.cmds.file(save=True, type='mayaAscii', force=True)
self.checkSolveResults(solres_list)
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
#
# 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
