Add access for vtkToTensorMesh function

discretize/mixins/vtkModule.py

@@ -469,26 +469,17 @@ class vtkTensorRead(object):
     """Provides a convienance method for reading VTK Rectilinear Grid files
     as ``TensorMesh`` objects."""
 
-    @classmethod
-    def readVTK(TensorMesh, fileName, directory=''):
-        """Read VTK Rectilinear (vtr xml file) and return Tensor mesh and model
-
-        Input:
 
-        :param str fileName: path to the vtr model file to read or just its name if directory is specified
-        :param str directory: directory where the UBC GIF file lives
+    @classmethod
+    def vtkToTensorMesh(TensorMesh, vtrGrid):
+        """Converts a ``vtkRectilinearGrid`` or :class:`vtki.RectilinearGrid`
+        to a :class:`discretize.TensorMesh` object.
 
         Output:
 
         :rtype: tuple
         :return: (TensorMesh, modelDictionary)
         """
-        fname = os.path.join(directory, fileName)
-        # Read the file
-        vtrReader = vtkXMLRectilinearGridReader()
-        vtrReader.SetFileName(fname)
-        vtrReader.Update()
-        vtrGrid = vtrReader.GetOutput()
         # Sort information
         hx = np.abs(np.diff(nps.vtk_to_numpy(vtrGrid.GetXCoordinates())))
         xR = nps.vtk_to_numpy(vtrGrid.GetXCoordinates())[0]
@@ -521,3 +512,25 @@ def readVTK(TensorMesh, fileName, directory=''):
 
         # Return the data
         return tensMsh, models
+
+    @classmethod
+    def readVTK(TensorMesh, fileName, directory=''):
+        """Read VTK Rectilinear (vtr xml file) and return Tensor mesh and model
+
+        Input:
+
+        :param str fileName: path to the vtr model file to read or just its name if directory is specified
+        :param str directory: directory where the UBC GIF file lives
+
+        Output:
+
+        :rtype: tuple
+        :return: (TensorMesh, modelDictionary)
+        """
+        fname = os.path.join(directory, fileName)
+        # Read the file
+        vtrReader = vtkXMLRectilinearGridReader()
+        vtrReader.SetFileName(fname)
+        vtrReader.Update()
+        vtrGrid = vtrReader.GetOutput()
+        return TensorMesh.vtkToTensorMesh(vtrGrid)