def approachBall(player): """ Once we are alligned with the ball, approach it """ if player.firstFrame(): player.hasAlignedOnce = False player.brain.tracker.trackBall() player.brain.CoA.setRobotGait(player.brain.motion) #if player.brain.ball.locDist > constants.APPROACH_ACTIVE_LOC_DIST: if transitions.shouldActiveLoc(player): player.brain.tracker.activeLoc() else : player.brain.tracker.trackBall() if player.penaltyKicking and \ player.ballInOppGoalBox(): return player.goNow('penaltyBallInOppGoalbox') # Switch to other states if we should if player.brain.playbook.role == pbc.GOALIE: if transitions.shouldKick(player): return player.goNow('waitBeforeKick') elif transitions.shouldPositionForKick(player): return player.goNow('positionForKick') elif transitions.shouldTurnToBall_ApproachBall(player): return player.goLater('turnToBall') elif not player.brain.tracker.activeLocOn and \ transitions.shouldScanFindBall(player): return player.goLater('scanFindBall') else: if transitions.shouldDribble(player): return player.goNow('dribble') elif transitions.shouldKick(player): return player.goNow('waitBeforeKick') elif transitions.shouldPositionForKick(player): return player.goNow('positionForKick') return approachBallWalk(player)
def approachBall(player): """ Once we are alligned with the ball, approach it """ if player.firstFrame(): player.hasAlignedOnce = False player.brain.tracker.trackBall() player.brain.CoA.setRobotGait(player.brain.motion) #if player.brain.ball.locDist > constants.APPROACH_ACTIVE_LOC_DIST: if transitions.shouldActiveLoc(player): player.brain.tracker.activeLoc() else: player.brain.tracker.trackBall() if player.penaltyKicking and \ player.ballInOppGoalBox(): return player.goNow('penaltyBallInOppGoalbox') # Switch to other states if we should if player.brain.play.isRole(GOALIE): if transitions.shouldKick(player): return player.goNow('waitBeforeKick') elif transitions.shouldPositionForKick(player): return player.goNow('positionForKick') elif transitions.shouldTurnToBall_ApproachBall(player): return player.goLater('turnToBall') elif not player.brain.tracker.activeLocOn and \ transitions.shouldScanFindBall(player): return player.goLater('scanFindBall') else: if transitions.shouldDribble(player): return player.goNow('dribble') elif transitions.shouldKick(player): return player.goNow('waitBeforeKick') elif transitions.shouldPositionForKick(player): return player.goNow('positionForKick') return approachBallWalk(player)
def approachBallWithLoc(player): if player.firstFrame(): player.brain.CoA.setRobotGait(player.brain.motion) player.hasAlignedOnce = False nav = player.brain.nav my = player.brain.my if player.brain.playbook.role == pbc.GOALIE: if transitions.shouldKick(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goNow('waitBeforeKick') elif transitions.shouldPositionForKickFromApproachLoc(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('positionForKick') elif my.locScoreFramesBad > constants.APPROACH_NO_LOC_THRESH: player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('approachBall') elif not player.brain.tracker.activeLocOn and \ transitions.shouldScanFindBall(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('scanFindBall') else: if transitions.shouldKick(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goNow('waitBeforeKick') elif transitions.shouldPositionForKickFromApproachLoc(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('positionForKick') elif transitions.shouldNotGoInBox(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('ballInMyBox') elif transitions.shouldChaseAroundBox(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('chaseAroundBox') elif transitions.shouldAvoidObstacleDuringApproachBall(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('avoidObstacle') elif my.locScoreFramesBad > constants.APPROACH_NO_LOC_THRESH: player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('approachBall') elif not player.brain.tracker.activeLocOn and \ transitions.shouldScanFindBall(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('scanFindBall') elif player.brain.tracker.activeLocOn and \ transitions.shouldScanFindBallActiveLoc(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('scanFindBall') #if player.brain.ball.locDist > constants.APPROACH_ACTIVE_LOC_DIST: if transitions.shouldActiveLoc(player): player.brain.tracker.activeLoc() else : player.brain.tracker.trackBall() dest = player.getApproachPosition() useOmni = MyMath.dist(my.x, my.y, dest[0], dest[1]) <= \ constants.APPROACH_OMNI_DIST changedOmni = False if useOmni != nav.movingOmni: player.changeOmniGoToCounter += 1 else : player.changeOmniGoToCounter = 0 if player.changeOmniGoToCounter > PositionConstants.CHANGE_OMNI_THRESH: changedOmni = True if player.firstFrame() or \ nav.destX != dest[0] or \ nav.destY != dest[1] or \ nav.destH != dest[2] or \ changedOmni: if not useOmni: player.brain.CoA.setRobotGait(player.brain.motion) nav.goTo(dest) else: player.brain.CoA.setRobotSlowGait(player.brain.motion) nav.omniGoTo(dest) return player.stay()
def approachBallWithLoc(player): if player.firstFrame(): player.brain.CoA.setRobotGait(player.brain.motion) player.hasAlignedOnce = False nav = player.brain.nav my = player.brain.my if player.brain.play.isRole(GOALIE): if transitions.shouldKick(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goNow('waitBeforeKick') elif transitions.shouldPositionForKickFromApproachLoc(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('positionForKick') elif my.locScoreFramesBad > constants.APPROACH_NO_LOC_THRESH: player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('approachBall') elif not player.brain.tracker.activeLocOn and \ transitions.shouldScanFindBall(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('scanFindBall') else: if transitions.shouldKick(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goNow('waitBeforeKick') elif transitions.shouldPositionForKickFromApproachLoc(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('positionForKick') elif transitions.shouldNotGoInBox(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('ballInMyBox') elif transitions.shouldChaseAroundBox(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('chaseAroundBox') elif transitions.shouldAvoidObstacleDuringApproachBall(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('avoidObstacle') elif my.locScoreFramesBad > constants.APPROACH_NO_LOC_THRESH: player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('approachBall') elif not player.brain.tracker.activeLocOn and \ transitions.shouldScanFindBall(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('scanFindBall') elif player.brain.tracker.activeLocOn and \ transitions.shouldScanFindBallActiveLoc(player): player.brain.CoA.setRobotGait(player.brain.motion) return player.goLater('scanFindBall') #if player.brain.ball.locDist > constants.APPROACH_ACTIVE_LOC_DIST: if transitions.shouldActiveLoc(player): player.brain.tracker.activeLoc() else: player.brain.tracker.trackBall() dest = player.getApproachPosition() useOmni = my.dist(dest) <= \ constants.APPROACH_OMNI_DIST changedOmni = False if useOmni != nav.movingOmni(): player.changeOmniGoToCounter += 1 else: player.changeOmniGoToCounter = 0 if player.changeOmniGoToCounter > PositionConstants.CHANGE_OMNI_THRESH: changedOmni = True if player.firstFrame() or \ nav.dest != dest or \ changedOmni: if not useOmni: player.brain.CoA.setRobotGait(player.brain.motion) nav.goTo(dest) else: player.brain.CoA.setRobotSlowGait(player.brain.motion) nav.omniGoTo(dest) return player.stay()