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