Difference between revisions of "Documentation/4.1/Developers/Python scripting"

From Slicer Wiki
Jump to: navigation, search
(Prepend documentation/versioncheck template. See http://na-mic.org/Mantis/view.php?id=2887)
 
(45 intermediate revisions by 4 users not shown)
Line 1: Line 1:
 +
<noinclude>{{documentation/versioncheck}}</noinclude>
 
= Background =
 
= Background =
  
This is an evolution of the [[Slicer3:Python|python implementation in slicer3]].  Slicer's APIs are now natively wrapped in python.  It is still experimental in slicer4.
+
This is an evolution of the [[Slicer3:Python|python implementation in slicer3]].  Slicer's APIs are now natively wrapped in python.   
 +
 
 +
Topics like plotting are still experimental in slicer4.
 +
 
 +
See [http://www.na-mic.org/Wiki/index.php/AHM2012-Slicer-Python this 2012 presentation on the state of python in slicer4].
 +
 
 +
'''See [[4.0/Training#Slicer4_Programming_Tutorial|the python slicer4 tutorial for more examples]].'''
  
 
= Usage options =
 
= Usage options =
Line 10: Line 17:
  
 
Most python code can be installed and run from this window, but because it exists in the event driven Qt GUI environment, some operations like, like parallel processing or headless operation, are not easily supported.
 
Most python code can be installed and run from this window, but because it exists in the event driven Qt GUI environment, some operations like, like parallel processing or headless operation, are not easily supported.
 +
 +
=== Examples ===
 +
 +
Start slicer4 and bring up python console.  Load a volume like this:
 +
 +
<pre>
 +
>>> slicer.util.loadVolume(slicer.app.slicerHome + "/share/MRML/Testing/TestData/fixed.nrrd")
 +
</pre>
 +
 +
Get the volume node for that volume:
 +
 +
<pre>
 +
>>> n = getNode('fixed')
 +
</pre>
 +
 +
You can use Tab to see lists of methods for a class instance.
 +
 +
 +
==== Running a CLI from Python ====
 +
 +
Here's an example to create a model from a volume using the [[Documentation/4.0/Modules/GrayscaleModelMaker|Grayscale Model Maker]]
 +
<pre>
 +
def grayModel(volumeNode):
 +
  parameters = {}
 +
  parameters["InputVolume"] = volumeNode.GetID()
 +
  outModel = slicer.vtkMRMLModelNode()
 +
  slicer.mrmlScene.AddNode( outModel )
 +
  parameters["OutputGeometry"] = outModel.GetID()
 +
  grayMaker = slicer.modules.grayscalemodelmaker
 +
  return (slicer.cli.run(grayMaker, None, parameters))
 +
</pre>
 +
 +
To try this, download the MRHead dataset from the [[Documentation/4.0/Modules/SampleData|Sample Data]] and paste the code into the python console and then run this:
 +
<pre>
 +
v = getNode('MRHead')
 +
cliNode = grayModel(v)
 +
</pre>
 +
 +
Note that the CLI module runs in a background thread, so the call to grayModel will return right away.  But the slicer.cli.run call returns a cliNode (an instance of [http://slicer.org/doc/html/classvtkMRMLCommandLineModuleNode.html vtkMRMLCommandLineModuleNode]) which can be used to monitor the progress of the module.
 +
 +
In order to view the names of the parameters required by a CLI module, consider this example on the Grayscale Model Maker:
 +
<pre>
 +
>>> gm = slicer.modules.grayscalemodelmaker
 +
>>> gmlogic = gm.cliModuleLogic()
 +
>>> node = gmLogic.CreateNode()
 +
>>> print node.GetParameterName(0,0)
 +
InputVolume
 +
>>> print node.GetParameterName(0,1)
 +
OutputGeometry
 +
</pre>
 +
 +
In this example we create a simple callback that will be called whenever the cliNode is modified.  The status will tell you if the nodes is Pending, Running, or Completed.
 +
 +
<pre>
 +
def printStatus(caller, event):
 +
  print("Got a %s from a %s" % (event, caller.GetClassName()))
 +
  if caller.IsA('vtkMRMLCommandLineModuleNode'):
 +
    print("Status is %s" % caller.GetStatusString())
 +
 +
cliNode.AddObserver('ModifiedEvent', printStatus)
 +
</pre>
 +
 +
==== Accessing slice vtkRenderWindows from slice views ====
 +
 +
The example below shows how to get the rendered slice window.
 +
 +
<pre>
 +
lm = slicer.app.layoutManager()
 +
redWidget = lm.sliceWidget('Red')
 +
redView = redWidget.sliceView()
 +
wti = vtk.vtkWindowToImageFilter()
 +
wti.SetInput(redView.renderWindow())
 +
wti.Update()
 +
v = vtk.vtkImageViewer()
 +
v.SetColorWindow(255)
 +
v.SetColorLevel(128)
 +
v.SetInputConnection(wti.GetOutputPort())
 +
v.Render()
 +
</pre>
 +
 +
''TODO: some more samples of the Qt console''
  
 
== In iPython ==
 
== In iPython ==
 +
 +
''Important: The example below was developed for an early beta version of slicer4 and is not supported in Slicer 4.0 or 4.1''
 +
 +
See [http://slicer-devel.65872.n3.nabble.com/import-slicer-problem-in-using-python-in-the-shell-launched-using-quot-slicer-xterm-amp-quot-td3968880.html this thread for information on adapting this approach to Slicer 4.1].
 +
  
 
[http://ipython.scipy.org iPython] is a powerful shell and can also be used to access the vtk and slicer APIs (but not Qt at the moment).
 
[http://ipython.scipy.org iPython] is a powerful shell and can also be used to access the vtk and slicer APIs (but not Qt at the moment).
Line 37: Line 130:
  
 
  git clone git://github.com/ipython/ipython.git
 
  git clone git://github.com/ipython/ipython.git
  git checkout 0.10.2
+
  (cd ./ipython; git checkout 0.10.2)
 
  (cd ./ipython;  ../Slicer-build/Slicer4 --launch ../python-build/bin/python setup.py install)
 
  (cd ./ipython;  ../Slicer-build/Slicer4 --launch ../python-build/bin/python setup.py install)
  
 
* Install matplotlib (remove the source after installing so python import will not get confused by it.)
 
* Install matplotlib (remove the source after installing so python import will not get confused by it.)
  
  svn co https://matplotlib.svn.sourceforge.net/svnroot/matplotlib/branches/v1_0_maint matplotlib-source
+
  git clone git://github.com/pieper/matplotlib.git
  (cd ./matplotlib-source;  ../Slicer-build/Slicer4 --launch ../python-build/bin/python setup.py install)
+
  (cd ./matplotlib;  ../Slicer-build/Slicer4 --launch ../python-build/bin/python setup.py install)
  rm -rf matplotlib-source
+
  rm -rf matplotlib
  
 
=== Now try it! ===
 
=== Now try it! ===
Line 51: Line 144:
 
Launch an xterm with all the paths set correctly to find the slicer python packages
 
Launch an xterm with all the paths set correctly to find the slicer python packages
  
  ./Slicer-build --launch xterm &
+
  ./Slicer-build/Slicer4 --launch xterm &
  
 
Now, ''inside the xterm'' launch ipython
 
Now, ''inside the xterm'' launch ipython
Line 88: Line 181:
 
If all goes well, you should see an image like the one shown here.
 
If all goes well, you should see an image like the one shown here.
  
===Issues===
+
=== Parallel Processing ===
 +
 
 +
In the shell, run this to install [http://packages.python.org/joblib/ joblib]
 +
 
 +
wget http://pypi.python.org/packages/source/j/joblib/joblib-0.4.6.dev.tar.gz
 +
tar xvfz joblib-0.4.6.dev.tar.gz
 +
(cd ./joblib-0.4.6.dev;  ../Slicer-build/Slicer4 --launch ../python-build/bin/python setup.py install)
 +
 
 +
Then in ipython you can run this example:
 +
 
 +
from joblib import *
 +
from math import *
 +
for jobs in xrange(1,8):
 +
  print (jobs, Parallel(n_jobs=jobs)(delayed(sqrt)(i**2) for i in range(10000))[-1])
 +
 
 +
=== Packaging/Saving Scenes ===
 +
 
 +
A basic example.  The following packages all of the scene's referenced files into one specified folder.  All images, csvs, volumes, etc. (everything except the scene .mrml) are copied into the folder "<package dir>/data."  The scene .mrml is in the "<package dir>" directory.
 +
 
 +
import os
 +
# Library for OS specific routines
 +
 +
tempDir = os.path.join(slicer.app.slicerHome, ‘testScene’)
 +
# Put our temp scene directory into the slicer directory.  os.path.join takes care of slash issues that you may encounter with UNIX-Windows compatibility.   
 +
 +
os.mkdir(tempDir)
 +
 
 +
l = slicer.app.applicationLogic()
 +
l.SaveSceneToSlicerDataBundleDirectory(tempDir, None)
 +
 
 +
=== Loading DICOM Sets ===
 +
 
 +
The approach is to point Slicer's DICOM database to the directory of the new files.  The command appends the existing database with the files found in the inputted directory.
 +
i = ctk.ctkDICOMIndexer()
 +
i.addDirectory(slicer.dicomDatabase, '/yourDICOMdir/')
 +
 
 +
One approach to begin the load process is to call on the DICOM module, which will automatically open the "DICOM Details" popup.  However, if the popup has been used already in the current Slicer session, a refresh may not occur and a restart may be required.
 +
m = slicer.util.mainWindow()
 +
m.moduleSelector().selectModule('DICOM')
 +
 
 +
=Issues=
 
* matplotlib currently uses Tk to show the window on Linux and it does not handle pan/zoom events correctly.  Ideally there would be a PythonQt wrapper for the plots and this is probably required for use on windows (and maybe mac).
 
* matplotlib currently uses Tk to show the window on Linux and it does not handle pan/zoom events correctly.  Ideally there would be a PythonQt wrapper for the plots and this is probably required for use on windows (and maybe mac).
  
Line 94: Line 227:
  
 
* in slicer4 the PythonQt package is loaded as 'qt', however matplotlib tries running 'import qt' as a way to determine if it is running in PyQt version 3.  Because of this a patched version of matplotlib is required (see [https://github.com/pieper/matplotlib/commit/23bca600c27924450128bfe6e68196eb87cf0654 this diff])
 
* in slicer4 the PythonQt package is loaded as 'qt', however matplotlib tries running 'import qt' as a way to determine if it is running in PyQt version 3.  Because of this a patched version of matplotlib is required (see [https://github.com/pieper/matplotlib/commit/23bca600c27924450128bfe6e68196eb87cf0654 this diff])
 +
 +
* Tested in Windows 7: Python 2.6.6 (used in Slicer 4.1) has errors in referencing its XML DOM parsers, such as ElementTree.  This is a Windows-specific issue -- the same code works in Linux Ubuntu.  Solution thus far is to write your own XML parser.

Latest revision as of 07:25, 14 June 2013

Home < Documentation < 4.1 < Developers < Python scripting


For the latest Slicer documentation, visit the read-the-docs.


Background

This is an evolution of the python implementation in slicer3. Slicer's APIs are now natively wrapped in python.

Topics like plotting are still experimental in slicer4.

See this 2012 presentation on the state of python in slicer4.

See the python slicer4 tutorial for more examples.

Usage options

Python Interactor

Use the Window->Python Interactor (Control-P on window/linux, Command-P on mac) to bring up the Qt-based console with access to the vtk, Qt, and Slicer wrapped APIs.

Most python code can be installed and run from this window, but because it exists in the event driven Qt GUI environment, some operations like, like parallel processing or headless operation, are not easily supported.

Examples

Start slicer4 and bring up python console. Load a volume like this:

>>> slicer.util.loadVolume(slicer.app.slicerHome + "/share/MRML/Testing/TestData/fixed.nrrd")

Get the volume node for that volume:

>>> n = getNode('fixed')

You can use Tab to see lists of methods for a class instance.


Running a CLI from Python

Here's an example to create a model from a volume using the Grayscale Model Maker

def grayModel(volumeNode):
  parameters = {}
  parameters["InputVolume"] = volumeNode.GetID()
  outModel = slicer.vtkMRMLModelNode()
  slicer.mrmlScene.AddNode( outModel )
  parameters["OutputGeometry"] = outModel.GetID()
  grayMaker = slicer.modules.grayscalemodelmaker
  return (slicer.cli.run(grayMaker, None, parameters))

To try this, download the MRHead dataset from the Sample Data and paste the code into the python console and then run this:

v = getNode('MRHead')
cliNode = grayModel(v)

Note that the CLI module runs in a background thread, so the call to grayModel will return right away. But the slicer.cli.run call returns a cliNode (an instance of vtkMRMLCommandLineModuleNode) which can be used to monitor the progress of the module.

In order to view the names of the parameters required by a CLI module, consider this example on the Grayscale Model Maker:

>>> gm = slicer.modules.grayscalemodelmaker
>>> gmlogic = gm.cliModuleLogic()
>>> node = gmLogic.CreateNode()
>>> print node.GetParameterName(0,0)
 InputVolume
>>> print node.GetParameterName(0,1)
 OutputGeometry

In this example we create a simple callback that will be called whenever the cliNode is modified. The status will tell you if the nodes is Pending, Running, or Completed.

def printStatus(caller, event):
  print("Got a %s from a %s" % (event, caller.GetClassName()))
  if caller.IsA('vtkMRMLCommandLineModuleNode'):
    print("Status is %s" % caller.GetStatusString())

cliNode.AddObserver('ModifiedEvent', printStatus)

Accessing slice vtkRenderWindows from slice views

The example below shows how to get the rendered slice window.

lm = slicer.app.layoutManager()
redWidget = lm.sliceWidget('Red')
redView = redWidget.sliceView()
wti = vtk.vtkWindowToImageFilter()
wti.SetInput(redView.renderWindow())
wti.Update()
v = vtk.vtkImageViewer()
v.SetColorWindow(255)
v.SetColorLevel(128)
v.SetInputConnection(wti.GetOutputPort())
v.Render()

TODO: some more samples of the Qt console

In iPython

Important: The example below was developed for an early beta version of slicer4 and is not supported in Slicer 4.0 or 4.1

See this thread for information on adapting this approach to Slicer 4.1.


iPython is a powerful shell and can also be used to access the vtk and slicer APIs (but not Qt at the moment).

As of Slicer4 beta in February 2011, it is possible to use these steps for installation. This has only been tested on a ubuntu linux system so far.

These instructions assume you have a Slicer4-superbuild directory created according to the Slicer4 Build Instructions.

See the [1] website for more example plot types.

Prerequisites

  • Get readline - it will make iPython more useful.
# do this before building slicer4 - or do "cd Slicer4-superbuild/; rm -rf python* ; make"
sudo apt-get install libreadline6-dev

cd to your Slicer4-superbuild directory for the rest of these steps

  • Install the vtk package in the python tree
# TODO: this should be done in superbuild script
(cd ./VTK-build/Wrapping/Python;  ../../../Slicer-build/Slicer4 --launch ../../../python-build/bin/python setup.py install)
  • Install ipython:
git clone git://github.com/ipython/ipython.git
(cd ./ipython; git checkout 0.10.2)
(cd ./ipython;  ../Slicer-build/Slicer4 --launch ../python-build/bin/python setup.py install)
  • Install matplotlib (remove the source after installing so python import will not get confused by it.)
git clone git://github.com/pieper/matplotlib.git
(cd ./matplotlib;  ../Slicer-build/Slicer4 --launch ../python-build/bin/python setup.py install)
rm -rf matplotlib

Now try it!

python plotting from ipython using slicer libraries

Launch an xterm with all the paths set correctly to find the slicer python packages

./Slicer-build/Slicer4 --launch xterm &

Now, inside the xterm launch ipython

./python-build/bin/ipython

Inside ipython you can past the following script that does:

  • create a mrml scene and add a volume
  • make a numpy aray from the image data
  • do calculations in numpy and vtk for comparision
  • make a histogram plot of the data



import vtk
import slicer
mrml = slicer.vtkMRMLScene()
vl = slicer.vtkSlicerVolumesLogic()
vl.SetAndObserveMRMLScene(mrml)
n = vl.AddArchetypeVolume('../Slicer4/Testing/Data/Input/MRHeadResampled.nhdr', 'CTC')
i = n.GetImageData()
print (i.GetScalarRange())

import vtk.util.numpy_support
a = vtk.util.numpy_support.vtk_to_numpy(i.GetPointData().GetScalars())
print(a.min(),a.max())

import matplotlib
import matplotlib.pyplot
n, bins, patches = matplotlib.pyplot.hist(a, 50, facecolor='g', alpha=0.75)
matplotlib.pyplot.show()

If all goes well, you should see an image like the one shown here.

Parallel Processing

In the shell, run this to install joblib

wget http://pypi.python.org/packages/source/j/joblib/joblib-0.4.6.dev.tar.gz
tar xvfz joblib-0.4.6.dev.tar.gz
(cd ./joblib-0.4.6.dev;  ../Slicer-build/Slicer4 --launch ../python-build/bin/python setup.py install)

Then in ipython you can run this example:

from joblib import *
from math import *
for jobs in xrange(1,8):
  print (jobs, Parallel(n_jobs=jobs)(delayed(sqrt)(i**2) for i in range(10000))[-1])

Packaging/Saving Scenes

A basic example. The following packages all of the scene's referenced files into one specified folder. All images, csvs, volumes, etc. (everything except the scene .mrml) are copied into the folder "<package dir>/data." The scene .mrml is in the "<package dir>" directory.

import os 
# Library for OS specific routines

tempDir = os.path.join(slicer.app.slicerHome, ‘testScene’) 
# Put our temp scene directory into the slicer directory.  os.path.join takes care of slash issues that you may encounter with UNIX-Windows compatibility.    

os.mkdir(tempDir) 
l = slicer.app.applicationLogic()
l.SaveSceneToSlicerDataBundleDirectory(tempDir, None)

Loading DICOM Sets

The approach is to point Slicer's DICOM database to the directory of the new files. The command appends the existing database with the files found in the inputted directory.

i = ctk.ctkDICOMIndexer()
i.addDirectory(slicer.dicomDatabase, '/yourDICOMdir/')

One approach to begin the load process is to call on the DICOM module, which will automatically open the "DICOM Details" popup. However, if the popup has been used already in the current Slicer session, a refresh may not occur and a restart may be required.

m = slicer.util.mainWindow()
m.moduleSelector().selectModule('DICOM')

Issues

  • matplotlib currently uses Tk to show the window on Linux and it does not handle pan/zoom events correctly. Ideally there would be a PythonQt wrapper for the plots and this is probably required for use on windows (and maybe mac).
  • the matplotlib window is in the same thread with the ipython window so you cannot keep the plot open while working on the next one. However you can save the plot to a file (png, pdf, etc...) and look at it with another program while working in ipython.
  • in slicer4 the PythonQt package is loaded as 'qt', however matplotlib tries running 'import qt' as a way to determine if it is running in PyQt version 3. Because of this a patched version of matplotlib is required (see this diff)
  • Tested in Windows 7: Python 2.6.6 (used in Slicer 4.1) has errors in referencing its XML DOM parsers, such as ElementTree. This is a Windows-specific issue -- the same code works in Linux Ubuntu. Solution thus far is to write your own XML parser.