/*******************************************************************************
 * Copyright (c) 2007, 2010 Association for Decentralized Information Management
 * in Industry THTH ry.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     VTT Technical Research Centre of Finland - initial API and implementation
 *******************************************************************************/
package org.simantics.diagram.adapter;

import java.awt.geom.AffineTransform;

import org.simantics.databoard.Bindings;
import org.simantics.db.AsyncReadGraph;
import org.simantics.db.Resource;
import org.simantics.db.common.procedure.adapter.AsyncProcedureAdapter;
import org.simantics.db.procedure.AsyncProcedure;
import org.simantics.diagram.stubs.G2DResource;
import org.simantics.diagram.synchronization.ErrorHandler;
import org.simantics.diagram.synchronization.ISynchronizationContext;
import org.simantics.diagram.synchronization.SynchronizationHints;
import org.simantics.diagram.synchronization.graph.DiagramGraphUtil;
import org.simantics.diagram.synchronization.graph.GraphSynchronizationHints;
import org.simantics.diagram.synchronization.graph.layer.GraphLayerManager;
import org.simantics.g2d.diagram.DiagramHints;
import org.simantics.g2d.diagram.IDiagram;
import org.simantics.g2d.element.ElementUtils;
import org.simantics.g2d.element.IElement;
import org.simantics.g2d.layers.ILayersEditor;
import org.simantics.utils.datastructures.hints.IHintContext.Key;

public class ElementFactoryUtil {

    public static void readTransform(AsyncReadGraph graph, final Resource resource, final IElement e) {
        readTransform(graph, resource, e, new AsyncProcedureAdapter<IElement>() {
            @Override
            public void exception(AsyncReadGraph graph, Throwable t) {
                // Nowhere to log properly.
                t.printStackTrace();
            }
        });
    }

    public static void readTransform(AsyncReadGraph graph, final Resource resource, final IElement e, final AsyncProcedure<IElement> procedure) {
        G2DResource G2D = graph.getService(G2DResource.class);
        graph.forPossibleRelatedValue(resource, G2D.HasTransform, Bindings.DOUBLE_ARRAY, new AsyncProcedure<double[]>() {
            @Override
            public void exception(AsyncReadGraph graph, Throwable throwable) {
                procedure.exception(graph, throwable);
            }

            @Override
            public void execute(AsyncReadGraph graph, double[] mat) {
                mat = DiagramGraphUtil.validateAffineTransform(resource, mat);
                AffineTransform tr = mat != null ? new AffineTransform(mat) : new AffineTransform();
                ElementUtils.setTransform(e, tr);
                procedure.execute(graph, e);
            }
        });
    }

    public static ISynchronizationContext getContext(IDiagram diagram) {
        return diagram.getHint(SynchronizationHints.CONTEXT);
    }

    public static ISynchronizationContext getContextChecked(IDiagram diagram) {
        ISynchronizationContext context = getContext(diagram);
        if (context == null)
            throw new IllegalStateException("diagram does not contain " + SynchronizationHints.CONTEXT + " hint");
        return context;
    }

    public static ErrorHandler getErrorHandler(IDiagram diagram) {
        ISynchronizationContext context = getContext(diagram);
        if (context == null)
            return null;
        return context.get(SynchronizationHints.ERROR_HANDLER);
    }

    public static ErrorHandler getErrorHandlerChecked(IDiagram diagram) {
        ISynchronizationContext context = getContextChecked(diagram);
        ErrorHandler errorHandler = context.get(SynchronizationHints.ERROR_HANDLER);
        if (errorHandler == null)
            throw new IllegalStateException("synchronization context " + context + " does not contain " + SynchronizationHints.ERROR_HANDLER + " hint");
        return errorHandler;
    }

    public static <T> T getContextHint(IDiagram diagram, Key key) {
        ISynchronizationContext context = getContext(diagram);
        if (context == null)
            return null;
        return context.get(key);
    }

    public static <T> T getContextHintChecked(IDiagram diagram, Key key) {
        return getContextChecked(diagram).get(key);
    }

    public static void loadLayersForElement(AsyncReadGraph graph, IDiagram diagram, IElement e, Resource element,
            AsyncProcedure<IElement> callback) {
        final GraphLayerManager layerManager = getContextHint(diagram, GraphSynchronizationHints.GRAPH_LAYER_MANAGER);
        if (layerManager != null)
            loadLayersForElement(graph, layerManager, diagram, e, element, callback);
        else
            callback.execute(graph, e);
    }

    public static void loadLayersForElement(AsyncReadGraph graph, GraphLayerManager layerManager, IDiagram diagram, IElement e, Resource element,
            AsyncProcedure<IElement> callback) {
        final ILayersEditor layers = diagram.getHint(DiagramHints.KEY_LAYERS_EDITOR);
        if (layers != null)
            layerManager.loadLayersForElement(graph, layers, e, element, callback);
        else
            callback.execute(graph, e);
    }

}