def get_markers_from_selection():
"""
Given a selection of nodes, find the associated markers.
:return: list of Marker objects.
"""
nodes = maya.cmds.ls(long=True, selection=True) or []
node_categories = filter_nodes.get_nodes(nodes)
marker_nodes = node_categories.get('marker', [])
camera_nodes = node_categories.get('camera', [])
for node in camera_nodes:
node_type = maya.cmds.nodeType(node)
cam = None
if node_type == 'transform':
cam = mmapi.Camera(transform=node)
if node_type == 'camera':
cam = mmapi.Camera(shape=node)
tfm_node = cam.get_transform_node()
below_nodes = maya.cmds.ls(tfm_node, dag=True, long=True)
marker_nodes += filter_nodes.get_marker_nodes(below_nodes)
marker_group_nodes = list(node_categories['markergroup'])
for node in marker_group_nodes:
below_nodes = maya.cmds.ls(node, dag=True, long=True)
marker_nodes += filter_nodes.get_marker_nodes(below_nodes)
# Convert nodes into Marker objects.
marker_nodes = list(set(marker_nodes))
marker_list = []
for node in marker_nodes:
mkr = mmapi.Marker(node=node)
marker_list.append(mkr)
return marker_list
def _get_active_viewport_camera():
"""Get the active viewport camera."""
cam = None
model_editor = utils_viewport.get_active_model_editor()
if model_editor is None:
msg = 'Please select an active 3D viewport.'
LOG.warning(msg)
return cam
cam_tfm, cam_shp = utils_viewport.get_viewport_camera(model_editor)
node = cam_shp
if node is None:
msg = 'Please select an active viewport to get a camera.'
LOG.error(msg)
return cam
if camera_utils.is_startup_cam(node) is True:
msg = "Cannot create Markers in 'persp' camera."
LOG.error(msg)
return cam
if maya.cmds.nodeType(node) == 'transform':
cam = mmapi.Camera(transform=node)
elif maya.cmds.nodeType(node) == 'camera':
cam = mmapi.Camera(shape=node)
else:
LOG.error('Camera node is invalid; %r', node)
return cam
return cam
def test_get_root_frames_from_markers_1(self):
# Time Range
start = 1001
end = 1101
maya.cmds.playbackOptions(animationStartTime=start,
minTime=start,
animationEndTime=end,
maxTime=end)
# Create Camera and Marker Group
cam_tfm = maya.cmds.createNode('transform', name='cam_tfm')
cam_shp = maya.cmds.createNode('camera',
name='cam_shp',
parent=cam_tfm)
cam = mmapi.Camera(shape=cam_shp)
mkr_grp = mmapi.MarkerGroup().create_node(cam=cam)
# Marker A
mkr_a = mmapi.Marker().create_node()
mkr_a_node = mkr_a.get_node()
times = [1000, 1001, 1101, 1102]
values = [0, 1, 1, 0]
anim_utils.create_anim_curve_node_apione(times, values,
mkr_a_node + '.enable')
# Calculate Root Frames
min_frames_per_marker = 2
mkr_list = [mkr_a]
start_frame, end_frame = time_utils.get_maya_timeline_range_inner()
frame_nums = mod.get_root_frames_from_markers(mkr_list,
min_frames_per_marker,
start_frame, end_frame)
print 'frames:', frame_nums
return
def _find_camera_from_selection(sel):
cam = None
filtered_nodes = mmapi.filter_nodes_into_categories(sel)
cam_nodes = filtered_nodes['camera']
mkr_grp_nodes = filtered_nodes['markergroup']
mkr_nodes = filtered_nodes['marker']
# Check selected camera.
if len(cam_nodes) > 0:
cam_node = cam_nodes[0]
cam_tfm, cam_shp = camera_utils.get_camera(cam_node)
if cam_tfm is not None and cam_shp is not None:
cam = mmapi.Camera(transform=cam_tfm, shape=cam_shp)
if cam is not None and cam.is_valid():
return cam
# Check selected marker group.
if len(mkr_grp_nodes) > 0:
mkr_grp_node = mkr_grp_nodes[0]
mkr_grp = mmapi.MarkerGroup(node=mkr_grp_node)
cam = mkr_grp.get_camera()
if cam is not None and cam.is_valid():
return cam
# Check selected marker.
if len(mkr_nodes) > 0:
mkr_node = mkr_nodes[0]
mkr = mmapi.Marker(node=mkr_node)
cam = mkr.get_camera()
if cam is not None and cam.is_valid():
return cam
return None
def main():
"""
Convert all selected transforms into 2D markers under a camera.
"""
# Get camera
model_editor = utils_viewport.get_active_model_editor()
if model_editor is None:
msg = 'Please select an active 3D viewport.'
LOG.warning(msg)
return
cam_tfm, cam_shp = utils_viewport.get_viewport_camera(model_editor)
if cam_shp is None:
LOG.error('Please select an active viewport to get a camera.')
return
if utils_camera.is_startup_cam(cam_shp) is True:
LOG.error("Cannot create Markers in 'persp' camera.")
return
# Get transforms
nodes = maya.cmds.ls(
selection=True,
long=True,
type='transform',
) or []
if len(nodes) == 0:
LOG.warning('Please select one or more transform nodes.')
return
mmapi.load_plugin()
try:
# Turn off Maya UI
maya.mel.eval('paneLayout -e -manage false $gMainPane')
# Compute the Marker Data.
start_frame, end_frame = utils_time.get_maya_timeline_range_outer()
mkr_data_list = __convert_nodes_to_marker_data_list(
cam_tfm,
cam_shp,
nodes,
start_frame,
end_frame,
)
cam = mmapi.Camera(shape=cam_shp)
mkr_list = mayareadfile.create_nodes(
mkr_data_list,
cam=cam,
mkr_grp=None,
with_bundles=True,
)
mkr_nodes = [mkr.get_node() for mkr in mkr_list]
except:
raise
finally:
# Turn on Maya UI
maya.mel.eval('paneLayout -e -manage true $gMainPane')
if len(mkr_nodes) > 0:
maya.cmds.select(mkr_nodes, replace=True)
return
def get_selected_cameras():
"""
Return the (associated) Camera objects from the selection.
:returns: Camera objects.
:rtype: mmSolver.api.Camera
"""
cams = []
nodes = maya.cmds.ls(sl=True, long=True) or []
added_cameras = []
objects = mmapi.filter_nodes_into_categories(nodes)
for node in objects['camera']:
cam = None
if maya.cmds.nodeType(node) == 'camera':
cam = mmapi.Camera(shape=node)
else:
cam = mmapi.Camera(transform=node)
if cam is None:
continue
shp_node = cam.get_shape_node()
if shp_node not in added_cameras:
cams.append(cam)
added_cameras.append(shp_node)
for node in objects['marker']:
mkr = mmapi.Marker(node=node)
cam = mkr.get_camera()
if cam is None:
continue
shp_node = cam.get_shape_node()
if shp_node not in added_cameras:
cams.append(cam)
added_cameras.append(shp_node)
for node in objects['markergroup']:
mkr_grp = mmapi.MarkerGroup(node=node)
cam = mkr_grp.get_camera()
if cam is None:
continue
shp_node = cam.get_shape_node()
if shp_node not in added_cameras:
cams.append(cam)
added_cameras.append(shp_node)
return cams
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 create_new_camera():
name = 'camera'
cam_tfm = maya.cmds.createNode('transform', name=name)
cam_tfm = mmapi.get_long_name(cam_tfm)
cam_shp = maya.cmds.createNode('camera',
name=name + 'Shape',
parent=cam_tfm)
cam_shp = mmapi.get_long_name(cam_shp)
cam = mmapi.Camera(transform=cam_tfm, shape=cam_shp)
return cam
def _get_camera_for_update(sel):
cam = None
if len(sel) == 0:
# Get active viewport camera.
model_editor = viewport_utils.get_active_model_editor()
if model_editor is not None:
cam_tfm, cam_shp = viewport_utils.get_viewport_camera(model_editor)
if cam_tfm is not None and cam_shp is not None:
cam = mmapi.Camera(transform=cam_tfm, shape=cam_shp)
if cam is not None and cam.is_valid():
return cam
else:
cam = _find_camera_from_selection(sel)
return cam
def create_new_camera():
"""
Create a new camera nodes and object.
:returns: Camera object.
:rtype: Camera
"""
name = 'camera'
cam_tfm = maya.cmds.createNode('transform', name=name)
cam_tfm = node_utils.get_long_name(cam_tfm)
cam_shp = maya.cmds.createNode('camera',
name=name + 'Shape',
parent=cam_tfm)
cam_shp = node_utils.get_long_name(cam_shp)
cam = mmapi.Camera(transform=cam_tfm, shape=cam_shp)
return cam
def get_cameras():
"""
Get all cameras in the scene file.
Start-up cameras are ignored ('persp', 'top', 'side', 'front', etc)
:rtype: list of mmSolver.api.Camera
"""
nodes = maya.cmds.ls(type='camera', long=True) or []
cam_nodes = mmapi.filter_camera_nodes(nodes)
cams = []
for node in cam_nodes:
startup = maya.cmds.camera(node, query=True, startupCamera=True)
if startup is True:
continue
cam = mmapi.Camera(shape=node)
cams.append(cam)
return cams
def convert_to_marker():
"""
Center the selected transform onto the camera view.
"""
# Get camera
cam_shp = __get_camera()
if cam_shp is None:
LOG.warning('Please select an active viewport to get a camera.')
return
cam_tfm = maya.cmds.listRelatives(cam_shp, parent=True)[0]
# Get transforms
nodes = maya.cmds.ls(
selection=True,
long=True,
type='transform',
) or []
if len(nodes) == 0:
LOG.warning('Please select one or more transform nodes.')
return
mmapi.load_plugin()
try:
# Turn off Maya UI
maya.mel.eval('paneLayout -e -manage false $gMainPane')
# Compute the Marker Data.
start_frame, end_frame = __get_timeline_range_inner()
mkr_data_list = __convert_nodes_to_marker_data_list(
cam_tfm, cam_shp, nodes, start_frame, end_frame)
cam = mmapi.Camera(shape=cam_shp)
mayareadfile.create_nodes(
mkr_data_list,
cam=cam,
mkr_grp=None,
with_bundles=False,
)
except:
raise
finally:
maya.mel.eval(
'paneLayout -e -manage true $gMainPane') # turn on Maya UI
return
def get_active_viewport_camera():
"""
Get the Camera that is attached to the active viewport.
:return: The Camera object, or None.
:rtype: Camera or None
"""
cam = None
# Get the camera from the active viewport.
model_editor = viewport_utils.get_active_model_editor()
if model_editor is None:
return cam
cam_tfm, cam_shp = viewport_utils.get_viewport_camera(model_editor)
if cam_shp is None:
return cam
if camera_utils.is_startup_cam(cam_shp) is True:
return cam
cam = mmapi.Camera(shape=cam_shp)
return cam
def test_get_root_frames_from_markers_3(self):
# Time Range
start = 0
end = 41
maya.cmds.playbackOptions(animationStartTime=start,
minTime=start,
animationEndTime=end,
maxTime=end)
# Create Camera
cam_tfm = maya.cmds.createNode('transform', name='cam_tfm')
cam_shp = maya.cmds.createNode('camera',
name='cam_shp',
parent=cam_tfm)
cam = mmapi.Camera(shape=cam_shp)
# 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)
# Load Marker Data
path = self.get_data_path('match_mover', 'loadmarker.rz2')
_, mkr_data_list = marker_read.read(path)
# Create Markers
mkr_grp = mmapi.MarkerGroup().create_node(cam=cam)
mkr_list = marker_read.create_nodes(mkr_data_list,
cam=cam,
mkr_grp=mkr_grp)
# Calculate Root Frames
min_frames_per_marker = 3
start_frame, end_frame = time_utils.get_maya_timeline_range_inner()
frame_nums = mod.get_root_frames_from_markers(mkr_list,
min_frames_per_marker,
start_frame, end_frame)
print 'frames:', frame_nums
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 main():
"""
Create a new marker under the current viewport camera, or under
the selected camera, if a camera is selected.
"""
mmapi.load_plugin()
sel = maya.cmds.ls(sl=True, long=True)
node_filtered = mmapi.filter_nodes_into_categories(sel)
cams = node_filtered['camera']
cams = filter(utils_camera.is_not_startup_cam, cams)
mkr_grps = node_filtered['markergroup']
cam = None
mkr_grp = None
if len(cams) > 0 and len(mkr_grps) > 0:
msg = ('Please select a camera or marker group; '
'both node types are selected.')
LOG.error(msg)
elif len(cams) == 0 and len(mkr_grps) == 0:
# Create a Marker under the active viewport camera.
model_editor = utils_viewport.get_active_model_editor()
if model_editor is None:
msg = 'Please select an active 3D viewport.'
LOG.warning(msg)
return
cam_tfm, cam_shp = utils_viewport.get_viewport_camera(model_editor)
node = cam_shp
if node is None:
msg = 'Please select an active viewport to get a camera.'
LOG.error(msg)
return
if utils_camera.is_startup_cam(node) is True:
msg = "Cannot create Markers in 'persp' camera."
LOG.error(msg)
return
if maya.cmds.nodeType(node) == 'transform':
cam = mmapi.Camera(transform=node)
elif maya.cmds.nodeType(node) == 'camera':
cam = mmapi.Camera(shape=node)
else:
LOG.error('Camera node is invalid; %r', node)
return
elif len(cams) > 0 and len(mkr_grps) == 0:
# Create a Marker under the selected camera.
node = cams[0]
if maya.cmds.nodeType(node) == 'transform':
cam = mmapi.Camera(transform=node)
elif maya.cmds.nodeType(node) == 'camera':
cam = mmapi.Camera(shape=node)
else:
LOG.error('Camera node is invalid; %r', node)
return
elif len(cams) == 0 and len(mkr_grps) > 0:
# Create a marker under the first selected Marker Group.
node = mkr_grps[0]
mkr_grp = mmapi.MarkerGroup(node=node)
else:
LOG.error('Should not get here.')
bnd_name = mmapi.get_new_bundle_name('bundle1')
bnd = mmapi.Bundle().create_node(name=bnd_name)
mkr_name = mmapi.get_new_marker_name('marker1')
mkr = mmapi.Marker().create_node(name=mkr_name,
cam=cam,
mkr_grp=mkr_grp,
bnd=bnd)
maya.cmds.select(mkr.get_node(), replace=True)
return
def create_marker():
"""
Create a new marker under the current viewport camera, or under
the selected camera, if a camera is selected.
"""
mmapi.load_plugin()
sel = maya.cmds.ls(sl=True, long=True)
node_filtered = filter_nodes.get_nodes(sel)
cams = node_filtered['camera']
cams = filter(is_not_startup_cam, cams)
mkr_grps = node_filtered['markergroup']
cam = None
mkr_grp = None
if len(cams) > 0 and len(mkr_grps) > 0:
msg = 'Please select a camera or marker group; '
msg += 'both node types are selected.'
LOG.error(msg)
elif len(cams) == 0 and len(mkr_grps) == 0:
node = __get_camera()
if node is None:
msg = 'Please activate a viewport to get a camera.'
LOG.error(msg)
return
if is_startup_cam(node) is True:
msg = "Cannot create Markers in 'persp' camera."
LOG.error(msg)
return
if maya.cmds.nodeType(node) == 'transform':
cam = mmapi.Camera(transform=node)
elif maya.cmds.nodeType(node) == 'camera':
cam = mmapi.Camera(shape=node)
else:
LOG.error('Camera node is invalid; %r', node)
return
elif len(cams) > 0 and len(mkr_grps) == 0:
node = cams[0]
if maya.cmds.nodeType(node) == 'transform':
cam = mmapi.Camera(transform=node)
elif maya.cmds.nodeType(node) == 'camera':
cam = mmapi.Camera(shape=node)
else:
LOG.error('Camera node is invalid; %r', node)
return
elif len(cams) == 0 and len(mkr_grps) > 0:
node = mkr_grps[0]
mkr_grp = mmapi.MarkerGroup(name=node)
else:
LOG.error('Should not get here.')
bnd_name = mmapi.get_bundle_name('bundle1')
bnd = mmapi.Bundle().create_node(
name=bnd_name
)
mkr_name = mmapi.get_marker_name('marker1')
mkr = mmapi.Marker().create_node(
name=mkr_name,
cam=cam,
mkr_grp=mkr_grp,
bnd=bnd
)
maya.cmds.select(mkr.get_node(), replace=True)
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
def create_markers_from_transforms(cam_tfm, cam_shp, tfm_nodes,
start_frame, end_frame,
bundle_position_mode,
delete_static_anim_curves):
"""
Create Markers from Maya transform nodes.
:param cam_tfm: Camera transform node.
:type cam_tfm: str
:param cam_shp: Camera shape node.
:type cam_shp: str
:param tfm_nodes: Input transform nodes to be converted.
:type tfm_nodes: [str, ...]
:param start_frame: The frame range to sample the input
transform nodes, and convert to animation.
:type start_frame: int
:param end_frame: The frame range to sample the input
transform nodes, and convert to animation.
:type end_frame: int
:param bundle_position_mode: The position for the newly created
Bundle (connected to the Marker).
:type bundle_position_mode: None or BUNDLE_POSITION_MODE_*
:param delete_static_anim_curves: When enabled, this will remove
all keyframes from the bundle, if the bundle is not animated.
:type delete_static_anim_curves: bool
:returns: A tuple of 2 lists; First list is the Marker nodes
created by the function, and the second list is the bundle
nodes created by this function.
:rtype: ([str, ...], [str, ...])
"""
# Compute the Marker Data.
mkr_data_list = convert_nodes_to_marker_data_list(
cam_tfm,
cam_shp,
tfm_nodes,
start_frame,
end_frame,
)
# Get Camera
cam = mmapi.Camera(shape=cam_shp)
# Get or create Marker Group.
mkr_grp = None
mkr_grp_nodes = maya.cmds.ls(cam_tfm, dag=True, long=True,
type='mmMarkerGroupTransform') or []
mkr_grp_nodes = sorted(mkr_grp_nodes)
if len(mkr_grp_nodes) == 0:
mkr_grp = mmapi.MarkerGroup().create_node(cam=cam)
else:
mkr_grp = mmapi.MarkerGroup(node=mkr_grp_nodes[0])
# Create Marker nodes
mkr_list = mayareadfile.create_nodes(
mkr_data_list,
cam=cam,
mkr_grp=mkr_grp,
with_bundles=True,
)
mkr_nodes = [mkr.get_node() for mkr in mkr_list]
bnd_list = [mkr.get_bundle() for mkr in mkr_list]
bnd_list = [bnd for bnd in bnd_list if bnd is not None]
bnd_nodes = [bnd.get_node() for bnd in bnd_list]
if bundle_position_mode == const.BUNDLE_POSITION_MODE_ORIGIN_VALUE:
# Bundle stays at the origin.
pass
elif bundle_position_mode == const.BUNDLE_POSITION_MODE_SOURCE_VALUE:
# Move the newly created bundle to the original transform's
# location.
_match_node_position(
bnd_nodes, tfm_nodes,
start_frame, end_frame,
delete_static_anim_curves)
return mkr_nodes, bnd_nodes
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_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 create_new_setup():
"""
Create the default setup
"""
maya.cmds.loadPlugin('matrixNodes', quiet=True)
mmapi.load_plugin()
# Create a camera and image plane (or validate and use the given one)
cam_tfm = maya.cmds.createNode('transform', name='camera')
cam_shp = maya.cmds.createNode('camera', name='cameraShape', parent=cam_tfm)
cam = mmapi.Camera(shape=cam_shp)
# Create image plane.
img_pl_tfm, img_pl_shp = maya.cmds.imagePlane(camera=cam_shp)
# Using a "To Size" fit mode will forcibly change the image to
# stretch it to match the size given. Therefore it is important
# the image plane size (including the aspect ratio) is correct.
maya.cmds.setAttr('{}.fit'.format(img_pl_shp), 4) # 4 = To Size
# Image plane is almost at far-clipping plane distance.
maya.cmds.setAttr('{}.depth'.format(img_pl_shp), 9990)
# Darken the imagePlane.
value = 0.5
maya.cmds.setAttr(
'{}.colorGain'.format(img_pl_shp),
value, value, value, type='double3')
# Make the image plane non-selectable.
maya.cmds.setAttr('{}.overrideEnabled'.format(img_pl_shp), 1)
maya.cmds.setAttr(
'{}.overrideDisplayType'.format(img_pl_shp),
2) # 2 == 'Reference' display type.
# Create a "Calibrate" marker group.
mkr_grp = mmapi.MarkerGroup().create_node(cam=cam, name='calibrate')
origin_mkr = mmapi.Marker().create_node(
mkr_grp=mkr_grp, name='originPoint')
origin_mkr_node = origin_mkr.get_node()
# Axis 1
line_a = _create_line(cam, mkr_grp, 'LineA')
line_b = _create_line(cam, mkr_grp, 'LineB')
intersect_node_a, vp_a_mkr = _create_vanishing_point(
line_a, line_b,
mkr_grp)
_set_default_axis_values(line_a, line_b, invert_x=False)
# TODO: Create visualiser node for the line.
# Axis 2
line_c = _create_line(cam, mkr_grp, 'LineC')
line_d = _create_line(cam, mkr_grp, 'LineD')
intersect_node_b, vp_b_mkr = _create_vanishing_point(
line_c, line_d,
mkr_grp)
_set_default_axis_values(line_c, line_d, invert_x=True)
# TODO: Create visualiser node for the line.
# Horizon Line
horizon_line_mkr1, horizon_line_mkr2 = _create_line(
cam, mkr_grp, 'HorizonLine')
_set_default_horizon_values(horizon_line_mkr1, horizon_line_mkr2)
horizon_line_mkr1_node = horizon_line_mkr1.get_node()
horizon_line_mkr2_node = horizon_line_mkr2.get_node()
# TODO: Create visualiser node for the line.
# Create a mmCameraCalibrate node.
calib_node = maya.cmds.createNode('mmCameraCalibrate')
maya.cmds.setAttr(
'{}.calibrationMode'.format(calib_node),
3) # 3 = "Two Vanishing Points".
decompose_node = maya.cmds.createNode('decomposeMatrix')
maya.cmds.setAttr(
'{}.inputRotateOrder'.format(decompose_node),
2) # 2 = ZXY (good default for cameras aimed at the horizon.)
maya.cmds.addAttr(
calib_node,
attributeType='float',
minValue=0.0,
defaultValue=1920.0,
longName='imageWidth')
maya.cmds.addAttr(
calib_node,
attributeType='float',
minValue=0.0,
defaultValue=1080.0,
longName='imageHeight')
maya.cmds.addAttr(
calib_node,
attributeType='float',
minValue=0.0,
defaultValue=1.0,
longName='imagePixelAspectRatio')
maya.cmds.addAttr(
calib_node,
attributeType='float',
minValue=0.0,
defaultValue=1.0,
longName='imageAspectRatio')
exp = (
'imageAspectRatio = (imageWidth * imagePixelAspectRatio) / imageHeight;'
'verticalFilmAperture = horizontalFilmAperture / imageAspectRatio;'
)
maya.cmds.expression(object=calib_node, string=exp)
maya.cmds.addAttr(
calib_node,
attributeType='message',
longName='vanishingPointNodeA')
maya.cmds.addAttr(
calib_node,
attributeType='message',
longName='vanishingPointNodeB')
maya.cmds.addAttr(
calib_node,
attributeType='message',
longName='horizonPointNodeA')
maya.cmds.addAttr(
calib_node,
attributeType='message',
longName='horizonPointNodeB')
maya.cmds.addAttr(
calib_node,
attributeType='message',
longName='originPointNode')
# Create connections.
src_dst_attr_list = [
# Camera shape viewing attributes.
['{}.outCameraAperture'.format(calib_node),
'{}.cameraAperture'.format(cam_shp)],
['{}.outFocalLength'.format(calib_node),
'{}.focalLength'.format(cam_shp)],
['{}.outMatrix'.format(calib_node),
'{}.inputMatrix'.format(decompose_node)],
['{}.outputTranslate'.format(decompose_node),
'{}.translate'.format(cam_tfm)],
['{}.outputRotate'.format(decompose_node),
'{}.rotate'.format(cam_tfm)],
['{}.inputRotateOrder'.format(decompose_node),
'{}.rotateOrder'.format(cam_tfm)],
# Connections so we can find the intersection values from the
# calibration, without creating a dependency
# 'attributeAffects' relationship in the DG.
['{}.message'.format(intersect_node_a),
'{}.vanishingPointNodeA'.format(calib_node)],
['{}.message'.format(intersect_node_b),
'{}.vanishingPointNodeB'.format(calib_node)],
['{}.message'.format(origin_mkr_node),
'{}.originPointNode'.format(calib_node)],
['{}.message'.format(horizon_line_mkr1_node),
'{}.horizonPointNodeA'.format(calib_node)],
['{}.message'.format(horizon_line_mkr2_node),
'{}.horizonPointNodeB'.format(calib_node)],
['{}.coverageX'.format(img_pl_shp),
'{}.imageWidth'.format(calib_node)],
['{}.coverageY'.format(img_pl_shp),
'{}.imageHeight'.format(calib_node)],
['{}.horizontalFilmAperture'.format(cam_shp),
'{}.sizeX'.format(img_pl_shp)],
['{}.verticalFilmAperture'.format(cam_shp),
'{}.sizeY'.format(img_pl_shp)],
]
for src, dst in src_dst_attr_list:
if maya.cmds.isConnected(src, dst) is False:
maya.cmds.connectAttr(src, dst)
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 main():
"""
Screen-Space Bake.
Usage:
1) Select transform nodes.
2) Activate viewport.
3) Run tool.
4) A new locator is created under the active camera
"""
mmapi.load_plugin()
editor = viewport_utils.get_active_model_editor()
if editor is None:
LOG.error('Please select an active viewport; editor=%r', editor)
return
cam_tfm, cam_shp = viewport_utils.get_viewport_camera(editor)
if cam_tfm is None or cam_shp is None:
LOG.error('Please select a camera; cam_tfm=%r cam_shp=%r',
cam_tfm, cam_shp)
return
cam = mmapi.Camera(transform=cam_tfm, shape=cam_shp)
img_width = maya.cmds.getAttr(cam_shp + '.horizontalFilmAperture') * 100.0
img_height = maya.cmds.getAttr(cam_shp + '.verticalFilmAperture') * 100.0
nodes = maya.cmds.ls(
selection=True,
long=True,
type='transform'
) or []
if len(nodes) == 0:
LOG.error('Please select transform nodes; %r', nodes)
return
start_frame, end_frame = utils_time.get_maya_timeline_range_inner()
times = range(start_frame, end_frame+1)
created_loc_tfms = []
for node in nodes:
grp_node, depth_tfm, loc_tfm, loc_shp = lib.create_screen_space_locator(cam)
created_loc_tfms.append(loc_tfm)
# BUG: If a camera has 'camera scale' attribute set other than
# 1.0, the reprojected values will not be correct.
values = maya.cmds.mmReprojection(
node,
camera=(cam_tfm, cam_shp),
time=times,
asMarkerCoordinate=True,
imageResolution=(int(img_width), int(img_height)),
)
stop = len(values)
step = 3
plug = loc_tfm + '.translateX'
values_x = values[0:stop:step]
animfn_x = anim_utils.create_anim_curve_node_apione(
times, values_x,
node_attr=plug
)
plug = loc_tfm + '.translateY'
values_y = values[1:stop:step]
animfn_y = anim_utils.create_anim_curve_node_apione(
times, values_y,
node_attr=plug
)
plug = depth_tfm + '.scaleX'
values_z = values[2:stop:step]
animfn_z = anim_utils.create_anim_curve_node_apione(
times, values_z,
node_attr=plug
)
if len(created_loc_tfms) > 0:
maya.cmds.select(created_loc_tfms, replace=True)
# Trigger Maya to refresh.
maya.cmds.currentTime(update=True)
maya.cmds.refresh()
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 main():
"""
Convert all selected transforms into 2D markers under a camera.
"""
# Get camera
model_editor = utils_viewport.get_active_model_editor()
if model_editor is None:
msg = 'Please select an active 3D viewport.'
LOG.warning(msg)
return
cam_tfm, cam_shp = utils_viewport.get_viewport_camera(model_editor)
if cam_shp is None:
LOG.error('Please select an active viewport to get a camera.')
return
if utils_camera.is_startup_cam(cam_shp) is True:
LOG.error("Cannot create Markers in 'persp' camera.")
return
# Get transforms
nodes = maya.cmds.ls(
selection=True,
long=True,
type='transform',
) or []
if len(nodes) == 0:
LOG.warning('Please select one or more transform nodes.')
return
mmapi.load_plugin()
try:
utils_viewport.viewport_turn_off()
# Compute the Marker Data.
start_frame, end_frame = utils_time.get_maya_timeline_range_outer()
mkr_data_list = lib.convert_nodes_to_marker_data_list(
cam_tfm,
cam_shp,
nodes,
start_frame,
end_frame,
)
# Get Camera
cam = mmapi.Camera(shape=cam_shp)
# Get or create Marker Group.
mkr_grp = None
mkr_grp_nodes = maya.cmds.ls(
cam_tfm, dag=True, long=True, type='mmMarkerGroupTransform') or []
mkr_grp_nodes = sorted(mkr_grp_nodes)
if len(mkr_grp_nodes) == 0:
mkr_grp = mmapi.MarkerGroup().create_node(cam=cam)
else:
mkr_grp = mmapi.MarkerGroup(node=mkr_grp_nodes[0])
# Create Marker nodes
mkr_list = mayareadfile.create_nodes(
mkr_data_list,
cam=cam,
mkr_grp=mkr_grp,
with_bundles=True,
)
mkr_nodes = [mkr.get_node() for mkr in mkr_list]
finally:
utils_viewport.viewport_turn_on()
if len(mkr_nodes) > 0:
maya.cmds.select(mkr_nodes, replace=True)
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
