psiprobe.controllers.RenderChartController Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of psi-probe-core Show documentation
Show all versions of psi-probe-core Show documentation
PSI Probe Core - Core logic, data models, and controllers
/**
* 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
*
* https://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 java.awt.BasicStroke;
import java.awt.Color;
import javax.inject.Inject;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartUtils;
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.ui.RectangleInsets;
import org.jfree.data.xy.DefaultTableXYDataset;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.ServletRequestUtils;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.AbstractController;
import psiprobe.Utils;
import psiprobe.beans.stats.providers.SeriesProvider;
import psiprobe.jfreechart.XYLine3DRenderer;
import psiprobe.model.stats.StatsCollection;
/**
* 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:
*
* - s1c, s2c, ... s9c - Series #i main color
* - s1o, s2o, ... s9o - Series #i outline color
* - bc - Chart background color
* - gc - Chart grid lines color
* - xl - X axis label
* - yl - Y axis label
* - xz - image width
* - yx - image height
* - l - show legend (boolean: true|false)
* - p - name of series provider bean
*
*/
@Controller
public class RenderChartController extends AbstractController {
/** The Constant logger. */
private static final Logger logger = LoggerFactory.getLogger(RenderChartController.class);
/** The stats collection. */
@Inject
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;
}
@RequestMapping(path = "/chart.png")
@Override
public ModelAndView handleRequest(HttpServletRequest request, HttpServletResponse response)
throws Exception {
return super.handleRequest(request, response);
}
@Override
protected ModelAndView handleRequestInternal(HttpServletRequest request,
HttpServletResponse response) throws Exception {
// the max number of series
final int seriesMaxCount = 9;
// 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(ChartUtils.encodeAsPNG(chart.createBufferedImage(width, height)));
}
return null;
}
}