bundles/org.simantics.g2d/src/org/simantics/g2d/diagram/handler/impl/DataElementMapImpl.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.g2d.diagram.handler.impl;\r
  13 \r
  14 import org.simantics.g2d.diagram.IDiagram;\r
  15 import org.simantics.g2d.diagram.handler.DataElementMap;\r
  16 import org.simantics.g2d.element.ElementHints;\r
  17 import org.simantics.g2d.element.ElementUtils;\r
  18 import org.simantics.g2d.element.IElement;\r
  19 \r
  20 /**\r
  21  * @author Toni Kalajainen\r
  22  */\r
  23 public class DataElementMapImpl implements DataElementMap {\r
  24 \r
  25         @Override\r
  26         public Object getData(IDiagram d, IElement element) {\r
  27                 assert(ElementUtils.getDiagram(element) == d);\r
  28                 return element.getHint(ElementHints.KEY_OBJECT);\r
  29         }\r
  30 \r
  31         @Override\r
  32         public IElement getElement(IDiagram d, Object data) {\r
  33                 for (IElement e : d.getElements())\r
  34                 {\r
  35                         Object ed = e.getHint(ElementHints.KEY_OBJECT); \r
  36                         if (ed!=null && ed.equals(data))\r
  37                                 return e;\r
  38                 }\r
  39                 return null;\r
  40         }\r
  41 \r
  42 \r
  43 }\r