mirror of
https://github.com/alicevision/Meshroom.git
synced 2025-04-29 02:08:08 +02:00
096da0689d
* plugins loader * graph IO in json * uid computation * commandLine expression evaluation * subprocess execution * generate statistics * export command logs to file * save/load status file * only compute nodes not previously computed
55 lines
No EOL
1.6 KiB
Python
55 lines
No EOL
1.6 KiB
Python
|
|
import processGraph as pg
|
|
|
|
import os
|
|
import sys
|
|
import re
|
|
import argparse
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='Create a new Node Type')
|
|
parser.add_argument('node', metavar='NodeName', type=str,
|
|
help='New node name')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if sys.stdin.isatty():
|
|
print('No input documentation.')
|
|
print('Usage: YOUR_COMMAND --help | {cmd} YourCommand'.format(cmd=os.path.splitext(__file__)[0]))
|
|
exit(-1)
|
|
|
|
inputDoc = [line for line in sys.stdin]
|
|
inputArgs = [line for line in inputDoc if '--' in line]
|
|
|
|
arg_re = re.compile('.*?--(?P<longName>\w+).*?')
|
|
|
|
|
|
def convertToLabel(name):
|
|
camelCaseToLabel = re.sub('(.)([A-Z][a-z]+)', r'\1 \2', name)
|
|
snakeToLabel = ' '.join(word.capitalize() for word in camelCaseToLabel.split('_'))
|
|
snakeToLabel = ' '.join(word.capitalize() for word in snakeToLabel.split(' '))
|
|
# print name, camelCaseToLabel, snakeToLabel
|
|
return snakeToLabel
|
|
|
|
|
|
outputNodeStr = '''
|
|
import processGraph as pg
|
|
|
|
class __COMMANDNAME__(pg.CommandLineNodeDesc):
|
|
internalFolder = '{nodeType}/{uid0}/'
|
|
cmdLineExpr = '{nodeType} {allParams}'
|
|
'''.replace('__COMMANDNAME__', 'args.node')
|
|
|
|
for inputArg in inputArgs:
|
|
paramName = arg_re.match(inputArg).group('longName')
|
|
|
|
inputArgLower = inputArg.lower()
|
|
isFile = 'path' in inputArgLower or 'folder' in inputArgLower or 'file' in inputArgLower
|
|
outputNodeStr += '''
|
|
{name} = pg.{attributeType}(
|
|
label='{label}',
|
|
uid=[0],
|
|
)'''.format(name=paramName, label=convertToLabel(paramName), attributeType='FileAttribute' if isFile else 'ParamAttribute')
|
|
|
|
|
|
print(outputNodeStr) |