def main(filePath):
readOptions = mfr.Options()
readOptions.filePath = filePath
proj = mfr.readRZML(readOptions)
for cam in proj.cameras:
mmCam = converter.convertCamera(cam, cdo.softwareType.mm)
tdeCam = converter.convertCamera(cam, cdo.softwareType.tde)
assert tdeCam != None
assert tdeCam.name == cam.name
assert tdeCam.index == cam.index
assert tdeCam.filmAspectRatio == cam.filmAspectRatio
assert tdeCam.width == cam.width
assert tdeCam.height == cam.height
assert tdeCam.imageAspectRatio == cam.imageAspectRatio
assert isinstance(tdeCam.distortion, cdo.KeyframeData)
assert isinstance(tdeCam.focalLength, cdo.KeyframeData)
assert isinstance(cam.sequences, list)
assert isinstance(tdeCam.sequences, list)
assert tdeCam.sequences == cam.sequences
assert tdeCam.distortion.getValue(0) != None
assert tdeCam.focalLength.getValue(0) != None
assert len(tdeCam.distortion.getTimeValues()) > 0
assert len(tdeCam.focalLength.getTimeValues()) > 0
# Test distortion converter.
distortion = converter.convertDistortion(cam, cdo.softwareType.tde)
distortion = converter.convertDistortion(cam, cdo.softwareType.mm)
# Test unit conversion, on float values and KeyframeData objects
# (static and not static).
assert converter.convertValue(1.0, cdo.units.cm, cdo.units.mm) == 10.0
assert converter.convertValue(1.0, cdo.units.mm, cdo.units.cm) == 0.1
assert converter.convertValue(1.0, cdo.units.cm, cdo.units.cm) == 1.0
assert converter.convertValue(1.0, cdo.units.mm, cdo.units.mm) == 1.0
keyData = cdo.KeyframeData(static=True, initialValue=1.0)
keyDataNew = converter.convertValue(keyData, cdo.units.mm,
cdo.units.mm)
assert keyDataNew.getValue(0) == 1.0
keyDataNew = converter.convertValue(keyData, cdo.units.cm,
cdo.units.mm)
assert keyDataNew.getValue(0) == 10.0
keyDataNew = converter.convertValue(keyData, cdo.units.mm,
cdo.units.cm)
assert keyDataNew.getValue(0) == 0.1
keyData = cdo.KeyframeData()
keyData.setValue(1.0, 2)
keyData.setValue(2.0, 3)
keyDataNew = converter.convertValue(keyData, cdo.units.mm,
cdo.units.mm)
assert keyDataNew.getValue(2) == 1.0
keyDataNew = converter.convertValue(keyData, cdo.units.cm,
cdo.units.mm)
assert keyDataNew.getValue(2) == 10.0
keyDataNew = converter.convertValue(keyData, cdo.units.mm,
cdo.units.cm)
assert keyDataNew.getValue(2) == 0.1
return True
def main(filePath):
outDirs = test.outDirTests()
times = test.timeStringTests()
for time, outDir in zip(times, outDirs):
readOptions = mfr.Options()
readOptions.filePath = filePath
readOptions.time = time
projData = mfr.readRZML(readOptions)
cams = projData.cameras
for cam in projData.cameras:
distoOptions = mmWriteDistoImaBatchScript.Options()
distoOptions.filePath = filePath
distoOptions.time = time
distoOptions.outDir = outDir
mmWriteDistoImaBatchScript.main(cam, distoOptions)
return True
def main(filePath):
outDirs = test.outDirTests()
times = test.timeStringTests()
for time, outDir in zip(times, outDirs):
readOptions = mfr.Options()
readOptions.filePath = filePath
readOptions.time = time
projData = mfr.readRZML(readOptions)
cams = projData.cameras
for cam in projData.cameras:
tdeCam = converter.convertCamera(cam, cdo.softwareType.tde)
nukeOptions = tdeWriteWetaNukeDistortionNode.Options()
nukeOptions.filePath = filePath
nukeOptions.time = time
nukeOptions.outDir = outDir
tdeWriteWetaNukeDistortionNode.main(tdeCam, nukeOptions)
return True
def main(filePath):
outDirs = test.outDirTests()
times = test.timeStringTests()
for time, outDir in zip(times, outDirs):
readOptions = mfr.Options()
readOptions.filePath = filePath
readOptions.time = time
projData = mfr.readRZML(readOptions)
cams = projData.cameras
for cam in projData.cameras:
tdeCam = converter.convertCamera(cam, cdo.softwareType.tde)
warpOptions = tdeWriteWarpBatchScript.Options()
warpOptions.filePath = filePath
warpOptions.time = time
warpOptions.outDir = outDir
tdeWriteWarpBatchScript.main(tdeCam, warpOptions)
return True
def main(filePath):
readOptions = mfr.Options()
readOptions.filePath = filePath
times = test.timeStringTests()
for time in times:
readOptions.time = time
project = mfr.readRZML(readOptions)
assert project._software != None
assert project._units != None
assert project.name != None
assert project.index != None
assert project.path != None
assert project.frameRange != None
assert project.cameras != None
assert project.sequences != None
assert p.isfile(project.path) == True
frameRangeLength = int(project.frameRange[1]-project.frameRange[0])
assert frameRangeLength > 0
for cam in project.cameras:
assert cam != None
assert isinstance(cam, cdo.MMCameraData)
assert cam._software != None
assert cam._software == cdo.softwareType.mm
assert cam._units != None
assert cam._units == cdo.units.mm
assert cam.name != None
assert cam.index != None
assert cam.index > 0
assert cam.focalLength != None
assert cam.focalLength.length > 0
assert cam.focal != None
assert cam.focal.length > 0
assert cam.distortion != None
assert cam.distortion.length > 0
assert cam.pixelAspectRatio != None
assert cam.pixelAspectRatio > 0.0
assert cam.imageAspectRatio != None
assert cam.imageAspectRatio > 0.0
assert cam.filmbackWidth != None
assert cam.filmbackWidth > 0.0
assert cam.filmbackHeight != None
assert cam.filmbackHeight > 0.0
for seq in project.sequences:
assert seq != None
assert isinstance(seq, cdo.MMSequenceData)
assert seq._software != None
assert seq._software == cdo.softwareType.mm
assert seq.name != None
assert seq.index != None
assert seq.index > 0
hasCam = False
camIndex = -1
camName = None
for cam in project.cameras:
assert isinstance(cam, cdo.MMCameraData)
if cam.index == seq.cameraIndex:
hasCam = True
camIndex = cam.index
camName = cam.name
assert hasCam == True
assert camIndex > 0
assert camName != None
assert seq.width != None
assert seq.height != None
assert seq.imageAspectRatio != None
assert seq.imageAspectRatio > 0.0
assert seq.imagePath != None
assert isinstance(seq.imagePath, str)
assert seq.frameRange != None
assert len(seq.frameRange) == 3
return True
def exportData(self, options):
"""Exports data.
options - A ConverterOptions class that has all options
filled out correctly.
Returns True or False."""
assert options.inputFile != None
assert options.inputTime != None
assert options.outputDir != None
assert options.exportNukeDistNode != None
assert options.exportRawText != None
assert options.exportTdeLens != None
assert options.exportWarpBatch != None
assert options.exportDistoBatch != None
if not cdo.vaildInputFile(options.inputFile):
msg = 'Incorrect file extension, must end with ".rzml", file path: %s.'
print msg % repr(options.inputFile)
return
# read the file, get camera data.
fileName = p.split(options.inputFile)[1]
print("Reading Matchmover File: '%s'" % fileName)
readOptions = mmFileReader.Options()
readOptions.filePath = options.inputFile
readOptions.time = options.inputTime
projData = mmFileReader.readRZML(readOptions)
cams = projData.cameras
for cam in projData.cameras:
print("Converting Camera: '%s'" % cam.name)
tdeCam = converter.convertCamera(cam, cdo.softwareType.tde)
if options.exportNukeDistNode:
nukeOptions = tdeWriteWetaNukeDistortionNode.Options()
nukeOptions.filePath = options.inputFile
nukeOptions.time = options.inputTime
nukeOptions.outDir = options.outputDir
tdeWriteWetaNukeDistortionNode.main(tdeCam, nukeOptions)
if options.exportRawText:
rawOptions = tdeWriteRawText.Options()
rawOptions.filePath = options.inputFile
rawOptions.time = options.inputTime
rawOptions.outDir = options.outputDir
tdeWriteRawText.main(tdeCam, rawOptions)
if options.exportTdeLens:
lensOptions = tdeWriteLensFile.Options()
lensOptions.filePath = options.inputFile
lensOptions.time = options.inputTime
lensOptions.outDir = options.outputDir
tdeWriteLensFile.main(tdeCam, lensOptions)
if options.exportWarpBatch:
warpOptions = tdeWriteWarpBatchScript.Options()
warpOptions.filePath = options.inputFile
warpOptions.time = options.inputTime
warpOptions.outDir = options.outputDir
# warpOptions.useOverscan = 'none'
# warpOptions.fileSuffix = '_warp4'
tdeWriteWarpBatchScript.main(tdeCam, warpOptions)
if options.exportDistoBatch:
distoOptions = mmWriteDistoImaBatchScript.Options()
distoOptions.filePath = options.inputFile
distoOptions.time = options.inputTime
distoOptions.outDir = options.outputDir
mmWriteDistoImaBatchScript.main(cam, distoOptions)
print('Distortion Converter Finished!')
return