Python vtk.vtkPlaneSource() Examples
The following are 9
code examples of vtk.vtkPlaneSource().
You can vote up the ones you like or vote down the ones you don't like,
and go to the original project or source file by following the links above each example.
You may also want to check out all available functions/classes of the module
vtk
, or try the search function
.
.
Example #1
| Source File: utility.py From ILCC with BSD 2-Clause "Simplified" License | 6 votes |
|
def draw_one_grd_vtk(ls): # arr:[a,b,c,d],a:orig, b, point1, c,point 2, d,color
source = vtk.vtkPlaneSource()
source.SetOrigin(ls[0])
source.SetPoint1(ls[1])
source.SetPoint2(ls[2])
source.Update()
# source.SetPoint1(0, 0, 0)
# source.SetPoint2(4, 3, 0)
# mapper
mapper = vtk.vtkPolyDataMapper()
color = vtk.vtkUnsignedCharArray()
color.SetName("colors")
color.SetNumberOfComponents(3)
# color_tup = np.random.randint(1, 255, 3)
color.SetNumberOfTuples(source.GetOutput().GetNumberOfCells())
for i in xrange(source.GetOutput().GetNumberOfCells()):
color_tup = np.array([255, 255, 255]) * ls[3]
color.InsertTuple(i, color_tup)
source.GetOutput().GetCellData().SetScalars(color)
mapper.SetInputConnection(source.GetOutputPort())
# actor
actor = vtk.vtkActor()
actor.SetMapper(mapper)
# assign actor to the renderer
# ren.AddActor(actor)
return actor
# generate the color list of the point cloud for different color styles. intens_rg: color by reflectance intensity (red:high green:low),
# intens: color by reflectance intensity (white:high back:low), autumn: matplotlib autumn color map, cool: matplotlib cool color map
Example #2
| Source File: vista_aux.py From gempy with GNU Lesser General Public License v3.0 | 6 votes |
|
def call_back_plane_move_changes(self, indices):
df_changes = self.model._orientations.df.loc[np.atleast_1d(indices).astype(int)][['X', 'Y', 'Z',
'G_x', 'G_y', 'G_z', 'id']]
for index, new_values_df in df_changes.iterrows():
new_center = new_values_df[['X', 'Y', 'Z']].values
new_normal = new_values_df[['G_x', 'G_y', 'G_z']].values
new_source = vtk.vtkPlaneSource()
new_source.SetCenter(new_center)
new_source.SetNormal(new_normal)
new_source.Update()
plane1 = self.orientations_widgets[index]
# plane1.SetInputData(new_source.GetOutput())
plane1.SetNormal(new_normal)
plane1.SetCenter(new_center[0], new_center[1], new_center[2])
_color_lot = self._get_color_lot(is_faults=True, is_basement=False, index='id')
plane1.GetPlaneProperty().SetColor(mcolors.hex2color(_color_lot[int(new_values_df['id'])]))
plane1.GetHandleProperty().SetColor(mcolors.hex2color(_color_lot[int(new_values_df['id'])]))
Example #3
| Source File: _vista.py From gempy with GNU Lesser General Public License v3.0 | 6 votes |
|
def call_back_plane_move_changes(self, indices):
df_changes = self.model._orientations.df.loc[np.atleast_1d(indices)][['X', 'Y', 'Z',
'G_x', 'G_y', 'G_z', 'id']]
for index, new_values_df in df_changes.iterrows():
new_center = new_values_df[['X', 'Y', 'Z']].values
new_normal = new_values_df[['G_x', 'G_y', 'G_z']].values
new_source = vtk.vtkPlaneSource()
new_source.SetCenter(new_center)
new_source.SetNormal(new_normal)
new_source.Update()
plane1 = self.p_widget.loc[index, 'val']
# plane1.SetInputData(new_source.GetOutput())
plane1.SetNormal(new_normal)
plane1.SetCenter(new_center[0], new_center[1], new_center[2])
plane1.GetPlaneProperty().SetColor(
parse_color(self.model._surfaces.df.set_index('id')['color'][new_values_df['id']])) # self.C_LOT[new_values_df['id']])
plane1.GetHandleProperty().SetColor(
parse_color(self.model._surfaces.df.set_index('id')['color'][new_values_df['id']]))
return True
Example #4
| Source File: graphics.py From robopy with MIT License | 5 votes |
|
def floor():
plane = vtk.vtkPlaneSource()
reader = vtk.vtkJPEGReader()
reader.SetFileName(pkg_resources.resource_filename("robopy", "media/imgs/floor.jpg"))
texture = vtk.vtkTexture()
texture.SetInputConnection(reader.GetOutputPort())
map_to_plane = vtk.vtkTextureMapToPlane()
map_to_plane.SetInputConnection(plane.GetOutputPort())
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputConnection(map_to_plane.GetOutputPort())
actor = vtk.vtkActor()
actor.SetMapper(mapper)
actor.SetTexture(texture)
return actor
Example #5
| Source File: test_code_vasp_01.py From PyChemia with MIT License | 5 votes |
|
def MakePlane():
"""
Make a plane as the source.
:return: vtkPolyData with normal and scalar data.
"""
source = vtk.vtkPlaneSource()
source.SetOrigin(-10.0, -10.0, 0.0)
source.SetPoint2(-10.0, 10.0, 0.0)
source.SetPoint1(10.0, -10.0, 0.0)
source.SetXResolution(20)
source.SetYResolution(20)
source.Update()
return MakeElevations(source.GetOutput())
Example #6
| Source File: pointobject.py From Det3D with Apache License 2.0 | 5 votes |
|
def CreatePlane(self, normal=None, origin=None):
"Create a plane (optionally with a given normal vector and origin)"
plane = vtk.vtkPlaneSource()
plane.SetXResolution(10)
plane.SetYResolution(10)
if not normal is None:
plane.SetNormal(normal)
if not origin is None:
plane.SetCenter(origin)
plane.Update()
self.pd = vtk.vtkPolyData()
self.pd.DeepCopy(plane.GetOutput())
self.scalars = None
self.SetupPipelineMesh()
Example #7
| Source File: geometric_objects.py From pyvista with MIT License | 4 votes |
|
def Plane(center=(0, 0, 0), direction=(0, 0, 1), i_size=1, j_size=1,
i_resolution=10, j_resolution=10):
"""Create a plane.
Parameters
----------
center : list or np.ndarray
Location of the centroid in [x, y, z]
direction : list or np.ndarray
Direction cylinder points to in [x, y, z]
i_size : float
Size of the plane in the i direction.
j_size : float
Size of the plane in the j direction.
i_resolution : int
Number of points on the plane in the i direction.
j_resolution : int
Number of points on the plane in the j direction.
Return
------
plane : pyvista.PolyData
Plane mesh
"""
planeSource = vtk.vtkPlaneSource()
planeSource.SetXResolution(i_resolution)
planeSource.SetYResolution(j_resolution)
planeSource.Update()
surf = pyvista.PolyData(planeSource.GetOutput())
surf.points[:, 0] *= i_size
surf.points[:, 1] *= j_size
surf.rotate_y(-90)
translate(surf, center, direction)
return surf
Example #8
| Source File: visualization_3d.py From gempy with GNU Lesser General Public License v3.0 | 4 votes |
|
def create_foliation(self, X, Y, Z, fn,
Gx, Gy, Gz,
n_plane=0, n_render=0, n_index=0, alpha=0.5):
"""
Method to create a plane given a foliation
Args:
X : X coord
Y: Y coord
Z: Z coord
fn (int): id
Gx (str): Component of the gradient x
Gy (str): Component of the gradient y
Gz (str): Component of the gradient z
n_plane (int): Number of the plane
n_render (int): Number of the render where the plane belongs
n_index (int): index value in the PandasDataframe of InupData.surface_points
alpha: Opacity of the plane
Returns:
vtk.vtkPlaneWidget
"""
Z = Z * self.ve
d = vtk.vtkPlaneWidget()
d.SetInteractor(self.interactor)
d.SetRepresentationToSurface()
# Position
source = vtk.vtkPlaneSource()
source.SetNormal(Gx, Gy, Gz)
source.SetCenter(X, Y, Z)
a, b, c, d_, e, f = self.extent
source.SetPoint1(X+self._e_dx*.01, Y-self._e_dy*.01, Z)
source.SetPoint2(X-self._e_dx*.01, Y+self._e_dy*.01, Z)
source.Update()
d.SetInputData(source.GetOutput())
d.SetHandleSize(.05)
min_extent = np.min([self._e_dx, self._e_dy, self._e_dz])
d.SetPlaceFactor(0.1)
d.PlaceWidget(a, b, c, d_, e, f)
d.SetNormal(Gx, Gy, Gz)
d.SetCenter(X, Y, Z)
d.GetPlaneProperty().SetColor(mcolors.hex2color(self.geo_model._surfaces.df.set_index('id')['color'][fn]))#self.C_LOT[fn])
d.GetHandleProperty().SetColor(mcolors.hex2color(self.geo_model._surfaces.df.set_index('id')['color'][fn]))#self.C_LOT[fn])
d.GetHandleProperty().SetOpacity(alpha)
d.SetCurrentRenderer(self.ren_list[n_render])
d.n_plane = n_plane
d.n_render = n_render
d.index = n_index
d.AddObserver("EndInteractionEvent", self.planesCallback)
d.AddObserver("InteractionEvent", self.Callback_camera_reset)
d.On()
return d
Example #9
| Source File: visualization_3d.py From gempy with GNU Lesser General Public License v3.0 | 4 votes |
|
def planesCallback_move_changes(self, indices):
df_changes = self.geo_model._orientations.df.loc[np.atleast_1d(indices)][['X', 'Y', 'Z', 'G_x', 'G_y', 'G_z', 'id']]
for index, new_values_df in df_changes.iterrows():
new_center = new_values_df[['X', 'Y', 'Z']].values
new_normal = new_values_df[['G_x', 'G_y', 'G_z']].values
new_source = vtk.vtkPlaneSource()
new_source.SetCenter(new_center)
new_source.SetNormal(new_normal)
new_source.Update()
plane1 = self.o_rend_1.loc[index, 'val']
# plane1.SetInputData(new_source.GetOutput())
plane1.SetNormal(new_normal)
plane1.SetCenter(new_center[0], new_center[1], new_center[2])
plane1.GetPlaneProperty().SetColor(mcolors.hex2color(
self.geo_model._surfaces.df.set_index('id')['color'][new_values_df['id']]))#self.C_LOT[new_values_df['id']])
plane1.GetHandleProperty().SetColor(mcolors.hex2color(
self.geo_model._surfaces.df.set_index('id')['color'][new_values_df['id']]))
plane2 = self.o_rend_2.loc[index, 'val']
plane2.SetInputData(new_source.GetOutput())
plane2.SetNormal(new_normal)
plane2.SetCenter(new_center[0], new_center[1], new_center[2])
plane2.GetPlaneProperty().SetColor(mcolors.hex2color(
self.geo_model._surfaces.df.set_index('id')['color'][new_values_df['id']]))
plane2.GetHandleProperty().SetColor(mcolors.hex2color(
self.geo_model._surfaces.df.set_index('id')['color'][new_values_df['id']]))
plane3 = self.o_rend_3.loc[index, 'val']
plane3.SetInputData(new_source.GetOutput())
plane3.SetNormal(new_normal)
plane3.SetCenter(new_center[0], new_center[1], new_center[2])
plane3.GetPlaneProperty().SetColor(mcolors.hex2color(
self.geo_model._surfaces.df.set_index('id')['color'][new_values_df['id']]))
plane3.GetHandleProperty().SetColor(mcolors.hex2color(
self.geo_model._surfaces.df.set_index('id')['color'][new_values_df['id']]))
plane4 = self.o_rend_4.loc[index, 'val']
plane4.SetInputData(new_source.GetOutput())
plane4.SetNormal(new_normal)
plane4.SetCenter(new_center[0], new_center[1], new_center[2])
plane4.GetPlaneProperty().SetColor(mcolors.hex2color(
self.geo_model._surfaces.df.set_index('id')['color'][new_values_df['id']]))
plane4.GetHandleProperty().SetColor(mcolors.hex2color(
self.geo_model._surfaces.df.set_index('id')['color'][new_values_df['id']]))
