path: root/src/org.gluster.storage.management.console/src/org/gluster/storage/management/console/utils/ChartUtil.java

/*******************************************************************************
 * Copyright (c) 2006-2011 Gluster, Inc. <http://www.gluster.com>
 * This file is part of Gluster Management Console.
 *
 * Gluster Management Console is free software; you can redistribute
 * it and/or modify it under the terms of the GNU General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * Gluster Management Console 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see
 * <http://www.gnu.org/licenses/>.
 *******************************************************************************/
package org.gluster.storage.management.console.utils;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.birt.chart.util.CDateTime;
import org.eclipse.jface.preference.IPreferenceStore;
import org.eclipse.swt.SWT;
import org.eclipse.swt.custom.CCombo;
import org.eclipse.swt.events.ModifyEvent;
import org.eclipse.swt.events.ModifyListener;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.GridLayout;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.ui.forms.events.HyperlinkAdapter;
import org.eclipse.ui.forms.events.HyperlinkEvent;
import org.eclipse.ui.forms.widgets.FormToolkit;
import org.eclipse.ui.forms.widgets.Hyperlink;
import org.gluster.storage.management.console.Activator;
import org.gluster.storage.management.console.preferences.PreferenceConstants;
import org.gluster.storage.management.core.constants.CoreConstants;
import org.gluster.storage.management.core.constants.GlusterConstants;
import org.gluster.storage.management.core.model.GlusterServer;
import org.gluster.storage.management.core.model.NetworkInterface;
import org.gluster.storage.management.core.model.ServerStats;
import org.gluster.storage.management.core.model.ServerStatsRow;

import com.ibm.icu.util.Calendar;

/**
 *
 */
public class ChartUtil {
	private static final ChartUtil instance = new ChartUtil();
	private static final int CHART_WIDTH = 350;
	private static final IPreferenceStore preferenceStore = Activator.getDefault().getPreferenceStore();

	private ChartUtil() {
	}

	public static ChartUtil getInstance() {
		return instance;
	}

	/**
	 * @param toolkit
	 * @param section
	 * @param stats
	 * @param dataColumnIndex
	 * @param unit
	 * @param timestampFormat
	 * @param listener
	 * @param maxValue
	 * @return The composite containing the various links related to the created chart
	 */
	public Composite createAreaChart(FormToolkit toolkit, Composite section, ServerStats stats, int dataColumnIndex,
			String unit, String timestampFormat, ChartPeriodLinkListener listener, double maxValue, int linkColumnCount) {
		if (stats == null) {
			toolkit.createLabel(section, "Server statistics not available. Try after some time!");
			return null;
		}

		List<Calendar> timestamps = new ArrayList<Calendar>();
		List<Double> data = new ArrayList<Double>();

		extractChartData(stats, timestamps, data, dataColumnIndex);

		if (timestamps.size() == 0) {
			toolkit.createLabel(section, "Server statistics not available!" + CoreConstants.NEWLINE
					+ "Check if all services are running properly " + CoreConstants.NEWLINE
					+ "on the cluster servers, or try after some time!");
			return null;
		}

		createAreaChart(section, timestamps.toArray(new Calendar[0]), data.toArray(new Double[0]), unit,
				timestampFormat, maxValue);

		// Calendar[] timestamps = new Calendar[] { new CDateTime(1000l*1310468100), new CDateTime(1000l*1310468400),
		// new CDateTime(1000l*1310468700),
		// new CDateTime(1000l*1310469000), new CDateTime(1000l*1310469300), new CDateTime(1000l*1310469600), new
		// CDateTime(1000l*1310469900),
		// new CDateTime(1000l*1310470200), new CDateTime(1000l*1310470500), new CDateTime(1000l*1310470800), new
		// CDateTime(1000l*1310471100),
		// new CDateTime(1000l*1310471400), new CDateTime(1000l*1310471700), new CDateTime(1000l*1310472000), new
		// CDateTime(1000l*1310472300),
		// new CDateTime(1000l*1310472600), new CDateTime(1000l*1310472900), new CDateTime(1000l*1310473200), new
		// CDateTime(1000l*1310473500),
		// new CDateTime(1000l*1310473800) };
		//
		// Double[] values = new Double[] { 10d, 11.23d, 17.92d, 18.69d, 78.62d, 89.11d, 92.43d, 89.31d, 57.39d, 18.46d,
		// 10.44d, 16.28d, 13.51d, 17.53d, 12.21, 20d, 21.43d, 16.45d, 14.86d, 15.27d };
		// createLineChart(section, timestamps, values, "%");
		Composite chartLinksComposite = createChartLinks(toolkit, section, linkColumnCount, listener);

		if (linkColumnCount == 5) {
			createNetworkInterfaceCombo(section, chartLinksComposite, toolkit,
					(NetworkChartPeriodLinkListener) listener);
		}
		return chartLinksComposite;
	}

