diff --git a/src/main/java/fr/univlille/sae/classification/model/Iris.java b/src/main/java/fr/univlille/sae/classification/model/Iris.java index abf271abb8ef1ba9bb655bab997a826b774bb18d..b91fd287cdb198a7bbd19255d89bbaaa600dd51f 100644 --- a/src/main/java/fr/univlille/sae/classification/model/Iris.java +++ b/src/main/java/fr/univlille/sae/classification/model/Iris.java @@ -141,23 +141,8 @@ public class Iris extends LoadableData { } - /** - * Renvoie la couleur associée à la variété de l'Iris. - * @return couleur correspondant à la variété. - */ - @Override - public Color getColor() { - switch (this.variety) { - case "Setosa": - return Color.RED; - case "Versicolor": - return Color.BLUE; - case "Virginica": - return Color.GREEN; - default: - return Color.BLACK; // Couleur par défaut si la variété est inconnue - } - } + + /** * Renvoie les noms des attributs de l'Iris. diff --git a/src/main/java/fr/univlille/sae/classification/model/LoadableData.java b/src/main/java/fr/univlille/sae/classification/model/LoadableData.java index e33aa56df15fe6e6a83e71adecae981d5508889c..11b71137b6bb3e9a51491dac47ab2c132f5ed4f0 100644 --- a/src/main/java/fr/univlille/sae/classification/model/LoadableData.java +++ b/src/main/java/fr/univlille/sae/classification/model/LoadableData.java @@ -55,11 +55,11 @@ public abstract class LoadableData { public abstract Map<String, Object> getAttributesNames(); - /** - * Renvoie la couleur associée à l'objet. - * @return couleur correspondant à la classification de l'objet. - */ - public abstract Color getColor(); + + + + + /** * Renvoie la valeur des données en fonction de l'axe spécifié. diff --git a/src/main/java/fr/univlille/sae/classification/model/Pokemon.java b/src/main/java/fr/univlille/sae/classification/model/Pokemon.java index f52f5b1a285687aab16c03cb5f9ff2c41298ef8d..8c8bdb0563c7e9c04aa76cbd2a8b7d3b4acace2f 100644 --- a/src/main/java/fr/univlille/sae/classification/model/Pokemon.java +++ b/src/main/java/fr/univlille/sae/classification/model/Pokemon.java @@ -112,54 +112,7 @@ public class Pokemon extends LoadableData { return attrNames; } - /** - * Renvoie la couleur associée à l'objet. - * - * @return couleur correspondant à la classification de l'objet. - */ - @Override - public Color getColor() { - switch (this.type1) { - case "normal": - return Color.LIGHTGREY; - case "grass": - return Color.GREEN; - case "electric": - return Color.YELLOW; - case "bug": - return Color.GREENYELLOW; - case "psychic": - return Color.PLUM; - case "poison": - return Color.PURPLE; - case "steel": - return Color.SILVER; - case "dragon": - return Color.WHITE; - case "flying": - return Color.SKYBLUE; - case "water": - return Color.BLUE; - case "rock": - return Color.SIENNA; - case "fire": - return Color.RED; - case "fairy": - return Color.PINK; - case "fighting": - return Color.FIREBRICK; - case "ice": - return Color.DARKTURQUOISE; - case "ghost": - return Color.DARKMAGENTA; - case "dark": - return Color.GREY; - case "ground": - return Color.KHAKI; - default: - return Color.BLACK; // Couleur par défaut si la variété est inconnue - } - } + /** * Renvoie la valeur des données en fonction de l'axe spécifié. diff --git a/src/test/java/fr/univlille/sae/classification/model/IrisTest.java b/src/test/java/fr/univlille/sae/classification/model/IrisTest.java index e0873a5f3a78475d65d874d2e7d40ab6788a9f0a..824be5f4bcf220de6165f03d342622ef2316bb37 100644 --- a/src/test/java/fr/univlille/sae/classification/model/IrisTest.java +++ b/src/test/java/fr/univlille/sae/classification/model/IrisTest.java @@ -40,7 +40,7 @@ class IrisTest { @Test void getColor() { - assertEquals(Color.RED, iris.getColor()); + assertEquals(Color.RED, iris.getClassifications().get(iris.getClassification())); } @Test