Release v2.1 pushed

Author: Loïc Vanden Bemden

README.md

@@ -1,6 +1,32 @@
 # Blob Simulation & Detection
 Détection d'un blob dans une image ainsi que détection des comprimés de nourriture. Simulation d'un blob dans PyGame à travers un système multi-agents à connaissance partagée.
 
+## Release 2.1 - *7/06/2019*
+
+### Modifications : 
++ Déplacement des scripts dans le dossier root
++ Changement du nom des fichiers enregistrés par la détection
++ Automatisation des différents scripts, ajout d'un fonctionnement en mode "caché".
++ Automatisation de la simulation avec ajouts de paramètres et sauvegarde finale
++ Ajout d'un paramètre pour initialiser de la nourriture dans la simulation
++ Ajout d'un paramètre d'affichage dans la simulation
++ Uniformisation des paramètres dans les différents scripts
++ Possibilité de raffiner un modèle détecté avec un fichier d'informations supplémentaires
++ Variables de couleur déplacées dans un fichier json séparé
++ Ajout de fichiers d'exemples pour les différents scripts
+
+### Scripts : 
++ Setup : `python setup.py -i data/example.jpg`
++ Detection : `python detect.py -i data/example.jpg`
++ Simulation : `python play.py -i data/output-examples/example-detect.board -s 3`
++ Compare : `python compare.py --first data/output-examples/simulation/10_loops/10_loops.board --second data/output-examples/simulation/100_loops/100_loops.board -s 3`
+
+Les différentes sorties produites sont les suivantes : 
+![Sortie du script de détection](data/output-examples/example-detect-details.jpg?raw=true "Sortie du script de détection") 
+![Sortie de la simulation après 100 itérations](data/output-examples/simulation/100_loops/100_loops.jpg?raw=true "Sortie de la simulation après 100 itérations")
+![Sortie du script de comparaison](data/compare_init_with_100.jpg?raw=true "Sortie du script de comparaison") 
+
+
 ## Release 2.0 - *27/05/2019*
 
 Le passage à cette release invalide les fichiers de sauvegarde précédents, le format ayant été modifié.
@@ -17,7 +43,6 @@ La nouvelle interface de la simulation produira des images comme celle-ci :
 
 ![Interface de simulation](example/test-run.jpg?raw=true "Interface de simulation") 
 
-
 Pour exécuter le script de comparaison sur un exemple : 
 `python compare.py --first example/test.board --second example/test-run.board`
 

compare.py

@@ -17,46 +17,44 @@
 import argparse
 from simulation import board
 import pygame
-from pygame.locals import QUIT
+from pygame.locals import QUIT, KEYDOWN, K_ESCAPE
 
 
 def main():
     ap = argparse.ArgumentParser()
     ap.add_argument("--first", required=True, help="first board file")
     ap.add_argument("--second", required=True, help="second board file")
-    ap.add_argument("-s", "--scale", type=float, default=10, help="scale display (default: x10")
-    ap.add_argument("-o", "--output", type=str, help="give a name to save the jpeg file")
+    ap.add_argument("-s", "--scale", type=float, default=10,
+                    help="Scales board resolution by this factor (default: x10)")
+    ap.add_argument("-o", "--output", type=str, help="Give a name to save the jpeg file")
     args = ap.parse_args()
 
-    board_1 = board.Board(0,0)
+    board_1 = board.Board(0, 0)
     board_1.load(args.first)
 
-    board_2 = board.Board(0,0)
+    board_2 = board.Board(0, 0)
     board_2.load(args.second)
 
     board_comp = board_1.compare(board_2)
 
     if board_comp is None:
         return
 
-    pygame.init()
-
     width = int(board_1.width * args.scale)
     height = int(board_1.height * args.scale)
-    window = pygame.display.set_mode((width, height))
 
     game_surface = pygame.Surface((board_1.width, board_1.height))
     pixel_array = pygame.PixelArray(game_surface)
     for x in range(board_1.width):
         for y in range(board_1.height):
-            pixel_array[x,y] = (0, 0, 0)
+            pixel_array[x, y] = (0, 0, 0)
 
             val = max(-255, min(board_comp.get_blob(x, y), 255))
             if val < 0:
-                val = - val # int(-val/2) + 125
+                val = - val  # int(-val/2) + 125
                 pixel_array[x, y] = (val/4, val/4, val)
             elif val > 0:
-                val = val # int(val/2) + 125
+                val = val  # int(val/2) + 125
                 pixel_array[x, y] = (val, val/4, val/4)
             else:
                 if not board_comp.is_touched(x, y):
@@ -66,24 +64,27 @@ def main():
                         pixel_array[x, y] = (125, 75, 75)
 
             if board_comp.has_food(x, y):
-                 pixel_array[x, y] = (0, board_comp.foods[x, y], 0)
+                pixel_array[x, y] = (0, board_comp.foods[x, y], 0)
 
     del pixel_array
 
     game_window = pygame.transform.scale(game_surface, (width, height))
 
-    window.blit(game_window, (0, 0))
-    pygame.display.flip()
-
-    ended = False
-    while not ended:
-        pygame.time.wait(10)
-        for event in pygame.event.get():
-            if event.type == QUIT:
-                ended = True
-
     if args.output is not None:
-        pygame.image.save(window, args.output)
+        pygame.image.save(game_window, args.output)
+
+    else:
+        pygame.init()
+        window = pygame.display.set_mode((width, height))
+        window.blit(game_window, (0, 0))
+        pygame.display.flip()
+
+        ended = False
+        while not ended:
+            pygame.time.wait(10)
+            for event in pygame.event.get():
+                if event.type == QUIT or event.type == KEYDOWN and event.key == K_ESCAPE:
+                    ended = True
 
 
 if __name__ == "__main__":