src/Eye.java

   1 import java.awt.Color;
   2 import java.util.Arrays;
   3
   4 public class Eye implements Comparable<Eye> {
   5     private String strColor;
   6     private Color color;
   7     private String[] colorsList = {"black", "blue", "brown", "green"};
   8
   9     Eye() {
  10         sortColorList();
  11     }
  12
  13     /**
  14      * @param color
  15      */
  16     Eye(String color) {
  17         setStrColor(color);
  18         setColor(color);
  19         sortColorList();
  20     }
  21
  22     public String getStrColor() {
  23         return strColor;
  24     }
  25
  26     public void setStrColor(String strColor) {
  27         if (validateColor(strColor)) {
  28             this.strColor = strColor;
  29         } else {
  30             throw new IllegalArgumentException("Color must be" + new Eye().colorListToString());
  31         }
  32     }
  33
  34     public String[] getColorsList() {
  35         return colorsList;
  36     }
  37
  38     /**
  39      * @return
  40      */
  41     public Color getColor() {
  42         return color;
  43     }
  44
  45     /**
  46      * @param color
  47      */
  48     public void setColor(String color) {
  49         if (validateColor(color)) {
  50             if (color.equals("brown"))
  51                 this.color = new Color(88, 41, 0);
  52             else
  53                 this.color = Color.getColor(color);
  54
  55         } else {
  56             throw new IllegalArgumentException("Color must be" + new Eye().colorListToString());
  57         }
  58     }
  59
  60     /**
  61      * @param color
  62      * @return
  63      */
  64     private boolean validateColor(String color) {
  65         for (String c : colorsList) {
  66             if (color.equals(c)) {
  67                 return true;
  68             }
  69         }
  70         return false;
  71     }
  72
  73     private void sortColorList() {
  74         Arrays.sort(this.colorsList);
  75     }
  76
  77     /**
  78      * @param eye
  79      * @return
  80      */
  81     @Override
  82     public int compareTo(Eye eye) {
  83         double r_diff = this.getColor().getRed() - eye.getColor().getRed();
  84         double g_diff = this.getColor().getGreen() - eye.getColor().getGreen();
  85         double b_diff = this.getColor().getBlue() - eye.getColor().getBlue();
  86         //See https://en.wikipedia.org/wiki/Color_difference
  87         Double distance = Math.sqrt(2 * Math.pow(r_diff, 2) + 4 * Math.pow(g_diff, 2) + 3 * Math.pow(b_diff, 2));
  88         return distance.intValue();
  89     }
  90
  91     private String colorListToString() {
  92         StringBuilder stringBuilder = new StringBuilder();
  93         for (String c : colorsList) {
  94             stringBuilder.append(" ");
  95             stringBuilder.append(c);
  96         }
  97         return stringBuilder.toString();
  98     }
  99 }