def apply_to_camera(pgroup_id, cam_id, lens_id, options, file_data):
"""
Replace the camera and lens with the given options.
"""
camera_data = file_data.get('data', dict())
# Set image file path
file_start_frame = camera_data.get('start_frame')
file_end_frame = camera_data.get('end_frame')
plate_load = options.get('plate_load')
plate_path = options.get('plate_path')
if (plate_load and file_start_frame is not None
and file_end_frame is not None and plate_path):
plate_path = os.path.normpath(plate_path)
tde4.setCameraPath(cam_id, plate_path)
# Set plate frame range
file_start = int(file_start_frame)
file_end = int(file_end_frame)
tde4.setCameraSequenceAttr(cam_id, file_start, file_end, 1)
if SUPPORT_CAMERA_FRAME_OFFSET is True:
tde4.setCameraFrameOffset(cam_id, file_start)
# Set pixel aspect ratio
par = options.get('par')
if par:
par = float(par)
tde4.setLensPixelAspect(lens_id, par)
# Set Camera name
set_name = options.get('set_cam_name')
if set_name:
cam_name = camera_data.get('name', '')
if cam_name:
tde4.setCameraName(cam_id, cam_name)
lens_name = cam_name + '_lens'
if lens_name:
tde4.setLensName(lens_id, lens_name)
attr_data = camera_data.get('attr', dict())
# Set film back
#
# Note: These values cannot be animated in 3DE, even if in Maya
# they were animated. We only take the first value in the list and
# assume the film back value is constant.
fbk_size = options.get('fbk_size')
filmBackWidthSamples = attr_data.get('filmBackWidth')
filmBackHeightSamples = attr_data.get('filmBackHeight')
if fbk_size and filmBackWidthSamples and filmBackHeightSamples:
value_x = filmBackWidthSamples[0][-1] * MILLIMETERS_TO_CENTIMETRES
value_y = filmBackHeightSamples[0][-1] * MILLIMETERS_TO_CENTIMETRES
tde4.setLensFBackWidth(lens_id, value_x)
tde4.setLensFBackHeight(lens_id, value_y)
fbk_offset = options.get('fbk_offset')
filmBackOffsetXSamples = attr_data.get('filmBackOffsetX')
filmBackOffsetYSamples = attr_data.get('filmBackOffsetY')
if fbk_offset and filmBackOffsetXSamples and filmBackOffsetYSamples:
value_x = filmBackOffsetXSamples[0][-1] * MILLIMETERS_TO_CENTIMETRES
value_y = filmBackOffsetYSamples[0][-1] * MILLIMETERS_TO_CENTIMETRES
tde4.setLensLensCenterX(lens_id, value_x)
tde4.setLensLensCenterY(lens_id, value_y)
# Set focal length
file_start_frame = camera_data.get('start_frame')
file_end_frame = camera_data.get('end_frame')
chosen_start_frame = options.get('start_frame')
chosen_end_frame = options.get('end_frame')
fl = options.get('fl')
focalLengthSamples = attr_data.get('focalLength')
if (fl and focalLengthSamples and isinstance(file_start_frame, (int, long))
and isinstance(file_end_frame, (int, long))
and isinstance(chosen_start_frame, basestring)
and isinstance(chosen_end_frame, basestring)):
file_start = int(file_start_frame)
file_end = int(file_end_frame)
chosen_start = int(chosen_start_frame)
chosen_end = int(chosen_end_frame)
focal_length_set = _set_camera_focal_length(
cam_id,
lens_id,
focalLengthSamples,
file_start,
file_end,
chosen_start,
chosen_end,
)
# Set translation and rotation
file_start_frame = camera_data.get('start_frame')
file_end_frame = camera_data.get('end_frame')
chosen_start_frame = options.get('start_frame')
chosen_end_frame = options.get('end_frame')
if (isinstance(file_start_frame,
(int, long)) and isinstance(file_end_frame, (int, long))
and isinstance(chosen_start_frame, basestring)
and isinstance(chosen_end_frame, basestring)):
file_start = int(file_start_frame)
file_end = int(file_end_frame)
chosen_start = int(chosen_start_frame)
chosen_end = int(chosen_end_frame)
# Set Translation
translate_set = False
translate = options.get('translate')
tx_samples = attr_data.get('translateX')
ty_samples = attr_data.get('translateY')
tz_samples = attr_data.get('translateZ')
if translate and tx_samples and ty_samples and tz_samples:
translate_set = _set_camera_translation(pgroup_id, cam_id,
tx_samples, ty_samples,
tz_samples, file_start,
file_end, chosen_start,
chosen_end)
# Set Rotation
rotate_set = False
rotate = options.get('rotate')
rx_samples = attr_data.get('rotateX')
ry_samples = attr_data.get('rotateY')
rz_samples = attr_data.get('rotateZ')
if rotate and rx_samples and ry_samples and rz_samples:
rotate_set = _set_camera_rotation(pgroup_id, cam_id, rx_samples,
ry_samples, rz_samples,
file_start, file_end,
chosen_start, chosen_end)
if translate_set or rotate_set:
tde4.setPGroupPostfilterMode(pgroup_id, 'POSTFILTER_OFF')
tde4.filterPGroup(pgroup_id, cam_id)
return
def pixel_aspect(self, val):
tde4.setLensPixelAspect(self._lens_id, val)
