diff options
Diffstat (limited to 'subprojects/visualization/src')
2 files changed, 79 insertions, 64 deletions
diff --git a/subprojects/visualization/src/main/java/tools/refinery/visualization/ModelVisualizerAdapter.java b/subprojects/visualization/src/main/java/tools/refinery/visualization/ModelVisualizerAdapter.java index 42f01242..2f71d4f3 100644 --- a/subprojects/visualization/src/main/java/tools/refinery/visualization/ModelVisualizerAdapter.java +++ b/subprojects/visualization/src/main/java/tools/refinery/visualization/ModelVisualizerAdapter.java | |||
@@ -1,8 +1,7 @@ | |||
1 | package tools.refinery.visualization; | 1 | package tools.refinery.visualization; |
2 | 2 | ||
3 | import guru.nidi.graphviz.engine.Format; | ||
4 | import guru.nidi.graphviz.model.MutableGraph; | ||
5 | import tools.refinery.store.adapter.ModelAdapter; | 3 | import tools.refinery.store.adapter.ModelAdapter; |
4 | import tools.refinery.store.tuple.Tuple; | ||
6 | import tools.refinery.visualization.internal.FileFormat; | 5 | import tools.refinery.visualization.internal.FileFormat; |
7 | import tools.refinery.visualization.internal.ModelVisualizerBuilderImpl; | 6 | import tools.refinery.visualization.internal.ModelVisualizerBuilderImpl; |
8 | 7 | ||
@@ -13,21 +12,25 @@ public interface ModelVisualizerAdapter extends ModelAdapter { | |||
13 | return new ModelVisualizerBuilderImpl(); | 12 | return new ModelVisualizerBuilderImpl(); |
14 | } | 13 | } |
15 | 14 | ||
16 | public MutableGraph createVisualizationForCurrentModelState(); | ||
17 | |||
18 | public MutableGraph createVisualizationForModelState(Long version); | ||
19 | |||
20 | public String createDotForCurrentModelState(); | 15 | public String createDotForCurrentModelState(); |
21 | 16 | ||
22 | public String createDotForModelState(Long version); | 17 | public String createDotForModelState(Long version); |
23 | 18 | ||
24 | public boolean saveVisualization(MutableGraph graph, String path); | ||
25 | |||
26 | public boolean saveVisualization(MutableGraph graph, Format format, String path); | ||
27 | |||
28 | public boolean saveDot(String dot, String filePath); | 19 | public boolean saveDot(String dot, String filePath); |
29 | 20 | ||
30 | public boolean renderDot(String dot, String filePath); | 21 | public boolean renderDot(String dot, String filePath); |
31 | 22 | ||
32 | public boolean renderDot(String dot, FileFormat format, String filePath); | 23 | public boolean renderDot(String dot, FileFormat format, String filePath); |
24 | |||
25 | public void addTransition(Long from, Long to, String action); | ||
26 | |||
27 | |||
28 | public void addTransition(Long from, Long to, String action, Tuple activation); | ||
29 | public void addSolution(Long state); | ||
30 | |||
31 | public boolean saveDesignSpace(String path); | ||
32 | |||
33 | public boolean renderDesignSpace(String path); | ||
34 | |||
35 | public boolean renderDesignSpace(String path, FileFormat format); | ||
33 | } | 36 | } |
diff --git a/subprojects/visualization/src/main/java/tools/refinery/visualization/internal/ModelVisualizerAdapterImpl.java b/subprojects/visualization/src/main/java/tools/refinery/visualization/internal/ModelVisualizerAdapterImpl.java index 7456d913..9a284e24 100644 --- a/subprojects/visualization/src/main/java/tools/refinery/visualization/internal/ModelVisualizerAdapterImpl.java +++ b/subprojects/visualization/src/main/java/tools/refinery/visualization/internal/ModelVisualizerAdapterImpl.java | |||
@@ -1,14 +1,9 @@ | |||
1 | package tools.refinery.visualization.internal; | 1 | package tools.refinery.visualization.internal; |
2 | 2 | ||
3 | import guru.nidi.graphviz.attribute.GraphAttr; | ||
4 | import guru.nidi.graphviz.attribute.Label; | ||
5 | import guru.nidi.graphviz.engine.Format; | ||
6 | import guru.nidi.graphviz.engine.Graphviz; | ||
7 | import guru.nidi.graphviz.model.MutableGraph; | ||
8 | import tools.refinery.store.model.Interpretation; | 3 | import tools.refinery.store.model.Interpretation; |
9 | import tools.refinery.store.model.Model; | 4 | import tools.refinery.store.model.Model; |
10 | import tools.refinery.store.representation.AnySymbol; | 5 | import tools.refinery.store.representation.AnySymbol; |
11 | import tools.refinery.store.representation.TruthValue; | 6 | import tools.refinery.store.tuple.Tuple; |
12 | import tools.refinery.visualization.ModelVisualizerAdapter; | 7 | import tools.refinery.visualization.ModelVisualizerAdapter; |
13 | import tools.refinery.visualization.ModelVisualizerStoreAdapter; | 8 | import tools.refinery.visualization.ModelVisualizerStoreAdapter; |
14 | 9 | ||
@@ -16,12 +11,13 @@ import java.io.*; | |||
16 | import java.util.HashMap; | 11 | import java.util.HashMap; |
17 | import java.util.Map; | 12 | import java.util.Map; |
18 | 13 | ||
19 | import static guru.nidi.graphviz.model.Factory.*; | ||
20 | |||
21 | public class ModelVisualizerAdapterImpl implements ModelVisualizerAdapter { | 14 | public class ModelVisualizerAdapterImpl implements ModelVisualizerAdapter { |
22 | private final Model model; | 15 | private final Model model; |
23 | private final ModelVisualizerStoreAdapter storeAdapter; | 16 | private final ModelVisualizerStoreAdapter storeAdapter; |
24 | private final Map<AnySymbol, Interpretation<?>> interpretations; | 17 | private final Map<AnySymbol, Interpretation<?>> interpretations; |
18 | private final StringBuilder designSpaceBuilder = new StringBuilder(); | ||
19 | private int transitionCounter = 0; | ||
20 | |||
25 | public ModelVisualizerAdapterImpl(Model model, ModelVisualizerStoreAdapter storeAdapter) { | 21 | public ModelVisualizerAdapterImpl(Model model, ModelVisualizerStoreAdapter storeAdapter) { |
26 | this.model = model; | 22 | this.model = model; |
27 | this.storeAdapter = storeAdapter; | 23 | this.storeAdapter = storeAdapter; |
@@ -46,6 +42,7 @@ public class ModelVisualizerAdapterImpl implements ModelVisualizerAdapter { | |||
46 | } | 42 | } |
47 | interpretations.put(symbol, castInterpretation); | 43 | interpretations.put(symbol, castInterpretation); |
48 | } | 44 | } |
45 | designSpaceBuilder.append("digraph designSpace {\n"); | ||
49 | } | 46 | } |
50 | 47 | ||
51 | @Override | 48 | @Override |
@@ -59,36 +56,6 @@ public class ModelVisualizerAdapterImpl implements ModelVisualizerAdapter { | |||
59 | } | 56 | } |
60 | 57 | ||
61 | @Override | 58 | @Override |
62 | public MutableGraph createVisualizationForCurrentModelState() { | ||
63 | var graph = mutGraph("model").setDirected(true).graphAttrs().add(GraphAttr.dpi(100)); | ||
64 | for (var entry : interpretations.entrySet()) { | ||
65 | var key = entry.getKey(); | ||
66 | var arity = key.arity(); | ||
67 | var cursor = entry.getValue().getAll(); | ||
68 | while (cursor.move()) { | ||
69 | if (arity == 1) { | ||
70 | var id = cursor.getKey().get(0); | ||
71 | graph.add(mutNode(String.valueOf(id)).add("label", key.name() + ": " + id)); | ||
72 | } else { | ||
73 | var from = cursor.getKey().get(0); | ||
74 | var to = cursor.getKey().get(1); | ||
75 | graph.add(mutNode(String.valueOf(from)).addLink(to(mutNode(String.valueOf(to))).with(Label.of(key.name())))); | ||
76 | } | ||
77 | } | ||
78 | } | ||
79 | return graph; | ||
80 | } | ||
81 | |||
82 | @Override | ||
83 | public MutableGraph createVisualizationForModelState(Long version) { | ||
84 | var currentVersion = model.getState(); | ||
85 | model.restore(version); | ||
86 | var graph = createVisualizationForCurrentModelState(); | ||
87 | model.restore(currentVersion); | ||
88 | return graph; | ||
89 | } | ||
90 | |||
91 | @Override | ||
92 | public String createDotForCurrentModelState() { | 59 | public String createDotForCurrentModelState() { |
93 | var sb = new StringBuilder(); | 60 | var sb = new StringBuilder(); |
94 | sb.append("digraph model {\n"); | 61 | sb.append("digraph model {\n"); |
@@ -123,22 +90,6 @@ public class ModelVisualizerAdapterImpl implements ModelVisualizerAdapter { | |||
123 | } | 90 | } |
124 | 91 | ||
125 | @Override | 92 | @Override |
126 | public boolean saveVisualization(MutableGraph graph, String path) { | ||
127 | return saveVisualization(graph, Format.PNG, path); | ||
128 | } | ||
129 | |||
130 | @Override | ||
131 | public boolean saveVisualization(MutableGraph graph, Format format, String path) { | ||
132 | try { | ||
133 | Graphviz.fromGraph(graph).render(format).toFile(new File(path)); | ||
134 | return true; | ||
135 | } catch (IOException e) { | ||
136 | e.printStackTrace(); | ||
137 | return false; | ||
138 | } | ||
139 | } | ||
140 | |||
141 | @Override | ||
142 | public boolean saveDot(String dot, String filePath) { | 93 | public boolean saveDot(String dot, String filePath) { |
143 | File file = new File(filePath); | 94 | File file = new File(filePath); |
144 | file.getParentFile().mkdirs(); | 95 | file.getParentFile().mkdirs(); |
@@ -172,4 +123,65 @@ public class ModelVisualizerAdapterImpl implements ModelVisualizerAdapter { | |||
172 | } | 123 | } |
173 | return true; | 124 | return true; |
174 | } | 125 | } |
126 | |||
127 | @Override | ||
128 | public void addTransition(Long from, Long to, String action) { | ||
129 | designSpaceBuilder.append(from).append(" -> ").append(to).append(" [label=\"").append(transitionCounter++) | ||
130 | .append(": ").append(action).append("\"]\n"); | ||
131 | |||
132 | } | ||
133 | |||
134 | @Override | ||
135 | public void addTransition(Long from, Long to, String action, Tuple activation) { | ||
136 | designSpaceBuilder.append(from).append(" -> ").append(to).append(" [label=\"").append(transitionCounter++) | ||
137 | .append(": ").append(action).append(" / "); | ||
138 | |||
139 | |||
140 | for (int i = 0; i < activation.getSize(); i++) { | ||
141 | designSpaceBuilder.append(activation.get(i)); | ||
142 | if (i < activation.getSize() - 1) { | ||
143 | designSpaceBuilder.append(", "); | ||
144 | } | ||
145 | } | ||
146 | designSpaceBuilder.append("\"]\n"); | ||
147 | } | ||
148 | |||
149 | @Override | ||
150 | public void addSolution(Long state) { | ||
151 | designSpaceBuilder.append(state).append(" [shape = doublecircle]\n"); | ||
152 | } | ||
153 | |||
154 | private String buildDesignSpaceDot() { | ||
155 | for (var state : storeAdapter.getStore().getStates()) { | ||
156 | designSpaceBuilder.append(state).append(" [URL=\"./").append(state).append(".svg\"]\n"); | ||
157 | } | ||
158 | designSpaceBuilder.append("}"); | ||
159 | return designSpaceBuilder.toString(); | ||
160 | } | ||
161 | |||
162 | @Override | ||
163 | public boolean saveDesignSpace(String path) { | ||
164 | saveDot(buildDesignSpaceDot(), path + "/designSpace.dot"); | ||
165 | for (var state : storeAdapter.getStore().getStates()) { | ||
166 | saveDot(createDotForModelState(state), path + "/" + state + ".dot"); | ||
167 | } | ||
168 | return true; | ||
169 | } | ||
170 | |||
171 | @Override | ||
172 | public boolean renderDesignSpace(String path) { | ||
173 | return renderDesignSpace(path, FileFormat.SVG); | ||
174 | } | ||
175 | |||
176 | @Override | ||
177 | public boolean renderDesignSpace(String path, FileFormat format) { | ||
178 | for (var state : storeAdapter.getStore().getStates()) { | ||
179 | var stateDot = createDotForModelState(state); | ||
180 | saveDot(stateDot, path + "/" + state + ".dot"); | ||
181 | renderDot(stateDot, path + "/" + state + "." + format.getFormat()); | ||
182 | } | ||
183 | var designSpaceDot = buildDesignSpaceDot(); | ||
184 | saveDot(designSpaceDot, path + "/designSpace.dot"); | ||
185 | return renderDot(designSpaceDot, format, path + "/designSpace." + format.getFormat()); | ||
186 | } | ||
175 | } | 187 | } |