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 78b880325e81201eab2ebc0eabdf0b38eb1612d7..20ba0c983f9cee3250b1e2899c0bcb6ba2ffc20b 100644 --- a/src/main/java/fr/univlille/sae/classification/model/Iris.java +++ b/src/main/java/fr/univlille/sae/classification/model/Iris.java @@ -137,23 +137,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 ae3f033e92c77040d04bb0342008e2819e4bd136..04aadaded9d64cec60d8f6df63bcbd2f5b2a3995 100644 --- a/src/main/java/fr/univlille/sae/classification/model/LoadableData.java +++ b/src/main/java/fr/univlille/sae/classification/model/LoadableData.java @@ -84,11 +84,11 @@ public abstract class LoadableData { */ public abstract String[] getAttributesName(); - /** - * 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 1311287cdc4219dda769e5a2c94a2311dbddf904..b6d65eed681d452df28b40406416c6f727a00f43 100644 --- a/src/main/java/fr/univlille/sae/classification/model/Pokemon.java +++ b/src/main/java/fr/univlille/sae/classification/model/Pokemon.java @@ -106,54 +106,7 @@ public class Pokemon extends LoadableData{ }; } - /** - * 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