class Output():
def __init__(self, cv_window):
self.commands = KeyboardCommands()
self.cv_window = cv_window
def send_output(self):
try:
while True:
self.commands.set_command(self.cv_window.get_output())
self.commands.send_command()
except KeyboardInterrupt:
pass
class MainWindow(QMainWindow):
"""
This is a class that creates a PyQt5 window containing start, stop, and exit buttons as
well as gesture icons. The window is the main window for our program.
Attributes:
commands: Instantiate keyboard command class object for sending commands to Google Earth
stop_commands (bool) : Flag for enabling and disabling worker thread
capture: Instantiate QtCapture class when start video button is pressed
google_earth: GoogleEarth class object
new_position (int) : X and Y position values of Google Earth window
window_resize (int) : Width and height of resized Google Earth window
title_bar_offset (int) : Offset value to account for windowless frame when positioning
window
threadpool : Instantiate QThreadpool class
layout : Instantiate QVBoxLayout or QHBoxLayout classes, used for aligning images and
buttons
label_dict (dict) : Contains labels that are used to display gesture icon images
image_list (list) : Contains names of image files for gesture icon images
title_list (list) : Contains names of titles for corresponding gesture images
label : Instantiate QLabel class, labels used to hold gesture icon images and text
button : Instantiate QPushButton class, buttons used to start, stop, and exit program
widget : Insantiate QWidget class, contains and displays labels and buttons
Args:
earth : GoogleEarth class object
"""
def __init__(self, earth, *args, **kwargs):
"""
Please see help(MainWindow) for more info
"""
super(MainWindow, self).__init__(*args, **kwargs)
# Instantiate KeyboardCommands class
self.commands = KeyboardCommands()
# Flag for stopping commands to GE window
self.stop_commands = False
# Will hold hand_recognition QtCapture class
self.capture = None
# Make Qt gesture icon window frameless
self.setWindowFlags(QtCore.Qt.FramelessWindowHint)
# Get resolution, window size, and offsets for positioning
self.google_earth = earth
self.new_position = self.google_earth.get_screen_position()
self.window_resize = self.google_earth.get_screen_resize()
self.title_bar_offset = 35
# Set geometry of Qt gesture icon window
# (this window is the parent of all other Qt windows)
self.setGeometry(
QtWidgets.QStyle.alignedRect(
QtCore.Qt.LeftToRight,
QtCore.Qt.AlignCenter,
# Width of Qt gesture window based on width of GE window
QtCore.QSize(int(self.window_resize[0]), 100),
QtWidgets.qApp.desktop().availableGeometry()))
# Initialize threadpool object
self.threadpool = QThreadPool()
# Create layouts for organizing Qt gesture icon window
self.layout = QVBoxLayout()
self.layout1 = QHBoxLayout()
self.layout2 = QHBoxLayout()
self.layout3 = QHBoxLayout()
self.label_dict = dict()
self.image_list = [
'index_up.png', 'fist.png', 'palm.png', 'thumb_left.png',
'thumb_right.png', 'five_wide.png'
]
self.title_list = [
'Move Up', 'Zoom In', 'Placeholder', 'Move Left', 'Move Right',
'Zoom Out'
]
# Create and add 6 labels containing hand gesture image to layout2 and 6
# labels with the gesture descriptions to layout1
for num in range(0, 6):
self.label = QLabel(self)
self.pixmap = QPixmap(self.image_list[num])
self.pixmap = self.pixmap.scaledToWidth(100)
self.label.setPixmap(self.pixmap)
self.label_title = QLabel(self.title_list[num])
# Modify left margins for images and labels positioning
if num == 0:
self.label.setContentsMargins(25, 0, 0, 0)
self.label_title.setContentsMargins(35, 0, 0, 0)
elif num == 5:
self.label.setContentsMargins(25, 0, 0, 0)
self.label_title.setContentsMargins(35, 0, 0, 0)
else:
self.label.setContentsMargins(15, 0, 0, 0)
self.label_title.setContentsMargins(25, 0, 0, 0)
self.label_dict[num] = self.label
self.layout2.addWidget(self.label_dict[num])
self.layout1.addWidget(self.label_title)
# Create start button and connect it to start_opencv function
self.start_button = QPushButton("Start Video")
self.start_button.setStyleSheet("background-color: silver")
self.start_button.pressed.connect(self.start_opencv)
# Create stop button and connect it to stop_opencv function
self.stop_button = QPushButton("Stop Video")
self.stop_button.setStyleSheet("background-color: silver")
self.stop_button.pressed.connect(self.stop_opencv)
# Create stop button and connect it to stop_opencv function
self.exit_button = QPushButton("Exit Program")
self.exit_button.setStyleSheet("background-color: silver")
self.exit_button.pressed.connect(self.exit)
# Add start and stop button to layout 3
self.layout3.addWidget(self.start_button)
self.layout3.addWidget(self.stop_button)
self.layout3.addWidget(self.exit_button)
# Add layout 1, 2, and 3 to layout
self.layout.addLayout(self.layout1)
self.layout.addLayout(self.layout2)
self.layout.addLayout(self.layout3)
# Create widget to hold layout, add layout to widget
self.widget = QWidget()
self.widget.setLayout(self.layout)
# Set widget with layouts as central widget
self.setCentralWidget(self.widget)
def start_opencv(self):
"""
Slot function for the start button signal. Instantiates Qt opencv window if not created,
then starts and shows the window. Once the window is opened, starts worker thread to send
commands to Google Earth.
"""
# If opencv window not initialized,
if not self.capture:
# Instantiate QtCapture class, set parent and window flags
self.capture = QtCapture(self.google_earth)
self.capture.setParent(self)
self.capture.setWindowFlags(QtCore.Qt.Tool)
self.capture.setWindowTitle("OpenCV Recording Window")
self.capture.setGeometry(
int(self.window_resize[0] + self.new_position[0]),
int(self.window_resize[1] + self.title_bar_offset), -1, -1)
# Start video capture and show it
self.capture.start()
self.capture.show()
# Set stop command flag, create worker attached to send_output
# function, start worker as new thread
self.stop_commands = False
worker_one = Worker(self.send_output)
self.threadpool.start(worker_one)
def stop_opencv(self):
"""
Slot function for stop button signal. Stops Qt timer in opencv loop and sets
stop_commands to True to kill worker thread.
"""
# Stop timer in hand_recognition, set flag to kill worker thread
self.capture.stop()
self.stop_commands = True
def exit(self):
"""
Slot function for exit button signal. Sets stop_commands to True to kill worker
thread, then calls close_earth to close Google Earth window, and finally terminates
the QApplication.
"""
self.stop_commands = True
self.google_earth.close_earth()
QtCore.QCoreApplication.instance().quit()
def send_output(self):
"""
Gets current output from opencv window and sends the command to the Google Earth window
"""
# While stop command false, get commands from hand_recognition
# and send commands to Google Earth window
while True:
self.commands.set_command(self.capture.get_output())
self.commands.send_command()
if self.stop_commands:
break