1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
import cv2
import numpy as np
import pygame
import random
import threading
from cockroach import Cockroach
from asteroid import Asteroid
# Initialize Pygame
pygame.init()
# Define screen dimensions
SCREEN_WIDTH = 600
SCREEN_HEIGHT = 400
# Load cockroach image
cockroach_image = pygame.image.load('res/cockroach.png')
class MyGame:
def __init__(self) -> None:
# Open the webcam (0 is usually the default)
self.cap = cv2.VideoCapture(0)
# Create the game screen
self.screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
pygame.display.set_caption('Camera Feed and Cockroach Panel')
# Track whether the Pygame window should be closed
self.running = True
# Create an event to signal the camera feed thread to stop
self.stop_event = threading.Event()
def run_cockroach_game(self):
# Create a list to hold cockroaches
cockroaches = []
asteroids = []
while not self.stop_event.is_set():
for event in pygame.event.get():
if event.type == pygame.QUIT:
# Set the stop event when the 'x' button is clicked
self.stop_event.set()
# Spawn a new cockroach randomly
if len(cockroaches) < 5 and random.random() < 0.02:
cockroaches.append(Cockroach())
# Move cockroaches downwards and remove off-screen ones
for cockroach in cockroaches:
cockroach.move_down()
if cockroach.y >= SCREEN_HEIGHT:
cockroaches.remove(cockroach)
# Spawn a new asteroid randomly
if len(asteroids) < 5 and random.random() < 0.02:
asteroids.append(Asteroid())
# Move asteroid downwards and remove off-screen ones
for asteroid in asteroids:
asteroid.move_down()
if asteroid.y >= SCREEN_HEIGHT:
asteroids.remove(asteroid)
# Read frame from the webcam
ret, frame = self.cap.read()
# If frame is not returned, break
if not ret:
break
# Rotate the frame counter-clockwise by 90 degrees
frame = cv2.rotate(frame, cv2.ROTATE_90_COUNTERCLOCKWISE)
# Convert frame from RGB to HSV, and define frame used for detection
det_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
# Define lower green and upper green
lower_green = np.array([60, 100, 100])
upper_green = np.array([100, 255, 255])
# Define green mask
green_mask = cv2.inRange(det_frame, lower_green, upper_green)
# Detect contours
contours, _ = cv2.findContours(green_mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
# Find largest contour
largest_contour = max(contours, key=cv2.contourArea) if contours else None
# If largest contour is present, draw a green rectangle
if largest_contour is not None:
x, y, w, h = cv2.boundingRect(largest_contour)
cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
# Convert frame to Pygame surface
frame_pygame = pygame.surfarray.make_surface(frame_rgb)
# Draw the camera feed as the background
self.screen.blit(frame_pygame, (0, 0))
# Draw the cockroach panel on top of the camera feed
for cockroach in cockroaches:
cockroach.draw(self.screen)
# Draw the cockroach panel on top of the camera feed
for asteroid in asteroids:
asteroid.draw(self.screen)
# Update the display
pygame.display.flip()
# Release the captured webcam
self.cap.release()
# Destroy all CV2 Windows
cv2.destroyAllWindows()
def loop(self) -> None:
# Start the webcam feed and cockroach panel in separate threads
cockroach_thread = threading.Thread(target=self.run_cockroach_game)
cockroach_thread.start()
while self.running:
# Handle events for the Pygame window
for event in pygame.event.get():
if event.type == pygame.QUIT:
# Close the Pygame window
self.running = False
# Quit if 'q' is pressed
if cv2.waitKey(1) & 0xFF == ord('q'):
self.running = False
# Signal the camera feed thread to stop
self.stop_event.set()
if __name__ == '__main__':
game = MyGame()
# Start the webcam feed and cockroach panel together
game.loop()
|
