package com.c2kernel.lifecycle.gui.view;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Shape;
import java.awt.geom.AffineTransform;
import java.awt.geom.GeneralPath;

import com.c2kernel.graph.model.DirectedEdge;
import com.c2kernel.graph.model.GraphPoint;
import com.c2kernel.graph.view.DirectedEdgeRenderer;
import com.c2kernel.lifecycle.NextDef;
public class WfDirectedEdgeDefRenderer implements DirectedEdgeRenderer
{
	private GeneralPath mArrowTemplate = new GeneralPath();
	public WfDirectedEdgeDefRenderer()
	{
		mArrowTemplate.moveTo(-5, 5);
		mArrowTemplate.lineTo(0, 0);
		mArrowTemplate.lineTo(5, 5);
	}
	@Override
	public void draw(Graphics2D g2d, DirectedEdge directedEdge)
	{
		GraphPoint originPoint = directedEdge.getOriginPoint();
		GraphPoint terminusPoint = directedEdge.getTerminusPoint();
		GraphPoint midPoint = new GraphPoint();
		AffineTransform transform = new AffineTransform();
		Shape arrow = null;
		NextDef nextDef = (NextDef) directedEdge;
		boolean hasError = !nextDef.verify();
		String text = (String) nextDef.getProperties().get("Alias");
		g2d.setPaint(hasError ? Color.red : Color.black);
		if (("Broken +".equals(nextDef.getProperties().get("Type"))))
		{
			g2d.drawLine(originPoint.x, originPoint.y, originPoint.x, (originPoint.y + terminusPoint.y) / 2);
			g2d.drawLine(originPoint.x, (originPoint.y + terminusPoint.y) / 2, terminusPoint.x, (originPoint.y + terminusPoint.y) / 2);
			g2d.drawLine(terminusPoint.x, (originPoint.y + terminusPoint.y) / 2, terminusPoint.x, terminusPoint.y);
			midPoint.x = (originPoint.x + terminusPoint.x) / 2;
			midPoint.y = (originPoint.y + terminusPoint.y) / 2;
			transform.translate(midPoint.x, midPoint.y);
			transform.rotate(
				calcArrowAngle(
					originPoint.x,
					originPoint.x - terminusPoint.x > -5
						&& originPoint.x - terminusPoint.x < 5 ? originPoint.y : (originPoint.y + terminusPoint.y) / 2,
					terminusPoint.x,
					originPoint.x - terminusPoint.x > -5
						&& originPoint.x - terminusPoint.x < 5 ? terminusPoint.y : (originPoint.y + terminusPoint.y) / 2));
		}
		else if (("Broken -".equals(nextDef.getProperties().get("Type"))))
		{
			g2d.drawLine(originPoint.x, originPoint.y, terminusPoint.x, originPoint.y);
			g2d.drawLine(terminusPoint.x, originPoint.y, terminusPoint.x, terminusPoint.y);
			boolean arrowOnY = !(originPoint.y - terminusPoint.y < 60 && originPoint.y - terminusPoint.y > -60);
			midPoint.x = arrowOnY ? terminusPoint.x : (originPoint.x + terminusPoint.x) / 2;
			midPoint.y = arrowOnY ? (originPoint.y + terminusPoint.y) / 2 : originPoint.y;
			transform.translate(midPoint.x, midPoint.y);
			transform
				.rotate(
					calcArrowAngle(
						arrowOnY ? terminusPoint.x : originPoint.x,
						arrowOnY ? originPoint.y : originPoint.y,
						arrowOnY ? terminusPoint.x : terminusPoint.x,
						arrowOnY ? terminusPoint.y : originPoint.y));
		}
		else if (("Broken |".equals(nextDef.getProperties().get("Type"))))
		{
			g2d.drawLine(originPoint.x, originPoint.y, originPoint.x, terminusPoint.y);
			g2d.drawLine(originPoint.x, terminusPoint.y, terminusPoint.x, terminusPoint.y);
			boolean arrowOnY = !(originPoint.y - terminusPoint.y < 60 && originPoint.y - terminusPoint.y > -60);
			midPoint.x = arrowOnY ? originPoint.x : (originPoint.x + terminusPoint.x) / 2;
			midPoint.y = arrowOnY ? (originPoint.y + terminusPoint.y) / 2 : terminusPoint.y;
			transform.translate(midPoint.x, midPoint.y);
			transform
				.rotate(
					calcArrowAngle(
						arrowOnY ? terminusPoint.x : originPoint.x,
						arrowOnY ? originPoint.y : originPoint.y,
						arrowOnY ? terminusPoint.x : terminusPoint.x,
						arrowOnY ? terminusPoint.y : originPoint.y));
		}
		else
		{
			g2d.drawLine(originPoint.x, originPoint.y, terminusPoint.x, terminusPoint.y);
			midPoint.x = originPoint.x + (terminusPoint.x - originPoint.x) / 2;
			midPoint.y = originPoint.y + (terminusPoint.y - originPoint.y) / 2;
			transform.translate(midPoint.x, midPoint.y);
			transform.rotate(calcArrowAngle(originPoint.x, originPoint.y, terminusPoint.x, terminusPoint.y));
		}

		arrow = mArrowTemplate.createTransformedShape(transform);
		g2d.draw(arrow);
		if (text != null)
			g2d.drawString(text, midPoint.x + 10, midPoint.y);
	}
	private static double calcArrowAngle(int originX, int originY, int terminusX, int terminusY)
	{
		double width = terminusX - originX;
		double height = terminusY - originY;
		if ((width == 0) && (height > 0))
		{
			return Math.PI;
		}
		if ((width == 0) && (height < 0))
		{
			return 0;
		}
		if ((width > 0) && (height == 0))
		{
			return Math.PI / 2.0;
		}
		if ((width < 0) && (height == 0))
		{
			return -1.0 * Math.PI / 2.0;
		}
		if ((width > 0) && (height > 0))
		{
			return Math.PI / 2.0 + Math.atan(Math.abs(height) / Math.abs(width));
		}
		if ((width > 0) && (height < 0))
		{
			return Math.atan(Math.abs(width) / Math.abs(height));
		}
		if ((width < 0) && (height < 0))
		{
			return -1.0 * Math.atan(Math.abs(width) / Math.abs(height));
		}
		if ((width < 0) && (height > 0))
		{
			return -1.0 * (Math.PI / 2.0 + Math.atan(Math.abs(height) / Math.abs(width)));
		}
		return 0.0;
	}
}