diff --git a/src/main/java/com/watschman/beeriffic/bees/EnumSpecies.java b/src/main/java/com/watschman/beeriffic/bees/EnumSpecies.java index 749d37f..73ebdc0 100644 --- a/src/main/java/com/watschman/beeriffic/bees/EnumSpecies.java +++ b/src/main/java/com/watschman/beeriffic/bees/EnumSpecies.java @@ -5,9 +5,12 @@ import forestry.api.core.EnumHumidity; import forestry.api.core.EnumTemperature; import forestry.api.genetics.*; +import net.minecraft.item.ItemStack; + +import java.util.Map; public enum EnumSpecies { - STONY("Stony", "Saxum", EnumBranches.STONUM, 0xffffff, 0x7a7a7a, EnumTemperature.COLD, EnumHumidity.NORMAL, false, true, false, true, false); + STONY("Stony", "Saxum", EnumBranches.STONUM, 0xffffff, 0x7a7a7a, EnumTemperature.COLD, EnumHumidity.NORMAL, false, true, false, true, false, null, null); private final EnumTemperature temperature; private final EnumHumidity humidity; @@ -28,7 +31,9 @@ boolean doesHaveEffect, boolean isDominant, boolean isCounted, - boolean isSecret + boolean isSecret, + Map products, + Map specialties ) { this.temperature = lovedTemperature; this.humidity = lovedHumidity; @@ -54,6 +59,17 @@ if(isNocturnal) { this.beeSpecies.setNocturnal(); } + if(products != null) { + for(Map.Entry entry : products.entrySet()) { + this.beeSpecies.addProduct(entry.getKey(), entry.getValue()); + } + } + + if(specialties != null) { + for(Map.Entry entry : specialties.entrySet()) { + this.beeSpecies.addSpecialty(entry.getKey(), entry.getValue()); + } + } this.beeSpecies.build(); }