mirror of
https://github.com/graphhopper/jsprit.git
synced 2020-01-24 07:45:05 +01:00
107 lines
3 KiB
Java
107 lines
3 KiB
Java
package jsprit.analysis.toolbox;
|
|
|
|
import java.awt.Color;
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.util.concurrent.ConcurrentHashMap;
|
|
|
|
import org.jfree.chart.ChartFactory;
|
|
import org.jfree.chart.ChartUtilities;
|
|
import org.jfree.chart.JFreeChart;
|
|
import org.jfree.chart.axis.NumberAxis;
|
|
import org.jfree.chart.plot.PlotOrientation;
|
|
import org.jfree.chart.plot.XYPlot;
|
|
import org.jfree.data.Range;
|
|
import org.jfree.data.xy.XYSeries;
|
|
import org.jfree.data.xy.XYSeriesCollection;
|
|
|
|
/**
|
|
*
|
|
* @author schroeder
|
|
*
|
|
*/
|
|
public class XYLineChartBuilder {
|
|
|
|
/**
|
|
* Helper that just saves the chart as specified png-file. The width of the image is 1000 and height 600.
|
|
*
|
|
* @param chart
|
|
* @param pngFilename
|
|
*/
|
|
public static void saveChartAsPNG(JFreeChart chart, String pngFilename){
|
|
try {
|
|
ChartUtilities.saveChartAsPNG(new File(pngFilename), chart, 1000, 600);
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Returns a new instance of the builder.
|
|
*
|
|
* @param chartTitle appears on top of the XYLineChart
|
|
* @param xDomainName appears below the xAxis
|
|
* @param yDomainName appears beside the yAxis
|
|
*
|
|
* @return the builder
|
|
*/
|
|
public static XYLineChartBuilder newInstance(String chartTitle, String xDomainName, String yDomainName){
|
|
return new XYLineChartBuilder(chartTitle, xDomainName, yDomainName);
|
|
}
|
|
|
|
private ConcurrentHashMap<String,XYSeries> seriesMap = new ConcurrentHashMap<String, XYSeries>();
|
|
|
|
private final String xDomain;
|
|
|
|
private final String yDomain;
|
|
|
|
private final String chartName;
|
|
|
|
private XYLineChartBuilder(String chartName, String xDomainName, String yDomainName) {
|
|
this.xDomain=xDomainName;
|
|
this.yDomain=yDomainName;
|
|
this.chartName=chartName;
|
|
}
|
|
|
|
/**
|
|
* Adds data to the according series (i.e. XYLine).
|
|
*
|
|
* @param seriesName
|
|
* @param xVal
|
|
* @param yVal
|
|
*/
|
|
public void addData(String seriesName, double xVal, double yVal){
|
|
if(!seriesMap.containsKey(seriesName)){
|
|
seriesMap.put(seriesName, new XYSeries(seriesName,true,true));
|
|
}
|
|
seriesMap.get(seriesName).add(xVal, yVal);
|
|
}
|
|
|
|
/**
|
|
* Builds and returns JFreeChart.
|
|
* @return
|
|
*/
|
|
public JFreeChart build(){
|
|
XYSeriesCollection collection = new XYSeriesCollection();
|
|
for(XYSeries s : seriesMap.values()){
|
|
collection.addSeries(s);
|
|
}
|
|
JFreeChart chart = ChartFactory.createXYLineChart(chartName, xDomain, yDomain, collection, PlotOrientation.VERTICAL, true, true, false);
|
|
XYPlot plot = chart.getXYPlot();
|
|
plot.setBackgroundPaint(Color.WHITE);
|
|
plot.setDomainGridlinePaint(Color.LIGHT_GRAY);
|
|
plot.setRangeGridlinePaint(Color.LIGHT_GRAY);
|
|
NumberAxis yAxis = (NumberAxis) plot.getRangeAxis();
|
|
Range rangeBounds = collection.getRangeBounds(true);
|
|
double upper = Math.min(rangeBounds.getUpperBound(), rangeBounds.getLowerBound()*5);
|
|
if(upper == 0.0){ upper = 10000; }
|
|
if(rangeBounds.getLowerBound() == upper){
|
|
yAxis.setRangeWithMargins(rangeBounds.getLowerBound()-rangeBounds.getLowerBound()*.1,upper+upper*.1);
|
|
}
|
|
else{
|
|
yAxis.setRangeWithMargins(rangeBounds.getLowerBound(),upper);
|
|
}
|
|
return chart;
|
|
}
|
|
|
|
}
|