added option to remove null elements

Author: akaszynski

doc/ansys_examples.rst

@@ -98,6 +98,7 @@ This ANSYS APDL script builds a bar and applies torque to it using SURF154 eleme
     SAVE
 
 Read and plot the results within python using:
+
 .. code:: python
 
     import pyansys

pyansys/_version.py

@@ -1,5 +1,5 @@
 # major, minor, patch
-version_info = 0, 24, 0
+version_info = 0, 24, 1
 
 # Nice string for the version
 __version__ = '.'.join(map(str, version_info))

pyansys/archive_reader.py

@@ -22,6 +22,7 @@
 from pyansys import _reader
 from pyansys import _relaxmidside
 from pyansys import _parser
+from pyansys.elements import valid_types
 
 # Attempt to load VTK dependent modules
 try:
@@ -48,7 +49,8 @@ def __init__(self, filename):
         """ Initializes a cdb object """
         self.raw = _reader.Read(filename)
 
-    def ParseVTK(self, force_linear=False, allowable_types=None):
+    def ParseVTK(self, force_linear=False, allowable_types=None,
+                 null_unallowed=False):
         """
         Parses raw data from cdb file to VTK format.
 
@@ -62,8 +64,12 @@ def ParseVTK(self, force_linear=False, allowable_types=None):
             Allowable element types.  Defaults to:
             ['45', '95', '185', '186', '92', '187']
 
-            Can include:
-            ['45', '95', '185', '186', '92', '187', '154']
+            See help(pyansys.elements) for available element types.
+
+        null_unallowed : bool, optional
+            Elements types not matching element types will be stored as empty
+            (null) elements.  Useful for debug or tracking element numbers.
+            Default False.
 
         Returns
         -------
@@ -86,8 +92,13 @@ def ParseVTK(self, force_linear=False, allowable_types=None):
         else:
             assert isinstance(allowable_types, list), \
                    'allowable_types must be a list'
+            for eletype in allowable_types:
+                if eletype not in valid_types:
+                    raise Exception('Element type "%s" ' % eletype +
+                                    'cannot be parsed in pyansys')
 
-        result = _parser.Parse(self.raw, force_linear, allowable_types)
+        result = _parser.Parse(self.raw, force_linear, allowable_types,
+                               null_unallowed)
         cells, offset, cell_type, numref, enum, etype, rcon = result
 
         # catch bug

pyansys/binary_reader.py

@@ -13,6 +13,7 @@
 from pyansys import _parsefull
 from pyansys import _rstHelper
 from pyansys import _parser
+from pyansys.elements import valid_types
 
 try:
     import vtk
@@ -848,8 +849,8 @@ def StoreGeometry(self):
                          'e_rcon': np.ones_like(enum)}
 
         # store the reference array
-        valid_types = ['45', '95', '185', '186', '92', '187', '154', '181']
-        result = _parser.Parse(self.geometry, False, valid_types)
+        # Allow quadradic and null unallowed
+        result = _parser.Parse(self.geometry, False, valid_types, True)
         cells, offset, cell_type, self.numref, _, _, _ = result
 
         # catch -1
@@ -860,11 +861,8 @@ def StoreGeometry(self):
 
             element_type = np.zeros_like(etype)
             for key, typekey in ekey:
-                # if str(typekey) in valid_types:
                 element_type[etype == key] = typekey
 
-            # validmask = element_type != 0
-            # element_type = element_type[validmask]
             nodes = nloc[:, :3]
             self.quadgrid = vtkInterface.UnstructuredGrid(offset, cells,
                                                           cell_type, nodes)
@@ -920,9 +918,8 @@ def ElementSolutionHeader(self, rnum):
             # = 2 - extrapolate always
             # print(rxtrap)
             if rxtrap == 0:
-                warnings.warn('Strains and stresses are moved and not ' +
-                              'extrapolated nodal stress calculations will ' +
-                              'be incorrect')
+                warnings.warn('Strains and stresses are being evaluated at ' +
+                              'gauss points and not extrapolated')
 
             # item 122  64-bit pointer to element solution
             f.seek((rpointers[rnum] + 120) * 4)

pyansys/cellquality.py

@@ -20,16 +20,22 @@ def CellQuality(grid):
     Elements may be linear or quadradic
 
     """
-    # Get cells and points from grid
-    cells = grid.GetNumpyCells(ctypes.c_long)
+    cells = grid.GetNumpyCells(ctypes.c_int64)
+
+    offset = grid.offset
+    if offset.dtype != ctypes.c_int64:
+        offset = offset.astype(ctypes.c_int64)
+
+    celltypes = grid.celltypes
+
     points = grid.points
     if not points.flags.c_contiguous:
         points = np.ascontiguousarray(points)
 
     if points.dtype == ctypes.c_float:
-        return _cellqualfloat.CompScJac_quad(cells, points)
+        return _cellqualfloat.CompScJac_quad(cells, points)  # TODO: update
     elif points.dtype == ctypes.c_double:
-        return _cellqual.CompScJac_quad(cells, points)
+        return _cellqual.ComputeQuality(cells, offset, celltypes, points)
     else:
         raise Exception('Invalid point precision %s.  ' % points.dtype +
                         'Must be either float or double')