X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2FOrigin.java;h=7d3e855c30ca7cb28690d408ffb61df29bde6ce9;hb=c8da8eb79910c7ea33dc2970ebd8d1d1db234a89;hp=60e7fc9e641a40a6690458dffa0e1f8395469f59;hpb=1c1189337121a1baa74961813e619c99b398a333;p=Persons_Comparator.git

diff --git a/src/Origin.java b/src/Origin.java
index 60e7fc9..7d3e855 100644
--- a/src/Origin.java
+++ b/src/Origin.java
@@ -1,8 +1,32 @@
+import java.util.ArrayList;
+import java.util.Arrays;
+
 public class Origin implements Comparable<Origin> {
+    private String continent = "Unknown";
     private String country;
+    private String[] continents = {"Africa", "Americas", "Asia", "Europe", "Oceania"};
+    private String[][] content2DArray = {
+            {"Mali"}, /* Africa */
+            {"United States of America", "Brasil", "Mexico"}, /* Americas */
+            {"India"}, /* Asia */
+            {"France", "Germany", "United Kingdom", "Belgium", "Norway"}, /* Europe */
+            {"Australia"} /* Oceania */
+    };
+
+    Origin() {
+    }
 
     Origin(String country) {
         setCountry(country);
+        setContinent(this.getContinentFromCountry(country));
+    }
+
+    public String getContinent() {
+        return continent;
+    }
+
+    public void setContinent(String continent) {
+        this.continent = continent;
     }
 
     public String getCountry() {
@@ -13,6 +37,31 @@ public class Origin implements Comparable<Origin> {
         this.country = country;
     }
 
+    public String[] getCountriesArray() {
+        ArrayList<String> stringArrayList = new ArrayList<>();
+        for (int i = 0; i < this.content2DArray.length; i++) {
+            for (int j = 0; j < this.content2DArray[i].length; j++) {
+                stringArrayList.add(this.content2DArray[i][j]);
+            }
+        }
+        stringArrayList.sort(String::compareToIgnoreCase);
+        String[] rtArray = new String[stringArrayList.size()];
+        return stringArrayList.toArray(rtArray);
+    }
+
+    public String getContinentFromCountry(String country) {
+        int continentKey = -1;
+        for (int i = 0; i < content2DArray.length; i++) {
+            if (Arrays.asList(content2DArray[i]).indexOf(country) != -1) {
+                continentKey = i;
+            }
+        }
+        if (continentKey != -1)
+            return this.continents[continentKey];
+        else
+            return "Unknown";
+    }
+
     @Override
     public int compareTo(Origin origin) {
         return 0;