scinote-web/app/services/experiments/generate_workflow_image_service.rb

104 lines
2.4 KiB
Ruby
Raw Normal View History

2019-01-11 17:16:50 +08:00
# frozen_string_literal: true
module Experiments
class GenerateWorkflowImageService
extend Service
require 'graphviz'
attr_reader :errors
def initialize(experiment:)
@exp = experiment
2021-03-26 00:42:57 +08:00
@graph_params = {
inputscale: 3,
size: '2,2',
pad: '0.4',
ratio: 'fill',
splines: true,
center: true,
pack: true,
bgcolor: Constants::COLOR_CONCRETE
}
2021-03-26 00:42:57 +08:00
@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'
}
2019-01-11 17:16:50 +08:00
2021-03-26 00:42:57 +08:00
@graph = Graphviz::Graph.new('G', @graph_params)
2019-01-11 17:16:50 +08:00
@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
2021-03-26 00:42:57 +08:00
node = @graph.add_node(
2019-01-11 17:16:50 +08:00
"#{subgraph_id}-#{index}",
2021-03-26 00:42:57 +08:00
@node_params.merge(
label: '',
pos: "#{my_module.x / 10},-#{my_module.y / 10}!"
)
2019-01-11 17:16:50 +08:00
)
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]
2021-03-26 00:42:57 +08:00
parent_node.connect(child_node, @edge_params)
2019-01-11 17:16:50 +08:00
end
end
end
# Draw orphan nodes
@exp.my_modules.without_group.each do |my_module|
2021-03-26 00:42:57 +08:00
@graph.add_node(
"Orphan-#{my_module.id}",
2021-03-26 00:42:57 +08:00
@node_params.merge(
label: '',
pos: "#{my_module.x / 10},-#{my_module.y / 10}!"
)
)
end
2019-01-11 17:16:50 +08:00
end
def save_file
file = Tempfile.open(%w(wimg .png), Rails.root.join('tmp'))
begin
2021-03-26 00:42:57 +08:00
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
2019-01-11 17:16:50 +08:00
end
end
end