def _convert_2d_to_3d_point_undistort(point_group, camera, fbw, fbh, lcox,
lcoy, camera_fov, frame, pos, depth):
"""
Convert a 2D point (undistorted) into a 3D point, in world space.
:param point_group: Camera Point Group for camera.
:param camera: The camera to use for rolling shutter calculations.
:param fbw: Camera's film back width value.
:param fbh: Camera's film back height value.
:param lcox: Camera lens lens center offset X value.
:param lcoy: Camera lens lens center offset Y value.
:param camera_fov: Camera Field of View as list of left, right,
bottom and top.
:param frame: The 2D point's frame number (in internal 3DE frame numbers).
:param pos: Input 2D data.
:param depth: The content distance to calculate rolling shutter at.
:return: Corrected 2D point.
:rtype: vec2d
"""
focal = tde4.getCameraFocalLength(camera, frame)
r3d = vl_sdv.mat3d(tde4.getPGroupRotation3D(point_group, camera, frame))
p3d = vl_sdv.vec3d(tde4.getPGroupPosition3D(point_group, camera, frame))
left, right, bottom, top = camera_fov
p2d = [0, 0]
p2d[0] = (pos[0] - left) / (right - left)
p2d[1] = (pos[1] - bottom) / (top - bottom)
p2d = tde4.removeDistortion2D(camera, frame, p2d)
p2d_cm = vl_sdv.vec2d((p2d[0] - 0.5) * fbw - lcox,
(p2d[1] - 0.5) * fbh - lcoy)
homogeneous_point = r3d * vl_sdv.vec3d(p2d_cm[0], p2d_cm[1], -focal).unit()
out_point = homogeneous_point * depth + p3d
return out_point
if LensDistortionValid:
f.write(
"setAttr ($cameraShape+\".horizontalFilmAperture\") %.15f;\n"
% (CameraApertureWidth * 1.1))
f.write(
"setAttr ($cameraShape+\".verticalFilmAperture\") %.15f;\n"
% (CameraApertureHeight * 1.1))
else:
f.write(
"setAttr ($cameraShape+\".horizontalFilmAperture\") %.15f;\n"
% CameraApertureWidth)
f.write(
"setAttr ($cameraShape+\".verticalFilmAperture\") %.15f;\n"
% CameraApertureHeight)
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")
# image plane...
f.write("\n")
f.write("// create image plane...\n")
f.write("string $imagePlane = `createNode imagePlane`;\n")
def exportCameras(f, cameras, camFrameStart, exportPath, campg, outSize, date):
for cameraIndex, cam in enumerate(cameras):
frameStart = int(camFrameStart[cameraIndex])
frame0 = frameStart - 1
camInfo = tools.TDECamInfo(cam, cameraIndex)
cameraName = camInfo.name
cameraPath = camInfo.cameraPath
projectName = camInfo.projectName
projectPath = camInfo.projectPath
rez_x = camInfo.rez_x
rez_y = camInfo.rez_y
firstFrame = camInfo.firstFrame
lastFrame = camInfo.lastFrame
offset = camInfo.offset
noframes = camInfo.noframes
offset = int(firstFrame) - int(frameStart)
model = camInfo.model
camType = camInfo.camType
model_maya = tools.getLDmodelNukeNodeName(model)
lens = camInfo.lens
if lens is not None:
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_in = -lco_x / 2.54
lco_y_in = -lco_y / 2.54
# convert focal length to mm...
focal = focal * 10.0
# get the focalLenght
# fotalLenght = tde4.getLensFocalLength(lens) # UNUSED
# get the focusDistance
# focus = tde4.getLensFocus(lens) # UNUSED
# create camera
f.write("\n")
f.write("// create camera %s...\n" % cameraName)
f.write(
"string $cameraNodes[] = `camera -name \"camera_%s\" -hfa %.15f -vfa %.15f -fl %.15f -ncp 0.01 -fcp 10000 -shutterAngle 180 -ff \"overscan\"`;\n"
% (cameraName, 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_in)
f.write("setAttr ($cameraShape+\".verticalFilmOffset\") %.15f;\n" %
lco_y_in)
f.write(
'addAttr -ln "distortion_model" -dt "string" $cameraShape;\n')
f.write(
'setAttr -type "string" ($cameraShape + ".distortion_model") %s;\n'
% model_maya)
'''
for para in (tools.getLDmodelParameterList(model)):
f.write('addAttr -ln %s -dt "string" $cameraShape;\n'%tools.getLDmodelMayaParameterName(para));
f.write('setAttr -type "string" ($cameraShape + ".%s") %s;\n'%(tools.getLDmodelMayaParameterName(para),tde4.getLensLDAdjustableParameter(lens, para, fotalLenght, focus)))
'''
f.write(
'addAttr -ln "pixel_aspect_ratio" -at "float" $cameraShape;\n')
f.write('setAttr ($cameraShape + ".pixel_aspect_ratio") %s;\n' %
p_aspect)
'''
'f.write('addAttr -ln "plate_resolution_x" -at "long" $cameraShape;\n');
f.write('setAttr ($cameraShape + ".plate_resolution_x") %s;\n'%rez_x)
f.write('addAttr -ln "plate_resolution_y" -at "long" $cameraShape;\n');
f.write('setAttr ($cameraShape + ".plate_resolution_y") %s;\n'%rez_y)
'''
f.write('addAttr -ln "frameOffset" -at "long" $cameraShape;\n')
f.write("setAttr ($cameraShape + \".frameOffset\") %s;\n" % offset)
f.write('addAttr -ln "lcoX" -at "double" $cameraShape;\n')
f.write("setAttr ($cameraShape + \".lcoX\") %s;\n" % lco_x)
f.write('addAttr -ln "lcoY" -at "double" $cameraShape;\n')
f.write("setAttr ($cameraShape + \".lcoY\") %s;\n" % lco_y)
f.write('addAttr -ln "lcoXin" -at "double" $cameraShape;\n')
f.write("setAttr ($cameraShape + \".lcoXin\") %s;\n" % lco_x_in)
f.write('addAttr -ln "lcoYin" -at "double" $cameraShape;\n')
f.write("setAttr ($cameraShape + \".lcoYin\") %s;\n" % lco_y_in)
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")
# image plane...
f.write("\n")
f.write("// create image plane...\n")
f.write("string $imagePlanes[] = `imagePlane`;\n")
f.write("string $imagePlane = $imagePlanes[1];\n")
f.write("select $imagePlane;\n")
f.write("string $imagePlaneTransfo[] = `pickWalk -d up`;\n")
f.write("select -cl;\n")
f.write("cameraImagePlaneUpdate ($cameraShape, $imagePlane);\n")
f.write("setAttr ($imagePlane + \".frameOffset\") %s;\n" % offset)
f.write("setAttr ($imagePlane + \".offsetX\") %.15f;\n" % lco_x)
f.write("setAttr ($imagePlane + \".offsetY\") %.15f;\n" % lco_y)
f.write("setAttr ($imagePlane + \".type\") 1;\n")
f.write("parent $imagePlaneTransfo[0] $cameraTransform;\n")
path = tde4.getCameraPath(cam)
sattr = tde4.getCameraSequenceAttr(cam)
if camType == "SEQUENCE":
f.write("setAttr ($imagePlane+\".useFrameExtension\") 1;\n")
f.write(
"expression -n \"frame_ext_expression\" -s($imagePlane+\".frameExtension=frame\");\n"
)
pathPrepared = prepareImagePath(path, sattr[0])
else:
f.write("setAttr ($imagePlane+\".useFrameExtension\") 0;\n")
pathPrepared = path.replace("\\", "/")
f.write(
"setAttr ($imagePlane + \".imageName\") -type \"string\" \"%s\";\n"
% (pathPrepared))
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" % cameraName)
f.write("playbackOptions -min %d -max %d;\n" %
(1 + frame0, noframes + frame0))
f.write("\n")
frame = 1
rot0 = None
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
# set Distortion
f.write('addAttr -ln "distortion" -at "bool" $cameraShape;\n')
if tools.cameraHasDistortion(cam):
f.write('setAttr ($cameraShape + ".distortion") 1;\n')
else:
f.write('setAttr ($cameraShape + ".distortion") 0;\n')
# change params as in old export to nuke
# create new ones
f.write(
'addAttr -ln "first_tracked_frame" -at long -dv 0 $cameraShape;\n')
f.write(
'addAttr -ln "last_tracked_frame" -at long -dv 0 $cameraShape;\n')
f.write('addAttr -ln "colorspace" -dt "string" $cameraShape;\n')
f.write('addAttr -ln "RenderRez_X" -at long -dv 0 $cameraShape;\n')
f.write('addAttr -ln "RenderRez_Y" -at long -dv 0 $cameraShape;\n')
f.write('addAttr -ln "disto_comp" -dt "string" $cameraShape;\n')
# set values
minFrame = firstFrame
f.write('int $minFrame = ' + str(minFrame) + '; \n')
f.write('setAttr ($cameraShape + ".first_tracked_frame") $minFrame;\n')
maxFrame = lastFrame
f.write('int $maxFrame = ' + str(maxFrame) + '; \n')
f.write('setAttr ($cameraShape + ".last_tracked_frame") $maxFrame;\n')
# if firstFrame == lastFrame:
if 'labo/log' in cameraPath:
colorspace = getdata.getRootColorSpace(projectName)
else:
colorspace = 'sRGB'
f.write('string $colorspace = "' + colorspace + '"; \n')
f.write(
'setAttr -type "string" ($cameraShape + ".colorspace") $colorspace;\n'
)
renderOutX = str(max(int(rez_x), int(outSize[cameraIndex].width)))
renderOutY = str(max(int(rez_y), int(outSize[cameraIndex].height)))
f.write('int $renderOutX = ' + str(renderOutX) + '; \n')
# TODO if positive disto is other value
f.write('setAttr ($cameraShape + ".RenderRez_X") $renderOutX; \n')
# TODO if positive disto is other value
f.write('int $renderOutY = ' + str(renderOutY) + '; \n')
f.write('setAttr ($cameraShape + ".RenderRez_Y") $renderOutY; \n')
f.write('string $exportFileName = "' + exportPath.replace("\\", "/") +
'_undisto/' + cameraName + '_undisto.nk";\n')
f.write(
'setAttr -type "string"($cameraShape + ".disto_comp") $exportFileName; \n'
)
# other values
coefX = 1.0 * int(renderOutX) / rez_x
coefY = 1.0 * int(renderOutY) / rez_y
oldhfa = fback_w
oldvfa = fback_h
newHfa = str(coefX * oldhfa)
newVfa = str(coefY * oldvfa)
f.write('setAttr ($cameraShape + ".hfa") ' + newHfa + '; \n')
f.write('setAttr ($cameraShape + ".vfa") ' + newVfa + '; \n')
f.write(
'string $initIplane[] = `listConnections -type "imagePlane" $cameraShape`; \n'
)
f.write('string $camIplaneShape = $initIplane[0]; \n')
f.write('setAttr ($camIplaneShape + ".sizeX") ' + newHfa + '; \n')
f.write('setAttr ($camIplaneShape + ".sizeY") ' + newVfa + '; \n')
undistoFilePath = camInfo.getUndistoImagePath(app='maya', date=date)
f.write('setAttr -type "string" ($camIplaneShape + ".imageName") "' +
undistoFilePath + '"; \n')
f.write('setAttr ($camIplaneShape + ".type") 0; \n')
# camera rendeable
f.write('setAttr ($cameraShape + ".renderable") 1; \n')
# keys in some params
touchParam(f, '$cameraShape + ".hfa"', insertKey=True, lock=False)
touchParam(f, '$cameraShape + ".vfa"', insertKey=True, lock=False)
pixelAspectRatioParamName = '$cameraShape + ".pixel_aspect_ratio"'
touchParam(f, pixelAspectRatioParamName, insertKey=True, lock=True)
renderRezXparamName = '$cameraShape + ".RenderRez_X"'
touchParam(f, renderRezXparamName, insertKey=True, lock=True)
renderRezTparamName = '$cameraShape + ".RenderRez_Y"'
touchParam(f, renderRezTparamName, insertKey=True, lock=True)
touchParam(f, '$cameraShape + ".distortion"')
touchParam(f, '$cameraShape + ".first_tracked_frame"', insertKey=False)
touchParam(f, '$cameraShape + ".last_tracked_frame"', insertKey=False)
touchParam(f, '$cameraShape + ".colorspace"', insertKey=False)
touchParam(f, '$cameraShape + ".frameOffset"', insertKey=False)
# set render globals
f.write('setAttr "defaultResolution.width" ' + renderOutX + '; \n')
f.write('setAttr "defaultResolution.height" ' + renderOutY + '; \n')
# delete old params
f.write('deleteAttr ($cameraShape + ".distortion_model"); \n')
f.write('deleteAttr ($cameraShape + ".lcoX"); \n')
f.write('deleteAttr ($cameraShape + ".lcoY"); \n')
f.write('deleteAttr ($cameraShape + ".lcoXin"); \n')
f.write('deleteAttr ($cameraShape + ".lcoYin"); \n')
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 _remove_rs_from_2d_point(point_group, camera, frame, input_2d, depth):
"""
Correct Rolling Shutter for the given input_2d point data, on frame.
:param point_group: Camera Point Group for camera.
:param camera: The camera to use for rolling shutter calculations.
:param frame: The 2D point's frame number (in internal 3DE frame numbers).
:param input_2d: Input 2D data.
:param depth: The content distance to calculate rolling shutter at.
:return: 2D point with corrected position.
:rtype: [float, float]
"""
assert isinstance(input_2d, vl_sdv.vec2d)
num_frames = tde4.getCameraNoFrames(camera)
if num_frames == 1:
return input_2d
# Static camera and lens values.
camera_fps = tde4.getCameraFPS(camera)
camera_fov = tde4.getCameraFOV(camera)
lens = tde4.getCameraLens(camera)
fbw = tde4.getLensFBackWidth(lens)
fbh = tde4.getLensFBackHeight(lens)
lcox = tde4.getLensLensCenterX(lens)
lcoy = tde4.getLensLensCenterY(lens)
rs_time_shift = tde4.getCameraRollingShutterTimeShift(camera)
rs_value = rs_time_shift * camera_fps
# Sample at previous frame
prev_pos = vl_sdv.vec3d(0, 0, 0)
prev_frame = frame - 1
if frame > 1:
prev_pos = _convert_2d_to_3d_point_undistort(
point_group, camera,
fbw, fbh, lcox, lcoy,
camera_fov,
prev_frame, input_2d, depth)
# Sample at next frame
next_pos = vl_sdv.vec3d(0, 0, 0)
next_frame = frame + 1
if frame < num_frames:
next_pos = _convert_2d_to_3d_point_undistort(
point_group, camera,
fbw, fbh, lcox, lcoy,
camera_fov,
next_frame, input_2d, depth)
# Sample at current frame
curr_pos = _convert_2d_to_3d_point_undistort(
point_group, camera,
fbw, fbh, lcox, lcoy,
camera_fov,
frame, input_2d, depth)
# Blend previous, next and current frame values based on the
# position of the 2D point vertical position and the rolling
# shutter value.
if frame == 1:
prev_pos = curr_pos + (curr_pos - next_pos)
if frame == num_frames:
next_pos = curr_pos + (curr_pos - prev_pos)
t = rs_value * (1.0 - input_2d[1])
curr_pos = _apply_rs_correction(-t, prev_pos, curr_pos, next_pos)
# Back-projection
focal = tde4.getCameraFocalLength(camera, frame)
r3d = vl_sdv.mat3d(tde4.getPGroupRotation3D(point_group, camera, frame))
p3d = vl_sdv.vec3d(tde4.getPGroupPosition3D(point_group, camera, frame))
d = r3d.trans() * (curr_pos - p3d)
p2d = [0, 0]
p2d[0] = (d[0] * focal / (-d[2] * fbw)) + (lcox / fbw) + 0.5
p2d[1] = (d[1] * focal / (-d[2] * fbh)) + (lcoy / fbh) + 0.5
p = tde4.applyDistortion2D(camera, frame, p2d)
left, right, bottom, top = camera_fov
p = vl_sdv.vec2d((p[0] * (right - left)) + left,
(p[1] * (top - bottom)) + bottom)
v = (input_2d + (input_2d - p)).list()
return v
def get_position_3d(self, cam, frame):
return tde4.getPGroupPosition3D(self._pg_id, cam.id_, frame)
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']