psiprobe.controllers.RenderChartController.java Source code

Java tutorial

Introduction

Here is the source code for psiprobe.controllers.RenderChartController.java

Source

/**
 * Licensed under the GPL License. You may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 *
 * THIS PACKAGE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
 * WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE.
 */
package psiprobe.controllers;

import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartUtilities;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.DateAxis;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.renderer.xy.XYAreaRenderer;
import org.jfree.chart.renderer.xy.XYLine3DRenderer;
import org.jfree.data.xy.DefaultTableXYDataset;
import org.jfree.ui.RectangleInsets;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.ServletRequestUtils;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.AbstractController;

import psiprobe.Utils;
import psiprobe.beans.stats.providers.SeriesProvider;
import psiprobe.model.stats.StatsCollection;

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

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * Plots data from "statsCollection" bean. The data is converted to XYSeries using SeriesProvider,
 * name of which would be passed as a request parameter. The servlet can only plot up to 9 series.
 * It is customizable using these request parameters:
 * <ul>
 * <li>s1c, s2c, ... s9c - Series #i main color</li>
 * <li>s1o, s2o, ... s9o - Series #i outline color</li>
 * <li>bc - Chart background color</li>
 * <li>gc - Chart grid lines color</li>
 * <li>xl - X axis label</li>
 * <li>yl - Y axis label</li>
 * <li>xz - image width</li>
 * <li>yx - image height</li>
 * <li>l - show legend (boolean: true|false)</li>
 * <li>p - name of series provider bean</li>
 * </ul>
 */
public class RenderChartController extends AbstractController {

    /** The Constant logger. */
    private static final Logger logger = LoggerFactory.getLogger(RenderChartController.class);

    /** The stats collection. */
    private StatsCollection statsCollection;

    /**
     * Gets the stats collection.
     *
     * @return the stats collection
     */
    public StatsCollection getStatsCollection() {
        return statsCollection;
    }

    /**
     * Sets the stats collection.
     *
     * @param statsCollection the new stats collection
     */
    public void setStatsCollection(StatsCollection statsCollection) {
        this.statsCollection = statsCollection;
    }

    @Override
    protected ModelAndView handleRequestInternal(HttpServletRequest request, HttpServletResponse response)
            throws Exception {

        final int seriesMaxCount = 9; // the max number of series

        // get Series Color from the request
        int[] seriesColor = new int[seriesMaxCount];
        seriesColor[0] = Utils.toIntHex(request.getParameter("s1c"), 0x9bd2fb);
        seriesColor[1] = Utils.toIntHex(request.getParameter("s2c"), 0xFF0606);
        for (int i = 2; i < seriesMaxCount; i++) {
            seriesColor[i] = Utils.toIntHex(request.getParameter("s" + (i + 1) + "c"), -1);
        }

        // get Series Outline Color from the request
        int[] seriesOutlineColor = new int[seriesMaxCount];
        seriesOutlineColor[0] = Utils.toIntHex(request.getParameter("s1o"), 0x0665aa);
        seriesOutlineColor[1] = Utils.toIntHex(request.getParameter("s2o"), 0x9d0000);
        for (int i = 2; i < seriesMaxCount; i++) {
            seriesOutlineColor[i] = Utils.toIntHex(request.getParameter("s" + (i + 1) + "o"), -1);
        }

        // background color
        int backgroundColor = Utils.toIntHex(request.getParameter("bc"), 0xFFFFFF);

        // grid color
        int gridColor = Utils.toIntHex(request.getParameter("gc"), 0);

        // X axis title
        String labelX = ServletRequestUtils.getStringParameter(request, "xl", "");

        // Y axis title
        String labelY = ServletRequestUtils.getStringParameter(request, "yl", "");

        // image width
        int width = ServletRequestUtils.getIntParameter(request, "xz", 800);

        // image height
        int height = ServletRequestUtils.getIntParameter(request, "yz", 400);

        // show legend?
        boolean showLegend = ServletRequestUtils.getBooleanParameter(request, "l", true);

        // Series provider
        String provider = ServletRequestUtils.getStringParameter(request, "p", null);

        // Chart type
        String chartType = ServletRequestUtils.getStringParameter(request, "ct", "area");

        DefaultTableXYDataset ds = new DefaultTableXYDataset();

        if (provider != null) {
            Object series = getApplicationContext().getBean(provider);
            if (series instanceof SeriesProvider) {
                ((SeriesProvider) series).populate(ds, statsCollection, request);
            } else {
                logger.error("SeriesProvider '{}' does not implement '{}'", provider, SeriesProvider.class);
            }

        }

        // Build series data from the give statistic
        JFreeChart chart = null;
        if ("area".equals(chartType)) {
            chart = ChartFactory.createXYAreaChart("", labelX, labelY, ds, PlotOrientation.VERTICAL, showLegend,
                    false, false);

            ((XYAreaRenderer) chart.getXYPlot().getRenderer()).setOutline(true);

        } else if ("stacked".equals(chartType)) {
            chart = ChartFactory.createStackedXYAreaChart("", labelX, labelY, ds, PlotOrientation.VERTICAL,
                    showLegend, false, false);

        } else if ("line".equals(chartType)) {
            chart = ChartFactory.createXYLineChart("", labelX, labelY, ds, PlotOrientation.VERTICAL, showLegend,
                    false, false);

            final XYLine3DRenderer renderer = new XYLine3DRenderer();
            renderer.setDrawOutlines(true);
            for (int i = 0; i < seriesMaxCount; i++) {
                renderer.setSeriesLinesVisible(i, true);
                renderer.setSeriesShapesVisible(i, true);
                renderer.setSeriesStroke(i, new BasicStroke(2));
            }
            renderer.setXOffset(1);
            renderer.setYOffset(1);
            chart.getXYPlot().setRenderer(renderer);
        }

        if (chart != null) {
            chart.setAntiAlias(true);
            chart.setBackgroundPaint(new Color(backgroundColor));
            for (int i = 0; i < seriesMaxCount; i++) {
                if (seriesColor[i] >= 0) {
                    chart.getXYPlot().getRenderer().setSeriesPaint(i, new Color(seriesColor[i]));
                }
                if (seriesOutlineColor[i] >= 0) {
                    chart.getXYPlot().getRenderer().setSeriesOutlinePaint(i, new Color(seriesOutlineColor[i]));
                }
            }
            chart.getXYPlot().setDomainGridlinePaint(new Color(gridColor));
            chart.getXYPlot().setRangeGridlinePaint(new Color(gridColor));
            chart.getXYPlot().setDomainAxis(0, new DateAxis());
            chart.getXYPlot().setDomainAxis(1, new DateAxis());
            chart.getXYPlot().setInsets(new RectangleInsets(-15, 0, 0, 10));

            response.setHeader("Content-type", "image/png");
            response.getOutputStream().write(ChartUtilities.encodeAsPNG(chart.createBufferedImage(width, height)));
        }

        return null;
    }
}