raster mesh plotting (#47)

addresses #44

Author: kip-hart

CHANGELOG.rst

@@ -6,6 +6,12 @@ All notable changes to this project will be documented in this file.
 The format is based on `Keep a Changelog`_,
 and this project adheres to `Semantic Versioning`_.
 
+`1.5.2`_ - 2021-09-09
+--------------------------
+Fixed
+'''''''
+- Plotting 3D raster meshes.
+
 `1.5.1`_ - 2021-09-09
 --------------------------
 Fixed
@@ -250,7 +256,8 @@ Added
 
 .. LINKS
 
-.. _`Unreleased`: https://github.com/kip-hart/MicroStructPy/compare/v1.5.1...HEAD
+.. _`Unreleased`: https://github.com/kip-hart/MicroStructPy/compare/v1.5.2...HEAD
+.. _`1.5.2`: https://github.com/kip-hart/MicroStructPy/compare/v1.5.1...v1.5.2
 .. _`1.5.1`: https://github.com/kip-hart/MicroStructPy/compare/v1.5.0...v1.5.1
 .. _`1.5.0`: https://github.com/kip-hart/MicroStructPy/compare/v1.4.10...v1.5.0
 .. _`1.4.10`: https://github.com/kip-hart/MicroStructPy/compare/v1.4.9...v1.4.10

src/microstructpy/__init__.py

@@ -4,4 +4,4 @@
 import microstructpy.seeding
 import microstructpy.verification
 
-__version__ = '1.5.1'
+__version__ = '1.5.2'

src/microstructpy/meshing/trimesh.py

@@ -1329,6 +1329,11 @@ def plot(self, index_by='element', material=[], loc=0, **kwargs):
                     plt_value = np.empty(inds.shape, dtype=object)
                     for i, val_i in enumerate(value):
                         plt_value[inds == i] = val_i
+                    if 'color' in key:
+                        unset_mask = plt_value == None
+                        plt_value[unset_mask] = 'k'
+                        inds[unset_mask] = -1
+
                 else:
                     plt_value = value
                 plt_kwargs[key] = plt_value
@@ -1905,7 +1910,10 @@ def _plot_2d(ax, mesh, index_by, **kwargs):
                 else:
                     e_str = 'Cannot index by {}.'.format(index_by)
                     raise ValueError(e_str)
-                v = value[ind]
+                try:
+                    v = value[ind]
+                except IndexError:
+                    v = 'none'
                 plt_value.append(v)
         else:
             plt_value = value