source: trunk/src/org/expeditee/items/widgets/charts/AbstractPie.java@ 919

/**
 *    AbstractPie.java
 *    Copyright (C) 2010 New Zealand Digital Library, http://expeditee.org
 *
 *    This program is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    This program is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.expeditee.items.widgets.charts;

import java.awt.BasicStroke;
import java.awt.Color;

import org.expeditee.gui.AttributeValuePair;
import org.expeditee.gui.Frame;
import org.expeditee.items.Text;
import org.jfree.chart.labels.PieSectionLabelGenerator;
import org.jfree.chart.plot.PiePlot;
import org.jfree.chart.plot.Plot;
import org.jfree.data.general.DefaultPieDataset;
import org.jfree.ui.RectangleInsets;

public abstract class AbstractPie extends AbstractChart {

        protected DefaultPieDataset _data;

        private PieSectionLabelGenerator _labelGenerator;

        protected DefaultPieDataset getChartData() {
                if (_data == null)
                        _data = new DefaultPieDataset();
                return _data;
        }

        public AbstractPie(Text source, String[] args) {
                super(source, args);
        }

        @Override
        protected void init() {
                super.init();
                _labelGenerator = ((PiePlot) getChart().getPlot()).getLabelGenerator();
        }

        /**
         * @param dataFrame
         */
        @Override
        protected void refreshPlot(Frame dataFrame, Plot plot) {
                if (dataFrame.hasAnnotation("labels")) {
                        ((PiePlot) plot).setLabelGenerator(_labelGenerator);
                } else {
                        ((PiePlot) plot).setLabelGenerator(null);
                }
        }

        @Override
        protected void clearData() {
                _data.clear();
        }

        /**
         * @param dataFrame
         */
        @Override
        protected void refreshData(Frame dataFrame) {
                for (Text t : dataFrame.getBodyTextItems(false)) {
                        // Ignore line ends for pie charts
                        //TODO find out HOW text became null when i was doing a graph!!
                        String text = t.getText();
                        if (t.isLineEnd() || text == null)
                                continue;
                        AttributeValuePair avp = new AttributeValuePair(text);
                        if (avp != null && avp.hasPair()) {
                                try {
                                        _data.setValue(avp.getAttribute(), avp.getDoubleValue());
                                        _paints.put(avp.getAttribute(), t.getBackgroundColor());
                                } catch (NumberFormatException nfe) {

                                }
                        }
                }
        }

        @Override
        public void setSourceBorderColor(Color c) {
                super.setSourceBorderColor(c);
                if (getChart() == null)
                        return;
                PiePlot plot = ((PiePlot) getChart().getPlot());
                if (c == null) {
                        plot.setBaseSectionOutlinePaint(Plot.DEFAULT_OUTLINE_PAINT);
                        plot.setLabelLinkPaint(PiePlot.DEFAULT_LABEL_OUTLINE_PAINT);
                        plot.setLabelOutlinePaint(PiePlot.DEFAULT_LABEL_OUTLINE_PAINT);
                        plot.setLabelShadowPaint(PiePlot.DEFAULT_LABEL_SHADOW_PAINT);
                        plot.setShadowPaint(PiePlot.DEFAULT_SHADOW_PAINT);
                } else {
                        plot.setBaseSectionOutlinePaint(c);
                        plot.setLabelLinkPaint(c);
                        plot.setLabelOutlinePaint(c);
                        Color newC = c.darker();
                        Color faded = new Color(newC.getRed(), newC.getGreen(), newC
                                        .getBlue(), 128);

                        plot.setLabelShadowPaint(faded);
                        plot.setShadowPaint(c);
                }
        }

        @Override
        public void setSourceColor(Color c) {
                super.setSourceColor(c);
                if (getChart() == null)
                        return;
                PiePlot plot = ((PiePlot) getChart().getPlot());
                if (c == null) {
                        plot.setLabelPaint(PiePlot.DEFAULT_LABEL_PAINT);
                } else {
                        plot.setLabelPaint(c);
                }
        }

        @Override
        public void setSourceThickness(float newThickness, boolean setConnected) {
                super.setSourceThickness(newThickness, setConnected);
                if (getChart() == null)
                        return;
                PiePlot plot = ((PiePlot) getChart().getPlot());
                plot.setBaseSectionOutlineStroke(new BasicStroke(newThickness));

                plot.setLabelLinkStroke(new BasicStroke(newThickness));
                plot.setLabelOutlineStroke(new BasicStroke(newThickness));

                plot.setLabelFont(plot.getLabelFont().deriveFont(
                                getFontSize(newThickness, PiePlot.DEFAULT_LABEL_FONT
                                                .getSize2D())));
                plot.setLabelPadding(new RectangleInsets(newThickness + 1,
                                newThickness + 1, newThickness + 1, newThickness + 1));
                plot.setLegendItemShape(Plot.getCircle((newThickness + 1) * 2));
        }
}