mirror of
https://github.com/OpenFOAM/ThirdParty-6.git
synced 2025-12-08 06:57:43 +00:00
ParaView-5.0.1: Added the source-tree to ThirdParty-dev and patched as described in the README file
Resolves bug-report http://bugs.openfoam.org/view.php?id=2098
This commit is contained in:
@ -0,0 +1,70 @@
|
||||
# Set up a basic scene for rendering.
|
||||
from paraview.simple import *
|
||||
import os
|
||||
import sys
|
||||
|
||||
script = """
|
||||
import paraview.numpy_support
|
||||
|
||||
# Utility to get next color
|
||||
def getNextColor():
|
||||
colors = 'bgrcmykw'
|
||||
for c in colors:
|
||||
yield c
|
||||
|
||||
# This function must be defined. It is where specific data arrays are requested.
|
||||
def setup_data(view):
|
||||
print "Setting up data"
|
||||
|
||||
# This function must be defined. It is where the actual rendering commands for matplotlib go.
|
||||
def render(view,width,height):
|
||||
from paraview import python_view
|
||||
figure = python_view.matplotlib_figure(width,height)
|
||||
|
||||
ax = figure.add_subplot(111)
|
||||
ax.hold = True
|
||||
numObjects = view.GetNumberOfVisibleDataObjects()
|
||||
print "num visible objects: ", numObjects
|
||||
for i, color in zip(xrange(0,numObjects), getNextColor()):
|
||||
dataObject = view.GetVisibleDataObjectForRendering(i)
|
||||
if dataObject:
|
||||
vtk_points = dataObject.GetPoints()
|
||||
if vtk_points:
|
||||
vtk_points_data = vtk_points.GetData()
|
||||
pts = paraview.numpy_support.vtk_to_numpy(vtk_points_data)
|
||||
x, y = pts[:,0], pts[:,1]
|
||||
ax.scatter(x, y, color=color)
|
||||
|
||||
ax.hold = False
|
||||
|
||||
return python_view.figure_to_image(figure)
|
||||
"""
|
||||
|
||||
view = CreateView("PythonView")
|
||||
view.Script = script
|
||||
|
||||
cone = Cone()
|
||||
Show(cone, view)
|
||||
|
||||
sphere = Sphere()
|
||||
Show(sphere, view)
|
||||
|
||||
Render()
|
||||
|
||||
try:
|
||||
baselineIndex = sys.argv.index('-B')+1
|
||||
baselinePath = sys.argv[baselineIndex]
|
||||
except:
|
||||
print "Could not get baseline directory. Test failed."
|
||||
|
||||
baseline_file = os.path.join(baselinePath, "TestPythonViewMatplotlibScript.png")
|
||||
import vtk.test.Testing
|
||||
vtk.test.Testing.VTK_TEMP_DIR = vtk.util.misc.vtkGetTempDir()
|
||||
vtk.test.Testing.compareImage(view.GetRenderWindow(), baseline_file, threshold=25)
|
||||
vtk.test.Testing.interact()
|
||||
|
||||
Delete(cone)
|
||||
del cone
|
||||
|
||||
Delete(sphere)
|
||||
del sphere
|
||||
Reference in New Issue
Block a user