Skip to content

Commit 8fdebfc

Browse files
fix class names and kick_decision_maker
1 parent 9a0445b commit 8fdebfc

File tree

6 files changed

+18
-19
lines changed

6 files changed

+18
-19
lines changed

src/behaviors/bhv_starter_clearball.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from pyrusgeom import vector_2d
77
from service_pb2 import *
88

9-
class ClearBall(IBehavior):
9+
class BhvStarterClearBall(IBehavior):
1010

1111
def __init__(self):
1212
pass

src/behaviors/bhv_starter_dribble.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from service_pb2 import *
77

88

9-
class Dribble(IBehavior):
9+
class BhvStarterDribble(IBehavior):
1010

1111
def __init__(self):
1212
pass

src/behaviors/bhv_starter_kick_planner.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -3,37 +3,37 @@
33
from pyrusgeom.soccer_math import *
44
from pyrusgeom.geom_2d import *
55
from service_pb2 import *
6-
from src.behaviors.bhv_starter_clearball import ClearBall
7-
from src.behaviors.bhv_starter_pass import Pass
8-
from src.behaviors.bhv_starter_dribble import Dribble
9-
from src.behaviors.bhv_starter_shoot import Shoot
6+
from src.behaviors.bhv_starter_clearball import BhvStarterClearBall
7+
from src.behaviors.bhv_starter_pass import BhvStarterPass
8+
from src.behaviors.bhv_starter_dribble import BhvStarterDribble
9+
from src.behaviors.bhv_starter_shoot import BhvStarterShoot
1010
from src.utils.tools import Tools
1111

1212
class BhvStarterKickPlanner(IBehavior):
1313
def __init__(self):
14-
self.shoot = Shoot()
15-
self.clear_ball = ClearBall()
16-
self.dribble = Dribble()
17-
self.pass_ball = Pass()
14+
self.starter_shoot = BhvStarterShoot()
15+
self.starter_clear_ball = BhvStarterClearBall()
16+
self.starter_dribble = BhvStarterDribble()
17+
self.starter_pass = BhvStarterPass()
1818

1919
def execute(self, agent: IAgent):
2020
agent.logger.debug("BhvStarterKickPlanner.execute")
2121
from src.sample_player_agent import SamplePlayerAgent # Local import to avoid circular import
2222
actions = []
23-
actions += [shoot] if (shoot := self.shoot.execute(agent)) is not None else []
23+
actions += [shoot] if (shoot := self.starter_shoot.execute(agent)) is not None else []
2424
opps = Tools.OpponentsFromSelf(agent)
2525
nearest_opp = opps[0] if opps else None
2626
nearest_opp_dist = nearest_opp.dist_from_self if nearest_opp else 1000.0
2727

2828
if nearest_opp_dist < 10:
29-
actions += [passing] if (passing := self.pass_ball.execute(agent)) is not None else []
29+
actions += [passing] if (passing := self.starter_pass.execute(agent)) is not None else []
3030

31-
actions += [dribble] if (dribble := self.dribble.execute(agent)) is not None else []
31+
actions += [dribble] if (dribble := self.starter_dribble.execute(agent)) is not None else []
3232

3333
if nearest_opp_dist > 2.5:
3434
actions.append(PlayerAction(body_hold_ball=Body_HoldBall()))
3535

36-
actions.append(self.clear_ball.execute(agent))
36+
actions.append(self.starter_clear_ball.execute(agent))
3737

3838
#Sending actions' queue
3939
for i in actions:

src/behaviors/bhv_starter_pass.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from service_pb2 import *
77

88

9-
class Pass(IBehavior):
9+
class BhvStarterPass(IBehavior):
1010

1111
def __init__(self):
1212
pass

src/behaviors/bhv_starter_shoot.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66

77

8-
class Shoot(IBehavior):
8+
class BhvStarterShoot(IBehavior):
99
def __init__(self):
1010
pass
1111

src/decision_makers/kick_decision_maker.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,10 @@
1010
class KickDecisionMaker(IDecisionMaker):
1111
def __init__(self):
1212
self.bhv_kick_planner = BhvKickPlanner()
13-
self.bhv_kick_planner = BhvStarterKickPlanner()
13+
#self.bhv_kick_planner = BhvStarterKickPlanner()
1414

1515
def make_decision(self, agent: IAgent):
1616
agent.logger.debug("--- WithBallDecisionMaker ---")
1717
from src.sample_player_agent import SamplePlayerAgent # Local import to avoid circular import
1818
assert isinstance(agent, SamplePlayerAgent)
19-
#self.bhv_kick_planner.execute(agent)
20-
self.bhv_kick_planner.execute(agent)
19+
self.bhv_kick_planner.execute(agent)

0 commit comments

Comments
 (0)