Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.browsing.ui.swt / src / org / simantics / browsing / ui / swt / AdapterImagerFactory.java

diff --git a/bundles/org.simantics.browsing.ui.swt/src/org/simantics/browsing/ui/swt/AdapterImagerFactory.java b/bundles/org.simantics.browsing.ui.swt/src/org/simantics/browsing/ui/swt/AdapterImagerFactory.java
new file mode 100644 (file)
index 0000000..6afe648
--- /dev/null
+++ b/bundles/org.simantics.browsing.ui.swt/src/org/simantics/browsing/ui/swt/AdapterImagerFactory.java
@@ -0,0 +1,123 @@
+/*******************************************************************************\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.swt;\r
+\r
+import java.util.Collection;\r
+import java.util.Collections;\r
+import java.util.HashMap;\r
+\r
+import org.eclipse.core.runtime.IAdaptable;\r
+import org.eclipse.jface.resource.ImageDescriptor;\r
+import org.eclipse.swt.graphics.Image;\r
+import org.simantics.browsing.ui.BuiltinKeys;\r
+import org.simantics.browsing.ui.BuiltinKeys.ImagerKey;\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.common.EvaluatorData.Transformer;\r
+import org.simantics.browsing.ui.common.imagers.ContainerImager;\r
+import org.simantics.browsing.ui.content.Imager;\r
+import org.simantics.browsing.ui.content.ImagerFactory;\r
+import org.simantics.browsing.ui.graph.impl.MissingImageDescriptor;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.ui.icons.ImageDescriptorProvider;\r
+import org.simantics.utils.datastructures.cache.ProvisionException;\r
+import org.simantics.utils.ui.ErrorLogger;\r
+\r
+/**\r
+ * @author Antti Villberg\r
+ */\r
+public class AdapterImagerFactory implements ImagerFactory {\r
+\r
+    final private Transformer transformer;\r
+    final private HashMap<String, Image> constants;\r
+    final private String resourceColumn;\r
+\r
+    public AdapterImagerFactory() {\r
+        this(null);\r
+    }\r
+\r
+    public AdapterImagerFactory(Transformer transformer) {\r
+        this(transformer, null, new HashMap<String, Image>(4));\r
+    }\r
+\r
+    public AdapterImagerFactory(Transformer transformer, String resourceColumn, HashMap<String, Image> constants) {\r
+        this.transformer = transformer;\r
+        this.constants = constants;\r
+        this.resourceColumn = resourceColumn;\r
+    }\r
+\r
+    private Resource getInputResource(final NodeContext context) {\r
+        Object input = context.getConstant(BuiltinKeys.INPUT);\r
+        if (transformer != null) {\r
+            input = transformer.transform(input);\r
+        }\r
+\r
+        if (input instanceof Resource)\r
+            return (Resource) context.getConstant(BuiltinKeys.INPUT);\r
+        else if (input instanceof IAdaptable)\r
+            return (Resource) (((IAdaptable) input).getAdapter(Resource.class));\r
+        return null;\r
+    }\r
+\r
+\r
+    @SuppressWarnings("unchecked")\r
+    @Override\r
+    public Imager create(final PrimitiveQueryUpdater updater, final NodeContext context, final ImagerKey key) {\r
+        @SuppressWarnings("rawtypes")\r
+        final ContainerImager result = new ContainerImager((HashMap<String, Image>) constants.clone());\r
+        result.setImage(resourceColumn, MissingImageDescriptor.getInstance());\r
+\r
+        final DataSource<ReadGraph> source = updater.getDataSource(ReadGraph.class);\r
+        final Resource inputResource = getInputResource(context);\r
+\r
+        source.schedule(graph -> {\r
+            ImageDescriptor image = null;\r
+            Collection<Resource> types = Collections.emptyList();\r
+            try {\r
+                types = graph.getPrincipalTypes(inputResource);\r
+//                System.out.println("AdapterImagerFactory " + source.adapt(inputResource, String.class));\r
+//                for(Resource r : types) {\r
+//                    System.out.println("-" + source.adapt(r, String.class));\r
+//                }\r
+            } catch (DatabaseException e1) {\r
+                ErrorLogger.defaultLogError(e1);\r
+            }\r
+            for (Resource type : types) {\r
+                try {\r
+                    ImageDescriptorProvider provider = graph.getPossibleAdapter(type, ImageDescriptorProvider.class);\r
+                    if (provider != null) {\r
+                        try {\r
+                            ImageDescriptor descriptor = provider.get();\r
+                            if (descriptor != null)\r
+                                image = descriptor;\r
+                            //System.out.println("*" + image);\r
+                        } catch (ProvisionException pe) {\r
+                            ErrorLogger.defaultLogError(pe);\r
+                        }\r
+                        break;\r
+                    }\r
+                } catch (DatabaseException e) {\r
+                    ErrorLogger.defaultLogError(e);\r
+                }\r
+            }\r
+\r
+            result.setImage(resourceColumn, image);\r
+            updater.scheduleReplace(context, key, result);\r
+        });\r
+\r
+        return result;\r
+    }\r
+\r
+}\r