"""Multi-input/output format converter"""
from __future__ import annotations
import os
import sys
import glob
from typing import Dict, Optional, Any, TYPE_CHECKING
from pyNastran import DEV
if TYPE_CHECKING: # pragma: no cover
from cpylog import SimpleLogger
# stl_to_plot3d ???
[docs]def process_nastran(bdf_filename: str, fmt2: str, fname2: str,
log: Optional[SimpleLogger]=None,
data: Optional[Dict[str, Any]]=None,
debug: bool=True,
quiet: bool=False) -> None:
"""
Converts Nastran to STL/Cart3d/Tecplot/UGRID3d
"""
assert fmt2 in ['stl', 'cart3d', 'tecplot', 'ugrid', 'nastran', 'abaqus'], 'format2=%s' % fmt2
if data is None:
data = {'--scale': 1.0,}
from pyNastran.bdf.bdf import BDF
xref = True
if fmt2 == 'ugrid':
xref = False
model = BDF(log=log, debug=debug)
model.read_bdf(bdf_filename, validate=False, xref=xref)
if data['--scale'] != 1.0:
scale = data['--scale']
data['--scale'] = 1.0
for node in model.nodes.values():
node.xyz = node.get_position() * scale
node.cp = 0
del node.cp_ref
if fmt2 == 'stl':
from pyNastran.converters.nastran.nastran_to_stl import nastran_to_stl
nastran_to_stl(model, fname2, is_binary=data['--binary'])
elif fmt2 == 'cart3d':
from pyNastran.converters.nastran.nastran_to_cart3d import nastran_to_cart3d
cart3d = nastran_to_cart3d(model)
cart3d.write_cart3d(fname2)
elif fmt2 == 'tecplot':
from pyNastran.converters.nastran.nastran_to_tecplot import nastran_to_tecplot
tecplot = nastran_to_tecplot(model)
tecplot_filename = fname2
tecplot.write_tecplot(tecplot_filename, adjust_nids=False)
elif fmt2 == 'ugrid':
from pyNastran.converters.nastran.nastran_to_ugrid import nastran_to_ugrid
nastran_to_ugrid(model, fname2)
elif fmt2 == 'abaqus':
from pyNastran.converters.nastran.nastran_to_abaqus import nastran_to_abaqus
nastran_to_abaqus(model, fname2)
elif fmt2 == 'nastran':
model.write_bdf(fname2, size=16)
else:
raise NotImplementedError('fmt2=%s is not supported by process_nastran' % fmt2)
[docs]def process_cart3d(cart3d_filename: str, fmt2: str, fname2: str,
log: SimpleLogger,
data: Dict[str, Any],
quiet: bool=False) -> None:
"""
Converts Cart3d to STL/Nastran/Tecplot/Cart3d
"""
assert fmt2 in ['stl', 'nastran', 'tecplot', 'cart3d'], 'format2=%s' % fmt2
if data is None:
data = {'--scale': 1.0,}
from pyNastran.converters.cart3d.cart3d import read_cart3d
model = read_cart3d(cart3d_filename, log=log)
if data['--scale'] != 1.0:
model.points *= data['--scale']
data['--scale'] = 1.0
if fmt2 == 'stl':
from pyNastran.converters.cart3d.cart3d_to_stl import cart3d_to_stl_filename
cart3d_to_stl_filename(model, fname2, log=log, is_binary=data['--binary'])
elif fmt2 == 'nastran':
from pyNastran.converters.cart3d.cart3d_to_nastran import cart3d_to_nastran_filename
cart3d_to_nastran_filename(model, fname2, log=log)
elif fmt2 == 'tecplot':
from pyNastran.converters.cart3d.cart3d_to_tecplot import cart3d_to_tecplot
cart3d_to_tecplot(model, fname2, log=log)
elif fmt2 == 'cart3d':
model.write_cart3d(fname2, is_binary=data['--binary'])
# elif fmt2 == 'ugrid':
# cart3d_to_ugrid(model, fname2)
else:
raise NotImplementedError('fmt2=%s is not supported by process_cart3d' % fmt2)
[docs]def process_stl(stl_filename: str, fmt2: str, fname2: str,
log: SimpleLogger,
data: Optional[Dict[str, Any]]=None,
quiet: bool=False) -> None:
"""
Converts STL to Nastran/Cart3d
"""
assert fmt2 in ['stl', 'nastran', 'cart3d'], 'format2=%s' % fmt2
if data is None:
data = {'--scale': 1.0,}
if '*' in stl_filename:
stl_filenames = glob.glob(stl_filename)
else:
stl_filenames = [stl_filename]
assert len(stl_filenames) > 0, stl_filenames
from pyNastran.converters.stl.utils import merge_stl_files
model = merge_stl_files(stl_filenames, stl_out_filename=None, log=log)
scale = data['--scale']
if scale is not None:
assert isinstance(scale, float), 'scale=%r type=%r' % (scale, type(scale))
model.nodes *= scale
# model = STL()
# model.read_stl(stl_filename)
if fmt2 == 'nastran':
from pyNastran.converters.stl.stl_to_nastran import stl_to_nastran
stl_to_nastran(model, fname2, log=log)
elif fmt2 == 'cart3d':
from pyNastran.converters.stl.stl_to_cart3d import stl_to_cart3d
stl_to_cart3d(model, fname2, log=log)
elif fmt2 == 'stl':
is_binary = data['--binary']
model.write_stl(fname2, is_binary=is_binary, float_fmt='%6.12f', stop_on_failure=False)
# elif fmt2 == 'tecplot':
# stl_to_tecplot(model, fname2)
# elif fmt2 == 'ugrid':
# stl_to_ugrid(model, fname2)
else:
raise NotImplementedError('fmt2=%s is not supported by process_stl' % fmt2)
[docs]def element_slice(tecplot, data: Dict[str, Any]) -> None:
"""removes solid elements from a tecplot model"""
xslice = data['--xx']
yslice = data['--yy']
zslice = data['--zz']
# if xslice is not None:
# xslice = data['--xx']
# tecplot.slice_x(xslice)
# if yslice is not None:
# yslice = data['--yy']
# tecplot.slice_y(yslice)
# if zslice is not None:
# zslice = data['--zz']
# tecplot.slice_z(zslice)
#print(tecplot)
tecplot.slice_xyz(xslice, yslice, zslice)
[docs]def process_tecplot(tecplot_filename: str, fmt2: str, fname2: str,
log: SimpleLogger,
data: Optional[Dict[str, Any]]=None,
quiet: bool=False) -> None:
"""
Converts Tecplot to Tecplot
Globs all input tecplot files (e.g. tecplot*.plt)
"""
assert fmt2 in ['stl', 'nastran', 'cart3d', 'tecplot'], 'format2=%s' % fmt2
if '*' in tecplot_filename:
tecplot_filenames = glob.glob(tecplot_filename)
else:
tecplot_filenames = [tecplot_filename]
assert len(tecplot_filenames) > 0, tecplot_filename
from pyNastran.converters.tecplot.utils import merge_tecplot_files
from pyNastran.converters.tecplot.tecplot_to_nastran import tecplot_to_nastran_filename
from pyNastran.converters.tecplot.tecplot_to_cart3d import tecplot_to_cart3d_filename
model = merge_tecplot_files(tecplot_filenames, tecplot_filename_out=None, log=log)
#if fmt2 == 'cart3d':
#tecplot_to_cart3d(model, fname2)
#elif fmt2 == 'stl':
#tecplot_to_stl(model, fname2)
# elif fmt2 == 'ugrid':
# tecplot_to_ugrid(model, fname2)
res_types = data['RESTYPE']
unused_is_points = not data['--block']
if fmt2 == 'tecplot':
if not quiet: # pragma: no cover
print(data)
element_slice(model, data)
# this is a good way to merge files
model.write_tecplot(fname2, res_types=res_types) # is_points=is_points
elif fmt2 == 'nastran':
tecplot_to_nastran_filename(model, fname2)
elif fmt2 == 'stl':
cart3d_filename = fname2 + '.tri'
tecplot_to_cart3d_filename(model, cart3d_filename, log=log)
process_cart3d(cart3d_filename, fmt2, fname2, log, data=data, quiet=quiet)
os.remove(cart3d_filename)
#tecplot_to_nastran_filename(model, fname2 + '.bdf')
#process_nastran(fname2 + '.bdf', fmt2, fname2, log, data=data, quiet=quiet)
elif fmt2 == 'cart3d':
# supports tris/quads, not loads
#tecplot_to_nastran_filename(model, fname2 + '.bdf')
#process_nastran(fname2 + '.bdf', fmt2, fname2, log, data=data, quiet=quiet)
# supports quads/loads, not tris
tecplot_to_cart3d_filename(model, fname2, log=log)
else:
raise NotImplementedError('fmt2=%s is not supported by process_tecplot' % fmt2)
[docs]def process_ugrid(ugrid_filename: str, fmt2: str, fname2: str,
log: SimpleLogger,
data: Optional[SimpleLogger]=None,
quiet: bool=False) -> None:
"""
Converts UGRID to Nastran/Cart3d/STL/Tecplot
"""
assert fmt2 in ['stl', 'nastran', 'cart3d', 'tecplot'], 'format2=%s' % fmt2
read_shells = True
read_solids = True
if fmt2 in ['stl', 'cart3d']:
read_shells = True
read_solids = False
from pyNastran.converters.aflr.ugrid.ugrid_reader import UGRID
model = UGRID(read_shells=read_shells, read_solids=read_solids, log=log)
model.read_ugrid(ugrid_filename)
if fmt2 == 'nastran':
# ugrid_to_nastran(model, fname2
include_shells = True
include_solids = True
bdf_filename = fname2
model.write_bdf(bdf_filename, include_shells=include_shells, include_solids=include_solids)
elif fmt2 == 'cart3d':
include_shells = True
include_solids = False
bdf_filename = fname2 + '.bdf'
model.write_bdf(bdf_filename, include_shells=include_shells, include_solids=include_solids)
# ugrid_to_cart3d(model, fname2)
process_nastran(bdf_filename, 'cart3d', fname2, data=None)
elif fmt2 == 'stl':
include_shells = True
include_solids = False
bdf_filename = fname2 + '.bdf'
model.write_bdf(bdf_filename, include_shells=include_shells, include_solids=include_solids)
process_nastran(bdf_filename, 'cart3d', fname2, data=None)
# ugrid_to_stl(model, fname2)
elif fmt2 == 'tecplot':
from pyNastran.converters.aflr.ugrid.ugrid3d_to_tecplot import ugrid_to_tecplot
# ugrid_to_tecplot(model, fname2)
tecplot, unused_zone = ugrid_to_tecplot(model)
element_slice(tecplot, data)
tecplot_filename = fname2
tecplot.write_tecplot(tecplot_filename)
else:
raise NotImplementedError('fmt2=%s is not supported by process_ugrid' % fmt2)
[docs]def process_vrml(vrml_filename: str, fmt2: str, fname2: str,
log: SimpleLogger,
data: Dict[str, Any],
quiet: bool=False) -> None:
"""
Converts VRML to Nastran
"""
assert fmt2 in ['nastran', 'stl'], 'format2=%s' % fmt2
#if data['--scale'] != 1.0:
#model.points *= data['--scale']
#data['--scale'] = 1.0
from pyNastran.converters.dev.vrml.vrml import vrml_to_nastran, vrml_to_stl
if fmt2 == 'nastran':
vrml_to_nastran(vrml_filename, fname2, log=log)
if fmt2 == 'stl':
vrml_to_stl(vrml_filename, fname2, log=log)
else:
raise NotImplementedError('fmt2=%s is not supported by process_vrml' % fmt2)
if __name__ == '__main__': # pragma: no cover
cmd_line_format_converter()