capacitytablemodel.java

来自「基于Java的软件信息系统－项目管理」· Java 代码 · 共 52 行

package ui;

import javax.swing.table.AbstractTableModel;
import java.util.List;
import javax.swing.table.TableModel;
import db.CapacityInfo;

public class CapacityTableModel extends AbstractTableModel
{
    private List m_capacityList;

    public CapacityTableModel (List capacityList)
    {
        m_capacityList = capacityList;
    }

    public String getColumnName (int col)
    {
        switch (col)
        {
        case 0:
            return "Description";
        case 1:
            return "# of Reservations";
        default:
            return "Unknown";
        }
    }

    public int getColumnCount()
    {
        return 2;
    }

    public int getRowCount()
    {
        return m_capacityList.size();
    }

    // col 0 is the description
    // col 1 is the count
    public Object getValueAt(int row, int col)
    {
        CapacityInfo info = (CapacityInfo) m_capacityList.get(row);
       if (col == 0)
            return info.getRoomDescription();
        else if (col == 1)
            return new Integer (info.getCount());
        return null;
    }
}