def exportCameraPoints(f, campg):
#
# write camera point group...
f.write("\n")
f.write("// create camera point group...\n")
name = "cameraPGroup_%s_1" % tools.validName(tde4.getPGroupName(campg))
f.write(
"string $pointGroupName = `group -em -name \"%s\" -parent $sceneGroupName`;\n"
% name)
f.write("$pointGroupName = ($sceneGroupName + \"|\" + $pointGroupName);\n")
f.write("\n")
# write points...
l = tde4.getPointList(campg)
particleExists = 0
f.write("\nparticle;")
for p in l:
if tde4.isPointCalculated3D(campg, p):
name = tde4.getPointName(campg, p)
name = "p%s" % tools.validName(name)
p3d = tde4.getPointCalcPosition3D(campg, p)
p3d = convertZup(p3d, yup)
f.write("\n")
if 'agisoft_' in name:
particleStr = "emit -object particle1 -pos %.15f %.15f %.15f ;"
f.write(particleStr % (p3d[0], p3d[1], p3d[2]))
particleExists = 1
else:
f.write("// create point %s...\n" % name)
f.write(
"string $locator = stringArrayToString(`spaceLocator -name %s`, \"\");\n"
% name)
f.write("$locator = (\"|\" + $locator);\n")
f.write("xform -t %.15f %.15f %.15f $locator;\n" %
(p3d[0], p3d[1], p3d[2]))
f.write("parent $locator $pointGroupName;\n")
f.write("\n")
f.write(
"xform -zeroTransformPivots -rotateOrder zxy -scale 1.000000 1.000000 1.000000 $pointGroupName;\n"
)
f.write("\n")
if particleExists == 1:
f.write("particle -q -ct particle1;")
f.write('disconnectAttr time1.outTime particleShape1.currentTime;\n')
f.write('saveInitialState -atr "pos" particle1;\n')
f.write('parent "particle1" $sceneGroupName;\n')
f.write('rename "Scene|particle1" "agisoft_points";\n')
else:
f.write('delete "particle1";')
def _generate_v2(point_group,
camera,
points,
start_frame=None,
undistort=False):
"""
Generate the UV file format contents, using JSON format.
:param point_group: The 3DE Point Group containing 'points'
:type point_group: str
:param camera: The 3DE Camera containing 2D 'points' data.
:type camera: str
:param points: The list of 3DE Points representing 2D data to
save.
:type points: list of str
:param start_frame: The frame number to be considered at
'first frame'. Defaults to 1001 if
set to None.
:type start_frame: None or int
:param undistort: Should we apply undistortion to the 2D points
data? Yes or no.
:type undistort: bool
Each point will store:
- Point name
- X, Y position (in UV coordinates, per-frame)
- Point weight (per-frame)
- Point Set name
- Point 'Persistent ID'
"""
assert isinstance(point_group, basestring)
assert isinstance(camera, basestring)
assert isinstance(points, (list, tuple))
assert start_frame is None or isinstance(start_frame, int)
assert isinstance(undistort, bool)
if start_frame is None:
start_frame = 1001
data = UV_TRACK_HEADER_VERSION_2.copy()
cam_num_frames = tde4.getCameraNoFrames(camera)
if len(points) == 0:
return ''
frame0 = int(start_frame)
frame0 -= 1
data['num_points'] = len(points)
data['is_undistorted'] = bool(undistort)
data['points'] = []
for point in points:
point_data = {}
# Query point information
name = tde4.getPointName(point_group, point)
uid = None
if SUPPORT_PERSISTENT_ID is True:
uid = tde4.getPointPersistentID(point_group, point)
point_set = tde4.getPointSet(point_group, point)
point_set_name = None
if point_set is not None:
point_set_name = tde4.getSetName(point_group, point_set)
point_data['name'] = name
point_data['id'] = uid
point_data['set_name'] = point_set_name
# Write per-frame position data
frame = 1 # 3DE starts at frame '1' regardless of the 'start frame'.
point_data['per_frame'] = []
pos_block = tde4.getPointPosition2DBlock(point_group, point, camera, 1,
cam_num_frames)
for pos in pos_block:
if pos[0] == -1.0 or pos[1] == -1.0:
# No valid data here.
frame += 1
continue
# Does the 2D point go outside the camera FOV? Is that ok?
valid_2d = tde4.isPointPos2DValid(point_group, point, camera,
frame)
if valid_2d != 1:
# No valid data here.
frame += 1
continue
f = frame + frame0
if undistort is True:
pos = tde4.removeDistortion2D(camera, frame, pos)
weight = 1.0
if SUPPORT_POINT_WEIGHT_BY_FRAME is True:
weight = tde4.getPointWeightByFrame(point_group, point, camera,
frame)
frame_data = {'frame': f, 'pos': pos, 'weight': weight}
point_data['per_frame'].append(frame_data)
frame += 1
data['points'].append(point_data)
data_str = json.dumps(data)
return data_str
def _generate_v1(point_group,
camera,
points,
start_frame=None,
undistort=False):
"""
Generate the UV file format contents, using a basic ASCII format.
:param point_group: The 3DE Point Group containing 'points'
:type point_group: str
:param camera: The 3DE Camera containing 2D 'points' data.
:type camera: str
:param points: The list of 3DE Points representing 2D data to
save.
:type points: list of str
:param start_frame: The frame number to be considered at
'first frame'. Defaults to 1001 if
set to None.
:type start_frame: None or int
:param undistort: Should we apply undistortion to the 2D points
data? Yes or no.
:type undistort: bool
Each point will store:
- Point name
- X, Y position (in UV coordinates, per-frame)
- Point weight (per-frame)
"""
assert isinstance(point_group, basestring)
assert isinstance(camera, basestring)
assert isinstance(points, (list, tuple))
assert start_frame is None or isinstance(start_frame, int)
assert isinstance(undistort, bool)
if start_frame is None:
start_frame = 1001
data_str = ''
cam_num_frames = tde4.getCameraNoFrames(camera)
if len(points) == 0:
return data_str
frame0 = int(start_frame)
frame0 -= 1
data_str += '{0:d}\n'.format(len(points))
for point in points:
name = tde4.getPointName(point_group, point)
c2d = tde4.getPointPosition2DBlock(point_group, point, camera, 1,
cam_num_frames)
# Write per-frame position data
num_valid_frame = 0
pos_list = []
weight_list = []
frame = 1 # 3DE starts at frame '1' regardless of the 'start-frame'.
for v in c2d:
if v[0] == -1.0 or v[1] == -1.0:
# No valid data here.
frame += 1
continue
# Does the 2D point go outside the camera FOV? Is that ok?
valid = tde4.isPointPos2DValid(point_group, point, camera, frame)
if valid == 0:
# No valid data here.
frame += 1
continue
# Number of points with valid positions
num_valid_frame += 1
f = frame + frame0
if undistort is True:
v = tde4.removeDistortion2D(camera, frame, v)
weight = 1.0
if SUPPORT_POINT_WEIGHT_BY_FRAME is True:
weight = tde4.getPointWeightByFrame(point_group, point, camera,
frame)
pos_list.append((f, v))
weight_list.append((f, weight))
frame += 1
# add data
data_str += name + '\n'
data_str += '{0:d}\n'.format(num_valid_frame)
for pos_data, weight_data in zip(pos_list, weight_list):
f = pos_data[0]
v = pos_data[1]
w = weight_data[1]
assert f == weight_data[0]
data_str += '%d %.15f %.15f %.8f\n' % (f, v[0], v[1], w)
return data_str
f.write("\n")
f.write("// create camera point group...\n")
name = 'trk_locator'
f.write(
"string $pointGroupName = `group -em -name \"%s\" -parent $sceneGroupName`;\n"
% name)
f.write(
"$pointGroupName = ($sceneGroupName + \"|\" + $pointGroupName);\n"
)
f.write("\n")
# write points...
l = tde4.getPointList(campg)
for p in l:
if tde4.isPointCalculated3D(campg, p):
name = tde4.getPointName(campg, p)
name = "Tracker%s" % validName(name)
p3d = tde4.getPointCalcPosition3D(campg, p)
p3d = convertZup(p3d, yup)
f.write("\n")
f.write("// create point %s...\n" % name)
f.write(
"string $locator = stringArrayToString(`spaceLocator -name %s`, \"\");\n"
% name)
f.write("$locator = (\"|\" + $locator);\n")
f.write("xform -t %.15f %.15f %.15f $locator;\n" %
(p3d[0], p3d[1], p3d[2]))
f.write("parent $locator $pointGroupName;\n")
f.write("\n")
f.write(
def _generate_v2_v3_and_v4(point_group,
camera,
points,
version=None,
**kwargs):
"""
Generate the UV file format contents, using JSON format.
Set the individual _generate_v2 or _generate_v3 functions for
details of what is stored.
:param point_group: The 3DE Point Group containing 'points'
:type point_group: str
:param camera: The 3DE Camera containing 2D 'points' data.
:type camera: str
:param points: The list of 3DE Points representing 2D data to
save.
:type points: list of str
:param version: The version of file to generate,
UV_TRACK_FORMAT_VERSION_2 or
UV_TRACK_FORMAT_VERSION_3.
:type version: int
:param start_frame: Format v2 and v3; The frame number to be
considered at 'first frame'.
Defaults to 1001 if set to None.
:type start_frame: None or int
:param undistort: Format v2; Should we apply undistortion to the 2D
points data? Yes or no.
:type undistort: bool
:returns: A JSON format string, with the UV Track data in it.
:rtype: str
"""
assert isinstance(point_group, basestring)
assert isinstance(camera, basestring)
assert isinstance(points, (list, tuple))
assert isinstance(version, (int, long))
assert version in [
UV_TRACK_FORMAT_VERSION_2, UV_TRACK_FORMAT_VERSION_3,
UV_TRACK_FORMAT_VERSION_4
]
start_frame = kwargs.get('start_frame')
assert start_frame is None or isinstance(start_frame, int)
if start_frame is None:
start_frame = 1001
undistort = None
if version == UV_TRACK_FORMAT_VERSION_2:
undistort = kwargs.get('undistort')
assert isinstance(undistort, bool)
data = None
if version == UV_TRACK_FORMAT_VERSION_2:
data = UV_TRACK_HEADER_VERSION_2.copy()
elif version == UV_TRACK_FORMAT_VERSION_3:
data = UV_TRACK_HEADER_VERSION_3.copy()
elif version == UV_TRACK_FORMAT_VERSION_4:
data = UV_TRACK_HEADER_VERSION_4.copy()
else:
raise ValueError("Version number is invalid; %r" % version)
cam_num_frames = tde4.getCameraNoFrames(camera)
camera_fov = tde4.getCameraFOV(camera)
if len(points) == 0:
return ''
frame0 = int(start_frame)
frame0 -= 1
data['num_points'] = len(points)
data['is_undistorted'] = None
if version == UV_TRACK_FORMAT_VERSION_2:
data['is_undistorted'] = bool(undistort)
data['points'] = []
for point in points:
point_data = {}
# Query point information
name = tde4.getPointName(point_group, point)
uid = None
if SUPPORT_PERSISTENT_ID is True:
uid = tde4.getPointPersistentID(point_group, point)
point_set = tde4.getPointSet(point_group, point)
point_set_name = None
if point_set is not None:
point_set_name = tde4.getSetName(point_group, point_set)
point_data['name'] = name
point_data['id'] = uid
point_data['set_name'] = point_set_name
valid_mode = _get_point_valid_mode(point_group, point)
# Get the 3D point position
if version in [UV_TRACK_FORMAT_VERSION_3, UV_TRACK_FORMAT_VERSION_4]:
point_data['3d'] = _get_3d_data_from_point(point_group, point)
# Write per-frame position data
frame = 1 # 3DE starts at frame '1' regardless of the 'start frame'.
point_data['per_frame'] = []
pos_block = tde4.getPointPosition2DBlock(point_group, point, camera, 1,
cam_num_frames)
for pos in pos_block:
if pos[0] == -1.0 or pos[1] == -1.0:
# No valid data here.
frame += 1
continue
# Is the 2D point obscured?
valid = tde4.isPointPos2DValid(point_group, point, camera, frame)
if valid == 0:
# No valid data here.
frame += 1
continue
# Check if we're inside the FOV / Frame or not.
valid_pos = _is_valid_position(pos, camera_fov, valid_mode)
if valid_pos is False:
frame += 1
continue
pos_undist = pos
if undistort is True or undistort is None:
pos_undist = tde4.removeDistortion2D(camera, frame, pos)
weight = _get_point_weight(point_group, point, camera, frame)
f = frame + frame0
frame_data = {'frame': f, 'pos': pos_undist, 'weight': weight}
if version in [
UV_TRACK_FORMAT_VERSION_3, UV_TRACK_FORMAT_VERSION_4
]:
frame_data['pos_dist'] = pos
point_data['per_frame'].append(frame_data)
frame += 1
data['points'].append(point_data)
if version == UV_TRACK_FORMAT_VERSION_4:
lens = tde4.getCameraLens(camera)
data['camera'] = _generate_camera_data(camera, lens, frame0)
data_str = json.dumps(data)
return data_str
def exportMocaObjects(f, frameStart):
#
# write object/mocap point groups...
frameStart = int(frameStart)
frame0 = frameStart - 1
camera = tde4.getCurrentCamera()
noframes = tde4.getCameraNoFrames(camera)
pgl = tde4.getPGroupList()
index = 1
for pg in pgl:
if tde4.getPGroupType(pg) == "OBJECT" and camera is not None:
f.write("\n")
f.write("// create object point group...\n")
groupValidName = tools.validName(tde4.getPGroupName(pg))
pgname = "objectPGroup_%s_%d_1" % (groupValidName, index)
index += 1
f.write(
"string $pointGroupName = `group -em -name \"%s\" -parent $sceneGroupName`;\n"
% pgname)
f.write(
"$pointGroupName = ($sceneGroupName + \"|\" + $pointGroupName);\n"
)
# write points...
l = tde4.getPointList(pg)
for p in l:
if tde4.isPointCalculated3D(pg, p):
name = tde4.getPointName(pg, p)
name = "p%s" % tools.validName(name)
p3d = tde4.getPointCalcPosition3D(pg, p)
p3d = convertZup(p3d, yup)
if 'agisoft' not in name:
f.write("\n")
f.write("// create point %s...\n" % name)
f.write(
"string $locator = stringArrayToString(`spaceLocator -name %s`, \"\");\n"
% name)
f.write("$locator = (\"|\" + $locator);\n")
f.write("xform -t %.15f %.15f %.15f $locator;\n" %
(p3d[0], p3d[1], p3d[2]))
f.write("parent $locator $pointGroupName;\n")
f.write("\n")
scale = tde4.getPGroupScale3D(pg)
f.write(
"xform -zeroTransformPivots -rotateOrder zxy -scale %.15f %.15f %.15f $pointGroupName;\n"
% (scale, scale, scale))
# animate object point group...
f.write("\n")
f.write("// animating point group %s...\n" % pgname)
frame = 1
rot0 = None
while frame <= noframes:
# rot/pos...
p3d = tde4.getPGroupPosition3D(pg, camera, frame)
p3d = convertZup(p3d, yup)
r3d = tde4.getPGroupRotation3D(pg, camera, frame)
rot = convertToAngles(r3d)
if frame > 1:
rot = [
angleMod360(rot0[0], rot[0]),
angleMod360(rot0[1], rot[1]),
angleMod360(rot0[2], rot[2])
]
rot0 = rot
f.write(
"setKeyframe -at translateX -t %d -v %.15f $pointGroupName; "
% (frame + frame0, p3d[0]))
f.write(
"setKeyframe -at translateY -t %d -v %.15f $pointGroupName; "
% (frame + frame0, p3d[1]))
f.write(
"setKeyframe -at translateZ -t %d -v %.15f $pointGroupName; "
% (frame + frame0, p3d[2]))
f.write(
"setKeyframe -at rotateX -t %d -v %.15f $pointGroupName; "
% (frame + frame0, rot[0]))
f.write(
"setKeyframe -at rotateY -t %d -v %.15f $pointGroupName; "
% (frame + frame0, rot[1]))
f.write(
"setKeyframe -at rotateZ -t %d -v %.15f $pointGroupName;\n"
% (frame + frame0, rot[2]))
frame += 1
# mocap point groups...
if tde4.getPGroupType(pg) == "MOCAP" and camera is not None:
f.write("\n")
f.write("// create mocap point group...\n")
groupValidName = tools.validName(tde4.getPGroupName(pg))
pgname = "objectPGroup_%s_%d_1" % (groupValidName, index)
index += 1
f.write(
"string $pointGroupName = `group -em -name \"%s\" -parent $sceneGroupName`;\n"
% pgname)
f.write(
"$pointGroupName = ($sceneGroupName + \"|\" + $pointGroupName);\n"
)
# write points...
l = tde4.getPointList(pg)
for p in l:
if tde4.isPointCalculated3D(pg, p):
name = tde4.getPointName(pg, p)
name = "p%s" % tools.validName(name)
p3d = tde4.getPointMoCapCalcPosition3D(pg, p, camera, 1)
p3d = convertZup(p3d, yup)
f.write("\n")
f.write("// create point %s...\n" % name)
f.write(
"string $locator = stringArrayToString(`spaceLocator -name %s`, \"\");\n"
% name)
f.write("$locator = (\"|\" + $locator);\n")
f.write("xform -t %.15f %.15f %.15f $locator;\n" %
(p3d[0], p3d[1], p3d[2]))
for frame in range(1, noframes + 1):
p3d = tde4.getPointMoCapCalcPosition3D(
pg, p, camera, frame)
p3d = convertZup(p3d, yup)
f.write(
"setKeyframe -at translateX -t %d -v %.15f $locator; "
% (frame + frame0, p3d[0]))
f.write(
"setKeyframe -at translateY -t %d -v %.15f $locator; "
% (frame + frame0, p3d[1]))
f.write(
"setKeyframe -at translateZ -t %d -v %.15f $locator; "
% (frame + frame0, p3d[2]))
f.write("parent $locator $pointGroupName;\n")
f.write("\n")
scale = tde4.getPGroupScale3D(pg)
f.write(
"xform -zeroTransformPivots -rotateOrder zxy -scale %.15f %.15f %.15f $pointGroupName;\n"
% (scale, scale, scale))
# animate mocap point group...
f.write("\n")
f.write("// animating point group %s...\n" % pgname)
frame = 1
while frame <= noframes:
# rot/pos...
p3d = tde4.getPGroupPosition3D(pg, camera, frame)
p3d = convertZup(p3d, yup)
r3d = tde4.getPGroupRotation3D(pg, camera, frame)
rot = convertToAngles(r3d)
if frame > 1:
rot = [
angleMod360(rot0[0], rot[0]),
angleMod360(rot0[1], rot[1]),
angleMod360(rot0[2], rot[2])
]
rot0 = rot
f.write(
"setKeyframe -at translateX -t %d -v %.15f $pointGroupName; "
% (frame + frame0, p3d[0]))
f.write(
"setKeyframe -at translateY -t %d -v %.15f $pointGroupName; "
% (frame + frame0, p3d[1]))
f.write(
"setKeyframe -at translateZ -t %d -v %.15f $pointGroupName; "
% (frame + frame0, p3d[2]))
f.write(
"setKeyframe -at rotateX -t %d -v %.15f $pointGroupName; "
% (frame + frame0, rot[0]))
f.write(
"setKeyframe -at rotateY -t %d -v %.15f $pointGroupName; "
% (frame + frame0, rot[1]))
f.write(
"setKeyframe -at rotateZ -t %d -v %.15f $pointGroupName;\n"
% (frame + frame0, rot[2]))
frame += 1
def _generate_v1(point_group, camera, points,
start_frame=None,
undistort=False,
rs_distance=None):
"""
Generate the UV file format contents, using a basic ASCII format.
Each point will store:
- Point name
- X, Y position (in UV coordinates, per-frame)
- Point weight (per-frame)
:param point_group: The 3DE Point Group containing 'points'
:type point_group: str
:param camera: The 3DE Camera containing 2D 'points' data.
:type camera: str
:param points: The list of 3DE Points representing 2D data to
save.
:type points: list of str
:param start_frame: The frame number to be considered at
'first frame'. Defaults to 1001 if
set to None.
:type start_frame: None or int
:param undistort: Should we apply undistortion to the 2D points
data? Yes or no.
:type undistort: bool
:param rs_distance: If not None, correct rolling shutter effects on
the 2D points at the content distance rs_distance.
:type rs_distance: None or float
:returns: A ASCII format string, with the UV Track data in it.
:rtype: str
"""
assert isinstance(point_group, TEXT_TYPE)
assert isinstance(camera, TEXT_TYPE)
assert isinstance(points, (list, tuple))
assert start_frame is None or isinstance(start_frame, int)
assert isinstance(undistort, bool)
assert rs_distance is None or isinstance(rs_distance, float)
if start_frame is None:
start_frame = 1001
data_str = ''
cam_num_frames = tde4.getCameraNoFrames(camera)
camera_fov = tde4.getCameraFOV(camera)
if len(points) == 0:
return data_str
frame0 = int(start_frame)
frame0 -= 1
data_str += '{0:d}\n'.format(len(points))
for point in points:
name = tde4.getPointName(point_group, point)
c2d = tde4.getPointPosition2DBlock(
point_group, point, camera,
1, cam_num_frames
)
valid_mode = _get_point_valid_mode(point_group, point)
# Write per-frame position data
num_valid_frame = 0
pos_list = []
weight_list = []
frame = 1 # 3DE starts at frame '1' regardless of the 'start-frame'.
for v in c2d:
if v[0] == -1.0 or v[1] == -1.0:
# No valid data here.
frame += 1
continue
# Does the 2D point go outside the camera FOV? Is that ok?
valid = tde4.isPointPos2DValid(
point_group,
point,
camera,
frame
)
if valid == 0:
# No valid data here.
frame += 1
continue
# Check if we're inside the FOV / Frame or not.
valid_pos = _is_valid_position(v, camera_fov, valid_mode)
if valid_pos is False:
frame += 1
continue
# Number of points with valid positions
num_valid_frame += 1
f = frame + frame0
if rs_distance is not None:
v = vl_sdv.vec2d(v[0], v[1])
v = _remove_rs_from_2d_point(
point_group, camera, frame, v, rs_distance)
if undistort is True:
v = tde4.removeDistortion2D(camera, frame, v)
weight = _get_point_weight(point_group, point, camera, frame)
pos_list.append((f, v))
weight_list.append((f, weight))
frame += 1
# add data
data_str += name + '\n'
data_str += '{0:d}\n'.format(num_valid_frame)
for pos_data, weight_data in zip(pos_list, weight_list):
f = pos_data[0]
v = pos_data[1]
w = weight_data[1]
assert f == weight_data[0]
data_str += '%d %.15f %.15f %.8f\n' % (f, v[0], v[1], w)
return data_str
def name(self):
return tde4.getPointName(self._group.id_, self._point_id)
def main():
campg = None
pgl = tde4.getPGroupList()
for pg in pgl:
if tde4.getPGroupType(pg) == "CAMERA": campg = pg
if campg == None:
tde4.postQuestionRequester("Export Maya...",
"Error, there is no camera point group.",
"Ok")
#
# open requester...
try:
req = _export_requester_maya
except (ValueError, NameError, TypeError):
_export_requester_maya = tde4.createCustomRequester()
req = _export_requester_maya
tde4.addFileWidget(req, "file_browser", "Exportfile...", "*.mel")
tde4.addTextFieldWidget(req, "startframe_field", "Startframe", "1")
# tde4.addOptionMenuWidget(req,"mode_menu","Orientation","Y-Up", "Z-Up")
tde4.addToggleWidget(req, "hide_ref_frames", "Hide Reference Frames",
0)
cam = tde4.getCurrentCamera()
offset = tde4.getCameraFrameOffset(cam)
tde4.setWidgetValue(req, "startframe_field", str(offset))
# ret = tde4.postCustomRequester(req,"Export Maya (MEL-Script)...",600,0,"Ok","Cancel")
ret = 1
if ret == 1:
# yup = tde4.getWidgetValue(req,"mode_menu")
# if yup==2: yup = 0
yup = 1
# path = tde4.getWidgetValue(req,"file_browser")
path = get_mel_filename()['path']
# frame0 = float(tde4.getWidgetValue(req,"startframe_field"))
# frame0 -= 1
framerange = get_frame_range()
playbackoptions = 'playbackOptions -min {0} -max {1};'
playbackoptions = playbackoptions.format(framerange['first'],
framerange['last'])
frame0 = framerange['first'] - 1
hide_ref = tde4.getWidgetValue(req, "hide_ref_frames")
if path != None:
if not path.endswith('.mel'): path = path + '.mel'
f = open(path, "w")
if not f.closed:
#
# write some comments...
f.write("//\n")
f.write("// Maya/MEL export data written by %s\n" %
tde4.get3DEVersion())
f.write("//\n")
f.write(
"// All lengths are in centimeter, all angles are in degree.\n"
)
f.write("//\n\n")
#
# write scene group...
groupname = """// create scene group...
string $sceneGroupName = `group -em -name "mm_{name}"`;
"""
# f.write("// create scene group...\n")
# f.write("string $sceneGroupName = `group -em -name \"Scene\"`;\n")
groupname = groupname.format(
name=get_mel_filename()['filename'][:-4])
f.write(groupname)
#
# write cameras...
cl = tde4.getCameraList()
index = 1
for cam in cl:
camType = tde4.getCameraType(cam)
noframes = tde4.getCameraNoFrames(cam)
lens = tde4.getCameraLens(cam)
if lens != None:
name = validName(tde4.getCameraName(cam))
cam_name = 'cam_mm_' + name
# name = "%s_%s_1"%(name,index)
# name = "%s_%s"%(name,index)
name = cam_name
index += 1
fback_w = tde4.getLensFBackWidth(lens)
fback_h = tde4.getLensFBackHeight(lens)
p_aspect = tde4.getLensPixelAspect(lens)
focal = tde4.getCameraFocalLength(cam, 1)
lco_x = tde4.getLensLensCenterX(lens)
lco_y = tde4.getLensLensCenterY(lens)
# convert filmback to inch...
fback_w = fback_w / 2.54
fback_h = fback_h / 2.54
lco_x = -lco_x / 2.54
lco_y = -lco_y / 2.54
# convert focal length to mm...
focal = focal * 10.0
# create camera...
f.write("\n")
f.write("// create camera %s...\n" % name)
f.write(
"string $cameraNodes[] = `camera -name \"%s\" -hfa %.15f -vfa %.15f -fl %.15f -ncp 0.01 -fcp 10000 -shutterAngle 180 -ff \"overscan\"`;\n"
% (name, fback_w, fback_h, focal))
f.write("string $cameraTransform = $cameraNodes[0];\n")
f.write("string $cameraShape = $cameraNodes[1];\n")
f.write(
"xform -zeroTransformPivots -rotateOrder zxy $cameraTransform;\n"
)
f.write(
"setAttr ($cameraShape+\".horizontalFilmOffset\") %.15f;\n"
% lco_x)
f.write(
"setAttr ($cameraShape+\".verticalFilmOffset\") %.15f;\n"
% lco_y)
f.write("setAttr ($cameraShape+\".renderable\") 1;\n")
p3d = tde4.getPGroupPosition3D(campg, cam, 1)
p3d = convertZup(p3d, yup)
f.write(
"xform -translation %.15f %.15f %.15f $cameraTransform;\n"
% (p3d[0], p3d[1], p3d[2]))
r3d = tde4.getPGroupRotation3D(campg, cam, 1)
rot = convertToAngles(r3d)
f.write(
"xform -rotation %.15f %.15f %.15f $cameraTransform;\n"
% rot)
f.write("xform -scale 1 1 1 $cameraTransform;\n")
"""add pipeline attributes to camerashape"""
# attribs = add_pipeline_attribs() ### OBSOLETE WAY
attribs = add_pipeline_parms()
f.write(attribs)
# image plane...
f.write("\n\n\n\n// create image plane...\n")
f.write(
"string $imagePlane = `createNode imagePlane`;\n")
f.write(
"cameraImagePlaneUpdate ($cameraShape, $imagePlane);\n"
)
f.write(
"setAttr ($imagePlane + \".offsetX\") %.15f;\n" %
lco_x)
f.write(
"setAttr ($imagePlane + \".offsetY\") %.15f;\n" %
lco_y)
if camType == "SEQUENCE":
f.write(
"setAttr ($imagePlane+\".useFrameExtension\") 1;\n"
)
else:
f.write(
"setAttr ($imagePlane+\".useFrameExtension\") 0;\n"
)
f.write(
"expression -n \"frame_ext_expression\" -s ($imagePlane+\".frameExtension=frame\");\n"
)
path = tde4.getCameraPath(cam)
sattr = tde4.getCameraSequenceAttr(cam)
path = prepareImagePath(path, sattr[0])
f.write(
"setAttr ($imagePlane + \".imageName\") -type \"string\" \"%s\";\n"
% (path))
f.write("setAttr ($imagePlane + \".fit\") 4;\n")
f.write(
"setAttr ($imagePlane + \".displayOnlyIfCurrent\") 1;\n"
)
f.write(
"setAttr ($imagePlane + \".depth\") (9000/2);\n")
# parent camera to scene group...
f.write("\n")
f.write("// parent camera to scene group...\n")
f.write("parent $cameraTransform $sceneGroupName;\n")
if camType == "REF_FRAME" and hide_ref:
f.write(
"setAttr ($cameraTransform +\".visibility\") 0;\n"
)
# animate camera...
if camType != "REF_FRAME":
f.write("\n")
f.write("// animating camera %s...\n" % name)
f.write(playbackoptions)
# f.write("playbackOptions -min %d -max %d;\n"%(1+frame0,noframes+frame0))
f.write("\n\n")
frame = 1
while frame <= noframes:
# rot/pos...
p3d = tde4.getPGroupPosition3D(campg, cam, frame)
p3d = convertZup(p3d, yup)
r3d = tde4.getPGroupRotation3D(campg, cam, frame)
rot = convertToAngles(r3d)
if frame > 1:
rot = [
angleMod360(rot0[0], rot[0]),
angleMod360(rot0[1], rot[1]),
angleMod360(rot0[2], rot[2])
]
rot0 = rot
f.write(
"setKeyframe -at translateX -t %d -v %.15f $cameraTransform; "
% (frame + frame0, p3d[0]))
f.write(
"setKeyframe -at translateY -t %d -v %.15f $cameraTransform; "
% (frame + frame0, p3d[1]))
f.write(
"setKeyframe -at translateZ -t %d -v %.15f $cameraTransform; "
% (frame + frame0, p3d[2]))
f.write(
"setKeyframe -at rotateX -t %d -v %.15f $cameraTransform; "
% (frame + frame0, rot[0]))
f.write(
"setKeyframe -at rotateY -t %d -v %.15f $cameraTransform; "
% (frame + frame0, rot[1]))
f.write(
"setKeyframe -at rotateZ -t %d -v %.15f $cameraTransform; "
% (frame + frame0, rot[2]))
# focal length...
focal = tde4.getCameraFocalLength(cam, frame)
focal = focal * 10.0
f.write(
"setKeyframe -at focalLength -t %d -v %.15f $cameraShape;\n"
% (frame + frame0, focal))
frame += 1
#
# write camera point group...
f.write("\n")
f.write("// create camera point group...\n")
name = "cameraPGroup_%s_1" % validName(
tde4.getPGroupName(campg))
f.write(
"string $pointGroupName = `group -em -name \"%s\" -parent $sceneGroupName`;\n"
% name)
# f.write("$pointGroupName = ($sceneGroupName + \"|\" + $pointGroupName);\n")
f.write("\n")
# write points...
l = tde4.getPointList(campg)
for p in l:
if tde4.isPointCalculated3D(campg, p):
name = tde4.getPointName(campg, p)
name = "p%s" % validName(name)
p3d = tde4.getPointCalcPosition3D(campg, p)
p3d = convertZup(p3d, yup)
f.write("\n")
f.write("// create point %s...\n" % name)
f.write(
"string $locator = stringArrayToString(`spaceLocator -name %s`, \"\");\n"
% name)
f.write("$locator = (\"|\" + $locator);\n")
f.write("xform -t %.15f %.15f %.15f $locator;\n" %
(p3d[0], p3d[1], p3d[2]))
f.write("parent $locator $pointGroupName;\n")
f.write("\n")
f.write(
"xform -zeroTransformPivots -rotateOrder zxy -scale 1.000000 1.000000 1.000000 $pointGroupName;\n"
)
f.write("\n")
#
# write object/mocap point groups...
camera = tde4.getCurrentCamera()
noframes = tde4.getCameraNoFrames(camera)
pgl = tde4.getPGroupList()
index = 1
for pg in pgl:
if tde4.getPGroupType(pg) == "OBJECT" and camera != None:
f.write("\n")
f.write("// create object point group...\n")
pgname = "objectPGroup_%s_%d_1" % (validName(
tde4.getPGroupName(pg)), index)
index += 1
f.write(
"string $pointGroupName = `group -em -name \"%s\" -parent $sceneGroupName`;\n"
% pgname)
f.write(
"$pointGroupName = ($sceneGroupName + \"|\" + $pointGroupName);\n"
)
# write points...
l = tde4.getPointList(pg)
for p in l:
if tde4.isPointCalculated3D(pg, p):
name = tde4.getPointName(pg, p)
name = "p%s" % validName(name)
p3d = tde4.getPointCalcPosition3D(pg, p)
p3d = convertZup(p3d, yup)
f.write("\n")
f.write("// create point %s...\n" % name)
f.write(
"string $locator = stringArrayToString(`spaceLocator -name %s`, \"\");\n"
% name)
f.write("$locator = (\"|\" + $locator);\n")
f.write(
"xform -t %.15f %.15f %.15f $locator;\n" %
(p3d[0], p3d[1], p3d[2]))
f.write("parent $locator $pointGroupName;\n")
f.write("\n")
scale = tde4.getPGroupScale3D(pg)
f.write(
"xform -zeroTransformPivots -rotateOrder zxy -scale %.15f %.15f %.15f $pointGroupName;\n"
% (scale, scale, scale))
# animate object point group...
f.write("\n")
f.write("// animating point group %s...\n" % pgname)
frame = 1
while frame <= noframes:
# rot/pos...
p3d = tde4.getPGroupPosition3D(pg, camera, frame)
p3d = convertZup(p3d, yup)
r3d = tde4.getPGroupRotation3D(pg, camera, frame)
rot = convertToAngles(r3d)
if frame > 1:
rot = [
angleMod360(rot0[0], rot[0]),
angleMod360(rot0[1], rot[1]),
angleMod360(rot0[2], rot[2])
]
rot0 = rot
f.write(
"setKeyframe -at translateX -t %d -v %.15f $pointGroupName; "
% (frame + frame0, p3d[0]))
f.write(
"setKeyframe -at translateY -t %d -v %.15f $pointGroupName; "
% (frame + frame0, p3d[1]))
f.write(
"setKeyframe -at translateZ -t %d -v %.15f $pointGroupName; "
% (frame + frame0, p3d[2]))
f.write(
"setKeyframe -at rotateX -t %d -v %.15f $pointGroupName; "
% (frame + frame0, rot[0]))
f.write(
"setKeyframe -at rotateY -t %d -v %.15f $pointGroupName; "
% (frame + frame0, rot[1]))
f.write(
"setKeyframe -at rotateZ -t %d -v %.15f $pointGroupName;\n"
% (frame + frame0, rot[2]))
frame += 1
# mocap point groups...
if tde4.getPGroupType(pg) == "MOCAP" and camera != None:
f.write("\n")
f.write("// create mocap point group...\n")
pgname = "objectPGroup_%s_%d_1" % (validName(
tde4.getPGroupName(pg)), index)
index += 1
f.write(
"string $pointGroupName = `group -em -name \"%s\" -parent $sceneGroupName`;\n"
% pgname)
f.write(
"$pointGroupName = ($sceneGroupName + \"|\" + $pointGroupName);\n"
)
# write points...
l = tde4.getPointList(pg)
for p in l:
if tde4.isPointCalculated3D(pg, p):
name = tde4.getPointName(pg, p)
name = "p%s" % validName(name)
p3d = tde4.getPointMoCapCalcPosition3D(
pg, p, camera, 1)
p3d = convertZup(p3d, yup)
f.write("\n")
f.write("// create point %s...\n" % name)
f.write(
"string $locator = stringArrayToString(`spaceLocator -name %s`, \"\");\n"
% name)
f.write("$locator = (\"|\" + $locator);\n")
f.write(
"xform -t %.15f %.15f %.15f $locator;\n" %
(p3d[0], p3d[1], p3d[2]))
for frame in range(1, noframes + 1):
p3d = tde4.getPointMoCapCalcPosition3D(
pg, p, camera, frame)
p3d = convertZup(p3d, yup)
f.write(
"setKeyframe -at translateX -t %d -v %.15f $locator; "
% (frame + frame0, p3d[0]))
f.write(
"setKeyframe -at translateY -t %d -v %.15f $locator; "
% (frame + frame0, p3d[1]))
f.write(
"setKeyframe -at translateZ -t %d -v %.15f $locator; "
% (frame + frame0, p3d[2]))
f.write("parent $locator $pointGroupName;\n")
f.write("\n")
scale = tde4.getPGroupScale3D(pg)
f.write(
"xform -zeroTransformPivots -rotateOrder zxy -scale %.15f %.15f %.15f $pointGroupName;\n"
% (scale, scale, scale))
# animate mocap point group...
f.write("\n")
f.write("// animating point group %s...\n" % pgname)
frame = 1
while frame <= noframes:
# rot/pos...
p3d = tde4.getPGroupPosition3D(pg, camera, frame)
p3d = convertZup(p3d, yup)
r3d = tde4.getPGroupRotation3D(pg, camera, frame)
rot = convertToAngles(r3d)
if frame > 1:
rot = [
angleMod360(rot0[0], rot[0]),
angleMod360(rot0[1], rot[1]),
angleMod360(rot0[2], rot[2])
]
rot0 = rot
f.write(
"setKeyframe -at translateX -t %d -v %.15f $pointGroupName; "
% (frame + frame0, p3d[0]))
f.write(
"setKeyframe -at translateY -t %d -v %.15f $pointGroupName; "
% (frame + frame0, p3d[1]))
f.write(
"setKeyframe -at translateZ -t %d -v %.15f $pointGroupName; "
% (frame + frame0, p3d[2]))
f.write(
"setKeyframe -at rotateX -t %d -v %.15f $pointGroupName; "
% (frame + frame0, rot[0]))
f.write(
"setKeyframe -at rotateY -t %d -v %.15f $pointGroupName; "
% (frame + frame0, rot[1]))
f.write(
"setKeyframe -at rotateZ -t %d -v %.15f $pointGroupName;\n"
% (frame + frame0, rot[2]))
frame += 1
#
# global (scene node) transformation...
p3d = tde4.getScenePosition3D()
p3d = convertZup(p3d, yup)
r3d = tde4.getSceneRotation3D()
rot = convertToAngles(r3d)
s = tde4.getSceneScale3D()
f.write(
"xform -zeroTransformPivots -rotateOrder zxy -translation %.15f %.15f %.15f -scale %.15f %.15f %.15f -rotation %.15f %.15f %.15f $sceneGroupName;\n\n"
%
(p3d[0], p3d[1], p3d[2], s, s, s, rot[0], rot[1], rot[2]))
f.write("\n")
f.close()
# tde4.postQuestionRequester("Export Maya...","Project successfully exported.","Ok")
print '--> successfully exported Maya Mel'
else:
tde4.postQuestionRequester("Export Maya...",
"Error, couldn't open file.", "Ok")
return get_mel_filename()['path']
