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