[cig-commits] r5434 - mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display

[luis at geodynamics.org]

Author: luis
Date: 2006-12-04 11:48:19 -0800 (Mon, 04 Dec 2006)
New Revision: 5434

Added:
   mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display/actions.py
Modified:
   mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display/__init__.py
   mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display/mayavi_custom_ui.py
Log:
Added OpenHDF5 and OpenVTK actions for mayavi2


Modified: mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display/__init__.py
===================================================================
--- mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display/__init__.py	2006-12-04 18:44:11 UTC (rev 5433)
+++ mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display/__init__.py	2006-12-04 19:48:19 UTC (rev 5434)
@@ -0,0 +1,3 @@
+"""
+CitcomS plugins for MayaVi2
+"""

Added: mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display/actions.py
===================================================================
--- mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display/actions.py	2006-12-04 18:44:11 UTC (rev 5433)
+++ mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display/actions.py	2006-12-04 19:48:19 UTC (rev 5434)
@@ -0,0 +1,57 @@
+# Standard library imports
+from os.path import isfile
+
+# Enthought library imports
+from enthought.pyface import FileDialog, OK
+
+# Mayavi plugin imports
+from enthought.mayavi.script import get_imayavi
+from enthought.mayavi.core.common import error
+from enthought.mayavi.action.common import WorkbenchAction, get_imayavi  # TODO: fix double import of get_imayavi
+
+
+class OpenVTKAction(WorkbenchAction):
+    """ Open a VTK file. """
+
+    def perform(self):
+        """Performs the action. """
+        wildcard = 'VTK files (*.vtk)|*.vtk|' + FileDialog.WILDCARD_ALL
+        parent = self.window.control
+        dialog = FileDialog(parent=parent,
+                            title='Open CitcomS VTK file',
+                            action='open',
+                            wildcard=wildcard)
+        if dialog.open() == OK:
+            if isfile(dialog):
+                from citcoms_display.plugins.VTKFileReader import VTKFileReader
+                r = VTKFileReader()
+                r.initialize(dialog.path)
+                mv = get_imayavi(self.window)
+                mv.add_source(r)
+            else:
+                error("File '%s' does not exist!" % dialog.path, parent)
+        return
+
+
+class OpenHDF5Action(WorkbenchAction):
+    """ Open an HDF5 file. """
+
+    def perform(self):
+        """ Performs the action. """
+        wildcard = 'HDF5 files (*.h5)|*.h5|' + FileDialog.WILDCARD_ALL
+        parent = self.window.control
+        dialog = FileDialog(parent=parent,
+                            title='Open CitcomS HDF5 file',
+                            action='open',
+                            wildcard=wildcard)
+        if dialog.open() == OK:
+            if isfile(dialog.path):
+                from citcoms_display.plugins.HDF5FileReader import HDF5FileReader
+                r = HDF5FileReader()
+                r.initialize(dialog.path)
+                mv = get_imayavi(self.window)
+                mv.add_source(r)
+            else:
+                error("File '%s' does not exist!" % dialog.path, parent)
+        return
+

Modified: mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display/mayavi_custom_ui.py
===================================================================
--- mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display/mayavi_custom_ui.py	2006-12-04 18:44:11 UTC (rev 5433)
+++ mc/3D/CitcomS/trunk/visual/Mayavi2/citcoms_display/mayavi_custom_ui.py	2006-12-04 19:48:19 UTC (rev 5434)
@@ -1,4 +1,7 @@
-###############################################################################
+"""
+This module customizes the MayaVi2 UI and adds callbacks to the CitcomS
+visualization plugins.
+"""
 
 # Enthought library imports.
 from enthought.envisage.workbench.action.action_plugin_definition import \
@@ -24,7 +27,7 @@
 # old name: enthought.mayavi.plugins.OpenCitcomSFILES.OpenCitcomSVTKFILE
 citcoms_open_vtk = Action(
     id          = "OpenCitcomsVTKFile",
-    class_name  = "citcoms_display.mayavi_actions.OpenVTK",
+    class_name  = "citcoms_display.actions.OpenVTKAction",
     name        = "&CitcomS VTK file",
     #image      = "images/new_scene.png",
     tooltip     = "Open a CitcomS VTK data file",
@@ -36,7 +39,7 @@
 # old name: enthought.mayavi.plugins.OpenCitcomSFILES.OpenCitcomSHDFFILE
 citcoms_open_hdf = Action(
     id          = "OpenCitcomsHDF5File",
-    class_name  = "citcoms_display.mayavi_actions.OpenHDF5",
+    class_name  = "citcoms_display.actions.OpenHDF5Action",
     name        = "CitcomS &HDF5 file",
     #image      = "images/new_scene.png",
     tooltip     = "Open a CitcomS HDF5 data file",