import os import json import psutil import shutil import tempfile import logging from meshroom.core import desc Viewpoint = [ desc.IntParam(name="viewId", label="Id", description="Image UID", value=-1, uid=[0], range=(0, 200, 1)), desc.IntParam(name="poseId", label="Pose Id", description="Pose Id", value=-1, uid=[0], range=(0, 200, 1)), desc.File(name="path", label="Image Path", description="Image Filepath", value="", uid=[0, 1]), desc.IntParam(name="intrinsicId", label="Intrinsic", description="Internal Camera Parameters", value=-1, uid=[0], range=(0, 200, 1)), desc.IntParam(name="rigId", label="Rig", description="Rig Parameters", value=-1, uid=[0], range=(0, 200, 1)), desc.IntParam(name="subPoseId", label="Rig Sub-Pose", description="Rig Sub-Pose Parameters", value=-1, uid=[0], range=(0, 200, 1)), desc.StringParam(name="metadata", label="Image Metadata", description="", value="", uid=[]), ] Intrinsic = [ desc.IntParam(name="intrinsicId", label="Id", description="Intrinsic UID", value=-1, uid=[0], range=(0, 200, 1)), desc.FloatParam(name="pxInitialFocalLength", label="Initial Focal Length", description="Initial Guess on the Focal Length", value=-1.0, uid=[0], range=(0.0, 200.0, 1.0)), desc.FloatParam(name="pxFocalLength", label="Focal Length", description="Known/Calibrated Focal Length", value=-1.0, uid=[0], range=(0.0, 200.0, 1.0)), desc.ChoiceParam(name="type", label="Camera Type", description="Camera Type", value="", values=['', 'pinhole', 'radial1', 'radial3', 'brown', 'fisheye4'], exclusive=True, uid=[0]), # desc.StringParam(name="deviceMake", label="Make", description="Camera Make", value="", uid=[]), # desc.StringParam(name="deviceModel", label="Model", description="Camera Model", value="", uid=[]), # desc.StringParam(name="sensorWidth", label="Sensor Width", description="Camera Sensor Width", value="", uid=[0]), desc.IntParam(name="width", label="Width", description="Image Width", value=0, uid=[], range=(0, 10000, 1)), desc.IntParam(name="height", label="Height", description="Image Height", value=0, uid=[], range=(0, 10000, 1)), desc.StringParam(name="serialNumber", label="Serial Number", description="Device Serial Number (camera and lens combined)", value="", uid=[]), desc.GroupAttribute(name="principalPoint", label="Principal Point", description="", groupDesc=[ desc.FloatParam(name="x", label="x", description="", value=0, uid=[], range=(0, 10000, 1)), desc.FloatParam(name="y", label="y", description="", value=0, uid=[], range=(0, 10000, 1)), ]), desc.ListAttribute( name="distortionParams", elementDesc=desc.FloatParam(name="p", label="", description="", value=0.0, uid=[0], range=(-0.1, 0.1, 0.01)), label="Distortion Params", description="Distortion Parameters", ), ] class CameraInit(desc.CommandLineNode): internalFolder = '{cache}/{nodeType}/{uid0}/' commandLine = 'aliceVision_cameraInit {allParams} --allowSingleView 1' # don't throw an error if there is only one image size = desc.DynamicNodeSize('viewpoints') inputs = [ desc.ListAttribute( name="viewpoints", elementDesc=desc.GroupAttribute(name="viewpoint", label="Viewpoint", description="", groupDesc=Viewpoint), label="Viewpoints", description="Input viewpoints", group="", ), desc.ListAttribute( name="intrinsics", elementDesc=desc.GroupAttribute(name="intrinsic", label="Intrinsic", description="", groupDesc=Intrinsic), label="Intrinsics", description="Camera Intrinsics", group="", ), desc.FloatParam( name='defaultFieldOfView', label='Default Field Of View', description='Empirical value for the field of view in degree.', value=45.0, range=(0, 180.0, 1), uid=[0], ), desc.File( name='sensorDatabase', label='Sensor Database', description='''Camera sensor width database path.''', value=os.environ.get('ALICEVISION_SENSOR_DB', ''), uid=[], ), desc.ChoiceParam( name='verboseLevel', label='Verbose Level', description='''verbosity level (fatal, error, warning, info, debug, trace).''', value='info', values=['fatal', 'error', 'warning', 'info', 'debug', 'trace'], exclusive=True, uid=[], ), ] outputs = [ desc.File( name='output', label='Output', description='''Output SfMData.''', value='{cache}/{nodeType}/{uid0}/cameraInit.sfm', uid=[], ), ] def buildIntrinsics(self, node, additionalViews=()): """ Build intrinsics from node current views and optional additional views Args: node: the CameraInit node instance to build intrinsics for additionalViews: (optional) the new views (list of path to images) to add to the node's viewpoints Returns: The updated views and intrinsics as two separate lists """ assert isinstance(node.nodeDesc, CameraInit) origCmdVars = node._cmdVars.copy() # Python3: with tempfile.TemporaryDirectory(prefix="Meshroom_CameraInit") as tmpCache tmpCache = tempfile.mkdtemp() localCmdVars = { 'cache': tmpCache, 'nodeType': node.nodeType, } node._buildCmdVars(localCmdVars) node._cmdVars = localCmdVars try: os.makedirs(os.path.join(tmpCache, node.internalFolder)) self.createViewpointsFile(node, additionalViews) cmd = self.buildCommandLine(node.chunks[0]) cmd += " --allowIncompleteOutput 1" # don't throw an error if the image intrinsic is undefined # logging.debug(' - commandLine:', cmd) subprocess = psutil.Popen(cmd, stdout=None, stderr=None, shell=True) stdout, stderr = subprocess.communicate() subprocess.wait() if subprocess.returncode != 0: logging.warning('CameraInit: Error on buildIntrinsics of node "{}".'.format(node.name)) # Reload result of aliceVision_cameraInit cameraInitSfM = node.output.value jsonData = open(cameraInitSfM, 'r').read() data = json.loads(jsonData) intrinsicsKeys = [i.name for i in Intrinsic] intrinsics = [{k: v for k, v in item.items() if k in intrinsicsKeys} for item in data.get("intrinsics", [])] for intrinsic in intrinsics: pp = intrinsic['principalPoint'] intrinsic['principalPoint'] = {} intrinsic['principalPoint']['x'] = pp[0] intrinsic['principalPoint']['y'] = pp[1] # print('intrinsics:', intrinsics) viewsKeys = [v.name for v in Viewpoint] views = [{k: v for k, v in item.items() if k in viewsKeys} for item in data.get("views", [])] for view in views: view['metadata'] = json.dumps(view['metadata']) # convert metadata to string # print('views:', views) return views, intrinsics except Exception: logging.warning('CameraInit: Error on buildIntrinsics of node "{}".'.format(node.name)) raise finally: node._cmdVars = origCmdVars node._buildCmdVars(localCmdVars) shutil.rmtree(tmpCache) def createViewpointsFile(self, node, additionalViews=()): node.viewpointsFile = "" if node.viewpoints or additionalViews: newViews = [] for path in additionalViews: # format additional views to match json format newViews.append({"path": path}) intrinsics = node.intrinsics.getPrimitiveValue(exportDefault=True) for intrinsic in intrinsics: intrinsic['principalPoint'] = [intrinsic['principalPoint']['x'], intrinsic['principalPoint']['y']] views = node.viewpoints.getPrimitiveValue(exportDefault=False) for view in views: # filter out unnecessary attributes if 'metadata' in view: del view['metadata'] sfmData = { "version": [1, 0, 0], "views": views + newViews, "intrinsics": intrinsics, "featureFolder": "", "matchingFolder": "", } node.viewpointsFile = '{cache}/{nodeType}/{uid0}/viewpoints.sfm'.format(**node._cmdVars) with open(node.viewpointsFile, 'w') as f: f.write(json.dumps(sfmData, indent=4)) # python3: json.dumps(node.viewpoints, f, indent=4) def buildCommandLine(self, chunk): cmd = desc.CommandLineNode.buildCommandLine(self, chunk) if chunk.node.viewpointsFile: cmd += ' --input "{}"'.format(chunk.node.viewpointsFile) return cmd def processChunk(self, chunk): self.createViewpointsFile(chunk.node) desc.CommandLineNode.processChunk(self, chunk)