bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/ElementFactoryUtil.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.diagram.adapter;
  13
  14 import java.awt.geom.AffineTransform;
  15
  16 import org.simantics.db.AsyncReadGraph;
  17 import org.simantics.db.ReadGraph;
  18 import org.simantics.db.Resource;
  19 import org.simantics.db.common.procedure.adapter.AsyncProcedureAdapter;
  20 import org.simantics.db.exception.DatabaseException;
  21 import org.simantics.db.procedure.AsyncProcedure;
  22 import org.simantics.diagram.stubs.DiagramResource;
  23 import org.simantics.diagram.synchronization.ErrorHandler;
  24 import org.simantics.diagram.synchronization.ISynchronizationContext;
  25 import org.simantics.diagram.synchronization.SynchronizationHints;
  26 import org.simantics.diagram.synchronization.graph.DiagramGraphUtil;
  27 import org.simantics.diagram.synchronization.graph.GraphSynchronizationHints;
  28 import org.simantics.diagram.synchronization.graph.layer.GraphLayerManager;
  29 import org.simantics.g2d.diagram.DiagramHints;
  30 import org.simantics.g2d.diagram.IDiagram;
  31 import org.simantics.g2d.element.ElementUtils;
  32 import org.simantics.g2d.element.IElement;
  33 import org.simantics.g2d.layers.ILayersEditor;
  34 import org.simantics.utils.datastructures.hints.IHintContext.Key;
  35 import org.slf4j.Logger;
  36 import org.slf4j.LoggerFactory;
  37
  38 public class ElementFactoryUtil {
  39
  40     private static final Logger LOGGER = LoggerFactory.getLogger(ElementFactoryUtil.class);
  41
  42     public static void readTransform(AsyncReadGraph graph, final Resource resource, final IElement e) {
  43         readTransform(graph, resource, e, new AsyncProcedureAdapter<IElement>() {
  44             @Override
  45             public void exception(AsyncReadGraph graph, Throwable t) {
  46                 LOGGER.error("Error reading transform from " + resource + " into element " + e, t);
  47             }
  48         });
  49     }
  50
  51     public static void readTransform(AsyncReadGraph graph, final Resource resource, final IElement e, final AsyncProcedure<IElement> procedure) {
  52         DiagramResource DIA = graph.getService(DiagramResource.class);
  53         try {
  54             AffineTransform tr = DiagramGraphUtil.getDynamicAffineTransform((ReadGraph)graph, null, resource, DIA.HasDynamicTransform, true);
  55             ElementUtils.setTransform(e, tr);
  56         } catch (DatabaseException e1) {
  57             ElementUtils.setTransform(e, new AffineTransform());
  58         }
  59         procedure.execute(graph, e);
  60     }
  61
  62     public static ISynchronizationContext getContext(IDiagram diagram) {
  63         return diagram.getHint(SynchronizationHints.CONTEXT);
  64     }
  65
  66     public static ISynchronizationContext getContextChecked(IDiagram diagram) {
  67         ISynchronizationContext context = getContext(diagram);
  68         if (context == null)
  69             throw new IllegalStateException("diagram does not contain " + SynchronizationHints.CONTEXT + " hint");
  70         return context;
  71     }
  72
  73     public static ErrorHandler getErrorHandler(IDiagram diagram) {
  74         ISynchronizationContext context = getContext(diagram);
  75         if (context == null)
  76             return null;
  77         return context.get(SynchronizationHints.ERROR_HANDLER);
  78     }
  79
  80     public static ErrorHandler getErrorHandlerChecked(IDiagram diagram) {
  81         ISynchronizationContext context = getContextChecked(diagram);
  82         ErrorHandler errorHandler = context.get(SynchronizationHints.ERROR_HANDLER);
  83         if (errorHandler == null)
  84             throw new IllegalStateException("synchronization context " + context + " does not contain " + SynchronizationHints.ERROR_HANDLER + " hint");
  85         return errorHandler;
  86     }
  87
  88     public static <T> T getContextHint(IDiagram diagram, Key key) {
  89         ISynchronizationContext context = getContext(diagram);
  90         if (context == null)
  91             return null;
  92         return context.get(key);
  93     }
  94
  95     public static <T> T getContextHintChecked(IDiagram diagram, Key key) {
  96         return getContextChecked(diagram).get(key);
  97     }
  98
  99     public static void loadLayersForElement(AsyncReadGraph graph, IDiagram diagram, IElement e, Resource element,
 100             AsyncProcedure<IElement> callback) {
 101         final GraphLayerManager layerManager = getContextHint(diagram, GraphSynchronizationHints.GRAPH_LAYER_MANAGER);
 102         if (layerManager != null)
 103             loadLayersForElement(graph, layerManager, diagram, e, element, callback);
 104         else
 105             callback.execute(graph, e);
 106     }
 107
 108     public static void loadLayersForElement(AsyncReadGraph graph, GraphLayerManager layerManager, IDiagram diagram, IElement e, Resource element,
 109             AsyncProcedure<IElement> callback) {
 110         final ILayersEditor layers = diagram.getHint(DiagramHints.KEY_LAYERS_EDITOR);
 111         if (layers != null)
 112             layerManager.loadLayersForElement(graph, layers, e, element, callback);
 113         else
 114             callback.execute(graph, e);
 115     }
 116
 117 }