From a55be7234204b100e7a29bad57ad942612e7bde5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Magdalena=20Wilczy=C5=84ska?= <magdawil26@gmail.com>
Date: Mon, 25 Mar 2019 16:03:25 +0100
Subject: [PATCH] Added statistics utils

---
 game.py                      |  3 ++-
 sprites/garbage_collector.py | 14 ++++++++++++++
 sprites/house.py             |  5 ++++-
 utils.py                     | 33 +++++++++++++++++++++++++++++++++
 4 files changed, 53 insertions(+), 2 deletions(-)

diff --git a/game.py b/game.py
index d9d4def..d3be8b2 100644
--- a/game.py
+++ b/game.py
@@ -7,7 +7,7 @@ from pygame.locals import *
 import utils
 
 ##INITIALIZE STATIC VARIABLES#########
-FPS = 60
+FPS = 20
 
 all_sprites = sprite.Group()
 fps_clock = time.Clock()
@@ -59,6 +59,7 @@ while(1):
         if(type(item) == House):
             item.generate_rubbish()
 
+    utils.get_statistics(all_sprites)
     display.flip()
     fps_clock.tick(FPS)
 ##################################################################################
diff --git a/sprites/garbage_collector.py b/sprites/garbage_collector.py
index 89b7544..1185e59 100644
--- a/sprites/garbage_collector.py
+++ b/sprites/garbage_collector.py
@@ -4,6 +4,8 @@ from sprites.cell import Cell
 
 class Garbage_collector(Cell):
     def __init__(self, x, y):
+        GC_CAPACITY = 10
+
         Cell.__init__(self, x, y)
         self.image = pygame.image.load("images/garbage_collector.png")
         self.move_options = {
@@ -12,6 +14,12 @@ class Garbage_collector(Cell):
             "left": lambda forbidden: ('x', self.x - 1) if (self.x - 1, self.y) not in forbidden else ('x', self.x),
             "right": lambda forbidden: ('x', self.x + 1) if (self.x + 1, self.y) not in forbidden else ('x', self.x)
         }
+        self.trash_space_taken = {
+            "plastic": 0,
+            "glass": 0,
+            "metal": 0
+        }
+        self.trash_collected = 0
 
     def move(self, direction, forbidden):
         (destination, value) = self.move_options[direction](forbidden)
@@ -21,3 +29,9 @@ class Garbage_collector(Cell):
             self.y = value
 
         self.update()
+    
+    def get_collect_data(self):
+        return self.trash_collected
+
+    def get_space_data(self):
+        return self.trash_space_taken
diff --git a/sprites/house.py b/sprites/house.py
index 8ac9576..a101078 100644
--- a/sprites/house.py
+++ b/sprites/house.py
@@ -22,7 +22,7 @@ class House(Cell):
         self.max_metal = max_metal
 
     def generate_rubbish(self):
-        if(random.randint(0, 50) == 1):  # 1/5 szansa na wyrzucenie śmiecia w klatce
+        if(random.randint(0, 25) == 1):  # 1/25 szansa na wyrzucenie śmiecia w klatce
             thrash_type = random.randint(0, 2)
             self.rubbish[thrash_type] = self.rubbish[thrash_type] + 1
 
@@ -48,3 +48,6 @@ class House(Cell):
     def check_rubbish_status(self):
         print("plastic: " + str(self.rubbish[PLASTIC]) + " glass: " + str(
             self.rubbish[GLASS]) + " metal: " + str(self.rubbish[METAL]))
+    
+    def get_rubbish_data(self):
+        return self.rubbish
diff --git a/utils.py b/utils.py
index 46cc6f8..4cd11bf 100644
--- a/utils.py
+++ b/utils.py
@@ -87,3 +87,36 @@ def generate_garbage_collector(all_sprites, obstacles_coords):
     all_sprites.add(gc)
     return gc
 ##################################################################################
+
+##GET STATISTICS##################################################################
+
+def get_statistics(all_sprites):
+    ###Garbage collector stats###
+    gc_taken_space_plastic = 0
+    gc_taken_space_metal = 0
+    gc_taken_space_glass = 0
+    gc_trash_space = 10
+
+    ###Board stats###############
+    plastic_left = 0
+    metal_left = 0
+    glass_left = 0
+    total_gathered = 0
+
+    for item in all_sprites:
+        if(type(item) == House):
+            rubbish = item.get_rubbish_data()
+            plastic_left += rubbish[0]
+            glass_left += rubbish[1]
+            metal_left += rubbish[2]
+        if(type(item) == Garbage_collector):
+            space_taken = item.get_space_data()
+            gc_taken_space_plastic += space_taken.get("plastic")
+            gc_taken_space_glass += space_taken.get("glass")
+            gc_taken_space_metal += space_taken.get("metal")
+            total_gathered += item.get_collect_data()
+            
+    print("plastic left: "+str(plastic_left)+" | glass left: "+str(glass_left)+" | metal left: "+str(metal_left))
+    print("    plastic: "+str(gc_taken_space_plastic)+"/"+str(gc_trash_space)+" | glass: "+str(gc_taken_space_glass)+"/"+str(gc_trash_space)+" | metal: "+str(gc_taken_space_metal)+"/"+str(gc_trash_space))
+    print("### TOTAL COLLECTED: "+str(total_gathered)+" ###")
+##################################################################################