private JFreeChart display(
String title, int height, int width, String category, List<Report> toolResults) {
JFrame f = new JFrame(title);
f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
// averages
ArrayList<Double> averageFalseRates = new ArrayList<Double>();
ArrayList<Double> averageTrueRates = new ArrayList<Double>();
double averageFalseRate = 0;
double averageTrueRate = 0;
XYSeriesCollection dataset = new XYSeriesCollection();
XYSeries series = new XYSeries("Scores");
for (int i = 0; i < toolResults.size(); i++) {
Report toolReport = toolResults.get(i);
OverallResult overallResults = toolReport.getOverallResults().getResults(category);
series.add(
overallResults.getFalsePositiveRate() * 100, overallResults.getTruePositiveRate() * 100);
if (toolReport.isCommercial()) {
averageFalseRates.add(overallResults.getFalsePositiveRate());
averageTrueRates.add(overallResults.getTruePositiveRate());
}
}
for (double d : averageFalseRates) {
averageFalseRate += d;
}
averageFalseRate = averageFalseRate / averageFalseRates.size();
for (double d : averageTrueRates) {
averageTrueRate += d;
}
averageTrueRate = averageTrueRate / averageTrueRates.size();
series.add(averageFalseRate * 100, averageTrueRate * 100);
dataset.addSeries(series);
afr = averageFalseRate;
atr = averageTrueRate;
chart =
ChartFactory.createScatterPlot(
title,
"False Positive Rate",
"True Positive Rate",
dataset,
PlotOrientation.VERTICAL,
true,
true,
false);
String fontName = "Arial";
DecimalFormat pctFormat = new DecimalFormat("0'%'");
theme = (StandardChartTheme) org.jfree.chart.StandardChartTheme.createJFreeTheme();
theme.setExtraLargeFont(new Font(fontName, Font.PLAIN, 24)); // title
theme.setLargeFont(new Font(fontName, Font.PLAIN, 20)); // axis-title
theme.setRegularFont(new Font(fontName, Font.PLAIN, 16));
theme.setSmallFont(new Font(fontName, Font.PLAIN, 12));
theme.setRangeGridlinePaint(Color.decode("#C0C0C0"));
theme.setPlotBackgroundPaint(Color.white);
theme.setChartBackgroundPaint(Color.white);
theme.setGridBandPaint(Color.red);
theme.setAxisOffset(new RectangleInsets(0, 0, 0, 0));
theme.setBarPainter(new StandardBarPainter());
theme.setAxisLabelPaint(Color.decode("#666666"));
theme.apply(chart);
XYPlot xyplot = chart.getXYPlot();
NumberAxis rangeAxis = (NumberAxis) xyplot.getRangeAxis();
NumberAxis domainAxis = (NumberAxis) xyplot.getDomainAxis();
xyplot.setOutlineVisible(true);
rangeAxis.setRange(-5, 109.99);
rangeAxis.setNumberFormatOverride(pctFormat);
rangeAxis.setTickLabelPaint(Color.decode("#666666"));
rangeAxis.setMinorTickCount(5);
rangeAxis.setTickUnit(new NumberTickUnit(10));
rangeAxis.setAxisLineVisible(true);
rangeAxis.setMinorTickMarksVisible(true);
rangeAxis.setTickMarksVisible(true);
rangeAxis.setLowerMargin(10);
rangeAxis.setUpperMargin(10);
xyplot.setRangeGridlineStroke(new BasicStroke());
xyplot.setRangeGridlinePaint(Color.lightGray);
xyplot.setRangeMinorGridlinePaint(Color.decode("#DDDDDD"));
xyplot.setRangeMinorGridlinesVisible(true);
domainAxis.setRange(-5, 105);
domainAxis.setNumberFormatOverride(pctFormat);
domainAxis.setTickLabelPaint(Color.decode("#666666"));
domainAxis.setMinorTickCount(5);
domainAxis.setTickUnit(new NumberTickUnit(10));
domainAxis.setAxisLineVisible(true);
domainAxis.setTickMarksVisible(true);
domainAxis.setMinorTickMarksVisible(true);
domainAxis.setLowerMargin(10);
domainAxis.setUpperMargin(10);
xyplot.setDomainGridlineStroke(new BasicStroke());
xyplot.setDomainGridlinePaint(Color.lightGray);
xyplot.setDomainMinorGridlinePaint(Color.decode("#DDDDDD"));
xyplot.setDomainMinorGridlinesVisible(true);
chart.setTextAntiAlias(true);
chart.setAntiAlias(true);
chart.removeLegend();
chart.setPadding(new RectangleInsets(20, 20, 20, 20));
xyplot.getRenderer().setSeriesPaint(0, Color.decode("#4572a7"));
// // setup item labels
// XYItemRenderer renderer = xyplot.getRenderer();
// Shape circle = new Ellipse2D.Float(-2.0f, -2.0f, 7.0f, 7.0f);
// for ( int i = 0; i < dataset.getSeriesCount(); i++ ) {
// renderer.setSeriesShape(i, circle);
// renderer.setSeriesPaint(i, Color.blue);
// String label = ""+((String)dataset.getSeries(i).getKey());
// int idx = label.indexOf( ':');
// label = label.substring( 0, idx );
// StandardXYItemLabelGenerator generator = new StandardXYItemLabelGenerator(label);
// renderer.setSeriesItemLabelGenerator(i, generator);
// renderer.setSeriesItemLabelsVisible(i, true);
// ItemLabelPosition position = new ItemLabelPosition(ItemLabelAnchor.OUTSIDE12,
// TextAnchor.BASELINE_CENTER );
// renderer.setSeriesPositiveItemLabelPosition(i, position);
// }
makeDataLabels(category, toolResults, xyplot);
makeLegend(category, toolResults, 57, 48, dataset, xyplot);
Stroke dashed =
new BasicStroke(1, BasicStroke.CAP_BUTT, BasicStroke.JOIN_BEVEL, 0, new float[] {6, 3}, 0);
for (XYDataItem item : (List<XYDataItem>) series.getItems()) {
double x = item.getX().doubleValue();
double y = item.getY().doubleValue();
double z = (x + y) / 2;
XYLineAnnotation score = new XYLineAnnotation(x, y, z, z, dashed, Color.blue);
xyplot.addAnnotation(score);
}
// // put legend inside plot
// LegendTitle lt = new LegendTitle(xyplot);
// lt.setItemFont(theme.getSmallFont());
// lt.setPosition(RectangleEdge.RIGHT);
// lt.setItemFont(theme.getSmallFont());
// XYTitleAnnotation ta = new XYTitleAnnotation(.7, .55, lt, RectangleAnchor.TOP_LEFT);
// ta.setMaxWidth(0.48);
// xyplot.addAnnotation(ta);
// draw guessing line
XYLineAnnotation guessing = new XYLineAnnotation(-5, -5, 105, 105, dashed, Color.red);
xyplot.addAnnotation(guessing);
XYPointerAnnotation worse =
makePointer(75, 5, "Worse than guessing", TextAnchor.TOP_CENTER, 90);
xyplot.addAnnotation(worse);
XYPointerAnnotation better =
makePointer(25, 100, "Better than guessing", TextAnchor.BOTTOM_CENTER, 270);
xyplot.addAnnotation(better);
XYTextAnnotation stroketext =
new XYTextAnnotation(" Random Guess", 88, 107);
stroketext.setTextAnchor(TextAnchor.CENTER_RIGHT);
stroketext.setBackgroundPaint(Color.white);
stroketext.setPaint(Color.red);
stroketext.setFont(theme.getRegularFont());
xyplot.addAnnotation(stroketext);
XYLineAnnotation strokekey = new XYLineAnnotation(58, 107, 68, 107, dashed, Color.red);
xyplot.setBackgroundPaint(Color.white);
xyplot.addAnnotation(strokekey);
ChartPanel cp =
new ChartPanel(
chart, height, width, 400, 400, 1200, 1200, false, false, false, false, false, false);
f.add(cp);
f.pack();
f.setLocationRelativeTo(null);
// f.setVisible(true);
return chart;
}