src/Origin.java

   1 import java.util.ArrayList;
   2 import java.util.Arrays;
   3
   4 public class Origin implements Comparable<Origin> {
   5     private String continent = "Unknown";
   6     private String country;
   7     private String[] continents = {"Africa", "Americas", "Asia", "Europe", "Oceania"};
   8     private String[][] content2DArray = {
   9             {"Mali"}, /* Africa */
  10             {"United States of America", "Brasil", "Mexico"}, /* Americas */
  11             {"India"}, /* Asia */
  12             {"France", "Germany", "United Kingdom", "Belgium", "Norway"}, /* Europe */
  13             {"Australia"} /* Oceania */
  14     };
  15
  16
  17     Origin() {
  18     }
  19
  20     Origin(String country) {
  21         setCountry(country);
  22         setContinent(this.getContinentFromCountry(country));
  23     }
  24
  25     public String getContinent() {
  26         return continent;
  27     }
  28
  29     public void setContinent(String continent) {
  30         this.continent = continent;
  31     }
  32
  33     public String getCountry() {
  34         return country;
  35     }
  36
  37     public void setCountry(String country) {
  38         this.country = country;
  39     }
  40
  41     public String[] getCountriesArray() {
  42         ArrayList<String> stringArrayList = new ArrayList<>();
  43         for (int i = 0; i < this.content2DArray.length; i++) {
  44             for (int j = 0; j < this.content2DArray[i].length; j++) {
  45                 stringArrayList.add(this.content2DArray[i][j]);
  46             }
  47         }
  48         stringArrayList.sort(String::compareToIgnoreCase);
  49         String[] rtArray = new String[stringArrayList.size()];
  50         return stringArrayList.toArray(rtArray);
  51     }
  52
  53     public String getContinentFromCountry(String country) {
  54         int continentKey = -1;
  55         for (int i = 0; i < content2DArray.length; i++) {
  56             if (Arrays.asList(content2DArray[i]).indexOf(country) != -1) {
  57                 continentKey = i;
  58             }
  59         }
  60         if (continentKey != -1)
  61             return this.continents[continentKey];
  62         else
  63             return "Unknown";
  64     }
  65
  66     @Override
  67     public int compareTo(Origin origin) {
  68         return 0;
  69     }
  70 }