1

2025-12-08 11:54:30 +08:00
parent ffe26b777c
commit 74d9a10e41
9 changed files with 556 additions and 111 deletions
--- a/Source/WulaFallenEmpire/Damage/DamageWorker_ExplosionWithTerrain.cs
+++ b/Source/WulaFallenEmpire/Damage/DamageWorker_ExplosionWithTerrain.cs
@@ -0,0 +1,89 @@
+using System.Collections.Generic;
+using System.Linq;
+using RimWorld;
+using UnityEngine;
+using Verse;
+using Verse.Sound;
+
+namespace WulaFallenEmpire
+{
+    public class DamageWorker_ExplosionWithTerrain : DamageWorker_AddInjury
+    {
+        // 获取地形覆盖扩展
+        private DamageDefExtension_TerrainCover GetTerrainCoverExtension(DamageDef damageDef)
+        {
+            if (damageDef?.modExtensions == null)
+                return null;
+
+            return damageDef.modExtensions
+                .OfType<DamageDefExtension_TerrainCover>()
+                .FirstOrDefault();
+        }
+
+        public override void ExplosionAffectCell(Explosion explosion, IntVec3 c, List<Thing> damagedThings, List<Thing> ignoredThings, bool canThrowMotes)
+        {
+            base.ExplosionAffectCell(explosion, c, damagedThings, ignoredThings, canThrowMotes);
+            var terrainCover = GetTerrainCoverExtension(explosion.damType);
+            // 处理地形覆盖
+            ProcessTerrainCover(explosion, c, terrainCover);
+        }
+
+        // 处理地形覆盖
+        private void ProcessTerrainCover(Explosion explosion, IntVec3 cellsToAffect, DamageDefExtension_TerrainCover terrainCover)
+        {
+            if (explosion.Map == null)
+                return;
+
+            Map map = explosion.Map;
+
+            ApplyTerrainCoverToCell(cellsToAffect, map, terrainCover, explosion);
+        }
+
+        // 对单个单元格应用地形覆盖
+        public void ApplyTerrainCoverToCell(IntVec3 cell, Map map,
+            DamageDefExtension_TerrainCover terrainCover, Explosion explosion = null)
+        {
+            // 检查单元格是否可影响
+            if (!terrainCover.CanAffectCell(cell, map, out string reason))
+            {
+                Log.Message($"Cannot affect cell {cell}: {reason}");
+                return;
+            }
+
+            TerrainDef currentTerrain = cell.GetTerrain(map);
+
+            // 检查框架
+            Frame frame = cell.GetFirstThing<Frame>(map);
+
+            // 检查植物（特别是大型树木）
+            Plant plant = cell.GetPlant(map);
+            if (plant != null && plant.def.plant.treeCategory == TreeCategory.Super)
+            {
+                Log.Message($"Large tree at {cell}, skipping");
+                return;
+            }
+
+            // 尝试设置地形
+            if (GenConstruct.CanBuildOnTerrain(terrainCover.terrainToSpawn, cell, map, Rot4.North))
+            {
+                // 销毁框架（如果需要）
+                if (frame != null)
+                {
+                    frame.Destroy(DestroyMode.Vanish);
+                }
+
+                // 保存原始地形（用于临时覆盖）
+                TerrainDef originalTerrain = map.terrainGrid.TerrainAt(cell);
+
+                // 设置新地形
+                map.terrainGrid.SetTerrain(cell, terrainCover.terrainToSpawn);
+
+                Log.Message($"Applied terrain {terrainCover.terrainToSpawn.defName} to cell {cell}");
+            }
+            else
+            {
+                Log.Message($"Cannot build terrain {terrainCover.terrainToSpawn.defName} at cell {cell}");
+            }
+        }
+    }
+}