package org.simantics.diagram.function;

import java.awt.geom.AffineTransform;
import java.awt.geom.Point2D;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;

import org.simantics.databoard.Bindings;
import org.simantics.db.ReadGraph;
import org.simantics.db.Resource;
import org.simantics.db.common.request.PossibleIndexRoot;
import org.simantics.db.exception.DatabaseException;
import org.simantics.db.layer0.exception.MissingVariableException;
import org.simantics.db.layer0.request.PossibleConfiguration;
import org.simantics.db.layer0.variable.Variable;
import org.simantics.db.layer0.variable.Variables;
import org.simantics.diagram.content.ElementContext;
import org.simantics.diagram.stubs.DiagramResource;
import org.simantics.diagram.stubs.G2DResource;
import org.simantics.diagram.synchronization.graph.DiagramGraphUtil;
import org.simantics.issues.common.IssueUtils;
import org.simantics.layer0.Layer0;
import org.simantics.modeling.ModelingResources;
import org.simantics.modeling.template2d.ontology.Template2dResource;
import org.simantics.scenegraph.loader.SceneGraphContext;
import org.simantics.scenegraph.loader.ScenegraphLoaderUtils;
import org.simantics.scl.reflection.annotations.SCLValue;


public class All {

    @SCLValue(type = "ReadGraph -> Resource -> Variable -> Variable")
    public static Variable defaultRuntimeVariable(ReadGraph graph, Resource converter, Variable context) throws DatabaseException {

        SceneGraphContext vc = ScenegraphLoaderUtils.getContext(graph, context);
        if(vc == null) return null;

        DiagramResource DIA = DiagramResource.getInstance(graph);
        String uri = graph.getRelatedValue(vc.getRuntime(), DIA.RuntimeDiagram_HasVariable, Bindings.STRING);
        return Variables.getVariable(graph, uri);

    }

    @SCLValue(type = "ReadGraph -> Resource -> Variable -> String")
    public static String referenceText(ReadGraph graph, Resource converter, Variable context) throws DatabaseException {

        //DiagramResource DIA = DiagramResource.getInstance(graph);

        //Resource runtime = ScenegraphLoaderUtils.getRuntime(graph, context);
        String path = context.getParent(graph).getPropertyValue(graph, "reference", Bindings.STRING);

        // diagram uri and diagram variable
//        String diagramURI = graph.getRelatedValue(runtime, DIA.RuntimeDiagram_HasVariable, Bindings.STRING);
//        Variable diagramVariable = org.simantics.db.layer0.variable.Variables.getVariable(graph, diagramURI);

//        Variable property = diagramVariable.browse(graph, path);

        try {
            Variable selection = ScenegraphLoaderUtils.getPossibleVariableSelection(graph, context);
        	PredefinedVariables vars = PredefinedVariables.getInstance();
        	Variable property = vars.getVariable(graph, path, converter, selection);
        	if (property  != null)
        		return property.getValue(graph).toString();
        } catch (MissingVariableException e){
        	return "Invalid path.";
        }
        return null;
    }

    @SCLValue(type = "ReadGraph -> Resource -> ElementContext -> a")
    public static Object modelledTransform(ReadGraph graph, Resource converter, ElementContext context) throws DatabaseException {
        return graph.getRelatedValue(context.element, G2DResource.getInstance(graph).HasTransform, Bindings.DOUBLE_ARRAY);
    }
    
    @SCLValue(type = "ReadGraph -> Resource -> ElementContext -> a")
    public static Object flagTransform(ReadGraph graph, Resource converter, ElementContext context) throws DatabaseException {
		Layer0 L0 = Layer0.getInstance(graph);
		DiagramResource DIA = DiagramResource.getInstance(graph);
        Template2dResource TEMPLATE2D = Template2dResource.getInstance(graph);

		Resource flag = context.element;
		Resource runtimeDiagram = context.runtime;

		Resource diagram = graph.getPossibleObject(runtimeDiagram, DIA.RuntimeDiagram_HasConfiguration);
		if (diagram == null)
			return flagTransformImpl(graph, converter, context);

		Resource template = getTemplateFromDiagram(graph, diagram);
		if (template == null)
			return flagTransformImpl(graph, converter, context);
		
		double gridSize = DiagramGraphUtil.getGridSize(graph, diagram, 0.0);

		String tableName = graph.getPossibleRelatedValue(flag, DIA.Flag_HasIOTableBinding, Bindings.STRING);
		Integer rowIndex = graph.getPossibleRelatedValue(flag, DIA.Flag_HasIOTableRowIndex, Bindings.INTEGER);
//        double[] mat = graph.getRelatedValue(flag, DIA.HasTransform, Bindings.DOUBLE_ARRAY);
        
        
        Collection<Resource> slotTables = graph.getObjects(template, L0.ConsistsOf);
        for (Resource slotTable:slotTables){
        	if (!graph.isInstanceOf(slotTable, TEMPLATE2D.FlagTable))
        		continue;
        	String name = graph.getPossibleRelatedValue(slotTable, L0.HasName, Bindings.STRING);
        	if (!name.equals(tableName))
        		continue;
        	
            double[] transform = graph.getPossibleRelatedValue2(slotTable, DIA.Scenegraph_Composite_transform, Bindings.getBindingUnchecked(double[].class));
        	Resource align = graph.getPossibleObject(slotTable, TEMPLATE2D.FlagTable_HasAlignment);
        	boolean isRightAlignment = false;
        	if (align.equals(TEMPLATE2D.FlagTable_Alignment_Right))
        		isRightAlignment = true;
        	Float width = graph.getPossibleRelatedValue(slotTable, TEMPLATE2D.FlagTable_HasWidth, Bindings.FLOAT);
        	Float rowHeight = graph.getPossibleRelatedValue(slotTable, TEMPLATE2D.FlagTable_HasRowHeigth, Bindings.FLOAT);
        	//Integer rowCount = g.getPossibleRelatedValue(slotTable, TEMPLATE2D.FlagTable_HasRowCount, Bindings.INTEGER);
        	
//			Rectangle2D rowBounds = new Rectangle2D.Double(0, rowHeight*rowIndex, width, rowHeight);
//			rowBounds = GeometryUtils.transformShape(rowBounds, new AffineTransform(transform)).getBounds2D();
//            e.setHint(FlagClass.KEY_FLAG_BOUNDS, rowBounds);
            
            double[] flagTransform = graph.getRelatedValue(flag, DIA.HasTransform, Bindings.DOUBLE_ARRAY);
            return calcTransform(transform, flagTransform, rowHeight, width, isRightAlignment, rowIndex, gridSize);
        }

        return flagTransformImpl(graph, converter, context);
    }

