added linear cell quality check

Author: akaszynski

pyansys/cellquality.py

@@ -2,7 +2,9 @@
 Cell quality module
 """
 import ctypes
-from pyansys import _cellqual
+import numpy as np
+
+from pyansys import _cellqual, _cellqualfloat
 
 
 def CellQuality(grid):
@@ -20,6 +22,14 @@ def CellQuality(grid):
     """
     # Get cells and points from grid
     cells = grid.GetNumpyCells(ctypes.c_long)
-    points = grid.GetNumpyPoints(ctypes.c_double)
+    points = grid.points
+    if not points.flags.c_contiguous:
+        points = np.ascontiguousarray(points)
 
-    return _cellqual.CompScJac_quad(cells, points)
+    if points.dtype == ctypes.c_float:
+        return _cellqualfloat.CompScJac_quad(cells, points)
+    elif points.dtype == ctypes.c_double:
+        return _cellqual.CompScJac_quad(cells, points)
+    else:
+        raise Exception('Invalid point precision %s.  ' % points.dtype +
+                        'Must be either float or double')

pyansys/cython/_cellqual.pyx

@@ -7,8 +7,7 @@
 # cython imports
 import numpy as np
 cimport numpy as np
-from libc.math cimport sqrt, fabs
-from numpy.math cimport INFINITY
+from libc.math cimport sqrt
 import ctypes
 
 
@@ -2398,4 +2397,3 @@ def CompScJac_quad(long [::1] cellarr, double [:, ::1] pts):
         cnum += 1
 
     return np.asarray(jacs)[:cnum]
-    

setup.py

@@ -6,11 +6,11 @@
 from io import open as io_open
 
 from setuptools import setup, Extension
-#from setuptools.command.build_ext import build_ext
-
 from setuptools.command.build_ext import build_ext as _build_ext
-# Create a build class that includes numpy directory
+
+
 class build_ext(_build_ext):
+    """ build class that includes numpy directory """
     def finalize_options(self):
         _build_ext.finalize_options(self)
         # Prevent numpy from thinking it is still in its setup process:
@@ -123,6 +123,12 @@ def compilerName():
                            extra_compile_args=cmp_arg,
                            language='c'),
 
+                 # cell quality module
+                 Extension("pyansys._cellqualfloat",
+                           ["pyansys/cython/_cellqualfloat.pyx"],
+                           extra_compile_args=cmp_arg,
+                           language='c'),
+
                  Extension("pyansys._rstHelper",
                            ["pyansys/cython/_rstHelper.pyx"],
                            extra_compile_args=cmp_arg,