@@ -222,12 +222,12 @@ def loop(self):
222222 self .head_pub .publish (self .head_msg )
223223 elif key == "0" :
224224 # Search for Ball and track it if found
225- self .head_mode_msg .head_mode = HeadMode .BALL_MODE
226- assert int (key ) == HeadMode .BALL_MODE
225+ self .head_mode_msg .head_mode = HeadMode .SEARCH_BALL
226+ assert int (key ) == HeadMode .SEARCH_BALL
227227 elif key == "1" :
228228 # Look generally for all features on the field (ball, goals, corners, center point)
229- self .head_mode_msg .head_mode = HeadMode .FIELD_FEATURES
230- assert int (key ) == HeadMode .FIELD_FEATURES
229+ self .head_mode_msg .head_mode = HeadMode .SEARCH_FIELD_FEATURES
230+ assert int (key ) == HeadMode .SEARCH_FIELD_FEATURES
231231 elif key == "2" :
232232 # Simply look directly forward
233233 self .head_mode_msg .head_mode = HeadMode .LOOK_FORWARD
@@ -238,12 +238,12 @@ def loop(self):
238238 assert int (key ) == HeadMode .DONT_MOVE
239239 elif key == "4" :
240240 # Ball Mode adapted for Penalty Kick
241- self .head_mode_msg .head_mode = HeadMode .BALL_MODE_PENALTY
242- assert int (key ) == HeadMode .BALL_MODE_PENALTY
241+ self .head_mode_msg .head_mode = HeadMode .SEARCH_BALL_PENALTY
242+ assert int (key ) == HeadMode .SEARCH_BALL_PENALTY
243243 elif key == "5" :
244244 # Do a pattern which only looks in front of the robot
245- self .head_mode_msg .head_mode = HeadMode .LOOK_FRONT
246- assert int (key ) == HeadMode .LOOK_FRONT
245+ self .head_mode_msg .head_mode = HeadMode .SEARCH_FRONT
246+ assert int (key ) == HeadMode .SEARCH_FRONT
247247 elif key == "y" :
248248 # kick left forward
249249 pass
0 commit comments