Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Vector image support for imagetoVTK #6

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Feb 25, 2019
Merged

Vector image support for imagetoVTK #6

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
d71b638
Vector image support for imagetoVTK
JonTong Feb 23, 2019
aa37d25
Altered docstring for imageToVTK and added example to image.py
JonTong Feb 23, 2019
190dff7
Formatting in docstring
JonTong Feb 23, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 11 additions & 0 deletions examples/image.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,3 +43,14 @@

imageToVTK("./image", cellData = {"pressure" : pressure}, pointData = {"temp" : temp} )

fluxx = np.random.rand(ncells).reshape( (nx, ny, nz), order='F')
fluxy = np.random.rand(ncells).reshape( (nx, ny, nz), order='F')
fluxz = np.random.rand(ncells).reshape( (nx, ny, nz), order='F')
flux = (fluxx, fluxy, fluxz)

Efieldx = np.random.rand(npoints).reshape( (nx + 1, ny + 1, nz + 1), order='F')
Efieldy = np.random.rand(npoints).reshape( (nx + 1, ny + 1, nz + 1), order='F')
Efieldz = np.random.rand(npoints).reshape( (nx + 1, ny + 1, nz + 1), order='F')
Efield = (Efieldx,Efieldy,Efieldz)

imageToVTK("./image", cellData={"flux" : flux}, pointData = {"Efield" : Efieldx} )
17 changes: 12 additions & 5 deletions pyevtk/hl.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -79,13 +79,14 @@ def imageToVTK(path, origin = (0.0,0.0,0.0), spacing = (1.0,1.0,1.0), cellData =
spacing: grid spacing (default = (1,1,1))
cellData: dictionary containing arrays with cell centered data.
Keys should be the names of the data arrays.
Arrays must have the same dimensions in all directions and must contain
only scalar data.
Arrays must have the same dimensions in all directions and can contain
scalar data ([n,n,n]) or vector data ([n,n,n],[n,n,n],[n,n,n]).
nodeData: dictionary containing arrays with node centered data.
Keys should be the names of the data arrays.
Arrays must have same dimension in each direction and
they should be equal to the dimensions of the cell data plus one and
must contain only scalar data.
can contain scalar data ([n+1,n+1,n+1]) or
vector data ([n+1,n+1,n+1],[n+1,n+1,n+1],[n+1,n+1,n+1]).

RETURNS:
Full path to saved file.
Expand All @@ -100,11 +101,17 @@ def imageToVTK(path, origin = (0.0,0.0,0.0), spacing = (1.0,1.0,1.0), cellData =
if cellData != None:
keys = list(cellData.keys())
data = cellData[keys[0]]
end = data.shape
if hasattr(data,'shape'):
end = data.shape
elif(data[0].ndim==3 and data[1].ndim==3 and data[2].ndim==3):
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

been forever since I've looked at this code but doesn't this assume 3D vector arrays?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I guess it does - I only thought about using it for 3D vector fields like velocity etc.

Would it be worth generalising it to higher dimensional vectors?

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think it should support whatever the format supports but I've long passed the reigns to @renefritze so I'll let him decide

end = data[0].shape
elif pointData != None:
keys = list(pointData.keys())
data = pointData[keys[0]]
end = data.shape
if hasattr(data,'shape'):
end = data.shape
elif(data[0].ndim==3 and data[1].ndim==3 and data[2].ndim==3):
end = data[0].shape
end = (end[0] - 1, end[1] - 1, end[2] - 1)

# Write data to file
Expand Down