package com.explosion.expfmodules.rdbmsconn.dbom;
   
   /*
    * =============================================================================
    * 
    * Copyright 2004 Stephen Cowx
    * 
    * Licensed 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.
   * 
   * =============================================================================
   */
  
  import java.sql.ResultSet;
  import java.util.ArrayList;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Vector;
  
  import javax.swing.table.DefaultTableModel;
  
  /***
   * This class is the result of merging a result set with a TableModel. It
   * contains a reference to the result set, a reference to the table model and
   * also various other information about the data.
   * 
   * @author Stephen Cowx
   */
  
  public class DataSet
  {
      private int numberOfRowsUpdated = 0;
      private int[] longestEntryLengths;
      private String[] longestEntries;
      private DefaultTableModel tableModel;
      private String message = null;
  
      private ResultSet resultSet = null;
      private Vector columnNames = new Vector();
      private HashMap columns = new HashMap();
      private List errors;
  
      /***
       * Construsts an empty dataset
       *  
       */
      public DataSet()
      {    
      }
  
      /***
       * Returns the widest value of the given column. In numberOfCharacters
       */
      public int getLongestEntryLength(int columnIndex)
      {
          return longestEntryLengths[columnIndex];
      }
  
      /***
       * Returns the all of the longest entries per column as an array of Strings
       */
      public String[] getLongestEntries()
      {
          return longestEntries;
      }
  
      /***
       * @param longestEntries The longestEntries to set.
       */
      public void setLongestEntries(String[] longestEntries)
      {
          this.longestEntries = longestEntries;
      }
  
      /***
       * Returns the all of the lengths of the widest(longest) entries per column
       * as an array of ints
       */
      public int[] getLongestEntryLengths()
      {
          return longestEntryLengths;
      }
  
      /***
       * @param longestEntryLengths The longestEntryLengths to set.
       */
      public void setLongestEntryLengths(int[] longestEntryLengths)
      {
          this.longestEntryLengths = longestEntryLengths;
      }
 
     /***
      * Returns the message created when this dataset was created.
      * 
      * @return
      */
     public String getMessage()
     {
         return this.message;
     }
 
     /***
      * @param message The message to set.
      */
     public void setMessage(String message)
     {
         this.message = message;
     }
 
     /***
      * returns the result set for this dataset
      * 
      * @return
      */
     public ResultSet getResultSet()
     {
         return this.resultSet;
     }
 
     /***
      * @param resultSet The resultSet to set.
      */
     public void setResultSet(ResultSet resultSet)
     {
         this.resultSet = resultSet;
     }
 
     /***
      * Returns the tabelModel containing the results
      */
     public DefaultTableModel getTableModel()
     {
         return this.tableModel;
     }
 
     /***
      * @param sets the data of this Dataset.  It should be a Vector of vectors;
      */
     public void setData(Vector data)
     {
         if (data == null)
             this.tableModel = new DefaultTableModel();
         else
             this.tableModel = new DefaultTableModel(data,getColumnNames());
     }
 
     /***
      * Returns the names o fall the columns as a Vector of Strings
      */
     public Vector getColumnNames()
     {
         return columnNames;
     }
 
     /***
      * Returns the columnAttributes for the named column
      * 
      * @param columnName
      * @return
      */
     public DataSetColumn getColumn(String columnName)
     {
         return (DataSetColumn) columns.get(columnName);
     }
 
     /***
      * Adds the column to this dataSet. 
      * 
      * @param columnName
      * @param attributes
      */
     public void addColumn(DataSetColumn column)
     {
         columns.put(column.getColumnName(), column);
         columnNames.add(column.getColumnName());
     }
 
     /***
      * Returns the numberOfRowsUpdated.
      * @return 
      */
     public int getNumberOfRowsUpdated()
     {
         return numberOfRowsUpdated;
     }
     /***
      * Sets the numberOfRows that were updated
      * @param numberOfRowsUpdated
      */
     public void setNumberOfRowsUpdated(int numberOfRowsUpdated)
     {
         this.numberOfRowsUpdated = numberOfRowsUpdated;
     }
     
     /***
      * This method returns a list of the error nmessages in this result set if there are any
      * @return
      */
     public List getErrors()
     {
         return errors;
     }
     
     /***
      * This method allows for errors to be added to this dataset
      * @param error
      */
     public void addError(Throwable error)
     {
         if (errors == null)
             errors = new ArrayList();
         
         errors.add(error);
     }
 }