MeshroomResearch/mrrs/nodes/colmap/AutomaticReconstructor.py at ad5cd6098f70469549567c86f888b123177f708e · meshroomHub/MeshroomResearch · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
__version__ = "4.0"

import os
from sys import platform

from meshroom.core import desc
from . import COLMAP

class ColmapAutomaticReconstructor(desc.CommandLineNode):
    commandLine = COLMAP+' automatic_reconstructor  {allParams}'

    category = 'Colmap'
    documentation = ''''''

    inputs = [
        desc.File(
            name='image_path',
            label='Image Directory',
            description='''Path to images.''',
            value='',
        ),

    ]

    outputs = [

        desc.File(
            name='workspace_path',
            label='Output Folder',
            description='''Output Folder.''',
            value='{nodeCacheFolder}',
        ),
        desc.File(
            name='output_pc',
            label='Output point cloud',
            description='''Output point cloud.''',
            value=os.path.join('{nodeCacheFolder}', 'dense\\0\\fused.ply'),
            group='',
        ),
    ]

    def buildCommandLine(self, chunk):
        cmd = desc.buildCommandLine(self, chunk)
        return cmd.replace(">", ".")