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

   1 package org.simantics.modeling.tests.commands;
   2
   3 import org.simantics.Simantics;
   4 import org.simantics.databoard.Bindings;
   5 import org.simantics.db.ReadGraph;
   6 import org.simantics.db.Resource;
   7 import org.simantics.db.common.request.ResourceRead;
   8 import org.simantics.db.exception.DatabaseException;
   9 import org.simantics.db.layer0.request.Model;
  10 import org.simantics.db.testing.common.CommandSequenceEnvironment;
  11 import org.simantics.diagram.elements.DiagramNodeUtil;
  12 import org.simantics.g2d.canvas.impl.CanvasContext;
  13 import org.simantics.g2d.diagram.DiagramHints;
  14 import org.simantics.g2d.diagram.IDiagram;
  15 import org.simantics.g2d.scenegraph.ICanvasSceneGraphProvider;
  16 import org.simantics.layer0.Layer0;
  17 import org.simantics.layer0.utils.triggers.IActivation;
  18 import org.simantics.layer0.utils.triggers.IActivationManager;
  19 import org.simantics.modeling.ModelingResources;
  20 import org.simantics.modeling.tests.traits.SingleResourceTrait;
  21 import org.simantics.modeling.tests.traits.SingleResourceTraitImpl;
  22 import org.simantics.utils.threads.WorkerThread;
  23
  24 public class LoadDiagram extends ResourceWriteCommand<CommandSequenceEnvironment> implements SingleResourceTrait {
  25     private static final boolean DEBUG = true;
  26         private SingleResourceTrait diagram;
  27
  28         transient private ICanvasSceneGraphProvider provider;
  29         transient private CanvasContext ctx;
  30         transient private IActivation activation;
  31         transient private WorkerThread workerThread;
  32
  33         public LoadDiagram(SingleResourceTrait diagram) {
  34                 this.diagram = diagram;
  35         }
  36
  37         public LoadDiagram(Resource diagram) {
  38                 this(new SingleResourceTraitImpl(diagram));
  39         }
  40
  41         @Override
  42         public Resource runResource(CommandSequenceEnvironment environment) throws DatabaseException {
  43
  44         try {
  45
  46                 Resource model = Simantics.sync(new Model(diagram.getResource()));
  47                 String diagramName = Simantics.sync(new ResourceRead<String>(diagram.getResource()) {
  48
  49                                 @Override
  50                                 public String perform(ReadGraph graph) throws DatabaseException {
  51                                         Layer0 L0 = Layer0.getInstance(graph);
  52                                         ModelingResources MOD = ModelingResources.getInstance(graph);
  53                                         Resource composite = graph.getPossibleObject(resource, MOD.DiagramToComposite);
  54                                         return graph.getPossibleRelatedValue(composite, L0.HasName, Bindings.STRING);
  55                                 }
  56
  57                 });
  58                 if (DEBUG)
  59                     System.out.println("DEBUG: load " + diagramName);
  60             workerThread = new WorkerThread("Load diagram");
  61             workerThread.start();
  62                 // Load diagram
  63             ctx = new CanvasContext(workerThread);
  64                         provider = DiagramNodeUtil.loadSceneGraphProvider(ctx, model, diagram.getResource(), "/" + diagramName);
  65                         // Activate mapping
  66                 IActivationManager activationManager = Simantics.getSession().peekService(IActivationManager.class);
  67                 if (activationManager != null) {
  68                     activation = activationManager.activate(diagram.getResource());
  69                 }
  70
  71                 } catch (InterruptedException e) {
  72                         throw new DatabaseException(e);
  73                 }
  74
  75         return diagram.getResource();
  76
  77         }
  78
  79         public ICanvasSceneGraphProvider getProvider() {
  80                 return provider;
  81         }
  82
  83         public IActivation getActivation() {
  84                 return activation;
  85         }
  86
  87         public CanvasContext getContext() {
  88                 return ctx;
  89         }
  90
  91         public WorkerThread getWorkerThread() {
  92                 return workerThread;
  93         }
  94
  95         public IDiagram getDiagram() {
  96                 return ctx.getDefaultHintContext().getHint(DiagramHints.KEY_DIAGRAM);
  97         }
  98
  99 }