bundles/org.simantics.browsing.ui.common/src/org/simantics/browsing/ui/common/imagers/ContainerImager.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.common.imagers;\r
  13 \r
  14 import java.util.HashMap;\r
  15 import java.util.Map;\r
  16 \r
  17 import org.simantics.browsing.ui.content.Imager;\r
  18 \r
  19 /**\r
  20  * @author Antti Villberg\r
  21  * \r
  22  */\r
  23 public class ContainerImager<Image> implements Imager {\r
  24 \r
  25     private final Map<String, Image> constants;\r
  26 \r
  27     private Image                    theImage;\r
  28 \r
  29     public ContainerImager(Map<String, Image> constants, Image image) {\r
  30         this.constants = constants;\r
  31         this.theImage = image;\r
  32     }\r
  33 \r
  34     public ContainerImager(Map<String, Image> constants) {\r
  35         this(constants, null);\r
  36     }\r
  37 \r
  38     public ContainerImager(Image image) {\r
  39         this(new HashMap<String, Image>(), image);\r
  40     }\r
  41 \r
  42     public ContainerImager() {\r
  43         this(new HashMap<String, Image>(), null);\r
  44     }\r
  45 \r
  46 \r
  47     public void setImage(String key, Image image) {\r
  48         if (key == null)\r
  49             theImage = image;\r
  50         else\r
  51             constants.put(key, image);\r
  52     }\r
  53 \r
  54     public void setImage(Image image) {\r
  55         theImage = image;\r
  56     }\r
  57 \r
  58     @SuppressWarnings("unchecked")\r
  59         @Override\r
  60     public Image getImage(String key) {\r
  61         if (theImage != null)\r
  62             return theImage;\r
  63         else\r
  64             return constants.get(key);\r
  65     }\r
  66 \r
  67 }\r