    private static Object flagTransformImpl(ReadGraph graph, Resource converter, ElementContext context) throws DatabaseException {
        double[] mat = graph.getRelatedValue(context.element, G2DResource.getInstance(graph).HasTransform, Bindings.DOUBLE_ARRAY);
        return mat;
    }

	private static double[] calcTransform(double[] tableTransform, double[] flagTransform, Float rowHeight, Float tableWidth, boolean isRightAlignment, Integer flagRowIndex, double gridSize) {
		double scale = tableTransform[3];
		double scaledRowHeigth = scale*rowHeight;

		// move to coordinate system of parent of iotable
		double rowMiddleY = tableTransform[5]+(flagRowIndex+0.5F)*scaledRowHeigth;
		if (gridSize != 0.0) {
			double modulo = rowMiddleY%gridSize;
			int count =(int) (rowMiddleY/gridSize);
			double gridOnRow = (modulo > (gridSize/2))?gridSize*(1+count):gridSize*count;
			double diff = (gridOnRow > rowMiddleY)? gridOnRow - rowMiddleY:rowMiddleY-gridOnRow;
			if (diff < (scaledRowHeigth/2))
				rowMiddleY = gridOnRow; 
		}
		// move back to iotable coordinate system
		rowMiddleY = (rowMiddleY-tableTransform[5])/scale;

		double xx = isRightAlignment ? tableWidth : 0.0; 

		AffineTransform trans = new AffineTransform(new double[]{tableTransform[0],tableTransform[1],tableTransform[2],tableTransform[3],0.0,0.0});
		Point2D point = new Point2D.Double(xx, rowMiddleY);
		trans.transform(point, point);

		double[] copy = Arrays.copyOf(flagTransform, flagTransform.length);
		copy[4] = tableTransform[4] + point.getX();
		copy[5] = tableTransform[5] + point.getY();
		return copy;
	}

    public static Resource getTemplate(ReadGraph graph, Resource runtimeDiagram) throws DatabaseException {
        DiagramResource DIA = DiagramResource.getInstance(graph);
        Resource diagram = graph.getPossibleObject(runtimeDiagram, DIA.RuntimeDiagram_HasConfiguration);
        if (diagram == null)
            return null;
        return getTemplateFromDiagram(graph, diagram);
    }

    public static Resource getTemplateFromDiagram(ReadGraph graph, Resource diagram) throws DatabaseException {
        Template2dResource TEMPLATE2D = Template2dResource.getInstance(graph);
        Resource template = graph.getPossibleObject(diagram, TEMPLATE2D.HasDrawingTemplate);
        return template;
    }

    @SCLValue(type = "ReadGraph -> Resource -> Variable -> String")
    public static String diagramElementIssuePath(ReadGraph graph, Resource converter, Variable property) throws DatabaseException {
        Layer0 L0 = Layer0.getInstance(graph);
        ModelingResources MOD = ModelingResources.getInstance(graph);
        DiagramResource DIA = DiagramResource.getInstance(graph);

        List<Resource> contexts = IssueUtils.getContextsForProperty(graph, property);
        if(contexts.isEmpty()) return "";

        Resource context = contexts.get(0);

        Resource mapped = graph.getPossibleObject(context, MOD.ComponentToElement);
        if(mapped != null) context = mapped;
        mapped = graph.getPossibleObject(context, MOD.ConnectionToDiagramConnection);
        if(mapped != null) context = mapped;

        if(!graph.isInstanceOf(context, DIA.Element)) return "";

        Resource configuration = graph.sync(new PossibleConfiguration(context));
        if(configuration == null) return "";

        Resource diagram = graph.getPossibleObject(context, L0.PartOf);
        if(diagram == null) return "";

        Resource composite = graph.getPossibleObject(diagram, MOD.DiagramToComposite);
        if(composite == null) return "";

        Resource issueRoot = Variables.getPossibleIndexRoot(graph, property);
        if(issueRoot == null) return "";
        Resource contextRoot = graph.sync(new PossibleIndexRoot(composite));
        if(contextRoot == null) return "";
        if(issueRoot.equals(contextRoot)) {
            String uri = graph.getURI(composite);
            String modelURI = graph.getURI(contextRoot);
            return IssueUtils.pathString(uri, modelURI.length()+1);
        } else {
            String uri = graph.getURI(composite);
            String modelURI = graph.getURI(graph.getRootLibrary());
            return IssueUtils.pathString(uri, modelURI.length()+1);
        }
    }

}