sascha@422: package de.intevation.gnv.jfreechart;
sascha@422: 
sascha@779: import java.awt.BasicStroke;
sascha@422: import java.awt.Color;
sascha@779: import java.awt.Font;
sascha@779: import java.awt.FontMetrics;
sascha@422: import java.awt.Graphics2D;
sascha@422: import java.awt.Paint;
sascha@422: import java.awt.Shape;
sascha@447: 
sascha@422: import java.awt.geom.GeneralPath;
sascha@779: 
sascha@779: import java.awt.geom.Rectangle2D.Double;
sascha@779: 
sascha@422: import java.awt.geom.Rectangle2D;
sascha@779: 
sascha@779: import java.util.ArrayList;
sascha@779: 
sascha@779: import org.apache.log4j.Logger;
sascha@779: 
sascha@779: import org.jfree.chart.axis.ValueAxis;
sascha@422: 
sascha@422: import org.jfree.data.Range;
sascha@422: 
sascha@779: import org.jfree.text.TextUtilities;
sascha@450: 
sascha@450: import org.jfree.ui.RectangleEdge;
sascha@450: 
sascha@422: /**
ingo@795:  * This renderer is used to draw polygons into a Graphics object.
sascha@803:  *
sascha@780:  * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
sascha@780:  * @author <a href="mailto:sascha.teichmann@intevation.de">Sascha L. Teichmann</a>
sascha@422:  */
sascha@422: public class PolygonRenderer
sascha@422: {
sascha@447: 	private static Logger log = Logger.getLogger(
sascha@447: 		PolygonRenderer.class);
sascha@447: 
ingo@795:         /**
ingo@795:          * This interfaces describes a single method to retrieve a Paint object
ingo@795:          * for a given index.
ingo@795:          */
ingo@795:         public interface PaintLookup {
sascha@422: 
ingo@795:             /**
ingo@795:              *
ingo@795:              * @param index Index.
ingo@795:              * @return Paint
ingo@795:              */
ingo@795:             Paint getPaint(int index);
sascha@422: 
sascha@422:     } // interface PaintLookup
sascha@422: 
ingo@795:         /**
ingo@795:          * This class is used to generate labels for a given series.
ingo@795:          */
ingo@795:         public static class DefaultLabelGenerator
sascha@450:     implements          PolygonSeriesLabelGenerator
sascha@450:     {
ingo@795:             /**
ingo@795:              * Construts an empty DefaultLabelGenerator.
ingo@795:              */
ingo@795:             public DefaultLabelGenerator() {
sascha@450:         }
sascha@422: 
ingo@795:             /**
ingo@795:              *
ingo@795:              * @param series A PolygonSeries.
ingo@795:              * @return The label of series.
ingo@795:              */
ingo@795:             public String generateLabel(PolygonSeries series) {
sascha@450:             Object label = series.getAttribute("label");
sascha@450:             return label != null
sascha@450:                 ? toString(label)
sascha@450:                 : null;
sascha@450:         }
sascha@422: 
ingo@795:             /**
ingo@795:              *
ingo@795:              * @param label Object
ingo@795:              * @return String representaton of label.
ingo@795:              */
ingo@795:             protected String toString(Object label) {
sascha@450:             return label.toString();
sascha@450:         }
sascha@450:     } // class DefaultLabelGenerator
sascha@422: 
ingo@795:         /**
ingo@795:          * Constructor.
ingo@795:          */
ingo@795:         public static final PolygonSeriesLabelGenerator
sascha@450:         DEFAULT_LABEL_GENERATOR_INSTANCE = new DefaultLabelGenerator();
sascha@450: 
ingo@795:         /**
ingo@795:          *
ingo@795:          */
ingo@795:         protected PaintLookup                 lookup;
ingo@795:     /**
ingo@795:      *
ingo@795:      */
sascha@450:     protected PolygonSeriesLabelGenerator labelGenerator;
sascha@422: 
ingo@795:     /**
ingo@795:      *
ingo@795:      * @param lookup
ingo@795:      */
sascha@422:     public PolygonRenderer(PaintLookup lookup) {
sascha@450:         this(lookup, null);
sascha@422:     }
sascha@422: 
ingo@795:     /**
ingo@795:      *
ingo@795:      * @param lookup
ingo@795:      * @param labelGenerator
ingo@795:      */
sascha@450:     public PolygonRenderer(
sascha@778:         PaintLookup                 lookup,
sascha@450:         PolygonSeriesLabelGenerator labelGenerator
sascha@450:     ) {
sascha@450:         this.lookup         = lookup;
sascha@450:         this.labelGenerator = labelGenerator;
sascha@422:     }
sascha@422: 
ingo@795:     /**
ingo@795:      * This method draws polygons of each series in <code>dataset</code> into
ingo@795:      * the given graphics object. If a polygon has no attribute 'fill', we
ingo@795:      * expect that it is a line, otherwise the polygon is filled.
ingo@795:      *
ingo@795:      * @param graphics
ingo@795:      * @param plot
ingo@795:      * @param area
ingo@795:      * @param dataset
ingo@795:      */
sascha@450:     public void drawPolygons(
sascha@422:         Graphics2D     graphics,
sascha@451:         PolygonPlot    plot,
sascha@451:         Rectangle2D    area,
sascha@422:         PolygonDataset dataset
sascha@422:     ) {
sascha@422:         int seriesCount = dataset.getSeriesCount();
sascha@422:         for (int i = 0; i < seriesCount; i++) {
sascha@422:             PolygonSeries series   = dataset.getSeries(i);
sascha@422:             Integer       colorIdx = (Integer)series.getAttribute("fill");
sascha@422: 
sascha@422:             if (colorIdx != null) {
sascha@422:                 Paint paint = lookup.getPaint(colorIdx.intValue());
sascha@422:                 graphics.setPaint(paint != null ? paint : Color.black);
sascha@451:                 graphics.fill(constructShape(plot, area, series, true));
sascha@422:             }
sascha@422:             else {
sascha@447:                 Number lineWidth = (Number)series.getAttribute("line.width");
sascha@447:                 BasicStroke stroke = new BasicStroke(
sascha@447:                     lineWidth != null ? lineWidth.floatValue() : 1f);
sascha@447:                 graphics.setStroke(stroke);
sascha@422:                 graphics.setPaint(Color.black);
sascha@451:                 graphics.draw(constructShape(plot, area, series, false));
sascha@422:             }
sascha@422:         }
sascha@450:     }
sascha@450: 
ingo@795:     /**
ingo@795:      * Draw labels at each item of a series in the given dataset. If the series
ingo@795:      * has no label attritue, no label is drawn.
sascha@803:      *
ingo@795:      * @param graphics
ingo@795:      * @param plot
ingo@795:      * @param area
ingo@795:      * @param dataset
ingo@795:      */
sascha@450:     public void drawLabels(
sascha@450:         final Graphics2D  graphics,
sascha@450:         final PolygonPlot plot,
sascha@450:         final Rectangle2D area,
sascha@450:         PolygonDataset    dataset
sascha@450:     ) {
sascha@450:         if (labelGenerator == null) {
sascha@450:             return;
sascha@450:         }
sascha@450: 
sascha@450:         final ArrayList<Rectangle2D> bboxes = new ArrayList<Rectangle2D>();
sascha@450: 
sascha@450:         Font font = graphics.getFont();
sascha@450:         font = font.deriveFont(Font.PLAIN, Math.max(8, font.getSize()-3));
sascha@450:         graphics.setFont(font);
sascha@450:         FontMetrics metrics = graphics.getFontMetrics(font);
sascha@450: 
sascha@450:         for (int i = dataset.getSeriesCount()-1; i >= 0; --i) {
sascha@450:             PolygonSeries series = dataset.getSeries(i);
sascha@450: 
sascha@450:             String label = labelGenerator.generateLabel(series);
sascha@450:             if (label == null) {
sascha@450:                 continue;
sascha@450:             }
sascha@450: 
sascha@450:             final Rectangle2D box = TextUtilities.getTextBounds(
sascha@450:                 label, graphics, metrics);
sascha@450: 
sascha@450:             for (int j = series.getItemCount()-1; j >= 0; --j) {
sascha@450:                 final CompactXYItems ring = series.getItem(j);
sascha@450:                 LevelOrderIndices loi = new LevelOrderIndices(ring.size()-1);
sascha@450:                 Rectangle2D r = (Rectangle2D)loi.visit(
sascha@778:                     new LevelOrderIndices.Visitor()
sascha@450:                 {
sascha@450:                     ValueAxis          da = plot.getDomainAxis();
sascha@450:                     ValueAxis          ra = plot.getRangeAxis();
sascha@450:                     RectangleEdge      de = plot.getDomainAxisEdge();
sascha@450:                     RectangleEdge      re = plot.getRangeAxisEdge();
sascha@450:                     Rectangle2D.Double r  = new Rectangle2D.Double(
sascha@450:                         0d, 0d, box.getWidth(), box.getHeight());
sascha@450: 
sascha@450:                     public Object visit(int index) {
sascha@778:                         r.x = da.valueToJava2D(ring.getX(index), area, de)
sascha@450:                             - 0.5*box.getWidth();
sascha@450:                         r.y = ra.valueToJava2D(ring.getY(index), area, re)
sascha@450:                             + 0.5*box.getHeight();
sascha@450: 
sascha@450:                         for (Rectangle2D b: bboxes) {
sascha@450:                             if (b.intersects(r)) {
sascha@450:                                 return null;
sascha@450:                             }
sascha@450:                         }
sascha@450:                         return r;
sascha@450:                     }
sascha@450:                 });
sascha@450: 
sascha@450:                 if (r != null) {
sascha@450:                     bboxes.add(r);
sascha@450:                     graphics.drawString(
sascha@450:                         label, (float)r.getX(), (float)r.getY());
sascha@450:                 }
sascha@450:             } // for all items in series
sascha@450:         } // for all series
sascha@422:     }
sascha@422: 
ingo@795:     /**
ingo@795:      * Creates a shape made up of the CompactXYItems object stored in the given
ingo@795:      * series.
ingo@795:      *
ingo@795:      * @param plot The plot.
ingo@795:      * @param area The boundary.
ingo@795:      * @param series The series storing the items.
ingo@795:      * @param close Specifies if the polygon should be closed or not.
ingo@795:      * @return
ingo@795:      */
sascha@451:     protected Shape constructShape(
sascha@451:         PolygonPlot   plot,
sascha@451:         Rectangle2D   area,
sascha@778:         PolygonSeries series,
sascha@451:         boolean       close
sascha@451:     ) {
sascha@451:         ValueAxis     da = plot.getDomainAxis();
sascha@451:         ValueAxis     ra = plot.getRangeAxis();
sascha@451:         RectangleEdge de = plot.getDomainAxisEdge();
sascha@451:         RectangleEdge re = plot.getRangeAxisEdge();
sascha@451: 
sascha@422:         CompactXYItems [] rings = series.getRings();
sascha@451:         GeneralPath       path  = new GeneralPath();
sascha@451: 
sascha@422:         for (int i = 0; i < rings.length; ++i) {
sascha@451: 
sascha@422:             CompactXYItems ring = rings[i];
sascha@451: 
sascha@422:             double [] data = ring.getData();
sascha@451: 
sascha@422:             if (data.length >= 2) {
sascha@451:                 path.moveTo(
sascha@451:                     (float)da.valueToJava2D(data[0], area, de),
sascha@451:                     (float)ra.valueToJava2D(data[1], area, re));
sascha@422:             }
sascha@422:             for (int j = 2; j < data.length;) {
sascha@451:                 path.lineTo(
sascha@451:                     (float)da.valueToJava2D(data[j++], area, de),
sascha@451:                     (float)ra.valueToJava2D(data[j++], area, re));
sascha@422:             }
sascha@437:             if (close) {
sascha@437:                 path.closePath();
sascha@437:             }
sascha@422:         }
sascha@422:         return path;
sascha@422:     }
sascha@422: 
ingo@795:     /**
ingo@795:      * Retrieves the bounding box of a dataset.
ingo@795:      *
ingo@795:      * @param dataset
ingo@795:      * @return
ingo@795:      */
sascha@422:     public Rectangle2D getBoundingBox(PolygonDataset dataset) {
sascha@422:         Rectangle2D bbox = null;
sascha@422: 
sascha@422:         for (int i = 0, N = dataset.getSeriesCount(); i < N; i++) {
sascha@422:             Range domain = dataset.getSeries(i).getDomainBounds();
sascha@422:             Range range  = dataset.getSeries(i).getRangeBounds();
sascha@422: 
sascha@422:             double x = domain.getLowerBound();
sascha@422:             double y = range.getLowerBound();
sascha@422:             double w = Math.abs(domain.getUpperBound() - x);
sascha@422:             double h = Math.abs(range.getUpperBound() - y);
sascha@422: 
sascha@422:             if (bbox == null) {
sascha@422:                 bbox = new Rectangle2D.Double(x, y, w, h);
sascha@422:             }
sascha@422:             else {
sascha@422:                 bbox.add(new Rectangle2D.Double(x, y, w, h));
sascha@422:             }
sascha@422:         }
sascha@422: 
sascha@422:         return bbox;
sascha@422:     }
sascha@422: 
ingo@795:     /**
ingo@795:      *
ingo@795:      * @param series
ingo@795:      * @return the bounds of a series.
ingo@795:      */
sascha@422:     public Rectangle2D getBounds(PolygonSeries series) {
sascha@422: 
sascha@422:         Range domain = series.getDomainBounds();
sascha@422:         Range range  = series.getRangeBounds();
sascha@422: 
sascha@422:         return new Rectangle2D.Double(
sascha@422:             domain.getLowerBound(), range.getLowerBound(),
sascha@422:             domain.getUpperBound(), range.getUpperBound()
sascha@422:         );
sascha@422:     }
sascha@422: }
sascha@422: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :