public static Thing PickBestArmorFor(Pawn pawn)
{
if (pawn.outfits == null)
{
return(null);
}
if (pawn.Faction != Faction.OfPlayer)
{
return(null);
}
if (pawn.IsQuestLodger())
{
return(null);
}
Outfit currentOutfit = pawn.outfits.CurrentOutfit;
Thing thing = null;
float num2 = 0f;
List <Thing> list = pawn.Map.listerThings.ThingsInGroup(ThingRequestGroup.Apparel);
if (list.Count == 0)
{
return(null);
}
neededWarmth = PawnApparelGenerator.CalculateNeededWarmth(pawn, pawn.Map.Tile, GenLocalDate.Twelfth(pawn));
wornApparelScores = new List <float>();
List <Apparel> wornApparel = pawn.apparel.WornApparel;
for (int i = 0; i < wornApparel.Count; i++)
{
wornApparelScores.Add(ApparelScoreRaw(pawn, wornApparel[i]));
}
for (int j = 0; j < list.Count; j++)
{
Apparel apparel = (Apparel)list[j]; // && apparel.IsInAnyStorage()
if (!apparel.IsBurning() && currentOutfit.filter.Allows(apparel) && !apparel.IsForbidden(pawn) &&
(apparel.def.apparel.gender == Gender.None || apparel.def.apparel.gender == pawn.gender))
{
float num3 = ApparelScoreGain_NewTmp(pawn, apparel, wornApparelScores);
if (!(num3 < 0.05f) && !(num3 < num2) && (!CompBiocodable.IsBiocoded(apparel) || CompBiocodable.IsBiocodedFor(apparel, pawn)) &&
ApparelUtility.HasPartsToWear(pawn, apparel.def) &&
pawn.CanReserveAndReach(apparel, PathEndMode.OnCell, pawn.NormalMaxDanger()))
{
thing = apparel;
num2 = num3;
}
}
}
return(thing);
}
public static void Postfix(Vector3 clickPos, Pawn pawn, List <FloatMenuOption> opts)
{
IntVec3 position = IntVec3.FromVector3(clickPos);
// Add options for equipment.
if (pawn.equipment != null)
{
List <Thing> things = position.GetThingList(pawn.Map);
foreach (Thing thing in things)
{
if (thing.TryGetComp <CompEquippable>() != null)
{
ThingWithComps equipment = (ThingWithComps)thing;
if (!SimpleSidearmUtility.IsActive &&
equipment.def.IsWeapon &&
!MassUtility.WillBeOverEncumberedAfterPickingUp(pawn, equipment, 1) &&
!pawn.WorkTagIsDisabled(WorkTags.Violent) &&
pawn.CanReach(equipment, PathEndMode.ClosestTouch, Danger.Deadly) &&
pawn.health.capacities.CapableOf(PawnCapacityDefOf.Manipulation) &&
!(pawn.IsQuestLodger() && (!equipment.def.IsWeapon || pawn.equipment.Primary != null)) &&
EquipmentUtility.CanEquip(equipment, pawn, out _))
{
string text3 = UIText.AIEquip.Translate(thing.LabelShort);
if (equipment.def.IsRangedWeapon && pawn.story != null && pawn.story.traits.HasTrait(TraitDefOf.Brawler))
{
text3 += " " + UIText.EquipWarningBrawler.Translate();
}
opts.Add(ContextMenuUtility.MakeDecoratedEquipOption(pawn, equipment, text3, () => AddToLoadoutDialog(equipment)));
}
}
}
}
// Add options for apparel.
if (pawn.apparel != null)
{
Apparel apparel = pawn.Map.thingGrid.ThingAt <Apparel>(position);
if (apparel != null)
{
if (pawn.CanReach(apparel, PathEndMode.ClosestTouch, Danger.Deadly) &&
!apparel.IsBurning() &&
!MassUtility.WillBeOverEncumberedAfterPickingUp(pawn, apparel, apparel.stackCount) &&
ApparelOptionUtility.CanWear(pawn, apparel))
{
opts.Add(
ContextMenuUtility.MakeDecoratedWearApparelOption(
pawn, apparel, UIText.AIForceWear.Translate(apparel.LabelShort), true, () => AddToLoadoutDialog(apparel)));
opts.Add(
ContextMenuUtility.MakeDecoratedWearApparelOption(
pawn, apparel, UIText.AIWear.Translate(apparel.LabelShort), false, () => AddToLoadoutDialog(apparel)));
}
}
}
void AddToLoadoutDialog(Thing thing1)
{
if (AwesomeInventoryMod.Settings.OpenLoadoutInContextMenu && pawn.UseLoadout(out Loadout.CompAwesomeInventoryLoadout comp))
{
Find.WindowStack.Add(
new Dialog_InstantMessage(
UIText.AddToLoadout.Translate(comp.Loadout.label) + Environment.NewLine + $"({UIText.DisableWindowInModSetting.TranslateSimple()})"
, _size
, "Yes".TranslateSimple()
, () => ContextMenuUtility.AddToLoadoutDialog(pawn, comp, thing1)
, "No".TranslateSimple()));
}
}
}
public static void Postfix(Vector3 clickPos, Pawn pawn, List <FloatMenuOption> opts)
{
if (!AwesomeInventoryServiceProvider.TryGetImplementation <IInventoryHelper>(out IInventoryHelper inventoryHelper))
{
Log.Error(string.Format(ErrorMessage.NotImplemented, typeof(IInventoryHelper).Name));
return;
}
IntVec3 position = IntVec3.FromVector3(clickPos);
// Add options for equipment.
if (pawn.equipment != null)
{
List <Thing> things = position.GetThingList(pawn.Map);
foreach (Thing thing in things)
{
if (thing.TryGetComp <CompEquippable>() != null)
{
ThingWithComps equipment = (ThingWithComps)thing;
if (equipment.def.IsWeapon &&
!pawn.WorkTagIsDisabled(WorkTags.Violent) &&
pawn.CanReach(equipment, PathEndMode.ClosestTouch, Danger.Deadly) &&
pawn.health.capacities.CapableOf(PawnCapacityDefOf.Manipulation) &&
!(pawn.IsQuestLodger() && (!equipment.def.IsWeapon || pawn.equipment.Primary != null)) &&
EquipmentUtility.CanEquip(equipment, pawn, out _))
{
string text3 = UIText.AIEquip.Translate(thing.LabelShort);
if (equipment.def.IsRangedWeapon && pawn.story != null && pawn.story.traits.HasTrait(TraitDefOf.Brawler))
{
text3 += " " + UIText.EquipWarningBrawler.Translate();
}
var option = FloatMenuUtility.DecoratePrioritizedTask(
new FloatMenuOption(
text3,
() =>
{
TaggedString equipWeaponConfirmationDialogText = ThingRequiringRoyalPermissionUtility.GetEquipWeaponConfirmationDialogText(equipment, pawn);
CompBladelinkWeapon compBladelinkWeapon = equipment.TryGetComp <CompBladelinkWeapon>();
if (compBladelinkWeapon != null && compBladelinkWeapon.bondedPawn != pawn)
{
if (!equipWeaponConfirmationDialogText.NullOrEmpty())
{
equipWeaponConfirmationDialogText += "\n\n";
}
equipWeaponConfirmationDialogText += "BladelinkEquipWarning".Translate();
}
if (!equipWeaponConfirmationDialogText.NullOrEmpty())
{
equipWeaponConfirmationDialogText += "\n\n" + "RoyalWeaponEquipConfirmation".Translate();
Find.WindowStack.Add(
new Dialog_MessageBox(
equipWeaponConfirmationDialogText,
"Yes".Translate(),
() =>
{
Equip();
},
"No".Translate()));
}
else
{
Equip();
}
},
MenuOptionPriority.High),
pawn,
equipment);
opts.Add(option);
}
void Equip()
{
equipment.SetForbidden(value: false);
pawn.jobs.TryTakeOrderedJob(JobMaker.MakeJob(AwesomeInventory_JobDefOf.AwesomeInventory_MapEquip, equipment));
MoteMaker.MakeStaticMote(equipment.DrawPos, equipment.Map, ThingDefOf.Mote_FeedbackEquip);
PlayerKnowledgeDatabase.KnowledgeDemonstrated(ConceptDefOf.EquippingWeapons, KnowledgeAmount.Total);
}
}
}
}
// Add options for apparel.
if (pawn.apparel != null)
{
Apparel apparel = pawn.Map.thingGrid.ThingAt <Apparel>(position);
if (apparel != null)
{
if (pawn.CanReach(apparel, PathEndMode.ClosestTouch, Danger.Deadly) &&
!apparel.IsBurning() &&
!pawn.apparel.WouldReplaceLockedApparel(apparel) &&
ApparelUtility.HasPartsToWear(pawn, apparel.def))
{
FloatMenuOption option = FloatMenuUtility.DecoratePrioritizedTask(
new FloatMenuOption(
UIText.AIForceWear.Translate(apparel.LabelShort),
() =>
{
DressJob dressJob = SimplePool <DressJob> .Get();
dressJob.def = AwesomeInventory_JobDefOf.AwesomeInventory_Dress;
dressJob.targetA = apparel;
dressJob.ForceWear = true;
apparel.SetForbidden(value: false);
pawn.jobs.TryTakeOrderedJob(dressJob);
},
MenuOptionPriority.High),
pawn,
apparel);
opts.Add(option);
option = FloatMenuUtility.DecoratePrioritizedTask(
new FloatMenuOption(
UIText.AIWear.Translate(apparel.LabelShort),
() =>
{
DressJob dressJob = SimplePool <DressJob> .Get();
dressJob.def = AwesomeInventory_JobDefOf.AwesomeInventory_Dress;
dressJob.targetA = apparel;
dressJob.ForceWear = false;
apparel.SetForbidden(value: false);
pawn.jobs.TryTakeOrderedJob(dressJob);
},
MenuOptionPriority.High),
pawn,
apparel);
opts.Add(option);
}
}
}
List <Thing> items = position.GetThingList(pawn.Map);
foreach (Thing item in items)
{
if (item.def.category == ThingCategory.Item)
{
int count = MassUtility.CountToPickUpUntilOverEncumbered(pawn, item);
if (count == 0)
{
continue;
}
count = Math.Min(count, item.stackCount);
string displayText = UIText.Pickup.Translate(item.LabelNoCount + " x" + count);
var option = FloatMenuUtility.DecoratePrioritizedTask(
new FloatMenuOption(
displayText
, () =>
{
Job job = JobMaker.MakeJob(JobDefOf.TakeInventory, item);
job.count = count;
pawn.jobs.TryTakeOrderedJob(job);
})
, pawn
, item);
opts.Add(option);
}
}
}
public static void Postfix(Vector3 clickPos, Pawn pawn, List <FloatMenuOption> opts)
{
IntVec3 position = IntVec3.FromVector3(clickPos);
// Add options for equipment.
if (pawn.equipment != null)
{
List <Thing> things = position.GetThingList(pawn.Map);
foreach (Thing thing in things)
{
if (thing.TryGetComp <CompEquippable>() != null)
{
ThingWithComps equipment = (ThingWithComps)thing;
if (equipment.def.IsWeapon &&
!pawn.WorkTagIsDisabled(WorkTags.Violent) &&
pawn.CanReach(equipment, PathEndMode.ClosestTouch, Danger.Deadly) &&
pawn.health.capacities.CapableOf(PawnCapacityDefOf.Manipulation) &&
!(pawn.IsQuestLodger() && (!equipment.def.IsWeapon || pawn.equipment.Primary != null)) &&
EquipmentUtility.CanEquip(equipment, pawn, out _))
{
string text3 = UIText.AIEquip.Translate(thing.LabelShort);
if (equipment.def.IsRangedWeapon && pawn.story != null && pawn.story.traits.HasTrait(TraitDefOf.Brawler))
{
text3 += " " + UIText.EquipWarningBrawler.Translate();
}
var option = FloatMenuUtility.DecoratePrioritizedTask(
new FloatMenuOption(
text3,
() =>
{
TaggedString equipWeaponConfirmationDialogText = ThingRequiringRoyalPermissionUtility.GetEquipWeaponConfirmationDialogText(equipment, pawn);
CompBladelinkWeapon compBladelinkWeapon = equipment.TryGetComp <CompBladelinkWeapon>();
if (compBladelinkWeapon != null && compBladelinkWeapon.bondedPawn != pawn)
{
if (!equipWeaponConfirmationDialogText.NullOrEmpty())
{
equipWeaponConfirmationDialogText += "\n\n";
}
equipWeaponConfirmationDialogText += "BladelinkEquipWarning".Translate();
}
if (!equipWeaponConfirmationDialogText.NullOrEmpty())
{
equipWeaponConfirmationDialogText += "\n\n" + "RoyalWeaponEquipConfirmation".Translate();
Find.WindowStack.Add(
new Dialog_MessageBox(
equipWeaponConfirmationDialogText,
"Yes".Translate(),
() =>
{
Equip();
},
"No".Translate()));
}
else
{
Equip();
}
},
MenuOptionPriority.High),
pawn,
equipment);
opts.Add(option);
}
void Equip()
{
equipment.SetForbidden(value: false);
pawn.jobs.TryTakeOrderedJob(JobMaker.MakeJob(AwesomeInventory_JobDefOf.AwesomeInventory_MapEquip, equipment));
MoteMaker.MakeStaticMote(equipment.DrawPos, equipment.Map, ThingDefOf.Mote_FeedbackEquip);
PlayerKnowledgeDatabase.KnowledgeDemonstrated(ConceptDefOf.EquippingWeapons, KnowledgeAmount.Total);
}
}
}
}
// Add options for apparel.
if (pawn.apparel != null)
{
Apparel apparel = pawn.Map.thingGrid.ThingAt <Apparel>(position);
if (apparel != null)
{
if (pawn.CanReach(apparel, PathEndMode.ClosestTouch, Danger.Deadly) &&
!apparel.IsBurning() &&
ApparelOptionUtility.CanWear(pawn, apparel))
{
FloatMenuOption option = FloatMenuUtility.DecoratePrioritizedTask(
new FloatMenuOption(
UIText.AIForceWear.Translate(apparel.LabelShort),
() =>
{
DressJob dressJob = SimplePool <DressJob> .Get();
dressJob.def = AwesomeInventory_JobDefOf.AwesomeInventory_Dress;
dressJob.targetA = apparel;
dressJob.ForceWear = true;
apparel.SetForbidden(value: false);
pawn.jobs.TryTakeOrderedJob(dressJob);
},
MenuOptionPriority.High),
pawn,
apparel);
opts.Add(option);
option = FloatMenuUtility.DecoratePrioritizedTask(
new FloatMenuOption(
UIText.AIWear.Translate(apparel.LabelShort),
() =>
{
DressJob dressJob = SimplePool <DressJob> .Get();
dressJob.def = AwesomeInventory_JobDefOf.AwesomeInventory_Dress;
dressJob.targetA = apparel;
dressJob.ForceWear = false;
apparel.SetForbidden(value: false);
pawn.jobs.TryTakeOrderedJob(dressJob);
},
MenuOptionPriority.High),
pawn,
apparel);
opts.Add(option);
}
}
}
}
public static bool TryGiveJob(JobGiver_OptimizeApparel __instance, ref Job __result, Pawn pawn)
{
if (pawn.outfits == null)
{
Log.ErrorOnce(string.Concat(pawn, " tried to run JobGiver_OptimizeApparel without an OutfitTracker"), 5643897);
__result = null;
return(false);
}
if (pawn.Faction != Faction.OfPlayer)
{
Log.ErrorOnce(string.Concat("Non-colonist ", pawn, " tried to optimize apparel."), 764323);
__result = null;
return(false);
}
if (pawn.IsQuestLodger())
{
__result = null;
return(false);
}
if (!DebugViewSettings.debugApparelOptimize)
{
if (Find.TickManager.TicksGame < pawn.mindState.nextApparelOptimizeTick)
{
__result = null;
return(false);
}
}
else
{
debugSb = new StringBuilder();
debugSb.AppendLine(string.Concat("Scanning for ", pawn, " at ", pawn.Position));
}
Outfit currentOutfit = pawn.outfits.CurrentOutfit;
List <Apparel> wornApparel = pawn.apparel.WornApparel;
for (int num = wornApparel.Count - 1; num >= 0; num--)
{
if (!currentOutfit.filter.Allows(wornApparel[num]) && pawn.outfits.forcedHandler.AllowedToAutomaticallyDrop(wornApparel[num]) && !pawn.apparel.IsLocked(wornApparel[num]))
{
Job job = JobMaker.MakeJob(JobDefOf.RemoveApparel, wornApparel[num]);
job.haulDroppedApparel = true;
__result = job;
return(false);
}
}
Thing thing = null;
float num2 = 0f;
List <Thing> list = pawn.Map.listerThings.ThingsInGroup(ThingRequestGroup.Apparel);
if (list.Count == 0)
{
SetNextOptimizeTick2(pawn);
__result = null;
return(false);
}
neededWarmth = PawnApparelGenerator.CalculateNeededWarmth(pawn, pawn.Map.Tile, GenLocalDate.Twelfth(pawn));
//wornApparelScores.Clear();
List <float> wornApparelScores = new List <float>();
for (int i = 0; i < wornApparel.Count; i++)
{
wornApparelScores.Add(JobGiver_OptimizeApparel.ApparelScoreRaw(pawn, wornApparel[i]));
}
for (int j = 0; j < list.Count; j++)
{
Apparel apparel = (Apparel)list[j];
if (currentOutfit.filter.Allows(apparel) && apparel.IsInAnyStorage() && !apparel.IsForbidden(pawn) && !apparel.IsBurning() && (apparel.def.apparel.gender == Gender.None || apparel.def.apparel.gender == pawn.gender))
{
float num3 = JobGiver_OptimizeApparel.ApparelScoreGain_NewTmp(pawn, apparel, wornApparelScores);
if (DebugViewSettings.debugApparelOptimize)
{
debugSb.AppendLine(apparel.LabelCap + ": " + num3.ToString("F2"));
}
if (!(num3 < 0.05f) && !(num3 < num2) && (!EquipmentUtility.IsBiocoded(apparel) || EquipmentUtility.IsBiocodedFor(apparel, pawn)) && ApparelUtility.HasPartsToWear(pawn, apparel.def) && pawn.CanReserveAndReach(apparel, PathEndMode.OnCell, pawn.NormalMaxDanger()))
{
thing = apparel;
num2 = num3;
}
}
}
if (DebugViewSettings.debugApparelOptimize)
{
debugSb.AppendLine("BEST: " + thing);
Log.Message(debugSb.ToString());
debugSb = null;
}
if (thing == null)
{
SetNextOptimizeTick2(pawn);
__result = null;
return(false);
}
__result = JobMaker.MakeJob(JobDefOf.Wear, thing);
return(false);
}
