Hi everyone,
I am currently trying to load a custom data format through a python plugin.
An external python module returns a 2D numpy array, which I want to process further in Paraview.
I initialized the plugin with VTKPythonAlgorithmBase.__init__(self, nInputPorts=0, nOutputPorts=1, outputType='vtkImageData')
For RequestData I modified the routine from the python plugin example in the paraview github repo as follows:
def RequestData(self, request, inInfoVec, outInfoVec):
from vtkmodules.vtkCommonDataModel import vtkImageData
from vtkmodules.vtkCommonCore import VTK_DOUBLE
#Get data for requested timestep
data_time = self._get_update_time(outInfoVec.GetInformationObject(0))
raw_data = self._get_raw_data(data_time)
#Create output adapter
output = vtkImageData.GetData(outInfoVec, 0)
#Convert numpy array to image
frame_data = paraview.python_view.numpy_to_image(raw_data)
#Set extent of data
frame_data.SetExtent(0,499,0,119,0,1)
#Give it a name
frame_data.GetPointData().GetScalars().SetName("sonar data")
#Copy to output
output.ShallowCopy(frame_data)
#Set scalars to "active" (no idea what that does, found another code example that did that)
output.GetPointData().SetActiveScalars("sonar data")
#Again, copied from another code example
output.Modified()
#Copied from the code example from the paraview github repo
if data_time is not None:
output.GetInformation().Set(output.DATA_TIME_STEP(), data_time)
#Return success
return 1
When I load a file using this code, Paraview seems to load the correct number of points and even the correct values (at least I think it does, because the scale is scaled correctly), but doesn’t show anything. I tried pressing the Reset Camera and Zoom To Data buttons, but to no avail.
from paraview import util
pdi = self.GetInput()
util.SetOutputWholeExtent(self,(0,499,0,120, 0,1))
to the RequestInformation function, but the only thing that did, is that the time shown in Paraview now only goes from zero to one (instead of the previous bounds).
I don’t think I quite understand.
If I use paraview.python_view.numpy_to_image on a 2D-array, do I still need to set dimensions, origin and spacing and allocate the scalars?
Or do I need to do that when copying the vtkImageData I get from numpy_to_image to the output?
In other words, is ido in your example frame_data or output from the RequestData function I pasted above?
I tried writing a source plugin that would return a random image as a minimum case.
from paraview.util.vtkAlgorithm import *
import numpy as np
from paraview import python_view
from vtk.numpy_interface import algorithms as algs
from vtk.numpy_interface import dataset_adapter as dsa
from vtkmodules.vtkCommonDataModel import vtkImageData
from vtkmodules.vtkCommonDataModel import vtkDataSet
import vtk
@smproxy.source(name="Random image source",
label="Python-based test source for random images")
class PythonTestPointSource(VTKPythonAlgorithmBase):
def __init__(self):
VTKPythonAlgorithmBase.__init__(self,
nInputPorts=0,
nOutputPorts=1,
outputType='vtkImageData')
def RequestData(self, request, inInfo, outInfo):
#output = vtkPolyData.GetData(outInfo, 0)
output = dsa.WrapDataObject(vtkDataSet.GetData(outInfo))
dims = (320,240)
random_array = np.random.rand(dims[0],dims[1])
image = python_view.numpy_to_image(random_array)
#from the discourse post
image.SetDimensions(320,240,1)
image.SetOrigin(0,0,0)
image.SetSpacing(1,1,1)
image.AllocateScalars(vtk.VTK_FLOAT, 1)
output.ShallowCopy(image)
return 1
This gives me the correct number of points and the correct dimensions in Paraview, but neither on the “Slice”, nor on the “Points” nor on the “Outline” views, anything is rendered.
If I remove the block with your previous suggestions, the only thing that changes is, that my x and y extent swap places.
