1 files changed, 108 insertions, 0 deletions
diff --git a/src/main/java/gregtech/api/util/GT_ApiaryUpgrade.java b/src/main/java/gregtech/api/util/GT_ApiaryUpgrade.java
new file mode 100644
index 0000000000..4c60e6f8bc
--- /dev/null
+++ b/src/main/java/gregtech/api/util/GT_ApiaryUpgrade.java
@@ -0,0 +1,108 @@
+package gregtech.api.util;
+
+import cpw.mods.fml.common.Loader;
+import gregtech.api.enums.OrePrefixes;
+import gregtech.common.items.GT_MetaGenerated_Item_03;
+import net.bdew.gendustry.api.ApiaryModifiers;
+import net.bdew.gendustry.api.items.IApiaryUpgrade;
+import net.minecraft.item.ItemStack;
+
+import java.util.*;
+import java.util.function.Consumer;
+
+public enum GT_ApiaryUpgrade {
+    speed1(UNIQUE_INDEX.SPEED_UPGRADE, 32200, 1, 1),
+    speed2(UNIQUE_INDEX.SPEED_UPGRADE, 32201, 1, 2),
+    speed3(UNIQUE_INDEX.SPEED_UPGRADE, 32202, 1, 3),
+    speed4(UNIQUE_INDEX.SPEED_UPGRADE, 32203, 1, 4),
+    speed5(UNIQUE_INDEX.SPEED_UPGRADE, 32204, 1, 5),
+    speed6(UNIQUE_INDEX.SPEED_UPGRADE, 32205, 1, 6),
+    speed7(UNIQUE_INDEX.SPEED_UPGRADE, 32206, 1, 7),
+    speed8(UNIQUE_INDEX.SPEED_UPGRADE, 32207, 1, 8),
+    ;
+
+    private enum UNIQUE_INDEX{
+        SPEED_UPGRADE,
+        ;
+        void apply(Consumer<GT_ApiaryUpgrade> fn){
+            UNIQUE_UPGRADE_LIST.get(this).forEach(fn);
+        }
+    }
+
+    private static final EnumMap<UNIQUE_INDEX, ArrayList<GT_ApiaryUpgrade>> UNIQUE_UPGRADE_LIST = new EnumMap<>(UNIQUE_INDEX.class);
+
+    private int meta = 0;
+    private int maxnumber = 1;
+    private int maxspeedmodifier = 0; // formula: maxspeed = modifier
+
+    private final GT_Utility.ItemId id;
+    private final UNIQUE_INDEX unique_index;
+
+    private final HashMap<GT_Utility.ItemId, ItemStack> additionalGendustryUpgrades = new HashMap<>();
+    private final HashSet<GT_Utility.ItemId> blacklistedUpgrades = new HashSet<>(); // additionalGendustryUpgrades are blacklisted by default
+
+    GT_ApiaryUpgrade(UNIQUE_INDEX unique_index, int meta, int maxnumber, int maxspeedmodifier){
+        this.unique_index = unique_index;
+        this.meta = meta;
+        this.maxnumber = maxnumber;
+        this.maxspeedmodifier = maxspeedmodifier;
+        this.id = GT_Utility.ItemId.createNoCopy(get(1));
+    }
+
+    private void setup_static_variables(){
+        quickLookup.put(this.meta, this);
+        ArrayList<GT_ApiaryUpgrade> un = UNIQUE_UPGRADE_LIST.get(this.unique_index);
+        if(un != null)
+            un.forEach((u) -> { u.blacklistedUpgrades.add(this.id); this.blacklistedUpgrades.add(u.id); });
+        else {
+            un = new ArrayList<>(1);
+            UNIQUE_UPGRADE_LIST.put(this.unique_index, un);
+        }
+        un.add(this);
+    }
+
+    public static GT_ApiaryUpgrade getUpgrade(ItemStack s){
+        if(s == null)
+            return null;
+        if(!isUpgrade(s))
+            return null;
+        return quickLookup.get(s.getItemDamage());
+    }
+
+    public boolean isAllowedToWorkWith(ItemStack s){
+        GT_Utility.ItemId id = GT_Utility.ItemId.createNoCopy(s);
+        return !additionalGendustryUpgrades.containsKey(id) && !blacklistedUpgrades.contains(id);
+    }
+
+    public int getMaxNumber(){
+        return maxnumber;
+    }
+
+    public void applyModifiers(ApiaryModifiers mods, ItemStack s){
+        additionalGendustryUpgrades.forEach((k, u) -> ((IApiaryUpgrade)u.getItem()).applyModifiers(mods, u));
+    }
+
+    public ItemStack get(int count){
+        return new ItemStack(GT_MetaGenerated_Item_03.INSTANCE, count, meta);
+    }
+
+    public static boolean isUpgrade(ItemStack s){
+        return OrePrefixes.apiaryUpgrade.contains(s);
+    }
+
+    public int applyMaxSpeedModifier(int maxspeed){
+        return Math.max(maxspeed, maxspeedmodifier);
+    }
+
+    private static final HashMap<Integer, GT_ApiaryUpgrade> quickLookup = new HashMap<>();
+
+    static{
+        EnumSet.allOf(GT_ApiaryUpgrade.class).forEach(GT_ApiaryUpgrade::setup_static_variables);
+
+        if(Loader.isModLoaded("gendustry")) {
+            ItemStack s = GT_ModHandler.getModItem("gendustry", "ApiaryUpgrade", 8L, 0);
+            GT_Utility.ItemId a = GT_Utility.ItemId.createNoCopy(s);
+            UNIQUE_INDEX.SPEED_UPGRADE.apply((u) -> u.additionalGendustryUpgrades.put(a, s));
+        }
+    }
+}