Dodano mozliwosc zarzadania slotami

This commit is contained in:
jakzar 2024-03-10 15:03:56 +01:00
parent 2ef0664c48
commit adb2b15164
2 changed files with 87 additions and 14 deletions

22
Slot.py Normal file
View File

@ -0,0 +1,22 @@
import pygame
CUBE_SIZE=128
WIDTH = 1024
HEIGHT = 512
BLACK=(0,0,0)
BORDER_COLOR=BLACK
BORDER_THICKNESS=1 #Has to be INT value
class Slot:
def __init__(self,x_axis,y_axis,color,screen): #TODO crop and related to it things handling. for now as a place holder crop=>color
self.x_axis=x_axis
self.y_axis=y_axis
self.color=color
self.screen=screen
self.field=pygame.Rect(self.x_axis*CUBE_SIZE,self.y_axis*CUBE_SIZE,CUBE_SIZE,CUBE_SIZE)
def draw(self):
pygame.draw.rect(self.screen,self.color,self.field,0) #Draw field
pygame.draw.rect(self.screen,BLACK,self.field,BORDER_THICKNESS) #Draw border
pygame.display.update()
def color_change(self,color):
self.color=color
self.draw()

79
main.py
View File

@ -1,40 +1,91 @@
import pygame import pygame
import Slot
import random
import time
pygame.init() pygame.init()
BLACK = (0, 0, 0)
BROWN = (139, 69, 19)
WHITE = (255, 255, 255)
RED=(255,0,0)
GREEN=(0,255,0)
CUBE_SIZE = 128 CUBE_SIZE = 128
NUM_X = 8 NUM_X = 8
NUM_Y = 4 NUM_Y = 4
WIDTH = 1024 WIDTH = 1024
HEIGHT = 512 HEIGHT = 512
screen = pygame.display.set_mode((WIDTH, HEIGHT)) screen = pygame.display.set_mode((WIDTH, HEIGHT))
WHITE = (255, 255, 255)
screen.fill(WHITE) screen.fill(WHITE)
pygame.display.update() pygame.display.update()
BLACK = (0, 0, 0)
BROWN = (139, 69, 19)
SLOT_DICT={} #Slot are stored in dictionary with key being a Tuple of x and y coordinates so top left slot key is (0,0) and value is slot object
"""#Draw grid and tractor (old one)
def draw_grid(): def draw_grid():
for x in range(NUM_X): for x in range(0, 8):
for y in range(NUM_Y): pygame.draw.line(screen, BLACK, (x*CUBE_SIZE, 0), (x*CUBE_SIZE, HEIGHT)) #We got 8 lines in Y axis so to draw them we use x from range <0,7) to make slot manage easier
pygame.draw.rect(screen, BROWN, (x * CUBE_SIZE, y * CUBE_SIZE, CUBE_SIZE, CUBE_SIZE)) for y in range(0, 5):
pygame.draw.line(screen, BLACK, (0, y*CUBE_SIZE), (WIDTH, y*CUBE_SIZE)) #We got 4 lines in X axis so to draw them we use y from range <0,5) to make slot manage easier
for x in range(0, WIDTH, CUBE_SIZE):
pygame.draw.line(screen, BLACK, (x, 0), (x, HEIGHT))
for y in range(0, HEIGHT, CUBE_SIZE):
pygame.draw.line(screen, BLACK, (0, y), (WIDTH, y))
# Draw tractor # Draw tractor
tractor_image = pygame.image.load('images/traktor.png')
tractor_image = pygame.transform.scale(tractor_image, (CUBE_SIZE, CUBE_SIZE))
screen.blit(tractor_image, (CUBE_SIZE - 128, CUBE_SIZE - 128))"""
def draw_tractor(): #TODO? I think we should move draw_tractor to tractor class in the future
tractor_image = pygame.image.load('images/traktor.png') tractor_image = pygame.image.load('images/traktor.png')
tractor_image = pygame.transform.scale(tractor_image, (CUBE_SIZE, CUBE_SIZE)) tractor_image = pygame.transform.scale(tractor_image, (CUBE_SIZE, CUBE_SIZE))
screen.blit(tractor_image, (CUBE_SIZE - 128, CUBE_SIZE - 128)) screen.blit(tractor_image, (CUBE_SIZE - 128, CUBE_SIZE - 128))
pygame.display.update()
#Draw grid and tractor (new one)
def draw_grid():
for x in range(0,8):
for y in range(0,5):
new_slot=Slot.Slot(x,y,BROWN,screen)
SLOT_DICT[(x,y)]=new_slot
for entity in SLOT_DICT:
SLOT_DICT[entity].draw()
draw_tractor()
def change_color_of_slot(coordinates,color):
SLOT_DICT[coordinates].color_change(color)
def random_color():
x=random.randint(0,3)
switcher={0:BROWN,
1:GREEN,
2:RED,
3:WHITE}
return switcher[x]
def randomize_colors():
font=pygame.font.Font('freesansbold.ttf',32)
text=font.render('Randomizing crops',True,BLACK,WHITE)
textRect=text.get_rect()
textRect.center=(WIDTH//2,HEIGHT//2)
screen.blit(text,textRect)
pygame.display.update()
time.sleep(3)
for coordinates in SLOT_DICT:
SLOT_DICT[coordinates].color_change(random_color())
draw_tractor()
def init_demo():
draw_grid()
time.sleep(2)
randomize_colors()
init_demo()
while True: while True:
for event in pygame.event.get(): for event in pygame.event.get():
if event.type == pygame.QUIT: if event.type == pygame.QUIT:
quit() quit()
draw_grid()
pygame.display.update()