152 lines
5.5 KiB
Python
152 lines
5.5 KiB
Python
from copy import deepcopy
|
|
|
|
from mower.data import agent_list
|
|
from mower.solvers.fight.battle_choose.battle_filter import BattleFilterSolver
|
|
from mower.solvers.fight.battle_choose.battle_tag_choose import BattleTagChoose
|
|
from mower.solvers.fight.battle_choose.choose_skill import ChooseSkillSolver
|
|
from mower.utils import config
|
|
from mower.utils.character_recognize import (
|
|
operator_team_select,
|
|
)
|
|
from mower.utils.graph.utils import SceneGraphSolver
|
|
from mower.utils.image import cropimg
|
|
from mower.utils.log import logger
|
|
from mower.utils.scene import Scene
|
|
|
|
|
|
class BattleAgentChooseSolver(SceneGraphSolver):
|
|
solver_name = "战斗选择指定干员"
|
|
|
|
def run(self, agents: list) -> list:
|
|
if agents == []:
|
|
raise ValueError("干员列表为空")
|
|
self.agents = agents
|
|
self.agents_copy = deepcopy(agents)
|
|
self.cleard = False
|
|
self.filter = False
|
|
self.agent = None
|
|
self.choosed = []
|
|
self.final_check = False
|
|
self.check = []
|
|
self.lack = []
|
|
self.tmp_data = None
|
|
self.tag = "ALL"
|
|
|
|
self.before_swipe = None
|
|
self.after_swipe = None
|
|
|
|
self.interval = config.screenshot_avg / 500 # 截图延迟
|
|
super().run()
|
|
|
|
return [self.lack, len(self.choosed)]
|
|
|
|
def choose_agents(self):
|
|
if pos := self.find("choose_agent/open_profession"):
|
|
self.tap(pos, interval=0.1)
|
|
|
|
if self.agent is None:
|
|
if len(self.agents_copy) > 0:
|
|
self.agent = self.agents_copy.pop(0)
|
|
|
|
if "opers" in self.agent:
|
|
tag = agent_list[self.agent["opers"][-1]["name"]]["profession"]
|
|
else:
|
|
if self.agent["name"] == "阿米娅":
|
|
tag = "ALL"
|
|
else:
|
|
tag = agent_list[self.agent["name"]]["profession"]
|
|
|
|
BattleTagChoose().run(tag)
|
|
else:
|
|
return True
|
|
agents = dict(operator_team_select(config.recog.img))
|
|
if self.after_swipe:
|
|
# 滑动后还在移动，先等停下来
|
|
if (
|
|
self.after_swipe[0] == list(agents.keys())[-1]
|
|
and self.after_swipe[1][0][0] != agents[list(agents.keys())[-1]][0][0]
|
|
):
|
|
self.after_swipe[1] = agents[list(agents.keys())[-1]]
|
|
return
|
|
|
|
self.after_swipe = None
|
|
|
|
if "opers" in self.agent:
|
|
for i in self.agent["opers"]:
|
|
if i["name"] in agents.keys():
|
|
self.agent = i
|
|
logger.debug(f"干员组选择：{self.agent}")
|
|
return
|
|
else:
|
|
agent_name = self.agent["name"]
|
|
if agent_name in agents.keys():
|
|
if agents[agent_name][0][0] > 1650:
|
|
self.swipe_noinertia((1000, 540), (-500, 0))
|
|
return
|
|
if self.is_choosed(agents[agent_name]):
|
|
logger.debug(f"干员选好：{self.agent}")
|
|
ChooseSkillSolver().run(self.agent["skill"])
|
|
if agent_name in self.choosed:
|
|
logger.debug(f"{self.agent} 已经选过了")
|
|
else:
|
|
self.choosed.append(agent_name)
|
|
self.agent = None
|
|
return
|
|
logger.info(f"干员选择：{self.agent}")
|
|
self.ctap(agents[agent_name], 1.5, self.interval)
|
|
return
|
|
if self.before_swipe is not None:
|
|
if (
|
|
list(agents.keys())[-1] == self.before_swipe[0]
|
|
or list(agents.keys())[-1] is None
|
|
):
|
|
raise ValueError(f"选择干员: {self.agent}失败 滑动次数过多")
|
|
|
|
self.before_swipe = [list(agents.keys())[-1], agents[list(agents.keys())[-1]]]
|
|
self.swipe_noinertia((1000, 540), (-2000, 0))
|
|
|
|
config.recog.update()
|
|
agents = dict(operator_team_select(config.recog.img))
|
|
self.after_swipe = [list(agents.keys())[-1], agents[list(agents.keys())[-1]]]
|
|
|
|
def is_choosed(self, scope) -> bool:
|
|
if (
|
|
cropimg(
|
|
config.recog.img,
|
|
[
|
|
[scope[0][0] + 60, scope[0][1] - 113],
|
|
[scope[0][0] + 61, scope[0][1] - 112],
|
|
],
|
|
)[0][0][0]
|
|
< 50
|
|
) and cropimg(
|
|
config.recog.img,
|
|
[
|
|
[scope[0][0] + 60, scope[1][1] + 216],
|
|
[scope[0][0] + 61, scope[1][1] + 217],
|
|
],
|
|
)[0][0][0] < 50:
|
|
return True
|
|
return False
|
|
|
|
def transition(self) -> bool:
|
|
if (scene := self.scene()) == Scene.OPERATOR_AGENT_SELECT:
|
|
if self.filter is False:
|
|
BattleTagChoose().run("ALL")
|
|
BattleFilterSolver().run("等级", ascending=False)
|
|
self.filter = True
|
|
if self.cleard is False:
|
|
if self.find("choose_agent/battle_empty"):
|
|
self.cleard = True
|
|
else:
|
|
self.tap("choose_agent/clear_battle", interval=0.1)
|
|
return
|
|
return self.choose_agents()
|
|
elif scene == Scene.OPERATOR_SELECT:
|
|
self.tap("choose_agent/fast_select")
|
|
elif scene == Scene.SSS_SQUAD:
|
|
self.tap("sss/choose/fast_select")
|
|
elif scene in self.waiting_scene:
|
|
self.waiting_solver()
|
|
else:
|
|
return True
|