Movement and stoping time for tree cutting is improved

2 files changed, 8 insertions, 5 deletions

diff --git a/cutter.py b/cutter.py
index 81c6c16..f228863 100644
--- a/cutter.py
+++ b/cutter.py
@@ -6,9 +6,10 @@ cutter_images_number = 2
 cutter_images = ["images/cutter_%02d.png" % cutter_images_number for cutter_images_number in xrange(1, cutter_images_number + 1) ]
 
 class Cutter:
-	def __init__(self, screen_w, screen_h, cutter_speed):
+	def __init__(self, screen_w, screen_h, cutter_speed, cut_time_left):
 		self.cut_tree = False
 		self.speed = cutter_speed
+		self.cut_time = cut_time_left
 		self.image = pygame.image.load(cutter_images[randint(0, cutter_images_number - 1)])
 		image_rect = self.image.get_rect()
 		size_reduction = 5
diff --git a/main.py b/main.py
index 26a2f73..38f3d35 100644
--- a/main.py
+++ b/main.py
@@ -108,13 +108,15 @@ while True:
 	for arbol in trees:
 		for cortador in cutter:
 			if cortador.collides_with(arbol):
-				cortador.cut_trees = True
-				tree_cutting_time += 1
-				if cortador.cut_trees == True and tree_cutting_time == 100:
+				cortador.cut_tree = True
+				cortador.cut_time += 1
+				if cortador.cut_tree == True and cortador.cut_time == 100:
 					trees.remove(arbol)
+					cortador.cut_time = 0
+					cortador.cut_tree = False
 	
 	if cutter_time == 300:
-		cutter.append( Cutter(window_h, window_w, cutter_speed) )
+		cutter.append( Cutter(window_h, window_w, cutter_speed, 0) )
 		cutter_time = 0
 	cutter_time += 1