Add to default requirements and a partially working triad in the vtk …

…plotter

setup.py

@@ -29,6 +29,7 @@
     'numpy',
     'scipy',
     'openpyxl',
+    'pandas',
     'pint',
     'pyproj',  # required for getting survey parameters
     'PyYAML',
@@ -40,7 +41,6 @@
 requirements_easy = set([
     'magnetic_field_calculator',    # used to get default mag data for survey
     'networkx',
-    'pandas',
     'tabulate',
     'trimesh',
     'utm',

welleng/visual.py

@@ -23,6 +23,10 @@
 from vtkmodules.vtkRenderingCore import (
     vtkRenderWindow, vtkRenderWindowInteractor, vtkRenderer
 )
+from vtkmodules.vtkInteractionWidgets import (
+    vtkCameraOrientationWidget,
+    vtkOrientationMarkerWidget
+)
 
 from .version import __version__ as VERSION
 
@@ -143,7 +147,7 @@ def get_center(self):
 
         return tuple(center)
 
-    def show(self, add_axes=True, **kwargs):
+    def show(self, add_axes=True, orientation_marker=False, **kwargs):
         """
         Convenient method for opening a window to view the rendered scene.
         """
@@ -180,6 +184,15 @@ def show(self, add_axes=True, **kwargs):
         self.GetActiveCamera().Zoom(0.8)
 
         interactive = kwargs.get('interactive', True)
+
+        if orientation_marker:
+            cow = vtkCameraOrientationWidget()
+            cow.SetParentRenderer(self)
+            # cow.SetDefaultRenderer(self)
+
+            cow.Off()
+            cow.EnabledOn()
+
         if interactive:
             renderWindowInteractor.Start()