added support for Kratos mdpa files

Author: e-dub

pygem/__init__.py

@@ -20,4 +20,5 @@
 from .stephandler import StepHandler
 from .igeshandler import IgesHandler
 from .khandler import KHandler
+from .mdpahandler import MdpaHandler
 from .params import *

pygem/mdpahandler.py

@@ -0,0 +1,89 @@
+"""
+Derived module from khandler.py to handle Kratos mesh (.mdpa) files.
+"""
+import numpy as np
+import pygem.filehandler as fh
+
+
+class MdpaHandler(fh.FileHandler):
+    """
+    Kratos mesh file handler class
+
+    :cvar string infile: name of the input file to be processed.
+    :cvar string outfile: name of the output file where to write in.
+    :cvar list extensions: extensions of the input/output files. It is equal
+            to '.k'.
+    """
+
+    def __init__(self):
+        super(MdpaHandler, self).__init__()
+        self.extensions = ['.mdpa']
+
+    def parse(self, filename):
+        """
+        Method to parse the file `filename`. It returns a matrix with all the
+        coordinates. It reads only the section after "Begin Nodes" of the
+        mdpa files.
+
+        :param string filename: name of the input file.
+
+        :return: mesh_points: it is a `n_points`-by-3 matrix containing the
+                coordinates of the points of the mesh.
+        :rtype: numpy.ndarray
+        """
+        self._check_filename_type(filename)
+        self._check_extension(filename)
+        self.infile = filename
+        index = -9
+        mesh_points = []
+        with open(self.infile, 'r') as input_file:
+            for num, line in enumerate(input_file):
+                if line.startswith('Begin Nodes'):
+                    index = num
+                if num == index + 1:
+                    if line.startswith('End Nodes'):
+                        break
+                    else:
+                        li = []
+                        for t in line.split()[1:]:
+                            try:
+                                li.append(float(t))
+                            except ValueError:
+                                pass
+                        mesh_points.append(li)
+                        index = num
+            mesh_points = np.array(mesh_points)
+        return mesh_points
+
+    def write(self, mesh_points, filename):
+        """
+        Writes a .mdpa file, called filename, copying all the lines from
+        self.filename but the coordinates. mesh_points is a matrix that
+        contains the new coordinates to write in the .mdpa file.
+
+        :param numpy.ndarray mesh_points: it is a `n_points`-by-3 matrix
+                          containing the coordinates of the points of the mesh
+        :param string filename: name of the output file.
+        """
+        self._check_filename_type(filename)
+        self._check_extension(filename)
+        self._check_infile_instantiation()
+        self.outfile = filename
+        index = -9
+        i = 0
+        with open(self.outfile, 'w') as output_file:
+            with open(self.infile, 'r') as input_file:
+                for num, line in enumerate(input_file):
+                    if line.startswith('Begin Nodes'):
+                        index = num
+                    if num == index + 1:
+                        if line.startswith('End Nodes'):
+                            index = -9
+                        else:
+                            for j in range(0, 3):
+                                line = (line[:5 + 15 * (j)] +
+                                        '{:15.10f}'.format(mesh_points[i][j]) +
+                                        line[5 + 15 * (j + 1):])
+                            i += 1
+                            index = num
+                    output_file.write(line)