public Chart(String title, String timeAxis, String valueAxis, TimeSeries data) {
try {
// Build the datasets
dataset.addSeries(data);
// Create the chart
JFreeChart chart =
ChartFactory.createTimeSeriesChart(
title, timeAxis, valueAxis, dataset, true, true, false);
// Setup the appearance of the chart
chart.setBackgroundPaint(Color.white);
XYPlot plot = chart.getXYPlot();
plot.setBackgroundPaint(Color.lightGray);
plot.setDomainGridlinePaint(Color.white);
plot.setRangeGridlinePaint(Color.white);
plot.setAxisOffset(new RectangleInsets(UnitType.ABSOLUTE, 5.0, 5.0, 5.0, 5.0));
plot.setDomainCrosshairVisible(true);
plot.setRangeCrosshairVisible(true);
// Tell the chart how we would like dates to read
DateAxis axis = (DateAxis) plot.getDomainAxis();
axis.setDateFormatOverride(new SimpleDateFormat("EEE HH"));
this.add(new ChartPanel(chart));
} catch (Exception e) {
e.printStackTrace();
}
}
public Chart(String filename) {
try {
// Get Stock Symbol
this.stockSymbol = filename.substring(0, filename.indexOf('.'));
// Create time series
TimeSeries open = new TimeSeries("Open Price", Day.class);
TimeSeries close = new TimeSeries("Close Price", Day.class);
TimeSeries high = new TimeSeries("High", Day.class);
TimeSeries low = new TimeSeries("Low", Day.class);
TimeSeries volume = new TimeSeries("Volume", Day.class);
BufferedReader br = new BufferedReader(new FileReader(filename));
String key = br.readLine();
String line = br.readLine();
while (line != null && !line.startsWith("<!--")) {
StringTokenizer st = new StringTokenizer(line, ",", false);
Day day = getDay(st.nextToken());
double openValue = Double.parseDouble(st.nextToken());
double highValue = Double.parseDouble(st.nextToken());
double lowValue = Double.parseDouble(st.nextToken());
double closeValue = Double.parseDouble(st.nextToken());
long volumeValue = Long.parseLong(st.nextToken());
// Add this value to our series'
open.add(day, openValue);
close.add(day, closeValue);
high.add(day, highValue);
low.add(day, lowValue);
// Read the next day
line = br.readLine();
}
// Build the datasets
dataset.addSeries(open);
dataset.addSeries(close);
dataset.addSeries(low);
dataset.addSeries(high);
datasetOpenClose.addSeries(open);
datasetOpenClose.addSeries(close);
datasetHighLow.addSeries(high);
datasetHighLow.addSeries(low);
JFreeChart summaryChart = buildChart(dataset, "Summary", true);
JFreeChart openCloseChart = buildChart(datasetOpenClose, "Open/Close Data", false);
JFreeChart highLowChart = buildChart(datasetHighLow, "High/Low Data", true);
JFreeChart highLowDifChart =
buildDifferenceChart(datasetHighLow, "High/Low Difference Chart");
// Create this panel
this.setLayout(new GridLayout(2, 2));
this.add(new ChartPanel(summaryChart));
this.add(new ChartPanel(openCloseChart));
this.add(new ChartPanel(highLowChart));
this.add(new ChartPanel(highLowDifChart));
} catch (Exception e) {
e.printStackTrace();
}
}
public static void main(String[] args) {
String logFileName = null;
long samplingInterval = 500;
boolean saveToFile = false;
try {
for (int i = 0; i < args.length; i++) {
if (args[i].equals("-h") || args[i].equals("-help")) {
usage();
System.exit(0);
} else if (args[i].equals("-save")) {
saveToFile = false;
} else if (args[i].equals("-interval")) {
samplingInterval = Long.valueOf(args[++i]);
} else if (args[i].equals("-log")) {
logFileName = args[++i];
} else if (args[i].equals("-debug") || args[i].equals("-d")) {
System.setProperty("PharosMiddleware.debug", "true");
} else {
printErr("Unknown option: " + args[i]);
usage();
System.exit(1);
}
}
} catch (Exception e) {
e.printStackTrace();
usage();
System.exit(1);
}
if (logFileName == null) {
printErr("Must specify log file.");
usage();
System.exit(1);
}
print("Log file: " + logFileName);
print("Debug: " + (System.getProperty("PharosMiddleware.debug") != null));
try {
new VisualizeHeadingError(logFileName, samplingInterval, saveToFile);
} catch (Exception e) {
e.printStackTrace();
}
}
protected Day getDay(String date) {
try {
String[] st = date.split("-");
int year = Integer.parseInt(st[0]);
int month = Integer.parseInt(st[1]);
int day = Integer.parseInt(st[2]);
// Build a new Day
return new Day(day, month, year);
} catch (Exception e) {
e.printStackTrace();
}
return null;
}