summaryrefslogtreecommitdiff
path: root/examples/viewer3DVolume.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/viewer3DVolume.py')
-rw-r--r--examples/viewer3DVolume.py202
1 files changed, 97 insertions, 105 deletions
diff --git a/examples/viewer3DVolume.py b/examples/viewer3DVolume.py
index 82022f9..d030fba 100644
--- a/examples/viewer3DVolume.py
+++ b/examples/viewer3DVolume.py
@@ -22,7 +22,7 @@
# THE SOFTWARE.
#
# ###########################################################################*/
-"""This script illustrates the use of silx.gui.plot3d.ScalarFieldView.
+"""This script illustrates the use of :class:`silx.gui.plot3d.ScalarFieldView`.
It loads a 3D scalar data set from a file and displays iso-surfaces and
an interactive cutting plane.
@@ -39,6 +39,7 @@ __date__ = "05/01/2017"
import argparse
import logging
import os.path
+import sys
import numpy
@@ -114,107 +115,98 @@ def default_isolevel(data):
return numpy.mean(data) + numpy.std(data)
-def main(argv=None):
- # Parse input arguments
- parser = argparse.ArgumentParser(
- description=__doc__)
- parser.add_argument(
- '-l', '--level', nargs='?', type=float, default=float('nan'),
- help="The value at which to generate the iso-surface")
- parser.add_argument(
- '-sx', '--xscale', nargs='?', type=float, default=1.,
- help="The scale of the data on the X axis")
- parser.add_argument(
- '-sy', '--yscale', nargs='?', type=float, default=1.,
- help="The scale of the data on the Y axis")
- parser.add_argument(
- '-sz', '--zscale', nargs='?', type=float, default=1.,
- help="The scale of the data on the Z axis")
- parser.add_argument(
- '-ox', '--xoffset', nargs='?', type=float, default=0.,
- help="The offset of the data on the X axis")
- parser.add_argument(
- '-oy', '--yoffset', nargs='?', type=float, default=0.,
- help="The offset of the data on the Y axis")
- parser.add_argument(
- '-oz', '--zoffset', nargs='?', type=float, default=0.,
- help="The offset of the data on the Z axis")
- parser.add_argument(
- 'filename',
- nargs='?',
- default=None,
- help="""Filename to open.
-
- It supports 3D volume saved as .npy or in .h5 files.
-
- It also support nD data set (n>=3) stored in a HDF5 file.
- For HDF5, provide the filename and path as: <filename>::<path_in_file>.
- If the data set has more than 3 dimensions, it is possible to choose a
- 3D data set as a subset by providing the indices along the first n-3
- dimensions with '#':
- <filename>::<path_in_file>#<1st_dim_index>...#<n-3th_dim_index>
-
- E.g.: data.h5::/data_5D#1#1
- """)
- args = parser.parse_args(args=argv)
-
- # Start GUI
- global app # QApplication must be global to avoid seg fault on quit
- app = qt.QApplication([])
-
- # Create the viewer main window
- window = ScalarFieldView()
- window.setAttribute(qt.Qt.WA_DeleteOnClose)
-
- # Create a parameter tree for the scalar field view
- treeView = SFViewParamTree.TreeView(window)
- treeView.setSfView(window) # Attach the parameter tree to the view
-
- # Add the parameter tree to the main window in a dock widget
- dock = qt.QDockWidget()
- dock.setWindowTitle('Parameters')
- dock.setWidget(treeView)
- window.addDockWidget(qt.Qt.RightDockWidgetArea, dock)
-
- # Load data from file
- if args.filename is not None:
- data = load(args.filename)
- _logger.info('Data:\n\tShape: %s\n\tRange: [%f, %f]',
- str(data.shape), data.min(), data.max())
- else:
- # Create dummy data
- _logger.warning('Not data file provided, creating dummy data')
- coords = numpy.linspace(-10, 10, 64)
- z = coords.reshape(-1, 1, 1)
- y = coords.reshape(1, -1, 1)
- x = coords.reshape(1, 1, -1)
- data = numpy.sin(x * y * z) / (x * y * z)
-
- # Set ScalarFieldView data
- window.setData(data)
-
- # Set scale of the data
- window.setScale(args.xscale, args.yscale, args.zscale)
-
- # Set offset of the data
- window.setTranslation(args.xoffset, args.yoffset, args.zoffset)
-
- # Set axes labels
- window.setAxesLabels('X', 'Y', 'Z')
-
- # Add an iso-surface
- if not numpy.isnan(args.level):
- # Add an iso-surface at the given iso-level
- window.addIsosurface(args.level, '#FF0000FF')
- else:
- # Add an iso-surface from a function
- window.addIsosurface(default_isolevel, '#FF0000FF')
-
- window.show()
- return app.exec_()
-
-
-if __name__ == '__main__':
- import sys
-
- sys.exit(main(argv=sys.argv[1:]))
+# Parse input arguments
+parser = argparse.ArgumentParser(
+ description=__doc__)
+parser.add_argument(
+ '-l', '--level', nargs='?', type=float, default=float('nan'),
+ help="The value at which to generate the iso-surface")
+parser.add_argument(
+ '-sx', '--xscale', nargs='?', type=float, default=1.,
+ help="The scale of the data on the X axis")
+parser.add_argument(
+ '-sy', '--yscale', nargs='?', type=float, default=1.,
+ help="The scale of the data on the Y axis")
+parser.add_argument(
+ '-sz', '--zscale', nargs='?', type=float, default=1.,
+ help="The scale of the data on the Z axis")
+parser.add_argument(
+ '-ox', '--xoffset', nargs='?', type=float, default=0.,
+ help="The offset of the data on the X axis")
+parser.add_argument(
+ '-oy', '--yoffset', nargs='?', type=float, default=0.,
+ help="The offset of the data on the Y axis")
+parser.add_argument(
+ '-oz', '--zoffset', nargs='?', type=float, default=0.,
+ help="The offset of the data on the Z axis")
+parser.add_argument(
+ 'filename',
+ nargs='?',
+ default=None,
+ help="""Filename to open.
+
+ It supports 3D volume saved as .npy or in .h5 files.
+
+ It also support nD data set (n>=3) stored in a HDF5 file.
+ For HDF5, provide the filename and path as: <filename>::<path_in_file>.
+ If the data set has more than 3 dimensions, it is possible to choose a
+ 3D data set as a subset by providing the indices along the first n-3
+ dimensions with '#':
+ <filename>::<path_in_file>#<1st_dim_index>...#<n-3th_dim_index>
+
+ E.g.: data.h5::/data_5D#1#1
+ """)
+args = parser.parse_args(args=sys.argv[1:])
+
+# Start GUI
+app = qt.QApplication([])
+
+# Create the viewer main window
+window = ScalarFieldView()
+
+# Create a parameter tree for the scalar field view
+treeView = SFViewParamTree.TreeView(window)
+treeView.setSfView(window) # Attach the parameter tree to the view
+
+# Add the parameter tree to the main window in a dock widget
+dock = qt.QDockWidget()
+dock.setWindowTitle('Parameters')
+dock.setWidget(treeView)
+window.addDockWidget(qt.Qt.RightDockWidgetArea, dock)
+
+# Load data from file
+if args.filename is not None:
+ data = load(args.filename)
+ _logger.info('Data:\n\tShape: %s\n\tRange: [%f, %f]',
+ str(data.shape), data.min(), data.max())
+else:
+ # Create dummy data
+ _logger.warning('Not data file provided, creating dummy data')
+ coords = numpy.linspace(-10, 10, 64)
+ z = coords.reshape(-1, 1, 1)
+ y = coords.reshape(1, -1, 1)
+ x = coords.reshape(1, 1, -1)
+ data = numpy.sin(x * y * z) / (x * y * z)
+
+# Set ScalarFieldView data
+window.setData(data)
+
+# Set scale of the data
+window.setScale(args.xscale, args.yscale, args.zscale)
+
+# Set offset of the data
+window.setTranslation(args.xoffset, args.yoffset, args.zoffset)
+
+# Set axes labels
+window.setAxesLabels('X', 'Y', 'Z')
+
+# Add an iso-surface
+if not numpy.isnan(args.level):
+ # Add an iso-surface at the given iso-level
+ window.addIsosurface(args.level, '#FF0000FF')
+else:
+ # Add an iso-surface from a function
+ window.addIsosurface(default_isolevel, '#FF0000FF')
+
+window.show()
+app.exec_()
generated by cgit v1.2.3 (git 2.25.1) at 2024-05-11 01:05:39 +0000