private static void PullAmmoData(ModdedSaveData data, Game game)
{
foreach (var ammo in AmmoDataUtils.GetAllAmmoData(game))
{
var modsInThis = new HashSet <SRMod>(ammo.Select((x) => ModdedIDRegistry.IsModdedID(x.id) ? ModdedIDRegistry.ModForID(x.id) : null));
var belongingMod = AmmoIdentifier.TryGetIdentifier(ammo, game, out var id) ? AmmoIdentifier.GetModForIdentifier(id) : null;
modsInThis.Add(belongingMod);
modsInThis.Remove(null);
foreach (var mod in modsInThis)
{
if (mod == null)
{
continue;
}
if (AmmoIdentifier.TryGetIdentifier(ammo, game, out var identifier))
{
var segment = data.GetSegmentForMod(mod);
segment.customAmmo[identifier] =
AmmoDataUtils.RipOutWhere(ammo, (x) => mod == belongingMod?ModdedIDRegistry.ModForID(x.id) == null:ModdedIDRegistry.ModForID(x.id) == mod, false);
}
else
{
Debug.LogError("Unknown ammo identifier, skipping...");
}
}
}
}
public static void Postfix(Ammo __instance, bool __result, Identifiable.Id id, Identifiable identifiable,
int slotIdx, int count, bool overflow)
{
if (!__result)
{
return;
}
if (AmmoIdentifier.TryGetIdentifier(__instance, out var identifier))
{
if (identifiable && ExtendedData.IsRegistered(identifiable.gameObject))
{
var ammo = PersistentAmmoManager.GetPersistentAmmoForAmmo(__instance.ammoModel);
for (int i = 0; i < count; i++)
{
ammo.DataModel.slots[slotIdx]
.PushTop(ExtendedData.extendedActorData[identifiable.GetActorId()]);
}
}
else
{
if (!PersistentAmmoManager.HasPersistentAmmo(identifier))
{
return;
}
var ammo = PersistentAmmoManager.PersistentAmmoData[identifier];
for (int i = 0; i < count; i++)
{
ammo.DataModel.slots[slotIdx].PushTop(null);
}
}
}
}
public static void Postfix(Ammo __instance, int amount)
{
if (AmmoIdentifier.TryGetIdentifier(__instance, out var identifier))
{
if (PersistentAmmoManager.HasPersistentAmmo(identifier))
{
PersistentAmmoManager.OnAmmoDecrement(identifier, __instance.selectedAmmoIdx, amount);
}
}
}
public static void Postfix(Ammo __instance, int idx, bool __result)
{
if (idx < __instance.ammoModel.usableSlots)
{
if (AmmoIdentifier.TryGetIdentifier(__instance, out var identifier))
{
if (PersistentAmmoManager.HasPersistentAmmo(identifier))
{
PersistentAmmoManager.PersistentAmmoData[identifier].OnSelected(__instance.GetSelectedId(), __instance.selectedAmmoIdx);
}
}
}
}
public static void Postfix(Ammo __instance, GameObject __result)
{
if (!__result)
{
return;
}
if ((new StackFrame(2)).GetMethod().DeclaringType == typeof(VacColorAnimator))
{
return;
}
if (AmmoIdentifier.TryGetIdentifier(__instance, out var identifier))
{
if (PersistentAmmoManager.HasPersistentAmmo(identifier))
{
PersistentAmmoManager.PersistentAmmoData[identifier].OnSelected(Identifiable.GetId(__result), __instance.selectedAmmoIdx);
}
}
}
public static void Postfix(Ammo __instance, bool __result, Identifiable.Id id, Identifiable identifiable)
{
if (!__result)
{
return;
}
if (AmmoIdentifier.TryGetIdentifier(__instance, out var identifier))
{
var count = -1;
for (int i = 0; i < __instance.Slots.Length; i++)
{
if (__instance.Slots[i]?.id == id)
{
count = i;
break;
}
}
if (count == -1)
{
throw new Exception();
}
if (identifiable && ExtendedData.HasExtendedData(identifiable.gameObject))
{
var ammo = PersistentAmmoManager.GetPersistentAmmoForAmmo(__instance.ammoModel);
ammo.DataModel.slots[count].PushTop(ExtendedData.ReadDataFromGameObject(identifiable.gameObject));
ammo.Sync();
}
else
{
if (!PersistentAmmoManager.HasPersistentAmmo(identifier))
{
return;
}
var ammo = PersistentAmmoManager.GetPersistentAmmoForAmmo(__instance.ammoModel);
ammo.DataModel.slots[count].PushTop(null);
ammo.Sync();
}
}
}
public static void Postfix(Ammo __instance, bool __result, Identifiable.Id id, Identifiable identifiable,
int slotIdx, int count, bool overflow)
{
if (!__result)
{
return;
}
if (AmmoIdentifier.TryGetIdentifier(__instance, out var identifier))
{
if (identifiable && ExtendedData.HasExtendedData(identifiable.gameObject))
{
var ammo = PersistentAmmoManager.GetPersistentAmmoForAmmo(__instance.ammoModel);
if (ammo == null)
{
return;
}
for (int i = 0; i < count; i++)
{
ammo.DataModel.slots[slotIdx]
.PushTop(ExtendedData.ReadDataFromGameObject(identifiable.gameObject));
}
ammo.Sync();
}
else
{
if (!PersistentAmmoManager.HasPersistentAmmo(identifier))
{
return;
}
var ammo = PersistentAmmoManager.PersistentAmmoData[identifier];
for (int i = 0; i < count; i++)
{
ammo.DataModel.slots[slotIdx].PushTop(null);
}
ammo.Sync();
}
}
}
public static void PullModdedData(GameV09 game)
{
data.segments.Clear();
data.ammoDataEntries.Clear();
foreach (var actor in game.actors.Where((x) => SaveRegistry.IsCustom(x)))
{
var segment = data.GetSegmentForMod(SaveRegistry.ModForData(actor));
segment.identifiableData.Add(new IdentifiedData()
{
data = actor,
dataID = new DataIdentifier()
{
longID = actor.actorId, stringID = "", Type = IdentifierType.ACTOR
}
});
}
foreach (var gadget in game.world.placedGadgets.Where((x) => SaveRegistry.IsCustom(x.Value)))
{
var segment = data.GetSegmentForMod(SaveRegistry.ModForData(gadget.Value));
segment.identifiableData.Add(new IdentifiedData()
{
data = gadget.Value,
dataID = new DataIdentifier()
{
longID = 0, stringID = gadget.Key, Type = IdentifierType.GADGET
}
});
}
foreach (var mod in ModPlayerData.FindAllModsWithData(game.player))
{
var segment = data.GetSegmentForMod(mod);
segment.playerData.Pull(game.player, mod);
}
PediaDataBuffer buf = new PediaDataBuffer(game.pedia);
foreach (var mod in ModPediaData.FindAllModsWithData(buf))
{
var segment = data.GetSegmentForMod(mod);
segment.pediaData.Pull(buf, mod);
}
foreach (var ammo in AmmoDataUtils.GetAllAmmoData(game).Where((x) => AmmoDataUtils.HasCustomData(x)))
{
var modsInThis = new HashSet <SRMod>(ammo.Select((x) => SaveRegistry.IsCustom(x.id) ? SaveRegistry.ModForID(x.id) : null));
modsInThis.Remove(null);
foreach (var mod in modsInThis)
{
if (mod == null)
{
continue;
}
if (AmmoIdentifier.TryGetIdentifier(ammo, game, out var identifier))
{
var segment = data.GetSegmentForMod(mod);
segment.customAmmo[identifier] =
AmmoDataUtils.RipOutWhere(ammo, (x) => SaveRegistry.ModForID(x.id) == mod, false);
}
else
{
throw new Exception("OH GOD ITS HAPPENING");
}
}
}
ExtendedData.Push(data);
PersistentAmmoManager.SyncAll();
PersistentAmmoManager.Push(data);
}
public static void Prefix(GameV12 __instance, ref RemovalData __state)
{
__state = new RemovalData();
__state.AddAndRemoveWhereCustom(__instance.actors);
__state.AddAndRemoveWhere(__instance.world.placedGadgets, (x) => SaveRegistry.IsCustom(x.Value) || ModdedStringRegistry.IsModdedString(x.Key));
__state.AddAndRemoveWhere(__instance.ranch.plots, (x) => SaveRegistry.IsCustom(x) || ModdedStringRegistry.IsModdedString(x.id));
__state.AddAndRemoveWhere(__instance.world.gordos, x => SaveRegistry.IsCustom(x.Value) || ModdedStringRegistry.IsModdedString(x.Key));
__state.AddAndRemoveWhere(__instance.world.treasurePods, x => SaveRegistry.IsCustom(x.Value) || ModdedStringRegistry.IsModdedString(x.Key));
__state.AddAndRemoveWhere(__instance.world.offers, x => SaveRegistry.IsCustom(x.Value) || ModdedIDRegistry.IsModdedID(x.Key) || ExchangeOfferRegistry.IsCustom(x.Value));
__state.AddAndRemoveWhere(__instance.world.econSaturations, (x) => ModdedIDRegistry.IsModdedID(x.Key));
__state.AddAndRemoveWhere(__instance.world.lastOfferRancherIds, ExchangeOfferRegistry.IsCustom);
__state.AddAndRemoveWhere(__instance.world.pendingOfferRancherIds, ExchangeOfferRegistry.IsCustom);
__state.AddAndRemoveWhereCustom(__instance.player.upgrades);
__state.AddAndRemoveWhereCustom(__instance.player.availUpgrades);
__state.AddAndRemoveWhere(__instance.player.upgradeLocks,
(x) => ModdedIDRegistry.IsModdedID(x.Key));
__state.AddAndRemoveWhereCustom(__instance.player.blueprints);
__state.AddAndRemoveWhereCustom(__instance.player.availBlueprints);
__state.AddAndRemoveWhere(__instance.player.blueprintLocks, (x) => ModdedIDRegistry.IsModdedID(x.Key));
__state.AddAndRemoveWhere(__instance.player.progress, (x) => ModdedIDRegistry.IsModdedID(x.Key));
__state.AddAndRemoveWhere(__instance.player.delayedProgress, (x) => ModdedIDRegistry.IsModdedID(x.Key));
__state.AddAndRemoveWhere(__instance.player.gadgets, (x) => ModdedIDRegistry.IsModdedID(x.Key));
__state.AddAndRemoveWhere(__instance.player.craftMatCounts, (x) => ModdedIDRegistry.IsModdedID(x.Key));
__state.AddAndRemoveWhereCustom(__instance.player.unlockedZoneMaps);
__state.AddAndRemoveWhere(__instance.player.mail, (x) => MailRegistry.GetModForMail(x.messageKey) != null);
__state.AddAndRemoveWhere(__instance.pedia.unlockedIds, (x) => ModdedIDRegistry.IsModdedID(Enum.Parse(typeof(PediaDirector.Id), x)));
__state.AddAndRemoveWhere(__instance.pedia.completedTuts, (x) => ModdedIDRegistry.IsModdedID(Enum.Parse(typeof(TutorialDirector.Id), x)));
__state.AddAndRemoveWhere(__instance.pedia.popupQueue, (x) => ModdedIDRegistry.IsModdedID(Enum.Parse(typeof(TutorialDirector.Id), x)));
foreach (var data in AmmoDataUtils.GetAllAmmoData(__instance))
{
if (AmmoIdentifier.TryGetIdentifier(data, __instance, out var id) && AmmoIdentifier.IsModdedIdentifier(id))
{
__state.addBacks.Add(AmmoDataUtils.RemoveAmmoDataWithAddBack(data, __instance));
}
else
{
var moddedData = AmmoDataUtils.RipOutModdedData(data);
__state.addBacks.Add(() =>
{
AmmoDataUtils.SpliceAmmoData(data, moddedData);
});
}
}
void RemovePartial(object actor, RemovalData data)
{
if (CustomChecker.GetCustomLevel(actor) == CustomChecker.CustomLevel.PARTIAL)
{
var partial = PartialData.GetPartialData(actor.GetType(), true);
partial.Pull(actor);
data.addBacks.Add(() =>
{
partial.Push(actor);
});
}
}
foreach (var actor in __instance.actors)
{
RemovePartial(actor, __state);
}
foreach (var actor in __instance.ranch.plots)
{
RemovePartial(actor, __state);
}
foreach (var actor in __instance.world.placedGadgets)
{
RemovePartial(actor.Value, __state);
}
foreach (var actor in __instance.world.gordos)
{
RemovePartial(actor.Value, __state);
}
foreach (var actor in __instance.world.treasurePods)
{
RemovePartial(actor.Value, __state);
}
foreach (var offer in __instance.world.offers)
{
RemovePartial(offer.Value, __state);
}
var partialAppearance = new PartialAppearancesData();
partialAppearance.Pull(__instance.appearances);
__state.addBacks.Add(() => partialAppearance.Push(__instance.appearances));
}