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)))
def copyVistrailController(vt_controller, view=DummyView()):
newvt_controller = VistrailController()
current_log = vt_controller.log
vistrail = vt_controller.vistrail
newvt_controller.log = current_log
newvt_controller.current_pipeline_view = view.scene()
newvt_controller.set_vistrail(vistrail, None)
newvt_controller.disable_autosave()
return newvt_controller
def copyVistrailController(vt_controller, view=DummyView()):
newvt_controller = VistrailController()
current_log = vt_controller.log
vistrail = vt_controller.vistrail
newvt_controller.log = current_log
newvt_controller.current_pipeline_view = view.scene()
newvt_controller.set_vistrail(vistrail, None)
newvt_controller.disable_autosave()
return newvt_controller
def get_wf_graph(w_list, workflow_info=None, pdf=False):
"""run_and_get_results(w_list: list of (locator, version),
workflow_info:str, pdf:bool)
Load all workflows in wf_list and dump their graph to workflow_info.
"""
result = []
if is_running_gui():
from gui.vistrail_controller import VistrailController as \
GUIVistrailController
for locator, workflow in w_list:
try:
(v, abstractions, thumbnails, mashups) = load_vistrail(locator)
controller = GUIVistrailController()
if type(workflow) == type("str"):
version = v.get_version_number(workflow)
elif type(workflow) in [type(1), long]:
version = workflow
elif workflow is None:
version = controller.get_latest_version_in_graph()
else:
msg = "Invalid version tag or number: %s" % workflow
raise VistrailsInternalError(msg)
controller.change_selected_version(version)
if (workflow_info is not None
and controller.current_pipeline is not None):
from gui.pipeline_view import QPipelineView
pipeline_view = QPipelineView()
controller.current_pipeline_view = pipeline_view.scene()
controller.set_vistrail(v, locator, abstractions,
thumbnails, mashups)
pipeline_view.scene().setupScene(
controller.current_pipeline)
if pdf:
base_fname = "%s_%s_pipeline.pdf" % (
locator.short_name, version)
filename = os.path.join(workflow_info, base_fname)
pipeline_view.scene().saveToPDF(filename)
else:
base_fname = "%s_%s_pipeline.png" % (
locator.short_name, version)
filename = os.path.join(workflow_info, base_fname)
pipeline_view.scene().saveToPNG(filename)
del pipeline_view
result.append((True, ""))
except Exception, e:
result.append((False, str(e)))
def get_wf_graph(w_list, workflow_info=None, pdf=False):
"""run_and_get_results(w_list: list of (locator, version),
workflow_info:str, pdf:bool)
Load all workflows in wf_list and dump their graph to workflow_info.
"""
result = []
if is_running_gui():
from gui.vistrail_controller import VistrailController as \
GUIVistrailController
for locator, workflow in w_list:
try:
(v, abstractions , thumbnails, mashups) = load_vistrail(locator)
controller = GUIVistrailController()
if type(workflow) == type("str"):
version = v.get_version_number(workflow)
elif type(workflow) in [ type(1), long]:
version = workflow
elif workflow is None:
version = controller.get_latest_version_in_graph()
else:
msg = "Invalid version tag or number: %s" % workflow
raise VistrailsInternalError(msg)
controller.change_selected_version(version)
if (workflow_info is not None and
controller.current_pipeline is not None):
from gui.pipeline_view import QPipelineView
pipeline_view = QPipelineView()
controller.current_pipeline_view = pipeline_view.scene()
controller.set_vistrail(v, locator, abstractions, thumbnails,
mashups)
pipeline_view.scene().setupScene(controller.current_pipeline)
if pdf:
base_fname = "%s_%s_pipeline.pdf" % (locator.short_name, version)
filename = os.path.join(workflow_info, base_fname)
pipeline_view.scene().saveToPDF(filename)
else:
base_fname = "%s_%s_pipeline.png" % (locator.short_name, version)
filename = os.path.join(workflow_info, base_fname)
pipeline_view.scene().saveToPNG(filename)
del pipeline_view
result.append((True, ""))
except Exception, e:
result.append((False, str(e)))
