From 53d88e23f578ac4bf25ef6e7b8aedb8af4538c05 Mon Sep 17 00:00:00 2001
From: Rohan Vazarkar <rvazarkr@gmail.com>
Date: Thu, 5 Nov 2020 13:01:45 -0500
Subject: [PATCH] Fix CAPALLBT, delay rebirth after harvesting ygg

---
 .../AllocationProfiles/CustomAllocation.cs       | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

diff --git a/NGUInjector/AllocationProfiles/CustomAllocation.cs b/NGUInjector/AllocationProfiles/CustomAllocation.cs
index bbb3aca..bcd1c69 100644
--- a/NGUInjector/AllocationProfiles/CustomAllocation.cs
+++ b/NGUInjector/AllocationProfiles/CustomAllocation.cs
@@ -92,7 +92,7 @@ private List<string> ValidatePriorities(List<string> priorities)
             }
 
             priorities.RemoveAll(x => x.Contains("AUG") && !IsAUGUnlocked(ParseIndex(x)));
-            priorities.RemoveAll(x => x.Contains("BT") && !IsBTUnlocked(ParseIndex(x)));
+            priorities.RemoveAll(x => x.Contains("BT") && !x.Contains("ALL") && !IsBTUnlocked(ParseIndex(x)));
             priorities.RemoveAll(x => x.Contains("AT") && ATTargetMet(ParseIndex(x)));
 
             return priorities;
@@ -398,10 +398,8 @@ public void DoRebirth()
                 }
 
                 YggdrasilManager.HarvestAll();
-                LoadoutManager.RestoreGear();
-                LoadoutManager.ReleaseLock();
-                DiggerManager.RestoreDiggers();
-                DiggerManager.ReleaseLock();
+                Main.Log("Delaying rebirth 1 loop to allow fruit effects");
+                return;
             }
 
             if (Main.Settings.CastBloodSpells)
@@ -1073,7 +1071,7 @@ private bool ReadMagicBreakpoint(string breakpoint)
             if (breakpoint.StartsWith("TM"))
             {
                 if (_character.machine.multiTarget > 0 &&
-                    _character.machine.goldMultiLevel >= _character.machine.multiTarget)
+                    _character.machine.levelGoldMulti >= _character.machine.multiTarget)
                 {
                     Main.LogAllocation($"Skipping {breakpoint} because target is met");
                     return true;
@@ -1088,7 +1086,7 @@ private bool ReadMagicBreakpoint(string breakpoint)
             if (breakpoint.StartsWith("CAPTM"))
             {
                 if (_character.machine.multiTarget > 0 &&
-                    _character.machine.goldMultiLevel >= _character.machine.multiTarget)
+                    _character.machine.levelGoldMulti >= _character.machine.multiTarget)
                 {
                     Main.LogAllocation($"Skipping {breakpoint} because target is met");
                     return true;
@@ -1310,7 +1308,7 @@ private bool ReadEnergyBreakpoint(string breakpoint)
             if (breakpoint.StartsWith("TM"))
             {
                 if (_character.machine.speedTarget > 0 &&
-                    _character.machine.speedLevel >= _character.machine.speedTarget)
+                    _character.machine.levelSpeed >= _character.machine.speedTarget)
                 {
                     Main.LogAllocation($"Skipping {breakpoint} because target is met");
                     return true;
@@ -1323,7 +1321,7 @@ private bool ReadEnergyBreakpoint(string breakpoint)
 
             if (breakpoint.StartsWith("CAPTM"))
             {
-                if (_character.machine.speedTarget > 0 && _character.machine.speedLevel >= _character.machine.speedTarget)
+                if (_character.machine.speedTarget > 0 && _character.machine.levelSpeed >= _character.machine.speedTarget)
                     return true;
                 var cap = _calcs.CalculateTMEnergyCap(false);
                 SetInput(cap);