bundles/org.simantics.g2d/src/org/simantics/g2d/diagram/handler/impl/DataElementMapImpl.java

   1 /*******************************************************************************
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management
   3  * in Industry THTH ry.
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Eclipse Public License v1.0
   6  * which accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *     VTT Technical Research Centre of Finland - initial API and implementation
  11  *******************************************************************************/
  12 package org.simantics.g2d.diagram.handler.impl;
  13
  14 import org.simantics.g2d.diagram.IDiagram;
  15 import org.simantics.g2d.diagram.handler.DataElementMap;
  16 import org.simantics.g2d.element.ElementHints;
  17 import org.simantics.g2d.element.ElementUtils;
  18 import org.simantics.g2d.element.IElement;
  19
  20 /**
  21  * @author Toni Kalajainen
  22  */
  23 public class DataElementMapImpl implements DataElementMap {
  24
  25         @Override
  26         public Object getData(IDiagram d, IElement element) {
  27                 assert(ElementUtils.getDiagram(element) == d);
  28                 return element.getHint(ElementHints.KEY_OBJECT);
  29         }
  30
  31         @Override
  32         public IElement getElement(IDiagram d, Object data) {
  33                 for (IElement e : d.getElements())
  34                 {
  35                         Object ed = e.getHint(ElementHints.KEY_OBJECT);
  36                         if (ed!=null && ed.equals(data))
  37                                 return e;
  38                 }
  39                 return null;
  40         }
  41
  42
  43 }