Export Maya

meshroom/nodes/aliceVision/ExportMaya.py

@@ -3,17 +3,13 @@
 from meshroom.core import desc
 from meshroom.core.utils import VERBOSE_LEVEL
 
-
-class ExportMaya(desc.AVCommandLineNode):
-    commandLine = 'aliceVision_exportMeshroomMaya {allParams}'
+class ExportMaya(desc.Node):
 
     category = 'Export'
     documentation = '''
-Export a scene for Autodesk Maya, with an Alembic file describing the SfM: cameras and 3D points.
-It will export half-size undistorted images to use as image planes for cameras and also export thumbnails.
-Use the MeshroomMaya plugin, to load the ABC file. It will recognize the file structure and will setup the scene.
-MeshroomMaya contains a user interface to browse all cameras.
-'''
+    Export a Maya script.
+    This script executed inside Maya, will gather the Meshroom computed elements.
+    '''
 
     inputs = [
         desc.File(
@@ -22,6 +18,24 @@ class ExportMaya(desc.AVCommandLineNode):
             description="Input SfMData file.",
             value="",
         ),
+        desc.File(
+            name="alembic",
+            label="Alembic file",
+            description="Input alembic file.",
+            value="",
+        ),
+        desc.File(
+            name="mesh",
+            label="Input Mesh",
+            description="Input Mesh file.",
+            value="",
+        ),
+        desc.File(
+            name="images",
+            label="Undistorted Images",
+            description="Undistorted images template.",
+            value="",
+        ),
         desc.ChoiceParam(
             name="verboseLevel",
             label="Verbose Level",
@@ -34,8 +48,122 @@ class ExportMaya(desc.AVCommandLineNode):
     outputs = [
         desc.File(
             name="output",
-            label="Folder",
-            description="Folder for MeshroomMaya outputs: undistorted images and thumbnails.",
-            value=desc.Node.internalFolder,
+            label="Mel script",
+            description="Generated mel script",
+            value=desc.Node.internalFolder + "import.mel",
         ),
     ]
+
+    def processChunk(self, chunk):
+
+        import pyalicevision
+        import pathlib
+
+        chunk.logManager.start(chunk.node.verboseLevel.value)
+
+        chunk.logger.info("Open input file")
+        data = pyalicevision.sfmData.SfMData()
+        ret = pyalicevision.sfmDataIO.load(data, chunk.node.input.value, pyalicevision.sfmDataIO.ALL)
+        if not ret:
+            chunk.logger.error("Cannot open input")
+            chunk.logManager.end()
+            raise RuntimeError()
+
+        #Check that we have Only one intrinsic
+        intrinsics = data.getIntrinsics()
+        if len(intrinsics) > 1:
+            chunk.logger.error("Only project with a single intrinsic are supported")
+            chunk.logManager.end()
+            raise RuntimeError()
+
+        intrinsicId = next(iter(intrinsics))
+        intrinsic = intrinsics[intrinsicId]
+        w = intrinsic.w()
+        h = intrinsic.h()
+
+        cam = pyalicevision.camera.Pinhole.cast(intrinsic)
+        if cam == None:
+            chunk.logger.error("Intrinsic is not a required pinhole model")
+            chunk.logManager.end()
+            raise RuntimeError()
+
+        offset = cam.getOffset()
+        pix2inches = cam.sensorWidth() / (25.4 * max(w, h));
+        ox = -pyalicevision.numeric.getX(offset) * pix2inches
+        oy = pyalicevision.numeric.getY(offset) * pix2inches
+
+        scale = cam.getScale()
+        fx = pyalicevision.numeric.getX(scale)
+        fy = pyalicevision.numeric.getY(scale)
+
+
+        #Retrieve the first frame
+
+        minIntrinsicId = 0
+        minFrameId = 0
+        minFrameName = ''
+        first = True
+        views = data.getViews()
+
+        for viewId in views:
+
+            view = views[viewId]
+            frameId = view.getFrameId()
+            intrinsicId = view.getIntrinsicId()
+            frameName = pathlib.Path(view.getImageInfo().getImagePath()).stem
+
+            if first or frameId < minFrameId:
+                minFrameId = frameId
+                minIntrinsicId = intrinsicId
+                minFrameName = frameName
+                first = False
+
+
+        #Generate the script itself
+
+        alembic = chunk.node.alembic.value
+        abcString = f'AbcImport -mode open -fitTimeRange "{alembic}";'
+
+        mesh = chunk.node.mesh.value
+        objString = f'file -import -type "OBJ"  -ignoreVersion -ra true -mbl true -mergeNamespacesOnClash false -namespace "mesh" -options "mo=1"  -pr  -importTimeRange "combine" "{mesh}";'
+
+        framePath = chunk.node.images.value.replace('<INTRINSIC_ID>', str(minIntrinsicId)).replace('<FILESTEM>', minFrameName)
+
+        camString = f'''
+        select -r mvgCameras ;
+        string $camName[] = `listRelatives`;
+
+        currentTime {minFrameId};
+
+        imagePlane -c $camName[0] -fileName "{framePath}";
+        
+        setAttr "imagePlaneShape1.useFrameExtension" 1;
+        setAttr "imagePlaneShape1.offsetX" {ox};
+        setAttr "imagePlaneShape1.offsetY" {oy};
+        '''
+
+        ipa = fx / fy
+        advCamString = ''
+
+        if abs(ipa - 1.0)  < 1e-6:
+            advCamString = f'''
+            setAttr "imagePlaneShape1.fit" 1;
+            '''
+        else:
+            advCamString = f'''
+            setAttr "imagePlaneShape1.fit" 4;
+            setAttr "imagePlaneShape1.squeezeCorrection" {ipa};
+            
+            select -r $camName[0];
+            float $vaperture = `getAttr ".verticalFilmAperture"`;
+            float $scaledvaperture = $vaperture * {ipa};
+            setAttr "imagePlaneShape1.sizeY" $scaledvaperture;
+            '''
+
+        with open(chunk.node.output.value, "w") as f:
+            f.write(abcString + '\n')
+            f.write(objString + '\n')
+            f.write(camString + '\n')
+            f.write(advCamString + '\n')
+
+        chunk.logManager.end()