gui/modtable/ModTableModel.java

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().getMods());
                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);
        }

}
Revision:	606
Committed:	Mon Jan 14 12:34:55 2013 UTC (12 years, 8 months ago) by alloc
Content type:	text/x-java
File size:	4510 byte(s)
Log Message:	AEI2: - Table columns "Type" and "Platform" removed - Column "Install" moved to front - Added Type+Platform to the description pane for now - Order of level-installation: now in alphabetical order (0, 10...19, 1...9) - Oni launched with WorkDir set to the Edition folder instead of AEI folder
#	Content
1	package net.oni2.aeinstaller.gui.modtable;
2
3	import java.io.File;
4	import java.util.HashSet;
5	import java.util.ResourceBundle;
6	import java.util.TreeSet;
7	import java.util.Vector;
8
9	import javax.swing.table.AbstractTableModel;
10	import javax.swing.table.TableColumn;
11
12	import net.oni2.aeinstaller.backend.mods.Mod;
13	import net.oni2.aeinstaller.backend.mods.ModManager;
14
15	/**
16	* @author Christian Illy
17	*/
18	public class ModTableModel extends AbstractTableModel {
19
20	private static final long serialVersionUID = -8278155705802697354L;
21
22	private ResourceBundle bundle = ResourceBundle.getBundle(getClass()
23	.getName());
24
25	private Vector<Mod> items = new Vector<Mod>();
26	private Vector<Boolean> install = new Vector<Boolean>();
27
28	private HashSet<DownloadSizeListener> listeners = new HashSet<DownloadSizeListener>();
29
30	/**
31	* Create a new model
32	*/
33	public ModTableModel() {
34	reloadData();
35	}
36
37	@Override
38	public Object getValueAt(int row, int col) {
39	Mod mod = items.get(row);
40	switch (col) {
41	case -1:
42	return mod;
43	case 0:
44	return install.get(row);
45	case 1:
46	return mod.getName();
47	case 2:
48	return mod.getPackageNumberString();
49	}
50	return null;
51	}
52
53	@Override
54	public String getColumnName(int col) {
55	switch (col) {
56	case 0:
57	return bundle.getString("mod.install");
58	case 1:
59	return bundle.getString("mod.name");
60	case 2:
61	return bundle.getString("mod.package_number");
62	}
63	return null;
64	}
65
66	@Override
67	public int getRowCount() {
68	return items.size();
69	}
70
71	@Override
72	public int getColumnCount() {
73	return 3;
74	}
75
76	@Override
77	public Class<?> getColumnClass(int col) {
78	switch (col) {
79	case 0:
80	return Boolean.class;
81	case 1:
82	return String.class;
83	case 2:
84	return String.class;
85	}
86	return null;
87	}
88
89	/**
90	* Set the constraints on the columns size for the given column
91	*
92	* @param colNum
93	* Column number
94	* @param col
95	* Column object
96	*/
97	public void setColumnConstraints(int colNum, TableColumn col) {
98	int w;
99	switch (colNum) {
100	case 0:
101	w = 50;
102	col.setPreferredWidth(w);
103	col.setMinWidth(w);
104	col.setMaxWidth(w);
105	break;
106	case 1:
107	col.setPreferredWidth(150);
108	break;
109	case 2:
110	w = 55;
111	col.setPreferredWidth(w);
112	col.setMinWidth(w);
113	col.setMaxWidth(w);
114	break;
115	}
116	}
117
118	/**
119	* Reload the nodes data after an update to the cache
120	*/
121	public void reloadData() {
122	items.clear();
123	items.addAll(ModManager.getInstance().getMods());
124	revertSelection();
125	}
126
127	/**
128	* Revert the selection to the mods that are currently installed
129	*/
130	public void revertSelection() {
131	install.clear();
132	for (int i = 0; i < items.size(); i++) {
133	install.add(i, ModManager.getInstance()
134	.isModInstalled(items.get(i)));
135	}
136	fireTableDataChanged();
137	}
138
139	/**
140	* Reload the selection after a config was loaded
141	*
142	* @param config
143	* Config to load
144	*/
145	public void reloadSelection(File config) {
146	Vector<Integer> selected = ModManager.getInstance().loadModSelection(
147	config);
148	install.clear();
149	for (int i = 0; i < items.size(); i++) {
150	install.add(i, selected.contains(items.get(i).getPackageNumber()));
151	}
152	fireTableDataChanged();
153	}
154
155	/**
156	* Get the items vector
157	*
158	* @return Items
159	*/
160	public Vector<Mod> getItems() {
161	return items;
162	}
163
164	/**
165	* @return Mods selected for installation
166	*/
167	public TreeSet<Mod> getSelectedMods() {
168	TreeSet<Mod> res = new TreeSet<Mod>();
169	for (int i = 0; i < items.size(); i++) {
170	if (install.get(i))
171	res.add(items.get(i));
172	}
173	return res;
174	}
175
176	@Override
177	public boolean isCellEditable(int rowIndex, int columnIndex) {
178	return columnIndex == 0;
179	}
180
181	@Override
182	public void setValueAt(Object aValue, int rowIndex, int columnIndex) {
183	super.setValueAt(aValue, rowIndex, columnIndex);
184	if (columnIndex == 0) {
185	install.set(rowIndex, (Boolean) aValue);
186
187	int size = 0;
188	for (int i = 0; i < items.size(); i++) {
189	if (install.get(i)) {
190	Mod m = items.get(i);
191	if (!m.isLocalAvailable())
192	size += m.getZipSize();
193	}
194	}
195	for (DownloadSizeListener dsl : listeners)
196	dsl.downloadSizeChanged(size);
197	}
198	}
199
200	/**
201	* @param lis
202	* Listener to receive download size changed events
203	*/
204	public void addDownloadSizeListener(DownloadSizeListener lis) {
205	listeners.add(lis);
206	}
207
208	/**
209	* @param lis
210	* Listener to no longer receive download size changed events
211	*/
212	public void removeDownloadSizeListener(DownloadSizeListener lis) {
213	listeners.remove(lis);
214	}
215
216	}