Add stone ability colors

src/main/java/anvil/infinity/items/ItemMindStone.java

@@ -8,6 +8,7 @@ import lucraft.mods.lucraftcore.infinity.ModuleInfinity;
 import lucraft.mods.lucraftcore.infinity.items.ItemInfinityStone;
 import lucraft.mods.lucraftcore.superpowers.abilities.Ability;
 import lucraft.mods.lucraftcore.superpowers.abilities.AbilityFlight;
+import lucraft.mods.lucraftcore.util.abilitybar.EnumAbilityBarColor;
 import lucraft.mods.lucraftcore.util.helper.StringHelper;
 import net.minecraft.entity.EntityLivingBase;
 
@@ -39,6 +40,11 @@ public class ItemMindStone extends ItemInfinityStone {
         if (CompatHandler.isHeroesExpansion && ConfigHandler.heAbilities) {
             abilities.put("telekinesis", CompatHandler.HeroesExpension.getAbilityGrabEntity(entity));
         }
+
+        for (Ability ability : abilities.values()) {
+            ability.setDataValue(Ability.BAR_COLOR, EnumAbilityBarColor.YELLOW);
+        }
+
         return super.addStoneAbilities(entity, abilities, context);
     }
 

src/main/java/anvil/infinity/items/ItemPowerStone.java

@@ -8,6 +8,7 @@ import lucraft.mods.lucraftcore.infinity.ModuleInfinity;
 import lucraft.mods.lucraftcore.infinity.items.ItemInfinityStone;
 import lucraft.mods.lucraftcore.superpowers.abilities.*;
 import lucraft.mods.lucraftcore.superpowers.abilities.supplier.IAbilityProvider;
+import lucraft.mods.lucraftcore.util.abilitybar.EnumAbilityBarColor;
 import lucraft.mods.lucraftcore.util.helper.StringHelper;
 import net.minecraft.entity.EntityLivingBase;
 import net.minecraft.item.ItemStack;
@@ -47,6 +48,11 @@ public class ItemPowerStone extends ItemInfinityStone implements IAbilityProvide
             abilities.put("power_rocket_burst", CompatHandler.StarTech.getAbilityPowerRocketBurst(entity).setMaxCooldown(100));
             abilities.put("power_tendrils", CompatHandler.StarTech.getAbilityTendrils(entity));
         }
+
+        for (Ability ability : abilities.values()) {
+            ability.setDataValue(Ability.BAR_COLOR, EnumAbilityBarColor.PURPLE);
+        }
+
         return super.addStoneAbilities(entity, abilities, context);
     }
 

src/main/java/anvil/infinity/items/ItemRealityStone.java

@@ -8,6 +8,7 @@ import lucraft.mods.lucraftcore.infinity.items.ItemInfinityStone;
 import lucraft.mods.lucraftcore.superpowers.abilities.Ability;
 import lucraft.mods.lucraftcore.superpowers.abilities.AbilityInvisibility;
 import lucraft.mods.lucraftcore.superpowers.abilities.AbilitySizeChange;
+import lucraft.mods.lucraftcore.util.abilitybar.EnumAbilityBarColor;
 import lucraft.mods.lucraftcore.util.helper.StringHelper;
 import net.minecraft.entity.EntityLivingBase;
 
@@ -42,6 +43,11 @@ public class ItemRealityStone extends ItemInfinityStone {
             abilities.put("change_block", CompatHandler.SpeedsterHeroes.getAbilityChangeBlock(entity));
             abilities.put("turn_into_bubbles", CompatHandler.SpeedsterHeroes.getAbilityTurnIntoBubbles(entity));
         }
+
+        for (Ability ability : abilities.values()) {
+            ability.setDataValue(Ability.BAR_COLOR, EnumAbilityBarColor.RED);
+        }
+
         return super.addStoneAbilities(entity, abilities, context);
     }
 

src/main/java/anvil/infinity/items/ItemSoulStone.java

@@ -12,6 +12,7 @@ import lucraft.mods.lucraftcore.superpowers.abilities.Ability;
 import lucraft.mods.lucraftcore.superpowers.abilities.AbilityAttributeModifier;
 import lucraft.mods.lucraftcore.superpowers.abilities.AbilityHealing;
 import lucraft.mods.lucraftcore.superpowers.abilities.AbilityHealth;
+import lucraft.mods.lucraftcore.util.abilitybar.EnumAbilityBarColor;
 import lucraft.mods.lucraftcore.util.helper.StringHelper;
 import net.minecraft.entity.EntityLivingBase;
 
@@ -50,6 +51,11 @@ public class ItemSoulStone extends ItemInfinityStone {
         abilities.put("healing", new AbilityHealing(entity).setDataValue(AbilityHealing.FREQUENCY, 1));
         abilities.put("kill", new AbilityKill(entity, killCond));
         abilities.put("snap", new AbilitySnap(entity));
+
+        for (Ability ability : abilities.values()) {
+            ability.setDataValue(Ability.BAR_COLOR, EnumAbilityBarColor.ORANGE);
+        }
+
         return super.addStoneAbilities(entity, abilities, context);
     }
 

src/main/java/anvil/infinity/items/ItemTimeStone.java

@@ -6,6 +6,7 @@ import lucraft.mods.lucraftcore.infinity.EnumInfinityStone;
 import lucraft.mods.lucraftcore.infinity.ModuleInfinity;
 import lucraft.mods.lucraftcore.infinity.items.ItemInfinityStone;
 import lucraft.mods.lucraftcore.superpowers.abilities.Ability;
+import lucraft.mods.lucraftcore.util.abilitybar.EnumAbilityBarColor;
 import lucraft.mods.lucraftcore.util.helper.StringHelper;
 import net.minecraft.entity.EntityLivingBase;
 
@@ -33,6 +34,11 @@ public class ItemTimeStone extends ItemInfinityStone {
     public Ability.AbilityMap addStoneAbilities(EntityLivingBase entity, Ability.AbilityMap abilities, Ability.EnumAbilityContext context) {
         abilities.put("fast_forward", new AbilityFastForward(entity));
         abilities.put("stop_time", new AbilityStopTime(entity));
+
+        for (Ability ability : abilities.values()) {
+            ability.setDataValue(Ability.BAR_COLOR, EnumAbilityBarColor.GREEN);
+        }
+
         return super.addStoneAbilities(entity, abilities, context);
     }