path: root/systemtap/org.eclipse.linuxtools.systemtap.graphingapi.ui/src/org/eclipse/linuxtools/systemtap/graphingapi/ui/datadisplay/DataGrid.java

/*******************************************************************************
 * Copyright (c) 2006 IBM Corporation.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     IBM Corporation - Jeff Briggs, Henry Hughes, Ryan Morse
 *******************************************************************************/

package org.eclipse.linuxtools.systemtap.graphingapi.ui.datadisplay;

import org.eclipse.jface.preference.IPreferenceStore;
import org.eclipse.jface.wizard.WizardDialog;
import org.eclipse.linuxtools.internal.systemtap.graphingapi.ui.GraphingAPIUIPlugin;
import org.eclipse.linuxtools.internal.systemtap.graphingapi.ui.Localization;
import org.eclipse.linuxtools.systemtap.graphingapi.core.datasets.IDataSet;
import org.eclipse.linuxtools.systemtap.graphingapi.core.datasets.IFilteredDataSet;
import org.eclipse.linuxtools.systemtap.graphingapi.core.filters.IDataSetFilter;
import org.eclipse.linuxtools.systemtap.graphingapi.ui.preferences.GraphingAPIPreferenceConstants;
import org.eclipse.linuxtools.systemtap.graphingapi.ui.wizards.dataset.DataSetFactory;
import org.eclipse.linuxtools.systemtap.graphingapi.ui.wizards.filter.AvailableFilterTypes;
import org.eclipse.linuxtools.systemtap.graphingapi.ui.wizards.filter.SelectFilterWizard;
import org.eclipse.linuxtools.systemtap.structures.IFormattingStyles;
import org.eclipse.linuxtools.systemtap.structures.StringFormatter;
import org.eclipse.linuxtools.systemtap.structures.listeners.IUpdateListener;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.MenuAdapter;
import org.eclipse.swt.events.MenuEvent;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.events.SelectionListener;
import org.eclipse.swt.graphics.Point;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Control;
import org.eclipse.swt.widgets.Event;
import org.eclipse.swt.widgets.Listener;
import org.eclipse.swt.widgets.Menu;
import org.eclipse.swt.widgets.MenuItem;
import org.eclipse.swt.widgets.Table;
import org.eclipse.swt.widgets.TableColumn;
import org.eclipse.swt.widgets.TableItem;
import org.eclipse.ui.IWorkbench;
import org.eclipse.ui.PlatformUI;



public class DataGrid implements IUpdateListener {
	public DataGrid(Composite composite, IDataSet set, int style) {
		prefs = GraphingAPIUIPlugin.getDefault().getPreferenceStore();
		manualResize = !prefs.getBoolean(GraphingAPIPreferenceConstants.P_AUTO_RESIZE);

		dataSet = set;
		filteredDataSet = (dataSet instanceof IFilteredDataSet)
								? (IFilteredDataSet)dataSet
								: DataSetFactory.createFilteredDataSet(dataSet);
		this.style = style;
		clickLocation = new Point(-1, -1);
		removedItems = 0;
		createPartControl(composite);
	}

	public void setLayoutData(Object data) {
		table.setLayoutData(data);
	}

	public IDataSet getDataSet() { return dataSet; }
	public Control getControl() { return table; }

	public void createPartControl(Composite parent) {
		table = new Table(parent, SWT.SINGLE | SWT.FULL_SELECTION);
		table.setHeaderVisible(true);
		table.setLinesVisible(true);
		table.getVerticalBar().setVisible(true);

		String[] names = dataSet.getTitles();
		TableColumn column = new TableColumn(table, SWT.LEFT);
		column.setText(Localization.getString("DataGrid.Row")); //$NON-NLS-1$
		column.pack();
		column.setMoveable(false);
		column.setResizable(false);

		columnFormat = new IFormattingStyles[names.length];
		for(int i=0; i<names.length; i++) {
			column = new TableColumn(table, SWT.LEFT);
			column.setText(names[i]);
			column.pack();
			column.setMoveable(true);

			columnFormat[i] = new StringFormatter();
		}

		table.setMenu(this.initMenus());

		table.addListener(SWT.MouseDown, new Listener() {
			@Override
			public void handleEvent(Event event) {
				clickLocation.x = event.x;
				clickLocation.y = event.y;
			}
		});
		handleUpdateEvent();
	}

