Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.diagram / src / org / simantics / diagram / adapter / CompositeClassFactory.java

diff --git a/bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/CompositeClassFactory.java b/bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/CompositeClassFactory.java
new file mode 100644 (file)
index 0000000..858a616
--- /dev/null
+++ b/bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/CompositeClassFactory.java
@@ -0,0 +1,139 @@
+/*******************************************************************************\r
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
+ * in Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ *     VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.diagram.adapter;\r
+\r
+import gnu.trove.map.hash.TObjectIntHashMap;\r
+\r
+import java.util.ArrayList;\r
+import java.util.Collections;\r
+import java.util.Comparator;\r
+import java.util.List;\r
+import java.util.concurrent.atomic.AtomicInteger;\r
+\r
+import org.simantics.db.AsyncReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.common.procedure.adapter.TransientCacheAsyncListener;\r
+import org.simantics.db.common.procedure.guarded.GuardedAsyncProcedureWrapper;\r
+import org.simantics.db.procedure.AsyncMultiProcedure;\r
+import org.simantics.db.procedure.AsyncProcedure;\r
+import org.simantics.diagram.synchronization.SynchronizationHints;\r
+import org.simantics.diagram.synchronization.graph.TransformSynchronizer;\r
+import org.simantics.g2d.canvas.ICanvasContext;\r
+import org.simantics.g2d.diagram.IDiagram;\r
+import org.simantics.g2d.element.ElementClass;\r
+import org.simantics.g2d.element.ElementHints;\r
+import org.simantics.g2d.element.IElement;\r
+import org.simantics.g2d.element.handler.impl.DefaultTransform;\r
+import org.simantics.g2d.element.handler.impl.StaticObjectAdapter;\r
+import org.simantics.g2d.element.handler.impl.StaticSymbolImageInitializer;\r
+import org.simantics.g2d.element.handler.impl.StaticSymbolImpl;\r
+import org.simantics.g2d.element.handler.impl.TextImpl;\r
+import org.simantics.g2d.elementclass.ImageClass;\r
+import org.simantics.g2d.image.DefaultImages;\r
+import org.simantics.g2d.image.Image;\r
+\r
+public class CompositeClassFactory extends ElementFactoryAdapter {\r
+\r
+    private static final StaticSymbolImpl STATIC_SYMBOL = new StaticSymbolImpl(DefaultImages.COMPOSITION.get());\r
+\r
+    private static ElementClass build(Resource elementType) {\r
+        String id = "Composite: " + elementType.getResourceId();\r
+        return ElementClass.compile(\r
+                TextImpl.INSTANCE,\r
+                new StaticObjectAdapter(elementType),\r
+                DefaultTransform.INSTANCE,\r
+                StaticSymbolImageInitializer.INSTANCE,\r
+                STATIC_SYMBOL,\r
+                ImageClass.ImageElementHandler.INSTANCE)\r
+                .setId(id);\r
+    }\r
+\r
+    @Override\r
+    public void create(AsyncReadGraph graph, ICanvasContext canvas, IDiagram diagram, Resource elementType,\r
+            AsyncProcedure<ElementClass> procedure) {\r
+        procedure.execute(graph, build(elementType));\r
+    }\r
+\r
+    @Override\r
+    public void load(AsyncReadGraph graph, final ICanvasContext canvas, final IDiagram diagram, final Resource element, final IElement e, AsyncProcedure<IElement> procedure) {\r
+        final GuardedAsyncProcedureWrapper<IElement> guard = new GuardedAsyncProcedureWrapper<IElement>(procedure, 1);\r
+\r
+        e.setHint(SynchronizationHints.HINT_SYNCHRONIZER, TransformSynchronizer.INSTANCE);\r
+\r
+        final List<IElement> elements = new ArrayList<IElement>();\r
+        final TObjectIntHashMap<IElement> elementOrderMap = new TObjectIntHashMap<IElement>();\r
+\r
+        final AtomicInteger ready = new AtomicInteger(1);\r
+        final AtomicInteger indexCounter = new AtomicInteger(0);\r
+\r
+        graph.forOrderedSet(element, new AsyncMultiProcedure<Resource>() {\r
+\r
+            @Override\r
+            public void exception(AsyncReadGraph graph, Throwable throwable) {\r
+                guard.exception(graph, throwable);\r
+            }\r
+\r
+            @Override\r
+            public void execute(AsyncReadGraph graph, Resource r) {\r
+\r
+                ready.incrementAndGet();\r
+                final int childIndex = indexCounter.getAndIncrement();\r
+\r
+                graph.asyncRequest(new NodeRequest(canvas, diagram, r, null), new TransientCacheAsyncListener<IElement>() {\r
+\r
+                    @Override\r
+                    public void exception(AsyncReadGraph graph, Throwable throwable) {\r
+                        guard.exception(graph, throwable);\r
+                    }\r
+\r
+                    @Override\r
+                    public void execute(AsyncReadGraph graph, IElement child) {\r
+                        elementReady(graph, child, childIndex);\r
+                    }\r
+                });\r
+\r
+            }\r
+\r
+            @Override\r
+            public void finished(AsyncReadGraph graph) {\r
+                elementReady(graph, null, -1);\r
+            }\r
+\r
+            void elementReady(AsyncReadGraph graph, IElement child, int index) {\r
+                if (child != null) {\r
+                    synchronized (elements) {\r
+                        elements.add(child);\r
+                        elementOrderMap.put(child, index);\r
+                    }\r
+                }\r
+                if (ready.decrementAndGet() == 0) {\r
+                    // Children ready, sort them and go!\r
+                    Collections.sort(elements, new Comparator<IElement>() {\r
+                        @Override\r
+                        public int compare(IElement o1, IElement o2) {\r
+                            int thisVal = elementOrderMap.get(o1);\r
+                            int anotherVal = elementOrderMap.get(o2);\r
+                            return (thisVal < anotherVal ? -1 : (thisVal == anotherVal ? 0 : 1));\r
+                        }\r
+                    });\r
+\r
+                    Image img = new CompositeImage(elements);\r
+                    e.setHint(ElementHints.KEY_IMAGE, img);\r
+\r
+                    guard.execute(graph, e);\r
+                }\r
+            }\r
+        });\r
+\r
+    }\r
+\r
+}
\ No newline at end of file