def _query_selection_state():
"""
Query the current selection state.
Get the active point group, selected camera and connected lens.
:returns: A tuple of Point Group ID, Camera ID and Lens ID. Any of
the tuple members may be None if it could not be found.
:rtype: (str or None, str or None, str or None)
"""
cam_id = None
lens_id = None
msg = None
pgroup_id = tde4.getCurrentPGroup()
if not pgroup_id:
msg = 'Error: Could not get active Point Group for camera.'
else:
with_selection = 1
cam_ids = tde4.getCameraList(with_selection)
if len(cam_ids) == 1:
cam_id = cam_ids[0]
lens_id = tde4.getCameraLens(cam_id)
if not lens_id:
msg = 'Error: Selected camera does not have a connected lens.'
else:
if len(cam_ids) == 0:
msg = 'Error: Please select a camera.'
elif len(cam_ids) > 1:
msg = 'Error: Please select only one camera.'
tde4.postQuestionRequester(TITLE, msg, 'Ok')
if msg is not None:
tde4.postQuestionRequester(TITLE, msg, 'Ok')
return pgroup_id, cam_id, lens_id
def cameraHasDistortion(camera):
if isClassicModel(camera):
return cameraHasDistortionClassic(camera)
else:
lens = tde4.getCameraLens(camera)
model = getCameraModel(camera)
for parameterName in getLDmodelParameterList(model):
defaultValue = tde4.getLDModelParameterDefault(
model, parameterName)
# distorsionMode
dyndistmode = tde4.getLensDynamicDistortionMode(lens)
if dyndistmode == "DISTORTION_STATIC":
actualValue = tde4.getLensLDAdjustableParameter(
lens, parameterName, 1, 1)
if defaultValue != actualValue:
return True
else:
num_frames = tde4.getCameraNoFrames(camera)
for frame in range(1, num_frames + 1):
focal = tde4.getCameraFocalLength(camera, frame)
focus = tde4.getCameraFocus(camera, frame)
actualValue = tde4.getLensLDAdjustableParameter(
lens, parameterName, focal, focus)
if defaultValue != actualValue:
return True
return False
def supportedCameraMode(camera):
lens = tde4.getCameraLens(camera)
model = tde4.getLensLDModel(lens)
return model in [
'3DE Classic LD Model',
'3DE4 Radial - Fisheye, Degree 8',
'3DE4 Radial - Standard, Degree 4',
'3DE4 Anamorphic - Standard, Degree 4',
'3DE4 Anamorphic, Degree 6',
]
def cameraHasDistortionClassic(camera):
cameraLens = tde4.getCameraLens(camera)
dpClassic = getDistorsionParametersFromLenClassic(cameraLens, '', 1)
if float(dpClassic.Distortion) == 0 and\
float(dpClassic.Anamorphic_Squeeze) == 1 and\
float(dpClassic.Curvature_X) == 0 and\
float(dpClassic.Curvature_Y) == 0 and\
float(dpClassic.Quartic_Distortion) == 0:
return False
else:
return True
def getOutSize(req, cameraIndex, cameraTmp):
# get the actual image size from cameraTmp
imageHeightInTmp = tde4.getCameraImageHeight(cameraTmp)
imageWidthInTmp = tde4.getCameraImageWidth(cameraTmp)
lensTmp = tde4.getCameraLens(cameraTmp)
folcalParameter = getFocalParameters(lensTmp)
# distorsionMode
dyndistmode = tde4.getLensDynamicDistortionMode(lensTmp)
log.debug('dyndistmode : ' + str(dyndistmode))
if dyndistmode == "DISTORTION_STATIC":
lensModelParameter = getLensModelParameter(camera=None,
lens=lensTmp,
frame=None)
resOutTmp = get_bounding_box.calculateBoundigBox(
imageWidthInTmp, imageHeightInTmp, "undistort", folcalParameter,
lensModelParameter)
imageWidthOuttmp = resOutTmp.width
imageHeightOuttmp = resOutTmp.height
else:
imageHeightOuttmp = imageHeightInTmp
imageWidthOuttmp = imageWidthInTmp
for frame in range(1, tde4.getCameraNoFrames(cameraTmp) + 1):
lensModelParameter = getLensModelParameter(camera=cameraTmp,
lens=lensTmp,
frame=frame)
resOuttmp = get_bounding_box.calculateBoundigBox(
imageWidthInTmp, imageHeightInTmp, "undistort",
folcalParameter, lensModelParameter)
if imageHeightOuttmp < resOuttmp.height:
imageHeightOuttmp = resOuttmp.height
if imageWidthOuttmp < resOuttmp.width:
imageWidthOuttmp = resOuttmp.width
if imageHeightInTmp > imageHeightOuttmp:
imageHeightOuttmp = imageHeightInTmp
if imageWidthInTmp > imageWidthOuttmp:
imageWidthOuttmp = imageWidthInTmp
log.debug('****************** lensModelParameter ********************')
for key in lensModelParameter.keys():
log.debug(key + ' : ' + str(lensModelParameter.get(key)))
return imageHeightOuttmp, imageWidthOuttmp
def getInfoCam(self):
cam = self.cam
index = self.index
self.offset = tde4.getCameraFrameOffset(cam)
self.cameraPath = tde4.getCameraPath(cam).replace('\\', '/')
self.camType = tde4.getCameraType(cam)
self.noframes = tde4.getCameraNoFrames(cam)
self.lens = tde4.getCameraLens(cam)
self.rez_x = tde4.getCameraImageWidth(cam)
self.rez_y = tde4.getCameraImageHeight(cam)
self.model = tde4.getLensLDModel(self.lens)
self.firstFrame = tde4.getCameraSequenceAttr(cam)[0]
self.name = "%s_%s_1" % (validName(tde4.getCameraName(cam)), index)
self.lastFrame = '%d' % (self.firstFrame + self.noframes - 1)
self.firstFrame = '%d' % self.firstFrame
self.focal_cm = tde4.getCameraFocalLength(cam, 1)
def _set_camera_lens_filmback(self, camera_id):
# get filmback from camera_preferences file
fields = self.parent.engine.context.as_template_fields()
cam_preferences = preferences.Preferences(
pref_file_name='camera_preferences.yaml',
role=fields.get("Step"),
seq_override=fields.get("Sequence"),
shot_override=fields.get("Shot"))
try:
primary_cam = cam_preferences['camera_type']
# setLensFBackHeight() expects cm
filmback_height = cam_preferences[primary_cam +
"_def"]["vertical_aperture"] / 10
except KeyError as ke:
warning_message = "Unable to load camera preferences (filmback)." \
"\nKeyError: {}".format(ke)
self._warn_with_pop_up("Filmback not set", warning_message)
else:
lens_id = tde4.getCameraLens(camera_id)
tde4.setLensFBackHeight(lens_id, filmback_height)
cam = tde4.getCurrentCamera()
offset = tde4.getCameraFrameOffset(cam)
tde4.setWidgetValue(req, "startframe_field", str(offset))
FootageWidth = tde4.getCameraImageWidth(cam)
FootageHeight = tde4.getCameraImageHeight(cam)
lens = tde4.getFirstLens()
CameraApertureWidth = tde4.getLensFBackWidth(lens) / 2.54
CameraApertureHeight = tde4.getLensFBackHeight(lens) / 2.54
#about lens distortion.
LensDistortionValid = False
camList = tde4.getCameraList()
for cam in camList:
lens = tde4.getCameraLens(cam)
for id in tde4.getLensList():
focus = tde4.getLensFocalLength(id)
lensmodel = tde4.getLensLDModel(id)
focallength = tde4.getLensFocalLength(id)
paraNumber = tde4.getLDModelNoParameters(lensmodel)
for i in range(paraNumber):
paraName = tde4.getLDModelParameterName(lensmodel, i)
currentValue = tde4.getLensLDAdjustableParameter(
id, paraName, focus, focallength)
defaultValue = tde4.getLDModelParameterDefault(lensmodel, paraName)
if currentValue != defaultValue:
LensDistortionValid = True
break
ret = tde4.postCustomRequester(req, "Export Maya (MEL-Script)...", 600, 0,
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 _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
# 3 means user-defined. In this case the GUI allows to enter a frame value. if val == 3: tde4.setWidgetSensitiveFlag(nuke_node_req,"text_initial_frame_nuke",True) elif val == 2: tde4.setWidgetSensitiveFlag(nuke_node_req,"text_initial_frame_nuke",False) tde4.setWidgetValue(nuke_node_req,"text_initial_frame_nuke",str(initial_frame_3de4)) elif val == 1: tde4.setWidgetSensitiveFlag(nuke_node_req,"text_initial_frame_nuke",False) tde4.setWidgetValue(nuke_node_req,"text_initial_frame_nuke",str(tde4.getCameraSequenceAttr(id_cam)[0])) # main try: # camera and lens id_cam = tde4.getCurrentCamera() id_lens = tde4.getCameraLens(id_cam) # check if variable 'nuke_path' exists already or is of type None try: nuke_path except: nuke_path = '' if nuke_path is None: nuke_path = '' # File widget nuke_node_req = tde4.createCustomRequester() tde4.addFileWidget(nuke_node_req,'file_nuke_path', 'Filename: ', '*.nk', nuke_path) # Get smart start value for initial frame option menu. try:
def main(melscript):
"""
melscript: melscript full path to save nuke node next to id
"""
try:
camlist = tde4.getCameraList(1)
if not camlist:
raise Exception(' Only selected cameras will be exported ')
for cam in camlist:
offset = (tde4.getCameraSequenceAttr(cam)[0]) - 1
lens = tde4.getCameraLens(cam)
# check if variable 'nuke_path' exists already or is of type None
try:
nuke_path
except:
nuke_path = ''
if nuke_path is None:
nuke_path = ''
# open requester
# nuke_node_req = tde4.createCustomRequester()
# tde4.addFileWidget(nuke_node_req, 'userInput', 'Filename: ', '*.nk', nuke_path)
# if tde4.getCameraZoomingFlag(cam) and offset != 0:
# tde4.addToggleWidget(nuke_node_req, "stfr_menu", ("Offset curves to frame " + str(offset+1)+": "), 1)
# ret = tde4.postCustomRequester(nuke_node_req,' Export nuke distortion node for camera '+tde4.getCameraName(cam)+' ',700,110,' Ok ',' Cancel ')
# if ret != 1:
# raise CancelException('Cancelled')
# nuke_path = tde4.getWidgetValue(nuke_node_req, 'userInput')
nuke_path = melscript.replace('mel', 'nk')
# check path and suffix
if not nuke_path:
raise Exception(' No path entered ')
if not nuke_path.endswith('.nk'):
nuke_path = nuke_path + '.nk'
### THIS IS IMPORTANT FOR ZOOMCURVES:
### SET OFFSET TO 0 ALL THE TIME.
# check if offset should be applied or not if there is one
# if tde4.getCameraZoomingFlag(cam) and offset != 0:
# n = tde4.getWidgetValue(nuke_node_req,"stfr_menu")
# if n != 1:
# offset = 0
# offset = 0
temp_cam = TDE4Wrapper()
offset, kk, ii = temp_cam.frange
# export
# if ret == 1:
print '------------------ Export tde4 Nuke Distortion Node ------------------'
exportNukeDewarpNode(cam, offset, nuke_path)
print 'file:', nuke_path, '\n'
except CancelException, e:
print e
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']
def lens(self):
return TDELens(self, tde4.getCameraLens(self._cam_id))
# 3 means user-defined. In this case the GUI allows to enter a frame value. if val == 3: tde4.setWidgetSensitiveFlag(nuke_node_req,"text_initial_frame_nuke",True) elif val == 2: tde4.setWidgetSensitiveFlag(nuke_node_req,"text_initial_frame_nuke",False) tde4.setWidgetValue(nuke_node_req,"text_initial_frame_nuke",str(initial_frame_3de4)) elif val == 1: tde4.setWidgetSensitiveFlag(nuke_node_req,"text_initial_frame_nuke",False) tde4.setWidgetValue(nuke_node_req,"text_initial_frame_nuke",str(tde4.getCameraSequenceAttr(id_cam)[0])) # main try: # camera and lens id_cam = tde4.getCurrentCamera() id_lens = tde4.getCameraLens(id_cam) # check if variable 'nuke_path' exists already or is of type None try: nuke_path except: nuke_path = '' if nuke_path is None: nuke_path = '' # File widget nuke_node_req = tde4.createCustomRequester() tde4.addFileWidget(nuke_node_req,'file_nuke_path', 'Filename: ', '*.nk', nuke_path) # Get smart start value for initial frame option menu. try:
def exportNukeDewarpNode(cam, offset, nuke_path):
lens = tde4.getCameraLens(cam)
model = tde4.getLensLDModel(lens)
num_frames = tde4.getCameraNoFrames(cam)
w_fb_cm = tde4.getLensFBackWidth(lens)
h_fb_cm = tde4.getLensFBackHeight(lens)
lco_x_cm = tde4.getLensLensCenterX(lens)
lco_y_cm = tde4.getLensLensCenterY(lens)
pxa = tde4.getLensPixelAspect(lens)
# xa,xb,ya,yb in unit coordinates, in this order.
fov = tde4.getCameraFOV(cam)
print 'camera: ', tde4.getCameraName(cam)
print 'offset:', offset
print 'lens:', tde4.getLensName(lens)
print 'model: ', model
f = open(nuke_path, "w")
try:
f.write(
'# Created by 3DEqualizer4 using Export Nuke Distortion Nodes export script\n'
)
f.write("LD" + nukify_name(model) + ' {\n')
f.write(' direction undistort\n')
# write focal length curve if dynamic
if tde4.getCameraZoomingFlag(cam):
print 'dynamic focal length'
f.write(' tde4_focal_length_cm {{curve ')
for frame in range(1, num_frames + 1):
f.write('x%i' % (frame + offset))
f.write(' %.7f ' % tde4.getCameraFocalLength(cam, frame))
f.write('}}\n')
# write static focal length else
else:
print 'static focal length'
f.write(' tde4_focal_length_cm %.7f \n' %
tde4.getCameraFocalLength(cam, 1))
# write focus distance curve if dynamic
try:
if tde4.getCameraFocusMode(cam) == "FOCUS_DYNAMIC":
print 'dynamic focus distance'
f.write(' tde4_custom_focus_distance_cm {{curve ')
for frame in range(1, num_frames + 1):
f.write('x%i' % (frame + offset))
f.write(' %.7f ' % tde4.getCameraFocus(cam, frame))
f.write('}}\n')
except:
# For 3DE4 Release 1:
pass
# write static focus distance else
else:
print 'static focus distance'
try:
f.write(' tde4_custom_focus_distance_cm %.7f \n' %
tde4.getCameraFocus(cam, 1))
except:
# For 3DE4 Release 1:
f.write(' tde4_custom_focus_distance_cm 100.0 \n')
# write camera
f.write(' tde4_filmback_width_cm %.7f \n' % w_fb_cm)
f.write(' tde4_filmback_height_cm %.7f \n' % h_fb_cm)
f.write(' tde4_lens_center_offset_x_cm %.7f \n' % lco_x_cm)
f.write(' tde4_lens_center_offset_y_cm %.7f \n' % lco_y_cm)
f.write(' tde4_pixel_aspect %.7f \n' % pxa)
f.write(' field_of_view_xa_unit %.7f \n' % fov[0])
f.write(' field_of_view_ya_unit %.7f \n' % fov[2])
f.write(' field_of_view_xb_unit %.7f \n' % fov[1])
f.write(' field_of_view_yb_unit %.7f \n' % fov[3])
# write distortion parameters
#
# dynamic distortion
try:
dyndistmode = tde4.getLensDynamicDistortionMode(lens)
except:
# For 3DE4 Release 1:
if tde4.getLensDynamicDistortionFlag(lens) == 1:
dyndistmode = "DISTORTION_DYNAMIC_FOCAL_LENGTH"
else:
dyndistmode = "DISTORTION_STATIC"
old_api = True
try:
tde4.getLensLDAdjustableParameter(lens, para, 1)
except:
old_api = False
if old_api:
if dyndistmode == "DISTORTION_DYNAMIC_FOCAL_LENGTH":
print 'dynamic lens distortion, focal length'
# dynamic focal length (zoom)
for para in (getLDmodelParameterList(model)):
f.write(' ' + nukify_name(para) + ' {{curve ')
for frame in range(1, num_frames + 1):
focal = tde4.getCameraFocalLength(cam, frame)
f.write('x%i' % (frame + offset))
f.write(' %.7f ' % tde4.getLensLDAdjustableParameter(
lens, para, focal))
f.write('}}\n')
if dyndistmode == "DISTORTION_DYNAMIC_FOCUS_DISTANCE":
print 'dynamic lens distortion, focus distance'
# dynamic focus distance
for para in (getLDmodelParameterList(model)):
f.write(' ' + nukify_name(para) + ' {{curve ')
for frame in range(1, num_frames + 1):
# Older Releases do not have Focus-methods.
try:
focus = tde4.getCameraFocus(cam, frame)
except:
focus = 100.0
f.write('x%i' % (frame + offset))
f.write(' %.7f ' % tde4.getLensLDAdjustableParameter(
lens, para, focus))
f.write('}}\n')
# static distortion
if dyndistmode == "DISTORTION_STATIC":
print 'static lens distortion'
for para in (getLDmodelParameterList(model)):
f.write(' ' + nukify_name(para) + ' %.7f \n' %
tde4.getLensLDAdjustableParameter(lens, para, 1))
else:
# new API
if dyndistmode == "DISTORTION_STATIC":
print 'static lens distortion'
for para in (getLDmodelParameterList(model)):
f.write(
' ' + nukify_name(para) + ' %.7f \n' %
tde4.getLensLDAdjustableParameter(lens, para, 1, 1))
else:
print 'dynamic lens distortion,'
# dynamic
for para in (getLDmodelParameterList(model)):
f.write(' ' + nukify_name(para) + ' {{curve ')
for frame in range(1, num_frames + 1):
focal = tde4.getCameraFocalLength(cam, frame)
focus = tde4.getCameraFocus(cam, frame)
f.write('x%i' % (frame + offset))
f.write(' %.7f ' % tde4.getLensLDAdjustableParameter(
lens, para, focal, focus))
f.write('}}\n')
f.write(' name LD_3DE4_' + decode_entities(tde4.getCameraName(cam)) +
'\n')
f.write('}\n')
finally:
f.close()
def exportNukeDewarpNode(id_cam,offset,nuke_path):
id_lens = tde4.getCameraLens(id_cam)
model = tde4.getLensLDModel(id_lens)
num_frames = tde4.getCameraNoFrames(id_cam)
w_fb_cm = tde4.getLensFBackWidth(id_lens)
h_fb_cm = tde4.getLensFBackHeight(id_lens)
lco_x_cm = tde4.getLensLensCenterX(id_lens)
lco_y_cm = tde4.getLensLensCenterY(id_lens)
pxa = tde4.getLensPixelAspect(id_lens)
# xa,xb,ya,yb in unit coordinates, in this order.
xa_unit,xb_unit,ya_unit,yb_unit = tde4.getCameraFOV(id_cam)
f = open(nuke_path,"w")
try:
f.write('# Created by 3DEqualizer4 using Export Nuke Distortion Nodes export script\n')
f.write("LD" + nukify_name(model) + ' {\n')
f.write(' direction undistort\n')
################################
# focal length #
################################
if is_focal_length_dynamic(id_cam):
# write focal length curve if dynamic
# print 'dynamic focal length'
f.write(' tde4_focal_length_cm {{curve ')
for frame in range(num_frames):
# Internally, frames start at 1.
focal = tde4.getCameraFocalLength(id_cam,frame + 1)
f.write ('x%i %.7f ' % (frame + offset,focal))
f.write('}}\n')
else:
# write static focal length otherwise
# print 'static focal length'
f.write(' tde4_focal_length_cm %.7f \n' % tde4.getCameraFocalLength(id_cam,1))
################################
# focus distance #
################################
# For Release 1 this function return False, so no problem with getCameraFocus.
if is_focus_distance_dynamic(id_cam):
# write focus distance curve if dynamic
# print 'dynamic focus distance'
f.write(' tde4_custom_focus_distance_cm {{curve ')
for frame in range(num_frames):
# Internally, frames start at 1.
focus = tde4.getCameraFocus(id_cam,frame + 1)
f.write ('x%i %.7f ' % (frame + offset,focus))
f.write('}}\n')
else:
try:
# write static focus distance otherwise
f.write(' tde4_custom_focus_distance_cm %.7f \n' % tde4.getCameraFocus(id_cam,1))
except:
# For Release 1 we simply write out the default value to Nuke.
f.write(' tde4_custom_focus_distance_cm 100.0 \n')
################################
# built-in parameters #
################################
# the remaining five built-in parameters
f.write(' tde4_filmback_width_cm %.7f \n' % w_fb_cm)
f.write(' tde4_filmback_height_cm %.7f \n' % h_fb_cm)
f.write(' tde4_lens_center_offset_x_cm %.7f \n' % lco_x_cm)
f.write(' tde4_lens_center_offset_y_cm %.7f \n' % lco_y_cm)
f.write(' tde4_pixel_aspect %.7f \n' % pxa)
################################
# field-of-view #
################################
f.write(' field_of_view_xa_unit %.7f \n' % xa_unit)
f.write(' field_of_view_xb_unit %.7f \n' % xb_unit)
f.write(' field_of_view_ya_unit %.7f \n' % ya_unit)
f.write(' field_of_view_yb_unit %.7f \n' % yb_unit)
# write distortion parameters
#
# dynamic distortion
dyndistmode = get_dynamic_distortion_mode(id_lens)
old_api = True
try:
for para in getLDmodelParameterList(model):
tde4.getLensLDAdjustableParameter(id_lens, para, 1)
break
except:
old_api = False
if old_api:
# dynamic focal length (zoom)
if dyndistmode=="DISTORTION_DYNAMIC_FOCAL_LENGTH":
# print 'dynamic lens distortion, focal length'
for para in getLDmodelParameterList(model):
f.write(' ' + nukify_name(para) + ' {{curve ')
for frame in range(num_frames):
# Internally, frames start at 1.
focal = tde4.getCameraFocalLength(id_cam,frame + 1)
f.write ('x%i %.7f ' % (frame + offset,tde4.getLensLDAdjustableParameter(id_lens,para,focal)))
f.write('}}\n')
# dynamic focus distance
if dyndistmode=="DISTORTION_DYNAMIC_FOCUS_DISTANCE":
# print 'dynamic lens distortion, focus distance'
for para in getLDmodelParameterList(model):
f.write(' ' + nukify_name(para) + ' {{curve ')
for frame in range(num_frames):
# Older Releases do not have Focus-methods.
try:
# Internally, frames start at 1.
focus = tde4.getCameraFocus(id_cam,frame + 1)
except:
focus = 100.0
f.write('x%i %.7f ' % (frame + offset,tde4.getLensLDAdjustableParameter(id_lens,para,focus)))
f.write('}}\n')
# static distortion
if dyndistmode=="DISTORTION_STATIC":
# print 'static lens distortion'
for para in getLDmodelParameterList(model):
f.write(' ' + nukify_name(para) + ' %.7f \n'%tde4.getLensLDAdjustableParameter(id_lens,para,1))
else:
# new API
if dyndistmode=="DISTORTION_STATIC":
# print 'static lens distortion'
for para in getLDmodelParameterList(model):
f.write(' ' + nukify_name(para) + ' %.7f \n'%tde4.getLensLDAdjustableParameter(id_lens,para,1,1))
else:
# print 'dynamic lens distortion,'
# dynamic
for para in getLDmodelParameterList(model):
f.write(' ' + nukify_name(para) + ' {{curve ')
for frame in range(num_frames):
# Internally, frames start at 1.
focal = tde4.getCameraFocalLength(id_cam,frame + 1)
focus = tde4.getCameraFocus(id_cam,frame + 1)
f.write('x%i %.7f ' % (frame + offset,tde4.getLensLDAdjustableParameter(id_lens,para,focal,focus)))
# print "%i[3DE4] -> %i[Nuke]" % (frame + tde4.getCameraFrameOffset(id_cam),frame + offset)
f.write('}}\n')
f.write(' name LD_3DE4_' + decode_entities(tde4.getCameraName(id_cam)) + '\n')
f.write('}\n')
finally:
f.close()
def generateNukeNode(cam, direction, offset=0, index=0):
lens = tde4.getCameraLens(cam)
model = tde4.getLensLDModel(lens)
num_frames = tde4.getCameraNoFrames(cam)
w_fb_cm = tde4.getLensFBackWidth(lens)
h_fb_cm = tde4.getLensFBackHeight(lens)
lco_x_cm = tde4.getLensLensCenterX(lens)
lco_y_cm = tde4.getLensLensCenterY(lens)
pxa = tde4.getLensPixelAspect(lens)
# xa,xb,ya,yb in unit coordinates, in this order.
fov = tde4.getCameraFOV(cam)
print 'camera: ', tde4.getCameraName(cam)
print 'offset:', offset
print 'lens:', tde4.getLensName(lens)
print 'model: ', model
nukeNode = []
nukeNode.append('# Created by 3DEqualizer4 ')
nukeNode.append('# using Export Nuke Distortion Nodes export script')
nukeNode.append("LD" + tools.nukify_name(model) + ' {')
nukeNode.append(' direction ' + direction)
# write focal length curve if dynamic
if tde4.getCameraZoomingFlag(cam):
print 'dynamic focal length'
dynFocalLength = []
for frame in range(1, num_frames + 1):
dynFocalLength.append('x%i' % (frame + offset))
dynFocalLength.append(' %.7f ' %
tde4.getCameraFocalLength(cam, frame))
focalLenghtStr = "".join(dynFocalLength)
nukeNode.append(' tde4_focal_length_cm {{curve ' + focalLenghtStr +
' }}')
# write static focal length else
else:
print 'static focal length'
focalLenghtStr = ' tde4_focal_length_cm %.7f '
focalLenghtStr = focalLenghtStr % tde4.getCameraFocalLength(cam, 1)
nukeNode.append(focalLenghtStr)
# write focus distance curve if dynamic
try:
if tde4.getCameraFocusMode(cam) == "FOCUS_DYNAMIC":
print 'dynamic focus distance'
dynFocusDistance = []
for frame in range(1, num_frames + 1):
dynFocusDistance.append('x%i' % (frame + offset))
dynFocusDistance.append(' %.7f ' %
tde4.getCameraFocus(cam, frame))
focusDStr = "".join(dynFocusDistance)
toWrite = ' tde4_custom_focus_distance_cm {{curve ' + focusDStr + '}}'
nukeNode.append(toWrite)
except:
# For 3DE4 Release 1:
pass
# write static focus distance else
else:
print 'static focus distance'
try:
cameraFocus = ' tde4_custom_focus_distance_cm %.7f '
cameraFocus = cameraFocus % tde4.getCameraFocus(cam, 1)
nukeNode.append(cameraFocus)
except:
# For 3DE4 Release 1:
nukeNode.append(' tde4_custom_focus_distance_cm 100.0 ')
# write camera
nukeNode.append(' tde4_filmback_width_cm %.7f ' % w_fb_cm)
nukeNode.append(' tde4_filmback_height_cm %.7f ' % h_fb_cm)
nukeNode.append(' tde4_lens_center_offset_x_cm %.7f ' % lco_x_cm)
nukeNode.append(' tde4_lens_center_offset_y_cm %.7f ' % lco_y_cm)
nukeNode.append(' tde4_pixel_aspect %.7f ' % pxa)
nukeNode.append(' field_of_view_xa_unit %.7f ' % fov[0])
nukeNode.append(' field_of_view_ya_unit %.7f ' % fov[2])
nukeNode.append(' field_of_view_xb_unit %.7f ' % fov[1])
nukeNode.append(' field_of_view_yb_unit %.7f ' % fov[3])
nukeNode.append(' filter Simon')
# dynamic distortion
try:
dyndistmode = tde4.getLensDynamicDistortionMode(lens)
except:
# For 3DE4 Release 1:
if tde4.getLensDynamicDistortionFlag(lens) == 1:
dyndistmode = "DISTORTION_DYNAMIC_FOCAL_LENGTH"
else:
dyndistmode = "DISTORTION_STATIC"
if dyndistmode == "DISTORTION_STATIC":
print 'static lens distortion'
for para in (tools.getLDmodelParameterList(model)):
distStr = ' %.7f ' % tde4.getLensLDAdjustableParameter(
lens, para, 1, 1)
nukeNode.append(' ' + tools.nukify_name(para) + distStr)
else:
print 'dynamic lens distortion,'
# dynamic
for para in (tools.getLDmodelParameterList(model)):
lensCurve = []
for frame in range(1, num_frames + 1):
focal = tde4.getCameraFocalLength(cam, frame)
focus = tde4.getCameraFocus(cam, frame)
lensCurve.append('x%i' % (frame + offset))
frameParam = tde4.getLensLDAdjustableParameter(
lens, para, focal, focus)
lensCurve.append(' %.7f ' % frameParam)
lensCurveStr = ' {{curve ' + "".join(lensCurve) + ' }}'
nukeNode.append(' ' + tools.nukify_name(para) + lensCurveStr)
nameNode = 'name ' + getLDNodeName(cam, direction, offset, index)
nukeNode.append(nameNode)
nukeNode.append('}')
return nukeNode
def exportNukeDewarpNode(id_cam,offset,nuke_path):
id_lens = tde4.getCameraLens(id_cam)
model = tde4.getLensLDModel(id_lens)
num_frames = tde4.getCameraNoFrames(id_cam)
w_fb_cm = tde4.getLensFBackWidth(id_lens)
h_fb_cm = tde4.getLensFBackHeight(id_lens)
lco_x_cm = tde4.getLensLensCenterX(id_lens)
lco_y_cm = tde4.getLensLensCenterY(id_lens)
pxa = tde4.getLensPixelAspect(id_lens)
# xa,xb,ya,yb in unit coordinates, in this order.
xa_unit,xb_unit,ya_unit,yb_unit = tde4.getCameraFOV(id_cam)
f = open(nuke_path,"w")
try:
f.write('# Created by 3DEqualizer4 using Export Nuke Distortion Nodes export script\n')
f.write("LD" + nukify_name(model) + ' {\n')
f.write(' direction undistort\n')
################################
# focal length #
################################
if is_focal_length_dynamic(id_cam):
# write focal length curve if dynamic
# print 'dynamic focal length'
f.write(' tde4_focal_length_cm {{curve ')
for frame in range(num_frames):
# Internally, frames start at 1.
focal = tde4.getCameraFocalLength(id_cam,frame + 1)
f.write ('x%i %.7f ' % (frame + offset,focal))
f.write('}}\n')
else:
# write static focal length otherwise
# print 'static focal length'
f.write(' tde4_focal_length_cm %.7f \n' % tde4.getCameraFocalLength(id_cam,1))
################################
# focus distance #
################################
# For Release 1 this function return False, so no problem with getCameraFocus.
if is_focus_distance_dynamic(id_cam):
# write focus distance curve if dynamic
# print 'dynamic focus distance'
f.write(' tde4_custom_focus_distance_cm {{curve ')
for frame in range(num_frames):
# Internally, frames start at 1.
focus = tde4.getCameraFocus(id_cam,frame + 1)
f.write ('x%i %.7f ' % (frame + offset,focus))
f.write('}}\n')
else:
try:
# write static focus distance otherwise
f.write(' tde4_custom_focus_distance_cm %.7f \n' % tde4.getCameraFocus(id_cam,1))
except:
# For Release 1 we simply write out the default value to Nuke.
f.write(' tde4_custom_focus_distance_cm 100.0 \n')
################################
# built-in parameters #
################################
# the remaining five built-in parameters
f.write(' tde4_filmback_width_cm %.7f \n' % w_fb_cm)
f.write(' tde4_filmback_height_cm %.7f \n' % h_fb_cm)
f.write(' tde4_lens_center_offset_x_cm %.7f \n' % lco_x_cm)
f.write(' tde4_lens_center_offset_y_cm %.7f \n' % lco_y_cm)
f.write(' tde4_pixel_aspect %.7f \n' % pxa)
################################
# field-of-view #
################################
f.write(' field_of_view_xa_unit %.7f \n' % xa_unit)
f.write(' field_of_view_xb_unit %.7f \n' % xb_unit)
f.write(' field_of_view_ya_unit %.7f \n' % ya_unit)
f.write(' field_of_view_yb_unit %.7f \n' % yb_unit)
# write distortion parameters
#
# dynamic distortion
dyndistmode = get_dynamic_distortion_mode(id_lens)
old_api = True
try:
for para in getLDmodelParameterList(model):
tde4.getLensLDAdjustableParameter(id_lens, para, 1)
break
except:
old_api = False
if old_api:
# dynamic focal length (zoom)
if dyndistmode=="DISTORTION_DYNAMIC_FOCAL_LENGTH":
# print 'dynamic lens distortion, focal length'
for para in getLDmodelParameterList(model):
f.write(' ' + nukify_name(para) + ' {{curve ')
for frame in range(num_frames):
# Internally, frames start at 1.
focal = tde4.getCameraFocalLength(id_cam,frame + 1)
f.write ('x%i %.7f ' % (frame + offset,tde4.getLensLDAdjustableParameter(id_lens,para,focal)))
f.write('}}\n')
# dynamic focus distance
if dyndistmode=="DISTORTION_DYNAMIC_FOCUS_DISTANCE":
# print 'dynamic lens distortion, focus distance'
for para in getLDmodelParameterList(model):
f.write(' ' + nukify_name(para) + ' {{curve ')
for frame in range(num_frames):
# Older Releases do not have Focus-methods.
try:
# Internally, frames start at 1.
focus = tde4.getCameraFocus(id_cam,frame + 1)
except:
focus = 100.0
f.write('x%i %.7f ' % (frame + offset,tde4.getLensLDAdjustableParameter(id_lens,para,focus)))
f.write('}}\n')
# static distortion
if dyndistmode=="DISTORTION_STATIC":
# print 'static lens distortion'
for para in getLDmodelParameterList(model):
f.write(' ' + nukify_name(para) + ' %.7f \n'%tde4.getLensLDAdjustableParameter(id_lens,para,1))
else:
# new API
if dyndistmode=="DISTORTION_STATIC":
# print 'static lens distortion'
for para in getLDmodelParameterList(model):
f.write(' ' + nukify_name(para) + ' %.7f \n'%tde4.getLensLDAdjustableParameter(id_lens,para,1,1))
else:
# print 'dynamic lens distortion,'
# dynamic
for para in getLDmodelParameterList(model):
f.write(' ' + nukify_name(para) + ' {{curve ')
for frame in range(num_frames):
# Internally, frames start at 1.
focal = tde4.getCameraFocalLength(id_cam,frame + 1)
focus = tde4.getCameraFocus(id_cam,frame + 1)
f.write('x%i %.7f ' % (frame + offset,tde4.getLensLDAdjustableParameter(id_lens,para,focal,focus)))
# print "%i[3DE4] -> %i[Nuke]" % (frame + tde4.getCameraFrameOffset(id_cam),frame + offset)
f.write('}}\n')
f.write(' name LD_3DE4_' + decode_entities(tde4.getCameraName(id_cam)) + '\n')
f.write('}\n')
finally:
f.close()
def getCameraModel(camera):
lens = tde4.getCameraLens(camera)
model = tde4.getLensLDModel(lens)
return model
