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();
}
}
private void addData(
JFreeChart chart,
DSLAMSource source,
Timeinterval time,
Serializable filter,
Properties properties,
Map<String, Object> colorScheme,
Properties outputProperties)
throws IOException {
if (chart != null) {
PerformanceCounterDataCollection cData = null;
long dataSize = 0;
try {
cData =
getPerformanceCounterDataCollection(source, time, filter, properties, outputProperties);
Collection<PerformanceCounterData> data = null;
Date d0 = null; // first timestamp for data
Date d1 = null; // last timestamp for data
RegularTimePeriod time0 = null;
if (cData != null) {
data = cData.getData();
{
int l = data.size();
outputProperties.setProperty("data.count", Integer.toString(l));
}
if (data != null && data.size() > 0) {
dataSize = data.size();
// Set 'd0':
{
// TODO: Avoid assumption "data is sorted"!
PerformanceCounterData e = data.iterator().next();
d0 = e.getTimestamp();
}
// Set 'd1':
{
// TODO: Avoid assumption "data is sorted"!
for (PerformanceCounterData e : data) {
d1 = e.getTimestamp();
}
}
time0 = createRegularTimePeriod(d0);
}
}
XYPlot plot = chart.getXYPlot();
// Set the first dataset:
{
TimeSeriesCollection dataset = new TimeSeriesCollection();
List<Paint> seriesPaint = new ArrayList<Paint>();
List<Stroke> seriesStroke = new ArrayList<Stroke>();
// Add series 'ES':
{
if (data != null && data.size() > 0) {
TimeSeries series = new TimeSeries("ES", time0.getClass());
for (PerformanceCounterData e : data) {
Date d = e.getTimestamp();
RegularTimePeriod timePeriod = createRegularTimePeriod(d);
// Add point:
{
int value = getES(e);
addOrUpdate(series, timePeriod, (double) value);
}
}
seriesPaint.add((Paint) colorScheme.get("color.counter.es"));
seriesStroke.add(STROKE_COUNTER_ES);
dataset.addSeries(series);
}
}
// Add series 'SES':
{
if (data != null && data.size() > 0) {
TimeSeries series = new TimeSeries("SES", time0.getClass());
for (PerformanceCounterData e : data) {
Date d = e.getTimestamp();
RegularTimePeriod timePeriod = createRegularTimePeriod(d);
// Add point:
{
int value = getSES(e);
addOrUpdate(series, timePeriod, (double) value);
}
}
seriesPaint.add((Paint) colorScheme.get("color.counter.ses"));
seriesStroke.add(STROKE_COUNTER_SES);
dataset.addSeries(series);
}
}
// Add series 'US':
{
if (data != null && data.size() > 0) {
TimeSeries series = new TimeSeries("US", time0.getClass());
for (PerformanceCounterData e : data) {
Date d = e.getTimestamp();
RegularTimePeriod timePeriod = createRegularTimePeriod(d);
// Add point:
{
int value = getUS(e);
addOrUpdate(series, timePeriod, (double) value);
}
}
seriesPaint.add((Paint) colorScheme.get("color.counter.us"));
seriesStroke.add(STROKE_COUNTER_US);
dataset.addSeries(series);
}
}
// superspeed
{
if (data != null && data.size() > 0) {
TimeSeries series = new TimeSeries("LEFTRS", time0.getClass());
for (PerformanceCounterData e : data) {
Date d = e.getTimestamp();
RegularTimePeriod timePeriod = createRegularTimePeriod(d);
{
int value = getPreviousLEFTRS(e);
addOrUpdate(series, timePeriod, (double) value);
}
}
seriesPaint.add((Paint) colorScheme.get("color.counter.previousleftrs"));
seriesStroke.add(STROKE_COUNTER_US);
dataset.addSeries(series);
}
}
// ends
{
if (data != null && data.size() > 0) {
boolean showLinearCurve = getShowLinearCurve();
if (showLinearCurve) {
// Add series for a linear curve:
{
TimeSeries series = new TimeSeries("Linear", time0.getClass());
long t0 = d0.getTime();
long t1 = d1.getTime();
if (t0 < t1) {
long timeX = 15 * 60 * 1000; // 15 minutes intervals
// TODO: Read length of intervals from obtained data!
double value0 = 0;
double value1 = 900;
// TODO: Read '900' from obtained data!
long timeDelta = t1 - t0;
double valueDelta = value1 - value0;
long t = t0;
int i = 0;
while (t <= t1) {
Date d = new Date(t);
RegularTimePeriod timePeriod = createRegularTimePeriod(d);
double value = value0 + ((t - t0) * valueDelta) / timeDelta;
// Add point:
{
addOrUpdate(series, timePeriod, (double) value);
}
t += timeX;
i += 1;
}
}
seriesPaint.add(Color.red);
seriesStroke.add(STROKE_COUNTER_DEFAULT);
dataset.addSeries(series);
}
}
}
}
setDefaultRenderer(chart, colorScheme, 0, seriesPaint, seriesStroke);
plot.setDataset(0, dataset);
plot.mapDatasetToRangeAxis(0, 0);
}
} finally {
if (cData != null) {
cData.dispose();
cData = null;
}
}
if (outputProperties != null) {
outputProperties.setProperty("data.count", Long.toString(dataSize));
}
}
}