refactoring

Author: educ8s

__pycache__/particle.cpython-312.pyc

@@ -1,6 +1,4 @@
 import pygame
-from particle import SandParticle
-from particle import RockParticle
 
 class Grid:
 	def __init__(self, width, height, cell_size):
@@ -10,7 +8,6 @@ def __init__(self, width, height, cell_size):
 		self.cells = [[None for _ in range(self.columns)] for _ in range(self.rows)]
 
 	def draw(self, window):
-
 		for row in range(self.rows): 
 			for column in range(self.columns):
 				particle = self.cells[row][column]

grid.py

@@ -1,4 +1,4 @@
-import pygame, sys
+import pygame
 from simulation import Simulation
 
 GREY = (29, 29, 29)

main.py

@@ -14,21 +14,18 @@ def draw(self, window):
 		self.grid.draw(window)
 		self.draw_brush(window)
 
-
 	def add_particle(self, row, column):
 	    if self.mode == "sand":
-	        # 50% chance to add a sand particle
-	        if random.random() < 0.2:  # 50% probability
+	        if random.random() < 0.2:
 	            particle = SandParticle
 	            self.grid.add_particle(row, column, particle)
 	    elif self.mode == "rock":
-	        # Always add a rock particle
 	        particle = RockParticle
 	        self.grid.add_particle(row, column, particle)
 
 	def update(self):
 		orientation = random.randint(0, 1)	
-		for row in range(self.grid.rows - 1, -1, -1):  # Start from the bottom		
+		for row in range(self.grid.rows - 1, -1, -1):
 			if orientation == 0:
 				for column in range(self.grid.columns - 1, -1, -1):
 					self.process_particle(row, column)
@@ -92,11 +89,9 @@ def apply_brush(self, row, column):
 					self.add_particle(current_row, current_col)
 
 	def draw_brush(self, window):
-
 		mouse_pos = pygame.mouse.get_pos()
 		column = mouse_pos[0] // self.cell_size
 		row = mouse_pos[1] // self.cell_size
-		
 		cursor_size = self.brush_size * self.cell_size
 		color = (255, 255, 255)
 
@@ -107,8 +102,4 @@ def draw_brush(self, window):
 		elif self.mode == "erase":
 			color = (255, 105, 180) 
 
-		pygame.draw.rect(
-			window,
-			color,
-			(column * self.cell_size, row * self.cell_size, cursor_size, cursor_size),
-		)
+		pygame.draw.rect(window, color, (column * self.cell_size, row * self.cell_size, cursor_size, cursor_size))