-
Notifications
You must be signed in to change notification settings - Fork 0
/
KeyboardControl.py
46 lines (40 loc) · 1.41 KB
/
KeyboardControl.py
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
from djitellopy import Tello
import time
import pygame
# Pygame window for keyboard press
pygame.init()
window = pygame.display.set_mode((400, 400))
tello = Tello()
tello.connect()
def process_key(events):
for event in events:
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_t:
tello.takeoff()
if event.key == pygame.K_l or event.key == pygame.K_q or event.key == pygame.K_ESCAPE:
tello.land()
if event.key == pygame.K_a:
tello.move_left(30)
if event.key == pygame.K_d:
tello.move_right(30)
if event.key == pygame.K_w:
tello.move_up(30)
if event.key == pygame.K_s:
tello.move_down(30)
if event.key == pygame.K_LEFT:
tello.rotate_counter_clockwise(30)
if event.key == pygame.K_RIGHT:
tello.rotate_clockwise(30)
if event.key == pygame.K_1:
tello.flip_left()
if event.key == pygame.K_2:
tello.flip_right()
if event.key == pygame.K_3:
tello.flip_back()
if event.key == pygame.K_4:
tello.flip_forward()
while True:
# Handle keyboard input to control drone
events = pygame.event.get()
process_key(events)
time.sleep(0.05)