def process_partition(args):
# Ensure outd is a directory
if not os.path.isdir(args.outd):
raise ValueError('Invalid output directory')
# Partition weights
if ':' in args.np:
pwts = [int(w) for w in args.np.split(':')]
else:
pwts = [1]*int(args.np)
# Element weights
if args.elewts:
ewts = {e: int(w) for e, w in (ew.split(':') for ew in args.elewts)}
else:
ewts = {'quad': 6, 'tri': 3, 'tet': 3, 'hex': 18, 'pri': 10, 'pyr': 6}
# Partitioner-specific options
opts = dict(s.split(':', 1) for s in args.popts)
# Create the partitioner
if args.partitioner:
part = get_partitioner(args.partitioner, pwts, ewts, opts=opts)
else:
for name in sorted(cls.name for cls in subclasses(BasePartitioner)):
try:
part = get_partitioner(name, pwts, ewts)
break
except OSError:
pass
else:
raise RuntimeError('No partitioners available')
# Partition the mesh
mesh, rnum, part_soln_fn = part.partition(NativeReader(args.mesh))
# Prepare the solutions
solnit = (part_soln_fn(NativeReader(s)) for s in args.solns)
# Output paths/files
paths = it.chain([args.mesh], args.solns)
files = it.chain([mesh], solnit)
# Iterate over the output mesh/solutions
for path, data in zip(paths, files):
# Compute the output path
path = os.path.join(args.outd, os.path.basename(path.rstrip('/')))
# Save to disk
write_pyfrms(path, data)
# Write out the renumbering table
if args.rnumf:
print('etype,pold,iold,pnew,inew', file=args.rnumf)
for etype, emap in sorted(rnum.items()):
for k, v in sorted(emap.items()):
print(etype, *k, *v, sep=',', file=args.rnumf)
def process_import(args):
# Get a suitable mesh reader instance
if args.type:
reader = get_reader_by_name(args.type, args.inmesh)
else:
extn = os.path.splitext(args.inmesh.name)[1]
reader = get_reader_by_extn(extn, args.inmesh)
# Get the mesh in the PyFR format
mesh = reader.to_pyfrm(args.lintol)
# Save to disk
write_pyfrms(args.outmesh, mesh)