	private ChartViewerComposite createAreaChart(Composite section, Calendar timestamps[], Double values[],
			String unit, String timestampFormat, double maxValue) {
		ChartViewerComposite chartViewerComposite = new ChartViewerComposite(section, SWT.NONE, timestamps, values,
				unit, timestampFormat, maxValue);
		GridData data = new GridData(SWT.FILL, SWT.FILL, false, false);
		data.widthHint = CHART_WIDTH;
		data.heightHint = 250;
		data.verticalAlignment = SWT.CENTER;
		chartViewerComposite.setLayoutData(data);
		return chartViewerComposite;
	}

	private void extractChartData(ServerStats stats, List<Calendar> timestamps, List<Double> data, int dataColumnIndex) {
		for (ServerStatsRow row : stats.getRows()) {
			Double cpuUsage = row.getUsageData().get(dataColumnIndex);
			if (!cpuUsage.isNaN()) {
				timestamps.add(new CDateTime(row.getTimestamp() * 1000));
				data.add(cpuUsage);
			}
		}
	}

	private Composite createChartLinks(FormToolkit toolkit, Composite section, int columnCount,
			ChartPeriodLinkListener listener) {
		GridLayout layout = new org.eclipse.swt.layout.GridLayout(columnCount, false);
		layout.marginBottom = 0;
		layout.marginTop = 0;
		layout.marginLeft = (CHART_WIDTH - (50 * columnCount)) / 2;
		Composite graphComposite = toolkit.createComposite(section, SWT.NONE);
		graphComposite.setLayout(layout);
		GridData data = new GridData(SWT.FILL, SWT.FILL, false, false);
		data.widthHint = CHART_WIDTH;
		graphComposite.setLayoutData(data);

		createStatsLink(toolkit, listener, graphComposite, "1 day", GlusterConstants.STATS_PERIOD_1DAY);
		createStatsLink(toolkit, listener, graphComposite, "1 week", GlusterConstants.STATS_PERIOD_1WEEK);
		createStatsLink(toolkit, listener, graphComposite, "1 month", GlusterConstants.STATS_PERIOD_1MONTH);
		createStatsLink(toolkit, listener, graphComposite, "1 year", GlusterConstants.STATS_PERIOD_1YEAR);

		return graphComposite;
	}

	private void createStatsLink(FormToolkit toolkit, ChartPeriodLinkListener listener, Composite parent, String label,
			String statsPeriod) {
		Hyperlink link1 = toolkit.createHyperlink(parent, label, SWT.NONE);
		link1.addHyperlinkListener(listener.getInstance(statsPeriod));
		if (listener.getStatsPeriod().equals(statsPeriod)) {
			link1.setEnabled(false);
		}
	}

	public abstract class ChartPeriodLinkListener extends HyperlinkAdapter {
		protected String statsPeriod;
		protected String unit;
		protected int columnCount;
		protected double maxValue;
		protected FormToolkit toolkit;
		protected String serverName;
		protected int dataColumnIndex;

		public String getStatsPeriod() {
			return this.statsPeriod;
		}

		public ChartPeriodLinkListener(String serverName, String statsPeriod, String unit, double maxValue,
				int columnCount, int dataColumnIndex, FormToolkit toolkit) {
			this.serverName = serverName;
			this.statsPeriod = statsPeriod;
			this.unit = unit;
			this.columnCount = columnCount;
			this.maxValue = maxValue;
			this.dataColumnIndex = dataColumnIndex;
			this.toolkit = toolkit;
		}

		public ChartPeriodLinkListener(String serverName, String statsPeriod, FormToolkit toolkit) {
			this.statsPeriod = statsPeriod;
			this.serverName = serverName;
			this.toolkit = toolkit;
		}

		@Override
		public void linkActivated(HyperlinkEvent e) {
			super.linkActivated(e);
			updatePreference(serverName);
		}

		public abstract ChartPeriodLinkListener getInstance(String statsPeriod);

		protected abstract void updatePreference(String serverName);
	}

	public class CpuChartPeriodLinkListener extends ChartPeriodLinkListener {
		public CpuChartPeriodLinkListener(String serverName, String statsPeriod, FormToolkit toolkit) {
			super(serverName, statsPeriod, toolkit);
		}

		private CpuChartPeriodLinkListener(String serverName, String statsPeriod, String unit, double maxValue,
				int columnCount, int dataColumnIndex, FormToolkit toolkit) {
			super(serverName, statsPeriod, unit, maxValue, columnCount, dataColumnIndex, toolkit);
		}

		@Override
		protected void updatePreference(String serverName) {
			if (serverName == null) {
				preferenceStore.setValue(PreferenceConstants.P_CPU_AGGREGATED_CHART_PERIOD, statsPeriod);
			} else {
				preferenceStore.setValue(PreferenceConstants.P_CPU_CHART_PERIOD, statsPeriod);
			}
		}

