Python Programmable Filter: Difference between revisions

From KitwarePublic
Jump to navigationJump to search
(Adding a mention of the ParaView guide section.)
(44 intermediate revisions by 8 users not shown)
Line 1: Line 1:
== ParaView3's python programmable filter. ==
The python programmable filter is a general purpose filter that the end user can program within the paraview GUI to manipulate datasets as needed. To use the filter, turn the  '''PARAVIEW_ENABLE_PYTHON''' cmake option on. This causes the make process to wrap paraview's classes into python callable format.  
The python programmable filter is a general purpose filter that the end user can program within the paraview GUI to manipulate datasets as needed. To use the filter, turn the  '''PARAVIEW_ENABLE_PYTHON''' cmake option on. This causes the make process to wrap paraview's classes into python callable format.  


Line 7: Line 5:
* an instance of the python interpreter with the wrapped paraview libraries imported
* an instance of the python interpreter with the wrapped paraview libraries imported
* the ability to easily change the output dataset type.
* the ability to easily change the output dataset type.
For more examples, see the ParaView Guide, section 12.4.
[https://www.paraview.org/paraview-guide/]


== Transform the input ==
== Transform the input ==
When the user selects '''Programmable Filter''' from the '''Filters''' menu, an empty programmable filter is created. The default behavior of the empty script is create a dataset if the same type as its input and to copy through the input dataset's structure. The GUI provides a selection menu where the user can choose from the five primary VTK dataset types for the output. The GUI also provides a text entry area where the user can type, edit or paste in a python script.
When the user selects '''Programmable Filter''' from the '''Filters''' menu, an empty programmable filter is created. The default behavior of the empty script is to create a dataset of the same type as its input and to copy through the input dataset's structure. The GUI provides a selection menu where the user can choose from the five primary VTK dataset types for the output. The GUI also provides a text entry area where the user can type, edit or paste in a python script.


The following figure shows a python script that modifies the geometry of its input dataset.
The following figure shows a python script that modifies the geometry of its input dataset.


This result is produced by the following python script set as the '''Script'''
<table>
<tr>
<td valign="top">
<source lang="python">
pdi = self.GetPolyDataInput()
pdo =  self.GetPolyDataOutput()
newPoints = vtk.vtkPoints()
numPoints = pdi.GetNumberOfPoints()
for i in range(0, numPoints):
    coord = pdi.GetPoint(i)
    x, y, z = coord[:3]
    x = x * 1
    y = y * 1
    z = 1 + z*0.3
    newPoints.InsertPoint(i, x, y, z)
pdo.SetPoints(newPoints)
</source>
</td>
<td>      </td>
<td valign="top">
[[Image:PyScriptFig1.jpg|400px]]
[[Image:PyScriptFig1.jpg|400px]]
</td>
</tr>
</table>


This result is produced by the following python script set as the '''Script'''
== Dealing with Composite Datasets==
Readers such as Ensight, Exodus produce multiblock datasets. Here's the above squish script modified to work with multiblock datasets.
 
<table>
<tr>
<td>
<source lang="python">
<source lang="python">
  from paraview import vtk
def flatten(input, output):
    # Copy the cells etc.
    output.ShallowCopy(input)
    newPoints = vtk.vtkPoints()
    numPoints = input.GetNumberOfPoints()
    for i in range(0, numPoints):
        coord = input.GetPoint(i)
        x, y, z = coord[:3]
        x = x * 1
        y = y * 1
        z = 10 + 0.1*z
        newPoints.InsertPoint(i, x, y, z)
    output.SetPoints(newPoints)


  pdi = self.GetPolyDataInput()
input = self.GetInputDataObject(0, 0)
  pdo = self.GetPolyDataOutput()
output = self.GetOutputDataObject(0)
  newPoints = vtk.vtkPoints()
 
  numPoints = pdi.GetNumberOfPoints()
if input.IsA("vtkMultiBlockDataSet"):
  for i in range(0, numPoints):
    output.CopyStructure(input)
      coord = pdi.GetPoint(i)
    iter = input.NewIterator()
      x, y, z = coord[:3]
    iter.UnRegister(None)
      x = x * 1
    iter.InitTraversal()
      y = y * 1
    while not iter.IsDoneWithTraversal():
      z = 1 + z*0.3
        curInput = iter.GetCurrentDataObject()
      newPoints.InsertPoint(i, x, y, z)
        curOutput = curInput.NewInstance()
   pdo.SetPoints(newPoints)
        curOutput.UnRegister(None)
        output.SetDataSet(iter, curOutput)
        flatten(curInput, curOutput)
        iter.GoToNextItem();
else:
   flatten(input, output)
</source>
</source>
</td>
<td valign="top">
[[Image:PythonProgrammableFIlterMultiBlock.png|400px]]
</td>
</tr>
</table>
==Changing Data Type==
Here we are taking in vtkPolyData and producing a vtkImageData from it. Note that the '''"Output Data Set Type"''' combo-box in the GUI has been changed to '''vtkImageData''' before hitting the ''Apply'' button.


==Asas==
The following figure shows a python script that produces an image data output with one cell per point in its input polygonal dataset.
The following figure shows a python script that produces an image data output with one cell per point in its input polygonal dataset.
[[Image:PyScriptFig2.jpg]]
<table>
<tr>
<td valign="top">
<source lang="python">
#this example creates an Nx1x1 imagedata output
#and populates its cells with the point centered
#scalars of the input dataset


from paraview import vtk
#
# This part of the code is to be placed in the "Script" entry:
#
 
#get a hold of the input
pdi = self.GetInput()
numPts = pdi.GetNumberOfPoints()
   
   
#this example creates an Nx1x1 imagedata output
#create the output dataset with one cell per point
#and populates its cells with the point centered
ido = self.GetOutput()
#scalars of the input dataset
ido.SetDimensions(numPts+1,2,2)
ido.SetOrigin(-1,-1,-1)
#get a hold of the input
ido.SetSpacing(.1,.1,.1)
pdi = self.GetInput()
 
numPts = pdi.GetNumberOfPoints()
# On ParaView 3.8 to 3.12
#ido.SetWholeExtent(0,numPts,0,1,0,1)
#create the output dataset with one cell per point
#ido.AllocateScalars()
ido = self.GetOutput()
 
ido.SetDimensions(numPts+1,2,2)
# On ParaView 3.98, 4.0 and 4.1
ido.SetOrigin(-1,-1,-1)
ido.SetExtent(0,numPts,0,1,0,1)
ido.SetSpacing(.1,.1,.1)
ido.AllocateScalars(vtk.VTK_FLOAT,1)
ido.SetWholeExtent(0,numPts,0,1,0,1)
 
ido.AllocateScalars()
#choose an input point data array to copy
ivals = pdi.GetPointData().GetScalars()
#choose an input point data array to copy
ca = vtk.vtkFloatArray()
ivals = pdi.GetPointData().GetScalars()
ca.SetName(ivals.GetName())
ca = vtk.vtkFloatArray()
ca.SetNumberOfComponents(1)
ca.SetName(ivals.GetName())
ca.SetNumberOfTuples(numPts)
ca.SetNumberOfComponents(1)
 
ca.SetNumberOfTuples(numPts)
#add the new array to the output
#add the new array to the output
ido.GetCellData().AddArray(ca)
ido.GetCellData().AddArray(ca)
 
#copy the values over element by element
#copy the values over element by element
for i in range(0, numPts):
for i in range(0, numPts):
  ca.SetValue(i, ivals.GetValue(i))
  ca.SetValue(i, ivals.GetValue(i))
 
#############################################
 
#
# This part of the code is to be placed in the "RequestInformation Script" entry:
#
 
from paraview import util
 
pdi = self.GetInput()
numPts = pdi.GetNumberOfPoints()
 
util.SetOutputWholeExtent(self, [0,numPts,0,1,0,1])
</source>
 
</td>
<td valign="top">
[[Image:PyScriptFig2.jpg|400px]]
</td>
</tr>
</table>
 
==Generating Data (Programmable Source)==
Along with processing data, python can be used to produce data i.e. act as a data source. Under the '''Sources''' menu, we have the '''Programmable Source''' which can be used to produce data using Python.


Here's an example that generates a helix. Similar to the programmable filter, the output type must be choosen using the '''"Output Data Set Type"''' combo-box before hitting ''Apply''.


The following is a 'Programmable Source' example that generates a Helix curve.
The following is a 'Programmable Source' example that generates a Helix curve.
[[Image:HelixExample.jpg]]
<table>
<tr>
<td>
<source lang="python">
#This script generates a helix curve.
#This is intended as the script of a 'Programmable Source'
import math


#This script generates a helix curve.
numPts = 80 # Points along Helix
#This is intended as the script of a 'Programmable Source'
length = 8.0 # Length of Helix
import math
rounds = 3.0 # Number of times around
 
numPts = 80.0 # Points along Helix
#Get a vtk.PolyData object for the output
length = 8.0 # Length of Helix
pdo = self.GetPolyDataOutput()
rounds = 3.0 # Number of times around
 
#This will store the points for the Helix
#Get a paraview.vtk.PolyData object for the output
newPts = vtk.vtkPoints()
pdo = self.GetPolyDataOutput()
for i in range(0, numPts):
  #Generate the Points along the Helix
#This will store the points for the Helix
  x = i*length/numPts
newPts = paraview.vtkPoints()
  y = math.sin(i*rounds*2*math.pi/numPts)
for i in range(0, numPts):
  z = math.cos(i*rounds*2*math.pi/numPts)
    #Generate the Points along the Helix
  #Insert the Points into the vtkPoints object
    x = i*length/numPts
  #The first parameter indicates the reference.
    y = math.sin(i*rounds*2*math.pi/numPts)
  #value for the point. Here we add them sequentially.
    z = math.cos(i*rounds*2*math.pi/numPts)
  #Note that the first point is at index 0 (not 1).
    #Insert the Points into the vtkPoints object
  newPts.InsertPoint(i, x,y,z)
    #The first parameter indicates the reference.
 
    #value for the point. Here we add them sequentially.
#Add the points to the vtkPolyData object
    #Note that the first point is at index 0 (not 1).
#Right now the points are not associated with a line -  
    newPts.InsertPoint(i, x,y,z)
#it is just a set of unconnected points. We need to
#create a 'cell' object that ties points together
#Add the points to the vtkPolyData object
#to make a curve (in this case). This is done below.
#Right now the points are not associated with a line -  
#A 'cell' is just an object that tells how points are
#it is just a set of unconnected points. We need to
#connected to make a 1D, 2D, or 3D object.
#create a 'cell' object that ties points together
pdo.SetPoints(newPts)
#to make a curve (in this case). This is done below.
 
#A 'cell' is just an object that tells how points are
#Make a vtkPolyLine which holds the info necessary
#connected to make a 1D, 2D, or 3D object.
#to create a curve composed of line segments. This
pdo.SetPoints(newPts)
#really just hold constructor data that will be passed
#to vtkPolyData to add a new line.
#Make a vtkPolyLine which holds the info necessary
aPolyLine = vtk.vtkPolyLine()
#to create a curve composed of line segments. This
 
#really just hold constructor data that will be passed
#Indicate the number of points along the line
#to vtkPolyData to add a new line.
aPolyLine.GetPointIds().SetNumberOfIds(numPts)
aPolyLine = paraview.vtkPolyLine()
for i in range(0,numPts):
  #Add the points to the line. The first value indicates
#Indicate the number of points along the line
  #the order of the point on the line. The second value
aPolyLine.GetPointIds().SetNumberOfIds(numPts)
  #is a reference to a point in a vtkPoints object. Depends
for i in range(0,numPts):
  #on the order that Points were added to vtkPoints object.
    #Add the points to the line. The first value indicates
  #Note that this will not be associated with actual points
    #the order of the point on the line. The second value
  #until it is added to a vtkPolyData object which holds a
    #is a reference to a point in a vtkPoints object. Depends
  #vtkPoints object.
    #on the order that Points were added to vtkPoints object.
  aPolyLine.GetPointIds().SetId(i, i)
    #Note that this will not be associated with actual points
 
    #until it is added to a vtkPolyData object which holds a
#Allocate the number of 'cells' that will be added. We are just
    #vtkPoints object.
#adding one vtkPolyLine 'cell' to the vtkPolyData object.
    aPolyLine.GetPointIds().SetId(i, i)
pdo.Allocate(1, 1)
#Allocate the number of 'cells' that will be added. We are just
#adding one vtkPolyLine 'cell' to the vtkPolyData object.
pdo.Allocate(1, 1)
#Add the poly line 'cell' to the vtkPolyData object.
pdo.InsertNextCell(aPolyLine.GetCellType(), aPolyLine.GetPointIds())
#The Helix is ready to plot! Click 'Apply'.


#Add the poly line 'cell' to the vtkPolyData object.
pdo.InsertNextCell(aPolyLine.GetCellType(), aPolyLine.GetPointIds())


#The Helix is ready to plot! Click 'Apply'.
</source>
</td>
<td valign="top">
[[Image:HelixExample.jpg|400px]]
</td>
</tr>
</table>
==Programmable Source: Double Helix==
An example that draws a double helix with connecting lines (like DNA). Provides an example of using multiple drawing objects 'cells' in the same vtkPolyData output object. The 'Output Data Set Type' should be set of 'vtkPolyData'.
An example that draws a double helix with connecting lines (like DNA). Provides an example of using multiple drawing objects 'cells' in the same vtkPolyData output object. The 'Output Data Set Type' should be set of 'vtkPolyData'.


[[Image:DoubleHelix.jpg]]
<table>
<tr>
<td>
<source lang="python">
#This script generates a helix double.
#This is intended as the script of a 'Programmable Source'
import math
 
numPts = 80 # Points along each Helix
length = 8.0 # Length of each Helix
rounds = 3.0 # Number of times around
phase_shift = math.pi/1.5 # Phase shift between Helixes
 
#Get a vtk.PolyData object for the output
pdo = self.GetPolyDataOutput()


#This script generates a helix double.
#This will store the points for the Helix
#This is intended as the script of a 'Programmable Source'
newPts = vtk.vtkPoints()
import math
for i in range(0, numPts):
  #Generate Points for first Helix
numPts = 80.0 # Points along each Helix
  x = i*length/numPts
length = 8.0 # Length of each Helix
  y = math.sin(i*rounds*2*math.pi/numPts)
rounds = 3.0 # Number of times around
  z = math.cos(i*rounds*2*math.pi/numPts)
phase_shift = math.pi/1.5 # Phase shift between Helixes
  newPts.InsertPoint(i, x,y,z)
 
#Get a paraview.vtk.PolyData object for the output
  #Generate Points for second Helix. Add a phase offset to y and z.
pdo = self.GetPolyDataOutput()
  y = math.sin(i*rounds*2*math.pi/numPts+phase_shift)
  z = math.cos(i*rounds*2*math.pi/numPts+phase_shift)
#This will store the points for the Helix
  #Offset Helix 2 pts by 'numPts' to keep separate from Helix 1 Pts
newPts = paraview.vtkPoints()
  newPts.InsertPoint(i+numPts, x,y,z)
for i in range(0, numPts):
 
    #Generate Points for first Helix
#Add the points to the vtkPolyData object
    x = i*length/numPts
pdo.SetPoints(newPts)
    y = math.sin(i*rounds*2*math.pi/numPts)
 
    z = math.cos(i*rounds*2*math.pi/numPts)
#Make two vtkPolyLine objects to hold curve construction data
    newPts.InsertPoint(i, x,y,z)
aPolyLine1 = vtk.vtkPolyLine()
   
aPolyLine2 = vtk.vtkPolyLine()
    #Generate Points for second Helix. Add a phase offset to y and z.
 
    y = math.sin(i*rounds*2*math.pi/numPts+phase_shift)
#Indicate the number of points along the line
    z = math.cos(i*rounds*2*math.pi/numPts+phase_shift)
aPolyLine1.GetPointIds().SetNumberOfIds(numPts)
    #Offset Helix 2 pts by 'numPts' to keep separate from Helix 1 Pts
aPolyLine2.GetPointIds().SetNumberOfIds(numPts)
    newPts.InsertPoint(i+numPts, x,y,z)
for i in range(0,numPts):
  #First Helix - use the first set of points
#Add the points to the vtkPolyData object
  aPolyLine1.GetPointIds().SetId(i, i)
pdo.SetPoints(newPts)
  #Second Helix - use the second set of points
  #(Offset the point reference by 'numPts').
#Make two vtkPolyLine objects to hold curve construction data
  aPolyLine2.GetPointIds().SetId(i,i+numPts)
aPolyLine1 = paraview.vtkPolyLine()
 
aPolyLine2 = paraview.vtkPolyLine()
#Allocate the number of 'cells' that will be added.  
#Two 'cells' for the Helix curves, and one 'cell'
#Indicate the number of points along the line
#for every 3rd point along the Helixes.
aPolyLine1.GetPointIds().SetNumberOfIds(numPts)
links = range(0,numPts,3)
aPolyLine2.GetPointIds().SetNumberOfIds(numPts)
pdo.Allocate(2+len(links), 1)
for i in range(0,numPts):
 
    #First Helix - use the first set of points
#Add the poly line 'cell' to the vtkPolyData object.
    aPolyLine1.GetPointIds().SetId(i, i)
pdo.InsertNextCell(aPolyLine1.GetCellType(), aPolyLine1.GetPointIds())
    #Second Helix - use the second set of points
pdo.InsertNextCell(aPolyLine2.GetCellType(), aPolyLine2.GetPointIds())
    #(Offset the point reference by 'numPts').
 
    aPolyLine2.GetPointIds().SetId(i,i+numPts)
for i in links:
  #Add a line connecting the two Helixes.
#Allocate the number of 'cells' that will be added.  
  aLine = vtk.vtkLine()
#Two 'cells' for the Helix curves, and one 'cell'
  aLine.GetPointIds().SetId(0, i)
#for every 3rd point along the Helixes.
  aLine.GetPointIds().SetId(1, i+numPts)
links = range(0,numPts,3)
  pdo.InsertNextCell(aLine.GetCellType(), aLine.GetPointIds())
pdo.Allocate(2+len(links), 1)
 
</source>
#Add the poly line 'cell' to the vtkPolyData object.
</td>
pdo.InsertNextCell(aPolyLine1.GetCellType(), aPolyLine1.GetPointIds())
<td valign="top">
pdo.InsertNextCell(aPolyLine2.GetCellType(), aPolyLine2.GetPointIds())
[[Image:DoubleHelix.jpg|400px]]
</td>
for i in links:
</tr>
    #Add a line connecting the two Helixes.
</table>
    aLine = paraview.vtkLine()
    aLine.GetPointIds().SetId(0, i)
    aLine.GetPointIds().SetId(1, i+numPts)
    pdo.InsertNextCell(aLine.GetCellType(), aLine.GetPointIds())


== Examples ==
== Examples ==


[[Here are some more examples of simple ParaView 3 python filters.]]
[[ParaView/Simple ParaView 3 Python Filters|Here are some more examples of simple ParaView 3 python filters.]]


Paraview is built from VTK, and the python bindings for Paraview mirror the python bindings for VTK although the package names seem to be different. For example, there is a vtk.vtkLine in VTK which seems to behave the same as paraview.vtkLine in Paraview. While the online documentation for Paraview seems to be pretty limited, there is more extensive documentation with examples on the VTK documentation site. Examples are organized by class name and often include Python examples along with c++ and TCL examples:
Paraview is built from VTK, and the python bindings for Paraview mirror the python bindings for VTK although the package names seem to be different. For example, there is a vtk.vtkLine in VTK which seems to behave the same as paraview.vtkLine in Paraview. While the online documentation for Paraview seems to be pretty limited, there is more extensive documentation with examples on the VTK documentation site. Examples are organized by class name and often include Python examples along with c++ and TCL examples:


[[http://www.vtk.org/doc/release/5.0/html/c2_vtk_e_0.html VTK Example Code]]
[http://www.vtk.org/doc/release/5.0/html/c2_vtk_e_0.html VTK Example Code]


The examples list is a little hard to navigate. Perhaps an easier way to find examples is to go to the class list, look up the class you are interested in, and click on the class name which brings up the documentation for that class. Many of the class pages include an 'examples' section that links to code examples for the class. From the examples page you can choose the 'Python' example associated with the particular class (not every class has a Python example). The VTK class index can be found here:
The examples list is a little hard to navigate. Perhaps an easier way to find examples is to go to the class list, look up the class you are interested in, and click on the class name which brings up the documentation for that class. Many of the class pages include an 'examples' section that links to code examples for the class. From the examples page you can choose the 'Python' example associated with the particular class (not every class has a Python example). The VTK class index can be found here:


[[http://www.vtk.org/doc/release/5.0/html/annotated.html VTK Class Descriptions]]
[http://www.vtk.org/doc/release/5.0/html/annotated.html VTK Class Descriptions]


NOTE: You may not be able to directly copy the examples from the VTK documentation because the package names may not match Paraview package names, but with a little modification I have been able to get examples from VTK to work in Paraview.
NOTE: You may not be able to directly copy the examples from the VTK documentation because the package names may not match Paraview package names, but with a little modification I have been able to get examples from VTK to work in Paraview.
===Add Colors to a PolyData===
<source lang="python">
input = self.GetPolyDataInput();
output =  self.GetPolyDataOutput();
colors = vtk.vtkUnsignedCharArray();
colors.SetNumberOfComponents(3);
colors.SetName("Colors");
numPoints = input.GetNumberOfPoints()
for i in xrange(0, numPoints):
    colors.InsertNextTuple3(255,0,0);
output.GetPointData().AddArray(colors)
del colors
</source>
==== Notes ====
* don't forget to un-check the map scalars display panel properties check box.
===Center Data===
<source lang="python">
input = self.GetPolyDataInput()
output = self.GetPolyDataOutput()
numPoints = input.GetNumberOfPoints()
sumx = 0
sumy = 0
sumz = 0
for i in xrange(0, numPoints):
    coord = input.GetPoint(i)
    x, y, z = coord[:3]
    sumx = sumx + x
    sumy = sumy + y
    sumz = sumz + z
tx = -sumx/numPoints
ty = -sumy/numPoints
tz = -sumz/numPoints
# note: thanks to pipeline improvements in VTK 6
# shallow copy of input is not needed for PV version >= 4.0
ic = input.NewInstance()
ic.ShallowCopy(input)
transform = vtk.vtkTransform()
transform.Translate(tx,ty,tz)
transformFilter=vtk.vtkTransformPolyDataFilter()
transformFilter.SetTransform(transform)
transformFilter.SetInputData(ic)
transformFilter.Update()
output.ShallowCopy(transformFilter.GetOutputPort(0))
</source>
===Center Data using numpy===
<source lang="python">
from paraview.vtk import dataset_adapter as DA
pdi = self.GetInputDataObject(0,0)
pdo = self.GetOutputDataObject(0)
pdo.CopyAttributes(pdi)
old_pts = inputs[0].Points
new_pts = old_pts - mean(old_pts,axis=0)
arr = DA.numpyTovtkDataArray(new_pts, 'newpts')
pdo.GetPoints().SetData(arr)
</source>
{{ParaView/Template/Footer}}

Revision as of 13:35, 18 February 2019

The python programmable filter is a general purpose filter that the end user can program within the paraview GUI to manipulate datasets as needed. To use the filter, turn the PARAVIEW_ENABLE_PYTHON cmake option on. This causes the make process to wrap paraview's classes into python callable format.

The filter is a wrapper around VTK's vtkProgrammableFilter class and adds to it:

  • a string containing the user's script for the filter to execute
  • an instance of the python interpreter with the wrapped paraview libraries imported
  • the ability to easily change the output dataset type.

For more examples, see the ParaView Guide, section 12.4. [1]

Transform the input

When the user selects Programmable Filter from the Filters menu, an empty programmable filter is created. The default behavior of the empty script is to create a dataset of the same type as its input and to copy through the input dataset's structure. The GUI provides a selection menu where the user can choose from the five primary VTK dataset types for the output. The GUI also provides a text entry area where the user can type, edit or paste in a python script.

The following figure shows a python script that modifies the geometry of its input dataset.

This result is produced by the following python script set as the Script

<source lang="python"> pdi = self.GetPolyDataInput() pdo = self.GetPolyDataOutput() newPoints = vtk.vtkPoints() numPoints = pdi.GetNumberOfPoints() for i in range(0, numPoints):

   coord = pdi.GetPoint(i)
   x, y, z = coord[:3]
   x = x * 1
   y = y * 1
   z = 1 + z*0.3
   newPoints.InsertPoint(i, x, y, z)

pdo.SetPoints(newPoints) </source>

PyScriptFig1.jpg

Dealing with Composite Datasets

Readers such as Ensight, Exodus produce multiblock datasets. Here's the above squish script modified to work with multiblock datasets.

<source lang="python"> def flatten(input, output):

   # Copy the cells etc.
   output.ShallowCopy(input)
   newPoints = vtk.vtkPoints()
   numPoints = input.GetNumberOfPoints()
   for i in range(0, numPoints):
       coord = input.GetPoint(i)
       x, y, z = coord[:3]
       x = x * 1
       y = y * 1
       z = 10 + 0.1*z
       newPoints.InsertPoint(i, x, y, z)
   output.SetPoints(newPoints)

input = self.GetInputDataObject(0, 0) output = self.GetOutputDataObject(0)

if input.IsA("vtkMultiBlockDataSet"):

   output.CopyStructure(input)
   iter = input.NewIterator()
   iter.UnRegister(None)
   iter.InitTraversal()
   while not iter.IsDoneWithTraversal():
       curInput = iter.GetCurrentDataObject()
       curOutput = curInput.NewInstance()
       curOutput.UnRegister(None)
       output.SetDataSet(iter, curOutput)
       flatten(curInput, curOutput)
       iter.GoToNextItem();

else:

 flatten(input, output)

</source>

PythonProgrammableFIlterMultiBlock.png

Changing Data Type

Here we are taking in vtkPolyData and producing a vtkImageData from it. Note that the "Output Data Set Type" combo-box in the GUI has been changed to vtkImageData before hitting the Apply button.

The following figure shows a python script that produces an image data output with one cell per point in its input polygonal dataset.

<source lang="python">

  1. this example creates an Nx1x1 imagedata output
  2. and populates its cells with the point centered
  3. scalars of the input dataset
  1. This part of the code is to be placed in the "Script" entry:
  1. get a hold of the input

pdi = self.GetInput() numPts = pdi.GetNumberOfPoints()

  1. create the output dataset with one cell per point

ido = self.GetOutput() ido.SetDimensions(numPts+1,2,2) ido.SetOrigin(-1,-1,-1) ido.SetSpacing(.1,.1,.1)

  1. On ParaView 3.8 to 3.12
  2. ido.SetWholeExtent(0,numPts,0,1,0,1)
  3. ido.AllocateScalars()
  1. On ParaView 3.98, 4.0 and 4.1

ido.SetExtent(0,numPts,0,1,0,1) ido.AllocateScalars(vtk.VTK_FLOAT,1)

  1. choose an input point data array to copy

ivals = pdi.GetPointData().GetScalars() ca = vtk.vtkFloatArray() ca.SetName(ivals.GetName()) ca.SetNumberOfComponents(1) ca.SetNumberOfTuples(numPts)

  1. add the new array to the output

ido.GetCellData().AddArray(ca)

  1. copy the values over element by element

for i in range(0, numPts):

 ca.SetValue(i, ivals.GetValue(i))
  1. This part of the code is to be placed in the "RequestInformation Script" entry:

from paraview import util

pdi = self.GetInput() numPts = pdi.GetNumberOfPoints()

util.SetOutputWholeExtent(self, [0,numPts,0,1,0,1]) </source>

PyScriptFig2.jpg

Generating Data (Programmable Source)

Along with processing data, python can be used to produce data i.e. act as a data source. Under the Sources menu, we have the Programmable Source which can be used to produce data using Python.

Here's an example that generates a helix. Similar to the programmable filter, the output type must be choosen using the "Output Data Set Type" combo-box before hitting Apply.

The following is a 'Programmable Source' example that generates a Helix curve.

<source lang="python">

  1. This script generates a helix curve.
  2. This is intended as the script of a 'Programmable Source'

import math

numPts = 80 # Points along Helix length = 8.0 # Length of Helix rounds = 3.0 # Number of times around

  1. Get a vtk.PolyData object for the output

pdo = self.GetPolyDataOutput()

  1. This will store the points for the Helix

newPts = vtk.vtkPoints() for i in range(0, numPts):

  #Generate the Points along the Helix
  x = i*length/numPts
  y = math.sin(i*rounds*2*math.pi/numPts)
  z = math.cos(i*rounds*2*math.pi/numPts)
  #Insert the Points into the vtkPoints object
  #The first parameter indicates the reference.
  #value for the point. Here we add them sequentially.
  #Note that the first point is at index 0 (not 1).
  newPts.InsertPoint(i, x,y,z)
  1. Add the points to the vtkPolyData object
  2. Right now the points are not associated with a line -
  3. it is just a set of unconnected points. We need to
  4. create a 'cell' object that ties points together
  5. to make a curve (in this case). This is done below.
  6. A 'cell' is just an object that tells how points are
  7. connected to make a 1D, 2D, or 3D object.

pdo.SetPoints(newPts)

  1. Make a vtkPolyLine which holds the info necessary
  2. to create a curve composed of line segments. This
  3. really just hold constructor data that will be passed
  4. to vtkPolyData to add a new line.

aPolyLine = vtk.vtkPolyLine()

  1. Indicate the number of points along the line

aPolyLine.GetPointIds().SetNumberOfIds(numPts) for i in range(0,numPts):

  #Add the points to the line. The first value indicates
  #the order of the point on the line. The second value
  #is a reference to a point in a vtkPoints object. Depends
  #on the order that Points were added to vtkPoints object.
  #Note that this will not be associated with actual points
  #until it is added to a vtkPolyData object which holds a
  #vtkPoints object.
  aPolyLine.GetPointIds().SetId(i, i)
  1. Allocate the number of 'cells' that will be added. We are just
  2. adding one vtkPolyLine 'cell' to the vtkPolyData object.

pdo.Allocate(1, 1)

  1. Add the poly line 'cell' to the vtkPolyData object.

pdo.InsertNextCell(aPolyLine.GetCellType(), aPolyLine.GetPointIds())

  1. The Helix is ready to plot! Click 'Apply'.

</source>

HelixExample.jpg

Programmable Source: Double Helix

An example that draws a double helix with connecting lines (like DNA). Provides an example of using multiple drawing objects 'cells' in the same vtkPolyData output object. The 'Output Data Set Type' should be set of 'vtkPolyData'.

<source lang="python">

  1. This script generates a helix double.
  2. This is intended as the script of a 'Programmable Source'

import math

numPts = 80 # Points along each Helix length = 8.0 # Length of each Helix rounds = 3.0 # Number of times around phase_shift = math.pi/1.5 # Phase shift between Helixes

  1. Get a vtk.PolyData object for the output

pdo = self.GetPolyDataOutput()

  1. This will store the points for the Helix

newPts = vtk.vtkPoints() for i in range(0, numPts):

  #Generate Points for first Helix
  x = i*length/numPts
  y = math.sin(i*rounds*2*math.pi/numPts)
  z = math.cos(i*rounds*2*math.pi/numPts)
  newPts.InsertPoint(i, x,y,z)
  
  #Generate Points for second Helix. Add a phase offset to y and z.
  y = math.sin(i*rounds*2*math.pi/numPts+phase_shift)
  z = math.cos(i*rounds*2*math.pi/numPts+phase_shift)
  #Offset Helix 2 pts by 'numPts' to keep separate from Helix 1 Pts
  newPts.InsertPoint(i+numPts, x,y,z)
  1. Add the points to the vtkPolyData object

pdo.SetPoints(newPts)

  1. Make two vtkPolyLine objects to hold curve construction data

aPolyLine1 = vtk.vtkPolyLine() aPolyLine2 = vtk.vtkPolyLine()

  1. Indicate the number of points along the line

aPolyLine1.GetPointIds().SetNumberOfIds(numPts) aPolyLine2.GetPointIds().SetNumberOfIds(numPts) for i in range(0,numPts):

  #First Helix - use the first set of points
  aPolyLine1.GetPointIds().SetId(i, i)
  #Second Helix - use the second set of points
  #(Offset the point reference by 'numPts').
  aPolyLine2.GetPointIds().SetId(i,i+numPts)
  1. Allocate the number of 'cells' that will be added.
  2. Two 'cells' for the Helix curves, and one 'cell'
  3. for every 3rd point along the Helixes.

links = range(0,numPts,3) pdo.Allocate(2+len(links), 1)

  1. Add the poly line 'cell' to the vtkPolyData object.

pdo.InsertNextCell(aPolyLine1.GetCellType(), aPolyLine1.GetPointIds()) pdo.InsertNextCell(aPolyLine2.GetCellType(), aPolyLine2.GetPointIds())

for i in links:

  #Add a line connecting the two Helixes.
  aLine = vtk.vtkLine()
  aLine.GetPointIds().SetId(0, i)
  aLine.GetPointIds().SetId(1, i+numPts)
  pdo.InsertNextCell(aLine.GetCellType(), aLine.GetPointIds())

</source>

DoubleHelix.jpg

Examples

Here are some more examples of simple ParaView 3 python filters.

Paraview is built from VTK, and the python bindings for Paraview mirror the python bindings for VTK although the package names seem to be different. For example, there is a vtk.vtkLine in VTK which seems to behave the same as paraview.vtkLine in Paraview. While the online documentation for Paraview seems to be pretty limited, there is more extensive documentation with examples on the VTK documentation site. Examples are organized by class name and often include Python examples along with c++ and TCL examples:

VTK Example Code

The examples list is a little hard to navigate. Perhaps an easier way to find examples is to go to the class list, look up the class you are interested in, and click on the class name which brings up the documentation for that class. Many of the class pages include an 'examples' section that links to code examples for the class. From the examples page you can choose the 'Python' example associated with the particular class (not every class has a Python example). The VTK class index can be found here:

VTK Class Descriptions

NOTE: You may not be able to directly copy the examples from the VTK documentation because the package names may not match Paraview package names, but with a little modification I have been able to get examples from VTK to work in Paraview.

Add Colors to a PolyData

<source lang="python"> input = self.GetPolyDataInput(); output = self.GetPolyDataOutput();

colors = vtk.vtkUnsignedCharArray(); colors.SetNumberOfComponents(3); colors.SetName("Colors");

numPoints = input.GetNumberOfPoints() for i in xrange(0, numPoints):

   colors.InsertNextTuple3(255,0,0);

output.GetPointData().AddArray(colors) del colors </source>

Notes

  • don't forget to un-check the map scalars display panel properties check box.

Center Data

<source lang="python"> input = self.GetPolyDataInput() output = self.GetPolyDataOutput()

numPoints = input.GetNumberOfPoints() sumx = 0 sumy = 0 sumz = 0 for i in xrange(0, numPoints):

   coord = input.GetPoint(i)
   x, y, z = coord[:3]
   sumx = sumx + x
   sumy = sumy + y
   sumz = sumz + z

tx = -sumx/numPoints ty = -sumy/numPoints tz = -sumz/numPoints

  1. note: thanks to pipeline improvements in VTK 6
  2. shallow copy of input is not needed for PV version >= 4.0

ic = input.NewInstance() ic.ShallowCopy(input)

transform = vtk.vtkTransform() transform.Translate(tx,ty,tz)

transformFilter=vtk.vtkTransformPolyDataFilter() transformFilter.SetTransform(transform) transformFilter.SetInputData(ic) transformFilter.Update()

output.ShallowCopy(transformFilter.GetOutputPort(0)) </source>

Center Data using numpy

<source lang="python"> from paraview.vtk import dataset_adapter as DA

pdi = self.GetInputDataObject(0,0) pdo = self.GetOutputDataObject(0) pdo.CopyAttributes(pdi)

old_pts = inputs[0].Points new_pts = old_pts - mean(old_pts,axis=0) arr = DA.numpyTovtkDataArray(new_pts, 'newpts') pdo.GetPoints().SetData(arr) </source>


ParaView: [Welcome | Site Map]