com.esofthead.mycollab.community.ui.chart.PieChartWrapper.java Source code

Java tutorial

Introduction

Here is the source code for com.esofthead.mycollab.community.ui.chart.PieChartWrapper.java

Source

/**
 * This file is part of mycollab-web-community.
 *
 * mycollab-web-community 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.
 *
 * mycollab-web-community 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 mycollab-web-community.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.esofthead.mycollab.community.ui.chart;

import com.esofthead.mycollab.core.arguments.SearchCriteria;
import com.esofthead.mycollab.vaadin.AppContext;
import com.esofthead.mycollab.web.CustomLayoutExt;
import com.vaadin.shared.ui.MarginInfo;
import com.vaadin.shared.ui.label.ContentMode;
import com.vaadin.ui.*;
import com.vaadin.ui.Button;
import com.vaadin.ui.Button.ClickEvent;
import com.vaadin.ui.Label;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.labels.PieSectionLabelGenerator;
import org.jfree.chart.plot.PiePlot3D;
import org.jfree.data.general.DefaultPieDataset;
import org.jfree.data.general.PieDataset;
import org.jfree.ui.RectangleInsets;
import org.jfree.util.Rotation;

import java.awt.*;
import java.text.AttributedString;
import java.util.List;

/**
 * 
 * @author MyCollab Ltd.
 * @since 1.0
 */
public abstract class PieChartWrapper<S extends SearchCriteria> extends GenericChartWrapper<S> {
    private static final long serialVersionUID = 1L;

    protected DefaultPieDataset pieDataSet;

    private Class<? extends Enum<?>> enumKeyCls;

    public PieChartWrapper(final String title, final int width, final int height) {
        super(title, width, height);
    }

    public PieChartWrapper(final String title, Class<? extends Enum<?>> emumKey, final int width,
            final int height) {
        super(title, width, height);
        this.enumKeyCls = emumKey;
    }

    @Override
    protected JFreeChart createChart() {
        // create the chart...
        pieDataSet = createDataset();
        final JFreeChart chart = ChartFactory.createPieChart3D("", // chart
                // title
                pieDataSet, // data
                false, // include legend
                true, // tooltips?
                false // URLs?
        );

        // NOW DO SOME OPTIONAL CUSTOMISATION OF THE CHART...
        chart.getTitle().setPaint(new Color(0x5E5E5E));
        chart.setBorderVisible(false);
        // set the background color for the chart...
        final PiePlot3D plot = (PiePlot3D) chart.getPlot();
        plot.setOutlineVisible(false);
        plot.setInsets(RectangleInsets.ZERO_INSETS);
        plot.setStartAngle(290);
        plot.setBackgroundPaint(Color.white);
        plot.setDirection(Rotation.CLOCKWISE);
        plot.setForegroundAlpha(0.5f);
        plot.setNoDataMessage("No data to display");
        plot.setLabelGenerator(new JFreeChartLabelCustom());

        final List keys = pieDataSet.getKeys();
        for (int i = 0; i < keys.size(); i++) {
            final Comparable key = (Comparable) keys.get(i);
            plot.setSectionPaint(key, Color.decode(
                    "0x" + GenericChartWrapper.CHART_COLOR_STR[i % GenericChartWrapper.CHART_COLOR_STR.length]));
        }
        // OPTIONAL CUSTOMISATION COMPLETED.
        return chart;
    }

    protected abstract DefaultPieDataset createDataset();

    protected abstract void onClickedDescription(String key);

    class JFreeChartLabelCustom implements PieSectionLabelGenerator {

        @Override
        public String generateSectionLabel(PieDataset dataset, Comparable key) {
            String result = null;
            if (dataset != null) {
                int value = dataset.getValue(key).intValue();
                if (value == 0) {
                    return null;
                }

                if (enumKeyCls == null) {
                    return String.format("%s (%d)", key.toString(), value);
                } else {
                    return String.format("%s (%d)", AppContext.getMessage(enumKeyCls, key.toString()), value);
                }
            }
            return result;
        }

        @Override
        public AttributedString generateAttributedSectionLabel(PieDataset dataset, Comparable key) {
            throw new UnsupportedOperationException("Not supported yet.");
        }
    }

    @Override
    protected final ComponentContainer createLegendBox() {
        final CustomLayout boxWrapper = CustomLayoutExt.createLayout("legendBox");
        final CssLayout mainLayout = new CssLayout();

        mainLayout.setSizeUndefined();
        final List keys = pieDataSet.getKeys();

        for (int i = 0; i < keys.size(); i++) {
            final HorizontalLayout layout = new HorizontalLayout();
            layout.setMargin(new MarginInfo(false, false, false, true));
            layout.addStyleName("inline-block");
            final Comparable key = (Comparable) keys.get(i);
            final String color = "<div style = \" width:8px;height:8px;border-radius:5px;background: #"
                    + GenericChartWrapper.CHART_COLOR_STR[i % GenericChartWrapper.CHART_COLOR_STR.length] + "\" />";
            final Label lblCircle = new Label(color);
            lblCircle.setContentMode(ContentMode.HTML);

            String btnCaption;
            if (enumKeyCls == null) {
                btnCaption = String.format("%s(%d)", key, pieDataSet.getValue(key).intValue());
            } else {
                btnCaption = String.format("%s(%d)", AppContext.getMessage(enumKeyCls, key.toString()),
                        pieDataSet.getValue(key).intValue());
            }
            final Button btnLink = new Button(btnCaption, new Button.ClickListener() {
                private static final long serialVersionUID = 1L;

                @Override
                public void buttonClick(final ClickEvent event) {
                    PieChartWrapper.this.onClickedDescription(key.toString());
                }
            });
            btnLink.addStyleName("link");
            layout.addComponent(lblCircle);
            layout.setComponentAlignment(lblCircle, Alignment.MIDDLE_CENTER);
            layout.addComponent(btnLink);
            layout.setComponentAlignment(btnLink, Alignment.MIDDLE_CENTER);
            layout.setSizeUndefined();
            mainLayout.addComponent(layout);
        }
        boxWrapper.setWidth("100%");
        boxWrapper.addComponent(mainLayout, "legendBoxContent");
        return boxWrapper;
    }
}