import argparse
import random
import sys
from . import stl
def _get_parser(description):
parser = argparse.ArgumentParser(description=description)
parser.add_argument(
'infile',
nargs='?',
type=argparse.FileType('rb'),
default=sys.stdin,
help='STL file to read',
)
parser.add_argument(
'outfile',
nargs='?',
type=argparse.FileType('wb'),
default=sys.stdout,
help='STL file to write',
)
parser.add_argument('--name', nargs='?', help='Name of the mesh')
parser.add_argument(
'-n',
'--use-file-normals',
action='store_true',
help='Read the normals from the file instead of recalculating them',
)
parser.add_argument(
'-r',
'--remove-empty-areas',
action='store_true',
help='Remove areas with 0 surface areas to prevent errors during '
'normal calculation',
)
parser.add_argument(
'-s',
'--disable-speedups',
action='store_true',
help='Disable Cython speedups',
)
return parser
def _get_name(args):
names = [
args.name,
getattr(args.outfile, 'name', None),
getattr(args.infile, 'name', None),
'numpy-stl-%06d' % random.randint(0, 1_000_000),
]
for name in names: # pragma: no branch
if not isinstance(name, str):
continue
elif name.startswith('<'): # pragma: no cover
continue
elif r'\AppData\Local\Temp' in name: # pragma: no cover
# Windows temp file
continue
else:
return name
return None # pragma: no cover
[docs]
def main():
parser = _get_parser('Convert STL files from ascii to binary and back')
parser.add_argument(
'-a',
'--ascii',
action='store_true',
help='Write ASCII file (default is binary)',
)
parser.add_argument(
'-b',
'--binary',
action='store_true',
help='Force binary file (for TTYs)',
)
args = parser.parse_args()
name = _get_name(args)
stl_file = stl.StlMesh(
filename=name,
fh=args.infile,
calculate_normals=False,
remove_empty_areas=args.remove_empty_areas,
speedups=not args.disable_speedups,
)
if args.binary:
mode = stl.BINARY
elif args.ascii:
mode = stl.ASCII
else:
mode = stl.AUTOMATIC
stl_file.save(
name, args.outfile, mode=mode, update_normals=not args.use_file_normals
)
[docs]
def to_ascii():
parser = _get_parser('Convert STL files to ASCII (text) format')
args = parser.parse_args()
name = _get_name(args)
stl_file = stl.StlMesh(
filename=name,
fh=args.infile,
calculate_normals=False,
remove_empty_areas=args.remove_empty_areas,
speedups=not args.disable_speedups,
)
stl_file.save(
name,
args.outfile,
mode=stl.ASCII,
update_normals=not args.use_file_normals,
)
[docs]
def to_binary():
parser = _get_parser('Convert STL files to binary format')
args = parser.parse_args()
name = _get_name(args)
stl_file = stl.StlMesh(
filename=name,
fh=args.infile,
calculate_normals=False,
remove_empty_areas=args.remove_empty_areas,
speedups=not args.disable_speedups,
)
stl_file.save(
name,
args.outfile,
mode=stl.BINARY,
update_normals=not args.use_file_normals,
)