TableListTableModel.java

/**
 * This file is part of JSkat.
 *
 * JSkat 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.
 *
 * JSkat 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 JSkat.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.jskat.gui.swing.iss;

import java.util.ArrayList;
import java.util.List;

import javax.swing.table.AbstractTableModel;

import org.jskat.util.JSkatResourceBundle;

/**
 * Provides a model for the skat list table
 */
class TableListTableModel extends AbstractTableModel {

	private static final long serialVersionUID = 1L;

	private final JSkatResourceBundle strings;

	private final List<List<String>> data;
	private final List<String> columns;

	/**
	 * Constructor
	 */
	public TableListTableModel() {

		this.strings = JSkatResourceBundle.INSTANCE;
		this.data = new ArrayList<>();
		this.columns = new ArrayList<>();
		this.columns.add(this.strings.getString("name")); //$NON-NLS-1$
		this.columns.add(this.strings.getString("seats")); //$NON-NLS-1$
		this.columns.add(this.strings.getString("games")); //$NON-NLS-1$
		this.columns.add(this.strings.getString("player") + " 1"); //$NON-NLS-1$
		this.columns.add(this.strings.getString("player") + " 2"); //$NON-NLS-1$
		this.columns.add(this.strings.getString("player") + " 3"); //$NON-NLS-1$
	}

	/**
	 * @see AbstractTableModel#getColumnCount()
	 */
	@Override
	public int getColumnCount() {

		return this.columns.size();
	}

	/**
	 * @see AbstractTableModel#getRowCount()
	 */
	@Override
	public int getRowCount() {

		return this.data.size();
	}

	/**
	 * @see AbstractTableModel#getValueAt(int, int)
	 */
	@Override
	public Object getValueAt(int rowIndex, int columnIndex) {

		return this.data.get(rowIndex).get(columnIndex);
	}

	/**
	 * @see AbstractTableModel#getColumnName(int)
	 */
	@Override
	public String getColumnName(int col) {

		return this.columns.get(col);
	}

	/**
	 * Updates the skat table table model
	 * 
	 * @param tableName
	 *            Table name
	 * @param maxPlayers
	 *            Table type (3 or 4 players)
	 * @param gamesPlayed
	 *            Games played
	 * @param player1
	 *            Player 1
	 * @param player2
	 *            Player 2
	 * @param player3
	 *            Player 3
	 */
	public void updateTable(String tableName, int maxPlayers, long gamesPlayed,
			String player1, String player2, String player3) {

		boolean tableFound = false;
		int index = 0;

		// first try to find a player already known
		while (!tableFound && index < this.data.size()) {

			List<String> currRow = this.data.get(index);
			// check table name
			if (currRow.get(0).equals(tableName)) {
				// table found
				updateRow(index, maxPlayers, gamesPlayed, player1, player2,
						player3);
				tableFound = true;
			}
			index++;
		}

		if (!tableFound) {
			// table not found --> new table
			// add this table
			addRow(tableName, maxPlayers, gamesPlayed, player1, player2,
					player3);
		}
	}

	private void updateRow(int index, int maxPlayers, long gamesPlayed,
			String player1, String player2, String player3) {

		List<String> row = this.data.get(index);
		// set updated values
		row.set(1, Integer.toString(maxPlayers));
		row.set(2, Long.toString(gamesPlayed));
		row.set(3, player1);
		row.set(4, player2);
		row.set(5, player3);

		this.fireTableDataChanged();
	}

	private void addRow(String tableName, int maxPlayers, long gamesPlayed,
			String player1, String player2, String player3) {

		ArrayList<String> newLine = new ArrayList<>();
		newLine.add(tableName);
		newLine.add(Integer.toString(maxPlayers));
		newLine.add(Long.toString(gamesPlayed));
		newLine.add(player1);
		newLine.add(player2);
		newLine.add(player3);
		this.data.add(newLine);

		this.fireTableDataChanged();
	}

	public void removeTable(String tableName) {

		int index = 0;
		int removeIndex = 0;
		for (List<String> currRow : this.data) {

			if (currRow.get(0).equals(tableName)) {

				removeIndex = index;
			}
			index++;
		}
		this.data.remove(removeIndex);

		this.fireTableDataChanged();
	}
}