001: /*
002: * <copyright>
003: *
004: * Copyright 2003-2004 BBNT Solutions, LLC
005: * under sponsorship of the Defense Advanced Research Projects
006: * Agency (DARPA).
007: *
008: * You can redistribute this software and/or modify it under the
009: * terms of the Cougaar Open Source License as published on the
010: * Cougaar Open Source Website (www.cougaar.org).
011: *
012: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
013: * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
014: * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
015: * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
016: * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
017: * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
018: * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
019: * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
020: * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
021: * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
022: * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
023: *
024: * </copyright>
025: */
026: package org.cougaar.lib.aggagent.client;
027:
028: import java.util.Iterator;
029: import java.util.Vector;
030:
031: import javax.swing.table.AbstractTableModel;
032:
033: import org.cougaar.lib.aggagent.query.AggregationResultSet;
034: import org.cougaar.lib.aggagent.query.ResultSetDataAtom;
035: import org.cougaar.lib.aggagent.query.UpdateListener;
036:
037: /**
038: * This class provides a table model of an AggregationResultSet
039: */
040: public class ResultSetTableModel extends AbstractTableModel {
041: private Vector idHeaders = new Vector();
042: private Vector valueHeaders = new Vector();
043: private Vector dataAtoms = new Vector();
044: private AggregationResultSet observedResultSet = null;
045: private UpdateListener resultSetListener = null;
046:
047: public ResultSetTableModel() {
048: resultSetListener = new UpdateListener() {
049: public void objectAdded(Object sourceObject) {
050: objectChanged(sourceObject);
051: };
052:
053: public void objectRemoved(Object sourceObject) {
054: /* not my problem (clear model?) */
055: };
056:
057: public void objectChanged(Object sourceObject) {
058: updateInternalRepOfResultSet(observedResultSet);
059: }
060: };
061: }
062:
063: public void setResultSet(final AggregationResultSet rs) {
064: if (observedResultSet != null) {
065: observedResultSet.removeUpdateListener(resultSetListener);
066: }
067: observedResultSet = rs;
068: if (observedResultSet != null) {
069: observedResultSet.addUpdateListener(resultSetListener);
070: }
071: updateInternalRepOfResultSet(observedResultSet);
072: }
073:
074: public AggregationResultSet getResultSet() {
075: return observedResultSet;
076: }
077:
078: private void updateInternalRepOfResultSet(AggregationResultSet rs) {
079: Vector oldIdHeaders = (Vector) idHeaders.clone();
080: Vector oldValueHeaders = (Vector) valueHeaders.clone();
081: idHeaders.clear();
082: valueHeaders.clear();
083: dataAtoms.clear();
084: if (rs != null) {
085: Iterator atoms = rs.getAllAtoms();
086: boolean first = true;
087: while (atoms.hasNext()) {
088: ResultSetDataAtom da = (ResultSetDataAtom) atoms.next();
089:
090: if (first) {
091: first = false;
092: for (Iterator i = da.getIdentifierNames(); i
093: .hasNext();) {
094: idHeaders.add(i.next());
095: }
096: for (Iterator i = da.getValueNames(); i.hasNext();) {
097: valueHeaders.add(i.next());
098: }
099: }
100:
101: dataAtoms.add(da);
102: }
103: }
104:
105: // fire proper table change event
106: if ((idHeaders.equals(oldIdHeaders))
107: && (valueHeaders.equals(oldValueHeaders))) {
108: fireTableDataChanged();
109: } else {
110: fireTableStructureChanged();
111: }
112: }
113:
114: public int getRowCount() {
115: return dataAtoms.size();
116: }
117:
118: public int getColumnCount() {
119: return idHeaders.size() + valueHeaders.size();
120: }
121:
122: public String getColumnName(int column) {
123: String name = null;
124: int idCount = idHeaders.size();
125: if (column < idCount) {
126: name = idHeaders.elementAt(column).toString();
127: } else {
128: name = valueHeaders.elementAt(column - idCount).toString();
129: }
130:
131: return name;
132: }
133:
134: public Object getValueAt(int row, int column) {
135: Object value = null;
136: ResultSetDataAtom da = (ResultSetDataAtom) dataAtoms
137: .elementAt(row);
138:
139: int idCount = idHeaders.size();
140: if (column < idCount) {
141: value = da.getIdentifier(idHeaders.elementAt(column));
142: } else {
143: value = da.getValue(valueHeaders
144: .elementAt(column - idCount));
145: }
146:
147: return value;
148: }
149: }
|