feat: 选择音擎时更易读，显示稀有度和专武对应代理人 (#87)

选择音擎时更易读，显示稀有度和专武对应代理人

Author: 0xVanfer

zsim/lib_webui/constants.py

@@ -104,25 +104,52 @@ def _init_char_mapping() -> dict[str, str]:
     "丽娜",
     "零号·安比",
 ]  # 这个值其实没啥意义，但是必须是三个角色，否则可能会报错
-__lf = pl.scan_csv("./zsim/data/character.csv")
-char_options = __lf.select("name").unique().collect().to_series().to_list()
+__lf_character = pl.scan_csv("./zsim/data/character.csv")
+char_options = __lf_character.select("name").unique().collect().to_series().to_list()
 # 角色名称->职业特性
 char_profession_map = {
-    row["name"]: row["角色特性"] for row in __lf.collect().iter_rows(named=True)
+    row["name"]: row["角色特性"] for row in __lf_character.collect().iter_rows(named=True)
 }
 
+# 职业特性->角色名称列表
+profession_chars_map = {}
+for char_name, profession in char_profession_map.items():
+    if profession not in profession_chars_map:
+        profession_chars_map[profession] = []
+    profession_chars_map[profession].append(char_name)
+
+profession_chars_map["不限特性"] = char_options
+
 # 武器选项
-__lf = pl.scan_csv("./zsim/data/weapon.csv")
-weapon_options = __lf.select("名称").unique().collect().to_series().to_list()
+__lf_weapon = pl.scan_csv("./zsim/data/weapon.csv")
+weapon_options = __lf_weapon.select("名称").unique().collect().to_series().to_list()
 # 音擎名称->职业
 weapon_profession_map = {
-    row["名称"]: row["职业"] for row in __lf.collect().iter_rows(named=True)
+    row["名称"]: row["职业"] for row in __lf_weapon.collect().iter_rows(named=True)
 }
+# 音擎名称->稀有度
+weapon_rarity_map = {
+    row["名称"]: row["稀有度"] for row in __lf_weapon.collect().iter_rows(named=True)
+}
+# 音擎名称->角色名称 (仅限部分 S 级和 A 级)
+weapon_char_map: dict[str, str] = {}
+for row in __lf_weapon.collect().iter_rows(named=True):
+    cid = row["ID"] % 1000 * 10 + 1
+    names = (
+        __lf_character
+        .filter(pl.col("CID") == cid)
+        .select("name")
+        .collect()
+        .to_series()
+        .to_list()
+    )
+    # 如果没有对应角色，默认用空字符串
+    weapon_char_map[row["名称"]] = names[0] if names else ""
 
 # 驱动盘套装选项
-__lf = pl.scan_csv("./zsim/data/equip_set_2pc.csv")
+__lf_equip = pl.scan_csv("./zsim/data/equip_set_2pc.csv")
 equip_set_ids = (
-    __lf.select("set_ID")
+    __lf_equip.select("set_ID")
     .filter(pl.col("set_ID").is_not_null())
     .unique()
     .collect()
@@ -222,5 +249,7 @@ class IDDuplicateError(Exception):
 
     pass
 
-
-del __lf  # 确保在文件末尾删除临时变量
+# 确保在文件末尾删除临时变量
+del __lf_character 
+del __lf_weapon
+del __lf_equip

zsim/page_character_config.py

@@ -15,39 +15,63 @@ def page_character_config():
         equip_set4_options,
         weapon_options,
         weapon_profession_map,
+        weapon_rarity_map,
+        weapon_char_map,
         char_profession_map,
+        profession_chars_map,
     )
 
-    col1, col2, col3 = st.columns(3)
+    col0, col1, col2, col3, col4, col5, col6, col7 = st.columns([1, 1, 1, 1, 1, 1, 1, 1])
+    with col0:
+        profession_0 = st.selectbox(
+            "角色1特性",
+            list(profession_chars_map.keys()),
+            index=list(profession_chars_map.keys()).index("不限特性"),
+            key="profession_select_0",
+        )
     with col1:
         name_box_0 = [
             st.selectbox(
                 "角色1",
-                char_options,
-                index=char_options.index(default_chars[0])
-                if len(default_chars) > 0
+                profession_chars_map[profession_0],
+                index=profession_chars_map[profession_0].index(default_chars[0])
+                if len(default_chars) > 0 and default_chars[0] in profession_chars_map[profession_0]
                 else 0,
                 key="char_select_0",
             )
         ]
