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

import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.geom.Line2D;
import java.awt.geom.Point2D;
import java.awt.geom.Rectangle2D;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import org.simantics.Simantics;
import org.simantics.db.ReadGraph;
import org.simantics.db.Resource;
import org.simantics.db.common.procedure.adapter.TransientCacheListener;
import org.simantics.db.common.request.ResourceRead;
import org.simantics.db.exception.DatabaseException;
import org.simantics.db.layer0.util.Layer0Utils;
import org.simantics.diagram.profile.StyleBase;
import org.simantics.diagram.stubs.DiagramResource;
import org.simantics.district.network.ontology.DistrictNetworkResource;
import org.simantics.layer0.Layer0;
import org.simantics.modeling.ModelingResources;
import org.simantics.scenegraph.INode;
import org.simantics.scenegraph.g2d.G2DNode;
import org.simantics.scenegraph.profile.EvaluationContext;
import org.simantics.scenegraph.profile.common.ProfileVariables;
import org.simantics.scenegraph.utils.GeometryUtils;
import org.simantics.structural.stubs.StructuralResource2;

public class ConnectionLineStyle extends StyleBase<List<Point2D>> {
	
	public static class ConnectionLineNode extends G2DNode {
		private static final BasicStroke STROKE = new BasicStroke(1.f, BasicStroke.CAP_BUTT, BasicStroke.JOIN_MITER, 1.f, new float[] {4.f, 2.f}, 0.f);
		private static final Color[] colors = { Color.RED, Color.GREEN, Color.BLUE, Color.ORANGE, Color.CYAN, Color.PINK };
		
		private float strokeWidth;
		private Line2D[] lines;

		public ConnectionLineNode() {
			super();
		}
		
		private static final long serialVersionUID = 1L;

		@Override
		public Rectangle2D getBoundsInLocal() {
			return null;
		}

		@Override
		public Rectangle2D getBoundsInLocal(boolean b) {
			return null;
		}

		@Override
		public Rectangle2D getBounds() {
			return null;
		}
		
		public void setStrokeWidth(float w) {
			strokeWidth = w;
		}
		
		public void setPoints(List<Point2D> result) {
			Point2D p0 = DistrictNetworkNodeUtils.calculatePoint2D(result.get(0), null);
			lines = new Line2D[result.size() - 1];
			for (int i = 1; i < result.size(); i++) 
			{
				Point2D p = result.get(i);
				lines[i-1] = p != null ? new Line2D.Double(p0, DistrictNetworkNodeUtils.calculatePoint2D(p, null)) : null;
			}
		}
		
		@Override
		public void render(Graphics2D g2d) {
			if (lines == null || lines.length == 0)
				return;
			
			// Keep fixed line width on screen
			float scaleRecip = (float) GeometryUtils.getScale(g2d.getTransform());
			g2d.setStroke(GeometryUtils.scaleStroke(STROKE, strokeWidth / scaleRecip));
			
			for (int i = 0; i < lines.length; i++) {
				if (lines[i] != null) {
					g2d.setColor(colors[i % colors.length]);
					g2d.draw(lines[i]);
				}
			}
		}
	}
	
	@Override
	public List<Point2D> calculateStyle(ReadGraph graph, Resource runtimeDiagram, Resource entry, Resource groupItem)
			throws DatabaseException {
		return graph.syncRequest(new ElementConnectionRequest(groupItem), TransientCacheListener.instance());
	}
	
	@Override
	public void applyStyleForNode(EvaluationContext observer, INode parent, List<Point2D> result) {
		if (result == null || result.size() < 2) {
			ProfileVariables.denyChild(parent, "*", "districtNetworkConnection");
			return;
		}
		
		ConnectionLineNode node = ProfileVariables.claimChild(parent, "*", "districtNetworkConnection", ConnectionLineNode.class, observer);
		if (node == null)
			return;
		
		node.setPoints(result);
		node.setZIndex(0);
		node.setStrokeWidth(2.f);
	}

	@Override
	protected void cleanupStyleForNode(EvaluationContext evaluationContext, INode parent) {
		ProfileVariables.denyChild(parent, "*", "districtNetworkConnection");
	}
	
	private static final class ElementConnectionRequest extends ResourceRead<List<Point2D>> {
		private ElementConnectionRequest(Resource resource) {
			super(resource);
		}
	
		@Override
		public List<Point2D> perform(ReadGraph graph) throws DatabaseException {
			DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
			ModelingResources MOD = ModelingResources.getInstance(graph);
			StructuralResource2 STR = StructuralResource2.getInstance(graph);
			
			Resource vertex = resource;
			if (!graph.isInstanceOf(vertex, DN.Vertex))
				return Collections.emptyList();
			
			double[] coords = graph.getRelatedValue(vertex, DiagramResource.getInstance(graph).HasLocation);
			
			Resource element = graph.getPossibleObject(vertex, DN.MappedComponent);
			if (element == null)
				return Collections.emptyList();
			Resource component = graph.getPossibleObject(element, MOD.ElementToComponent);
			if (component == null)
				return Collections.emptyList();
			Resource componentType = graph.getPossibleType(component, STR.Component);
			if (componentType == null)
				return Collections.emptyList();
			
			Resource actionsModule = Layer0Utils.getPossibleChild(graph, componentType, "Actions");
			if (actionsModule == null || !graph.isInstanceOf(actionsModule, Layer0.getInstance(graph).SCLModule))
				return Collections.emptyList();
			
			List<Resource> components;
			try {
				components = Simantics.applySCL(graph.getURI(actionsModule), "getConnectedComponents", graph, component);
			}
			catch (DatabaseException e) {
				return Collections.emptyList();
			}
			
			if (components == null || components.isEmpty())
				return Collections.emptyList();
			
			List<Point2D> result = new ArrayList<>(components.size() + 1);
			result.add(new Point2D.Double(coords[0], coords[1]));
			for (Resource comp : components) {
				Resource e = comp != null ? graph.getPossibleObject(comp, MOD.ComponentToElement) : null;
				Resource mappingElement = e != null ? graph.getPossibleObject(e, DN.MappedFromElement) : null;
				if (mappingElement != null) {
					double[] coords2 = graph.getRelatedValue(mappingElement, DiagramResource.getInstance(graph).HasLocation);
					result.add(new Point2D.Double(coords2[0], coords2[1]));
				}
				else {
					result.add(null);
				}
			}
			
			return result;
		}
	}
}