	public Menu initMenus() {
		Menu menu = new Menu(table.getShell(), SWT.POP_UP);
		menu.addMenuListener(new MainMenuListener());

		Menu formatMenu = new Menu(menu);
		MenuItem item = new MenuItem(menu, SWT.CASCADE);
		item.setText(Localization.getString("DataGrid.FormatAs")); //$NON-NLS-1$
		item.setMenu(formatMenu);

			filterMenu = new Menu(menu);
			item = new MenuItem(menu, SWT.CASCADE);
			item.setText(Localization.getString("DataGrid.AddFilter")); //$NON-NLS-1$
			item.addSelectionListener(new AddFilterSelection());

			item = new MenuItem(menu, SWT.CASCADE);
			item.setText(Localization.getString("DataGrid.RemoveFilter")); //$NON-NLS-1$
			item.setMenu(filterMenu);

			IDataSetFilter[] filters = filteredDataSet.getFilters();
			if(null != filters) {
				for(int i=0; i<filters.length; i++) {
					item = new MenuItem(filterMenu, SWT.CASCADE);
					item.setText(AvailableFilterTypes.getFilterName(filters[i].getID()));
					item.setData(filters[i]);
					item.addSelectionListener(new RemoveFilterSelection());
				}
			}

		item = new MenuItem(menu, SWT.CHECK);
		item.setText(Localization.getString("DataGrid.ManualyResize")); //$NON-NLS-1$
		item.addSelectionListener(new MenuManualyResizedSelection());

		for(int i=0; i<IFormattingStyles.FORMAT_TITLES.length; i++) {
			item = new MenuItem(formatMenu, SWT.RADIO);
			item.setText(IFormattingStyles.FORMAT_TITLES[i]);
			item.addSelectionListener(new MenuFormatSelection());
		}

		formatMenu.addMenuListener(new FormatMenuListener());
		return menu;
	}

	private int getSelectedColumn() {
		TableColumn[] cols = table.getColumns();
		int location = 0;
		for(int i=0; i<cols.length; i++)
			if(clickLocation.x > location && clickLocation.x < (location+=cols[i].getWidth()))
				return i;

		return cols.length-1;
	}

	public class MainMenuListener extends MenuAdapter {
		@Override
		public void menuShown(MenuEvent e) {
			MenuItem item = ((Menu)e.widget).getItem(1);
			item.setSelection(manualResize);
		}
	}

	public class MenuManualyResizedSelection extends SelectionAdapter {
		@Override
		public void widgetSelected(SelectionEvent e) {
			manualResize = !manualResize;
		}
	}

	public class AddFilterSelection extends SelectionAdapter {
		@Override
		public void widgetSelected(SelectionEvent e) {
			SelectFilterWizard wizard = new SelectFilterWizard(dataSet.getTitles());
			IWorkbench workbench = PlatformUI.getWorkbench();
			wizard.init(workbench, null);
			WizardDialog dialog = new WizardDialog(workbench.getActiveWorkbenchWindow().getShell(), wizard);
			dialog.create();
			dialog.open();

			IDataSetFilter filter = wizard.getFilter();
			wizard.dispose();

			if(null != filter) {
				filteredDataSet.addFilter(filter);
				table.removeAll();
				handleUpdateEvent();

				MenuItem item = new MenuItem(filterMenu, SWT.CASCADE);
				item.setText(AvailableFilterTypes.getFilterName(filter.getID()));
				item.setData(filter);
				item.addSelectionListener(new RemoveFilterSelection());
			}
		}
	}

	public class RemoveFilterSelection implements SelectionListener {
		@Override
		public void widgetSelected(SelectionEvent e) {
			IDataSetFilter idsf = (IDataSetFilter)((MenuItem)e.widget).getData();
			e.widget.dispose();

			if(filteredDataSet.removeFilter(idsf)) {
				table.removeAll();
				handleUpdateEvent();
			}
		}

		@Override
		public void widgetDefaultSelected(SelectionEvent e) {}
	}

