Add python requirements (#102)

* Sort imported modules

* Import VTK module once

* Typo fix

* Add python requirements

requirements.txt

@@ -0,0 +1,2 @@
+numpy
+vtk

src/join_mesh.py

@@ -1,12 +1,12 @@
 #!/usr/bin/env python3
 import argparse
+import json
 import logging
 import os
-import json
-from partition_mesh import MeshPartitioner
-import vtk
 import os.path
 
+import vtk
+
 
 class MeshJoiner:
     """MeshJoiner class joins meshes partitioned by MeshPartitioner class.
@@ -76,7 +76,7 @@ def read_meshes(prefix: str, partitions=None, recoveryPath=None):
         """
         Reads meshes with given prefix.
         """
-        logger = MeshPartitioner.get_logger()
+        logger = MeshJoiner.get_logger()
         if not partitions:
             partitions = MeshJoiner.count_partitions(prefix)
             logger.info("Detected " + str(partitions) + " partitions with prefix " + prefix)

src/partition_mesh.py

@@ -1,14 +1,15 @@
 #!/usr/bin/env python3
 import argparse
+import json
 import logging
 import math
 import os
-from typing import List
-import numpy as np
+import platform
 import shutil
 from ctypes import *
-import json
-import platform
+
+import numpy as np
+import vtk
 
 
 class Mesh:
@@ -352,7 +353,6 @@ def apply_partition(orig_mesh: Mesh, part, numparts: int):
 
     @staticmethod
     def read_mesh(filename: str) -> Mesh:
-        import vtk
         extension = os.path.splitext(filename)[1]
         if (extension == ".vtu"):
             reader = vtk.vtkXMLUnstructuredGridReader()
@@ -386,7 +386,6 @@ def write_mesh(filename: str, points: List, data_index: List, cells=None, cell_t
         if (cell_types is not None):
             assert (len(cell_types) in [0, len(cells)])
         assert (len(points) == len(data_index))
-        import vtk
         logger = MeshPartitioner.get_logger()
 
         vtkGrid = vtk.vtkUnstructuredGrid()
@@ -482,7 +481,6 @@ def write_meshes(meshes, recoveryInfo, meshname: str, orig_mesh, directory=None)
 
     @staticmethod
     def vtu2vtk(inmesh, outmesh):
-        import vtk
         reader = vtk.vtkXMLUnstructuredGridReader()
         reader.SetFileName(inmesh)
         reader.Update()

src/vtk_calculator.py

@@ -1,13 +1,13 @@
 #!/usr/bin/env python3
 import argparse
+import json
 import logging
 import os.path
-import vtk
-import json
-import sys
+
 import numpy as np
-from vtk.util.numpy_support import vtk_to_numpy as v2n
+import vtk
 from vtk.util.numpy_support import numpy_to_vtk as n2v
+from vtk.util.numpy_support import vtk_to_numpy as v2n
 
 
 class Calculator: