incomplete tops

Author: illarionovEA

deepfield/field/field.py

@@ -969,11 +969,12 @@ def _add_welltracks(self, plotter):
                                      np.arange(0, count, 2),
                                      np.arange(1, count, 2)]).T
 
-        mesh = pv.PolyData(np.vstack(vertices), lines=np.vstack(faces))
-        plotter.add_mesh(mesh, name='wells', color='b', line_width=3)
+        if vertices:
+            mesh = pv.PolyData(np.vstack(vertices), lines=np.vstack(faces))
+            plotter.add_mesh(mesh, name='wells', color='b', line_width=3)
 
-        mesh = pv.PolyData(vertices_connectors, lines=faces_connectors)
-        plotter.add_mesh(mesh, name='well_connectors', color='k', line_width=2)
+            mesh = pv.PolyData(vertices_connectors, lines=faces_connectors)
+            plotter.add_mesh(mesh, name='well_connectors', color='k', line_width=2)
 
         return labeled_points
 

deepfield/field/grids.py

@@ -215,8 +215,13 @@ def _to_spatial(self, attr, **kwargs):
         _ = kwargs
         data = getattr(self, attr)
         if isinstance(data, np.ndarray) and data.ndim == 1:
-            if attr in ['ACTNUM', 'DX', 'DY', 'DZ', 'TOPS']:
+            if attr in ['ACTNUM', 'DX', 'DY', 'DZ']:
                 data = data.reshape(self.dimens, order='F')
+            elif attr == 'TOPS':
+                if data.size == np.prod(self.dimens):
+                    data = data.reshape(self.dimens, order='F')
+                else:
+                    data = data.reshape(self.dimens[:2], order='F')
             elif attr == 'COORD':
                 nx, ny, nz = self.dimens
                 data = data.reshape(-1, 6)
@@ -271,6 +276,11 @@ def __init__(self, **kwargs):
             tops = np.zeros(self.dimens)
             tops[..., 1:] = np.cumsum(self.dz, axis=-1)[..., :-1]
             setattr(self, 'TOPS', tops)
+        elif self.tops.ndim == 2 and 'DZ' in self:
+            tops = np.zeros(self.dimens)
+            tops[..., 1:] = np.cumsum(self.dz, axis=-1)[..., :-1]
+            tops += self.tops[:, :, None]
+            setattr(self, 'TOPS', tops)
 
     def get_xyz(self, ijk=None):
         """Get x, y, z coordinates of cell vertices."""