		@Override
		public ChartPeriodLinkListener getInstance(String statsPeriod) {
			return new CpuChartPeriodLinkListener(serverName, statsPeriod, "%", 100, 4, dataColumnIndex, toolkit);
		}
	}

	public class MemoryChartPeriodLinkListener extends ChartPeriodLinkListener {
		public MemoryChartPeriodLinkListener(String serverName, String statsPeriod, FormToolkit toolkit) {
			super(serverName, statsPeriod, toolkit);
		}

		private MemoryChartPeriodLinkListener(String serverName, String statsPeriod, String unit, double maxValue,
				int columnCount, int dataColumnIndex, FormToolkit toolkit) {
			super(serverName, statsPeriod, unit, maxValue, columnCount, dataColumnIndex, toolkit);
		}

		@Override
		protected void updatePreference(String serverName) {
			preferenceStore.setValue(PreferenceConstants.P_MEM_CHART_PERIOD, statsPeriod);
		}

		@Override
		public ChartPeriodLinkListener getInstance(String statsPeriod) {
			return new MemoryChartPeriodLinkListener(serverName, statsPeriod, "%", 100, 4, dataColumnIndex, toolkit);
		}
	}

	public class NetworkChartPeriodLinkListener extends ChartPeriodLinkListener {
		private GlusterServer server;

		public NetworkChartPeriodLinkListener(GlusterServer server, String statsPeriod, FormToolkit toolkit) {
			super(server == null ? null : server.getName(), statsPeriod, toolkit);
			this.setServer(server);
		}

		private NetworkChartPeriodLinkListener(GlusterServer server, String statsPeriod, String unit, double maxValue,
				int columnCount, int dataColumnIndex, FormToolkit toolkit) {
			super(server == null ? null : server.getName(), statsPeriod, unit, maxValue, columnCount, dataColumnIndex,
					toolkit);
			this.setServer(server);
		}

		@Override
		protected void updatePreference(String serverName) {
			if (serverName == null) {
				preferenceStore.setValue(PreferenceConstants.P_NETWORK_AGGREGATED_CHART_PERIOD, statsPeriod);
			} else {
				preferenceStore.setValue(PreferenceConstants.P_NETWORK_CHART_PERIOD, statsPeriod);
			}
		}

		@Override
		public ChartPeriodLinkListener getInstance(String statsPeriod) {
			// if serverName is null, it means we are showing aggregated stats - so the "network interface" combo will
			// not be shown in the "links" composite.
			int columnCount = (serverName == null ? 4 : 5);
			return new NetworkChartPeriodLinkListener(server, statsPeriod, "KiB/s", -1d, columnCount, dataColumnIndex,
					toolkit);
		}

		public void setServer(GlusterServer server) {
			this.server = server;
		}

		public GlusterServer getServer() {
			return server;
		}
	}

	public void createNetworkInterfaceCombo(final Composite section, final Composite graphComposite,
			final FormToolkit toolkit, final NetworkChartPeriodLinkListener networkChartPeriodLinkListener) {
		final GlusterServer server = networkChartPeriodLinkListener.getServer();
		final CCombo interfaceCombo = new CCombo(graphComposite, SWT.DROP_DOWN | SWT.READ_ONLY | SWT.BORDER | SWT.FLAT);
		final List<NetworkInterface> niList = server.getNetworkInterfaces();
		final String[] interfaces = new String[niList.size()];

		for (int i = 0; i < interfaces.length; i++) {
			interfaces[i] = niList.get(i).getName();
		}
		interfaceCombo.setItems(interfaces);
		interfaceCombo.select(0);
		interfaceCombo.addModifyListener(new ModifyListener() {
			@Override
			public void modifyText(ModifyEvent e) {
				preferenceStore.setValue(PreferenceConstants.P_DEFAULT_NETWORK_INTERFACE_PFX + server.getName(), interfaces[interfaceCombo.getSelectionIndex()]);
			}
		});
	}

	public void refreshChartSection(FormToolkit toolkit, Composite section, ServerStats stats, String statsPeriod,
			String unit, double maxValue, int columnCount, ChartPeriodLinkListener linkListener, int dataColumnIndex) {
		GUIHelper.getInstance().clearSection(section);
		createAreaChart(toolkit, section, stats, dataColumnIndex, unit, getTimestampFormatForPeriod(statsPeriod),
				linkListener, maxValue, columnCount);
		section.layout();
	}

	public String getTimestampFormatForPeriod(String statsPeriod) {
		if (statsPeriod.equals(GlusterConstants.STATS_PERIOD_1DAY)) {
			return "HH:mm";
		} else if (statsPeriod.equals(GlusterConstants.STATS_PERIOD_1WEEK)) {
			return "dd-MMM HH:mm";
		} else {
			return "dd-MMM";
		}
	}
}