	public class FormatMenuListener extends MenuAdapter {
		@Override
		public void menuShown(MenuEvent e) {
			MenuItem[] items = ((Menu)e.widget).getItems();
			boolean doubleValid = false, longValid = false;
			String itemText;

			int selectedCol = Math.max(1, getSelectedColumn());

			for(int i=0; i<items.length; i++)
				items[i].setSelection(false);
			items[columnFormat[selectedCol-1].getFormat()].setSelection(true);

			itemText = dataSet.getRow(0)[selectedCol-1].toString();
			items[IFormattingStyles.UNFORMATED].setEnabled(true);
			items[IFormattingStyles.STRING].setEnabled(true);

			try {
				Double.parseDouble(itemText);
				doubleValid = true;

				try {
					Long.parseLong(itemText);
					longValid = true;
				} catch(NumberFormatException nfe2) {}
			} catch(NumberFormatException nfe) {}

			items[IFormattingStyles.DOUBLE].setEnabled(doubleValid);
			items[IFormattingStyles.HEX].setEnabled(longValid);
			items[IFormattingStyles.OCTAL].setEnabled(longValid);
			items[IFormattingStyles.BINARY].setEnabled(longValid);
			items[IFormattingStyles.DATE].setEnabled(longValid);
		}
	}

	public class MenuFormatSelection extends SelectionAdapter {
		@Override
		public void widgetSelected(SelectionEvent e) {
			int format = IFormattingStyles.UNFORMATED;
			int column = Math.max(1, getSelectedColumn());
			int i;
			for(i=0; i<IFormattingStyles.FORMAT_TITLES.length; i++)
				if(IFormattingStyles.FORMAT_TITLES[i].equals(((MenuItem)e.getSource()).getText()))
					format = i;

			Object[] data = dataSet.getColumn(column-1);
			columnFormat[column-1].setFormat(format);
			for(i=0; i<table.getItemCount(); i++)
				table.getItem(i).setText(column, columnFormat[column-1].format(data[i].toString()));
			table.redraw();
		}
	}

	@Override
	public void handleUpdateEvent() {
		if(table.isDisposed()) return;

		table.getDisplay().asyncExec(new Runnable() {
			@Override
			public void run() {
				TableItem item;
				int startLocation, endLocation = filteredDataSet.getRowCount();

				if(FULL_UPDATE == (style & FULL_UPDATE)) {
					//Remove extra items so save memory.
					removedItems += table.getItemCount();
					table.removeAll();
					startLocation = 0;
				} else {
					startLocation = table.getItemCount()+removedItems;
				}

				//Add all the new items to the table
				Object[] os;
				for(int j,i=startLocation; i<endLocation; i++) {
					item = new TableItem(table, SWT.NONE);
					os = filteredDataSet.getRow(i);

					item.setText(0, "" + i); //$NON-NLS-1$
					for(j=0; j<os.length; j++) {
						//Ignore null items
						if (os[j] != null) {
							item.setText(j+1, columnFormat[j].format(os[j].toString()));
						}
					}
				}

				if(FULL_UPDATE != (style & FULL_UPDATE)) {
					//Remove extra items so save memory.
					if(table.getItemCount() > prefs.getInt(GraphingAPIPreferenceConstants.P_MAX_DATA_ITEMS)) {
						int items = table.getItemCount()-prefs.getInt(GraphingAPIPreferenceConstants.P_MAX_DATA_ITEMS);
						table.remove(0, items-1);
						removedItems += items;
					}
				}

				//Resize the columns
				TableColumn col = table.getColumn(0);
				col.pack();
				if(!manualResize) {
					TableColumn[] cols = table.getColumns();
					for(int i=1; i<cols.length; i++)
						cols[i].pack();
				}
				//Use if we want to set focus to newly added item
				if(prefs.getBoolean(GraphingAPIPreferenceConstants.P_JUMP_NEW_TABLE_ENTRY))
					table.showItem(table.getItem(table.getItemCount()-1));
			}
		});
	}

	public void dispose() {
		dataSet = null;
		table.dispose();
		table = null;
		clickLocation = null;
		columnFormat = null;
	}

	protected IDataSet dataSet;
	protected IFilteredDataSet filteredDataSet;
	protected IFormattingStyles columnFormat[];
	protected int removedItems;
	protected Table table;
	protected Point clickLocation;
	protected IPreferenceStore prefs;
	protected boolean manualResize;
	protected Menu filterMenu;
	protected int style;

	public static final int NONE = 0;
	public static final int FULL_UPDATE = 1;
}