pavel/kick-it
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

killed some refactoring bugs

Browse Source
This commit is contained in:
Pavel Lutskov
2018-06-04 10:37:03 +02:00
parent 194f6cda9d
commit 16aa1422e4
3 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
scripts/ball_approach.py
View File

@@ -22,6 +22,7 @@ class BallFollower(object):
self.counter = 0
def update(self):
print('in update loop')
try:
(x, y), radius = self.finder.find_colored_ball(
self.video_top.get_frame()

3
scripts/finders.py
View File

@@ -45,12 +45,13 @@ class BallFinder(object):
((x, y), radius) = cv2.minEnclosingCircle(c)
if radius < self.min_radius:
print('Nothin there')
self.history.appendleft(None)
return None
M = cv2.moments(c)
center = (int(M["m10"] / M["m00"]),int(M["m01"] // M["m00"]))
self.history.appendleft(center, int(radius))
self.history.appendleft((center, int(radius)))
return center, int(radius)
def visualize(self, frame):

4
scripts/movements.py
View File

@@ -67,14 +67,14 @@ class NaoMover(object):
self.mp.setStiffnesses("RKneePitch", stiffness)
def get_head_angles(self):
return self.mp.getAngles(('HeadYaw', 'HeadPitch'), useSensors=False)
return self.mp.getAngles(('HeadYaw', 'HeadPitch'), False)
def change_head_angles(self, d_yaw, d_pitch, speed):
self.mp.changeAngles(('HeadYaw', 'HeadPitch'),
(d_yaw, d_pitch), speed)
def set_head_angles(self, yaw, pitch, speed):
self.mp.setHeadAngles(('HeadYaw', 'HeadPitch'),
self.mp.setAngles(('HeadYaw', 'HeadPitch'),
(yaw, pitch), speed)
def move_to(self, front, side, rotation, wait=False):