mirror of
https://github.com/csd4ni3l/connect-the-current.git
synced 2026-01-01 04:13:41 +01:00
89 lines
3.4 KiB
Python
89 lines
3.4 KiB
Python
import arcade, arcade.gui, random
|
|
|
|
from utils.constants import button_style
|
|
from utils.preload import button_texture, button_hovered_texture
|
|
|
|
from collections import deque
|
|
|
|
from game.cells import *
|
|
|
|
class Game(arcade.gui.UIView):
|
|
def __init__(self, pypresence_client, difficulty):
|
|
super().__init__()
|
|
|
|
self.pypresence_client = pypresence_client
|
|
self.pypresence_client.update(state='In Game', start=self.pypresence_client.start_time)
|
|
|
|
self.difficulty = difficulty
|
|
self.cells = []
|
|
self.power_sources = []
|
|
self.houses = []
|
|
|
|
self.anchor = self.add_widget(arcade.gui.UIAnchorLayout(size_hint=(1, 1)))
|
|
self.grid_size = list(map(int, difficulty.split("x")))
|
|
self.power_grid = self.anchor.add(arcade.gui.UIGridLayout(horizontal_spacing=0, vertical_spacing=0, row_count=self.grid_size[0], column_count=self.grid_size[1]))
|
|
|
|
def on_show_view(self):
|
|
super().on_show_view()
|
|
|
|
self.back_button = arcade.gui.UITextureButton(texture=button_texture, texture_hovered=button_hovered_texture, text='<--', style=button_style, width=100, height=50)
|
|
self.back_button.on_click = lambda event: self.main_exit()
|
|
self.anchor.add(self.back_button, anchor_x="left", anchor_y="top", align_x=5, align_y=-5)
|
|
|
|
for row in range(self.grid_size[0]):
|
|
self.cells.append([])
|
|
for col in range(self.grid_size[1]):
|
|
left_neighbour = self.cells[row][col - 1] if col > 0 else None
|
|
top_neighbour = self.cells[row - 1][col] if row > 0 else None
|
|
|
|
cell_type = random.choice(["line", "corner", "t_junction", "cross", "power_source", "house"])
|
|
|
|
if cell_type in ["line", "corner", "t_junction", "cross"]:
|
|
cell = PowerLine(cell_type, left_neighbour, top_neighbour)
|
|
elif cell_type == "power_source":
|
|
cell = PowerSource(left_neighbour, top_neighbour)
|
|
self.power_sources.append(cell)
|
|
elif cell_type == "house":
|
|
cell = House(left_neighbour, top_neighbour)
|
|
self.houses.append(cell)
|
|
|
|
self.power_grid.add(cell, row=row, column=col)
|
|
self.cells[row].append(cell)
|
|
|
|
if left_neighbour:
|
|
left_neighbour.right_neighbour = cell
|
|
if top_neighbour:
|
|
top_neighbour.bottom_neighbour = cell
|
|
|
|
arcade.schedule(self.update_grid, 1 / 8)
|
|
|
|
def update_grid(self, _):
|
|
for row in self.cells:
|
|
for power_line in row:
|
|
if power_line.cell_type != "power_source":
|
|
power_line.powered = False
|
|
|
|
queue = deque(self.power_sources)
|
|
visited = set()
|
|
|
|
while queue:
|
|
current = queue.popleft()
|
|
|
|
if id(current) in visited:
|
|
continue
|
|
|
|
visited.add(id(current))
|
|
|
|
current.powered = True
|
|
|
|
for connected_neighbour in current.get_connected_neighbours():
|
|
if id(connected_neighbour) not in visited:
|
|
queue.append(connected_neighbour)
|
|
|
|
for row in self.cells:
|
|
for power_line in row:
|
|
power_line.update_visual()
|
|
|
|
def main_exit(self):
|
|
from menus.main import Main
|
|
self.window.show_view(Main(self.pypresence_client)) |