* Add Levenshtein distance to Firstname class Comparable.
[Persons_Comparator.git] / src / Eye.java
1 import java.awt.Color;
2 import java.util.List;
3 import java.util.Arrays;
4
5 public class Eye implements Comparable<Eye> {
6 private Color color;
7 private List<String> colorList = Arrays.asList("black", "green", "blue", "brown");
8
9 /**
10 *
11 * @param color
12 */
13 Eye(String color) {
14 setColor(color);
15 }
16
17 /**
18 *
19 * @return
20 */
21 public Color getColor() {
22 return color;
23 }
24
25 /**
26 *
27 * @param color
28 */
29 public void setColor(String color) {
30 if (validateColor(color)) {
31 if (color.equals("brown"))
32 this.color = new Color(88, 41, 0);
33 else
34 this.color = Color.getColor(color);
35
36 } else {
37 throw new IllegalArgumentException("Color must be " + colorList.toString());
38 }
39 }
40
41 /**
42 *
43 * @param color
44 * @return
45 */
46 private boolean validateColor(String color) {
47 return colorList.contains(color);
48 }
49
50 /**
51 *
52 * @param eye
53 * @return
54 */
55 @Override
56 public int compareTo(Eye eye) {
57 double r_diff = this.getColor().getRed() - eye.getColor().getRed();
58 double g_diff = this.getColor().getGreen() - eye.getColor().getGreen();
59 double b_diff = this.getColor().getBlue() - eye.getColor().getBlue();
60 // See https://en.wikipedia.org/wiki/Color_difference
61 Double distance = Math.sqrt(2 * Math.pow(r_diff, 2) + 4 * Math.pow(g_diff, 2) + 3 * Math.pow(b_diff, 2));
62 return distance.intValue();
63 }
64 }