bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/CompositeClassFactory.java

   1 /*******************************************************************************
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management
   3  * in Industry THTH ry.
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Eclipse Public License v1.0
   6  * which accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *     VTT Technical Research Centre of Finland - initial API and implementation
  11  *******************************************************************************/
  12 package org.simantics.diagram.adapter;
  13
  14 import gnu.trove.map.hash.TObjectIntHashMap;
  15
  16 import java.util.ArrayList;
  17 import java.util.Collections;
  18 import java.util.Comparator;
  19 import java.util.List;
  20 import java.util.concurrent.atomic.AtomicInteger;
  21
  22 import org.simantics.db.AsyncReadGraph;
  23 import org.simantics.db.Resource;
  24 import org.simantics.db.common.procedure.adapter.TransientCacheAsyncListener;
  25 import org.simantics.db.common.procedure.guarded.GuardedAsyncProcedureWrapper;
  26 import org.simantics.db.procedure.AsyncMultiProcedure;
  27 import org.simantics.db.procedure.AsyncProcedure;
  28 import org.simantics.diagram.synchronization.SynchronizationHints;
  29 import org.simantics.diagram.synchronization.graph.TransformSynchronizer;
  30 import org.simantics.g2d.canvas.ICanvasContext;
  31 import org.simantics.g2d.diagram.IDiagram;
  32 import org.simantics.g2d.element.ElementClass;
  33 import org.simantics.g2d.element.ElementHints;
  34 import org.simantics.g2d.element.IElement;
  35 import org.simantics.g2d.element.handler.impl.DefaultTransform;
  36 import org.simantics.g2d.element.handler.impl.StaticObjectAdapter;
  37 import org.simantics.g2d.element.handler.impl.StaticSymbolImageInitializer;
  38 import org.simantics.g2d.element.handler.impl.StaticSymbolImpl;
  39 import org.simantics.g2d.element.handler.impl.TextImpl;
  40 import org.simantics.g2d.elementclass.ImageClass;
  41 import org.simantics.g2d.image.DefaultImages;
  42 import org.simantics.g2d.image.Image;
  43
  44 public class CompositeClassFactory extends ElementFactoryAdapter {
  45
  46     private static final StaticSymbolImpl STATIC_SYMBOL = new StaticSymbolImpl(DefaultImages.COMPOSITION.get());
  47
  48     private static ElementClass build(Resource elementType) {
  49         String id = "Composite: " + elementType.getResourceId();
  50         return ElementClass.compile(
  51                 TextImpl.INSTANCE,
  52                 new StaticObjectAdapter(elementType),
  53                 DefaultTransform.INSTANCE,
  54                 StaticSymbolImageInitializer.INSTANCE,
  55                 STATIC_SYMBOL,
  56                 ImageClass.ImageElementHandler.INSTANCE)
  57                 .setId(id);
  58     }
  59
  60     @Override
  61     public void create(AsyncReadGraph graph, ICanvasContext canvas, IDiagram diagram, Resource elementType,
  62             AsyncProcedure<ElementClass> procedure) {
  63         procedure.execute(graph, build(elementType));
  64     }
  65
  66     @Override
  67     public void load(AsyncReadGraph graph, final ICanvasContext canvas, final IDiagram diagram, final Resource element, final IElement e, AsyncProcedure<IElement> procedure) {
  68         final GuardedAsyncProcedureWrapper<IElement> guard = new GuardedAsyncProcedureWrapper<IElement>(procedure, 1);
  69
  70         e.setHint(SynchronizationHints.HINT_SYNCHRONIZER, TransformSynchronizer.INSTANCE);
  71
  72         final List<IElement> elements = new ArrayList<IElement>();
  73         final TObjectIntHashMap<IElement> elementOrderMap = new TObjectIntHashMap<IElement>();
  74
  75         final AtomicInteger ready = new AtomicInteger(1);
  76         final AtomicInteger indexCounter = new AtomicInteger(0);
  77
  78         graph.forOrderedSet(element, new AsyncMultiProcedure<Resource>() {
  79
  80             @Override
  81             public void exception(AsyncReadGraph graph, Throwable throwable) {
  82                 guard.exception(graph, throwable);
  83             }
  84
  85             @Override
  86             public void execute(AsyncReadGraph graph, Resource r) {
  87
  88                 ready.incrementAndGet();
  89                 final int childIndex = indexCounter.getAndIncrement();
  90
  91                 graph.asyncRequest(new NodeRequest(canvas, diagram, r, null), new TransientCacheAsyncListener<IElement>() {
  92
  93                     @Override
  94                     public void exception(AsyncReadGraph graph, Throwable throwable) {
  95                         guard.exception(graph, throwable);
  96                     }
  97
  98                     @Override
  99                     public void execute(AsyncReadGraph graph, IElement child) {
 100                         elementReady(graph, child, childIndex);
 101                     }
 102                 });
 103
 104             }
 105
 106             @Override
 107             public void finished(AsyncReadGraph graph) {
 108                 elementReady(graph, null, -1);
 109             }
 110
 111             void elementReady(AsyncReadGraph graph, IElement child, int index) {
 112                 if (child != null) {
 113                     synchronized (elements) {
 114                         elements.add(child);
 115                         elementOrderMap.put(child, index);
 116                     }
 117                 }
 118                 if (ready.decrementAndGet() == 0) {
 119                     // Children ready, sort them and go!
 120                     Collections.sort(elements, new Comparator<IElement>() {
 121                         @Override
 122                         public int compare(IElement o1, IElement o2) {
 123                             int thisVal = elementOrderMap.get(o1);
 124                             int anotherVal = elementOrderMap.get(o2);
 125                             return (thisVal < anotherVal ? -1 : (thisVal == anotherVal ? 0 : 1));
 126                         }
 127                     });
 128
 129                     Image img = new CompositeImage(elements);
 130                     e.setHint(ElementHints.KEY_IMAGE, img);
 131
 132                     guard.execute(graph, e);
 133                 }
 134             }
 135         });
 136
 137     }
 138
 139 }