Java tutorial
/** * This file is part of mycollab-web. * * mycollab-web 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 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. If not, see <http://www.gnu.org/licenses/>. */ package com.esofthead.mycollab.ui.chart; import com.esofthead.mycollab.common.domain.GroupItem; import com.esofthead.mycollab.core.arguments.SearchCriteria; import com.esofthead.mycollab.core.utils.StringUtils; import com.esofthead.mycollab.vaadin.AppContext; import com.esofthead.mycollab.vaadin.web.ui.UIConstants; 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 org.vaadin.viritin.layouts.MHorizontalLayout; 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 { private static final long serialVersionUID = 1L; protected DefaultPieDataset pieDataSet; protected S searchCriteria; protected List<GroupItem> groupItems; private Class<? extends Enum<?>> enumKeyCls; public PieChartWrapper(final int width, final int height) { super(width, height); } public PieChartWrapper(Class<? extends Enum<?>> emumKey, final int width, final int height) { super(width, height); this.enumKeyCls = emumKey; } public void displayChart(final S criteria) { removeAllComponents(); this.searchCriteria = criteria; this.groupItems = loadGroupItems(); displayChart(); } abstract protected List<GroupItem> loadGroupItems(); @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); int colorIndex = i % CHART_COLOR_STR.size(); plot.setSectionPaint(key, Color.decode("0x" + CHART_COLOR_STR.get(colorIndex))); } // OPTIONAL CUSTOMISATION COMPLETED. return chart; } protected abstract DefaultPieDataset createDataset(); 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) { if (key instanceof Key) { return String.format("%s (%d)", StringUtils.trim(((Key) key).getDisplayName(), 20, true), value); } else { 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 CssLayout mainLayout = new CssLayout(); mainLayout.addStyleName("legendBoxContent"); mainLayout.setSizeUndefined(); final List keys = pieDataSet.getKeys(); for (int i = 0; i < keys.size(); i++) { MHorizontalLayout layout = new MHorizontalLayout() .withMargin(new MarginInfo(false, false, false, true)); layout.addStyleName("inline-block"); layout.setSizeUndefined(); layout.setDefaultComponentAlignment(Alignment.MIDDLE_CENTER); final Comparable key = (Comparable) keys.get(i); int colorIndex = i % CHART_COLOR_STR.size(); final String color = "<div style = \" width:13px;height:13px;background: #" + CHART_COLOR_STR.get(colorIndex) + "\" />"; final Label lblCircle = new Label(color); lblCircle.setContentMode(ContentMode.HTML); String btnCaption; if (enumKeyCls == null) { if (key instanceof Key) { btnCaption = String.format("%s (%d)", StringUtils.trim(((Key) key).getDisplayName(), 20, true), pieDataSet.getValue(key).intValue()); } else { 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(StringUtils.trim(btnCaption, 25, true), new Button.ClickListener() { private static final long serialVersionUID = 1L; @Override public void buttonClick(final ClickEvent event) { if (key instanceof Key) { clickLegendItem(((Key) key).getKey()); } else { clickLegendItem(key.toString()); } } }); btnLink.setDescription(btnCaption); btnLink.addStyleName(UIConstants.BUTTON_LINK); layout.with(lblCircle, btnLink); mainLayout.addComponent(layout); } mainLayout.setWidth("100%"); return mainLayout; } abstract protected void clickLegendItem(String key); }