Warning
This document is for an in-development version of Galaxy. You can alternatively view this page in the latest release if it exists or view the top of the latest release's documentation.
Source code for galaxy.workflow.render
import svgwrite
MARGIN = 5
LINE_SPACING = 15
STANDALONE_SVG_TEMPLATE = """<?xml version="1.0" standalone="no"?>
<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN"
"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">
%s"""
[docs]class WorkflowCanvas(object):
[docs] def __init__(self):
self.canvas = svgwrite.Drawing(profile='full')
self.connectors = []
self.boxes = []
self.text = []
self.in_pos = {}
self.out_pos = {}
self.widths = {}
self.max_x = 0
self.max_y = 0
self.max_width = 0
self.data = []
[docs] def finish(self):
# max_x, max_y, max_width = self.max_x, self.max_y, self.max_width
for box in self.boxes:
self.canvas.add(box)
for connector in self.connectors:
self.canvas.add(connector)
text_style_layer = self.canvas.g(style="font-family: Helvetica, Arial, FreeSans, Sans, sans, sans-serif;")
for text in self.text:
text_style_layer.add(text)
self.canvas.add(text_style_layer)
return self.canvas
[docs] def add_boxes(self, step_dict, width, name_fill):
x, y = step_dict['position']['left'], step_dict['position']['top']
self.boxes.append(svgwrite.shapes.Rect((x - MARGIN, y), (width, 30), fill=name_fill, stroke='#000000'))
box_height = (len(step_dict['data_inputs']) + len(step_dict['data_outputs'])) * LINE_SPACING + MARGIN
# Draw separator line.
if len(step_dict['data_inputs']) > 0:
box_height += LINE_SPACING
sep_y = y + len(step_dict['data_inputs']) * LINE_SPACING + 40
self.text.append(svgwrite.shapes.Line((x - MARGIN, sep_y), (x + width - MARGIN, sep_y), stroke=svgwrite.rgb(0, 0, 0)))
# Define an input/output box.
self.boxes.append(svgwrite.shapes.Rect((x - MARGIN, y + 30), (width, box_height), fill="#ffffff", stroke=svgwrite.rgb(0, 0, 0)))
[docs] def add_text(self, module_data_inputs, module_data_outputs, step, module_name):
left, top = step.position['left'], step.position['top']
x, y = left, top
order_index = step.order_index
max_len = len(module_name) * 1.5
self.text.append(svgwrite.text.Text(module_name, (x, y + 20), style='font-size:14px'))
y += 45
count = 0
in_pos = self.in_pos
out_pos = self.out_pos
for di in module_data_inputs:
cur_y = y + count * LINE_SPACING
if order_index not in in_pos:
in_pos[order_index] = {}
in_pos[order_index][di['name']] = (x, cur_y)
self.text.append(svgwrite.text.Text(di['label'], (x, cur_y), style='font-size:10px'))
count += 1
max_len = max(max_len, len(di['label']))
if len(module_data_inputs) > 0:
y += LINE_SPACING
for do in module_data_outputs:
cur_y = y + count * LINE_SPACING
if order_index not in out_pos:
out_pos[order_index] = {}
out_pos[order_index][do['name']] = (x, cur_y)
self.text.append(svgwrite.text.Text(do['name'], (x, cur_y), style='font-size:10px'))
count += 1
max_len = max(max_len, len(do['name']))
self.widths[order_index] = max_len * 5.5
self.max_x = max(self.max_x, left)
self.max_y = max(self.max_y, top)
self.max_width = max(self.max_width, self.widths[order_index])
[docs] def add_connection(self, step_dict, conn, output_dict):
in_coords = self.in_pos[step_dict['id']][conn]
# out_pos_index will be a step number like 1, 2, 3...
out_pos_index = output_dict['id']
# out_pos_name will be a string like 'o', 'o2', etc.
out_pos_name = output_dict['output_name']
if out_pos_index in self.out_pos:
# out_conn_index_dict will be something like:
# 7: {'o': (824.5, 618)}
out_conn_index_dict = self.out_pos[out_pos_index]
if out_pos_name in out_conn_index_dict:
out_conn_pos = out_conn_index_dict[out_pos_name]
else:
# Take any key / value pair available in out_conn_index_dict.
# A problem will result if the dictionary is empty.
if out_conn_index_dict:
key = next(iter(out_conn_index_dict.keys()))
out_conn_pos = self.out_pos[out_pos_index][key]
adjusted = (out_conn_pos[0] + self.widths[output_dict['id']], out_conn_pos[1])
self.text.append(svgwrite.shapes.Circle(center=(out_conn_pos[0] + self.widths[output_dict['id']] - MARGIN,
out_conn_pos[1] - MARGIN),
r=5,
fill="#ffffff",
stroke="#000000"))
marker = self.canvas.marker(overflow='visible',
refX="0", refY="5",
viewBox="0 0 10 5",
markerWidth="8",
markerHeight="10",
markerUnits="strokeWidth",
orient="auto", stroke="none", fill="black")
marker.add(self.canvas.path(d="M 0 0 L 10 5 L 0 10 z"))
self.canvas.defs.add(marker)
conn = svgwrite.shapes.Line((adjusted[0], adjusted[1] - MARGIN),
(in_coords[0] - 10, in_coords[1]),
stroke="#000000")
conn['marker-end'] = marker.get_funciri()
self.connectors.append(conn)
[docs] def add_steps(self, highlight_errors=False):
# Only highlight missing tools if displaying in the tool shed.
for step_dict in self.data:
tool_unavailable = step_dict.get('tool_errors', False)
if highlight_errors and tool_unavailable:
fill = "#EBBCB2"
else:
fill = "#EBD9B2"
width = self.widths[step_dict['id']]
self.add_boxes(step_dict, width, fill)
for conn, output_dict in step_dict['input_connections'].items():
self.add_connection(step_dict, conn, output_dict)
[docs] def populate_data_for_step(self, step, module_name, module_data_inputs, module_data_outputs, tool_errors=None):
step_dict = {
'id': step.order_index,
'data_inputs': module_data_inputs,
'data_outputs': module_data_outputs,
'position': step.position
}
if tool_errors:
step_dict['tool_errors'] = tool_errors
input_conn_dict = {}
for conn in step.input_connections:
input_conn_dict[conn.input_name] = \
dict(id=conn.output_step.order_index, output_name=conn.output_name)
step_dict['input_connections'] = input_conn_dict
self.data.append(step_dict)
self.add_text(module_data_inputs, module_data_outputs, step, module_name)