static void Postfix(Pawn_MeleeVerbs __instance, Thing target)
{
if (Settings.AutoSwitchMelee == false)
{
return;
}
Pawn pawn = __instance.Pawn;
if (pawn != null &&
WorldComp.TryGetAssignedWeapons(pawn, out AssignedWeaponContainer c))
{
Verb attackVerb = pawn.TryGetAttackVerb(target, !pawn.IsColonist);
if (attackVerb != null && attackVerb.verbProps.IsMeleeAttack)
{
float blunt = target.GetStatValue(StatDefOf.ArmorRating_Blunt);
float sharp = target.GetStatValue(StatDefOf.ArmorRating_Sharp);
DamageType dt = (blunt > sharp) ? DamageType.Sharp : DamageType.Blunt;
if (blunt == sharp)
{
if (Settings.PreferredDamageType == PreferredDamageTypeEnum.WeaponStorage_None)
{
return;
}
dt = (Settings.PreferredDamageType == PreferredDamageTypeEnum.ArmorBlunt) ? DamageType.Blunt : DamageType.Sharp;
}
if (TryGetBestWeapon(dt, pawn.equipment.Primary, c, out ThingWithComps bestWeapon))
{
HarmonyPatchUtil.EquipWeapon(bestWeapon, pawn, c);
}
}
}
}
public static void EquipWeapon(ThingWithComps weapon, Pawn pawn)
{
if (WorldComp.TryGetAssignedWeapons(pawn, out AssignedWeaponContainer c))
{
EquipWeapon(weapon, pawn, c);
}
}
static void Postfix(ref bool __result, ref Pawn __state, ThingWithComps eq)
{
if (__state != null)
{
if (WorldComp.TryGetAssignedWeapons(__state, out AssignedWeaponContainer c) &&
c.Contains(eq))
{
if (!Settings.AllowPawnsToDropWeapon)
{
if (!WorldComp.Add(eq))
{
Log.Warning($"unable to find weapon storage that can hold {eq.ThingID} so it will be dropped.");
WorldComp.Drop(eq);
}
}
else
{
if (c.Remove(eq))
{
if (eq.def.IsRangedWeapon)
{
if (!HarmonyPatchUtil.EquipRanged(c))
{
HarmonyPatchUtil.EquipMelee(c);
}
}
else
{
if (!HarmonyPatchUtil.EquipMelee(c))
{
HarmonyPatchUtil.EquipRanged(c);
}
}
}
if (Settings.PlaceDroppedWeaponsInStorage)
{
if (!WorldComp.Add(eq))
{
Log.Warning($"unable to find weapon storage that can hold {eq.ThingID} so it will be dropped.");
WorldComp.Drop(eq);
}
}
}
}
else
{
foreach (SharedWeaponFilter swf in WorldComp.SharedWeaponFilter)
{
if (swf.Allows(eq) &&
!WorldComp.Add(eq))
{
Log.Warning($"unable to find weapon storage that can hold {eq.ThingID} so it will be dropped.");
WorldComp.Drop(eq);
break;
}
}
}
}
}
static void Prefix(Verb_ShootOneUse __instance)
{
if (__instance.caster is Pawn pawn &&
WorldComp.TryGetAssignedWeapons(pawn, out AssignedWeaponContainer c))
{
c.Remove(__instance.EquipmentSource);
}
}
static void Postfix(Pawn_DraftController __instance)
{
Pawn pawn = __instance.pawn;
if (WorldComp.TryGetAssignedWeapons(pawn, out AssignedWeaponContainer weapons) &&
weapons.TryGetLastThingUsed(pawn, out ThingWithComps w))
{
HarmonyPatchUtil.EquipWeapon(w, pawn, weapons);
}
}
static void Prefix(Pawn_EquipmentTracker __instance, ThingWithComps eq)
{
if (eq.def.equipmentType == EquipmentType.Primary && __instance.Primary != null)
{
if (WorldComp.TryGetAssignedWeapons(__instance.pawn, out AssignedWeaponContainer c) &&
c.Contains(eq))
{
__instance.Remove(eq);
}
}
}
static void Prefix(Pawn_HealthTracker __instance, ref ThingWithComps __state)
{
if (Settings.EnableAssignWeapons &&
pawnFI.GetValue(__instance) is Pawn pawn &&
!__instance.Downed &&
pawn.IsColonist &&
pawn.equipment?.Primary != null)
{
if (WorldComp.TryGetAssignedWeapons(pawn, out AssignedWeaponContainer c))
{
HarmonyPatchUtil.UnequipPrimaryWeapon(pawn, c);
}
}
}
static void Postfix(Pawn __instance, ref State __state)
{
if (__instance.Dead && __instance.IsColonist && __instance.apparel?.LockedApparel?.Count == 0 && __state.Weapon != null)
{
if (WorldComp.Add(__state.Weapon))
{
__instance.equipment?.Remove(__state.Weapon);
}
if (WorldComp.TryGetAssignedWeapons(__instance, out AssignedWeaponContainer c))
{
WorldComp.RemoveAssignedWeapons(__instance);
foreach (ThingWithComps w in c.Weapons)
{
if (!WorldComp.Add(w))
{
BuildingUtil.DropSingleThing(w, __instance.Position, __state.Map);
}
}
}
}
}
static void Prefix(Pawn_HealthTracker __instance)
{
if (!Settings.EnableAssignWeapons)
{
return;
}
Pawn pawn = (Pawn)__instance.GetType().GetField(
"pawn", BindingFlags.NonPublic | BindingFlags.Instance).GetValue(__instance);
if (pawn != null &&
pawn.Faction == Faction.OfPlayer &&
pawn.def.race.Humanlike)
{
if (WorldComp.TryGetAssignedWeapons(pawn, out AssignedWeaponContainer c) &&
pawn.equipment?.Primary == null)
{
if (c.TryGetLastThingUsed(pawn, out ThingWithComps w))
{
HarmonyPatchUtil.EquipWeapon(w, pawn, c);
}
}
}
}
static void Postfix(Pawn_DraftController __instance, ref IEnumerable <Gizmo> __result)
{
try
{
Pawn pawn = __instance.pawn;
if (pawn.Faction == Faction.OfPlayer)
{
List <Gizmo> l = null;
if (Settings.ShowWeaponStorageButtonForPawns && WorldComp.HasStorages())
{
l = new List <Gizmo>()
{
new Command_Action
{
icon = AssignUI.weaponStorageTexture,
defaultLabel = "WeaponStorage.UseWeaponStorage".Translate(),
activateSound = SoundDef.Named("Click"),
action = delegate
{
Find.WindowStack.Add(new AssignUI(null, pawn));
}
}
};
l.AddRange(__result);
}
if (WorldComp.TryGetAssignedWeapons(__instance.pawn, out AssignedWeaponContainer weapons))
{
if (l == null)
{
l = new List <Gizmo>();
l.AddRange(__result);
}
//if (pawn.equipment.Primary != null)
// l.Add(CreateUnequipGizmo(pawn, weapons));
foreach (ThingWithComps weapon in weapons.Weapons)
{
bool isTool = Settings.IsTool(weapon);
bool show = false;
if (pawn.Drafted)
{
show = true;
}
else // Not drafted
{
if (isTool || Settings.ShowWeaponsWhenNotDrafted)
{
show = true;
}
}
if (show)
{
show = pawn.equipment.Primary != weapon;
}
if (show)
{
l.Add(CreateEquipWeaponGizmo(weapon.def, pawn,
delegate
{
HarmonyPatchUtil.EquipWeapon(weapon, pawn, weapons);
weapons.SetLastThingUsed(pawn, weapon, false);
}));
}
}
}
foreach (SharedWeaponFilter f in WorldComp.SharedWeaponFilter)
{
if (l == null)
{
l = new List <Gizmo>();
l.AddRange(__result);
}
f.UpdateFoundDefCache();
if (f.AssignedPawns.Contains(pawn))
{
if (l == null)
{
l = new List <Gizmo>();
if (__result != null)
{
l.AddRange(__result);
}
}
foreach (ThingDef d in f.AllowedDefs)
{
if (d != pawn.equipment.Primary?.def &&
f.FoundDefCacheContains(d))
{
l.Add(CreateEquipWeaponGizmo(d, pawn,
delegate
{
if (WorldComp.TryRemoveWeapon(d, f, false, out ThingWithComps weapon))
{
HarmonyPatchUtil.EquipWeapon(weapon, pawn);
f.UpdateDefCache(d);
}
}, "WeaponStorage.EquipShared"));
}