edu.uci.ics.jung.visualization.renderers.BasicNodeRenderer Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2005, The JUNG Authors
 * All rights reserved.
 *
 * This software is open-source under the BSD license; see either "license.txt"
 * or https://github.com/jrtom/jung/blob/master/LICENSE for a description.
 *
 * Created on Aug 23, 2005
 */
package edu.uci.ics.jung.visualization.renderers;

import edu.uci.ics.jung.layout.model.Point;
import edu.uci.ics.jung.visualization.MultiLayerTransformer.Layer;
import edu.uci.ics.jung.visualization.RenderContext;
import edu.uci.ics.jung.visualization.VisualizationModel;
import edu.uci.ics.jung.visualization.transform.shape.GraphicsDecorator;
import java.awt.Paint;
import java.awt.Shape;
import java.awt.Stroke;
import java.awt.geom.AffineTransform;
import java.awt.geom.Point2D;
import javax.swing.Icon;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class BasicNodeRenderer implements Renderer.Node {

  private static final Logger log = LoggerFactory.getLogger(BasicNodeRenderer.class);

  public void paintNode(
      RenderContext renderContext, VisualizationModel visualizationModel, N v) {
    if (renderContext.getNodeIncludePredicate().test(v)) {
      paintIconForNode(renderContext, visualizationModel, v);
    }
  }

  /**
   * Returns the node shape in layout coordinates.
   *
   * @param v the node whose shape is to be returned
   * @param coords the x and y view coordinates
   * @return the node shape in view coordinates
   */
  protected Shape prepareFinalNodeShape(
      RenderContext renderContext,
      VisualizationModel visualizationModel,
      N v,
      int[] coords) {

    // get the shape to be rendered
    Shape shape = renderContext.getNodeShapeFunction().apply(v);
    Point p = visualizationModel.getLayoutModel().apply(v);
    // p is the node location in layout coordinates
    log.trace("prepared a shape for " + v + " to go at " + p);
    Point2D p2d =
        renderContext
            .getMultiLayerTransformer()
            .transform(Layer.LAYOUT, new Point2D.Double(p.x, p.y));
    // now p is in view coordinates, ready to be further transformed by any transform in the
    // graphics context
    float x = (float) p2d.getX();
    float y = (float) p2d.getY();
    coords[0] = (int) x;
    coords[1] = (int) y;
    // create a transform that translates to the location of
    // the node to be rendered
    AffineTransform xform = AffineTransform.getTranslateInstance(x, y);
    // transform the node shape with xtransform
    shape = xform.createTransformedShape(shape);
    return shape;
  }

  /**
   * Paint v's icon on g at (x,y).
   *
   * @param v the node to be painted
   */
  protected void paintIconForNode(
      RenderContext renderContext, VisualizationModel visualizationModel, N v) {
    GraphicsDecorator g = renderContext.getGraphicsContext();
    int[] coords = new int[2];
    Shape shape = prepareFinalNodeShape(renderContext, visualizationModel, v, coords);

    if (renderContext.getNodeIconFunction() != null) {
      Icon icon = renderContext.getNodeIconFunction().apply(v);
      if (icon != null) {

        g.draw(icon, renderContext.getScreenDevice(), shape, coords[0], coords[1]);

      } else {
        paintShapeForNode(renderContext, visualizationModel, v, shape);
      }
    } else {
      paintShapeForNode(renderContext, visualizationModel, v, shape);
    }
  }

  protected void paintShapeForNode(
      RenderContext renderContext,
      VisualizationModel visualizationModel,
      N v,
      Shape shape) {
    GraphicsDecorator g = renderContext.getGraphicsContext();
    Paint oldPaint = g.getPaint();
    Paint fillPaint = renderContext.getNodeFillPaintFunction().apply(v);
    if (fillPaint != null) {
      g.setPaint(fillPaint);
      g.fill(shape);
      g.setPaint(oldPaint);
    }
    Paint drawPaint = renderContext.getNodeDrawPaintFunction().apply(v);
    if (drawPaint != null) {
      g.setPaint(drawPaint);
      Stroke oldStroke = g.getStroke();
      Stroke stroke = renderContext.getNodeStrokeFunction().apply(v);
      if (stroke != null) {
        g.setStroke(stroke);
      }
      g.draw(shape);
      g.setPaint(oldPaint);
      g.setStroke(oldStroke);
    }
  }
}