package com.revolsys.swing.map.overlay;

import java.awt.Color;
import java.awt.Graphics2D;

import com.revolsys.awt.WebColors;
import com.revolsys.geometry.model.GeometryFactory;
import com.revolsys.geometry.model.vertex.Vertex;
import com.revolsys.swing.map.Viewport2D;
import com.revolsys.swing.map.layer.record.renderer.MarkerStyleRenderer;
import com.revolsys.swing.map.layer.record.style.MarkerStyle;
import com.revolsys.util.Property;

public class VertexStyleRenderer {
  private final MarkerStyle fromVertexStyle;

  private final MarkerStyle toVertexStyle;

  private final MarkerStyle vertexStyle;

  public VertexStyleRenderer(final Color color) {
    this.vertexStyle = MarkerStyle.marker(SelectedRecordsVertexRenderer.vertexShape(), 9,
      WebColors.Black, 1, color);
    this.vertexStyle.setMarkerOrientationType("auto");

    this.fromVertexStyle = MarkerStyle.marker(SelectedRecordsVertexRenderer.firstVertexShape(), 9,
      WebColors.Black, 1, color);
    this.fromVertexStyle.setMarkerOrientationType("auto");
    this.fromVertexStyle.setMarkerPlacementType("vertex(0)");
    this.fromVertexStyle.setMarkerHorizontalAlignment("center");

    this.toVertexStyle = MarkerStyle.marker(SelectedRecordsVertexRenderer.lastVertexShape(), 9,
      WebColors.Black, 1, color);
    this.toVertexStyle.setMarkerOrientationType("auto");
    this.toVertexStyle.setMarkerPlacementType("vertex(n)");
    this.toVertexStyle.setMarkerHorizontalAlignment("right");
  }

  public void paintSelected(final Viewport2D viewport, final Graphics2D graphics,
    final GeometryFactory viewportGeometryFactory, final Vertex vertex) {
    if (Property.hasValue(vertex)) {
      MarkerStyle style;
      if (vertex.isFrom()) {
        style = this.fromVertexStyle;
      } else if (vertex.isTo()) {
        style = this.toVertexStyle;
      } else {
        style = this.vertexStyle;
      }
      final double orientation = vertex.getOrientaton();
      MarkerStyleRenderer.renderMarker(viewport, graphics, vertex, style, orientation);
    }
  }
}