diff --git a/src/main/java/org/saturnclient/modules/mods/ArmorDisplay.java b/src/main/java/org/saturnclient/modules/mods/ArmorDisplay.java index a9d26e7..87458fb 100644 --- a/src/main/java/org/saturnclient/modules/mods/ArmorDisplay.java +++ b/src/main/java/org/saturnclient/modules/mods/ArmorDisplay.java @@ -14,22 +14,31 @@ import net.minecraft.item.Items; public class ArmorDisplay extends Module implements HudMod { private static Property enabled = Property.bool(false); - private static ModDimensions dimensions = new ModDimensions(40, 60); + private static ModDimensions dimensions = new ModDimensions(40, 75); + private static Property useMainHand = Property.bool(false); public ArmorDisplay() { super(new ModuleDetails("Armor Display", "armor") - .description("Displays armor durability") - .version("v0.1.0") - .tags("Utility"), - enabled.named("Enabled"), - dimensions.prop()); + .description("Displays armor durability") + .version("v0.1.0") + .tags("Utility"), + enabled.named("Enabled"), + useMainHand.named("Use Main Hand"), + dimensions.prop()); } - public void renderArmor(RenderScope scope, ItemStack helmet, ItemStack chestplate, ItemStack leggings, ItemStack boots) { + public void renderArmor(RenderScope scope, ItemStack mainHand, ItemStack helmet, ItemStack chestplate, + ItemStack leggings, ItemStack boots) { int row = 0; + if (useMainHand.value && !mainHand.isEmpty()) { + scope.drawItem(mainHand, 0, 15 * row); + renderHealth(scope, row, mainHand.getMaxDamage(), mainHand.getDamage()); + row++; + } + if (!helmet.isEmpty()) { - scope.drawItem(helmet, 0, 0); + scope.drawItem(helmet, 0, 15 * row); renderHealth(scope, row, helmet.getMaxDamage(), helmet.getDamage()); row++; } @@ -53,23 +62,27 @@ public class ArmorDisplay extends Module implements HudMod { } public void renderHealth(RenderScope scope, int i, int max_damage, int damage) { - if (max_damage > 0 ) { - scope.drawText(0.5f, "" + (max_damage - damage), 17, (15 * i) + 3, dimensions.font.value, dimensions.fgColor.value); + if (max_damage > 0) { + scope.drawText(0.5f, "" + (max_damage - damage), 17, (15 * i) + 3, dimensions.font.value, + dimensions.fgColor.value); } } @Override public void renderHud(RenderScope scope) { renderArmor(scope, - SaturnClient.client.player.getEquippedStack(EquipmentSlot.HEAD), - SaturnClient.client.player.getEquippedStack(EquipmentSlot.CHEST), - SaturnClient.client.player.getEquippedStack(EquipmentSlot.LEGS), - SaturnClient.client.player.getEquippedStack(EquipmentSlot.FEET)); + SaturnClient.client.player.getEquippedStack(EquipmentSlot.MAINHAND), + SaturnClient.client.player.getEquippedStack(EquipmentSlot.HEAD), + SaturnClient.client.player.getEquippedStack(EquipmentSlot.CHEST), + SaturnClient.client.player.getEquippedStack(EquipmentSlot.LEGS), + SaturnClient.client.player.getEquippedStack(EquipmentSlot.FEET)); } @Override public void renderDummy(RenderScope scope) { - renderArmor(scope, new ItemStack(Items.DIAMOND_HELMET), new ItemStack(Items.DIAMOND_CHESTPLATE), new ItemStack(Items.DIAMOND_LEGGINGS), new ItemStack(Items.DIAMOND_BOOTS)); + renderArmor(scope, new ItemStack(Items.DIAMOND_SWORD), new ItemStack(Items.DIAMOND_HELMET), + new ItemStack(Items.DIAMOND_CHESTPLATE), + new ItemStack(Items.DIAMOND_LEGGINGS), new ItemStack(Items.DIAMOND_BOOTS)); } @Override