79 lines
2.4 KiB
C#
79 lines
2.4 KiB
C#
using RimWorld;
|
|
using Verse;
|
|
|
|
namespace WulaFallenEmpire
|
|
{
|
|
public class CompProperties_FactionSetter : CompProperties
|
|
{
|
|
// 派系配置
|
|
public FactionDef factionDef = null; // 指定的派系定义
|
|
public bool usePlayerFactionIfNull = true; // 如果没有指定派系，是否使用玩家派系
|
|
public bool overrideExistingFaction = true; // 是否覆盖已有的派系归属
|
|
|
|
public CompProperties_FactionSetter()
|
|
{
|
|
compClass = typeof(CompFactionSetter);
|
|
}
|
|
}
|
|
|
|
public class CompFactionSetter : ThingComp
|
|
{
|
|
private bool factionSet = false;
|
|
|
|
public CompProperties_FactionSetter Props => (CompProperties_FactionSetter)props;
|
|
|
|
public override void PostSpawnSetup(bool respawningAfterLoad)
|
|
{
|
|
base.PostSpawnSetup(respawningAfterLoad);
|
|
|
|
if (!respawningAfterLoad && !factionSet)
|
|
{
|
|
SetFaction();
|
|
factionSet = true;
|
|
}
|
|
}
|
|
|
|
public override void PostExposeData()
|
|
{
|
|
base.PostExposeData();
|
|
Scribe_Values.Look(ref factionSet, "factionSet", false);
|
|
}
|
|
|
|
private void SetFaction()
|
|
{
|
|
// 如果不需要覆盖已有派系且建筑已有派系，则跳过
|
|
if (!Props.overrideExistingFaction && parent.Faction != null)
|
|
return;
|
|
|
|
Faction faction = GetTargetFaction();
|
|
if (faction != null && faction != parent.Faction)
|
|
{
|
|
parent.SetFaction(faction);
|
|
Log.Message($"Set faction for {parent.Label} to {faction.Name}");
|
|
}
|
|
}
|
|
|
|
private Faction GetTargetFaction()
|
|
{
|
|
// 1. 如果指定了派系定义，使用该派系
|
|
if (Props.factionDef != null)
|
|
{
|
|
Faction faction = Find.FactionManager.FirstFactionOfDef(Props.factionDef);
|
|
if (faction != null)
|
|
return faction;
|
|
}
|
|
|
|
// 2. 默认使用玩家派系
|
|
if (Props.usePlayerFactionIfNull)
|
|
return Faction.OfPlayer;
|
|
|
|
return null;
|
|
}
|
|
|
|
public override string CompInspectStringExtra()
|
|
{
|
|
return base.CompInspectStringExtra();
|
|
}
|
|
}
|
|
}
|