main2.py

@@ -1,33 +1,111 @@
 import pygame
 import sys
-import tractor
-import src.screen as screen
+import random
+#import tractor
+#import src.screen as screen
 import src.plant as plant
 
-class Tractor2(pygame.sprite.Sprite):
-    def __init__(self,engine,transmission,pos_x,pos_y):
-        super.__init__()
-        self.image=pygame.image.load("../assets/tractor/tractor.png")
-        self.image=pygame.transform.scale(self,(36,36))
+screen_width=900
+screen_height=900
+SIZE = (screen_width, screen_height)
+SPECIES=["carrot","potatoe","beetroot","wheat"]
+# collected=0
+
+
+#agent class
+class Tractor(pygame.sprite.Sprite):
+    def __init__(self,engine,transmission):
+        super().__init__()
+        self.image=pygame.image.load("assets/tractor/tractor.png")
+        self.image=pygame.transform.scale(self.image,(36,36))
         self.rect = self.image.get_rect()
 
         self.engine=engine
         self.transmission=transmission
-        self.pos_x=pos_x
-        self.pos_y=pos_y
         self.fuel=100
     
+    def collect(self):
+         print("collected plant")
+         pygame.sprite.spritecollide(tractor,plant_grup,True)
+        #  collected=collected+1
+        #  print("plants in trunk "+collected)
+
     def update(self):
         self.rect.center=pygame.mouse.get_pos()
-        # self.pos_x=pos_x
-        # self.pos_y=pos_y
+
+#plant class
+class Plant(pygame.sprite.Sprite):
+    def __init__(self,species,is_ill,pos_x,pos_y):
+        super().__init__()
+        self.species=species
+        self.is_ill=is_ill
+
+        if species=="carrot":
+            self.growth_time=100
+            self.weight=50
+            self.fertilizer="carrot_fertilizer"
+            self.pic_path="assets/Carrot.png"
+        
+        if species=="beetroot":
+            self.growth_time=200
+            self.weight=200
+            self.fertilizer="beetroot_fertilizer"
+            self.pic_path="assets/Beetroot.png"
+
+        if species=="potato":
+            self.growth_time=100
+            self.weight=100
+            self.fertilizer="potatoe_fertilizer"
+            self.pic_path="assets/Potato.png"
+
+        else:
+            self.growth_time=250
+            self.weight=75
+            self.fertilizer="wheat_fertilizer"
+            self.pic_path="assets/Wheat.png"
+
+        self.image = pygame.image.load(self.pic_path) #zmienic
+        self.image = pygame.transform.scale(self.image,(36,36))
+        self.rect = self.image.get_rect()
+        self.rect.center = [pos_x,pos_y]
+        
+
+
 
 # pygame initialization
 pygame.init()
+clock = pygame.time.Clock()
 pygame.mouse.set_visible(False)
 
+#GAME SCREEN
+screen = pygame.display.set_mode(SIZE)
+pygame.display.set_caption("Traktor_interaktor")
+background = pygame.image.load("assets/farmland.jpg")
+background = pygame.transform.scale(background,SIZE)
+screen.fill((90,50,20))
+background.fill((90,50,20))
 
-myTractor = tractor.Tractor
+for line in range(25):
+            pygame.draw.line(background, (0, 0, 0), (0, line * 36), (SIZE[0], line * 36))
+            pygame.draw.line(background, (0, 0, 0), (line * 36, 0), (line * 36, SIZE[1]))
+
+# size in pixels of one tile = 36px/36px
+tile = (36, 36)
+
+# later move it to another class "barn"?
+barn_img = pygame.image.load('assets/barn.png')
+barn = pygame.transform.scale(barn_img, tile) 
+
+#Tractor
+tractor = Tractor('oil','manual')
+tractor_group = pygame.sprite.Group()
+tractor_group.add(tractor)
+
+#PLANTS
+plant_grup = pygame.sprite.Group()
+for plant in range(30):
+     new_plant = Plant(random.choice(SPECIES),0,random.randrange(0,screen_width),random.randrange(0,screen_height))
+     plant_grup.add(new_plant)
 
 if __name__ == "__main__":
     running = True
@@ -38,8 +116,13 @@ if __name__ == "__main__":
                 running = False
                 pygame.quit()
                 sys.exit()
-            # defines agent movement
-            tractor.movement(myTractor)
-        # screen visualisation
-        screen.set_screen(myTractor)
+            if event.type == pygame.MOUSEBUTTONDOWN:
+                 tractor.collect()
+            
+        pygame.display.flip()
+        screen.blit(background,(0,0))
+        plant_grup.draw(screen)
+        tractor_group.draw(screen)
+        tractor_group.update()
+        clock.tick(60)