def initialize(self, VTKWebApp, args):
dataid = args.id
treedata = getDBdata(dataid)
VTKWebApp.tree1 = treedata["tree1"]
VTKWebApp.tree2 = treedata["tree2"]
VTKWebApp.table = dataid+ ".csv"
# Create default pipeline (Only once for all the session)
if not VTKWebApp.view:
# read the trees
treeReader1 = vtk.vtkNewickTreeReader()
treeReader1.SetReadFromInputString(1)
treeReader1.SetInputString(VTKWebApp.tree1)
treeReader1.Update()
tree1 = treeReader1.GetOutput()
treeReader2 = vtk.vtkNewickTreeReader()
treeReader2.SetReadFromInputString(1)
treeReader2.SetInputString(VTKWebApp.tree2)
treeReader2.Update()
tree2 = treeReader2.GetOutput()
# read the table
tableReader = vtk.vtkDelimitedTextReader()
tableReader.SetFileName(VTKWebApp.table)
tableReader.SetHaveHeaders(True)
tableReader.DetectNumericColumnsOn()
tableReader.ForceDoubleOn()
tableReader.Update()
table = tableReader.GetOutput()
# setup the tanglegram
tanglegram = vtk.vtkTanglegramItem()
tanglegram.SetTree1(tree1);
tanglegram.SetTree2(tree2);
tanglegram.SetTable(table);
tanglegram.SetTree1Label("tree1");
tanglegram.SetTree2Label("tree2");
# setup the window
view = vtk.vtkContextView()
view.GetRenderer().SetBackground(1,1,1)
view.GetRenderWindow().SetSize(800,600)
iren = view.GetInteractor()
iren.SetRenderWindow(view.GetRenderWindow())
transformItem = vtk.vtkContextTransform()
transformItem.AddItem(tanglegram)
transformItem.SetInteractive(1)
view.GetScene().AddItem(transformItem)
view.GetRenderWindow().SetMultiSamples(0)
iren.Initialize()
view.GetRenderWindow().Render()
# VTK Web application specific
VTKWebApp.view = view.GetRenderWindow()
self.Application.GetObjectIdMap().SetActiveObject("VIEW", view.GetRenderWindow())
def initialize(self):
global renderer, renderWindow, renderWindowInteractor, cone, mapper, actor
# Bring used components
self.registerVtkWebProtocol(protocols.vtkWebMouseHandler())
self.registerVtkWebProtocol(protocols.vtkWebViewPort())
self.registerVtkWebProtocol(protocols.vtkWebViewPortImageDelivery())
self.registerVtkWebProtocol(protocols.vtkWebViewPortGeometryDelivery())
# Update authentication key to use
self.updateSecret(_PhylogeneticTree.authKey)
# Create default pipeline (Only once for all the session)
if not _PhylogeneticTree.view:
# read in a tree
treeReader = vtk.vtkNewickTreeReader()
treeReader.SetFileName(_PhylogeneticTree.treeFilePath)
treeReader.Update()
reader = treeReader.GetOutput()
# read in a table
tableReader = vtk.vtkDelimitedTextReader()
tableReader.SetFileName(_PhylogeneticTree.csvFilePath)
tableReader.SetHaveHeaders(True)
tableReader.DetectNumericColumnsOn()
tableReader.Update()
table = tableReader.GetOutput()
# play with the heatmap vis
treeHeatmapItem = vtk.vtkTreeHeatmapItem()
treeHeatmapItem.SetTree(reader)
treeHeatmapItem.SetTable(table)
# setup the window
view = vtk.vtkContextView()
view.GetRenderer().SetBackground(1, 1, 1)
view.GetRenderWindow().SetSize(800, 600)
iren = view.GetInteractor()
iren.SetRenderWindow(view.GetRenderWindow())
transformItem = vtk.vtkContextTransform()
transformItem.AddItem(treeHeatmapItem)
transformItem.SetInteractive(1)
view.GetScene().AddItem(transformItem)
view.GetRenderWindow().SetMultiSamples(0)
iren.Initialize()
view.GetRenderWindow().Render()
# VTK Web application specific
_PhylogeneticTree.view = view.GetRenderWindow()
self.Application.GetObjectIdMap().SetActiveObject(
"VIEW", view.GetRenderWindow())
def initialize(self):
global renderer, renderWindow, renderWindowInteractor, cone, mapper, actor
# Bring used components
self.registerVtkWebProtocol(protocols.vtkWebMouseHandler())
self.registerVtkWebProtocol(protocols.vtkWebViewPort())
self.registerVtkWebProtocol(protocols.vtkWebViewPortImageDelivery())
self.registerVtkWebProtocol(protocols.vtkWebViewPortGeometryDelivery())
# Update authentication key to use
self.updateSecret(_PhylogeneticTree.authKey)
# Create default pipeline (Only once for all the session)
if not _PhylogeneticTree.view:
# read in a tree
treeReader = vtk.vtkNewickTreeReader()
treeReader.SetFileName(_PhylogeneticTree.treeFilePath)
treeReader.Update()
reader = treeReader.GetOutput()
# read in a table
tableReader = vtk.vtkDelimitedTextReader()
tableReader.SetFileName(_PhylogeneticTree.csvFilePath)
tableReader.SetHaveHeaders(True)
tableReader.DetectNumericColumnsOn()
tableReader.Update()
table = tableReader.GetOutput()
# play with the heatmap vis
treeHeatmapItem = vtk.vtkTreeHeatmapItem()
treeHeatmapItem.SetTree(reader);
treeHeatmapItem.SetTable(table);
# setup the window
view = vtk.vtkContextView()
view.GetRenderer().SetBackground(1,1,1)
view.GetRenderWindow().SetSize(800,600)
iren = view.GetInteractor()
iren.SetRenderWindow(view.GetRenderWindow())
transformItem = vtk.vtkContextTransform()
transformItem.AddItem(treeHeatmapItem)
transformItem.SetInteractive(1)
view.GetScene().AddItem(transformItem)
view.GetRenderWindow().SetMultiSamples(0)
iren.Initialize()
view.GetRenderWindow().Render()
# VTK Web application specific
_PhylogeneticTree.view = view.GetRenderWindow()
self.Application.GetObjectIdMap().SetActiveObject("VIEW", view.GetRenderWindow())
def initialize(self, VTKWebApp, args):
VTKWebApp.treeFilePath = args.tree
VTKWebApp.csvFilePath = args.table
# Create default pipeline (Only once for all the session)
if not VTKWebApp.view:
# read in a tree
treeReader = vtk.vtkNewickTreeReader()
treeReader.SetFileName(VTKWebApp.treeFilePath)
treeReader.Update()
reader = treeReader.GetOutput()
# read in a table
tableReader = vtk.vtkDelimitedTextReader()
tableReader.SetFileName(VTKWebApp.csvFilePath)
tableReader.SetHaveHeaders(1)
tableReader.DetectNumericColumnsOn()
tableReader.Update()
table = tableReader.GetOutput()
# play with the heatmap vis
treeHeatmapItem = vtk.vtkTreeHeatmapItem()
treeHeatmapItem.SetTree(reader)
treeHeatmapItem.SetTable(table)
# setup the window
view = vtk.vtkContextView()
view.GetRenderer().SetBackground(1, 1, 1)
view.GetRenderWindow().SetSize(800, 600)
iren = view.GetInteractor()
iren.SetRenderWindow(view.GetRenderWindow())
transformItem = vtk.vtkContextTransform()
transformItem.AddItem(treeHeatmapItem)
transformItem.SetInteractive(1)
view.GetScene().AddItem(transformItem)
view.GetRenderWindow().SetMultiSamples(0)
iren.Initialize()
view.GetRenderWindow().Render()
# VTK Web application specific
VTKWebApp.view = view.GetRenderWindow()
self.Application.GetObjectIdMap().SetActiveObject("VIEW", view.GetRenderWindow())
def initialize(self, VTKWebApp, args):
# Create default pipeline (Only once for all the session)
if not VTKWebApp.view:
# create our visualization item
treeHeatmapItem = vtk.vtkTreeHeatmapItem()
# get our input data, read it into VTK format,
# and load it into our visualization item.
if args.treeName:
r = requests.get(
"%s/arborapi/projmgr/project/%s/PhyloTree/%s/phyloxml" % (args.baseURL, args.projectName, args.treeName))
phyloxmlTree = r.text
tree = vtk_arbor_utils.PhyloXMLToVTKTree(phyloxmlTree)
treeHeatmapItem.SetTree(tree)
if tree.GetVertexData().GetArray("property.differences"):
treeHeatmapItem.GetDendrogram().SetColorArray("property.differences")
treeHeatmapItem.GetDendrogram().SetLineWidth(2.0)
if args.tableName:
r = requests.get(
"%s/arborapi/projmgr/project/%s/CharacterMatrix/%s/csv" % (args.baseURL, args.projectName, args.tableName))
csvTable = r.text
table = vtk_arbor_utils.CSVToVTKTable(csvTable)
treeHeatmapItem.SetTable(table)
# setup the window
view = vtk.vtkContextView()
view.GetRenderWindow().SetSize(int(args.width), int(args.height))
view.GetRenderer().SetBackground(1,1,1)
iren = view.GetInteractor()
iren.SetRenderWindow(view.GetRenderWindow())
transformItem = vtk.vtkContextTransform()
transformItem.AddItem(treeHeatmapItem)
transformItem.SetInteractive(1)
view.GetScene().AddItem(transformItem)
view.GetRenderWindow().SetMultiSamples(0)
iren.Initialize()
view.GetRenderWindow().Render()
# adjust zoom so the item nicely fills the screen
itemSize = [0, 0]
treeHeatmapItem.GetSize(itemSize)
itemSize.append(0)
transformItem.GetTransform().MultiplyPoint(itemSize, itemSize)
newWidth = view.GetScene().GetSceneWidth()
newHeight = view.GetScene().GetSceneHeight()
pageWidth = newWidth
pageHeight = newHeight
sx = pageWidth / itemSize[0]
sy = pageHeight / itemSize[1]
if sy < sx:
scale = sy;
else:
scale = sx;
if scale > 1:
scale = scale * 0.5
else:
scale = scale * 0.9
transformItem.Scale(scale, scale)
# center the item within the screen
itemCenter = [0, 0]
treeHeatmapItem.GetCenter(itemCenter)
itemCenter.append(0)
centerPt = vtk.vtkPoints2D()
centerPt.InsertNextPoint(newWidth / 2.0, newHeight / 2.0)
transformItem.GetTransform().InverseTransformPoints(centerPt, centerPt)
sceneCenter = [0, 0]
centerPt.GetPoint(0, sceneCenter)
dx = -1 * (itemCenter[0] - sceneCenter[0])
dy = -1 * (itemCenter[1] - sceneCenter[1])
transformItem.Translate(dx, dy)
# VTK Web application specific
VTKWebApp.view = view.GetRenderWindow()
self.Application.GetObjectIdMap().SetActiveObject("VIEW", view.GetRenderWindow())
def initialize(self, VTKWebApp, args):
# Create default pipeline (Only once for all the session)
if not VTKWebApp.view:
baseURL = args.baseURL
# support for overriding the base URL
scriptDir = os.path.dirname(os.path.realpath(__file__))
configPath = scriptDir + "/baseURL.txt"
if os.path.isfile(configPath):
f = file(configPath, "r")
baseURL = f.read().rstrip()
f.close()
# get our input data from romanesco
r = requests.get(baseURL + args.tree1URI, verify=False)
tree1JSON = r.json()
tree1Str = tree1JSON["data"]
r = requests.get(baseURL + args.tree2URI, verify=False)
tree2JSON = r.json()
tree2Str = tree2JSON["data"]
r = requests.get(baseURL + args.tableURI, verify=False)
tableJSON = r.json()
tableStr = tableJSON["data"]
# get the tree names (TODO: consider better ways to get this info)
tree1LabelURI = args.tree1URI[0:args.tree1URI.find("romanesco")]
r = requests.get(baseURL + tree1LabelURI, verify=False)
tree1LabelJSON = r.json()
tree1Label = tree1LabelJSON["name"]
tree1Label = tree1Label[0:tree1Label.find(".")]
tree2LabelURI = args.tree2URI[0:args.tree2URI.find("romanesco")]
r = requests.get(baseURL + tree2LabelURI, verify=False)
tree2LabelJSON = r.json()
tree2Label = tree2LabelJSON["name"]
tree2Label = tree2Label[0:tree2Label.find(".")]
# convert our input data into deserialized VTK objects
tree1Reader = vtk.vtkTreeReader()
tree1Reader.ReadFromInputStringOn()
tree1Reader.SetInputString(tree1Str, len(tree1Str))
tree1Reader.Update()
tree1 = tree1Reader.GetOutput()
tree2Reader = vtk.vtkTreeReader()
tree2Reader.ReadFromInputStringOn()
tree2Reader.SetInputString(tree2Str, len(tree2Str))
tree2Reader.Update()
tree2 = tree2Reader.GetOutput()
tableReader = vtk.vtkDelimitedTextReader()
tableReader.ReadFromInputStringOn()
tableReader.SetInputString(tableStr, len(tableStr))
tableReader.SetHaveHeaders(True)
tableReader.DetectNumericColumnsOn()
tableReader.ForceDoubleOn()
table = tableReader.GetOutput()
tableReader.Update()
# create our visualization item and load the data into it.
tanglegram = vtk.vtkTanglegramItem()
tanglegram.SetTree1(tree1)
tanglegram.SetTree2(tree2)
tanglegram.SetTable(table)
tanglegram.SetTree1Label(tree1Label)
tanglegram.SetTree2Label(tree2Label)
tanglegram.SetCorrespondenceLineWidth(4.0)
tanglegram.SetTreeLineWidth(2.0)
# setup the window
view = vtk.vtkContextView()
view.GetRenderWindow().SetSize(int(args.width), int(args.height))
view.GetRenderer().SetBackground(1, 1, 1)
iren = view.GetInteractor()
iren.SetRenderWindow(view.GetRenderWindow())
transformItem = vtk.vtkContextTransform()
transformItem.AddItem(tanglegram)
transformItem.SetInteractive(1)
view.GetScene().AddItem(transformItem)
view.GetRenderWindow().SetMultiSamples(0)
iren.Initialize()
view.GetRenderWindow().Render()
# VTK Web application specific
VTKWebApp.view = view.GetRenderWindow()
self.Application.GetObjectIdMap().SetActiveObject(
"VIEW", view.GetRenderWindow())
def main():
colors = vtk.vtkNamedColors()
filename = get_program_parameters()
# Create the reader for the data.
print('Loading ', filename)
reader = vtk.vtkUnstructuredGridReader()
reader.SetFileName(filename)
reader.Update()
extractEdges = vtk.vtkExtractEdges()
extractEdges.SetInputConnection(reader.GetOutputPort())
legendValues = vtk.vtkVariantArray()
it = reader.GetOutput().NewCellIterator()
it.InitTraversal()
while not it.IsDoneWithTraversal():
cell = vtk.vtkGenericCell()
it.GetCell(cell)
cellName = vtk.vtkCellTypes.GetClassNameFromTypeId(cell.GetCellType())
print(cellName, 'NumberOfPoints:', cell.GetNumberOfPoints(),
'CellDimension:', cell.GetCellDimension())
legendValues.InsertNextValue(cellName)
it.GoToNextCell()
# Tube the edges
tubes = vtk.vtkTubeFilter()
tubes.SetInputConnection(extractEdges.GetOutputPort())
tubes.SetRadius(.05)
tubes.SetNumberOfSides(21)
edgeMapper = vtk.vtkPolyDataMapper()
edgeMapper.SetInputConnection(tubes.GetOutputPort())
edgeMapper.SetScalarRange(0, 26)
edgeActor = vtk.vtkActor()
edgeActor.SetMapper(edgeMapper)
edgeActor.GetProperty().SetSpecular(0.6)
edgeActor.GetProperty().SetSpecularPower(30)
# Glyph the points
sphere = vtk.vtkSphereSource()
sphere.SetPhiResolution(21)
sphere.SetThetaResolution(21)
sphere.SetRadius(0.08)
pointMapper = vtk.vtkGlyph3DMapper()
pointMapper.SetInputConnection(reader.GetOutputPort())
pointMapper.SetSourceConnection(sphere.GetOutputPort())
pointMapper.ScalingOff()
pointMapper.ScalarVisibilityOff()
pointActor = vtk.vtkActor()
pointActor.SetMapper(pointMapper)
pointActor.GetProperty().SetDiffuseColor(colors.GetColor3d('Banana'))
pointActor.GetProperty().SetSpecular(0.6)
pointActor.GetProperty().SetSpecularColor(1.0, 1.0, 1.0)
pointActor.GetProperty().SetSpecularPower(100)
# Label the points
labelMapper = vtk.vtkLabeledDataMapper()
labelMapper.SetInputConnection(reader.GetOutputPort())
labelActor = vtk.vtkActor2D()
labelActor.SetMapper(labelMapper)
# The geometry
geometryShrink = vtk.vtkShrinkFilter()
geometryShrink.SetInputConnection(reader.GetOutputPort())
geometryShrink.SetShrinkFactor(0.8)
# NOTE: We must copy the originalLut because the CategoricalLegend
# needs an indexed lookup table, but the geometryMapper uses a
# non-index lookup table
categoricalLut = vtk.vtkLookupTable()
originalLut = reader.GetOutput().GetCellData().GetScalars().GetLookupTable(
)
categoricalLut.DeepCopy(originalLut)
categoricalLut.IndexedLookupOn()
geometryMapper = vtk.vtkDataSetMapper()
geometryMapper.SetInputConnection(geometryShrink.GetOutputPort())
geometryMapper.SetScalarModeToUseCellData()
geometryMapper.SetScalarRange(0, 11)
geometryActor = vtk.vtkActor()
geometryActor.SetMapper(geometryMapper)
geometryActor.GetProperty().SetLineWidth(3)
geometryActor.GetProperty().EdgeVisibilityOn()
geometryActor.GetProperty().SetEdgeColor(0, 0, 0)
# Legend
for v in range(0, legendValues.GetNumberOfTuples()):
categoricalLut.SetAnnotation(legendValues.GetValue(v),
legendValues.GetValue(v).ToString())
legend = vtk.vtkCategoryLegend()
legend.SetScalarsToColors(categoricalLut)
legend.SetValues(legendValues)
legend.SetTitle('Cell Type')
legend.GetBrush().SetColor(colors.GetColor4ub('Silver'))
placeLegend = vtk.vtkContextTransform()
placeLegend.AddItem(legend)
placeLegend.Translate(640 - 20, 480 - 12 * 16)
contextView = vtk.vtkContextView()
contextView.GetScene().AddItem(placeLegend)
renderer = contextView.GetRenderer()
renderWindow = contextView.GetRenderWindow()
renderWindowInteractor = vtk.vtkRenderWindowInteractor()
renderWindowInteractor.SetRenderWindow(renderWindow)
renderer.AddActor(geometryActor)
renderer.AddActor(labelActor)
renderer.AddActor(edgeActor)
renderer.AddActor(pointActor)
renderer.SetBackground(colors.GetColor3d('SlateGray'))
aCamera = vtk.vtkCamera()
aCamera.Azimuth(-40.0)
aCamera.Elevation(50.0)
renderer.SetActiveCamera(aCamera)
renderer.ResetCamera()
renderWindow.SetSize(640, 480)
renderWindow.SetWindowName('ReadLegacyUnstructuredGrid')
renderWindow.Render()
renderWindowInteractor.Start()
def initialize(self, VTKWebApp, args):
# Create default pipeline (Only once for all the session)
if not VTKWebApp.view:
baseURL = args.baseURL
# support for overriding the base URL
scriptDir = os.path.dirname(os.path.realpath(__file__))
configPath = scriptDir + "/baseURL.txt"
if os.path.isfile(configPath):
f = file(configPath, "r")
baseURL = f.read().rstrip()
f.close()
# get our input data from romanesco
r = requests.get(baseURL + args.tableURI, verify=False)
tableJSON = r.json()
tableStr = tableJSON["data"]
r = requests.get(baseURL + args.treeURI, verify=False)
treeJSON = r.json()
treeStr = treeJSON["data"]
# deserialize our input data
tableReader = vtk.vtkTableReader()
tableReader.ReadFromInputStringOn()
tableReader.SetInputString(tableStr, len(tableStr))
tableReader.Update()
table = tableReader.GetOutput()
treeReader = vtk.vtkTreeReader()
treeReader.ReadFromInputStringOn()
treeReader.SetInputString(treeStr, len(treeStr))
treeReader.Update()
tree = treeReader.GetOutput()
# create our visualization item and load the data into it.
treeHeatmapItem = vtk.vtkTreeHeatmapItem()
treeHeatmapItem.SetTree(tree)
treeHeatmapItem.SetTable(table)
# detect if we are visualizing the results of a tree comparison
if tree.GetVertexData().GetArray("property.differences"):
treeHeatmapItem.GetDendrogram().SetColorArray(
"property.differences")
treeHeatmapItem.GetDendrogram().SetLineWidth(2.0)
# setup the window
view = vtk.vtkContextView()
view.GetRenderWindow().SetSize(int(args.width), int(args.height))
view.GetRenderer().SetBackground(1, 1, 1)
iren = view.GetInteractor()
iren.SetRenderWindow(view.GetRenderWindow())
transformItem = vtk.vtkContextTransform()
transformItem.AddItem(treeHeatmapItem)
transformItem.SetInteractive(1)
view.GetScene().AddItem(transformItem)
view.GetRenderWindow().SetMultiSamples(0)
iren.Initialize()
view.GetRenderWindow().Render()
# adjust zoom so the item nicely fills the screen
itemSize = [0, 0]
treeHeatmapItem.GetSize(itemSize)
itemSize.append(0)
transformItem.GetTransform().MultiplyPoint(itemSize, itemSize)
newWidth = view.GetScene().GetSceneWidth()
newHeight = view.GetScene().GetSceneHeight()
pageWidth = newWidth
pageHeight = newHeight
sx = pageWidth / itemSize[0]
sy = pageHeight / itemSize[1]
if sy < sx:
scale = sy
else:
scale = sx
if scale > 1:
scale = scale * 0.5
else:
scale = scale * 0.9
transformItem.Scale(scale, scale)
# center the item within the screen
itemCenter = [0, 0]
treeHeatmapItem.GetCenter(itemCenter)
itemCenter.append(0)
centerPt = vtk.vtkPoints2D()
centerPt.InsertNextPoint(newWidth / 2.0, newHeight / 2.0)
transformItem.GetTransform().InverseTransformPoints(centerPt, centerPt)
sceneCenter = [0, 0]
centerPt.GetPoint(0, sceneCenter)
dx = -1 * (itemCenter[0] - sceneCenter[0])
dy = -1 * (itemCenter[1] - sceneCenter[1])
transformItem.Translate(dx, dy)
# VTK Web application specific
VTKWebApp.view = view.GetRenderWindow()
self.Application.GetObjectIdMap().SetActiveObject(
"VIEW", view.GetRenderWindow())
def initialize(self, VTKWebApp, args):
dataid = args.id
treedata = getDBdata(dataid)
VTKWebApp.tree = treedata["tree"]
VTKWebApp.table = dataid+ ".csv"
# Create default pipeline (Only once for all the session)
if not VTKWebApp.view:
treeHeatmapItem = vtk.vtkTreeHeatmapItem()
# read in a tree
treeReader = vtk.vtkNewickTreeReader()
treeReader.SetReadFromInputString(1)
treeReader.SetInputString(VTKWebApp.tree)
treeReader.Update()
tree = treeReader.GetOutput()
treeHeatmapItem.SetTree(tree)
if (VTKWebApp.table != "none" and os.path.isfile(VTKWebApp.table)):
# read in a table
print("read table"+ VTKWebApp.table)
tableReader = vtk.vtkDelimitedTextReader()
tableReader.SetFileName(VTKWebApp.table)
tableReader.SetHaveHeaders(1)
tableReader.DetectNumericColumnsOn()
tableReader.Update()
table = tableReader.GetOutput()
if (table):
treeHeatmapItem.SetTable(table)
# setup the window
view = vtk.vtkContextView()
view.GetRenderer().SetBackground(1,1,1)
view.GetRenderWindow().SetSize(800,600)
iren = view.GetInteractor()
iren.SetRenderWindow(view.GetRenderWindow())
transformItem = vtk.vtkContextTransform()
transformItem.AddItem(treeHeatmapItem)
transformItem.SetInteractive(1)
view.GetScene().AddItem(transformItem)
view.GetRenderWindow().SetMultiSamples(0)
iren.Initialize()
view.GetRenderWindow().Render()
# adjust zoom so the item nicely fills the screen
itemSize = [0, 0]
treeHeatmapItem.GetSize(itemSize)
itemSize.append(0)
transformItem.GetTransform().MultiplyPoint(itemSize, itemSize)
newWidth = view.GetScene().GetSceneWidth()
newHeight = view.GetScene().GetSceneHeight()
pageWidth = newWidth
pageHeight = newHeight
sx = pageWidth / itemSize[0]
sy = pageHeight / itemSize[1]
if sy < sx:
scale = sy;
else:
scale = sx;
if scale > 1:
scale = scale * 0.5
else:
scale = scale * 0.9
transformItem.Scale(scale, scale)
# center the item within the screen
itemCenter = [0, 0]
treeHeatmapItem.GetCenter(itemCenter)
itemCenter.append(0)
centerPt = vtk.vtkPoints2D()
centerPt.InsertNextPoint(newWidth / 2.0, newHeight / 2.0)
transformItem.GetTransform().InverseTransformPoints(centerPt, centerPt)
sceneCenter = [0, 0]
centerPt.GetPoint(0, sceneCenter)
dx = -1 * (itemCenter[0] - sceneCenter[0])
dy = -1 * (itemCenter[1] - sceneCenter[1])
transformItem.Translate(dx, dy)
# VTK Web application specific
VTKWebApp.view = view.GetRenderWindow()
self.Application.GetObjectIdMap().SetActiveObject("VIEW", view.GetRenderWindow())
def initialize(self, VTKWebApp, args):
# Create default pipeline (Only once for all the session)
if not VTKWebApp.view:
baseURL = args.baseURL
# support for overriding the base URL
scriptDir = os.path.dirname(os.path.realpath(__file__))
configPath = scriptDir + "/baseURL.txt"
if os.path.isfile(configPath):
f = file(configPath, "r")
baseURL = f.read().rstrip()
f.close()
# get our input data from romanesco
r = requests.get(baseURL + args.tableURI, verify=False)
tableJSON = r.json()
tableStr = tableJSON["data"]
r = requests.get(baseURL + args.treeURI, verify=False)
treeJSON = r.json()
treeStr = treeJSON["data"]
# deserialize our input data
tableReader = vtk.vtkTableReader()
tableReader.ReadFromInputStringOn()
tableReader.SetInputString(tableStr, len(tableStr))
tableReader.Update()
table = tableReader.GetOutput()
treeReader = vtk.vtkTreeReader()
treeReader.ReadFromInputStringOn()
treeReader.SetInputString(treeStr, len(treeStr))
treeReader.Update()
tree = treeReader.GetOutput()
# create our visualization item and load the data into it.
treeHeatmapItem = vtk.vtkTreeHeatmapItem()
treeHeatmapItem.SetTree(tree)
treeHeatmapItem.SetTable(table)
# detect if we are visualizing the results of a tree comparison
if tree.GetVertexData().GetArray("property.differences"):
treeHeatmapItem.GetDendrogram().SetColorArray("property.differences")
treeHeatmapItem.GetDendrogram().SetLineWidth(2.0)
# setup the window
view = vtk.vtkContextView()
view.GetRenderWindow().SetSize(int(args.width), int(args.height))
view.GetRenderer().SetBackground(1,1,1)
iren = view.GetInteractor()
iren.SetRenderWindow(view.GetRenderWindow())
transformItem = vtk.vtkContextTransform()
transformItem.AddItem(treeHeatmapItem)
transformItem.SetInteractive(1)
view.GetScene().AddItem(transformItem)
view.GetRenderWindow().SetMultiSamples(0)
iren.Initialize()
view.GetRenderWindow().Render()
# adjust zoom so the item nicely fills the screen
itemSize = [0, 0]
treeHeatmapItem.GetSize(itemSize)
itemSize.append(0)
transformItem.GetTransform().MultiplyPoint(itemSize, itemSize)
newWidth = view.GetScene().GetSceneWidth()
newHeight = view.GetScene().GetSceneHeight()
pageWidth = newWidth
pageHeight = newHeight
sx = pageWidth / itemSize[0]
sy = pageHeight / itemSize[1]
if sy < sx:
scale = sy;
else:
scale = sx;
if scale > 1:
scale = scale * 0.5
else:
scale = scale * 0.9
transformItem.Scale(scale, scale)
# center the item within the screen
itemCenter = [0, 0]
treeHeatmapItem.GetCenter(itemCenter)
itemCenter.append(0)
centerPt = vtk.vtkPoints2D()
centerPt.InsertNextPoint(newWidth / 2.0, newHeight / 2.0)
transformItem.GetTransform().InverseTransformPoints(centerPt, centerPt)
sceneCenter = [0, 0]
centerPt.GetPoint(0, sceneCenter)
dx = -1 * (itemCenter[0] - sceneCenter[0])
dy = -1 * (itemCenter[1] - sceneCenter[1])
transformItem.Translate(dx, dy)
# VTK Web application specific
VTKWebApp.view = view.GetRenderWindow()
self.Application.GetObjectIdMap().SetActiveObject("VIEW", view.GetRenderWindow())
