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

package net.oni2.aeinstaller.gui.modtable;

import java.io.File;
import java.util.HashSet;
import java.util.ResourceBundle;
import java.util.TreeSet;
import java.util.Vector;

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

import net.oni2.aeinstaller.backend.mods.Mod;
import net.oni2.aeinstaller.backend.mods.ModManager;

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

        private static final long serialVersionUID = -8278155705802697354L;

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

        private Vector<Mod> items = new Vector<Mod>();
        private Vector<Boolean> install = new Vector<Boolean>();

        private HashSet<DownloadSizeListener> listeners = new HashSet<DownloadSizeListener>();

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

        @Override
        public Object getValueAt(int row, int col) {
                Mod mod = items.get(row);
                switch (col) {
                        case -1:
                                return mod;
                        case 0:
                                return install.get(row);
                        case 1:
                                return mod.getName();
                        case 2:
                                return mod.getPackageNumberString();
                }
                return null;
        }

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

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

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

        @Override
        public Class<?> getColumnClass(int col) {
                switch (col) {
                        case 0:
                                return Boolean.class;
                        case 1:
                                return String.class;
                        case 2:
                                return String.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:
                                w = 50;
                                col.setPreferredWidth(w);
                                col.setMinWidth(w);
                                col.setMaxWidth(w);
                                break;
                        case 1:
                                col.setPreferredWidth(150);
                                break;
                        case 2:
                                w = 55;
                                col.setPreferredWidth(w);
                                col.setMinWidth(w);
                                col.setMaxWidth(w);
                                break;
                }
        }

        /**
         * Reload the nodes data after an update to the cache
         */
        public void reloadData() {
                items.clear();
                items.addAll(ModManager.getInstance().getModsValidAndNotMandatory());
                revertSelection();
        }

        /**
         * Revert the selection to the mods that are currently installed
         */
        public void revertSelection() {
                install.clear();
                for (int i = 0; i < items.size(); i++) {
                        install.add(i, ModManager.getInstance()
                                        .isModInstalled(items.get(i)));
                }
                fireTableDataChanged();
        }

        /**
         * Reload the selection after a config was loaded
         * 
         * @param config
         *            Config to load
         */
        public void reloadSelection(File config) {
                Vector<Integer> selected = ModManager.getInstance().loadModSelection(
                                config);
                install.clear();
                for (int i = 0; i < items.size(); i++) {
                        install.add(i, selected.contains(items.get(i).getPackageNumber()));
                }
                fireTableDataChanged();
        }

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

        /**
         * @return Mods selected for installation
         */
        public TreeSet<Mod> getSelectedMods() {
                TreeSet<Mod> res = new TreeSet<Mod>();
                for (int i = 0; i < items.size(); i++) {
                        if (install.get(i))
                                res.add(items.get(i));
                }
                return res;
        }

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

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

                        int size = 0;
                        for (int i = 0; i < items.size(); i++) {
                                if (install.get(i)) {
                                        Mod m = items.get(i);
                                        if (!m.isLocalAvailable())
                                                size += m.getZipSize();
                                }
                        }
                        for (DownloadSizeListener dsl : listeners)
                                dsl.downloadSizeChanged(size);
                }
        }

        /**
         * @param lis
         *            Listener to receive download size changed events
         */
        public void addDownloadSizeListener(DownloadSizeListener lis) {
                listeners.add(lis);
        }

        /**
         * @param lis
         *            Listener to no longer receive download size changed events
         */
        public void removeDownloadSizeListener(DownloadSizeListener lis) {
                listeners.remove(lis);
        }

}