Source type plugin with hard-coded path
This commit is contained in:
Yeongdo Park
parent
07163d86bb
commit
b23a2e3df5
1 changed files with 104 additions and 15 deletions
|
|
@ -1,4 +1,5 @@
|
|||
# same imports as earlier.
|
||||
import numpy as np
|
||||
|
||||
from vtkmodules.vtkCommonDataModel import vtkDataSet
|
||||
from vtkmodules.util.vtkAlgorithm import VTKPythonAlgorithmBase
|
||||
from vtkmodules.numpy_interface import dataset_adapter as dsa
|
||||
|
|
@ -7,8 +8,8 @@ from vtkmodules.numpy_interface import dataset_adapter as dsa
|
|||
from paraview.util.vtkAlgorithm import smproxy, smproperty, smdomain
|
||||
|
||||
# to add a source, instead of a filter, use the `smproxy.source` decorator.
|
||||
@smproxy.source(label="Python-based Superquadric Source Example")
|
||||
class PythonSuperquadricSource(VTKPythonAlgorithmBase):
|
||||
@smproxy.source(label="POSTECH Turbulent Combustion DNS Reader")
|
||||
class PythonDnsDataReader(VTKPythonAlgorithmBase):
|
||||
"""This is dummy VTKPythonAlgorithmBase subclass that
|
||||
simply puts out a Superquadric poly data using a vtkSuperquadricSource
|
||||
internally"""
|
||||
|
|
@ -16,15 +17,76 @@ class PythonSuperquadricSource(VTKPythonAlgorithmBase):
|
|||
VTKPythonAlgorithmBase.__init__(self,
|
||||
nInputPorts=0,
|
||||
nOutputPorts=1,
|
||||
outputType='vtkPolyData')
|
||||
from vtkmodules.vtkFiltersSources import vtkSuperquadricSource
|
||||
self._realAlgorithm = vtkSuperquadricSource()
|
||||
outputType='vtkRectilinearGrid')
|
||||
|
||||
def RequestData(self, request, inInfo, outInfo):
|
||||
from vtkmodules.vtkCommonDataModel import vtkPolyData
|
||||
self._realAlgorithm.Update()
|
||||
output = vtkPolyData.GetData(outInfo, 0)
|
||||
output.ShallowCopy(self._realAlgorithm.GetOutput())
|
||||
import vtk
|
||||
from vtkmodules.vtkCommonDataModel import vtkRectilinearGrid
|
||||
output = vtkRectilinearGrid.GetData(outInfo, 0)
|
||||
|
||||
filename = 'E:/cygwin64/home/ignis/jupyter_notebook_home/FSD-IC1-IC4/IC4-samples/fort.15550'
|
||||
t, nx, ny, nz, U, V, W, Y0, Y1 = self._read_data(filename)
|
||||
|
||||
u = U.ravel()
|
||||
v = V.ravel()
|
||||
w = W.ravel()
|
||||
y1 = Y1.ravel()
|
||||
|
||||
x = np.arange(nx) * 4*np.pi/nx
|
||||
y = np.arange(ny) * 2*np.pi/ny
|
||||
z = np.arange(nz) * 2*np.pi/nz
|
||||
|
||||
# Create a rectilinear grid by defining three arrays specifying the
|
||||
# coordinates in the x-y-z directions.
|
||||
xCoords = vtk.vtkFloatArray()
|
||||
xCoords.SetNumberOfTuples(len(x))
|
||||
for i, xi in enumerate(x):
|
||||
xCoords.SetTuple1(i, xi)
|
||||
|
||||
yCoords = vtk.vtkFloatArray()
|
||||
yCoords.SetNumberOfTuples(len(y))
|
||||
for i, yi in enumerate(y):
|
||||
yCoords.SetTuple1(i, yi)
|
||||
|
||||
zCoords = vtk.vtkFloatArray()
|
||||
zCoords.SetNumberOfTuples(len(z))
|
||||
for i, zi in enumerate(z):
|
||||
zCoords.SetTuple1(i, zi)
|
||||
|
||||
# The coordinates are assigned to the rectilinear grid. Make sure that
|
||||
# the number of values in each of the XCoordinates, YCoordinates,
|
||||
# and ZCoordinates is equal to what is defined in SetDimensions().
|
||||
#
|
||||
output.SetDimensions(len(x), len(y), len(z))
|
||||
output.SetXCoordinates(xCoords)
|
||||
output.SetYCoordinates(yCoords)
|
||||
output.SetZCoordinates(zCoords)
|
||||
|
||||
numPoints = output.GetNumberOfPoints()
|
||||
|
||||
velocity = vtk.vtkFloatArray();
|
||||
velocity.SetNumberOfComponents(3);
|
||||
velocity.SetNumberOfTuples(numPoints);
|
||||
velocity.SetName("U");
|
||||
|
||||
for i in range(0, numPoints):
|
||||
velocity.SetTuple3(i, u[i], v[i], w[i]);
|
||||
|
||||
output.GetPointData().AddArray(velocity)
|
||||
del velocity
|
||||
|
||||
c = vtk.vtkFloatArray();
|
||||
c.SetNumberOfTuples(numPoints);
|
||||
c.SetName("c");
|
||||
|
||||
for i in range(0, numPoints):
|
||||
c.SetTuple1(i, 1.0-y1[i]);
|
||||
|
||||
output.GetPointData().AddArray(c)
|
||||
del c
|
||||
|
||||
del t, nx, ny, nz, u, v, w, Y0, Y1, x, y, z
|
||||
|
||||
return 1
|
||||
|
||||
# for anything too complex or not yet supported, you can explicitly
|
||||
|
|
@ -38,29 +100,56 @@ class PythonSuperquadricSource(VTKPythonAlgorithmBase):
|
|||
<Documentation>Set center of the superquadric</Documentation>
|
||||
</DoubleVectorProperty>""")
|
||||
def SetCenter(self, x, y, z):
|
||||
self._realAlgorithm.SetCenter(x,y,z)
|
||||
self.Modified()
|
||||
|
||||
# In most cases, one can simply use available decorators.
|
||||
@smproperty.doublevector(name="Scale", default_values=[1, 1, 1])
|
||||
@smdomain.doublerange()
|
||||
def SetScale(self, x, y, z):
|
||||
self._realAlgorithm.SetScale(x,y,z)
|
||||
self.Modified()
|
||||
|
||||
@smproperty.intvector(name="ThetaResolution", default_values=16)
|
||||
def SetThetaResolution(self, x):
|
||||
self._realAlgorithm.SetThetaResolution(x)
|
||||
self.Modified()
|
||||
|
||||
@smproperty.intvector(name="PhiResolution", default_values=16)
|
||||
@smdomain.intrange(min=0, max=1000)
|
||||
def SetPhiResolution(self, x):
|
||||
self._realAlgorithm.SetPhiResolution(x)
|
||||
self.Modified()
|
||||
|
||||
@smproperty.doublevector(name="Thickness", default_values=0.3333)
|
||||
@smdomain.doublerange(min=1e-24, max=1.0)
|
||||
def SetThickness(self, x):
|
||||
self._realAlgorithm.SetThickness(x)
|
||||
self.Modified()
|
||||
|
||||
def _read_data (self, fname):
|
||||
import struct
|
||||
import sys
|
||||
import os
|
||||
with open(fname, 'rb') as f1 :
|
||||
f1.seek(0)
|
||||
|
||||
raw_info = f1.read(4+8*6+4)[4:-4]
|
||||
t = struct.unpack('d', raw_info[ 0: 8])[0]
|
||||
nx = struct.unpack('q', raw_info[ 8:16])[0]
|
||||
ny = struct.unpack('q', raw_info[16:24])[0]
|
||||
nz = struct.unpack('q', raw_info[24:32])[0]
|
||||
count = nx*ny*nz
|
||||
bSize = count*8 # size in bytes for a variable
|
||||
|
||||
dummy_len = (4+8*3+4) + (4+8*2+4) + (4+8*2+4) + (4+8*2+4) + 4
|
||||
dummy = f1.read(dummy_len)
|
||||
#dummy = f1.read(4)
|
||||
|
||||
#raw_field = f1.read(4+bSize*5+4)[4:-4]
|
||||
V = np.fromfile(f1, dtype=np.double, count=(3*count)).reshape((3,nz,ny,nx))
|
||||
s = np.fromfile(f1, dtype=np.double, count=(2*count)).reshape((2,nz,ny,nx))
|
||||
|
||||
u = V[0]
|
||||
v = V[1]
|
||||
w = V[2]
|
||||
|
||||
Y0 = s[0]
|
||||
Y1 = s[1]
|
||||
|
||||
return t, nx, ny, nz, u, v, w, Y0, Y1
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue