feat: 识别青春杯友情并且作为训练选择权重的一部分

Author: ZJChe

module/umamusume/hook.py

@@ -24,7 +24,7 @@ def after_hook(ctx: UmamusumeContext):
     if ctx.cultivate_detail and ctx.cultivate_detail.turn_info is not None:
         if ctx.cultivate_detail.turn_info.parse_train_info_finish and ctx.cultivate_detail.turn_info.parse_main_menu_finish:
             if not ctx.cultivate_detail.turn_info.turn_info_logged:
-                ctx.cultivate_detail.turn_info.log_turn_info()
+                ctx.cultivate_detail.turn_info.log_turn_info(ctx.task.detail.scenario)
                 ctx.cultivate_detail.turn_info.turn_info_logged = True
             if ctx.cultivate_detail.turn_info.turn_operation is None:
                 ctx.cultivate_detail.turn_info.turn_operation = get_operation(ctx)

module/umamusume/scenario/aoharuhai_scenario.py

@@ -98,18 +98,21 @@ def parse_training_result(self, img: any) -> list[int]:
         skill_point_incr = (0 if skill_point_incr_text == "" else int(skill_point_incr_text)) + (0 if skill_point_incr_extra_text == "" else int(skill_point_incr_extra_text))
 
         return [speed_icr, stamina_incr, power_incr, will_incr, intelligence_incr, skill_point_incr]
-    
-    def parse_training_support_cord(self, img: any) -> list[SupportCardInfo]:
-        # TODO: 目前没有识别青春杯参数和青春杯友情条, 也没有将其作为训练权重的一部分,
-        base_x = 590
+
+    def parse_training_support_card(self, img: any) -> list[SupportCardInfo]:
+        base_x = 550
         base_y = 177
         inc = 115
         support_card_list_info_result: list[SupportCardInfo] = []
         for i in range(5):
-            support_card_icon = img[base_y:base_y + inc, base_x: base_x + 105]
+            support_card_icon = img[base_y:base_y + inc, base_x: base_x + 145]
+            
+            # 有青春杯训练, 且青春杯友情未满
+            can_incr_aoharu_train = detect_aoharu_train_arrow(support_card_icon) and aoharu_train_not_full(support_card_icon)
+            
             # 判断好感度
             support_card_icon = cv2.cvtColor(support_card_icon, cv2.COLOR_BGR2RGB)
-            favor_process_check_list = [support_card_icon[106, 16], support_card_icon[106, 20]]
+            favor_process_check_list = [support_card_icon[106, 56], support_card_icon[106, 60]]
             support_card_favor_process = SupportCardFavorLevel.SUPPORT_CARD_FAVOR_LEVEL_UNKNOWN
             for support_card_favor_process_pos in favor_process_check_list:
                 if compare_color_equal(support_card_favor_process_pos, [255, 235, 120]):
@@ -124,13 +127,6 @@ def parse_training_support_cord(self, img: any) -> list[SupportCardInfo]:
                 if support_card_favor_process != SupportCardFavorLevel.SUPPORT_CARD_FAVOR_LEVEL_UNKNOWN:
                     break
 
-            # 判断是否有事件
-            support_card_event_pos = support_card_icon[5, 83]
-            support_card_event_available = False
-            if (support_card_event_pos[0] >= 250
-                    and 55 <= support_card_event_pos[1] <= 90
-                    and 115 <= support_card_event_pos[2] <= 150):
-                support_card_event_available = True
             # 判断支援卡类型
             support_card_type = SupportCardType.SUPPORT_CARD_TYPE_UNKNOWN
             support_card_icon = cv2.cvtColor(support_card_icon, cv2.COLOR_RGB2GRAY)
@@ -146,11 +142,107 @@ def parse_training_support_cord(self, img: any) -> list[SupportCardInfo]:
                 support_card_type = SupportCardType.SUPPORT_CARD_TYPE_INTELLIGENCE
             elif image_match(support_card_icon, REF_SUPPORT_CARD_TYPE_FRIEND).find_match:
                 support_card_type = SupportCardType.SUPPORT_CARD_TYPE_FRIEND
-            if support_card_favor_process is not SupportCardFavorLevel.SUPPORT_CARD_FAVOR_LEVEL_UNKNOWN:
+            if (can_incr_aoharu_train) or \
+               (support_card_favor_process is not SupportCardFavorLevel.SUPPORT_CARD_FAVOR_LEVEL_UNKNOWN):
                 info = SupportCardInfo(card_type=support_card_type,
-                                    favor=support_card_favor_process,
-                                    has_event=support_card_event_available)
+                                       favor=support_card_favor_process,
+                                       can_incr_aoharu_train=can_incr_aoharu_train)
                 support_card_list_info_result.append(info)
             base_y += inc
 
-        return support_card_list_info_result
+        return support_card_list_info_result
+    
+# 检测支援卡右上角是否有箭头图标, 同时排除感叹号防止false positive
+# 输入的图片必须是彩色的
+def detect_aoharu_train_arrow(support_card_icon):
+    support_card_icon = cv2.cvtColor(support_card_icon, cv2.COLOR_BGR2RGB)
+    # 定义右上角检测区域
+    arrow_region_x_start = 110
+    arrow_region_x_end = 145  
+    arrow_region_y_start = 0
+    arrow_region_y_end = 40
+    
+    arrow_region = support_card_icon[arrow_region_y_start:arrow_region_y_end, 
+                                     arrow_region_x_start:arrow_region_x_end]
+    
+    # 定义箭头可能的颜色范围 (检查橙色)
+    orange_lower = [240, 100, 50]
+    orange_upper = [255, 180, 100]
+    
+    # 定义红色像素范围（用于检测感叹号）
+    red_lower = [180, 30,50]
+    red_upper = [255, 100, 150]
+    
+    # 计算橙色像素和红色像素的数量
+    orange_pixels = 0
+    red_pixels = 0
+    total_pixels = arrow_region.shape[0] * arrow_region.shape[1]
+    
+    for y in range(arrow_region.shape[0]):
+        for x in range(arrow_region.shape[1]):
+            pixel = arrow_region[y, x]
+            
+            # 检测橙色像素
+            if (orange_lower[0] <= pixel[0] <= orange_upper[0] and
+                orange_lower[1] <= pixel[1] <= orange_upper[1] and
+                orange_lower[2] <= pixel[2] <= orange_upper[2]):
+                orange_pixels += 1
+            # 检测红色像素
+            elif (red_lower[0] <= pixel[0] <= red_upper[0] and
+                  red_lower[1] <= pixel[1] <= red_upper[1] and
+                  red_lower[2] <= pixel[2] <= red_upper[2]):
+                red_pixels += 1
+    
+    orange_ratio = orange_pixels / total_pixels if total_pixels > 0 else 0
+    red_ratio = red_pixels / total_pixels if total_pixels > 0 else 0
+    
+    has_arrow = False
+    
+    # 首先排除感叹号：如果红色像素比例过高，判断为感叹号
+    if red_ratio > 0.2:
+        has_arrow = False
+    # 如果橙色像素比例超过阈值
+    elif (orange_ratio > 0.05):
+        has_arrow = True
+ 
+    return has_arrow
+
+
+# 检测左下角青春杯训练值是否未满
+# 如果已满或者不存在UI(比如已经触发了魂爆, 则返回false)
+# 否则返回true
+def aoharu_train_not_full(support_card_icon) -> bool:
+    support_card_icon = cv2.cvtColor(support_card_icon, cv2.COLOR_BGR2RGB)
+    avatar_region_x_start = 5
+    avatar_region_x_end = 45
+    avatar_region_y_start = 70  
+    avatar_region_y_end = 110
+    
+    avatar_region = support_card_icon[avatar_region_y_start:avatar_region_y_end,
+                                      avatar_region_x_start:avatar_region_x_end]
+    
+    total_pixels = avatar_region.shape[0] * avatar_region.shape[1]
+    if total_pixels == 0:
+        return False
+    
+    # 检测灰色
+    grey_lower = [100, 100, 100]
+    grey_upper = [150, 150, 150]
+    grey_pixels = 0
+
+    for y in range(avatar_region.shape[0]):
+        for x in range(avatar_region.shape[1]):
+            pixel = avatar_region[y, x]
+            if (grey_lower[0] <= pixel[0] <= grey_upper[0] and
+                grey_lower[1] <= pixel[1] <= grey_upper[1] and
+                grey_lower[2] <= pixel[2] <= grey_upper[2]):
+                grey_pixels += 1
+    
+    grey_ratio = grey_pixels / total_pixels
+    
+    if grey_ratio > 0.05:
+        status = True
+    else:
+        status = False
+    
+    return status

module/umamusume/scenario/base_scenario.py

@@ -32,6 +32,6 @@ def parse_training_result(self, img: any) -> list[int]:
         pass
 
     @abstractmethod
-    def parse_training_support_cord(self, img: any) -> list[SupportCardInfo]:
+    def parse_training_support_card(self, img: any) -> list[SupportCardInfo]:
         """从屏幕上获取每一张支援卡的信息"""
         pass

module/umamusume/scenario/ura_scenario.py

@@ -63,7 +63,7 @@ def parse_training_result(self, img: any) -> list[int]:
 
         return [speed_icr, stamina_incr, power_incr, will_incr, intelligence_incr, skill_point_incr]
 
-    def parse_training_support_cord(self, img: any) -> list[SupportCardInfo]:
+    def parse_training_support_card(self, img: any) -> list[SupportCardInfo]:
         base_x = 590
         base_y = 190
         inc = 120

module/umamusume/script/cultivate_task/parse.py

@@ -206,7 +206,7 @@ def parse_train_main_menu_operations_availability(ctx: UmamusumeContext, img):
 
 
 def parse_training_support_card(ctx: UmamusumeContext, img, train_type: TrainingType):
-    support_card_info_list = ctx.cultivate_detail.scenario.parse_training_support_cord(img)
+    support_card_info_list = ctx.cultivate_detail.scenario.parse_training_support_card(img)
     ctx.cultivate_detail.turn_info.training_info_list[train_type.value - 1].support_card_info_list = support_card_info_list
 
 def parse_train_type(ctx: UmamusumeContext, img) -> TrainingType:

module/umamusume/script/cultivate_task/support_card.py

@@ -10,6 +10,16 @@ def get_support_card_score(ctx: UmamusumeContext, info: SupportCardInfo):
         score = SCORE_DICT[info.card_type][info.name](ctx, info)
     else:
         score = SCORE_DICT[info.card_type][DEFAULT](ctx, info)
+
+    # 青春杯友情值提高
+    if info.can_incr_aoharu_train:
+        date = ctx.cultivate_detail.turn_info.date
+        if date <= 24:
+            score += 1
+        elif date <= 48:
+            score += 0.5
+        else:
+            score += 0.1
     return score
 
 

module/umamusume/types.py

@@ -15,16 +15,20 @@ class SupportCardInfo:
     card_type: SupportCardType
     favor: SupportCardFavorLevel
     has_event: bool
+    # 青春杯部分
+    can_incr_aoharu_train: bool
 
     def __init__(self,
                  name: str = "support_card",
                  card_type: SupportCardType = SupportCardType.SUPPORT_CARD_TYPE_UNKNOWN,
                  favor: SupportCardFavorLevel = SupportCardFavorLevel.SUPPORT_CARD_FAVOR_LEVEL_UNKNOWN,
-                 has_event: bool = False):
+                 has_event: bool = False,
+                 can_incr_aoharu_train: bool = False):
         self.name = name
         self.card_type = card_type
         self.favor = favor
         self.has_event = has_event
+        self.can_incr_aoharu_train = can_incr_aoharu_train
 
 
 class TrainingInfo:
@@ -45,14 +49,18 @@ def __init__(self):
         self.skill_point_incr = 0
         self.support_card_info_list = []
 
-    def log_training_info(self):
+    def log_training_info(self, scenario_type: ScenarioType):
         log.info("训练结果：速度：%s, 耐力：%s, 力量：%s, 毅力：%s, 智力：%s, 技能点：%s", self.speed_incr,
                  self.stamina_incr, self.power_incr, self.will_incr,
                  self.intelligence_incr, self.skill_point_incr)
         text = "此训练附带支援卡列表：["
-        for c in self.support_card_info_list:
-            if c.favor != SupportCardFavorLevel.SUPPORT_CARD_FAVOR_LEVEL_UNKNOWN:
-                text += "[支援卡名称：" + str(c.name) + "支援卡类型：" + str(c.card_type.name) + ", 支援卡羁绊阶段：" + str(c.favor.name) + "] "
+        if scenario_type == ScenarioType.SCENARIO_TYPE_AOHARUHAI:       
+            for c in self.support_card_info_list:
+                text += "[支援卡名称：" + str(c.name) + "支援卡类型：" + str(c.card_type.name) + ", 支援卡羁绊阶段：" + str(c.favor.name) + ", 可提升青春杯友情: " + str(c.can_incr_aoharu_train) + "] "
+        else:
+            for c in self.support_card_info_list:
+                if c.favor != SupportCardFavorLevel.SUPPORT_CARD_FAVOR_LEVEL_UNKNOWN:
+                    text += "[支援卡名称：" + str(c.name) + "支援卡类型：" + str(c.card_type.name) + ", 支援卡羁绊阶段：" + str(c.favor.name) + "] "
         text += "]"
         log.info(text)
 
@@ -123,22 +131,22 @@ def __init__(self):
         self.turn_info_logged = False
         self.turn_learn_skill_done = False
 
-    def log_turn_info(self):
+    def log_turn_info(self, scenario_type : ScenarioType):
         log.info("当前回合时间 >" + str(self.date))
         log.info("干劲状态 " + str(self.motivation_level.name))
         log.info("体力剩余" + str(self.remain_stamina))
         log.info("当前属性值 速度：%s, 耐力：%s, 力量：%s, 毅力：%s, 智力：%s, 技能点：%s", self.uma_attribute.speed,
                  self.uma_attribute.stamina, self.uma_attribute.power, self.uma_attribute.will, self.uma_attribute.intelligence, self.uma_attribute.skill_point)
         log.info("速度训练结果：")
-        self.training_info_list[0].log_training_info()
+        self.training_info_list[0].log_training_info(scenario_type)
         log.info("耐力训练结果：")
-        self.training_info_list[1].log_training_info()
+        self.training_info_list[1].log_training_info(scenario_type)
         log.info("力量训练结果：")
-        self.training_info_list[2].log_training_info()
+        self.training_info_list[2].log_training_info(scenario_type)
         log.info("毅力训练结果：")
-        self.training_info_list[3].log_training_info()
+        self.training_info_list[3].log_training_info(scenario_type)
         log.info("智力训练结果：")
-        self.training_info_list[4].log_training_info()
+        self.training_info_list[4].log_training_info(scenario_type)
 
 
 class CultivateContextDetail: