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.List;

import org.simantics.g2d.utils.GeometryUtils;
import org.simantics.scenegraph.g2d.G2DNode;

public class ConnectionLineNode extends G2DNode {
    
    public static final String NODE_NAME = "districtNetworkConnection";
    
	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]);
			}
		}
	}
}