Skip to content

Commit

Permalink
black modified files
Browse files Browse the repository at this point in the history
  • Loading branch information
hillyuan committed Jun 12, 2018
1 parent a489af3 commit f48e560
Show file tree
Hide file tree
Showing 3 changed files with 19 additions and 8 deletions.
12 changes: 7 additions & 5 deletions meshio/exodus_io.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,12 +211,14 @@ def write(filename, mesh):
)
for k, (name, data) in enumerate(mesh.point_data.items()):
node_data[0, k] = data
#node sets

# node sets
num_nsets = len(mesh.nsets)
if num_nsets>0:
data = rootgrp.createVariable("ns_prop1", "i4", "num_node_sets")
data_names = rootgrp.createVariable("ns_names", "S1", ("num_node_sets", "len_string"))
if num_nsets > 0:
data = rootgrp.createVariable("ns_prop1", "i4", "num_node_sets")
data_names = rootgrp.createVariable(
"ns_names", "S1", ("num_node_sets", "len_string")
)
for k, name in enumerate(mesh.nsets.keys()):
data[k] = k
for i, letter in enumerate(name):
Expand Down
12 changes: 10 additions & 2 deletions meshio/mesh.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,15 @@


class Mesh(object):
def __init__(self, points, cells, point_data=None, cell_data=None, field_data=None, nsets=None):
def __init__(
self,
points,
cells,
point_data=None,
cell_data=None,
field_data=None,
nsets=None,
):
self.points = points
self.cells = cells
self.point_data = point_data if point_data else {}
Expand All @@ -19,7 +27,7 @@ def __repr__(self):
lines.append("Elements:")
for tpe, elems in self.cells.items():
lines.append(" Number of {}s: {}".format(tpe, len(elems)))

if self.nsets:
lines.append("Node sets: {}".format(", ".join(self.nsets.keys())))

Expand Down
3 changes: 2 additions & 1 deletion test/helpers.py
Original file line number Diff line number Diff line change
Expand Up @@ -215,9 +215,10 @@ def add_field_data(mesh, value, dtype):
mesh2.field_data = {"a": numpy.array(value, dtype=dtype)}
return mesh2


def add_node_sets(mesh):
mesh2 = copy.deepcopy(mesh)
mesh2.nsets = {"nset": numpy.array([1,2])}
mesh2.nsets = {"nset": numpy.array([1, 2])}
return mesh2


Expand Down

0 comments on commit f48e560

Please sign in to comment.