1 package org.simantics.modeling.tests.commands;
3 import java.util.ArrayList;
5 import org.simantics.db.ReadGraph;
6 import org.simantics.db.Resource;
7 import org.simantics.db.WriteGraph;
8 import org.simantics.db.common.utils.NameUtils;
9 import org.simantics.db.exception.DatabaseException;
10 import org.simantics.db.testing.common.CommandSequenceEnvironment;
11 import org.simantics.db.testing.common.WriteCommand;
12 import org.simantics.diagram.stubs.DiagramResource;
13 import org.simantics.modeling.ModelingResources;
14 import org.simantics.modeling.tests.traits.SingleResourceTrait;
15 import org.simantics.modeling.tests.traits.SingleResourceTraitImpl;
16 import org.simantics.structural.stubs.StructuralResource2;
18 public class MergeFlags extends WriteCommand<CommandSequenceEnvironment> {
20 private SingleResourceTrait diagram;
22 public MergeFlags(SingleResourceTrait diagram) {
23 this.diagram = diagram;
26 public MergeFlags(Resource diagram) {
27 this(new SingleResourceTraitImpl(diagram));
30 public Resource getComposite(ReadGraph graph) throws DatabaseException {
31 DiagramResource DIA = DiagramResource.getInstance(graph);
32 ModelingResources MOD = ModelingResources.getInstance(graph);
33 StructuralResource2 STR = StructuralResource2.getInstance(graph);
34 Resource resource = diagram.getResource();
35 if(graph.isInstanceOf(resource, DIA.Diagram)) return graph.getSingleObject(resource, MOD.DiagramToComposite);
36 else if (graph.isInstanceOf(resource, STR.Composite)) return resource;
37 else throw new DatabaseException("Expected diagram or composite, got " + NameUtils.getSafeName(graph, resource));
41 protected void run(WriteGraph graph, CommandSequenceEnvironment environment) throws DatabaseException {
43 System.err.println("Merging flags in: " + NameUtils.getSafeName(graph, getComposite(graph)));
44 ArrayList<ArrayList<Resource>> groups = new ArrayList<ArrayList<Resource>>();
45 org.simantics.modeling.flags.MergeFlags.collectFlagGroupsInComposite(graph, getComposite(graph), groups);
46 for(ArrayList<Resource> group : groups) {
47 org.simantics.modeling.flags.MergeFlags.merge(graph, group);