def get_wrap(o):
# aqui para no importarlo siempre
import vtk
src = vtk.vtkPassThrough()
if vtk.vtkVersion.GetVTKMajorVersion() < 6:
src.SetInput(o)
else:
src.SetInputData(o)
return src
def get_void():
""" return an empty object suitable for representing """
# aqui para no importarlo siempre
import vtk
u = vtk.vtkUnstructuredGrid()
src = vtk.vtkPassThrough()
if vtk.vtkVersion.GetVTKMajorVersion() < 6:
src.SetInput(u)
else:
src.SetInputData(u)
# print 'void bounds', src.GetOutput().GetBounds() # 1 -1 1 -1 1 -1
return src
def __init__(self, imageData, intensity, color=None):
self._uuid = uuid.uuid1()
if intensity!=None:
self.intensity = intensity
if color==None:
color=self.color_
self.set_color(color)
self.connect = ConnectFilter()
self.deci = DecimateFilter()
self.marchingCubes = vtk.vtkMarchingCubes()
self.marchingCubes.SetInput(imageData)
self.output = vtk.vtkPassThrough()
self.prog = ProgressBarDialog(
title='Rendering surface %s' % self.label,
parent=None,
msg='Marching cubes ....',
size=(300,40),
)
self.prog.set_modal(True)
def start(o, event):
self.prog.show()
while gtk.events_pending(): gtk.main_iteration()
def progress(o, event):
val = o.GetProgress()
self.prog.bar.set_fraction(val)
while gtk.events_pending(): gtk.main_iteration()
def end(o, event):
self.prog.hide()
while gtk.events_pending(): gtk.main_iteration()
self.marchingCubes.AddObserver('StartEvent', start)
self.marchingCubes.AddObserver('ProgressEvent', progress)
self.marchingCubes.AddObserver('EndEvent', end)
self.update_pipeline()
self.notify_add_surface(self.output)
# Setup the query pipeline ...
query_document_array = vtk.vtkIdTypeArray()
query_document_array.SetName("document")
query_text_array = vtk.vtkUnicodeStringArray()
query_text_array.SetName("text")
query_document_array.InsertNextValue(query_document_array.GetNumberOfTuples())
query_text_array.InsertNextValue(unicode(query_string))
query_document_table = vtk.vtkTable()
query_document_table.AddColumn(query_document_array)
query_document_table.AddColumn(query_text_array)
query_document_reader = vtk.vtkPassThrough()
query_document_reader.SetInputConnection(0, query_document_table.GetProducerPort())
query_tokens = vtkTokenization()
query_tokens.SetInputConnection(0, query_document_reader.GetOutputPort())
query_projection = vtkTrilinosLSAProjection()
query_projection.SetInputConnection(0, query_document_reader.GetOutputPort())
query_projection.SetInputConnection(1, query_tokens.GetOutputPort())
query_projection.SetInputConnection(2, fd_reader.GetOutputPort())
query_projection.SetInputConnection(3, fw_reader.GetOutputPort())
query_projection.SetInputConnection(4, lsv_reader.GetOutputPort())
similarity_table = vtkCosineSimilarityTable()
similarity_table.SetInputConnection(0, query_projection.GetOutputPort())
similarity_table.SetInputConnection(1, rsv_reader.GetOutputPort())
