Tweaked loot fixes

BlackJar72 · Dec 18, 2018 · 4f3ccdb · 4f3ccdb
1 parent 14dd1e8
commit 4f3ccdb
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/java/jaredbgreat/dldungeons/pieces/chests/LootCategory.java b/java/jaredbgreat/dldungeons/pieces/chests/LootCategory.java
@@ -96,15 +96,13 @@ public static ItemStack getLoot(LootType type, int level, Random random) {
 		if(level < 0) level = 0;
 		switch(type) {
 		case GEAR:
-			if(level > 6) level = 6;
 			if(random.nextBoolean()) {
 				return getEnchantedGear(level, random);
 			} else {
-				return gear.levels[level].getLoot(random).getStack(random);
+				return gear.levels[Math.min(6, level)].getLoot(random).getStack(random);
 			}
 		case HEAL:
-			if(level > 6) level = 6;
-			return heal.levels[level].getLoot(random).getStack(random);
+			return heal.levels[Math.min(6, level)].getLoot(random).getStack(random);
 		case LOOT:
 			if(level > 6) {
 				if(level > random.nextInt(100)) {
@@ -152,7 +150,7 @@ private static ItemStack getEnchantedGear(int lootLevel, Random random) {
 			out = item.getStack(random);
 			out = EnchantmentHelper.addRandomEnchantment(random, out, enchPart, true);
 		} else {
-			out = gear.levels[lootLevel].getLoot(random).getStack(random);
+			out = gear.levels[Math.min(6, lootLevel)].getLoot(random).getStack(random);
 		}		
 		return out;
 	}