powertablemodel.java

测试工具

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * 
 */

package org.apache.jmeter.gui.util;

import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationTargetException;
import java.util.List;

import javax.swing.table.DefaultTableModel;

import org.apache.jorphan.collections.Data;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;

/**
 * @author mstover
 */
public class PowerTableModel extends DefaultTableModel {
	private static final Logger log = LoggingManager.getLoggerForClass();

	Data model = new Data();

	Class[] columnClasses;

	public PowerTableModel(String[] headers, Class[] cc) {
		if (headers.length != cc.length){
			throw new IllegalArgumentException("Header and column array sizes differ");
		}
		model.setHeaders(headers);
		columnClasses = cc;
	}

	public PowerTableModel() {
	}

	public void setRowValues(int row, Object[] values) {
		if (values.length != model.getHeaderCount()){
			throw new IllegalArgumentException("Incorrect number of data items");
		}
		model.setCurrentPos(row);
		for (int i = 0; i < values.length; i++) {
			model.addColumnValue(model.getHeaders()[i], values[i]);
		}
	}

	public Data getData() {
		return model;
	}

	public void addNewColumn(String colName, Class colClass) {
		model.addHeader(colName);
		Class[] newClasses = new Class[columnClasses.length + 1];
		System.arraycopy(columnClasses, 0, newClasses, 0, columnClasses.length);
		newClasses[newClasses.length - 1] = colClass;
		columnClasses = newClasses;
		Object defaultValue = createDefaultValue(columnClasses.length - 1);
		model.setColumnData(colName, defaultValue);
		this.fireTableStructureChanged();
	}

	public void removeRow(int row) {
		log.debug("remove row: " + row);
		if (model.size() > row) {
			log.debug("Calling remove row on Data");
			model.removeRow(row);
		}
	}

	public void removeColumn(int col) {
		model.removeColumn(col);
		this.fireTableStructureChanged();
	}

	public void setColumnData(int col, List data) {
		model.setColumnData(col, data);
	}

	public List getColumnData(String colName) {
		return model.getColumnAsObjectArray(colName);
	}

	public void clearData() {
		String[] headers = model.getHeaders();
		model = new Data();
		model.setHeaders(headers);
		this.fireTableDataChanged();
	}

	public void addRow(Object data[]) {
		if (data.length != model.getHeaderCount()){
			throw new IllegalArgumentException("Incorrect number of data items");
		}
		model.setCurrentPos(model.size());
		for (int i = 0; i < data.length; i++) {
			model.addColumnValue(model.getHeaders()[i], data[i]);
		}
	}

	public void addNewRow() {
		addRow(createDefaultRow());
	}

	private Object[] createDefaultRow() {
		Object[] rowData = new Object[getColumnCount()];
		for (int i = 0; i < rowData.length; i++) {
			rowData[i] = createDefaultValue(i);
		}
		return rowData;
	}

	public Object[] getRowData(int row) {
		Object[] rowData = new Object[getColumnCount()];
		for (int i = 0; i < rowData.length; i++) {
			rowData[i] = model.getColumnValue(i, row);
		}
		return rowData;
	}

	private Object createDefaultValue(int i) {
		Class colClass = getColumnClass(i);
		try {
			return colClass.newInstance();
		} catch (Exception e) {
			try {
				Constructor constr = colClass.getConstructor(new Class[] { String.class });
				return constr.newInstance(new Object[] { "" });
			} catch (NoSuchMethodException err) {
            } catch (InstantiationException err) {
            } catch (IllegalAccessException err) {
            } catch (InvocationTargetException err) {
            }
			try {
				Constructor constr = colClass.getConstructor(new Class[] { Integer.TYPE });
				return constr.newInstance(new Object[] { new Integer(0) });
            } catch (NoSuchMethodException err) {
            } catch (InstantiationException err) {
            } catch (IllegalAccessException err) {
            } catch (InvocationTargetException err) {
			}
			try {
				Constructor constr = colClass.getConstructor(new Class[] { Long.TYPE });
				return constr.newInstance(new Object[] { new Long(0L) });
            } catch (NoSuchMethodException err) {
            } catch (InstantiationException err) {
            } catch (IllegalAccessException err) {
            } catch (InvocationTargetException err) {
			}
			try {
				Constructor constr = colClass.getConstructor(new Class[] { Boolean.TYPE });
				return constr.newInstance(new Object[] { Boolean.FALSE });
            } catch (NoSuchMethodException err) {
            } catch (InstantiationException err) {
            } catch (IllegalAccessException err) {
            } catch (InvocationTargetException err) {
			}
			try {
				Constructor constr = colClass.getConstructor(new Class[] { Float.TYPE });
				return constr.newInstance(new Object[] { new Float(0F) });
            } catch (NoSuchMethodException err) {
            } catch (InstantiationException err) {
            } catch (IllegalAccessException err) {
            } catch (InvocationTargetException err) {
			}
			try {
				Constructor constr = colClass.getConstructor(new Class[] { Double.TYPE });
				return constr.newInstance(new Object[] { new Double(0D) });
            } catch (NoSuchMethodException err) {
            } catch (InstantiationException err) {
            } catch (IllegalAccessException err) {
            } catch (InvocationTargetException err) {
			}
			try {
				Constructor constr = colClass.getConstructor(new Class[] { Character.TYPE });
				return constr.newInstance(new Object[] { new Character(' ') });
            } catch (NoSuchMethodException err) {
            } catch (InstantiationException err) {
            } catch (IllegalAccessException err) {
            } catch (InvocationTargetException err) {
			}
			try {
				Constructor constr = colClass.getConstructor(new Class[] { Byte.TYPE });
				return constr.newInstance(new Object[] { new Byte(Byte.MIN_VALUE) });
            } catch (NoSuchMethodException err) {
            } catch (InstantiationException err) {
            } catch (IllegalAccessException err) {
            } catch (InvocationTargetException err) {
			}
			try {
				Constructor constr = colClass.getConstructor(new Class[] { Short.TYPE });
				return constr.newInstance(new Object[] { new Short(Short.MIN_VALUE) });
            } catch (NoSuchMethodException err) {
            } catch (InstantiationException err) {
            } catch (IllegalAccessException err) {
            } catch (InvocationTargetException err) {
			}
		}
		return "";
	}

	/**
	 * Required by table model interface.
	 * 
	 * @return the RowCount value
	 */
	public int getRowCount() {
		if (model == null) {
			return 0;
		}
		return model.size();
	}

	/**
	 * Required by table model interface.
	 * 
	 * @return the ColumnCount value
	 */
	public int getColumnCount() {
		return model.getHeaders().length;
	}

	/**
	 * Required by table model interface.
	 * 
	 * @return the ColumnName value
	 */
	public String getColumnName(int column) {
		return model.getHeaders()[column];
	}

	public boolean isCellEditable(int row, int column) {
		// all table cells are editable
		return true;
	}

	public Class getColumnClass(int column) {
		return columnClasses[column];
	}

	/**
	 * Required by table model interface. return the ValueAt value
	 */
	public Object getValueAt(int row, int column) {
		return model.getColumnValue(column, row);
	}

	/**
	 * Sets the ValueAt attribute of the Arguments object.
	 * 
	 * @param value
	 *            the new ValueAt value
	 */
	public void setValueAt(Object value, int row, int column) {
		if (row < model.size()) {
			model.setCurrentPos(row);
			model.addColumnValue(model.getHeaders()[column], value);
		}
	}
}