Fix collecting aeion items

randovania · Sep 17, 2023 · 384e851 · 384e851
1 parent d40980b
commit 384e851
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 1 deletion.
diff --git a/open_samus_returns_rando/files/randomizer_powerup.lua b/open_samus_returns_rando/files/randomizer_powerup.lua
@@ -300,3 +300,31 @@ function RandomizerBabyHatchling.OnPickedUp(actor, progression)
     RandomizerPowerup.OnPickedUp(actor, progression)
     Game.GetDefaultPlayer("Samus").BABYHATCHLINGCREATION:SpawnBaby()
 end
+
+RandomizerScanningPulse = {}
+setmetatable(RandomizerScanningPulse, {__index = RandomizerPowerup})
+function RandomizerScanningPulse.OnPickedUp(actor, progression)
+    RandomizerPowerup.OnPickedUp(actor, progression)
+    Player.SetAbilityUnlocked("ScanningPulse", true)
+end
+
+RandomizerEnergyShield = {}
+setmetatable(RandomizerEnergyShield, {__index = RandomizerPowerup})
+function RandomizerEnergyShield.OnPickedUp(actor, progression)
+    RandomizerPowerup.OnPickedUp(actor, progression)
+    Player.SetAbilityUnlocked("EnergyShield", true)
+end
+
+RandomizerEnergyWave = {}
+setmetatable(RandomizerEnergyWave, {__index = RandomizerPowerup})
+function RandomizerEnergyWave.OnPickedUp(actor, progression)
+    RandomizerPowerup.OnPickedUp(actor, progression)
+    Player.SetAbilityUnlocked("EnergyWave", true)
+end
+
+RandomizerPhaseDisplacement = {}
+setmetatable(RandomizerPhaseDisplacement, {__index = RandomizerPowerup})
+function RandomizerPhaseDisplacement.OnPickedUp(actor, progression)
+    RandomizerPowerup.OnPickedUp(actor, progression)
+    Player.SetAbilityUnlocked("PhaseDisplacement", true)
+end
diff --git a/open_samus_returns_rando/lua_editor.py b/open_samus_returns_rando/lua_editor.py
@@ -21,7 +21,11 @@ def _read_level_lua(level_id: str) -> str:
     "ITEM_WEAPON_SUPER_MISSILE_MAX": "RandomizerSuperMissile",
     "ITEM_RANDO_LOCKED_SUPERS": "RandomizerSuperMissileTank",
     "ITEM_WEAPON_POWER_BOMB_MAX": "RandomizerPowerBomb",
-    "ITEM_RANDO_LOCKED_PBS": "RandomizerPowerBombTank"
+    "ITEM_RANDO_LOCKED_PBS": "RandomizerPowerBombTank",
+    "ITEM_SPECIAL_ENERGY_SCANNING_PULSE": "RandomizerScanningPulse",
+    "ITEM_SPECIAL_ENERGY_ENERGY_SHIELD": "RandomizerEnergyShield",
+    "ITEM_SPECIAL_ENERGY_ENERGY_WAVE": "RandomizerEnergyWave",
+    "ITEM_SPECIAL_ENERGY_PHASE_DISPLACEMENT": "RandomizerPhaseDisplacement",
 }