source: AE/installer2/src/net/oni2/aeinstaller/gui/modtable/ModTableModel.java@ 594

package net.oni2.aeinstaller.gui.modtable;

import java.util.ResourceBundle;
import java.util.Vector;

import javax.swing.table.AbstractTableModel;
import javax.swing.table.TableColumn;

import net.oni2.aeinstaller.backend.depot.DepotConfig;
import net.oni2.aeinstaller.backend.depot.DepotManager;
import net.oni2.aeinstaller.backend.depot.model.NodeMod;

/**
 * @author Christian Illy
 */
public class ModTableModel extends AbstractTableModel {

        private static final long serialVersionUID = -8278155705802697354L;

        private ResourceBundle bundle = ResourceBundle.getBundle(getClass()
                        .getName());

        private Vector<NodeMod> items;
        private Vector<Boolean> install;
        private int vocabModTypeID = -1;
        private int vocabPlatformID = -1;

        /**
         * Create a new model
         */
        public ModTableModel() {
                reloadData();
        }

        @Override
        public Object getValueAt(int row, int col) {
                NodeMod node = items.get(row);
                switch (col) {
                        case -1:
                                return node;
                        case 0:
                                return node.getTitle();
                        case 1:
                                return node.getFields().get("package_number");
                        case 2:
                                String type = "";
                                if (vocabModTypeID < 0) {
                                        vocabModTypeID = DepotManager
                                                        .getInstance()
                                                        .getVocabulary(
                                                                        DepotConfig.getVocabularyName_ModType())
                                                        .getVid();
                                }
                                for (int tid : node.getTaxonomyTerms().get(vocabModTypeID)) {
                                        if (type.length() > 0)
                                                type += ", ";
                                        type += DepotManager.getInstance().getTaxonomyTerm(tid)
                                                        .getName();
                                }
                                return type;
                        case 3:
                                if (vocabPlatformID < 0) {
                                        vocabPlatformID = DepotManager
                                                        .getInstance()
                                                        .getVocabulary(
                                                                        DepotConfig.getVocabularyName_Platform())
                                                        .getVid();
                                }
                                int tid = node.getTaxonomyTerms().get(vocabPlatformID)
                                                .iterator().next();
                                return DepotManager.getInstance().getTaxonomyTerm(tid)
                                                .getName();
                        case 4:
                                return install.get(row);
                }
                return null;
        }

        @Override
        public String getColumnName(int col) {
                switch (col) {
                        case 0:
                                return bundle.getString("mod.name");
                        case 1:
                                return bundle.getString("mod.package_number");
                        case 2:
                                return bundle.getString("mod.type");
                        case 3:
                                return bundle.getString("mod.platform");
                        case 4:
                                return bundle.getString("mod.install");
                }
                return null;
        }

        @Override
        public int getRowCount() {
                return items.size();
        }

        @Override
        public int getColumnCount() {
                return 5;
        }

        @Override
        public Class<?> getColumnClass(int col) {
                switch (col) {
                        case 0:
                                return String.class;
                        case 1:
                                return Integer.class;
                        case 2:
                                return String.class;
                        case 3:
                                return String.class;
                        case 4:
                                return Boolean.class;
                }
                return null;
        }

        /**
         * Set the constraints on the columns size for the given column
         * 
         * @param colNum
         *            Column number
         * @param col
         *            Column object
         */
        public void setColumnConstraints(int colNum, TableColumn col) {
                int w;
                switch (colNum) {
                        case 0:
                                col.setPreferredWidth(150);
                                break;
                        case 1:
                                w = 55;
                                col.setPreferredWidth(w);
                                col.setMinWidth(w);
                                col.setMaxWidth(w);
                                break;
                        case 2:
                                col.setPreferredWidth(100);
                                break;
                        case 3:
                                w = 70;
                                col.setPreferredWidth(w);
                                col.setMinWidth(w);
                                col.setMaxWidth(w);
                                break;
                        case 4:
                                w = 60;
                                col.setPreferredWidth(w);
                                col.setMinWidth(w);
                                col.setMaxWidth(w);
                                break;
                }
        }

        /**
         * Reload the nodes data after an update to the cache
         */
        public void reloadData() {
                items = DepotManager.getInstance().getModPackageNodes();
                install = new Vector<Boolean>();
                // TODO check installed
                for (int i = 0; i < items.size(); i++) {
                        install.add(i, false);
                }
        }

        /**
         * Get the items vector
         * 
         * @return Items
         */
        public Vector<NodeMod> getItems() {
                return items;
        }

        @Override
        public boolean isCellEditable(int rowIndex, int columnIndex) {
                return columnIndex == 4;
        }

        @Override
        public void setValueAt(Object aValue, int rowIndex, int columnIndex) {
                super.setValueAt(aValue, rowIndex, columnIndex);
                if (columnIndex == 4) {
                        install.set(rowIndex, (Boolean) aValue);
                }
        }

}