Source code for tile_periodic_mesh

#!/usr/bin/env python
"""
The program scales a periodic input mesh (a rectangle or box) in
filename_in by a scale factor and generates a new mesh by repeating the
scaled original mesh in a regular grid (scale x scale [x scale]) if
repeat option is None, or in a grid nx x ny x nz for repeat 'nx,ny,nz',
producing again a periodic rectangle or box mesh.
"""
from __future__ import absolute_import
import sys
sys.path.append('.')

from sfepy.base.base import Output
from sfepy.discrete.fem.mesh import Mesh
from sfepy.mesh.mesh_generators import gen_tiled_mesh
from argparse import ArgumentParser, Action

helps = {
    'scale' : 'scale factor [default: %(default)s]',
    'repeat' : 'repetition counts in each axial direction'
               ' [default: %(default)s]',
    'eps'   : 'coordinate precision [default: %(default)s]',
}

[docs]class ParseRepeat(Action): def __call__(self, parser, namespace, values, option_string=None): setattr(namespace, self.dest, [int(r) for r in values.split(',')])
[docs]def main(): parser = ArgumentParser(description=__doc__) parser.add_argument("--version", action="version", version="%(prog)s 42") parser.add_argument("-s", "--scale", type=int, metavar='scale', action="store", dest="scale", default=2, help=helps['scale']) parser.add_argument("-r", "--repeat", type=str, metavar='nx,ny[,nz]', action=ParseRepeat, dest="repeat", default=None, help=helps['repeat']) parser.add_argument("-e", "--eps", type=float, metavar='eps', action="store", dest="eps", default=1e-8, help=helps['eps']) parser.add_argument('filename_in') parser.add_argument('filename_out') options = parser.parse_args() filename_in = options.filename_in filename_out = options.filename_out output = Output('tpm:') output('scale:', options.scale) output('repeat:', options.repeat) output('eps:', options.eps) mesh_in = Mesh.from_file(filename_in) mesh_out = gen_tiled_mesh(mesh_in, options.repeat, 1./options.scale, options.eps) mesh_out.write(filename_out, io='auto') output('done.')
if __name__ == '__main__': main()