// Plotter
// Chris Wilcox
// wilcox
// 10/19/2012
// CS160
import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Font;
import java.awt.GradientPaint;
import java.awt.Point;
import java.awt.RadialGradientPaint;
import java.awt.geom.Point2D;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.StandardChartTheme;
import org.jfree.chart.axis.AxisLocation;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.plot.CategoryPlot;
import org.jfree.chart.plot.PiePlot;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.renderer.category.LineAndShapeRenderer;
import org.jfree.chart.title.TextTitle;
import org.jfree.data.category.CategoryDataset;
import org.jfree.data.category.DefaultCategoryDataset;
import org.jfree.data.general.DatasetUtilities;
import org.jfree.data.general.DefaultPieDataset;
import org.jfree.ui.ApplicationFrame;
import org.jfree.ui.HorizontalAlignment;
import org.jfree.ui.RectangleInsets;
import org.jfree.ui.RefineryUtilities;
// Pie charts, bar charts, and line graphs
public class Plotter extends ApplicationFrame {
// Serialization identifier
private static final long serialVersionUID = 1L;
// Set a theme using shadow generator
{
ChartFactory.setChartTheme(new StandardChartTheme("JFree/Shadow", true));
}
// Enumeration
public enum eType {
PIECHART, BARCHART, LINEGRAPH
}
// Chart title
String chartTitle;
// Instance data
int pieCount;
int barCount;
int lineCount;
// Static allocation!
double[] pieData;
String[] pieLabels;
double[][] barData;
double[][] lineData;
public Plotter(String title) {
super(title);
chartTitle = title;
}
public void drawGraph(eType type)
{
JFreeChart chart;
switch (type)
{
case PIECHART:
chart = createPieChart();
break;
case BARCHART:
chart = createBarChart();
break;
case LINEGRAPH:
default:
chart = createLineGraph();
}
// add the chart to a panel...
ChartPanel chartPanel = new ChartPanel(chart);
chartPanel.setPreferredSize(new java.awt.Dimension(600, 400));
chart.setPadding(new RectangleInsets(4, 8, 2, 2));
setContentPane(chartPanel);
pack();
RefineryUtilities.positionFrameRandomly(this);
setVisible(true);
}
// Initialize pie chart data
public void pieChartData(int count, double data[]) {
// Copy data from user
pieCount = count;
pieData = new double[count];
for (int i = 0; i < count; i++) {
pieData[i] = data[i];
}
}
// Initialize pie chart labels
public void pieChartLabels(int count, String labels[]) {
// Copy labels from user
pieCount = count;
pieLabels = new String[count];
for (int i = 0; i < count; i++) {
pieLabels[i] = labels[i];
}
}
// Create pie chart
private JFreeChart createPieChart()
{
// Fill in data set
DefaultPieDataset dataset = new DefaultPieDataset();
for (int i = 0; i < pieCount; i++)
dataset.setValue(pieLabels[i], pieData[i]);
JFreeChart chart = ChartFactory.createPieChart(chartTitle,
dataset, // data
false, // no legend
true, // tooltips
false // no URL generation
);
// Set a custom background
chart.setBackgroundPaint(new GradientPaint(new Point(0, 0),
new Color(20, 20, 20), new Point(400, 200), Color.DARK_GRAY));
// Customize the title position and font
TextTitle title = chart.getTitle();
title.setHorizontalAlignment(HorizontalAlignment.LEFT);
title.setPaint(new Color(240, 240, 240));
title.setFont(new Font("Arial", Font.BOLD, 26));
PiePlot plot = (PiePlot) chart.getPlot();
plot.setBackgroundPaint(null);
plot.setOutlineVisible(false);
// Use gradients for section colors
plot.setSectionPaint(pieLabels[0], createGradientPaint(new Color(200, 200, 255), Color.BLUE));
plot.setSectionPaint(pieLabels[1], createGradientPaint(new Color(255, 200, 200), Color.RED));
plot.setSectionPaint(pieLabels[2], createGradientPaint(new Color(200, 255, 200), Color.GREEN));
plot.setSectionPaint(pieLabels[3], createGradientPaint(new Color(200, 255, 200), Color.YELLOW));
plot.setBaseSectionOutlinePaint(Color.WHITE);
plot.setSectionOutlinesVisible(true);
plot.setBaseSectionOutlineStroke(new BasicStroke(2.0f));
// Customize the section label appearance
plot.setLabelFont(new Font("Courier New", Font.BOLD, 20));
plot.setLabelLinkPaint(Color.WHITE);
plot.setLabelLinkStroke(new BasicStroke(2.0f));
plot.setLabelOutlineStroke(null);
plot.setLabelPaint(Color.WHITE);
plot.setLabelBackgroundPaint(null);
return chart;
}
// Set bar chart data
public void barChartData(int series, int count, double data[]) {
// Copy data from user
barCount = count;
if (series == 0) barData = new double[2][count];
for (int i = 0; i < count; i++) {
barData[series][i] = data[i];
}
}
// Create bar chart
private JFreeChart createBarChart()
{
// Fill in data set
CategoryDataset dataset;
dataset = DatasetUtilities.createCategoryDataset("Series ", "Factor ", barData);
JFreeChart chart = ChartFactory.createBarChart(
chartTitle, // chart title
"Category", // domain axis label
"Score (%)", // range axis label
dataset, // data
PlotOrientation.HORIZONTAL, // orientation
true, // include legend
true,
false
);
// Set the background color for the chart...
chart.setBackgroundPaint(Color.lightGray);
// Get a reference to the plot
CategoryPlot plot = chart.getCategoryPlot();
plot.setRangeAxisLocation(AxisLocation.BOTTOM_OR_LEFT);
// Change the tick unit selection to integer units only
NumberAxis rangeAxis = (NumberAxis) plot.getRangeAxis();
rangeAxis.setRange(0.0, 100.0);
rangeAxis.setStandardTickUnits(NumberAxis.createIntegerTickUnits());
return chart;
}
// Set line graph data
public void lineGraphData(int series, int count, double data[]) {
// Copy data from user
lineCount = count;
if (series == 0) lineData = new double[3][count];
for (int i = 0; i < count; i++) {
lineData[series][i] = data[i];
}
}
// Create line graph
private JFreeChart createLineGraph()
{
DefaultCategoryDataset dataset = new DefaultCategoryDataset();
for (int i = 0; i < 3; i++) // series
{
for (int j = 0; j < lineCount; j++) // data
{
String series = "Series " + i;
String type = "Type " + j;
dataset.addValue(lineData[i][j], series, type);
}
}
// create the chart...
final JFreeChart chart = ChartFactory.createLineChart(
chartTitle, // chart title
"Type", // domain axis label
"Value", // range axis label
dataset, // data
PlotOrientation.VERTICAL, // orientation
true, // include legend
true, // tooltips
false // urls
);
// Chart customization
chart.setBackgroundPaint(Color.white);
CategoryPlot plot = (CategoryPlot) chart.getPlot();
plot.setBackgroundPaint(Color.lightGray);
plot.setRangeGridlinePaint(Color.white);
// Customize the range axis
NumberAxis rangeAxis = (NumberAxis) plot.getRangeAxis();
rangeAxis.setStandardTickUnits(NumberAxis.createIntegerTickUnits());
rangeAxis.setAutoRangeIncludesZero(true);
// Customize the renderer
LineAndShapeRenderer renderer = (LineAndShapeRenderer) plot.getRenderer();
renderer.setSeriesStroke(
0, new BasicStroke(
2.0f, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND,
1.0f, new float[] {10.0f, 6.0f}, 0.0f
)
);
renderer.setSeriesStroke(
1, new BasicStroke(
2.0f, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND,
1.0f, new float[] {6.0f, 6.0f}, 0.0f
)
);
renderer.setSeriesStroke(
2, new BasicStroke(
2.0f, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND,
1.0f, new float[] {2.0f, 6.0f}, 0.0f
)
);
return chart;
}
// Utility method for creating gradient paints.
private static RadialGradientPaint createGradientPaint(Color c1, Color c2)
{
Point2D center = new Point2D.Float(0, 0);
float radius = 200;
float[] dist = {0.0f, 1.0f};
return new RadialGradientPaint(center, radius, dist,
new Color[] {c1, c2});
}
}
