def GetArray(self): input = self.__export.GetInput() input.UpdateInformation() type = input.GetScalarType() extent = input.GetWholeExtent() numComponents = input.GetNumberOfScalarComponents() dim = (extent[5] - extent[4] + 1, extent[3] - extent[2] + 1, extent[1] - extent[0] + 1) if (numComponents > 1): dim = dim + (numComponents, ) size = dim[0] * dim[1] * dim[2] * numComponents * self.__sizeDict[type] if _NEW_NUMERIC: imArray = Numeric.zeros((size, ), Numeric.UnsignedInt8) self.__export.Export(imArray) else: imString = Numeric.zeros((size, ), Numeric.UnsignedInt8).tostring() self.__export.Export(imString) imArray = Numeric.fromstring(imString, self.__typeDict[type]) # just to remind myself of the dangers of memory management del imString # reshape array appropriately. imArray = Numeric.reshape(imArray, dim) # convert unsigned short to int to avoid sign issues if (type == VTK_UNSIGNED_SHORT and self.__ConvertUnsignedShortToInt): imArray = umath.bitwise_and(imArray.astype(Numeric.Int32), 0xffff) return imArray
def GetArray(self): input = self.__export.GetInput() input.UpdateInformation() type = input.GetScalarType() extent = input.GetWholeExtent() numComponents = input.GetNumberOfScalarComponents() dim = (extent[5]-extent[4]+1, extent[3]-extent[2]+1, extent[1]-extent[0]+1) if (numComponents > 1): dim = dim + (numComponents,) size = dim[0]*dim[1]*dim[2]*numComponents*self.__sizeDict[type] if _NEW_NUMERIC: imArray = Numeric.zeros((size,),Numeric.UnsignedInt8) self.__export.Export(imArray) else: imString = Numeric.zeros((size,), Numeric.UnsignedInt8).tostring() self.__export.Export(imString) imArray = Numeric.fromstring(imString,self.__typeDict[type]) # just to remind myself of the dangers of memory management del imString # reshape array appropriately. imArray = Numeric.reshape(imArray, dim) # convert unsigned short to int to avoid sign issues if (type == VTK_UNSIGNED_SHORT and self.__ConvertUnsignedShortToInt): imArray = umath.bitwise_and(imArray.astype(Numeric.Int32),0xffff) return imArray
def GetArray(self): self.__export.Update() input = self.__export.GetInput() extent = input.GetExtent() type = input.GetScalarType() numComponents = input.GetNumberOfScalarComponents() dim = (extent[5] - extent[4] + 1, extent[3] - extent[2] + 1, extent[1] - extent[0] + 1) if (numComponents > 1): dim = dim + (numComponents, ) imArray = numpy.zeros(dim, self.__typeDict[type]) self.__export.Export(imArray) # convert unsigned short to int to avoid sign issues if (type == VTK_UNSIGNED_SHORT and self.__ConvertUnsignedShortToInt): imArray = umath.bitwise_and(imArray.astype('i'), 0xffff) return imArray
def GetArray(self): input = self.__export.GetInput() input.UpdateInformation() type = input.GetScalarType() extent = input.GetWholeExtent() numComponents = input.GetNumberOfScalarComponents() dim = (extent[5]-extent[4]+1, extent[3]-extent[2]+1, extent[1]-extent[0]+1) if (numComponents > 1): dim = dim + (numComponents,) imArray = numpy.zeros(dim, self.__typeDict[type]) self.__export.Export(imArray) # convert unsigned short to int to avoid sign issues if (type == VTK_UNSIGNED_SHORT and self.__ConvertUnsignedShortToInt): imArray = umath.bitwise_and(imArray.astype('i'),0xffff) return imArray