package org.rrd4j.graph;
import java.awt.Font;
import java.awt.Paint;
import java.util.Calendar;
import java.util.Date;
class TimeAxis implements RrdGraphConstants {
private static final TimeAxisSetting[] tickSettings = {
new TimeAxisSetting(0, SECOND, 30, MINUTE, 5, MINUTE, 5, 0, HH_MM),
new TimeAxisSetting(2, MINUTE, 1, MINUTE, 5, MINUTE, 5, 0, HH_MM),
new TimeAxisSetting(5, MINUTE, 2, MINUTE, 10, MINUTE, 10, 0, HH_MM),
new TimeAxisSetting(10, MINUTE, 5, MINUTE, 20, MINUTE, 20, 0, HH_MM),
new TimeAxisSetting(30, MINUTE, 10, HOUR, 1, HOUR, 1, 0, HH_MM),
new TimeAxisSetting(60, MINUTE, 30, HOUR, 2, HOUR, 2, 0, HH_MM),
new TimeAxisSetting(180, HOUR, 1, HOUR, 6, HOUR, 6, 0, HH_MM),
new TimeAxisSetting(600, HOUR, 6, DAY, 1, DAY, 1, 24 * 3600, "EEE"),
new TimeAxisSetting(1800, HOUR, 12, DAY, 1, DAY, 2, 24 * 3600, "EEE"),
new TimeAxisSetting(3600, DAY, 1, WEEK, 1, WEEK, 1, 7 * 24 * 3600, "'Week 'w"),
new TimeAxisSetting(3 * 3600, WEEK, 1, MONTH, 1, WEEK, 2, 7 * 24 * 3600, "'Week 'w"),
new TimeAxisSetting(6 * 3600, MONTH, 1, MONTH, 1, MONTH, 1, 30 * 24 * 3600, "MMM"),
new TimeAxisSetting(48 * 3600, MONTH, 1, MONTH, 3, MONTH, 3, 30 * 24 * 3600, "MMM"),
new TimeAxisSetting(10 * 24 * 3600, YEAR, 1, YEAR, 1, YEAR, 1, 365 * 24 * 3600, "yy"),
new TimeAxisSetting(-1, MONTH, 0, MONTH, 0, MONTH, 0, 0, "")
};
private final ImageParameters im;
private final ImageWorker worker;
private final RrdGraphDef gdef;
private final Mapper mapper;
private TimeAxisSetting tickSetting;
private final double secPerPix;
private final Calendar calendar;
TimeAxis(RrdGraph rrdGraph) {
this.im = rrdGraph.im;
this.worker = rrdGraph.worker;
this.gdef = rrdGraph.gdef;
this.mapper = rrdGraph.mapper;
this.secPerPix = (im.end - im.start) / (double) im.xsize;
this.calendar = Calendar.getInstance(gdef.tz, gdef.locale);
this.calendar.setFirstDayOfWeek(gdef.firstDayOfWeek);
}
/**
* Used for test
* @param im
* @param worker
* @param gdef
* @param mapper
*/
TimeAxis(ImageParameters im, ImageWorker worker, RrdGraphDef gdef, Mapper mapper) {
this.im = im;
this.worker = worker;
this.gdef = gdef;
this.mapper = mapper;
this.secPerPix = (im.end - im.start) / (double) im.xsize;
this.calendar = Calendar.getInstance(gdef.tz, gdef.locale);
this.calendar.setFirstDayOfWeek(gdef.firstDayOfWeek);
}
void draw() {
chooseTickSettings();
// early return, avoid exceptions
if (tickSetting == null) return;
drawMinor();
drawMajor();
drawLabels();
}
private void drawMinor() {
if (!gdef.noMinorGrid) {
adjustStartingTime(tickSetting.minorUnit, tickSetting.minorUnitCount);
Paint color = gdef.colors[COLOR_GRID];
int y0 = im.yorigin, y1 = y0 - im.ysize;
for (int status = getTimeShift(); status <= 0; status = getTimeShift()) {
if (status == 0) {
long time = calendar.getTime().getTime() / 1000L;
int x = mapper.xtr(time);
worker.drawLine(x, y0 - 1, x, y0 + 1, color, gdef.tickStroke);
worker.drawLine(x, y0, x, y1, color, gdef.gridStroke);
}
findNextTime(tickSetting.minorUnit, tickSetting.minorUnitCount);
}
}
}
private void drawMajor() {
adjustStartingTime(tickSetting.majorUnit, tickSetting.majorUnitCount);
Paint color = gdef.colors[COLOR_MGRID];
int y0 = im.yorigin, y1 = y0 - im.ysize;
for (int status = getTimeShift(); status <= 0; status = getTimeShift()) {
if (status == 0) {
long time = calendar.getTime().getTime() / 1000L;
int x = mapper.xtr(time);
worker.drawLine(x, y0 - 2, x, y0 + 2, color, gdef.tickStroke);
worker.drawLine(x, y0, x, y1, color, gdef.gridStroke);
}
findNextTime(tickSetting.majorUnit, tickSetting.majorUnitCount);
}
}
private void drawLabels() {
Font font = gdef.getFont(FONTTAG_AXIS);
Paint color = gdef.colors[COLOR_FONT];
adjustStartingTime(tickSetting.labelUnit, tickSetting.labelUnitCount);
int y = im.yorigin + (int) worker.getFontHeight(font) + 2;
for (int status = getTimeShift(); status <= 0; status = getTimeShift()) {
String label = tickSetting.format.format(calendar, gdef.locale);
long time = calendar.getTime().getTime() / 1000L;
int x1 = mapper.xtr(time);
int x2 = mapper.xtr(time + tickSetting.labelSpan);
int labelWidth = (int) worker.getStringWidth(label, font);
int x = x1 + (x2 - x1 - labelWidth) / 2;
if (x >= im.xorigin && x + labelWidth <= im.xorigin + im.xsize) {
worker.drawString(label, x, y, font, color);
}
findNextTime(tickSetting.labelUnit, tickSetting.labelUnitCount);
}
}
private void findNextTime(int timeUnit, int timeUnitCount) {
switch (timeUnit) {
case SECOND:
calendar.add(Calendar.SECOND, timeUnitCount);
break;
case MINUTE:
calendar.add(Calendar.MINUTE, timeUnitCount);
break;
case HOUR:
calendar.add(Calendar.HOUR_OF_DAY, timeUnitCount);
break;
case DAY:
calendar.add(Calendar.DAY_OF_MONTH, timeUnitCount);
break;
case WEEK:
calendar.add(Calendar.DAY_OF_MONTH, 7 * timeUnitCount);
break;
case MONTH:
calendar.add(Calendar.MONTH, timeUnitCount);
break;
case YEAR:
calendar.add(Calendar.YEAR, timeUnitCount);
break;
}
}
private int getTimeShift() {
long time = calendar.getTime().getTime() / 1000L;
return (time < im.start) ? -1 : (time > im.end) ? +1 : 0;
}
private void adjustStartingTime(int timeUnit, int timeUnitCount) {
calendar.setTime(new Date(im.start * 1000L));
switch (timeUnit) {
case SECOND:
calendar.add(Calendar.SECOND, -(calendar.get(Calendar.SECOND) % timeUnitCount));
break;
case MINUTE:
calendar.set(Calendar.SECOND, 0);
calendar.add(Calendar.MINUTE, -(calendar.get(Calendar.MINUTE) % timeUnitCount));
break;
case HOUR:
calendar.set(Calendar.SECOND, 0);
calendar.set(Calendar.MINUTE, 0);
calendar.add(Calendar.HOUR_OF_DAY, -(calendar.get(Calendar.HOUR_OF_DAY) % timeUnitCount));
break;
case DAY:
calendar.set(Calendar.SECOND, 0);
calendar.set(Calendar.MINUTE, 0);
calendar.set(Calendar.HOUR_OF_DAY, 0);
break;
case WEEK:
calendar.set(Calendar.SECOND, 0);
calendar.set(Calendar.MINUTE, 0);
calendar.set(Calendar.HOUR_OF_DAY, 0);
int diffDays = calendar.get(Calendar.DAY_OF_WEEK) - calendar.getFirstDayOfWeek();
if (diffDays < 0) {
diffDays += 7;
}
calendar.add(Calendar.DAY_OF_MONTH, -diffDays);
break;
case MONTH:
calendar.set(Calendar.SECOND, 0);
calendar.set(Calendar.MINUTE, 0);
calendar.set(Calendar.HOUR_OF_DAY, 0);
calendar.set(Calendar.DAY_OF_MONTH, 1);
calendar.add(Calendar.MONTH, -(calendar.get(Calendar.MONTH) % timeUnitCount));
break;
case YEAR:
calendar.set(Calendar.SECOND, 0);
calendar.set(Calendar.MINUTE, 0);
calendar.set(Calendar.HOUR_OF_DAY, 0);
calendar.set(Calendar.DAY_OF_MONTH, 1);
calendar.set(Calendar.MONTH, 0);
calendar.add(Calendar.YEAR, -(calendar.get(Calendar.YEAR) % timeUnitCount));
break;
}
}
private void chooseTickSettings() {
if (gdef.timeAxisSetting != null) {
tickSetting = new TimeAxisSetting(gdef.timeAxisSetting);
}
else {
for (int i = 0; tickSettings[i].secPerPix >= 0 && secPerPix > tickSettings[i].secPerPix; i++) {
tickSetting = tickSettings[i];
}
}
if (gdef.timeLabelFormat != null) {
tickSetting = tickSetting.withLabelFormat(gdef.timeLabelFormat);
}
}
}