import java.util.ArrayList;
import javax.swing.table.AbstractTableModel;
public class CountriesTable extends AbstractTableModel
{
	private static final long serialVersionUID = 1L;
	private ArrayList<Country> kraje = new ArrayList<Country>();
	private String[] nazwyKolumn = {"Panstwo", "Stolica", "Ludnosc", "Flaga"};

	public void add(Country nowyElement)
	{
		kraje.add(nowyElement);
	}

	public String getColumnName(int col)
	{
		return nazwyKolumn[col].toString();
	}

	public Class getColumnClass(int column)
    {
        return getValueAt(0, column).getClass();
    }

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

	public int getColumnCount()
	{
		return nazwyKolumn.length;
	}

	public Object getValueAt(int wiersz, int kolumna)
	{
		Country kraj = kraje.get(wiersz);
		switch(kolumna)
		{
			case 0:
				return kraj.getPanstwo();
			case 1:
				return kraj.getStolica();
			case 2:
				return kraj.getLudnosc();
			case 3:
				return kraj.getFlaga();
			default:
				return 0;
		}
	}

	public boolean isCellEditable(int row, int col)
	{
		return false;
	}
}