def nastran_to_stl(bdf_filename, stl_filename, is_binary=False, log=None):
"""
Converts a Nastran model to an STL
Parameters
----------
bdf_filename : varies
str : the path to a BDF input file
BDF() : a BDF() model object
stl_filename : str
the output STL path
is_binary : bool; default=False
should the output file be binary
"""
if isinstance(bdf_filename, str):
model = read_bdf(bdf_filename, log=None)
else:
model = bdf_filename
#log.info('card_count = %s' % model.card_count)
nnodes = len(model.nodes)
nodes = zeros((nnodes, 3), dtype='float64')
elements = []
i = 0
nodeid_to_i_map = {}
for node_id, node in sorted(iteritems(model.nodes)):
xyz = node.get_position()
nodes[i, :] = xyz
nodeid_to_i_map[node_id] = i
i += 1
assert len(model.nodes) == i, 'model.nodes=%s i=%s' % (len(model.nodes), i)
for eid, element in sorted(iteritems(model.elements)):
if element.type in ['CQUADR']:
continue
elif element.type in ['CBAR', 'CBEAM', 'CONM2', 'RBE2', 'RBE3',
'CBUSH', 'CBUSH1D', 'CBUSH2D',
'CONROD', 'CROD',
'CELAS1', 'CELAS2', 'CELAS3', 'CELAS4',
'CDAMP1', 'CDAMP2', 'CDAMP3', 'CDAMP4',]:
continue
elif element.type in ['CQUAD4']:
n1, n2, n3, n4 = element.node_ids
i1, i2, i3, i4 = nodeid_to_i_map[n1], nodeid_to_i_map[n2], nodeid_to_i_map[n3], nodeid_to_i_map[n4]
elements.append([i1, i2, i3])
elements.append([i3, i4, i1])
elif element.type in ['CTRIA3', 'CTRIAR']:
n1, n2, n3 = element.node_ids
i1, i2, i3 = nodeid_to_i_map[n1], nodeid_to_i_map[n2], nodeid_to_i_map[n3]
elements.append([i1, i2, i3])
else:
print(element.type)
elements = array(elements, dtype='int32')
stl = STL()
stl.nodes = nodes
stl.elements = elements
stl.write_stl(stl_filename, is_binary=is_binary)
return stl
def cart3d_to_stl(cart3d, stl_filename=None, is_binary=False, log=None, debug=False):
"""
Converts a Cart3D object to STL format.
Parameters
----------
cart3d : Cart3D()
a Cart3D object
stl_filename : str; default=None
path to the output STL file (or None to skip)
log : log
a logger object (or None)
debug : bool; default=False
True/False (used if log is not defined)
Returns
-------
stl : STL()
an STL object
"""
#normals = cart3d.get_normals()
stl = STL(log=log, debug=debug)
stl.nodes = cart3d.nodes - 1
stl.elements = cart3d.elements - 1
if stl_filename:
stl.write_stl(stl_filename, is_binary=is_binary)
return stl
def cart3d_to_stl(cart3d,
stl_filename=None,
is_binary=False,
log=None,
debug=False):
"""
Converts a Cart3D object to STL format.
Parameters
----------
cart3d : Cart3D()
a Cart3D object
stl_filename : str; default=None
path to the output STL file (or None to skip)
log : log
a logger object (or None)
debug : bool; default=False
True/False (used if log is not defined)
Returns
-------
stl : STL()
an STL object
.. todo:: this seems to be broken...
"""
#normals = cart3d.get_normals()
stl = STL(log=log, debug=debug)
stl.nodes = cart3d.nodes #- 1
stl.elements = cart3d.elements #- 1
assert stl.elements.min() == 0, stl.elements.min()
if stl_filename:
stl.write_stl(stl_filename, is_binary=is_binary)
return stl
def merge_stl_files(stl_filenames, stl_out_filename=None, remove_bad_elements=False,
is_binary=True, float_fmt='%6.12f'):
"""
Combines multiple STLs into a single file
Parameters
----------
stl_filenames : List[str, str, ...]
list of stl filenames or a string filename
(useful for removing bad elements)
remove_bad_elements : bool; default=False
should elements with invalid normals be removed?
stl_out_filename : str; default=None -> no writing
string of stl output filename
is_binary : bool; default=True
should the output file be binary
float_fmt : str; default='%6.12f'
the ascii float format
Returns
-------
stl : STL()
the stl object
"""
if isinstance(stl_filenames, str):
stl_filenames = [stl_filenames]
assert isinstance(stl_filenames, (list, tuple)), type(stl_filenames)
assert len(stl_filenames) > 0, stl_filenames
if len(stl_filenames) == 1:
model = STL()
model.read_stl(stl_filenames[0])
if remove_bad_elements:
model.remove_elements_with_bad_normals(model.elements, nodes=model.nodes)
if stl_out_filename is not None:
model.write_stl(stl_out_filename, is_binary=is_binary)
return model
nodes = []
elements = []
n0 = 0
for fname in stl_filenames:
model = STL() # TODO: you shouldn't need to to reinstantiate the STL
model.read_stl(fname)
nnodes = model.nodes.shape[0]
nodes.append(model.nodes)
elements.append(model.elements + n0)
n0 += nnodes
model.nodes = vstack(nodes)
model.elements = vstack(elements)
if remove_bad_elements:
model.remove_elements_with_bad_normals(model.elements, nodes=model.nodes)
if stl_out_filename is not None:
model.write_stl(stl_out_filename, is_binary=is_binary, float_fmt=float_fmt)
return model
def merge_stl_files(stl_filenames, stl_out_filename=None, remove_bad_elements=False,
is_binary=True, float_fmt='%6.12f'):
"""
Combines multiple STLs into a single file
Parameters
----------
stl_filenames : List[str, str, ...]
list of stl filenames or a string filename
(useful for removing bad elements)
remove_bad_elements : bool; default=False
should elements with invalid normals be removed?
stl_out_filename : str; default=None -> no writing
string of stl output filename
is_binary : bool; default=True
should the output file be binary
float_fmt : str; default='%6.12f'
the ascii float format
Returns
-------
stl : STL()
the stl object
"""
if isinstance(stl_filenames, str):
stl_filenames = [stl_filenames]
assert isinstance(stl_filenames, (list, tuple)), type(stl_filenames)
assert len(stl_filenames) > 0, stl_filenames
if len(stl_filenames) == 1:
model = STL()
model.read_stl(stl_filenames[0])
if remove_bad_elements:
model.remove_elements_with_bad_normals(model.elements, nodes=model.nodes)
if stl_out_filename is not None:
model.write_stl(stl_out_filename, is_binary=is_binary)
return model
nodes = []
elements = []
n0 = 0
for fname in stl_filenames:
model = STL() # TODO: you shouldn't need to to reinstantiate the STL
model.read_stl(fname)
nnodes = model.nodes.shape[0]
nodes.append(model.nodes)
elements.append(model.elements + n0)
n0 += nnodes
model.nodes = vstack(nodes)
model.elements = vstack(elements)
if remove_bad_elements:
model.remove_elements_with_bad_normals(model.elements, nodes=model.nodes)
if stl_out_filename is not None:
model.write_stl(stl_out_filename, is_binary=is_binary, float_fmt=float_fmt)
return model
def nastran_to_stl(model, stl_filename, is_binary=False):
#log.info('card_count = %s' % model.card_count)
nnodes = len(model.nodes)
nodes = zeros((nnodes, 3), dtype='float64')
elements = []
i = 0
nodeid_to_i_map = {}
for node_id, node in sorted(iteritems(model.nodes)):
xyz = node.get_position()
nodes[i, :] = xyz
nodeid_to_i_map[node_id] = i
i += 1
assert len(model.nodes) == i, 'model.nodes=%s i=%s' % (len(model.nodes), i)
for eid, element in sorted(iteritems(model.elements)):
if element.type in ['CQUADR']:
continue
elif element.type in ['CBAR', 'CBEAM', 'CONM2', 'RBE2', 'RBE3',
'CBUSH', 'CBUSH1D', 'CBUSH2D',
'CONROD', 'CROD',
'CELAS1', 'CELAS2', 'CELAS3', 'CELAS4',
'CDAMP1', 'CDAMP2', 'CDAMP3', 'CDAMP4',]:
continue
elif element.type in ['CQUAD4']:
n1, n2, n3, n4 = element.node_ids
i1, i2, i3, i4 = nodeid_to_i_map[n1], nodeid_to_i_map[n2], nodeid_to_i_map[n3], nodeid_to_i_map[n4]
elements.append([i1, i2, i3])
elements.append([i3, i4, i1])
elif element.type in ['CTRIA3', 'CTRIAR']:
n1, n2, n3 = element.node_ids
i1, i2, i3 = nodeid_to_i_map[n1], nodeid_to_i_map[n2], nodeid_to_i_map[n3]
elements.append([i1, i2, i3])
else:
print(element.type)
elements = array(elements, dtype='int32')
stl = STL()
stl.nodes = nodes
stl.elements = elements
stl.write_stl(stl_filename, is_binary=is_binary)
return stl
def nastran_to_stl(bdf_filename,
stl_filename,
is_binary=False,
log=None,
stop_on_failure=False):
"""
Converts a Nastran model to an STL
Parameters
----------
bdf_filename : varies
str : the path to a BDF input file
BDF() : a BDF() model object
stl_filename : str
the output STL path
is_binary : bool; default=False
should the output file be binary
log : Logger()
a Python logging object
stop_on_failure : bool; default=False
should the code stop if an error is encountered
"""
if isinstance(bdf_filename, str):
model = read_bdf(bdf_filename, log=log)
else:
model = bdf_filename
#log.info('card_count = %s' % model.card_count)
nnodes = len(model.nodes)
nodes = np.zeros((nnodes, 3), dtype='float64')
elements = []
i = 0
nodeid_to_i_map = {}
offset = False
if offset:
nid = list(model.nodes.keys())[0]
xyz0 = model.nodes[nid].get_position()
else:
xyz0 = np.zeros(3, dtype='float64')
for node_id, node in sorted(model.nodes.items()):
xyz = node.get_position()
nodes[i, :] = xyz - xyz0
nodeid_to_i_map[node_id] = i
i += 1
assert len(model.nodes) == i, 'model.nodes=%s i=%s' % (len(model.nodes), i)
for unused_eid, element in sorted(model.elements.items()):
if element.type in ['CQUADR']:
continue
elif element.type in [
'CBAR',
'CBEAM',
'CONM2',
'RBE2',
'RBE3',
'CBUSH',
'CBUSH1D',
'CBUSH2D',
'CONROD',
'CROD',
'CELAS1',
'CELAS2',
'CELAS3',
'CELAS4',
'CDAMP1',
'CDAMP2',
'CDAMP3',
'CDAMP4',
]:
continue
elif element.type in ['CQUAD4']:
n1, n2, n3, n4 = element.node_ids
i1, i2, i3, i4 = (nodeid_to_i_map[n1], nodeid_to_i_map[n2],
nodeid_to_i_map[n3], nodeid_to_i_map[n4])
elements.append([i1, i2, i3])
elements.append([i3, i4, i1])
elif element.type in ['CTRIA3', 'CTRIAR']:
nids = element.node_ids
unids = np.unique(nids)
if len(unids) == 2:
continue
n1, n2, n3 = nids
i1, i2, i3 = nodeid_to_i_map[n1], nodeid_to_i_map[
n2], nodeid_to_i_map[n3]
elements.append([i1, i2, i3])
else:
model.log.warning('skipping %s' % element.type)
elements = np.array(elements, dtype='int32')
stl = STL(log=model.log)
stl.nodes = nodes
#stl.nodes -= nodes[0, :]
stl.elements = elements
stl.write_stl(stl_filename,
is_binary=is_binary,
stop_on_failure=stop_on_failure)
return stl
def nastran_to_stl(bdf_filename, stl_filename, is_binary=False, log=None):
"""
Converts a Nastran model to an STL
Parameters
----------
bdf_filename : varies
str : the path to a BDF input file
BDF() : a BDF() model object
stl_filename : str
the output STL path
is_binary : bool; default=False
should the output file be binary
log : Logger()
a Python logging object
"""
if isinstance(bdf_filename, str):
model = read_bdf(bdf_filename, log=log)
else:
model = bdf_filename
#log.info('card_count = %s' % model.card_count)
nnodes = len(model.nodes)
nodes = zeros((nnodes, 3), dtype='float64')
elements = []
i = 0
nodeid_to_i_map = {}
for node_id, node in sorted(iteritems(model.nodes)):
xyz = node.get_position()
nodes[i, :] = xyz
nodeid_to_i_map[node_id] = i
i += 1
assert len(model.nodes) == i, 'model.nodes=%s i=%s' % (len(model.nodes), i)
for eid, element in sorted(iteritems(model.elements)):
if element.type in ['CQUADR']:
continue
elif element.type in [
'CBAR',
'CBEAM',
'CONM2',
'RBE2',
'RBE3',
'CBUSH',
'CBUSH1D',
'CBUSH2D',
'CONROD',
'CROD',
'CELAS1',
'CELAS2',
'CELAS3',
'CELAS4',
'CDAMP1',
'CDAMP2',
'CDAMP3',
'CDAMP4',
]:
continue
elif element.type in ['CQUAD4']:
n1, n2, n3, n4 = element.node_ids
i1, i2, i3, i4 = nodeid_to_i_map[n1], nodeid_to_i_map[
n2], nodeid_to_i_map[n3], nodeid_to_i_map[n4]
elements.append([i1, i2, i3])
elements.append([i3, i4, i1])
elif element.type in ['CTRIA3', 'CTRIAR']:
n1, n2, n3 = element.node_ids
i1, i2, i3 = nodeid_to_i_map[n1], nodeid_to_i_map[
n2], nodeid_to_i_map[n3]
elements.append([i1, i2, i3])
else:
print(element.type)
elements = array(elements, dtype='int32')
stl = STL(log=model.log)
stl.nodes = nodes
stl.elements = elements
stl.write_stl(stl_filename, is_binary=is_binary)
return stl
