mirror of
https://github.com/scinote-eln/scinote-web.git
synced 2024-11-14 05:05:55 +08:00
103 lines
2.4 KiB
Ruby
103 lines
2.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Experiments
|
|
class GenerateWorkflowImageService
|
|
extend Service
|
|
require 'graphviz'
|
|
|
|
attr_reader :errors
|
|
|
|
def initialize(experiment:)
|
|
@exp = experiment
|
|
@graph_params = {
|
|
inputscale: 3,
|
|
size: '2,2',
|
|
pad: '0.4',
|
|
ratio: 'fill',
|
|
splines: true,
|
|
center: true,
|
|
pack: true,
|
|
bgcolor: Constants::COLOR_CONCRETE
|
|
}
|
|
@node_params = {
|
|
color: Constants::COLOR_VOLCANO,
|
|
style: :filled,
|
|
fontcolor: Constants::COLOR_VOLCANO,
|
|
shape: 'circle',
|
|
fontname: 'Arial',
|
|
fontsize: '16.0'
|
|
}
|
|
@edge_params = {
|
|
color: Constants::COLOR_VOLCANO,
|
|
penwidth: '3.0'
|
|
}
|
|
|
|
@graph = Graphviz::Graph.new('G', @graph_params)
|
|
@errors = []
|
|
end
|
|
|
|
def call
|
|
draw_diagram
|
|
save_file
|
|
self
|
|
end
|
|
|
|
def succeed?
|
|
@errors.none?
|
|
end
|
|
|
|
private
|
|
|
|
def draw_diagram
|
|
# Draw grouped modules
|
|
@exp.my_module_groups.each_with_index do |group, gindex|
|
|
subgraph_id = "cluster-#{gindex}"
|
|
nodes = {}
|
|
|
|
group.my_modules.workflow_ordered.each_with_index do |my_module, index|
|
|
# draw nodes
|
|
node = @graph.add_node(
|
|
"#{subgraph_id}-#{index}",
|
|
@node_params.merge(
|
|
label: '',
|
|
pos: "#{my_module.x / 10},-#{my_module.y / 10}!"
|
|
)
|
|
)
|
|
nodes[my_module.id] = node
|
|
end
|
|
|
|
# draw edges
|
|
group.my_modules.workflow_ordered.each do |m|
|
|
m.outputs.each do |output|
|
|
parent_node = nodes[m.id]
|
|
child_node = nodes[output.input_id]
|
|
parent_node.connect(child_node, @edge_params)
|
|
end
|
|
end
|
|
end
|
|
|
|
# Draw orphan nodes
|
|
@exp.my_modules.without_group.each do |my_module|
|
|
@graph.add_node(
|
|
"Orphan-#{my_module.id}",
|
|
@node_params.merge(
|
|
label: '',
|
|
pos: "#{my_module.x / 10},-#{my_module.y / 10}!"
|
|
)
|
|
)
|
|
end
|
|
end
|
|
|
|
def save_file
|
|
file = Tempfile.open(%w(wimg .png), Rails.root.join('tmp'))
|
|
begin
|
|
Graphviz.output(@graph, path: file.path, format: 'png', dot: 'neato')
|
|
file.rewind
|
|
@exp.workflowimg.attach(io: file, filename: File.basename(file.path))
|
|
ensure
|
|
file.close
|
|
file.unlink
|
|
end
|
|
end
|
|
end
|
|
end
|