View Javadoc
   /*
    * Angkor Web Framework
    *
    * Distributable under LGPL license.
    * See terms of license at gnu.org.
    */
   
   package com.tirsen.angkor.beans;
   
  import com.tirsen.angkor.table.AbstractTableModel;
  import com.tirsen.angkor.widget.ValueModel;
  
  import java.beans.BeanInfo;
  import java.beans.IntrospectionException;
  import java.beans.Introspector;
  import java.beans.PropertyDescriptor;
  import java.io.Serializable;
  import java.util.ArrayList;
  import java.util.Arrays;
  import java.util.Collection;
  import java.util.Iterator;
  import java.util.List;
  
  public class JavaBeanTableModel extends AbstractTableModel
  {
      private Class rowClass;
      private List rows;
      private List columns;
  
      public JavaBeanTableModel()
      {
      }
  
      public JavaBeanTableModel(Class rowClass)
      {
          this.rowClass = rowClass;
      }
  
      public interface TableColumnModel extends Serializable
      {
          String getName();
  
          ValueModel getValue(Object row);
      }
  
      private static class PropertyTableColumnModel implements TableColumnModel
      {
          private String name;
          private transient PropertyDescriptor property;
  
          public PropertyTableColumnModel(PropertyDescriptor property)
          {
              this.property = property;
          }
  
          public PropertyTableColumnModel(String name, PropertyDescriptor property)
          {
              this.name = name;
              this.property = property;
          }
  
          public String getName()
          {
              return name == null ? property.getDisplayName() : name;
          }
  
          public ValueModel getValue(Object row)
          {
              return new PropertyValueModel(row, property);
          }
      }
  
      /***
       * Resets the columns.
       */
      public void resetColumns()
      {
          columns = null;
      }
  
      /***
       * If the columns are not already created creates columns based on all the properties in the beans.
       */
      private void createDefaultColumns()
      {
          if (columns == null)
          {
              try
              {
                  PropertyDescriptor[] properties = getProperties();
                  columns = new ArrayList(properties.length);
                  for (int i = 0; i < properties.length; i++)
                  {
                      PropertyDescriptor property = properties[i];
                      addColumn(new PropertyTableColumnModel(property));
                  }
              }
              catch (IntrospectionException e)
              {
                 throw new RuntimeException(e.getMessage());
             }
         }
     }
 
     private PropertyDescriptor[] getProperties() throws IntrospectionException
     {
         Class beanClass = getRowClass();
         BeanInfo beanInfo = Introspector.getBeanInfo(beanClass, Object.class);
         PropertyDescriptor[] properties = beanInfo.getPropertyDescriptors();
         return properties;
     }
 
     private Class getRowClass()
     {
         Class cls = rowClass;
         if (cls == null && getRowCount() > 0) cls = getRow(0).getClass();
         if (cls == null) cls = getRows().getClass().getComponentType();
         if (cls == null) throw new IllegalStateException("Could not determine class of objects used for rows.");
         return cls;
     }
 
     public void addColumn(TableColumnModel column)
     {
         columns.add(column);
     }
 
     public PropertyDescriptor findProperty(String property)
     {
         try
         {
             PropertyDescriptor[] properties = getProperties();
             for (int i = 0; i < properties.length; i++)
             {
                 PropertyDescriptor propertyDesc = properties[i];
                 if (propertyDesc.getName().equals(property)) return propertyDesc;
             }
             throw new IntrospectionException("No such property " + property);
         }
         catch (IntrospectionException e)
         {
             throw new RuntimeException(e.getMessage());
         }
     }
 
     public void addColumn(String name, String property)
     {
         if (columns == null) columns = new ArrayList(1);
         columns.add(new PropertyTableColumnModel(name, findProperty(property)));
     }
 
     public String getColumnName(int column)
     {
         return getColumn(column).getName();
     }
 
     public int getColumnCount()
     {
         createDefaultColumns();
         return columns.size();
     }
 
     public ValueModel getValueAt(int row, int column)
     {
         return (getColumn(column)).getValue(getRow(row));
     }
 
     private TableColumnModel getColumn(int column)
     {
         createDefaultColumns();
         return (TableColumnModel) columns.get(column);
     }
 
     /***
      * Gets the value object for a specified row, also checks wheather row is within range specified by
      * {@link #setRange(int,int)}.
      * @throws IllegalArgumentException if <code>row</code> parameter is outside specified range.
      */
     public Object getRow(int row)
     {
         if (row < getStart() || row > getEnd())
         {
             throw new IllegalArgumentException("Tried to access row " + row +
                     " outside range (" + getStart() + ", " + getEnd() + ").");
         }
         return getRows().get(row);
     }
 
     private List getRows()
     {
         if (rows == null) throw new IllegalStateException("Tried to access model before initializing rows.");
         return rows;
     }
 
     public int getRowCount()
     {
         int rowCount = super.getRowCount();
         if (rowCount == -1)
         {
             if (rows == null)
                 rowCount = 0;
             else
                 rowCount = getRows().size();
         }
         return rowCount;
     }
 
     public void setRow(int row, Object object)
     {
         ensureCapable(row);
         rows.set(row, object);
     }
 
     /***
      * Sets a segment of rows starting at specified startRow.
      */
     public void setRows(int startRow, Collection objects)
     {
         ensureCapable(startRow + objects.size());
         int row = startRow;
         for (Iterator iterator = objects.iterator(); iterator.hasNext();)
         {
             Object o = (Object) iterator.next();
             rows.set(row++, o);
         }
     }
 
     private void ensureCapable(int row)
     {
         if (rows == null) rows = new ArrayList(row);
         int size = rows.size();
         for (int i = size - row; i < size + 1; i++)
         {
             rows.add(null);
         }
     }
 
     public void setRows(Object[] rows)
     {
         if (rows != null)
             this.rows = Arrays.asList(rows);
         else
             this.rows = null;
     }
 
     public void setRows(List rows)
     {
         this.rows = new ArrayList(rows);
     }
 
     public void setRows(Iterator iterator)
     {
         rows = new ArrayList();
         while (iterator.hasNext())
         {
             Object o = (Object) iterator.next();
             rows.add(o);
         }
     }
 
     public void empty()
     {
         super.empty();
         rows = null;
     }
 }