暂存

2025-08-01 21:27:01 +08:00
parent 78ab937b2b
commit f087d995bb
8 changed files with 970 additions and 213 deletions
--- a/Source/WulaFallenEmpire/PsychicRitual_TechOffering.cs
+++ b/Source/WulaFallenEmpire/PsychicRitual_TechOffering.cs
@@ -0,0 +1,188 @@
+using System.Collections.Generic;
+using System.Linq;
+using RimWorld;
+using Verse;
+using Verse.AI.Group;
+
+namespace WulaFallenEmpire
+{
+    // 用于在XML中定义祭品
+    public class OfferingItem
+    {
+        public ThingDef thingDef;
+        public float power;
+    }
+
+    public class QualityThreshold
+    {
+        public float threshold;
+        public QualityCategory quality;
+    }
+
+    public class PsychicRitual_TechOffering : PsychicRitualDef_InvocationCircle
+    {
+        // 从XML加载的额外祭品列表
+        public List<OfferingItem> extraOfferings = new List<OfferingItem>();
+
+        // 从XML加载的奖励池
+        public List<ThingDef> rewardWeaponPool = new List<ThingDef>();
+
+        // 从XML加载的品质阈值
+        public List<QualityThreshold> qualityThresholds = new List<QualityThreshold>();
+
+        // 重写计算最大能量的方法
+        public override void CalculateMaxPower(PsychicRitualRoleAssignments assignments, List<QualityFactor> powerFactorsOut, out float power)
+        {
+            // 首先调用基类方法
+            base.CalculateMaxPower(assignments, powerFactorsOut, out power);
+
+            IntVec3 center = assignments.Target.Cell;
+            Map map = assignments.Target.Map;
+            float offeringRadius = 8f;
+
+            float extraPowerFromOfferings = 0f;
+            if (!extraOfferings.NullOrEmpty())
+            {
+                var offeringThings = new Dictionary<ThingDef, float>();
+                foreach(var offering in extraOfferings)
+                {
+                    offeringThings[offering.thingDef] = offering.power;
+                }
+
+                foreach (Thing thing in GenRadial.RadialDistinctThingsAround(center, map, offeringRadius, useCenter: true))
+                {
+                    if (offeringThings.TryGetValue(thing.def, out float value))
+                    {
+                        extraPowerFromOfferings += value * thing.stackCount;
+                    }
+                }
+            }
+
+            if (extraPowerFromOfferings > 0)
+            {
+                powerFactorsOut?.Add(new QualityFactor
+                {
+                    label = "WULA_ExtraOfferings".Translate(),
+                    positive = true,
+                    quality = extraPowerFromOfferings,
+                    toolTip = "WULA_ExtraOfferings_Tooltip".Translate()
+                });
+                power += extraPowerFromOfferings;
+            }
+            
+            power = UnityEngine.Mathf.Clamp01(power);
+        }
+
+        // 重写创建仪式步骤的方法
+        public override List<PsychicRitualToil> CreateToils(PsychicRitual psychicRitual, PsychicRitualGraph parent)
+        {
+            // 获取基类的仪式步骤
+            List<PsychicRitualToil> toils = base.CreateToils(psychicRitual, parent);
+            
+            // 在最后添加我们自定义的奖励步骤
+            toils.Add(new PsychicRitualToil_TechOfferingOutcome(psychicRitual, this));
+
+            return toils;
+        }
+    }
+
+    // 自定义的仪式步骤，用于处理奖励
+    public class PsychicRitualToil_TechOfferingOutcome : PsychicRitualToil
+    {
+        private PsychicRitual psychicRitual;
+        private PsychicRitualDef def;
+
+        public PsychicRitualToil_TechOfferingOutcome(PsychicRitual psychicRitual, PsychicRitualDef def)
+        {
+            this.psychicRitual = psychicRitual;
+            this.def = def;
+        }
+
+        public override bool Tick(PsychicRitual psychicRitual, PsychicRitualGraph parent)
+        {
+            float power = psychicRitual.power;
+
+            // 消耗祭品
+            IntVec3 center = psychicRitual.assignments.Target.Cell;
+            Map map = psychicRitual.assignments.Target.Map;
+            float offeringRadius = 8f;
+
+            PsychicRitual_TechOffering ritualDef = (PsychicRitual_TechOffering)def;
+
+            if (!ritualDef.extraOfferings.NullOrEmpty())
+            {
+                var offeringThings = new Dictionary<ThingDef, float>();
+                foreach(var offering in ritualDef.extraOfferings)
+                {
+                    offeringThings[offering.thingDef] = offering.power;
+                }
+
+                foreach (Thing thing in GenRadial.RadialDistinctThingsAround(center, map, offeringRadius, useCenter: true))
+                {
+                    if (offeringThings.ContainsKey(thing.def))
+                    {
+                        thing.Destroy(DestroyMode.Vanish);
+                    }
+                }
+            }
+
+            // 从奖励池中随机选择一个武器
+            if (ritualDef.rewardWeaponPool.NullOrEmpty())
+            {
+                Log.Error($"[WulaFallenEmpire] Reward weapon pool is empty for {def.defName}");
+                return true;
+            }
+            ThingDef weaponDef = ritualDef.rewardWeaponPool.RandomElement();
+            if (weaponDef == null)
+            {
+                Log.Error($"[WulaFallenEmpire] Could not find weapon Def: {weaponDef.defName}");
+                return true;
+            }
+
+            // 根据能量值决定物品品质
+            QualityCategory quality = QualityCategory.Awful; // 默认最低品质
+            if (!ritualDef.qualityThresholds.NullOrEmpty())
+            {
+                // 对阈值列表按阈值从高到低排序
+                var sortedThresholds = ritualDef.qualityThresholds.OrderByDescending(t => t.threshold).ToList();
+                foreach (var threshold in sortedThresholds)
+                {
+                    if (power >= threshold.threshold)
+                    {
+                        quality = threshold.quality;
+                        break; // 找到第一个满足的阈值就跳出
+                    }
+                }
+            }
+            else // 如果XML中没有定义，则使用硬编码的默认值
+            {
+                if (power >= 1.0f) { quality = QualityCategory.Legendary; }
+                else if (power >= 0.8f) { quality = QualityCategory.Masterwork; }
+                else if (power >= 0.5f) { quality = QualityCategory.Excellent; }
+                else if (power >= 0.2f) { quality = QualityCategory.Normal; }
+                else { quality = QualityCategory.Poor; }
+            }
+
+            // 创建物品并设置品质
+            Thing reward = ThingMaker.MakeThing(weaponDef);
+            CompQuality compQuality = reward.TryGetComp<CompQuality>();
+            if (compQuality != null)
+            {
+                compQuality.SetQuality(quality, ArtGenerationContext.Colony);
+            }
+
+            // 在仪式中心点生成奖励物品
+            GenPlace.TryPlaceThing(reward, psychicRitual.assignments.Target.Cell, map, ThingPlaceMode.Near);
+
+            // 发送消息通知玩家
+            Find.LetterStack.ReceiveLetter(
+                "WULA_RitualReward_Label".Translate(),
+                "WULA_RitualReward_Description".Translate(reward.Label, quality.GetLabel()),
+                LetterDefOf.PositiveEvent,
+                new LookTargets(psychicRitual.assignments.Target.Cell, map)
+            );
+            
+            return true;
+        }
+    }
+}