| 123456789101112131415161718192021222324252627 |
- """Load an existing dataset and draw
- the streamlines of the velocity field"""
- from vedo import *
- ######################## vtk
- import vtk
- # Read the data and specify which scalars and vectors to read.
- pl3d = vtk.vtkMultiBlockPLOT3DReader()
- fpath = download(dataurl+"combxyz.bin")
- pl3d.SetXYZFileName(fpath)
- fpath = download(dataurl+"combq.bin")
- pl3d.SetQFileName(fpath)
- pl3d.SetScalarFunctionNumber(100)
- pl3d.SetVectorFunctionNumber(202)
- pl3d.Update()
- # this vtkStructuredGridData already has a vector field:
- sdata = pl3d.GetOutput().GetBlock(0)
- ######################## vedo
- domain = UnstructuredGrid(sdata).alpha(0.1).c('white')
- probe = Grid(s=[5,5], res=[6,6], c='white').rotate_y(90).pos(5,0,29)
- streamlines = domain.compute_streamlines(probe)
- streamlines.celldata.select("SeedIds")
- streamlines.cmap("Set1", on='cells')
- print(streamlines)
- show(domain, streamlines, probe, __doc__, axes=7, bg='bb').close()
|
