timeaxis.java

jrobin,使用纯java实现的RRD数据库,使用RRD数据库来统计数据.

/* ============================================================
 * JRobin : Pure java implementation of RRDTool's functionality
 * ============================================================
 *
 * Project Info:  http://www.jrobin.org
 * Project Lead:  Sasa Markovic ([e-mail])
 *
 * Developers:    Sasa Markovic ([e-mail])
 *
 *
 * (C) Copyright 2003-2005, by Sasa Markovic.
 *
 * This library is free software; you can redistribute it and/or modify it under the terms
 * of the GNU Lesser General Public License as published by the Free Software Foundation;
 * either version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License along with this
 * library; if not, write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330,
 * Boston, MA 02111-1307, USA.
 */
package org.jrobin.graph;

import java.awt.*;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.Locale;

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 TimeAxisSetting tickSetting;
	private RrdGraph rrdGraph;
	private double secPerPix;
	private Calendar calendar;

	TimeAxis(RrdGraph rrdGraph) {
		this.rrdGraph = rrdGraph;
		if (rrdGraph.im.xsize > 0) {
			this.secPerPix = (rrdGraph.im.end - rrdGraph.im.start) / rrdGraph.im.xsize;
		}
		this.calendar = Calendar.getInstance(Locale.getDefault());
		this.calendar.setFirstDayOfWeek(rrdGraph.gdef.firstDayOfWeek);
	}

	void draw() {
		chooseTickSettings();
		if (tickSetting == null) {
			return;
		}
		drawMinor();
		drawMajor();
		drawLabels();
	}

	private void drawMinor() {
		if (!rrdGraph.gdef.noMinorGrid) {
			adjustStartingTime(tickSetting.minorUnit, tickSetting.minorUnitCount);
			Paint color = rrdGraph.gdef.colors[COLOR_GRID];
			int y0 = rrdGraph.im.yorigin, y1 = y0 - rrdGraph.im.ysize;
			for (int status = getTimeShift(); status <= 0; status = getTimeShift()) {
				if (status == 0) {
					long time = calendar.getTime().getTime() / 1000L;
					int x = rrdGraph.mapper.xtr(time);
					rrdGraph.worker.drawLine(x, y0 - 1, x, y0 + 1, color, TICK_STROKE);
					rrdGraph.worker.drawLine(x, y0, x, y1, color, GRID_STROKE);
				}
				findNextTime(tickSetting.minorUnit, tickSetting.minorUnitCount);
			}
		}
	}

	private void drawMajor() {
		adjustStartingTime(tickSetting.majorUnit, tickSetting.majorUnitCount);
		Paint color = rrdGraph.gdef.colors[COLOR_MGRID];
		int y0 = rrdGraph.im.yorigin, y1 = y0 - rrdGraph.im.ysize;
		for (int status = getTimeShift(); status <= 0; status = getTimeShift()) {
			if (status == 0) {
				long time = calendar.getTime().getTime() / 1000L;
				int x = rrdGraph.mapper.xtr(time);
				rrdGraph.worker.drawLine(x, y0 - 2, x, y0 + 2, color, TICK_STROKE);
				rrdGraph.worker.drawLine(x, y0, x, y1, color, GRID_STROKE);
			}
			findNextTime(tickSetting.majorUnit, tickSetting.majorUnitCount);
		}
	}

	private void drawLabels() {
		// escape strftime like format string
		String labelFormat = tickSetting.format.replaceAll("([^%]|^)%([^%t])", "$1%t$2");
		Font font = rrdGraph.gdef.smallFont;
		Paint color = rrdGraph.gdef.colors[COLOR_FONT];
		adjustStartingTime(tickSetting.labelUnit, tickSetting.labelUnitCount);
		int y = rrdGraph.im.yorigin + (int) rrdGraph.worker.getFontHeight(font) + 2;
		for (int status = getTimeShift(); status <= 0; status = getTimeShift()) {
			String label = formatLabel(labelFormat, calendar.getTime());
			long time = calendar.getTime().getTime() / 1000L;
			int x1 = rrdGraph.mapper.xtr(time);
			int x2 = rrdGraph.mapper.xtr(time + tickSetting.labelSpan);
			int labelWidth = (int) rrdGraph.worker.getStringWidth(label, font);
			int x = x1 + (x2 - x1 - labelWidth) / 2;
			if (x >= rrdGraph.im.xorigin && x + labelWidth <= rrdGraph.im.xorigin + rrdGraph.im.xsize) {
				rrdGraph.worker.drawString(label, x, y, font, color);
			}
			findNextTime(tickSetting.labelUnit, tickSetting.labelUnitCount);
		}
	}

	private static String formatLabel(String format, Date date) {
		if (format.contains("%")) {
			// strftime like format string
			return String.format(format, date);
		}
		else {
			// simple date format
			return new SimpleDateFormat(format).format(date);
		}
	}

	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 < rrdGraph.im.start) ? -1 : (time > rrdGraph.im.end) ? +1 : 0;
	}

	private void adjustStartingTime(int timeUnit, int timeUnitCount) {
		calendar.setTime(new Date(rrdGraph.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 (rrdGraph.gdef.timeAxisSetting != null) {
			tickSetting = new TimeAxisSetting(rrdGraph.gdef.timeAxisSetting);
		}
		else {
			for (int i = 0; tickSettings[i].secPerPix >= 0 && secPerPix > tickSettings[i].secPerPix; i++) {
				tickSetting = tickSettings[i];
			}
		}
	}

}