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 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157
|
#!/usr/bin/env python
import vtk
from vtk.test import Testing
from vtk.util.misc import vtkGetDataRoot
VTK_DATA_ROOT = vtkGetDataRoot()
ren1 = vtk.vtkRenderer()
renWin = vtk.vtkRenderWindow()
renWin.SetMultiSamples(0)
renWin.AddRenderer(ren1)
iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)
pnmReader = vtk.vtkBMPReader()
pnmReader.SetFileName(VTK_DATA_ROOT + "/Data/masonry.bmp")
texture = vtk.vtkTexture()
texture.SetInputConnection(pnmReader.GetOutputPort())
triangleStripPoints = vtk.vtkPoints()
triangleStripPoints.SetNumberOfPoints(5)
triangleStripPoints.InsertPoint(0, 0, 1, 0)
triangleStripPoints.InsertPoint(1, 0, 0, .5)
triangleStripPoints.InsertPoint(2, 1, 1, .3)
triangleStripPoints.InsertPoint(3, 1, 0, .6)
triangleStripPoints.InsertPoint(4, 2, 1, .1)
triangleStripTCoords = vtk.vtkFloatArray()
triangleStripTCoords.SetNumberOfComponents(2)
triangleStripTCoords.SetNumberOfTuples(5)
triangleStripTCoords.InsertTuple2(0, 0, 1)
triangleStripTCoords.InsertTuple2(1, 0, 0)
triangleStripTCoords.InsertTuple2(2, .5, 1)
triangleStripTCoords.InsertTuple2(3, .5, 0)
triangleStripTCoords.InsertTuple2(4, 1, 1)
triangleStripPointScalars = vtk.vtkFloatArray()
triangleStripPointScalars.SetNumberOfTuples(5)
triangleStripPointScalars.InsertValue(0, 1)
triangleStripPointScalars.InsertValue(1, 0)
triangleStripPointScalars.InsertValue(2, 0)
triangleStripPointScalars.InsertValue(3, 0)
triangleStripPointScalars.InsertValue(4, 0)
triangleStripCellScalars = vtk.vtkFloatArray()
triangleStripCellScalars.SetNumberOfTuples(1)
triangleStripCellScalars.InsertValue(0, 1)
triangleStripPointNormals = vtk.vtkFloatArray()
triangleStripPointNormals.SetNumberOfComponents(3)
triangleStripPointNormals.SetNumberOfTuples(5)
triangleStripPointNormals.InsertTuple3(0, 0, 0, 1)
triangleStripPointNormals.InsertTuple3(1, 0, 1, 0)
triangleStripPointNormals.InsertTuple3(2, 0, 1, 1)
triangleStripPointNormals.InsertTuple3(3, 1, 0, 0)
triangleStripPointNormals.InsertTuple3(4, 1, 0, 1)
triangleStripCellNormals = vtk.vtkFloatArray()
triangleStripCellNormals.SetNumberOfComponents(3)
triangleStripCellNormals.SetNumberOfTuples(1)
triangleStripCellNormals.InsertTuple3(0, 1, 1, 1)
aTriangleStrip = vtk.vtkTriangleStrip()
aTriangleStrip.GetPointIds().SetNumberOfIds(5)
aTriangleStrip.GetPointIds().SetId(0, 0)
aTriangleStrip.GetPointIds().SetId(1, 1)
aTriangleStrip.GetPointIds().SetId(2, 2)
aTriangleStrip.GetPointIds().SetId(3, 3)
aTriangleStrip.GetPointIds().SetId(4, 4)
lut = vtk.vtkLookupTable()
lut.SetNumberOfColors(5)
lut.SetTableValue(0, 0, 0, 1, 1)
lut.SetTableValue(1, 0, 1, 0, 1)
lut.SetTableValue(2, 0, 1, 1, 1)
lut.SetTableValue(3, 1, 0, 0, 1)
lut.SetTableValue(4, 1, 0, 1, 1)
masks = [0, 1, 2, 3, 4, 5, 6, 7, 10, 11, 14, 15, 16, 18, 20, 22, 26, 30]
types = ["strip", "triangle"]
i = 0
j = 0
k = 0
for type in types:
for mask in masks:
idx = str(i)
exec("grid" + idx + " = vtk.vtkUnstructuredGrid()")
eval("grid" + idx).Allocate(1, 1)
eval("grid" + idx).InsertNextCell(
aTriangleStrip.GetCellType(), aTriangleStrip.GetPointIds())
eval("grid" + idx).SetPoints(triangleStripPoints)
exec("geometry" + idx + " = vtk.vtkGeometryFilter()")
eval("geometry" + idx).SetInputData(eval("grid" + idx))
exec("triangles" + idx + " = vtk.vtkTriangleFilter()")
eval("triangles" + idx).SetInputConnection(
eval("geometry" + idx).GetOutputPort())
exec("mapper" + idx + " = vtk.vtkPolyDataMapper()")
if (type == "strip"):
eval("mapper" + idx).SetInputConnection(
eval("geometry" + idx).GetOutputPort())
if (type == "triangle"):
eval("mapper" + idx).SetInputConnection(
eval("triangles" + idx).GetOutputPort())
eval("mapper" + idx).SetLookupTable(lut)
eval("mapper" + idx).SetScalarRange(0, 4)
exec("actor" + idx + " = vtk.vtkActor()")
eval("actor" + idx).SetMapper(eval("mapper" + idx))
if mask & 1 != 0:
eval("grid" + idx).GetPointData().SetNormals(
triangleStripPointNormals)
if mask & 2 != 0:
eval("grid" + idx).GetPointData().SetScalars(
triangleStripPointScalars)
eval("mapper" + idx).SetScalarModeToUsePointData()
if mask & 4 != 0:
eval("grid" + idx).GetPointData().SetTCoords(
triangleStripTCoords)
eval("actor" + idx).SetTexture(texture)
if mask & 8 != 0:
eval("grid" + idx).GetCellData().SetScalars(
triangleStripCellScalars)
eval("mapper" + idx).SetScalarModeToUseCellData()
if mask & 16 != 0:
eval("grid" + idx).GetCellData().SetNormals(
triangleStripCellNormals)
eval("actor" + idx).AddPosition(j * 2, k * 2, 0)
ren1.AddActor(eval("actor" + idx))
eval("actor" + idx).GetProperty().SetRepresentationToWireframe()
j += 1
if (j >= 6):
j = 0
k += 1
i += 1
renWin.SetSize(480, 480)
ren1.SetBackground(.7, .3, .1)
ren1.ResetCameraClippingRange()
renWin.Render()
# render the image
#
iren.Initialize()
threshold = 15
#iren.Start()
|