/*
* This program is free software; you can redistribute it and/or modify it under the
* terms of the GNU Lesser General Public License, version 2.1 as published by the Free Software
* Foundation.
*
* You should have received a copy of the GNU Lesser General Public License along with this
* program; if not, you can obtain a copy at http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html
* or from the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*
* This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
* See the GNU Lesser General Public License for more details.
*
* Copyright (c) 2001 - 2013 Object Refinery Ltd, Pentaho Corporation and Contributors.. All rights reserved.
*/
package org.pentaho.reporting.engine.classic.demo.ancient.demo.invoice;
import java.util.ArrayList;
import java.util.Date;
import javax.swing.table.AbstractTableModel;
import org.pentaho.reporting.engine.classic.demo.ancient.demo.invoice.model.Article;
import org.pentaho.reporting.engine.classic.demo.ancient.demo.invoice.model.Invoice;
public class InvoiceTableModel extends AbstractTableModel
{
private static Class[] COLUMN_TYPES =
{
Invoice.class,
String.class, String.class, String.class,
String.class, String.class, String.class,
String.class, Date.class, String.class,
String.class, String.class, String.class,
Float.class, Integer.class
};
private static String[] COLUMN_NAMES =
{
"invoice",
"customer.firstName", "customer.lastName", "customer.street",
"customer.town", "customer.postalCode", "customer.country",
"customer.salutation", "invoice.date", "invoice.number",
"article.name", "article.number", "article.details",
"article.price", "article.count"
};
private transient Invoice[] invoicePerRow;
private transient Article[] articlesPerRow;
private transient int[] articleCountPerRow;
private ArrayList invoices;
private int totalSize;
public InvoiceTableModel()
{
invoices = new ArrayList();
}
public void addInvoice(final Invoice invoice)
{
invoices.add(invoice);
invalidateCaches();
fireTableDataChanged();
}
public void removeInvoice(final Invoice invoice)
{
invoices.remove(invoice);
invalidateCaches();
fireTableDataChanged();
}
public Invoice getInvoice(final int invoice)
{
return (Invoice) invoices.get(invoice);
}
public void invalidateCaches()
{
int size = 0;
for (int i = 0; i < invoices.size(); i++)
{
final Invoice inv = getInvoice(i);
size += inv.getArticleCount();
}
this.totalSize = size;
this.invoicePerRow = null;
this.articlesPerRow = null;
this.articleCountPerRow = null;
}
/**
* Returns the number of columns in the model. A <code>JTable</code> uses this method to determine how many columns it
* should create and display by default.
*
* @return the number of columns in the model
* @see #getRowCount
*/
public int getColumnCount()
{
return COLUMN_NAMES.length;
}
/**
* Returns the number of rows in the model. A <code>JTable</code> uses this method to determine how many rows it
* should display. This method should be quick, as it is called frequently during rendering.
*
* @return the number of rows in the model
* @see #getColumnCount
*/
public int getRowCount()
{
return totalSize;
}
/**
* Returns a default name for the column using spreadsheet conventions: A, B, C, ... Z, AA, AB, etc. If
* <code>column</code> cannot be found, returns an empty string.
*
* @param column the column being queried
* @return a string containing the default name of <code>column</code>
*/
public String getColumnName(final int column)
{
return COLUMN_NAMES[column];
}
/**
* Returns <code>Object.class</code> regardless of <code>columnIndex</code>.
*
* @param columnIndex the column being queried
* @return the Object.class
*/
public Class getColumnClass(final int columnIndex)
{
return COLUMN_TYPES[columnIndex];
}
private void fillCache()
{
if (invoicePerRow != null && articlesPerRow != null)
{
// nothing to do...
return;
}
// ensure that we have enough space ...
this.invoicePerRow = new Invoice[totalSize];
this.articlesPerRow = new Article[totalSize];
this.articleCountPerRow = new int[totalSize];
int currentRow = 0;
final int invoiceSize = invoices.size();
for (int i = 0; i < invoiceSize; i++)
{
final Invoice inv = (Invoice) invoices.get(i);
final int articleCount = inv.getArticleCount();
for (int ac = 0; ac < articleCount; ac++)
{
invoicePerRow[currentRow] = inv;
articlesPerRow[currentRow] = inv.getArticle(ac);
articleCountPerRow[currentRow] = inv.getArticleCount(ac);
currentRow += 1;
}
}
}
/**
* Returns the value for the cell at <code>columnIndex</code> and <code>rowIndex</code>.
*
* @param rowIndex the row whose value is to be queried
* @param columnIndex the column whose value is to be queried
* @return the value Object at the specified cell
*/
public Object getValueAt(final int rowIndex, final int columnIndex)
{
// just make sure we can access the invoices by the array
fillCache();
final Invoice inv = invoicePerRow[rowIndex];
final Article art = articlesPerRow[rowIndex];
switch (columnIndex)
{
case 0:
return inv;
case 1:
return inv.getCustomer().getFirstName();
case 2:
return inv.getCustomer().getLastName();
case 3:
return inv.getCustomer().getStreet();
case 4:
return inv.getCustomer().getTown();
case 5:
return inv.getCustomer().getPostalCode();
case 6:
return inv.getCustomer().getCountry();
case 7:
return inv.getCustomer().getSalutation();
case 8:
return inv.getDate();
case 9:
return inv.getInvoiceNumber();
case 10:
return art.getName();
case 11:
return art.getArticleNumber();
case 12:
return art.getArticleDetails();
case 13:
return new Float(art.getPrice());
case 14:
return new Integer(articleCountPerRow[rowIndex]);
}
throw new IndexOutOfBoundsException("ColumnIndex is invalid: " + columnIndex);
}
}