Wyświetlanie domów i wysypiska

board

@@ -4,75 +4,119 @@ import pygame
 screen = []
 objectArray = []
 
-class Agent:
+
-    def __init__(self, name, xPos, yPos):
+class Position:
+    def __init__(self, x, y):
+        self.x = x
+        self.y = y
+
+
+class Object:
+    def __init__(self, name, pos):
         self.name = name
-        self.xPos = xPos
+        self.pos = pos
-        self.yPos = yPos
+
+    def draw(self, square):
+        pass
+
+
+class Agent(Object):
+    def __init__(self, name, pos):
+        super().__init__(name, pos)
+
+    def draw(self, square):
+        ## RYSUJEMY AGENTA
+        # agent_color = (255, 0, 0)
+        circleX = self.pos.x * square + square + square / 2  # dodane jedno +square, by śmieciara nie wychodziła poza kratę
+        circleY = self.pos.y * square - square / 2
+        # radius = 10
+        # pygame.draw.circle(screen, agent_color, (a + circleX, b + circleY), radius)
+        truck = pygame.image.load("car.png").convert_alpha()  # tu ścieżka do zdjęcia w tle
+        truck = pygame.transform.scale(truck, (square, square))
+        screen.blit(truck, (circleX, circleY))
+
+
+class House(Object):
+    def __init__(self, name, pos):
+        super().__init__(name, pos)
+
+    def draw(self, square):
+        leftTopX, leftTopY = 50 + self.pos.x * square, 10 + self.pos.y * square
+        pygame.draw.rect(screen, (0, 255, 0), (leftTopX, leftTopY, square, square))
+
+
+class Junkyard(Object):
+    def __init__(self, name, pos):
+        super().__init__(name, pos)
+
+    def draw(self, square):
+        leftTopX, leftTopY = 50 + self.pos.x * square, 10 + self.pos.y * square
+        pygame.draw.rect(screen, (255, 0, 0), (leftTopX, leftTopY, square, square))
 
 def draw(square_num, objectArr):
-    #następne dwie linijki do odkomentowania, jak będzie wgrane zdjęcie do tła
+    # następne dwie linijki do odkomentowania, jak będzie wgrane zdjęcie do tła
-    #background = pygame.image.load("ścieżka do pliku").convert() #tu ścieżka do zdjęcia w tle
+    # background = pygame.image.load("ścieżka do pliku").convert() #tu ścieżka do zdjęcia w tle
-    #screen.blit(background, (0, 0))
+    # screen.blit(background, (0, 0))
-    grid_color = (0, 0, 0) #kolor czarny
+    grid_color = (0, 0, 0)  # kolor czarny
 
-    grid_size = 500  #rozmiar kraty
+    grid_size = 500  # rozmiar kraty
-    square = grid_size/square_num #rozmiar pojedyńczego kwadracika
+    square = grid_size / square_num  # rozmiar pojedyńczego kwadracika
 
     a = 50
-    b = 10  #odległości kraty od krawędzi okna
+    b = 10  # odległości kraty od krawędzi okna
+
+    for o in objectArr:
+        o.draw(square)
 
     for i in range(square_num):
-        pygame.draw.line(screen, grid_color, (a + i*square, b), (a + i*square, b + grid_size), 2)
+        pygame.draw.line(screen, grid_color, (a + i * square, b), (a + i * square, b + grid_size), 2)
 
-        pygame.draw.line(screen, grid_color, (a, b + i*square), (a + grid_size, b + i*square), 2)
+        pygame.draw.line(screen, grid_color, (a, b + i * square), (a + grid_size, b + i * square), 2)
 
     pygame.draw.line(screen, grid_color, (a, b + grid_size), (a + grid_size,
-                     b + grid_size), 2)
+                                                              b + grid_size), 2)
     pygame.draw.line(screen, grid_color, (a + grid_size, b),
                      (a + grid_size, b + grid_size), 2)
 
     ## tutaj rysujemy agenta i inne obiekty juz na gotowej mapie
 
-    ## RYSUJEMY AGENTA
-    #agent_color = (255, 0, 0)
-    circleX = objectArr[0].xPos * square + square + square/ 2 #dodane jedno +square, by śmieciara nie wychodziła poza kratę
-    circleY = objectArr[0].yPos * square - square / 2
-    #radius = 10
-    #pygame.draw.circle(screen, agent_color, (a + circleX, b + circleY), radius)
-    truck = pygame.image.load("car.png").convert_alpha() #tu ścieżka do zdjęcia w tle
-    truck = pygame.transform.scale(truck, (square, square))
-    screen.blit(truck, (circleX, circleY))
 
 def kb_listen(objectArray, gridLength):
     for event in pygame.event.get():
         if event.type == pygame.KEYDOWN:
-            if event.key == pygame.K_LEFT and objectArray[0].xPos > 0:
+            if event.key == pygame.K_LEFT and objectArray[0].pos.x > 0:
-                objectArray[0].xPos = objectArray[0].xPos - 1
+                objectArray[0].pos.x = objectArray[0].pos.x - 1
-            if event.key == pygame.K_RIGHT and objectArray[0].xPos < gridLength - 1:
+            if event.key == pygame.K_RIGHT and objectArray[0].pos.x < gridLength - 1:
-                objectArray[0].xPos = objectArray[0].xPos + 1
+                objectArray[0].pos.x = objectArray[0].pos.x + 1
-            if event.key == pygame.K_UP and objectArray[0].yPos > 1:
+            if event.key == pygame.K_UP and objectArray[0].pos.y > 1:
-                objectArray[0].yPos = objectArray[0].yPos - 1
+                objectArray[0].pos.y = objectArray[0].pos.y - 1
-            if event.key == pygame.K_DOWN and objectArray[0].yPos < gridLength:
+            if event.key == pygame.K_DOWN and objectArray[0].pos.y < gridLength:
-                objectArray[0].yPos = objectArray[0].yPos + 1
+                objectArray[0].pos.y = objectArray[0].pos.y + 1
         if event.type == pygame.QUIT:
             sys.exit()
 
 
 if __name__ == '__main__':
-    pygame.init()  #inicjalizacja modułów, na razie niepotrzebna
+    pygame.init()  # inicjalizacja modułów, na razie niepotrzebna
 
-    #Tworzymy nowego playera, czy tam agenta
+    # Tworzymy nowego playera, czy tam agenta
-    agent = Agent("smieciarka", 5, 7)
+    agent = Agent("smieciarka", Position(5, 7))
+    junkyard = Junkyard("wysypisko", Position(10, 10))
+    housePositions = [Position(x, y) for x, y in [
+        (7, 4), (3, 10), (8, 10), (4, 5), (1, 2), (10, 4), (14, 14), (6, 9)
+    ]]
+    houses = [House(f'dom-{i}', pos) for i, pos in enumerate(housePositions)]
     objectArray.append(agent)
+    objectArray.append(junkyard)
+    objectArray += houses
 
     width = 600
     height = 530
-    screen = pygame.display.set_mode((width, height))  #ustalanie rozmiarów okna
+    screen = pygame.display.set_mode((width, height))  # ustalanie rozmiarów okna
 
     while 1:
-        c = (255, 255, 255)  #tymczasowy kolor tła - do usunięcia, jak już będzie zdjęcie
+        c = (255, 255, 255)  # tymczasowy kolor tła - do usunięcia, jak już będzie zdjęcie
         screen.fill(c)
         draw(15, objectArray)
         kb_listen(objectArray, 15)
-        pygame.display.update()  #by krata pojawiła się w okienku - update powierzchni
+        pygame.display.update()  # by krata pojawiła się w okienku - update powierzchni