* Code cleanup;

[Persons_Comparator.git] / src / Region.java

import java.io.File;
import java.io.FileNotFoundException;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class Region {
    private String csvFile = "data/countries.csv";
    private String[] continents = {"Africa", "Americas", "Asia", "Europe", "Oceania"};
    private ArrayList<Country> countryArrayListEurope = new ArrayList<>();
    private ArrayList<Country> countryArrayListAfrica = new ArrayList<>();
    private ArrayList<Country> countryArrayListAmericas = new ArrayList<>();
    private ArrayList<Country> countryArrayListAsia = new ArrayList<>();
    private ArrayList<Country> countryArrayListOceania = new ArrayList<>();

    public Region() {
        this.loadCSVCountries();
    }

    public String[] getContinents() {
        return continents;
    }

    public ArrayList<Country> getCountryArrayListAfrica() {
        return countryArrayListAfrica;
    }

    public ArrayList<Country> getCountryArrayListAsia() {
        return countryArrayListAsia;
    }

    public ArrayList<Country> getCountryArrayListAmericas() {
        return countryArrayListAmericas;
    }

    public ArrayList<Country> getCountryArrayListEurope() {
        return countryArrayListEurope;
    }

    public ArrayList<Country> getCountryArrayListOceania() {
        return countryArrayListOceania;
    }

    public void loadCSVCountries() {
        Scanner scanner = null;
        try {
            scanner = new Scanner(new File(csvFile));
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }
        while (scanner.hasNext()) {
            List<String> line = CSVUtils.parseLine(scanner.nextLine());
            String countryList = line.get(0);
            String[] countryArray = countryList.split(",");
            String latLng = line.get(16);
            String[] latLngArray = latLng.split(",");
            if (line.get(12).equals("Europe")) {
                Country c = new Country(countryArray[1], line.get(12), Double.parseDouble(latLngArray[0]), Double.parseDouble(latLngArray[1]));
                countryArrayListEurope.add(c);
            } else if (line.get(12).equals("Africa")) {
                Country c = new Country(countryArray[1], line.get(12), Double.parseDouble(latLngArray[0]), Double.parseDouble(latLngArray[1]));
                countryArrayListAfrica.add(c);
            } else if (line.get(12).equals("Americas")) {
                Country c = new Country(countryArray[1], line.get(12), Double.parseDouble(latLngArray[0]), Double.parseDouble(latLngArray[1]));
                countryArrayListAmericas.add(c);
            } else if (line.get(12).equals("Asia")) {
                Country c = new Country(countryArray[1], line.get(12), Double.parseDouble(latLngArray[0]), Double.parseDouble(latLngArray[1]));
                countryArrayListAsia.add(c);
            } else if (line.get(12).equals("Oceania")) {
                Country c = new Country(countryArray[1], line.get(12), Double.parseDouble(latLngArray[0]), Double.parseDouble(latLngArray[1]));
                countryArrayListOceania.add(c);
            }
        }
        scanner.close();
        /*Utils.displayArrayList(countryArrayListOceania);
        Utils.displayArrayList(countryArrayListAfrica);
        Utils.displayArrayList(countryArrayListAmericas);
        Utils.displayArrayList(countryArrayListAsia);
        Utils.displayArrayList(countryArrayListEurope);*/
    }
}