1 files changed, 10 insertions, 4 deletions

diff --git a/src/ch/epfl/maze/graphics/Animation.java b/src/ch/epfl/maze/graphics/Animation.java
index 9882231..1858057 100644
--- a/src/ch/epfl/maze/graphics/Animation.java
+++ b/src/ch/epfl/maze/graphics/Animation.java
@@ -21,7 +21,6 @@ import java.util.TreeMap;
  * of animals.
  *
  * @author EPFL
- * @author Pacien TRAN-GIRARD
  */
 public final class Animation {
 
@@ -220,15 +219,22 @@ public final class Animation {
      */
     private BufferedImage loadImage(Animal animal) {
         // path = "img/superclass/class.png"
-        String folder = animal.getClass().getSuperclass().getSimpleName();
+        Class<?> superClass = animal.getClass().getSuperclass();
+        File f = new File("");
         String file = animal.getClass().getSimpleName();
-        String path = "img/" + folder + File.separator + file + ".png";
+        String folder = "";
+        while (superClass != null && !f.exists()) {
+            folder = superClass.getSimpleName();
+            String path = "img/" + folder + File.separator + file + ".png";
+            f = new File(path);
+            superClass = superClass.getSuperclass();
+        }
 
         // adds image to buffer if not already there
         BufferedImage img = mImages.get(folder + "." + file);
         if (img == null) {
             try {
-                img = ImageIO.read(new File(path));
+                img = ImageIO.read(f);
                 mImages.put(folder + "." + file, img);
             } catch (IOException e) {
                 e.printStackTrace();