model-converter-python/convert.py

56 lines
1.9 KiB
Python
Raw Permalink Normal View History

2016-11-21 15:20:24 +01:00
#!/usr/bin/env python3
import argparse
import os
2016-11-25 15:03:03 +01:00
import d3.model.tools as mt
import functools as fc
2016-11-30 15:37:58 +01:00
from d3.model.basemodel import Vector
2016-11-21 15:20:24 +01:00
def check_path(path, should_exist):
""" Check that a path (file or folder) exists or not and return it.
"""
path = os.path.normpath(path)
if should_exist != os.path.exists(path):
msg = "path " + ("does not" if should_exist else "already") + " exist: " + path
raise argparse.ArgumentTypeError(msg)
return path
def main(args):
2016-11-30 15:37:58 +01:00
if (args.from_up is None) != (args.to_up is None):
raise Exception("from-up and to-up args should be both present or both absent")
up_conversion = None
if args.from_up is not None:
up_conversion = (args.from_up, args.to_up)
2016-11-22 15:23:53 +01:00
output = args.output if args.output is not None else '.' + args.type
2016-11-30 15:37:58 +01:00
result = mt.convert(args.input, output, up_conversion)
2016-11-21 15:20:24 +01:00
if args.output is None:
print(result)
else:
with open(args.output, 'w') as f:
f.write(result)
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.set_defaults(func=main)
parser.add_argument('-v', '--version', action='version', version='1.0')
parser.add_argument('-i', '--input', metavar='input',
2016-11-25 15:03:03 +01:00
type=fc.partial(check_path, should_exist=True), default=None,
2016-11-30 15:37:58 +01:00
help='Input file')
2016-11-21 15:20:24 +01:00
parser.add_argument('-o', '--output', metavar='output',
help='Output path')
2016-11-22 15:23:53 +01:00
parser.add_argument('-t', '--type', metavar='type',
help='Export type, useless if output is specified')
2016-11-30 15:37:58 +01:00
parser.add_argument('-fu', '--from-up', metavar='fup', default=None,
help="Initial up vector")
parser.add_argument('-tu', '--to-up', metavar='fup', default=None,
help="Output up vector")
2016-11-21 15:20:24 +01:00
args = parser.parse_args()
args.func(args)