203 lines
3.9 KiB
Python
203 lines
3.9 KiB
Python
|
|
|
|
import random
|
|
|
|
from typing import List, Dict
|
|
|
|
|
|
|
|
class AIDecision:
|
|
|
|
def __init__(self, difficulty: str = "medium"):
|
|
|
|
self.difficulty = difficulty
|
|
|
|
self.last_decision_time = 0
|
|
|
|
self.decision_cooldown = self._get_decision_cooldown()
|
|
|
|
|
|
|
|
def decide(self, snapshot: Dict, dt: float) -> List[Dict]:
|
|
|
|
self.last_decision_time += dt
|
|
|
|
if self.last_decision_time < self.decision_cooldown:
|
|
|
|
return []
|
|
|
|
|
|
|
|
commands = []
|
|
|
|
commands.extend(self.plan_expand(snapshot))
|
|
|
|
commands.extend(self.plan_attack(snapshot))
|
|
|
|
commands.extend(self.plan_build(snapshot))
|
|
|
|
|
|
|
|
self.last_decision_time = 0
|
|
|
|
return commands
|
|
|
|
|
|
|
|
def plan_expand(self, snapshot: Dict) -> List[Dict]:
|
|
|
|
commands = []
|
|
|
|
resources = snapshot.get('resources', {})
|
|
|
|
minerals = resources.get('minerals', 0)
|
|
|
|
|
|
|
|
if minerals >= 400 and random.random() < 0.3:
|
|
|
|
commands.append({
|
|
|
|
'type': 'create_entity',
|
|
|
|
'entity_type': 'supply_depot',
|
|
|
|
'position': self._find_expansion_position(snapshot)
|
|
|
|
})
|
|
|
|
|
|
|
|
return commands
|
|
|
|
|
|
|
|
def plan_attack(self, snapshot: Dict) -> List[Dict]:
|
|
|
|
commands = []
|
|
|
|
entities = snapshot.get('entities', [])
|
|
|
|
ai_units = [e for e in entities if e.get('owner') == 'ai']
|
|
|
|
player_units = [e for e in entities if e.get('owner') == 'player']
|
|
|
|
|
|
|
|
if len(ai_units) >= 5 and player_units and random.random() < 0.4:
|
|
|
|
target_unit = random.choice(player_units)
|
|
|
|
for unit in ai_units:
|
|
|
|
if unit.get('type') == 'unit' and unit.get('state') == 'idle':
|
|
|
|
commands.append({
|
|
|
|
'type': 'attack',
|
|
|
|
'entity_id': unit['id'],
|
|
|
|
'target_id': target_unit['id']
|
|
|
|
})
|
|
|
|
|
|
|
|
return commands
|
|
|
|
|
|
|
|
def plan_build(self, snapshot: Dict) -> List[Dict]:
|
|
|
|
commands = []
|
|
|
|
resources = snapshot.get('resources', {})
|
|
|
|
minerals = resources.get('minerals', 0)
|
|
|
|
entities = snapshot.get('entities', [])
|
|
|
|
ai_units = [e for e in entities if e.get('owner') == 'ai']
|
|
|
|
|
|
|
|
if minerals >= 150 and len(ai_units) < 8 and random.random() < 0.5:
|
|
|
|
commands.append({
|
|
|
|
'type': 'create_entity',
|
|
|
|
'entity_type': 'worker',
|
|
|
|
'position': self._find_build_position(snapshot)
|
|
|
|
})
|
|
|
|
|
|
|
|
if minerals >= 300 and random.random() < 0.2:
|
|
|
|
commands.append({
|
|
|
|
'type': 'create_entity',
|
|
|
|
'entity_type': 'barracks',
|
|
|
|
'position': self._find_build_position(snapshot)
|
|
|
|
})
|
|
|
|
|
|
|
|
return commands
|
|
|
|
|
|
|
|
def _find_expansion_position(self, snapshot: Dict) -> tuple:
|
|
|
|
base_x = snapshot.get('ai_base_x', 100)
|
|
|
|
base_y = snapshot.get('ai_base_y', 100)
|
|
|
|
offset_x = random.randint(-200, 200)
|
|
|
|
offset_y = random.randint(-200, 200)
|
|
|
|
return (base_x + offset_x, base_y + offset_y)
|
|
|
|
|
|
|
|
def _find_build_position(self, snapshot: Dict) -> tuple:
|
|
|
|
base_x = snapshot.get('ai_base_x', 100)
|
|
|
|
base_y = snapshot.get('ai_base_y', 100)
|
|
|
|
offset_x = random.randint(-50, 50)
|
|
|
|
offset_y = random.randint(-50, 50)
|
|
|
|
return (base_x + offset_x, base_y + offset_y)
|
|
|
|
|
|
|
|
def _get_decision_cooldown(self) -> float:
|
|
|
|
if self.difficulty == "easy":
|
|
|
|
return 3.0
|
|
|
|
elif self.difficulty == "medium":
|
|
|
|
return 2.0
|
|
|
|
elif self.difficulty == "hard":
|
|
|
|
return 1.0
|
|
|
|
else:
|
|
|
|
return 2.0
|
|
|