Source code for capsul.qt_gui.widgets.pipeline_user_view

# -*- coding: utf-8 -*-
'''
A widget to visualize a pipeline as a simple workflow

Classes
=======
:class:`PipelineUserView`
-------------------------
'''

from __future__ import print_function

# System import
from __future__ import absolute_import
import sys
import tempfile
from soma.subprocess import check_call

# Soma import
from soma.qt_gui.qt_backend import QtGui




class PipelineUserView(QtGui.QWidget):
    """ A widget to visualize a pipeline as a simple workflow.

    Uses Graphviz `dot` tool.
    """
    def __init__(self, pipeline):
        """ Initialize the WorkflowViewer class
        """
        # Inheritance
        super(PipelineUserView, self).__init__()

        # Class attributets
        self.pipeline = pipeline

        # Initialize the widget
        layout = QtGui.QVBoxLayout(self)
        self.label = QtGui.QLabel()
        layout.addWidget(self.label)
        #self.setLayout(layout)

        self.update()



    def update(self):
        image = tempfile.NamedTemporaryFile(suffix=".png")
        dot = tempfile.NamedTemporaryFile(suffix=".png")
        self.write(dot)
        dot.flush()
        check_call(["dot", "-Tpng", "-o", image.name, dot.name])
        self.pixmap = QtGui.QPixmap(image.name)  # .scaledToHeight(600)
        self.label.setPixmap(self.pixmap)


    def write(self, out=sys.stdout):
        graph = self.pipeline.workflow_graph()
        out.write('digraph workflow {\n'.encode())
        ids = {}
        for n in graph._nodes:
            id = str(len(ids))
            ids[n] = id
            out.write(('  %s [label="%s"];\n' % (id, n)).encode())
        for n, v in graph._links:
            out.write(('  %s -> %s;\n' % (ids[n], ids[v])).encode())
        out.write('}\n'.encode())