BaseListTableModel.java

/*
 *  UFO Saved Game Editor
 *  Copyright (C) 2010 Christopher Davoren
 *
 * 
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 * 
 *  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 General Public License for more details.
 * 
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package net.rubikscomplex.ufosge.gui.table;

import java.util.ArrayList;
import java.util.Iterator;
import javax.swing.table.AbstractTableModel;
import net.rubikscomplex.ufosge.data.Base;
import net.rubikscomplex.ufosge.data.SavedGame;


/**
 *
 * @author Chris Davoren
 */
public class BaseListTableModel extends AbstractTableModel {
    public static final int COLUMN_NAME = 0;
    public static final int COLUMN_SCIENTISTS = 1;
    public static final int COLUMN_ENGINEERS = 2;
    public static final int COLUMN_ISACTIVE = 3;

    protected SavedGame game;
    protected boolean listInactive;

    protected ArrayList<Base> listedBases;

    public BaseListTableModel(SavedGame game, boolean listInactive) {
        super();
        
        this.game = game;
        this.listInactive = listInactive;

        updateList(false);
    }

    public void updateList(boolean fireChangedEvent) {
        listedBases = new ArrayList<Base>();

        Iterator<Base> ib = game.bases.iterator();
        while(ib.hasNext()) {
            Base b = ib.next();
            if(b.isActive || (listInactive && b.isUsed())) {
                listedBases.add(b);
            }
        }

        if (fireChangedEvent) {
            fireTableDataChanged();
        }
    }
    public boolean listsUnused() {
        return listInactive;
    }

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

    public int getColumnCount() {
        return listInactive ? 4 : 3;
    }

    public Object getValueAt(int row, int column) {
        switch(column) {
            case COLUMN_NAME: return listedBases.get(row).name;
            case COLUMN_SCIENTISTS: return listedBases.get(row).scientists;
            case COLUMN_ENGINEERS: return listedBases.get(row).engineers;
            case COLUMN_ISACTIVE: return listedBases.get(row).isActive;
            default: return null;
        }
    }

    public Base getBaseAt(int row) {
        return listedBases.get(row);
    }

    @Override
    public String getColumnName(int column) {
        switch(column) {
            case COLUMN_NAME: return "Name";
            case COLUMN_SCIENTISTS: return "Scientists";
            case COLUMN_ENGINEERS: return "Engineers";
            case COLUMN_ISACTIVE: return "Active";
            default: return "Unknown";
        }
    }

    @Override
    public Class<?> getColumnClass(int column) {
        switch(column) {
            case COLUMN_NAME: return String.class;
            case COLUMN_SCIENTISTS: return Short.class;
            case COLUMN_ENGINEERS: return Short.class;
            case COLUMN_ISACTIVE: return Boolean.class;
            default: return Object.class;
        }
    }
}