edu.uci.ics.jung.visualization.control.CubicCurveEdgeEffects Maven / Gradle / Ivy

Go to download
package edu.uci.ics.jung.visualization.control;

import edu.uci.ics.jung.visualization.BasicVisualizationServer;
import edu.uci.ics.jung.visualization.VisualizationServer;
import edu.uci.ics.jung.visualization.util.ArrowFactory;
import java.awt.Color;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Shape;
import java.awt.geom.AffineTransform;
import java.awt.geom.CubicCurve2D;
import java.awt.geom.Point2D;

public class CubicCurveEdgeEffects implements EdgeEffects {

  protected CubicCurve2D rawEdge = new CubicCurve2D.Float();
  protected Shape edgeShape;
  protected Shape rawArrowShape;
  protected Shape arrowShape;
  protected VisualizationServer.Paintable edgePaintable;
  protected VisualizationServer.Paintable arrowPaintable;

  public CubicCurveEdgeEffects() {
    this.rawEdge.setCurve(0.0f, 0.0f, 0.33f, 100, 0.66f, -50, 1.0f, 0.0f);
    rawArrowShape = ArrowFactory.getNotchedArrow(20, 16, 8);
    this.edgePaintable = new EdgePaintable();
    this.arrowPaintable = new ArrowPaintable();
  }

  @Override
  public void startEdgeEffects(BasicVisualizationServer vv, Point2D down, Point2D out) {
    transformEdgeShape(down, out);
    vv.addPostRenderPaintable(edgePaintable);
  }

  @Override
  public void midEdgeEffects(BasicVisualizationServer vv, Point2D down, Point2D out) {
    transformEdgeShape(down, out);
  }

  @Override
  public void endEdgeEffects(BasicVisualizationServer vv) {
    vv.removePostRenderPaintable(edgePaintable);
  }

  //	@Override
  public void startArrowEffects(BasicVisualizationServer vv, Point2D down, Point2D out) {
    transformArrowShape(down, out);
    vv.addPostRenderPaintable(arrowPaintable);
  }

  @Override
  public void midArrowEffects(BasicVisualizationServer vv, Point2D down, Point2D out) {
    transformArrowShape(down, out);
  }

  @Override
  public void endArrowEffects(BasicVisualizationServer vv) {
    vv.removePostRenderPaintable(arrowPaintable);
  }

  /** code lifted from PluggableRenderer to move an edge shape into an arbitrary position */
  private void transformEdgeShape(Point2D down, Point2D out) {
    float x1 = (float) down.getX();
    float y1 = (float) down.getY();
    float x2 = (float) out.getX();
    float y2 = (float) out.getY();

    AffineTransform xform = AffineTransform.getTranslateInstance(x1, y1);

    float dx = x2 - x1;
    float dy = y2 - y1;
    float thetaRadians = (float) Math.atan2(dy, dx);
    xform.rotate(thetaRadians);
    float dist = (float) Math.sqrt(dx * dx + dy * dy);
    xform.scale(dist / rawEdge.getBounds().getWidth(), 1.0);
    edgeShape = xform.createTransformedShape(rawEdge);
  }

  private void transformArrowShape(Point2D down, Point2D out) {
    float x1 = (float) down.getX();
    float y1 = (float) down.getY();
    float x2 = (float) out.getX();
    float y2 = (float) out.getY();

    AffineTransform xform = AffineTransform.getTranslateInstance(x2, y2);

    float dx = x2 - x1;
    float dy = y2 - y1;
    float thetaRadians = (float) Math.atan2(dy, dx);
    xform.rotate(thetaRadians);
    arrowShape = xform.createTransformedShape(rawArrowShape);
  }
  /** Used for the edge creation visual effect during mouse drag */
  class EdgePaintable implements VisualizationServer.Paintable {

    public void paint(Graphics g) {
      if (edgeShape != null) {
        Color oldColor = g.getColor();
        g.setColor(Color.black);
        ((Graphics2D) g).draw(edgeShape);
        g.setColor(oldColor);
      }
    }

    public boolean useTransform() {
      return false;
    }
  }

  /** Used for the directed edge creation visual effect during mouse drag */
  class ArrowPaintable implements VisualizationServer.Paintable {

    public void paint(Graphics g) {
      if (arrowShape != null) {
        Color oldColor = g.getColor();
        g.setColor(Color.black);
        ((Graphics2D) g).fill(arrowShape);
        g.setColor(oldColor);
      }
    }

    public boolean useTransform() {
      return false;
    }
  }
}