X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2FRegion.java;h=7c50c89a3184138ed043acb7cd13ccbc81de8328;hb=e16e3b1587a729dd1e2d39bd4426831136376aa7;hp=1deb36d938721e772dba6902620f455859b13f58;hpb=3fc1aed88123babd636df3a0de26a4a87e472497;p=Persons_Comparator.git

diff --git a/src/Region.java b/src/Region.java
index 1deb36d..7c50c89 100644
--- a/src/Region.java
+++ b/src/Region.java
@@ -1,15 +1,81 @@
+import java.io.File;
+import java.io.FileNotFoundException;
 import java.util.ArrayList;
 import java.util.List;
-import java.util.TreeMap;
+import java.util.Scanner;
 
-public class Region extends TreeMap<String, ArrayList<Country>> {
-    private List<Country> country;
+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 void setCountry(List<Country> origins) {
-        this.country = country;
+    public Region() {
+        this.loadCSVCountries();
     }
 
-    public List<Country> getCountry() {
-        return country;
+    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);*/
+    }
+}
\ No newline at end of file