You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tello_modules/test.py

35 lines
861 B

4 years ago
from TelloModule import TelloModule
import time
4 years ago
import matplotlib.pyplot as plt
import matplotlib.animation as animation
4 years ago
mod = TelloModule()
4 years ago
fig = plt.figure()
plt.ion()
x = 100
y = 150
line1 = plt.plot([x, 0],[0, y], c = 'b')
line2 = plt.plot([-x, 0],[0, y], c = 'b')
line3 = plt.plot([-x, 0],[0, -y], c = 'b')
line4 = plt.plot([x, 0],[0, -y], c = 'b')
line5 = plt.plot([-x/3, x/3], [y-y/3, y-y/3], c = 'b')
mod.module_init()
point = [None, None, None, None]
while 1:
if (point[0] != None):
for p in point:
p.remove()
point[0] = plt.scatter(0, mod.get_forward_dist(), c = 'r')
point[1] = plt.scatter(mod.get_right_dist(), 0, c = 'r')
point[2] = plt.scatter(-mod.get_left_dist(), 0, c = 'r')
point[3] = plt.scatter(0, -mod.get_back_dist(), c = 'r')
plt.pause(0.1)
plt.ioff()
plt.show()
mod.module_deinit()