X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2FEye.java;h=5177333340ccaae5405fce6264d927319ef3aa16;hb=7add5cb9b2213df01b559a59f39b033e27f50e72;hp=6edc8a563d99af95b373270e2936592551bcf2c7;hpb=089fcbfccd75b77daf843b22e128beeee1d1dc07;p=Persons_Comparator.git

diff --git a/src/Eye.java b/src/Eye.java
index 6edc8a5..5177333 100644
--- a/src/Eye.java
+++ b/src/Eye.java
@@ -1,16 +1,39 @@
 import java.awt.Color;
-import java.util.List;
 import java.util.Arrays;
 
-public class Eye implements Comparable<Eye> {
+public class Eye {
+    private String strColor;
     private Color color;
-    private List<String> colorList = Arrays.asList("black", "green", "blue", "brown");
+    private String[] colorsList = {"black", "blue", "brown", "green"};
+
+    Eye() {
+        //this.color = Color.white;
+        sortColorList();
+    }
 
     /**
      * @param color
      */
     Eye(String color) {
+        setStrColor(color);
         setColor(color);
+        sortColorList();
+    }
+
+    public String getStrColor() {
+        return strColor;
+    }
+
+    public void setStrColor(String strColor) {
+        if (validateColor(strColor)) {
+            this.strColor = strColor;
+        } else {
+            throw new IllegalArgumentException("Color must be" + this.colorListToString());
+        }
+    }
+
+    public String[] getColorsList() {
+        return colorsList;
     }
 
     /**
@@ -25,13 +48,16 @@ public class Eye implements Comparable<Eye> {
      */
     public void setColor(String color) {
         if (validateColor(color)) {
-            if (color.equals("brown"))
+            if (color.equals("black"))
+                this.color = Color.black;
+            else if (color.equals("blue"))
+                this.color = Color.blue;
+            else if (color.equals("brown"))
                 this.color = new Color(88, 41, 0);
-            else
-                this.color = Color.getColor(color);
-
+            else if (color.equals("green"))
+                this.color = Color.green;
         } else {
-            throw new IllegalArgumentException("Color must be " + colorList.toString());
+            throw new IllegalArgumentException("Color must be" + this.colorListToString());
         }
     }
 
@@ -40,20 +66,37 @@ public class Eye implements Comparable<Eye> {
      * @return
      */
     private boolean validateColor(String color) {
-        return colorList.contains(color);
+        for (String c : colorsList) {
+            if (color.equals(c)) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    private void sortColorList() {
+        Arrays.sort(this.colorsList);
     }
 
     /**
      * @param eye
      * @return
      */
-    @Override
-    public int compareTo(Eye eye) {
+    public int distanceTo(Eye eye) {
         double r_diff = this.getColor().getRed() - eye.getColor().getRed();
         double g_diff = this.getColor().getGreen() - eye.getColor().getGreen();
         double b_diff = this.getColor().getBlue() - eye.getColor().getBlue();
-        // See https://en.wikipedia.org/wiki/Color_difference
+        //See https://en.wikipedia.org/wiki/Color_difference
         Double distance = Math.sqrt(2 * Math.pow(r_diff, 2) + 4 * Math.pow(g_diff, 2) + 3 * Math.pow(b_diff, 2));
         return distance.intValue();
     }
+
+    private String colorListToString() {
+        StringBuilder stringBuilder = new StringBuilder();
+        for (String c : colorsList) {
+            stringBuilder.append(" ");
+            stringBuilder.append(c);
+        }
+        return stringBuilder.toString();
+    }
 }