package org.simantics.district.network.ui.adapters;

import java.awt.geom.AffineTransform;

import org.simantics.db.AsyncReadGraph;
import org.simantics.db.ReadGraph;
import org.simantics.db.Resource;
import org.simantics.db.exception.DatabaseException;
import org.simantics.db.procedure.AsyncProcedure;
import org.simantics.diagram.adapter.SyncElementFactory;
import org.simantics.diagram.stubs.DiagramResource;
import org.simantics.district.network.ontology.DistrictNetworkResource;
import org.simantics.g2d.canvas.ICanvasContext;
import org.simantics.g2d.diagram.IDiagram;
import org.simantics.g2d.element.ElementClass;
import org.simantics.g2d.element.ElementUtils;
import org.simantics.g2d.element.IElement;
import org.simantics.g2d.element.handler.impl.StaticObjectAdapter;

public class DistrictNetworkVertexElementFactory extends SyncElementFactory {

    public static final ElementClass CLASS = DistrictNetworkVertexElement.CLASS;
    private DistrictNetworkResource DN;
    private DiagramResource DIA;

    public DistrictNetworkVertexElementFactory(ReadGraph graph) {
        this.DN = DistrictNetworkResource.getInstance(graph);
        this.DIA = DiagramResource.getInstance(graph);
    }

    @Override
    public void create(AsyncReadGraph graph, ICanvasContext canvas, IDiagram diagram, Resource elementType, AsyncProcedure<ElementClass> procedure) {
        procedure.execute(graph, CLASS.newClassWith(false, new StaticObjectAdapter(elementType)));
    }

    @Override
    protected Resource getElementClassBaseType(AsyncReadGraph graph) {
        return DN.Vertex;
    }

    @Override
    public void load(ReadGraph graph, ICanvasContext canvas, IDiagram diagram, Resource vertexResource, IElement element) throws DatabaseException {
        
        double[] coords = graph.getRelatedValue(vertexResource, DIA.HasLocation);
        DistrictNetworkVertex vertex = new DistrictNetworkVertex(coords);
        element.setHint(DistrictNetworkVertexElement.KEY_DN_VERTEX, vertex);
        
        // set element transform based on layer SRS and coords
        AffineTransform at = new AffineTransform(1, 0, 0, 1, coords[0], coords[1]);
        ElementUtils.setTransform(element, at);
    }
}