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

import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.RenderingHints;
import java.awt.Stroke;
import java.awt.geom.AffineTransform;
import java.awt.geom.Point2D;
import java.awt.geom.Rectangle2D;

import org.simantics.district.network.ModelledCRS;
import org.simantics.district.network.ui.adapters.DistrictNetworkVertex;
import org.simantics.scenegraph.ParentNode;
import org.simantics.scenegraph.g2d.G2DNode;
import org.simantics.scenegraph.utils.GeometryUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class DistrictNetworkVertexNode extends G2DNode {

    private static final Logger LOGGER = LoggerFactory.getLogger(DistrictNetworkVertexNode.class);
    
    private static final long serialVersionUID = -2641639101400236719L;
    private DistrictNetworkVertex vertex;

    private static final double left = -0.25;
    private static final double top = -0.25;
    private static final double width = 0.5;
    private static final double height = 0.5;
    
    private static final Rectangle2D NORMAL = new Rectangle2D.Double(left, top, width, height);
    private static final Rectangle2D HOVERED = new Rectangle2D.Double(left * 3, top * 3, width * 3, height * 3);
    
    private Stroke stroke = new BasicStroke(2);
    private boolean scaleStroke = true;
    private boolean hover;

    private Color color;

    private Rectangle2D bounds;

    private Double strokee;

    @Override
    public void init() {
        setZIndex(2);
    }
    
    @Override
    public void render(Graphics2D g2d) {
        AffineTransform ot = null;
        AffineTransform t = getTransform();
        if (t != null && !t.isIdentity()) {
            ot = g2d.getTransform();
            g2d.transform(getTransform());
        }
        
        Object aaHint = g2d.getRenderingHint(RenderingHints.KEY_ANTIALIASING);
        g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_OFF);
        
        Color oldColor = g2d.getColor();
        Stroke oldStroke = g2d.getStroke();
        
        g2d.setColor(color);
        
        double scaleRecip = 1;
        if (scaleStroke) {
            double scale = GeometryUtils.getScale(g2d.getTransform());
            double str;
            if (strokee != null)
                str = strokee;
            else
                str = 1.0;
            //System.out.println("scale: " + scale);
            scaleRecip = 1.0 / scale;
        }
        scaleRecip = 8.0 * scaleRecip;
        
        // Translate lat and lon to X and Y
        Point2D res = calculatePoint2D(vertex);
        
        Rectangle2D toDraw;
        if (hover) {
            toDraw = new Rectangle2D.Double(res.getX() - (HOVERED.getWidth() / 2 * scaleRecip), res.getY() - (HOVERED.getHeight() / 2 * scaleRecip), HOVERED.getWidth() * scaleRecip, HOVERED.getHeight() * scaleRecip);
        } else {
            toDraw = new Rectangle2D.Double(res.getX() - (NORMAL.getWidth() / 2 * scaleRecip), res.getY() - (NORMAL.getHeight() / 2 * scaleRecip), NORMAL.getWidth() * scaleRecip, NORMAL.getHeight() * scaleRecip);
        }
        // render
        g2d.fill(toDraw);
        
        // Reset stats
        g2d.setColor(oldColor);
        g2d.setStroke(oldStroke);
        
        g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, aaHint);
        
        if (ot != null)
            g2d.setTransform(ot);
    }

    @Override
    public Rectangle2D getBounds() {
        return super.getBounds();
    }
    
    @Override
    public Rectangle2D getBoundsInLocal() {
        return bounds;
    }
    
    private void updateBounds() {
        Rectangle2D oldBounds = bounds;
        if (oldBounds == null)
            oldBounds = new Rectangle2D.Double();
        bounds = calculateBounds(oldBounds);
    }

    @Override
    public void setTransform(AffineTransform transform) {
        super.setTransform(transform);
        // Update bounds
        updateBounds();
    }
    
    @Override
    public AffineTransform getTransform() {
        return super.getTransform();
    }
    
    private Rectangle2D calculateBounds(Rectangle2D rect) {
        Point2D calcPoint = calculatePoint2D(vertex);
        AffineTransform at = getTransform();
        return new Rectangle2D.Double(calcPoint.getX(), calcPoint.getY(), 1 / at.getScaleX(), 1 / at.getScaleY()).getBounds2D();
    }

    private static Point2D calculatePoint2D(DistrictNetworkVertex vertex) {
        Point2D point= vertex.getPoint();
        double x = ModelledCRS.longitudeToX(point.getX());
        double y = ModelledCRS.latitudeToY(-point.getY()); // Inverse because Simantics Diagram is inverted

        // Apply the scaling
        Point2D res = new Point2D.Double(x, y);
        return res;
    }

    public void setVertex(DistrictNetworkVertex vertex) {
        this.vertex = vertex;
        updateBounds();
    }
    
    public boolean hover(boolean hover) {
//        if (hover && LOGGER.isDebugEnabled())
//            LOGGER.debug("Hovering " + this);
        boolean changed = false;
        if (this.hover != hover) {
            this.hover = hover;
            changed = true;
        }
        return changed;
    }

    public void setColor(Color color) {
        this.color = color;
    }
    
    public Color getColor() {
        return color;
    }

    @PropertySetter(value = "stroke")
    public void setStroke(Double stroke) {
        this.strokee = stroke / 10;
    }
}