bundles/org.simantics.browsing.ui.swt/src/org/simantics/browsing/ui/swt/AdapterImagerFactory.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
   3  * in Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.browsing.ui.swt;\r
  13 \r
  14 import java.util.Collection;\r
  15 import java.util.Collections;\r
  16 import java.util.HashMap;\r
  17 \r
  18 import org.eclipse.core.runtime.IAdaptable;\r
  19 import org.eclipse.jface.resource.ImageDescriptor;\r
  20 import org.eclipse.swt.graphics.Image;\r
  21 import org.simantics.browsing.ui.BuiltinKeys;\r
  22 import org.simantics.browsing.ui.BuiltinKeys.ImagerKey;\r
  23 import org.simantics.browsing.ui.DataSource;\r
  24 import org.simantics.browsing.ui.NodeContext;\r
  25 import org.simantics.browsing.ui.PrimitiveQueryUpdater;\r
  26 import org.simantics.browsing.ui.common.EvaluatorData.Transformer;\r
  27 import org.simantics.browsing.ui.common.imagers.ContainerImager;\r
  28 import org.simantics.browsing.ui.content.Imager;\r
  29 import org.simantics.browsing.ui.content.ImagerFactory;\r
  30 import org.simantics.browsing.ui.graph.impl.MissingImageDescriptor;\r
  31 import org.simantics.db.ReadGraph;\r
  32 import org.simantics.db.Resource;\r
  33 import org.simantics.db.exception.DatabaseException;\r
  34 import org.simantics.ui.icons.ImageDescriptorProvider;\r
  35 import org.simantics.utils.datastructures.cache.ProvisionException;\r
  36 import org.simantics.utils.ui.ErrorLogger;\r
  37 \r
  38 /**\r
  39  * @author Antti Villberg\r
  40  */\r
  41 public class AdapterImagerFactory implements ImagerFactory {\r
  42 \r
  43     final private Transformer transformer;\r
  44     final private HashMap<String, Image> constants;\r
  45     final private String resourceColumn;\r
  46 \r
  47     public AdapterImagerFactory() {\r
  48         this(null);\r
  49     }\r
  50 \r
  51     public AdapterImagerFactory(Transformer transformer) {\r
  52         this(transformer, null, new HashMap<String, Image>(4));\r
  53     }\r
  54 \r
  55     public AdapterImagerFactory(Transformer transformer, String resourceColumn, HashMap<String, Image> constants) {\r
  56         this.transformer = transformer;\r
  57         this.constants = constants;\r
  58         this.resourceColumn = resourceColumn;\r
  59     }\r
  60 \r
  61     private Resource getInputResource(final NodeContext context) {\r
  62         Object input = context.getConstant(BuiltinKeys.INPUT);\r
  63         if (transformer != null) {\r
  64             input = transformer.transform(input);\r
  65         }\r
  66 \r
  67         if (input instanceof Resource)\r
  68             return (Resource) context.getConstant(BuiltinKeys.INPUT);\r
  69         else if (input instanceof IAdaptable)\r
  70             return (Resource) (((IAdaptable) input).getAdapter(Resource.class));\r
  71         return null;\r
  72     }\r
  73 \r
  74 \r
  75     @SuppressWarnings("unchecked")\r
  76     @Override\r
  77     public Imager create(final PrimitiveQueryUpdater updater, final NodeContext context, final ImagerKey key) {\r
  78         @SuppressWarnings("rawtypes")\r
  79         final ContainerImager result = new ContainerImager((HashMap<String, Image>) constants.clone());\r
  80         result.setImage(resourceColumn, MissingImageDescriptor.getInstance());\r
  81 \r
  82         final DataSource<ReadGraph> source = updater.getDataSource(ReadGraph.class);\r
  83         final Resource inputResource = getInputResource(context);\r
  84 \r
  85         source.schedule(graph -> {\r
  86             ImageDescriptor image = null;\r
  87             Collection<Resource> types = Collections.emptyList();\r
  88             try {\r
  89                 types = graph.getPrincipalTypes(inputResource);\r
  90 //                System.out.println("AdapterImagerFactory " + source.adapt(inputResource, String.class));\r
  91 //                for(Resource r : types) {\r
  92 //                    System.out.println("-" + source.adapt(r, String.class));\r
  93 //                }\r
  94             } catch (DatabaseException e1) {\r
  95                 ErrorLogger.defaultLogError(e1);\r
  96             }\r
  97             for (Resource type : types) {\r
  98                 try {\r
  99                     ImageDescriptorProvider provider = graph.getPossibleAdapter(type, ImageDescriptorProvider.class);\r
 100                     if (provider != null) {\r
 101                         try {\r
 102                             ImageDescriptor descriptor = provider.get();\r
 103                             if (descriptor != null)\r
 104                                 image = descriptor;\r
 105                             //System.out.println("*" + image);\r
 106                         } catch (ProvisionException pe) {\r
 107                             ErrorLogger.defaultLogError(pe);\r
 108                         }\r
 109                         break;\r
 110                     }\r
 111                 } catch (DatabaseException e) {\r
 112                     ErrorLogger.defaultLogError(e);\r
 113                 }\r
 114             }\r
 115 \r
 116             result.setImage(resourceColumn, image);\r
 117             updater.scheduleReplace(context, key, result);\r
 118         });\r
 119 \r
 120         return result;\r
 121     }\r
 122 \r
 123 }\r