UPItemTableModel.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.SavedGame;
import net.rubikscomplex.ufosge.data.UPItem;

/**
 *
 * @author Chris Davoren <chris@rubikscomplex.net>
 */
public class UPItemTableModel extends AbstractTableModel {
    public static final int COLUMN_NAME = 0;
    public static final int COLUMN_CATEGORY = 1;
    public static final int COLUMN_RESEARCHED = 2;

    protected SavedGame game;

    ArrayList<UPItem> upCache;

    public UPItemTableModel(SavedGame game, ArrayList<UPItem> uCache) {
        this.game = game;

        setCache(uCache);
    }

    protected void setCache(ArrayList<UPItem> uCache) {
        upCache = uCache;
    }

    public Object getValueAt(int row, int column) {
        UPItem up = upCache.get(row);
        // System.out.println(up.category + " : " + up.subCategory);
        switch(column) {
            case COLUMN_NAME: return up.getName();
            case COLUMN_CATEGORY: return UPItem.ARTICLE_CATEGORIES[up.category];
            case COLUMN_RESEARCHED: return UPItem.researchFlagToString(up.researchFlag);
            default: return null;
        }
    }

    @Override
    public Class<?> getColumnClass(int column) {
        switch(column) {
            case COLUMN_NAME: return String.class;
            case COLUMN_CATEGORY: return String.class;
            case COLUMN_RESEARCHED: return String.class;
            default: return Object.class;
        }
    }

    @Override
    public String getColumnName(int column) {
        switch(column) {
            case COLUMN_NAME: return "Name";
            case COLUMN_CATEGORY: return "Category";
            case COLUMN_RESEARCHED: return "Availability";
            default: return "Unknown";
        }
    }

    @Override
    public void setValueAt(Object value, int row, int column) {
        UPItem up = upCache.get(row);

        switch(column) {
            case COLUMN_RESEARCHED: up.researchFlag = UPItem.stringToResearchFlag(value.toString());
            default: return;
        }
    }

    public int getColumnCount() {
        return 3;
    }

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

    @Override
    public boolean isCellEditable(int row, int column) {
        return (column == COLUMN_RESEARCHED && upCache.get(row).researchFlag != UPItem.RESEARCH_ALWAYS);
    }
}