-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #9 from dwesterwick/RestoreExcludedLayers
Allow Removed Layers to Be Restored during a Raid
- Loading branch information
Showing
7 changed files
with
295 additions
and
89 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,125 @@ | ||
| using BepInEx.Logging; | ||
| using DrakiaXYZ.BigBrain.Brains; | ||
| using EFT; | ||
| using HarmonyLib; | ||
| using System; | ||
| using System.Collections.Generic; | ||
| using System.Linq; | ||
| using System.Reflection; | ||
| using System.Text; | ||
| using System.Threading.Tasks; | ||
|
|
||
| using AICoreLogicLayerClass = AICoreLayerClass<BotLogicDecision>; | ||
|
|
||
| namespace DrakiaXYZ.BigBrain.Internal | ||
| { | ||
| internal static class BrainHelpers | ||
| { | ||
| private static FieldInfo _layerDictionary = null; | ||
|
|
||
| public static Dictionary<int, AICoreLogicLayerClass> GetBrainLayerDictionary(this BaseBrain brain) | ||
| { | ||
| if (_layerDictionary == null) | ||
| { | ||
| Type baseBrainType = typeof(BaseBrain); | ||
| Type aiCoreStrategyType = baseBrainType.BaseType; | ||
|
|
||
| _layerDictionary = AccessTools.Field(aiCoreStrategyType, "dictionary_0"); | ||
| } | ||
|
|
||
| return _layerDictionary.GetValue(brain) as Dictionary<int, AICoreLogicLayerClass>; | ||
| } | ||
|
|
||
| internal static void RemoveAllExcludedLayers(this BotOwner botOwner) | ||
| { | ||
| foreach (BrainManager.ExcludeLayerInfo excludeLayer in BrainManager.Instance.ExcludeLayers) | ||
| { | ||
| botOwner.RemoveLayerForBot(excludeLayer); | ||
| } | ||
| } | ||
|
|
||
| internal static void RemoveLayerForBot(this BotOwner botOwner, BrainManager.ExcludeLayerInfo excludeLayer) | ||
| { | ||
| // Only remove the layer if ExcludeLayer contains the bot's brain name | ||
| if (!excludeLayer.ExcludeLayerBrains.Contains(botOwner.Brain.BaseBrain.ShortName())) | ||
| { | ||
| return; | ||
| } | ||
|
|
||
| botOwner.RemoveLayerForBot(excludeLayer.excludeLayerName); | ||
| } | ||
|
|
||
| internal static void RemoveLayerForBot(this BotOwner botOwner, string layerName) | ||
| { | ||
| // Get all brain layers the bot currently has | ||
| Dictionary<int, AICoreLogicLayerClass> botBrainLayerDictionary = botOwner.Brain.BaseBrain.GetBrainLayerDictionary(); | ||
|
|
||
| int layerIndexToRemove = -1; | ||
|
|
||
| foreach (int index in botBrainLayerDictionary.Keys) | ||
| { | ||
| if (botBrainLayerDictionary[index].Name() != layerName) | ||
| { | ||
| continue; | ||
| } | ||
|
|
||
| // Remove the brain layer from the bot's brain | ||
| layerIndexToRemove = index; | ||
| botOwner.Brain.BaseBrain.method_3(index); | ||
|
|
||
| // Ensure there is no longer a brain layer of the same type in the bot's brain | ||
| if (botOwner.Brain.BaseBrain.method_2(botBrainLayerDictionary[index])) | ||
| { | ||
| throw new InvalidOperationException($"Could not remove brain layer '{layerName}' from {botOwner.name}"); | ||
| } | ||
|
|
||
| // Cache the removed layer so it can be restored later if needed | ||
| BrainManager.Instance.ExcludedLayers.Add(new BrainManager.ExcludedLayerInfo(botOwner, botBrainLayerDictionary[index], botOwner.Brain.BaseBrain.ShortName(), index)); | ||
|
|
||
| break; | ||
| } | ||
|
|
||
| // If a matching layer was found, remove it from the bot's brain-layer dictionary. This is not done in method_3. | ||
| if (layerIndexToRemove > -1) | ||
| { | ||
| botBrainLayerDictionary.Remove(layerIndexToRemove); | ||
| } | ||
| } | ||
|
|
||
| internal static void RestoreLayerForBot(this BotOwner botOwner, string layerName) | ||
| { | ||
| // Get all brain layers the bot currently has | ||
| Dictionary<int, AICoreLogicLayerClass> botBrainLayerDictionary = botOwner.Brain.BaseBrain.GetBrainLayerDictionary(); | ||
|
|
||
| List<BrainManager.ExcludedLayerInfo> restoredLayers = new List<BrainManager.ExcludedLayerInfo>(); | ||
|
|
||
| foreach (BrainManager.ExcludedLayerInfo excludedLayer in BrainManager.Instance.ExcludedLayers) | ||
| { | ||
| if ((excludedLayer.BotOwner != botOwner) || (excludedLayer.LayerName != layerName)) | ||
| { | ||
| continue; | ||
| } | ||
|
|
||
| // Ensure the brain-layer index doesn't already exist in the bot's brain | ||
| if (botBrainLayerDictionary.ContainsKey(excludedLayer.Index)) | ||
| { | ||
| throw new InvalidOperationException($"Cannot restore '{excludedLayer.LayerName}' for {botOwner.name}. Index already exists in bot brain."); | ||
| } | ||
|
|
||
| // Add the brain layer back to the bot's brain | ||
| if (!botOwner.Brain.BaseBrain.method_0(excludedLayer.Index, excludedLayer.Layer, true)) | ||
| { | ||
| throw new InvalidOperationException($"Cannot restore '{excludedLayer.LayerName}' for {botOwner.name}. Failed to add layer."); | ||
| } | ||
|
|
||
| restoredLayers.Add(excludedLayer); | ||
| } | ||
|
|
||
| // After a brain layer has been restored, the cached version of it can be removed | ||
| foreach (BrainManager.ExcludedLayerInfo restoredLayer in restoredLayers) | ||
| { | ||
| BrainManager.Instance.ExcludedLayers.Remove(restoredLayer); | ||
| } | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.