Java tutorial
/** * Copyright (c) 2008 Pyxis Technologies inc. * * This 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 2 of the License, or (at your option) any later * version. * * This software 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, write to the Free Software Foundation, Inc., 51 * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA, or see the FSF site: * http://www.fsf.org. */ package info.novatec.testit.livingdoc.confluence.macros.historic; import java.awt.Color; import java.awt.image.BufferedImage; import java.io.IOException; import java.text.DecimalFormat; import java.text.NumberFormat; import java.util.HashMap; import java.util.List; import java.util.Map; import info.novatec.testit.livingdoc.confluence.velocity.LivingDocConfluenceManager; import org.jfree.chart.ChartFactory; import org.jfree.chart.JFreeChart; import org.jfree.chart.axis.CategoryAxis; import org.jfree.chart.axis.CategoryLabelPositions; import org.jfree.chart.axis.NumberAxis; import org.jfree.chart.axis.NumberTickUnit; import org.jfree.chart.axis.ValueAxis; import org.jfree.chart.labels.AbstractCategoryItemLabelGenerator; import org.jfree.chart.labels.CategoryToolTipGenerator; import org.jfree.chart.plot.CategoryPlot; import org.jfree.chart.plot.PlotOrientation; import org.jfree.chart.renderer.category.StackedBarRenderer; import org.jfree.chart.title.TextTitle; import org.jfree.chart.urls.CategoryURLGenerator; import org.jfree.data.DataUtilities; import org.jfree.data.category.CategoryDataset; import org.jfree.data.category.DefaultCategoryDataset; import info.novatec.testit.livingdoc.server.LivingDocServerException; import info.novatec.testit.livingdoc.server.domain.Execution; public class AggregationExecutionChartBuilder extends AbstractChartBuilder { private static final NumberFormat PERCENT_FORMATTER = new DecimalFormat("0%"); private final List<Execution> executions; private LivingDocConfluenceManager ldUtil; private AggregationExecutionChartBuilder(HistoricParameters settings, List<Execution> executions, LivingDocConfluenceManager ldUtil) { super(settings); this.executions = executions; this.ldUtil = ldUtil; } public static AggregationExecutionChartBuilder newInstance(HistoricParameters settings, List<Execution> executions, LivingDocConfluenceManager confluenceLivingDoc) { return new AggregationExecutionChartBuilder(settings, executions, confluenceLivingDoc); } @Override public String generateChart() throws LivingDocServerException { try { CategoryDataset dataset = generateDataset(); JFreeChart chart = createChart(dataset); BufferedImage chartImage = createChartImage(chart); return getDownloadPath(chartImage); } catch (IOException e) { throw new LivingDocServerException("livingdoc.server.generalexeerror", e.getMessage(), e); } } private CategoryDataset generateDataset() { final String successLabel = ldUtil.getText("livingdoc.historic.success"); final String failuresLabel = ldUtil.getText("livingdoc.historic.failures"); final String ignoredLabel = ldUtil.getText("livingdoc.historic.ignored"); DefaultCategoryDataset dataset = new DefaultCategoryDataset(); for (Execution exec : executions) { String category = exec.getSpecification().getName(); ChartLongValue value = new ChartLongValue(category, exec.getId()); double failureCount = exec.getErrors() + exec.getFailures() + (exec.hasException() ? 1 : 0); dataset.addValue(new Double(exec.getSuccess()), successLabel, value); if (settings.isShowIgnored()) { dataset.addValue(new Double(exec.getIgnored()), ignoredLabel, value); } dataset.addValue(new Double(failureCount), failuresLabel, value); } return dataset; } private JFreeChart createChart(CategoryDataset dataset) throws IOException { JFreeChart chart = ChartFactory.createAreaChart(settings.getTitle(), null, null, dataset, PlotOrientation.VERTICAL, true, false, false); customizeChart(chart); return chart; } @SuppressWarnings("deprecation") private void customizeChart(JFreeChart chart) throws IOException { chart.setBackgroundPaint(Color.white); chart.setBorderVisible(settings.isBorder()); TextTitle chartTitle = chart.getTitle(); customizeTitle(chartTitle, DEFAULT_TITLE_FONT); addSubTitle(chart, settings.getSubTitle(), DEFAULT_SUBTITLE_FONT); addSubTitle(chart, settings.getSubTitle2(), DEFAULT_SUBTITLE2_FONT); CategoryPlot plot = (CategoryPlot) chart.getPlot(); plot.setNoDataMessage(ldUtil.getText("livingdoc.historic.nodata")); StackedBarRenderer renderer = new StackedBarRenderer(true); plot.setRenderer(renderer); int index = 0; renderer.setSeriesPaint(index++, GREEN_COLOR); if (settings.isShowIgnored()) { renderer.setSeriesPaint(index++, Color.yellow); } renderer.setSeriesPaint(index, Color.red); renderer.setToolTipGenerator(new DefaultTooltipGenerator()); renderer.setItemURLGenerator(new CategoryURLGenerator() { @Override public String generateURL(CategoryDataset data, int series, int category) { Comparable<?> valueKey = data.getColumnKey(category); ChartLongValue value = (ChartLongValue) valueKey; return "javascript:" + settings.getExecutionUID() + "_showHistoricChart('" + value.getId() + "');"; } }); CategoryAxis domainAxis = plot.getDomainAxis(); customizeAxis(domainAxis); domainAxis.setCategoryLabelPositions(CategoryLabelPositions.UP_90); domainAxis.setCategoryMargin(0.01); ValueAxis rangeAxis = plot.getRangeAxis(); customizeAxis(rangeAxis); rangeAxis.setLowerBound(0); rangeAxis.setUpperBound(1.0); if (rangeAxis instanceof NumberAxis) { NumberAxis numberAxis = (NumberAxis) rangeAxis; numberAxis.setTickUnit(new NumberTickUnit(.10)); numberAxis.setNumberFormatOverride(PERCENT_FORMATTER); } plot.setForegroundAlpha(0.8f); } @SuppressWarnings("serial") private class DefaultTooltipGenerator extends AbstractCategoryItemLabelGenerator implements CategoryToolTipGenerator { private final Map<Integer, Double> totals = new HashMap<Integer, Double>(); public DefaultTooltipGenerator() { super("", PERCENT_FORMATTER); } @Override public String generateToolTip(CategoryDataset dataset, int row, int column) { final double base = getColumnTotal(dataset, column); final Number value = dataset.getValue(row, column); final double v = value.doubleValue(); return String.valueOf(dataset.getColumnKey(column)) + " - " + dataset.getRowKey(row) + " - " + getNumberFormat().format(v / base); } private double getColumnTotal(CategoryDataset dataset, int column) { Double total = totals.get(column); if (total == null) { total = DataUtilities.calculateColumnTotal(dataset, column); totals.put(column, total); } return total; } } }