Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.browsing.ui.graph.impl / src / org / simantics / browsing / ui / graph / impl / contribution / FinalLabelDecoratorContributionImpl.java

diff --git a/bundles/org.simantics.browsing.ui.graph.impl/src/org/simantics/browsing/ui/graph/impl/contribution/FinalLabelDecoratorContributionImpl.java b/bundles/org.simantics.browsing.ui.graph.impl/src/org/simantics/browsing/ui/graph/impl/contribution/FinalLabelDecoratorContributionImpl.java
new file mode 100644 (file)
index 0000000..348e823
--- /dev/null
+++ b/bundles/org.simantics.browsing.ui.graph.impl/src/org/simantics/browsing/ui/graph/impl/contribution/FinalLabelDecoratorContributionImpl.java
@@ -0,0 +1,163 @@
+/*******************************************************************************\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.browsing.ui.graph.impl.contribution;\r
+\r
+import org.simantics.browsing.ui.BuiltinKeys.LabelDecoratorKey;\r
+import org.simantics.browsing.ui.DataSource;\r
+import org.simantics.browsing.ui.NodeContext;\r
+import org.simantics.browsing.ui.PrimitiveQueryUpdater;\r
+import org.simantics.browsing.ui.content.LabelDecorator;\r
+import org.simantics.browsing.ui.graph.impl.request.ResourceQuery;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.procedure.Listener;\r
+import org.simantics.db.procedure.Procedure;\r
+import org.simantics.utils.ui.ErrorLogger;\r
+\r
+public abstract class FinalLabelDecoratorContributionImpl extends LabelDecorator.Stub {\r
+\r
+    final LabelDecorator FRESH = new LabelDecorator.Stub();\r
+    final LabelDecorator PENDING = new LabelDecorator.Stub();\r
+    final LabelDecorator NO_DECORATION = new LabelDecorator.Stub();\r
+\r
+    protected final PrimitiveQueryUpdater       updater;\r
+    private final ResourceQuery<LabelDecorator> labelQuery;\r
+    private final LabelDecoratorKey             key;\r
+\r
+    /**\r
+     * The current decorator result produced for this contribution. The value\r
+     * should never be <code>null</code>, but one of {@link #FRESH},\r
+     * {@link #PENDING}, {@link #NO_DECORATION} instead.\r
+     */\r
+    protected LabelDecorator                      decorator = FRESH;\r
+\r
+    final protected NodeContext                   context;\r
+\r
+    public Object getIdentity(LabelDecoratorKey key) {\r
+        return key;\r
+    }\r
+\r
+    private void request() {\r
+\r
+        final DataSource<ReadGraph> source = updater.getDataSource(ReadGraph.class);\r
+        assert(source != null);\r
+        \r
+        final Procedure<LabelDecorator> procedure = createProcedure();\r
+\r
+        source.schedule(graph -> {\r
+            if(procedure instanceof Listener<?>)\r
+                graph.asyncRequest(labelQuery, (Listener<LabelDecorator>)procedure);\r
+            else\r
+                graph.asyncRequest(labelQuery, procedure);\r
+        });\r
+\r
+    }\r
+\r
+\r
+\r
+    @Override\r
+    public <Color_> Color_ decorateBackground(Color_ color, String column, int itemIndex) {\r
+        if(FRESH == decorator) {\r
+            decorator = PENDING;\r
+            request();\r
+        }\r
+        return decorator.decorateBackground(color, column, itemIndex);\r
+    }\r
+\r
+    @Override\r
+    public <Font_> Font_ decorateFont(Font_ font, String column, int itemIndex) {\r
+        if(FRESH == decorator) {\r
+            decorator = PENDING;\r
+            request();\r
+        }\r
+        return decorator.decorateFont(font, column, itemIndex);\r
+    }\r
+\r
+    @Override\r
+    public String decorateLabel(String label, String column, int itemIndex) {\r
+        if(FRESH == decorator) {\r
+            decorator = PENDING;\r
+            request();\r
+        }\r
+        return decorator.decorateLabel(label, column, itemIndex);\r
+    }\r
+\r
+    @Override\r
+    public <Color_> Color_ decorateForeground(Color_ color, String column, int itemIndex) {\r
+        if(FRESH == decorator) {\r
+            decorator = PENDING;\r
+            request();\r
+        }\r
+        return decorator.decorateForeground(color, column, itemIndex);\r
+    }\r
+\r
+    public FinalLabelDecoratorContributionImpl(final PrimitiveQueryUpdater updater, final NodeContext context, final LabelDecoratorKey key) {\r
+\r
+        this.updater = updater;\r
+        this.context = context;\r
+        this.key = key;\r
+\r
+        labelQuery = new ResourceQuery<LabelDecorator>(getIdentity(key), context) {\r
+\r
+            @Override\r
+            public LabelDecorator perform(ReadGraph graph) throws DatabaseException {\r
+                try {\r
+                    return getDecorator(graph, context);\r
+                } catch (DatabaseException e) {\r
+                    throw e;\r
+                } catch (Throwable t) {\r
+                    ErrorLogger.defaultLogError("LazyGraphLabeler.labelQuery produced unexpected exception.", t);\r
+                    return null;\r
+                }\r
+            }\r
+\r
+            @Override\r
+            public String toString() {\r
+                return FinalLabelDecoratorContributionImpl.this + " with context " + context;\r
+            }\r
+\r
+        };\r
+\r
+    }\r
+\r
+    protected Procedure<LabelDecorator> createProcedure() {\r
+\r
+        return new Procedure<LabelDecorator>() {\r
+\r
+            @Override\r
+            public void execute(LabelDecorator result) {\r
+                replaceResult(result);\r
+            }\r
+\r
+            @Override\r
+            public void exception(Throwable t) {\r
+                ErrorLogger.defaultLogError("LazyContributionImpl.childQuery failed, see exception for details.", t);\r
+            }\r
+\r
+        };\r
+\r
+    }\r
+\r
+    protected void replaceResult(LabelDecorator result) {\r
+        // Never let decorator become null, use a stub decorator instead.\r
+        if (result == null)\r
+            result = NO_DECORATION;\r
+\r
+        decorator = result;\r
+        updater.scheduleReplace(context, key, this);\r
+    }\r
+\r
+    // OVERRIDE\r
+\r
+    public abstract LabelDecorator getDecorator(ReadGraph graph, NodeContext context) throws DatabaseException;\r
+\r
+}\r