Meshroom/meshroom/nodes/aliceVision/ExportImages.py
Candice Bentéjac ef33aabe5e [nodes] ExportImages: Fix category and namingMode parameter
Change the category to "Export" instead of "Utils" and align the
default value of the `namingMode` parameter with the one used in the
AliceVision's executable. Moreover, the value "viewId" was incorrect,
as "viewid" if the expected one.
2025-03-06 14:42:48 +01:00

83 lines
2.7 KiB
Python

__version__ = "1.0"
from meshroom.core import desc
from meshroom.core.utils import VERBOSE_LEVEL
class ExportImages(desc.AVCommandLineNode):
commandLine = 'aliceVision_exportImages {allParams}'
size = desc.DynamicNodeSize('input')
parallelization = desc.Parallelization(blockSize=40)
commandLineRange = '--rangeStart {rangeStart} --rangeSize {rangeBlockSize}'
category = 'Export'
documentation = '''
Export images referenced in the input sfmData by transforming
them to adapt to the required target intrinsics. For example, the target
intrinsics may be the same without the distortion.
'''
inputs = [
desc.File(
name="input",
label="SfMData",
description="Input SfMData file. Contains the original intrinsics of the images.",
value="",
),
desc.File(
name="target",
label="Target",
description="This sfmData file contains the required intrinsics for the output images.",
value="",
),
desc.ChoiceParam(
name="outputFileType",
label="Output File Type",
description="Output file type for the exported images.",
value="exr",
values=["jpg", "png", "tif", "exr"],
advanced=True,
),
desc.BoolParam(
name="evCorrection",
label="Correct Images Exposure",
description="Apply a correction on images' exposure value.",
value=False,
advanced=True,
),
desc.ChoiceParam(
name="namingMode",
label="Naming mode",
description="image naming mode :\n"
" - viewid: viewid.ext.\n"
" - frameid: Frameid.ext.\n"
" - keep: Keep original name.\n",
value="frameid",
values=["viewid", "frameid", "keep"],
),
desc.ChoiceParam(
name="verboseLevel",
label="Verbose Level",
description="Verbosity level (fatal, error, warning, info, debug, trace).",
values=VERBOSE_LEVEL,
value="info",
),
]
outputs = [
desc.File(
name="output",
label="Images Folder",
description="Output folder.",
value=desc.Node.internalFolder,
),
desc.File(
name="undistorted",
label="Undistorted Images",
description="List of undistorted images.",
semantic="image",
value=desc.Node.internalFolder + "<VIEW_ID>.{outputFileTypeValue}",
group="",
advanced=True,
),
]