添加自动生成作业工具

tools/CopilotDesinger/CopilotDesinger.py

@@ -0,0 +1,244 @@
+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):
+        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):
+        super().__init__(name, battle)
+        self._requirements = requirements
+        self._skill_usage = skill_usage
+        self._skill = skill
+
+    def skill_usage_change(self, change_to: int):
+        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,
+            "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
+
+    def create_operator(self, name: str, skill: int, skill_usage: int = SkillUsageType.NotAutoUse,
+                        requirements: Requirements = None) -> Operator:
+        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:
+        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):
+        if not self._speedup:
+            self._speed_change()
+            self._speedup = True
+
+    def speed_down(self):
+        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 = "battle.json"):
+        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

tools/CopilotDesinger/main.py

@@ -0,0 +1,32 @@
+from CopilotDesinger import Battle, Document, Operator, SkillUsageType, Direction
+
+危机合约十八 = Battle("obt/rune/level_rune_11-01",
+                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)
+
+危机合约十八.speed_up()
+棘刺.deploy((8, 6), Direction.UP)
+对空高速单体狙.deploy((2, 4), Direction.UP)
+伊芙利特.deploy((10, 4), Direction.LEFT)
+蜜莓.deploy((10, 5), Direction.LEFT)
+对空高速单体狙.wait_kills(16).retreat()
+红.wait_kills(18).deploy((3, 5), Direction.LEFT)
+史尔特尔.wait_kills(21).rear_delay(1500).deploy((2, 3), Direction.DOWN)
+史尔特尔.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()))
+危机合约十八.save_as_json("YanFengRongDong.json")