BaseInventoryTableModel.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 javax.swing.table.AbstractTableModel;
import java.util.ArrayList;
import java.util.Iterator;
import net.rubikscomplex.ufosge.data.Base;
import net.rubikscomplex.ufosge.data.SavedGame;


/**
 *
 * @author Chris Davoren
 */
public class BaseInventoryTableModel extends AbstractTableModel {
    public static final int COLUMN_NAME = 0;
    public static final int COLUMN_QUANTITY = 1;

    protected SavedGame game;
    protected Base base;

    static ArrayList<Integer> validIndices = null;
    static ArrayList<String> validNames = null;

    ArrayList<Short> valuesCache;

    public BaseInventoryTableModel(SavedGame game, Base base) {
        super();
        
        this.game = game;
        this.base = base;

        if (validIndices == null) {
            validIndices = new ArrayList<Integer>();
            validNames = new ArrayList<String>();

            for(int i = 0; i < Base.INV_NAMES.length; i++) {
                /* A terminating asterisk marks an unused/invalid inventory slot */
                if (!Base.INV_NAMES[i].endsWith("*")) {
                    validIndices.add(i);
                    validNames.add(Base.INV_NAMES[i]);
                }
            }
        }

        valuesCache = new ArrayList<Short>(validIndices.size());

        updateCache();
    }

    public void updateCache() {
        valuesCache.clear();
        Iterator<Integer> vi = validIndices.iterator();

        while(vi.hasNext()) {
            valuesCache.add(base.inventory[vi.next()]);
        }
    }

    public void update() {
        for(int i = 0; i < valuesCache.size(); i++) {
            base.inventory[validIndices.get(i)] = valuesCache.get(i);
        }
    }

    @Override
    public String getColumnName(int column) {
        switch(column) {
            case COLUMN_NAME: return "Item";
            case COLUMN_QUANTITY: return "Quantity";
            default: return "Uknown";
        }
    }

    public int getColumnCount() {
        return 2;
    }

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

    @Override
    public boolean isCellEditable(int row, int column) {
        return column == COLUMN_QUANTITY;
    }

    @Override
    public Object getValueAt(int row, int column) {
        if (column == COLUMN_NAME) {
            return validNames.get(row);
        }
        else {
            return valuesCache.get(row);
        }
    }

    @Override
    public void setValueAt(Object value, int row, int column) {
        if (column != COLUMN_QUANTITY) {
            return;
        }

        short sval;

        try {
            sval = Short.parseShort(value.toString());
        }
        catch(NumberFormatException e) {
            return;
        }


        valuesCache.set(row, sval);
    }

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