from src import template
from core.api import *

class Player(template.Player):
	def __init__(self):
		super().__init__()

	def run_planet(self):
		pass

	def run_destroyer(self):
		pass

	def run_miner(self):
		pass

	def run_scout(self):
		pass

import random

def try_build(self, type, energy, direction=None):
	# 先定义一个helper function，用以帮助建造实体
	# 尝试建造一个单位，如果没有指定方向，则顺时针依次尝试每一个方向。
	if self.controller.get_energy() < energy:  # 如果能量不足，直接返回False
		return False
	if direction is None:  # 如果没有指定方向，则尝试所有方向
		direction = Direction.all_directions()
	else:
		direction = [direction]
	for d in direction:  # 遍历所有方向
		if self.controller.can_build(type, d, energy):
			self.controller.build(type, d, energy)
			return True
	return False

def run_planet(self):
	# 所有的操作都通过self.controller执行，包括获取信息和行动
	current_round = self.controller.get_round_num()  # 获取当前回合数
	current_energy = self.controller.get_energy()  # 获取当前能量

	if self.controller.is_ready():  # 如果现在冷却值足够行动，则开始决定建造什么
		if current_round <= 100:  # 前100回合（前期）
			if current_energy > 100:
				# 如果能量大于100，则有30%概率建造一个destroyer，70%概率建造一个miner
				if random.random() > 0.7:
					self.try_build(EntityType("destroyer"), 30)
				else:
					self.try_build(EntityType("miner"), 75)
			elif current_energy > 30:
				# 如果能量值在30到100之间，则建造一个miner
				self.try_build(EntityType("miner"), 30)
			else:
				self.try_build(EntityType("scout"), 1)  # 如果能量小于30，则建造一个最便宜的scout
		else:  # 后期
			if current_energy > 200:
				# 如果能量大于200，则有80%概率建造一个destroyer，20%概率建造一个miner
				if random.random() > 0.2:
					self.try_build(EntityType("destroyer"), 60)
				else:
					self.try_build(EntityType("miner"), 100)
			elif current_energy > 75:  # 如果能量值在75到200之间，制造小型destroyer
				# 我们需要保证星球有一定能量储备，防止出现意外
				self.try_build(EntityType("destroyer"), 30)
			else:
				self.try_build(EntityType("scout"), 1)
	
	# 充能不占冷却，因此每轮都进行充能判定
	if self.controller.get_charge_point() > 500:
		return  # 如果已经有大于500点的充能，那我们就不需要再充能了
	if current_round < 100:
		pts = 1  # 前100回合，充能1点，如果对面不充能的话就可以获得一点
	elif current_round < 500:  # 中期充能20%
		pts = self.controller.get_energy() // 5
	else:  # 后期充能1/3
		pts = self.controller.get_energy() // 3
	
	if self.controller.can_charge(pts):
		self.controller.charge(pts)

def __init__(self):
	super().__init__()
	self.explore_dir = None  # 当前探索方向
	self.home_loc = None  # 母星位置，初始为None

def explore(self):  # 探索逻辑
	if self.explore_dir is None:
		# 如果探索方向没有被初始化，说明此时这个实体刚被创建出来。这意味着制造它的星球就在旁边。
		# 我们将初始探索方向设置为这个实体面向的方向。
		for entity in self.controller.sense_nearby_entities(radius=2, teams=[self.controller.get_team()]):
			# 遍历四周所有的同队实体
			if entity.type == EntityType("planet"):
				# 找到母星后，将探索方向设置为母星指向当前实体的方向
				self.explore_dir = entity.location.direction_to(self.controller.get_location())
				self.home_loc = entity.location  # 记录母星位置
				break
	
	if self.controller.is_ready():  # 如果实体可以行动
		for _ in range(8):  # 尝试8次探索，均失败说明四周全被堵死
			if self.controller.can_move(self.explore_dir):
				self.controller.move(self.explore_dir)
				return
			self.explore_dir = self.explore_dir.rotate_left()  # 左手法则

def run_scout(self):
	# 检查四周是否有敌人的miner
	for entity in self.controller.sense_nearby_entities(teams=self.controller.get_opponent()):
		if entity.type == EntityType("miner"):
			# 如果发现敌人的miner，则攻击它
			if self.controller.can_analyze(entity.location):
				self.controller.analyze(entity.location)
				return  # 如果直接分析成功，则进入冷却，可以直接跳过回合
			# 如果不能分析，那么说明要么自身在冷却中或者距离不够
			# 我们直接向敌人的方向接近
			d = self.controller.get_location().direction_to(entity.location)
			if self.controller.can_move(d):
				self.controller.move(d)
				return  # 如果移动成功，同理，可以直接跳过回合
	
	# 如果没有发现敌人的miner，则继续探索
	self.explore()

def run_destroyer(self):
	# 检查四周是否有敌方planet
	for entity in self.controller.sense_nearby_entities(teams=self.controller.get_opponent()):
		if entity.type == EntityType("planet"):
			# 如果发现敌人的planet，为了最大化伤害，我们优先接近
			d = self.controller.get_location().direction_to(entity.location)
			if self.controller.can_move(d):
				self.controller.move(d)
				return
			# 如果无法更接近了，我们就直接攻击
			dis = self.controller.get_location().distance_to(entity.location)
			if self.controller.can_overdrive(dis):
				self.controller.overdrive(dis)
				return
	
	# 如果没有发现敌人的planet，则继续探索
	self.explore()

def run_miner(self):
	if self.explore_dir is None:
		self.explore()
		return  # 如果探索方向没有被初始化，则直接开始explore
	if not self.controller.is_ready():
		return  # 如果实体不能行动，则直接返回
	# 检查四周是否有敌人的scout
	for entity in self.controller.sense_nearby_entities(teams=self.controller.get_opponent()):
		if entity.type == EntityType("scout"):
			# 如果发现敌人的scout，则向反方向逃跑
			d = entity.location.direction_to(self.controller.get_location())  # 对方指向自己的方向
			if self.controller.can_move(d):
				self.controller.move(d)
				return  # 直接逃跑

	# 如果没有发现敌人的scout，则继续探索
	dis = self.controller.get_location().add(self.explore_dir).distance_to(self.home_loc)  # 探索后距离母星的距离
	for _ in range(8):  # 尝试8次移动，均失败说明四周全被堵死
		if dis > 20:  # 如果距离母星大于20，则旋转方向
			self.explore_dir = self.explore_dir.rotate_left()  # 左手法则
		else:
			self.explore()  # 如果距离母星小于20，则继续探索
		dis = self.controller.get_location().add(self.explore_dir).distance_to(self.home_loc)