def addImage(self, name):
if name in self.images:
return
image = vtk.vtkImageData()
tex = vtk.vtkTexture()
tex.SetInput(image)
tex.EdgeClampOn()
tex.RepeatOff()
self.imageUtimes[name] = 0
self.images[name] = image
self.textures[name] = tex
def addImage(self, name):
if name in self.images:
return
image = vtk.vtkImageData()
tex = vtk.vtkTexture()
tex.SetInput(image)
tex.EdgeClampOn()
tex.RepeatOff()
self.imageUtimes[name] = 0
self.images[name] = image
self.textures[name] = tex
def getDepthMapData(self, viewId):
mapId = self.source.GetCurrentMapId(viewId)
if mapId < 0:
return None, None
depthImage = vtk.vtkImageData()
transform = vtk.vtkTransform()
# print "getting depth image for viewId {:d} mapId {:d}".format(viewId, mapId)
self.source.GetDataForMapId(viewId, mapId, depthImage, transform)
dims = depthImage.GetDimensions()
d = vnp.getNumpyFromVtk(depthImage, "ImageScalars")
d = d.reshape(dims[1], dims[0])
t = np.array([[transform.GetMatrix().GetElement(r, c) for c in xrange(4)] for r in xrange(4)])
return d, t
def getDepthMapData(self, viewId):
mapId = self.source.GetCurrentMapId(viewId)
if mapId < 0:
return None, None
depthImage = vtk.vtkImageData()
transform = vtk.vtkTransform()
# print "getting depth image for viewId {:d} mapId {:d}".format(viewId, mapId)
self.source.GetDataForMapId(viewId, mapId, depthImage, transform)
dims = depthImage.GetDimensions()
d = vnp.getNumpyFromVtk(depthImage, 'ImageScalars')
d = d.reshape(dims[1], dims[0])
t = np.array(
[[transform.GetMatrix().GetElement(r, c) for c in xrange(4)]
for r in xrange(4)])
return d, t
