package carpooling;

import java.util.Vector;
import javax.swing.table.AbstractTableModel;

public class TableModel extends AbstractTableModel {

    private Vector tratte_vector[] = new Vector[10];
    private Vector user_vector[] = new Vector[9];
    private Vector prenotazioni_vector[] = new Vector[2];

    public TableModel(int length) {

        if (length == 10) {
            for (int i = 0; i < length; i++) {
                tratte_vector[i] = new Vector<>();

            }
        } else if (length == 2) {
            for (int i = 0; i < length; i++) {
                prenotazioni_vector[i] = new Vector<>();

            }
        } else if (length == 9) {
            for (int i = 0; i < length; i++) {
                user_vector[i] = new Vector<>();

            }
        }
    }

    //TRATTE
    public int addElemento(String idTratta, String luogoPartenza, String luogoArrivo, String data, String oraPartenza, String oraArrivo, int costi, int postiDisponibili, String autista, String info) {
        int id = tratte_vector[0].size();
        tratte_vector[0].add(idTratta);
        tratte_vector[1].add(luogoPartenza);
        tratte_vector[2].add(luogoArrivo);
        tratte_vector[3].add(data);
        tratte_vector[4].add(oraPartenza);
        tratte_vector[5].add(oraArrivo);
        tratte_vector[6].add(costi);
        tratte_vector[7].add(postiDisponibili);
        tratte_vector[8].add(autista);
        tratte_vector[9].add(info);

        return id;
    }

    //UTENTI
    public int addElemento(String users, String md5, String nome, String cognome, String data_nascita, String cf, String email, String num_patente, String info) {
        int id = user_vector[0].size();
        user_vector[0].add(users);
        user_vector[1].add(md5);
        user_vector[2].add(nome);
        user_vector[3].add(cognome);
        user_vector[4].add(data_nascita);
        user_vector[5].add(cf);
        user_vector[6].add(email);
        user_vector[7].add(num_patente);
        user_vector[8].add(info);

        return id;
    }

    //PRENOTAZIONI
    public int addElemento(String idTratta, String users) {
        int id = prenotazioni_vector[0].size();
        prenotazioni_vector[0].add(idTratta);
        prenotazioni_vector[1].add(users);
        return id;
    }

//ADD_BLOCCO_ELEMENTI //
    //TRATTE
    public void addBloccoElementi(Tratta[] array_tratta) {


        for (int i = 0; i < array_tratta.length; i++) {
            addElemento(
                    array_tratta[i].getId(),
                    array_tratta[i].getLuogoPartenza(),
                    array_tratta[i].getLuogoArrivo(),
                    array_tratta[i].getData(),
                    array_tratta[i].getOraPartenza(),
                    array_tratta[i].getOraArrivo(),
                    array_tratta[i].getCosto(),
                    array_tratta[i].getPostiDisponibili(),
                    array_tratta[i].getAutista(),
                    array_tratta[i].getInfo());
        }

    }

    //PRENOTAZIONI
    public void addBloccoElementi(Prenotazione[] array_prenotazione) {


        for (int i = 0; i < array_prenotazione.length; i++) {
            addElemento(
                    array_prenotazione[i].getID(),
                    array_prenotazione[i].getUsers());
        }

    }

    //UTENTI
    public void addBloccoElementi(User[] array_user) {


        for (int i = 0; i < array_user.length; i++) {
            addElemento(
                    array_user[i].getUsers(),
                    array_user[i].getMd5(),
                    array_user[i].getNome(),
                    array_user[i].getCognome(),
                    array_user[i].getData_nascita(),
                    array_user[i].getCf(),
                    array_user[i].getEmail(),
                    array_user[i].getNum_patente(),
                    array_user[i].getInfo());
        }

    }

    public void remElemento(int index, String flag) {
        switch (flag) {
            case "tratte":
                for (int i = 0; i < tratte_vector.length; i++) {
                    tratte_vector[i].remove(index);
                }
                break;
            case "utenti":
                for (int i = 0; i < user_vector.length; i++) {
                    user_vector[i].remove(index);
                }
                break;
            case "prenotazioni":
                for (int i = 0; i < prenotazioni_vector.length; i++) {
                    prenotazioni_vector[i].remove(index);
                }
                break;
        }
    }

    public void clearList(String flag) {
        switch (flag) {
            case "tratte":
                for (int i = 0; i < tratte_vector.length; i++) {
                    tratte_vector[i].removeAllElements();
                }
                break;
            case "utenti":
                for (int i = 0; i < user_vector.length; i++) {
                    user_vector[i].removeAllElements();
                }
                break;
            case "prenotazioni":
                for (int i = 0; i < prenotazioni_vector.length; i++) {
                    prenotazioni_vector[i].removeAllElements();
                }
                break;
        }
    }

    public int getRowCount(String flag) {
        switch (flag) {
            case "tratte":
                return tratte_vector[0].size();

            case "utenti":
                return user_vector[0].size();

            case "prenotazioni":
                return prenotazioni_vector[0].size();
        }
        return 0;
    }

    public int getColumnCount(String flag) {
        switch (flag) {
            case "tratte":
                return tratte_vector.length;

            case "utenti":
                return user_vector.length;

            case "prenotazioni":
                return prenotazioni_vector.length;
        }
        return -1;

    }

    public Object getValueAt(int row, int col, String flag) {
        switch (flag) {
            case "tratte":
                return tratte_vector[col].get(row);

            case "utenti":
                return user_vector[col].get(row);

            case "prenotazioni":
                return prenotazioni_vector[col].get(row);
        }
        return -1;

    }


    public Class getColumnClass(int c, String flag) {
        return getValueAt(0, c, flag).getClass();
    }
};