gerrit.simantics Code Review - simantics/sysdyn.git/blob

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2012 Association for Decentralized Information Management in\r
   3  * 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.sysdyn.ui.browser.imagerules;\r
  13 \r
  14 import java.util.Collections;\r
  15 import java.util.Map;\r
  16 \r
  17 import org.eclipse.jface.resource.ImageDescriptor;\r
  18 import org.simantics.browsing.ui.common.ColumnKeys;\r
  19 import org.simantics.browsing.ui.model.images.ImageRule;\r
  20 import org.simantics.db.ReadGraph;\r
  21 import org.simantics.db.Resource;\r
  22 import org.simantics.db.exception.DatabaseException;\r
  23 import org.simantics.db.layer0.variable.Variable;\r
  24 import org.simantics.db.layer0.variable.Variables;\r
  25 import org.simantics.layer0.Layer0;\r
  26 import org.simantics.sysdyn.SysdynResource;\r
  27 import org.simantics.sysdyn.ui.Activator;\r
  28 import org.simantics.ui.utils.AdaptionUtils;\r
  29 \r
  30 public class VariableImageRule implements ImageRule {\r
  31 \r
  32     @Override\r
  33     public boolean isCompatible(Class<?> contentType) {\r
  34         return Variable.class.equals(contentType);\r
  35     }\r
  36 \r
  37     @Override\r
  38     public Map<String, ImageDescriptor> getImage(ReadGraph graph, Object content) throws DatabaseException {\r
  39         Variable var = AdaptionUtils.adaptToSingle(content, Variable.class);\r
  40 \r
  41         String image =  "icons/brick.png";\r
  42         \r
  43         Resource r =  (Resource)var.getPropertyValue(graph, Variables.REPRESENTS);\r
  44         if(r != null) {\r
  45             Layer0 L0 = Layer0.getInstance(graph);\r
  46             SysdynResource sr = SysdynResource.getInstance(graph);\r
  47             Resource type = graph.getSingleObject(r, L0.InstanceOf);\r
  48             if(graph.isInheritedFrom(type, sr.Module)) {\r
  49                 image = "icons/bricks.png";\r
  50             }\r
  51         }\r
  52 \r
  53         return Collections.singletonMap(ColumnKeys.SINGLE, \r
  54                 ImageDescriptor.createFromURL(Activator.getDefault().getBundle().getResource(image))\r
  55                 );\r
  56     }\r
  57 \r
  58 }\r