FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
Engineering_Python/PointsToUSGrid.py at master · OnScale/Engineering_Python · GitHub
Uh oh!
There was an error while loading.
Please reload this page
.
OnScale
/
Engineering_Python
Public
Notifications
You must be signed in to change notification settings
Fork
1
Star
0
Code
Issues
0
Pull requests
1
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Actions
Projects
Security and quality
Insights
Expand file tree
Breadcrumbs
Engineering_Python
/
PointsToUSGrid.py
Copy path
More file actions
More file actions
Latest commit
History
History
History
127 lines (104 loc) · 3.09 KB
Breadcrumbs
Engineering_Python
/
PointsToUSGrid.py
Copy path
File metadata and controls
127 lines (104 loc) · 3.09 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
# -*- coding: utf-8 -*-
"""
Created on Thu Nov 20 14:39:07 2014
Author: Gerry Harvey
- Read in XYZ point cloud mesh
- Read in Scalar values and apply a single one
- Create mesh from delaunay
- Export to VTK format for visualisation in Paraview
Updates:
- Read all 3 velocities in as a vector and apply to dataset (SetVectors?)
"""
# This example shows how to use Delaunay3D with alpha shapes.
import
vtk
from
vtk
import
*
# prevent error window from popping up
ow
=
vtk
.
vtkOutputWindow
()
ow
.
DebugOff
()
ow
.
GlobalWarningDisplayOff
()
large
=
1
if
(
large
==
1
):
filename
=
'0010.csv'
Name
=
'Vol'
cols
=
[
0
,
1
,
2
,
3
,
4
,
5
]
else
:
filename
=
'grid_vels.csv'
Name
=
'Surf'
cols
=
[
1
,
2
,
3
,
4
,
5
,
6
]
# Read positional & Vel data
data
=
loadtxt
(
filename
,
delimiter
=
','
,
usecols
=
cols
)
X
=
data
[:,
3
];
Y
=
data
[:,
4
];
Z
=
data
[:,
5
]
Xv
=
data
[:,
0
];
Yv
=
data
[:,
1
];
Zv
=
data
[:,
2
]
# Get number of unique points
Np
=
len
(
data
)
# The points to be triangulated are generated randomly in the unit
# cube located at the origin. The points are then associated with a
# vtkUnstructuredGrid
points
=
vtk
.
vtkPoints
()
# Set up arrays for data on points
xvel
=
vtk
.
vtkFloatArray
()
xvel
.
SetName
(
"X-velocity"
)
xvel
.
SetNumberOfComponents
(
1
)
xvel
.
SetNumberOfValues
(
Np
)
# Also create vel DataSet
for
i
in
range
(
0
,
Np
):
points
.
InsertPoint
(
i
,
X
[
i
],
Y
[
i
],
Z
[
i
])
xvel
.
SetValue
(
i
,
Xv
[
i
])
# Create unstructured grid object
profile
=
vtk
.
vtkUnstructuredGrid
()
profile
.
SetPoints
(
points
)
profile
.
GetPointData
().
SetScalars
(
xvel
)
# Option to subsample the points using MaskPoints
subsamp
=
0
if
(
subsamp
==
1
):
ptMask
=
vtk
.
vtkMaskPoints
()
ptMask
.
SetInput
(
profile
)
ptMask
.
SetOnRatio
(
3
)
ptMask
.
RandomModeOff
()
ptMask
.
GenerateVerticesOn
()
# perform delaunay on points
delny
=
vtk
.
vtkDelaunay3D
()
delny
.
SetInput
(
ptMask
.
GetOutput
())
delny
.
SetTolerance
(
0.0001
)
delny
.
SetAlpha
(
0
)
delny
.
BoundingTriangulationOff
()
else
:
# perform delaunay on points
delny
=
vtk
.
vtkDelaunay3D
()
delny
.
SetInput
(
profile
)
delny
.
SetTolerance
(
0.0001
)
delny
.
SetAlpha
(
0.002
)
delny
.
BoundingTriangulationOff
()
# Extract convex hull data
ch
=
vtkDataSetSurfaceFilter
()
ch
.
SetInputConnection
(
delny
.
GetOutputPort
())
ch
.
Update
()
chOUT
=
vtkPolyDataWriter
()
chOUT
.
SetFileName
(
"ConvexHull.vtk"
)
chOUT
.
SetInput
(
ch
.
GetOutput
())
chOUT
.
Write
()
# Copy to 'new' Unstructured Grid
us
=
vtk
.
vtkUnstructuredGrid
()
delny
.
Update
()
us
.
DeepCopy
(
delny
.
GetOutputDataObject
(
0
))
# Allocate Scalers to points in Unstructured Grid
Nc
=
us
.
GetNumberOfCells
()
mat
=
vtk
.
vtkFloatArray
()
mat
.
SetName
(
"Material"
)
mat
.
SetNumberOfComponents
(
1
)
mat
.
SetNumberOfValues
(
Nc
)
for
i
in
range
(
0
,
Nc
):
mat
.
SetValue
(
i
,
1
)
us
.
GetPointData
().
SetScalars
(
xvel
)
us
.
GetCellData
().
SetScalars
(
mat
)
us
.
Update
()
# write data to file for paraview
vtkDataOut
=
vtk
.
vtkUnstructuredGridWriter
()
vtkDataOut
.
SetFileName
(
Name
+
"Mesh.vtk"
)
vtkDataOut
.
SetInputConnection
(
us
.
GetProducerPort
())
vtkDataOut
.
Write
()
# Read data in
reader
=
vtk
.
vtkUnstructuredGridReader
()
reader
.
SetFileName
(
Name
+
"Mesh.vtk"
)
reader
.
Update
()
vtkDataIn
=
reader
.
GetOutput
()
Back
|
FazBrowse Home
|
New Git URL