def get_vt_graph(vt_list, tree_info, pdf=False):
"""get_vt_graph(vt_list: list of locator, tree_info:str)
Load all vistrails in vt_list and dump their tree to tree_info.
"""
result = []
if is_running_gui():
from gui.vistrail_controller import VistrailController as \
GUIVistrailController
for locator in vt_list:
try:
(v, abstractions, thumbnails, mashups) = load_vistrail(locator)
controller = GUIVistrailController()
if tree_info is not None:
from gui.version_view import QVersionTreeView
version_view = QVersionTreeView()
from gui.pipeline_view import QPipelineView
pipeline_view = QPipelineView()
controller.current_pipeline_view = pipeline_view.scene()
controller.set_vistrail(v, locator, abstractions,
thumbnails, mashups)
version_view.scene().setupScene(controller)
if pdf:
base_fname = "graph_%s.pdf" % locator.short_name
filename = os.path.join(tree_info, base_fname)
version_view.scene().saveToPDF(filename)
else:
base_fname = "graph_%s.png" % locator.short_name
filename = os.path.join(tree_info, base_fname)
version_view.scene().saveToPNG(filename)
del version_view
result.append((True, ""))
except Exception, e:
result.append((False, str(e)))
def get_vt_graph(vt_list, tree_info, pdf=False):
"""get_vt_graph(vt_list: list of locator, tree_info:str)
Load all vistrails in vt_list and dump their tree to tree_info.
"""
result = []
if is_running_gui():
from gui.vistrail_controller import VistrailController as \
GUIVistrailController
for locator in vt_list:
try:
(v, abstractions , thumbnails, mashups) = load_vistrail(locator)
controller = GUIVistrailController()
if tree_info is not None:
from gui.version_view import QVersionTreeView
version_view = QVersionTreeView()
from gui.pipeline_view import QPipelineView
pipeline_view = QPipelineView()
controller.current_pipeline_view = pipeline_view.scene()
controller.set_vistrail(v, locator, abstractions, thumbnails,
mashups)
version_view.scene().setupScene(controller)
if pdf:
base_fname = "graph_%s.pdf" % locator.short_name
filename = os.path.join(tree_info, base_fname)
version_view.scene().saveToPDF(filename)
else:
base_fname = "graph_%s.png" % locator.short_name
filename = os.path.join(tree_info, base_fname)
version_view.scene().saveToPNG(filename)
del version_view
result.append((True, ""))
except Exception, e:
result.append((False, str(e)))
class QVersionTab(QDockContainer, QToolWindowInterface):
"""
QVersionTab is a tab widget setting QVersionTreeView in a
center while having surrounding tool windows
"""
def __init__(self, parent=None):
""" QVersionTab(parent: QWidget) -> QVersionTab
Make it a main window with dockable area and a
QVersionTreeView in the middle
"""
QDockContainer.__init__(self, parent)
self.setWindowTitle('Version Tree')
self.versionView = QVersionTreeView()
self.setCentralWidget(self.versionView)
self.toolWindow().setFeatures(QtGui.QDockWidget.NoDockWidgetFeatures)
self.toolWindow().hide()
self.versionProp = QVersionProp(self)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea,
self.versionProp.toolWindow())
self.controller = None
self.connect(self.toolWindow(),
QtCore.SIGNAL('topLevelChanged(bool)'),
self.updateWindowTitle)
self.connect(self.versionView.scene(),
QtCore.SIGNAL('twoVersionsSelected(int,int)'),
self.twoVersionsSelected)
self.connect(self.versionView.scene(),
QtCore.SIGNAL('addToBookmarks'),
self.add_bookmark)
self.connect(self.versionView,
QtCore.SIGNAL('resetQuery()'),
self.resetQuery)
def updateWindowTitle(self, topLevel):
""" updateWindowTitle(topLevel: bool) -> None
Change the current widget title depends on the top level status
"""
if topLevel:
self.setWindowTitle('Version Tree <' +
self.toolWindow().parent().windowTitle()+'>')
else:
self.setWindowTitle('Pipeline')
def twoVersionsSelected(self, id1, id2):
""" twoVersionsSelected(id1: Int, id2: Int) -> None
Two versions are selected in the version tree, emit a signal
"""
self.emit(QtCore.SIGNAL('twoVersionsSelected(int,int)'), id1, id2)
def setController(self, controller):
""" setController(controller: VistrailController) -> None
Assign a vistrail controller to this version tree view
"""
oldController = self.versionView.scene().controller
if oldController!=controller:
if oldController!=None:
self.disconnect(oldController,
QtCore.SIGNAL('vistrailChanged()'),
self.vistrailChanged)
self.disconnect(oldController,
QtCore.SIGNAL('invalidateSingleNodeInVersionTree'),
self.single_node_changed)
self.disconnect(oldController,
QtCore.SIGNAL('notesChanged()'),
self.notesChanged)
self.disconnect(oldController,
QtCore.SIGNAL('audioChanged()'),
self.audioChanged)
self.controller = controller
self.versionView.scene().controller = controller
self.connect(controller,
QtCore.SIGNAL('vistrailChanged()'),
self.vistrailChanged)
self.connect(controller,
QtCore.SIGNAL('invalidateSingleNodeInVersionTree'),
self.single_node_changed)
self.connect(controller,
QtCore.SIGNAL("notesChanged()"),
self.notesChanged)
self.connect(controller,
QtCore.SIGNAL("audioChanged()"),
self.audioChanged)
if controller:
self.vistrailChanged()
self.versionProp.updateController(controller)
self.versionView.versionProp.updateController(controller)
def vistrailChanged(self):
""" vistrailChanged() -> None
An action was performed on the current vistrail
"""
self.versionView.scene().setupScene(self.controller)
if self.controller and self.controller.reset_version_view:
self.versionView.scene().fitToAllViews()
if self.controller:
self.versionProp.updateVersion(self.controller.current_version)
self.versionView.versionProp.updateVersion(self.controller.current_version)
self.emit(QtCore.SIGNAL("vistrailChanged()"))
def single_node_changed(self, old_version, new_version):
""" single_node_changed(old_version, new_version)
Handle single node change on version tree by not recomputing
entire scene.
"""
self.versionView.scene().update_scene_single_node_change(self.controller,
old_version,
new_version)
if self.controller and self.controller.reset_version_view:
self.versionView.scene().fitToAllViews()
if self.controller:
self.versionProp.updateVersion(self.controller.current_version)
self.versionView.versionProp.updateVersion(self.controller.current_version)
self.emit(QtCore.SIGNAL("vistrailChanged()"))
def audioChanged(self):
""" audioChanged() -> None
The audio for the current vistrail version changed
"""
if self.controller:
self.versionView.versionProp.updateVersion(self.controller.current_version)
def notesChanged(self):
""" notesChanged() -> None
The notes for the current vistrail version changed
"""
if self.controller:
self.versionView.versionProp.updateVersion(self.controller.current_version)
def add_bookmark(self, id, name):
"""add_bookmark(id: int, label:name) -> None
Gather all information concerning the new bookmark and send it to
BookmarksManager
"""
vistrailsFile = self.controller.file_name
BookmarksManager.add_bookmark('',vistrailsFile,id,name)
def resetQuery(self):
""" resetQuery() -> None
pass along the signal
"""
self.emit(QtCore.SIGNAL('resetQuery()'))
self.versionProp.resetSearch(emit_signal=False)
class QVersionTab(QDockContainer, QToolWindowInterface):
"""
QVersionTab is a tab widget setting QVersionTreeView in a
center while having surrounding tool windows
"""
def __init__(self, parent=None):
""" QVersionTab(parent: QWidget) -> QVersionTab
Make it a main window with dockable area and a
QVersionTreeView in the middle
"""
QDockContainer.__init__(self, parent)
self.setWindowTitle('Version Tree')
self.versionView = QVersionTreeView()
self.setCentralWidget(self.versionView)
self.toolWindow().setFeatures(QtGui.QDockWidget.NoDockWidgetFeatures)
self.toolWindow().hide()
self.versionProp = QVersionProp(self)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea,
self.versionProp.toolWindow())
self.controller = None
self.connect(self.toolWindow(), QtCore.SIGNAL('topLevelChanged(bool)'),
self.updateWindowTitle)
self.connect(self.versionView.scene(),
QtCore.SIGNAL('twoVersionsSelected(int,int)'),
self.twoVersionsSelected)
self.connect(self.versionView, QtCore.SIGNAL('resetQuery()'),
self.resetQuery)
def updateWindowTitle(self, topLevel):
""" updateWindowTitle(topLevel: bool) -> None
Change the current widget title depends on the top level status
"""
if topLevel:
self.setWindowTitle('Version Tree <' +
self.toolWindow().parent().windowTitle() + '>')
else:
self.setWindowTitle('Version Tree')
def twoVersionsSelected(self, id1, id2):
""" twoVersionsSelected(id1: Int, id2: Int) -> None
Two versions are selected in the version tree, emit a signal
"""
self.emit(QtCore.SIGNAL('twoVersionsSelected(int,int)'), id1, id2)
def setController(self, controller):
""" setController(controller: VistrailController) -> None
Assign a vistrail controller to this version tree view
"""
oldController = self.versionView.scene().controller
if oldController != controller:
if oldController != None:
self.disconnect(oldController,
QtCore.SIGNAL('vistrailChanged()'),
self.vistrailChanged)
self.disconnect(
oldController,
QtCore.SIGNAL('invalidateSingleNodeInVersionTree'),
self.single_node_changed)
self.disconnect(oldController, QtCore.SIGNAL('notesChanged()'),
self.notesChanged)
self.controller = controller
self.versionView.scene().controller = controller
self.connect(controller, QtCore.SIGNAL('vistrailChanged()'),
self.vistrailChanged)
self.connect(controller,
QtCore.SIGNAL('invalidateSingleNodeInVersionTree'),
self.single_node_changed)
self.connect(controller, QtCore.SIGNAL("notesChanged()"),
self.notesChanged)
if controller:
self.vistrailChanged()
self.versionProp.updateController(controller)
self.versionView.versionProp.updateController(controller)
def vistrailChanged(self):
""" vistrailChanged() -> None
An action was performed on the current vistrail
"""
self.versionView.scene().setupScene(self.controller)
if self.controller and self.controller.reset_version_view:
self.versionView.scene().fitToAllViews()
if self.controller:
self.versionProp.updateVersion(self.controller.current_version)
self.versionView.versionProp.updateVersion(
self.controller.current_version)
self.emit(QtCore.SIGNAL("vistrailChanged()"))
def single_node_changed(self, old_version, new_version):
""" single_node_changed(old_version, new_version)
Handle single node change on version tree by not recomputing
entire scene.
"""
self.versionView.scene().update_scene_single_node_change(
self.controller, old_version, new_version)
if self.controller and self.controller.reset_version_view:
self.versionView.scene().fitToAllViews()
if self.controller:
self.versionProp.updateVersion(self.controller.current_version)
self.versionView.versionProp.updateVersion(
self.controller.current_version)
self.emit(QtCore.SIGNAL("vistrailChanged()"))
def notesChanged(self):
""" notesChanged() -> None
The notes for the current vistrail version changed
"""
if self.controller:
self.versionView.versionProp.updateVersion(
self.controller.current_version)
def resetQuery(self):
""" resetQuery() -> None
pass along the signal
"""
self.emit(QtCore.SIGNAL('resetQuery()'))
self.versionProp.resetSearch(emit_signal=False)