diff --git a/tools/CopilotDesinger/CopilotDesinger.py b/tools/CopilotDesinger/CopilotDesinger.py new file mode 100644 index 0000000000..d4301165a2 --- /dev/null +++ b/tools/CopilotDesinger/CopilotDesinger.py @@ -0,0 +1,353 @@ +from __future__ import annotations + +import json + + +# ======================= constants begin ========================= +class Direction: + UP = "上" + DOWN = "下" + RIGHT = "右" + LEFT = "左" + + +class ActionType: + Deploy = "部署" + Skill = "技能" + Retreat = "撤退" + SpeedUp = "二倍速" + BulletTime = "子弹时间" + SkillUsage = "技能用法" + + +class SkillUsageType: + NotAutoUse = 0 # 0 - 不自动使用 默认 + UseWhenOk = 1 # 1 - 好了就用,有多少次用多少次(例如干员 棘刺 3 技能、桃金娘 1 技能等) + UseWhenOkOnce = 2 # 2 - 好了就用,仅使用一次(例如干员 山 2 技能) + AutoUse = 3 # 3 - 自动判断使用时机(画饼.jpg) + + +# ========================= constants end ============================= + +class Document: + def __init__(self, title: str, details: str, title_color: str = "dark", details_color: str = "dark"): + self._details_color = details_color + self._details = details + self._title_color = title_color + self._title = title + + def to_dict(self) -> dict: + return { + "title": self._title, + "title_color": self._title_color, + "details": self._details, + "details_color": self._details_color + } + + +class Requirements: + def __init__(self, elite: int, level: int, skill_level: int, module: int, potentiality: int): + self._potentiality = potentiality + self._module = module + self._skill_level = skill_level + self._level = level + self._elite = elite + + def to_dict(self): + return { + "elite": self._elite, + "level": self._level, + "skill_level": self._skill_level, + "module": self._module, + "potentiality": self._potentiality + } + + +class Action: + def __init__(self, action_type: str, name: str = None, location: tuple = None, direction: str = "无", kills: int = 0, + skill_usage: int = None, pre_delay: int = 0, rear_delay: int = 0, timeout: int = 999999999, + doc: str = None, doc_color: str = None): + self._doc_color = doc_color + self._doc = doc + self._timeout = timeout + self._rear_delay = rear_delay + self._pre_delay = pre_delay + self._skill_usage = skill_usage + self._kills = kills + self._direction = direction + self._name = name + self._location = location + self._action_type = action_type + + def to_dict(self) -> dict: + res = { + "type": self._action_type, + } + if self._kills != 0: + res["kills"] = self._kills + if self._pre_delay != 0: + res["pre_delay"] = self._pre_delay + if self._rear_delay != 0: + res["rear_delay"] = self._rear_delay + if self._action_type == ActionType.Deploy or self._action_type == ActionType.Skill \ + or self._action_type == ActionType.Retreat: + res["name"] = self._name + + if self._action_type == ActionType.Deploy: + res["location"] = self._location + res["direction"] = self._direction + return res + + +class OperatorOrGroup: + def __init__(self, name: str, battle: Battle): + self._battle = battle + self._name = name + self._pre_delay = 0 + self._rear_delay = 0 + self._wait_kills = 0 + + def pre_delay(self, times: int) -> OperatorOrGroup: + self._pre_delay = times + return self + + def rear_delay(self, times: int) -> OperatorOrGroup: + self._rear_delay = times + return self + + def wait_kills(self, kills: int) -> OperatorOrGroup: + self._wait_kills = kills + return self + + def deploy(self, location: tuple, direction: str): + self._battle.add_action(self._add_conditions(Action(ActionType.Deploy, self._name, location, direction))) + + def use_skill(self): + self._battle.add_action(self._add_conditions(Action(ActionType.Skill, self._name))) + + def retreat(self): + self._battle.add_action(self._add_conditions(Action(ActionType.Retreat, self._name))) + + def _add_conditions(self, action: Action) -> Action: + if self._pre_delay != 0: + action._pre_delay = self._pre_delay + if self._rear_delay != 0: + action._rear_delay = self._rear_delay + if self._wait_kills != 0: + action._kills = self._wait_kills + self._clear_conditions() + return action + + def _clear_conditions(self): + self._pre_delay = 0 + self._rear_delay = 0 + self._wait_kills = 0 + + def to_dict(self) -> dict: + # 此处不应该被执行到 + raise Exception("请使用子类方法") + + +class Group(OperatorOrGroup): + def __init__(self, name: str, battle: Battle, operators: tuple): + """ + 请使用 Battle.create_group() + :param name: + :param battle: + :param operators: + """ + super().__init__(name, battle) + self._operators = operators + + def to_dict(self) -> dict: + return { + "name": self._name, + "opers": [operator.to_dict() for operator in self._operators] + } + + +class Operator(OperatorOrGroup): + def __init__(self, name: str, battle: Battle, skill: int, skill_usage: int = SkillUsageType.NotAutoUse, + requirements: Requirements = None): + """ + 仅当要创建群组时才应该被调用 + 正常使用请用 Battle.create_operator() + :param name: + :param battle: + :param skill: + :param skill_usage: + :param requirements: + """ + super().__init__(name, battle) + self._requirements = requirements + self._skill_usage = skill_usage + self._skill = skill + + def skill_usage_change(self, change_to: int): + """ + 改变技能使用的类型 + 由于需要兼容群组类型,有时此方法可能无法通过代码提示调用,手动调用即可 + :param change_to: 需要调整到的 SkillUsageType + :return: + """ + self._battle.add_action(self._add_conditions(Action(ActionType.SkillUsage, skill_usage=change_to))) + + def to_dict(self) -> dict: + res = { + "name": self._name, + "skill": self._skill, + } + if self._skill_usage != SkillUsageType.NotAutoUse: + res["skill_usage"] = self._skill_usage + if self._requirements is not None: + res["requirements"] = self._requirements.to_dict() + return res + + +class Battle: + def __init__(self, stage_name: str, version="v4.0", doc: Document = None): + self._doc = doc + self._stage_name = stage_name + self._version = version + self._operators = [] + self._groups = [] + self._actions = [] + self._wait_times = 0 + self._wait_kills = 0 + self._speedup = False + self._last_save_file_name = "battle.json" + + def create_operator(self, name: str, skill: int, skill_usage: int = SkillUsageType.NotAutoUse, + requirements: Requirements = None) -> Operator: + """ + 创建并添加一个干员 + :param name: 干员名称 + :param skill: 要使用的技能 + :param skill_usage: 技能使用类型 参见 SkillUsageType + :param requirements: 可选 + :return: 新生成的干员 + """ + new_operator = Operator(name, self, skill, skill_usage, requirements) + self._add_operator(new_operator) + return new_operator + + def create_group(self, name: str, *operators: Operator) -> Group: + """ + 创建并添加一个群组 + :param name: 群组名 + :param operators: 所包含的干员,构造的时候 battle 一项置 None 即可 + :return: 新创建的群组 + """ + new_group = Group(name, self, operators) + self._add_group(new_group) + return new_group + + def _add_operator(self, operator: Operator): + self._operators.append(operator) + + def _add_group(self, group: Group): + self._groups.append(group) + + def add_action(self, action: Action): + self._actions.append(action) + + def speed_up(self): + """ + 调整为二倍速 + :return: + """ + if not self._speedup: + self._speed_change() + self._speedup = True + + def speed_down(self): + """ + 调整为一倍速 + :return: + """ + if self._speedup: + self._speed_change() + self._speedup = False + + def _speed_change(self): + self.add_action(Action(ActionType.SpeedUp)) + + def save_as_json(self, file_name: str = None): + """ + 保存为 json 文件,生成在同目录下 + :param file_name: 要保存的文件名 + :return: + """ + if file_name is None: + file_name = self._last_save_file_name + else: + self._last_save_file_name = file_name + with open(file_name, 'w', encoding='utf-8') as f: + json.dump(self.to_dict(), f, ensure_ascii=False) + + def to_dict(self) -> dict: + res = { + "stage_name": self._stage_name, + "minimum_required": self._version, + "opers": [operator.to_dict() for operator in self._operators], + "groups": [group.to_dict() for group in self._groups], + "actions": [action.to_dict() for action in self._actions], + } + if self._doc is not None: + res["doc"] = self._doc.to_dict() + + return res + + def test(self, enable_debug_output: bool = True): + """ + 自动调用 MeoAssistantArknights 并测试生成自动战斗的文件 + 测试的是上次生成的 json 需先调用 save_as_json() + 请自行配置 MeoAssistantArknights 所在地址等 + 同时确保能导入 asst.py + + :param enable_debug_output: 是否开启 debug 输出 + :return: + """ + # MeoAssistantArknights所在的路径 + mma_path = "D:\\Program Files\\MeoAssistantArknights" + # adb的路径 + adb_path = "D:\\Program Files\\MeoAssistantArknights\\platform-tools\\adb.exe" + # adb连接的地址 + adb_address = "127.0.0.1:8873" + try: + from asst import Asst, Message + except ImportError: + import sys + import pathlib + sys.path.append(str(pathlib.Path.cwd().parent.parent/"src"/"Python")) + try: + from asst import Asst, Message + except ImportError: + print("asst导入失败,请自行解决,或者下载" + " https://github.com/MaaAssistantArknights/MaaAssistantArknights/blob/master/src/Python/asst.py " + "到同目录下") + return + + @Asst.CallBackType + def callback(msg, details, arg): + print(Message(msg), json.loads(details.decode('utf-8')), arg) + + if not Asst.load(mma_path): + print("加载MeoAssistantArknights失败,请确定路径是否正确") + + asst = Asst(callback) if enable_debug_output else Asst() + if asst.connect(adb_path, adb_address): + print('连接成功') + else: + print('连接失败') + print('请检查adb地址或者端口是否有误') + exit() + asst.append_task('Copilot', { + 'stage_name': self._stage_name, + 'filename': self._last_save_file_name, + 'formation': False, + }) + asst.start() + while True: + if "stop" == str(input('>')): + exit() diff --git a/tools/CopilotDesinger/main.py b/tools/CopilotDesinger/main.py new file mode 100644 index 0000000000..a2ccee5299 --- /dev/null +++ b/tools/CopilotDesinger/main.py @@ -0,0 +1,57 @@ +from CopilotDesinger import * + +# 使用样例 渊默行动 18 +# 按道理是能用的,我照着作者的作业抄的,不能用一定是玛丽的问题( + +# 阅读过此文会对使用本程序有很大的帮助(毕竟本来就是为了生成这种json文件的( +# https://github.com/MaaAssistantArknights/MaaAssistantArknights/blob/master/docs/%E6%88%98%E6%96%97%E6%B5%81%E7%A8%8B%E5%8D%8F%E8%AE%AE.md + +# (其实我觉得有个 ide 会点英文 应该是个人都会用,不过我还是稍微写点注释啥的 + +# 创建作业对象 要啥填啥就完事了 别问为什么用中文命名变量 我怕改成英文了自己过会都看不懂了( +危机合约十八 = Battle(stage_name="obt/rune/level_rune_11-01", + version="v4.0", + doc=Document("《全自动危机合约 18》", "作业抄自 https://www.bilibili.com/video/BV13S4y187NW")) + +# 创建要使用的干员(老婆们) +# 第一个参数是名字,第二个是几技能 +史尔特尔 = 危机合约十八.create_operator("史尔特尔", 3) +伊芙利特 = 危机合约十八.create_operator("伊芙利特", 2) +# 鸡翅技能设定为好了就开 +棘刺 = 危机合约十八.create_operator("棘刺", 3, SkillUsageType.UseWhenOk) +# 创建一个群组 +对空高速单体狙 = 危机合约十八.create_group("对空高速单体狙", Operator("克洛丝", None, 1), Operator("能天使", None, 3)) +红 = 危机合约十八.create_operator("红", 2) +蜜莓 = 危机合约十八.create_operator("蜜莓", 1) + +# 开始写作业 QWQ +# 调成二倍速 +危机合约十八.speed_up() + +# 部署鸡翅 部署在(8,6)格子上,方向朝上 +# 不知道哪格是哪格的可以先随便写点然后 save_as_json() 和 test() +# 运行后会在同目录下生成 map.png 的 +棘刺.deploy((8, 6), Direction.UP) +对空高速单体狙.deploy((2, 4), Direction.UP) +伊芙利特.deploy((10, 4), Direction.LEFT) +蜜莓.deploy((10, 5), Direction.LEFT) +# 等待 16 杀后撤退 +对空高速单体狙.wait_kills(16).retreat() +红.wait_kills(18).deploy((3, 5), Direction.LEFT) +史尔特尔.wait_kills(21).pre_delay(1500).deploy((2, 3), Direction.DOWN) +# 使用 42 技能 +史尔特尔.use_skill() +史尔特尔.wait_kills(23).retreat() +红.wait_kills(25).pre_delay(10000).deploy((3, 5), Direction.LEFT) +红.retreat() +史尔特尔.wait_kills(29).rear_delay(1500).deploy((9, 4), Direction.LEFT) +红.pre_delay(1000).deploy((4, 4), Direction.DOWN) +史尔特尔.use_skill() +史尔特尔.wait_kills(34).retreat() +红.pre_delay(5000).deploy((3, 5), Direction.LEFT) + +# print(json.dumps(危机合约十八.to_dict())) +# 保存为json文件 +危机合约十八.save_as_json("YanFengRongDong.json") +# 直接进行测试 要用的需要点进去配置一下 不然直接注释掉也行( +危机合约十八.test()