-    with col2:
+    with col3:
+        profession_1 = st.selectbox(
+            "角色2特性",
+            list(profession_chars_map.keys()),
+            index=list(profession_chars_map.keys()).index("不限特性"),
+            key="profession_select_1",
+        )
+    with col4:
         name_box_1 = [
             st.selectbox(
                 "角色2",
-                char_options,
-                index=char_options.index(default_chars[1])
-                if len(default_chars) > 1
+                profession_chars_map[profession_1],
+                index=profession_chars_map[profession_1].index(default_chars[1])
+                if len(default_chars) > 1 and default_chars[1] in profession_chars_map[profession_1]
                 else 0,
                 key="char_select_1",
             )
         ]
-    with col3:
+    with col6:
+        profession_2 = st.selectbox(
+            "角色3特性",
+            list(profession_chars_map.keys()),
+            index=list(profession_chars_map.keys()).index("不限特性"),
+            key="profession_select_2",
+        )
+    with col7:
         name_box_2 = [
             st.selectbox(
                 "角色3",
-                char_options,
-                index=char_options.index(default_chars[2])
-                if len(default_chars) > 2
+                profession_chars_map[profession_2],
+                index=profession_chars_map[profession_2].index(default_chars[2])
+                if len(default_chars) > 2 and default_chars[2] in profession_chars_map[profession_2]
                 else 0,
                 key="char_select_2",
             )
@@ -62,9 +86,11 @@ def page_character_config():
         with st.expander(f"{name}的配置"):
             col_weapon, col_level, col_cinema = st.columns(3)
             with col_weapon:
-                show_adapted_weapon = st.session_state.get(
-                    f"{name}_show_adapted_weapon", True
-                )
+                show_adapted_weapon = st.session_state.get(f"{name}_show_adapted_weapon", True)
+                show_rarity_s = st.session_state.get(f"{name}_show_rarity_s", True)
+                show_rarity_a = st.session_state.get(f"{name}_show_rarity_a", True)
+                show_rarity_b = st.session_state.get(f"{name}_show_rarity_b", False)
+                
                 char_profession = char_profession_map.get(name)
                 if show_adapted_weapon and char_profession:
                     filtered_weapon_options = [
@@ -74,6 +100,20 @@ def page_character_config():
                     ]
                 else:
                     filtered_weapon_options = list(weapon_options)
+                
+                # 根据稀有度筛选
+                filtered_weapon_options = [
+                    w for w in filtered_weapon_options
+                    if (show_rarity_s and weapon_rarity_map.get(w) == "S") or
+                       (show_rarity_a and weapon_rarity_map.get(w) == "A") or
+                       (show_rarity_b and weapon_rarity_map.get(w) == "B")
+                ]
+                rarity_order = {"S": 0, "A": 1, "B": 2}
+                filtered_weapon_options = sorted(
+                    filtered_weapon_options,
+                    key=lambda w: (rarity_order.get(weapon_rarity_map.get(w), 3), w)
+                )
+                
                 if name in saved_char_config:
                     current_weapon = saved_char_config[name].get("weapon")
                 else:
@@ -93,12 +133,37 @@ def page_character_config():
                         filtered_weapon_options,
                         index=filtered_weapon_options.index(current_weapon),
                         key=f"{name}_weapon",
+                        format_func=lambda x: (
+                            f"({weapon_rarity_map.get(x, '未知')}"
+                            f"{' ' + weapon_char_map.get(x) if weapon_char_map.get(x) else ''}) {x}"
+                        ),
+                    )
+                
+                col_rarity = st.columns(4)
+                with col_rarity[0]:
+                    show_adapted_weapon = st.checkbox(
+                        "只显示适配音擎",
+                        value=show_adapted_weapon,
+                        key=f"{name}_show_adapted_weapon",
+                    )
+                with col_rarity[1]:
+                    show_rarity_s = st.checkbox(
+                        "S",
+                        value=show_rarity_s,
+                        key=f"{name}_show_rarity_s",
+                    )
+                with col_rarity[2]:
+                    show_rarity_a = st.checkbox(
+                        "A",
+                        value=show_rarity_a,
+                        key=f"{name}_show_rarity_a",
+                    )
+                with col_rarity[3]:
+                    show_rarity_b = st.checkbox(
+                        "B",
+                        value=show_rarity_b,
+                        key=f"{name}_show_rarity_b",
                     )
-                show_adapted_weapon = st.checkbox(
-                    "只显示适配音擎",
-                    value=show_adapted_weapon,
-                    key=f"{name}_show_adapted_weapon",
-                )
             with col_level:
                 st.number_input(
                     "音擎精炼等级",