Skip to content

Update bot_ai.py #227

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: develop
Choose a base branch
from
Open

Update bot_ai.py #227

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 7 additions & 7 deletions sc2/bot_ai.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -273,7 +273,7 @@ def is_near_to_expansion(t):

return closest

async def distribute_workers(self, resource_ratio: float = 2) -> None:
async def distribute_workers(self, resource_ratio: float = 2) -> None:
"""
Distributes workers across all the bases taken.
Keyword `resource_ratio` takes a float. If the current minerals to gas
Expand Down Expand Up @@ -353,13 +353,13 @@ async def distribute_workers(self, resource_ratio: float = 2) -> None:
if not possible_mining_places:
possible_mining_places = deficit_mining_places
# find closest mining place
current_place = min(deficit_mining_places, key=lambda place: place.distance_to(worker))
current_place = min(possible_mining_places, key=lambda place: place.distance_to(worker))

# remove it from the list
deficit_mining_places.remove(current_place)
# if current place is a gas extraction site, go there
if current_place.vespene_contents:
worker.gather(current_place)
# if current place is a gas extraction site,
self.do(worker.gather(current_place))
# go to the mineral field that is near and has the most minerals left
else:
local_minerals = (
Expand All @@ -368,17 +368,17 @@ async def distribute_workers(self, resource_ratio: float = 2) -> None:
# local_minerals can be empty if townhall is misplaced
target_mineral = max(local_minerals, key=lambda mineral: mineral.mineral_contents, default=None)
if target_mineral:
worker.gather(target_mineral)
self.do(worker.gather(target_mineral))
# more workers to distribute than free mining spots
# send to closest if worker is doing nothing
elif worker.is_idle and all_minerals_near_base:
target_mineral = min(all_minerals_near_base, key=lambda mineral: mineral.distance_to(worker))
worker.gather(target_mineral)
self.do(worker.gather(target_mineral))
else:
# there are no deficit mining places and worker is not idle
# so dont move him
pass

@property_cache_once_per_frame
def owned_expansions(self) -> dict[Point2, Unit]:
"""Dict of expansions owned by the player with mapping {expansion_location: townhall_structure}."""
Expand Down
Loading