bundles/org.simantics.modeling.tests/src/org/simantics/modeling/tests/commands/MergeFlags.java

   1 package org.simantics.modeling.tests.commands;
   2
   3 import java.util.ArrayList;
   4
   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;
  17
  18 public class MergeFlags extends WriteCommand<CommandSequenceEnvironment> {
  19
  20         private SingleResourceTrait diagram;
  21
  22         public MergeFlags(SingleResourceTrait diagram) {
  23                 this.diagram = diagram;
  24         }
  25
  26         public MergeFlags(Resource diagram) {
  27                 this(new SingleResourceTraitImpl(diagram));
  28         }
  29
  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));
  38         }
  39
  40         @Override
  41         protected void run(WriteGraph graph, CommandSequenceEnvironment environment) throws DatabaseException {
  42
  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);
  48         }
  49
  50         }
  51
  52 }