inteligentny-traktor/main.py

import pygame


class Field:
    def __init__(self, fieldType, plantType, isWet, wetTime, isFertilized, fertilizedTime):
        self.fieldType = fieldType              # good/bad
        self.plantType = plantType              # wheat/carrot/cabbage
        self.isWet = isWet                      # yes/no
        self.wetTime = wetTime                  # number
        self.isFertilized = isFertilized        # yes/no
        self.fertilizedTime = fertilizedTime    # number


class Plant:
    def __init__(self, plantType, growthState):
        self.plantType = plantType              # wheat/carrot/cabbage
        self.growthState = growthState          # growing/grown


class Fertilizer:
    def __init__(self, fertilizerType):
        self.fertilizerType = fertilizerType    # wheat/carrot/cabbage


class Player:
    x = 0
    y = 0
    rotation = 0


T = [[Field(1,0,0,0,0,0),Field(0,0,1,0,0,0),Field(1,1,1,0,0,0),Field(1,2,0,0,0,0),Field(0,3,1,0,0,0),Field(0,0,0,0,0,0),Field(0,0,0,0,0,0),Field(1,0,1,0,0,0),Field(1,0,0,0,0,0),Field(1,0,1,0,0,0)],
     [Field(1,0,0,0,0,0),Field(0,0,1,0,0,0),Field(1,1,1,0,0,0),Field(1,2,0,0,0,0),Field(0,3,1,0,0,0),Field(0,0,0,0,0,0),Field(0,0,0,0,0,0),Field(1,0,1,0,0,0),Field(1,0,0,0,0,0),Field(1,0,1,0,0,0)],
     [Field(1,0,0,0,0,0),Field(0,0,1,0,0,0),Field(1,1,1,0,0,0),Field(1,2,0,0,0,0),Field(0,3,1,0,0,0),Field(0,0,0,0,0,0),Field(0,0,0,0,0,0),Field(1,0,1,0,0,0),Field(1,0,0,0,0,0),Field(1,0,1,0,0,0)],
     [Field(1,0,0,0,0,0),Field(0,0,1,0,0,0),Field(1,1,1,0,0,0),Field(1,2,0,0,0,0),Field(0,3,1,0,0,0),Field(0,0,0,0,0,0),Field(0,0,0,0,0,0),Field(1,0,1,0,0,0),Field(1,0,0,0,0,0),Field(1,0,1,0,0,0)],
     [Field(1,0,0,0,0,0),Field(0,0,1,0,0,0),Field(1,1,1,0,0,0),Field(1,2,0,0,0,0),Field(0,3,1,0,0,0),Field(0,0,0,0,0,0),Field(0,0,0,0,0,0),Field(1,0,1,0,0,0),Field(1,0,0,0,0,0),Field(1,0,1,0,0,0)],
     [Field(1,0,0,0,0,0),Field(0,0,1,0,0,0),Field(1,1,1,0,0,0),Field(1,2,0,0,0,0),Field(0,3,1,0,0,0),Field(0,0,0,0,0,0),Field(0,0,0,0,0,0),Field(1,0,1,0,0,0),Field(1,0,0,0,0,0),Field(1,0,1,0,0,0)],
     [Field(1,0,0,0,0,0),Field(0,0,1,0,0,0),Field(1,1,1,0,0,0),Field(1,2,0,0,0,0),Field(0,3,1,0,0,0),Field(0,0,0,0,0,0),Field(0,0,0,0,0,0),Field(1,0,1,0,0,0),Field(1,0,0,0,0,0),Field(1,0,1,0,0,0)],
     [Field(1,0,0,0,0,0),Field(0,0,1,0,0,0),Field(1,1,1,0,0,0),Field(1,2,0,0,0,0),Field(0,3,1,0,0,0),Field(0,0,0,0,0,0),Field(0,0,0,0,0,0),Field(1,0,1,0,0,0),Field(1,0,0,0,0,0),Field(1,0,1,0,0,0)],
     [Field(1,0,0,0,0,0),Field(0,0,1,0,0,0),Field(1,1,1,0,0,0),Field(1,2,0,0,0,0),Field(0,3,1,0,0,0),Field(0,0,0,0,0,0),Field(0,0,0,0,0,0),Field(1,0,1,0,0,0),Field(1,0,0,0,0,0),Field(1,0,1,0,0,0)],
     [Field(1,0,0,0,0,0),Field(0,0,1,0,0,0),Field(1,1,1,0,0,0),Field(1,2,0,0,0,0),Field(0,3,1,0,0,0),Field(0,0,0,0,0,0),Field(0,0,0,0,0,0),Field(1,0,1,0,0,0),Field(1,0,0,0,0,0),Field(1,0,1,0,0,0)]]

pygame.init()

player = Player()

screen = pygame.display.set_mode([600, 600])

running = True
clock = pygame.time.Clock()
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_LEFT:
                if player.x > 0:
                    player.x = player.x - 1
            if event.key == pygame.K_UP:
                if player.y > 0:
                    player.y = player.y - 1
            if event.key == pygame.K_RIGHT:
                if player.x < 9:
                    player.x = player.x + 1
            if event.key == pygame.K_DOWN:
                if player.y < 9:
                    player.y = player.y + 1

    screen.fill((175, 255, 50))
    imgWheat = pygame.image.load('img/wheat.png')
    imgCarrot = pygame.image.load('img/carrot.png')
    imgCabbage = pygame.image.load('img/cabbage.png')
    imgPlayer = pygame.image.load('img/player.png')

    i = 0
    while i < len(T):
        j = 0
        while j < len(T[i]):
            color = (0, 0, 0)
            if T[i][j].isWet == 0:
                color = (160, 80, 40)
            else:
                color = (50, 25, 0)
            pygame.draw.rect(screen, color, pygame.Rect(50 + 50 * i, 50 + 50 * j, 50, 50))
            if T[i][j].plantType == 1:
                screen.blit(imgWheat, (50 + 50 * i, 50 + 50 * j))
            if T[i][j].plantType == 2:
                screen.blit(imgCarrot, (50 + 50 * i, 50 + 50 * j))
            if T[i][j].plantType == 3:
                screen.blit(imgCabbage, (50 + 50 * i, 50 + 50 * j))
            j = j + 1
        i = i + 1

    i = 0
    while i < len(T)+1:
        pygame.draw.line(screen, (0, 0, 0), (50 + i * 50, 50), (50 + i * 50, 50 + len(T) * 50), 5)
        pygame.draw.line(screen, (0, 0, 0), (50, 50 + i * 50), (50 + len(T) * 50, 50 + i * 50), 5)
        i = i + 1

    tmpImg = pygame.transform.rotate(imgPlayer, player.rotation)
    screen.blit(tmpImg, (55 + 50 * player.x, 55 + 50 * player.y))
    pygame.display.flip()
# clock.tick(30)

# Done! Time to quit.

pygame.quit()