1 files changed, 57 insertions, 0 deletions
diff --git a/src/main/java/gregtech/loaders/misc/bees/GTAlleleEffect.java b/src/main/java/gregtech/loaders/misc/bees/GTAlleleEffect.java
new file mode 100644
index 0000000000..128e3dfeb1
--- /dev/null
+++ b/src/main/java/gregtech/loaders/misc/bees/GTAlleleEffect.java
@@ -0,0 +1,57 @@
+package gregtech.loaders.misc.bees;
+
+import forestry.api.apiculture.EnumBeeChromosome;
+import forestry.api.apiculture.IAlleleBeeEffect;
+import forestry.api.apiculture.IBeeGenome;
+import forestry.api.apiculture.IBeeHousing;
+import forestry.api.genetics.AlleleManager;
+import forestry.api.genetics.IEffectData;
+import forestry.core.genetics.alleles.Allele;
+
+// helper class for implementing custom bee effects, based on MagicBees' implementation
+public abstract class GTAlleleEffect extends Allele implements IAlleleBeeEffect {
+
+    public static final IAlleleBeeEffect FORESTRY_BASE_EFFECT = (IAlleleBeeEffect) AlleleManager.alleleRegistry
+        .getAllele("forestry.effectNone");
+    protected boolean combinable;
+    protected int tickThrottle; // If set, this amount of ticks have to pass before an effect ticks
+
+    public GTAlleleEffect(String id, boolean isDominant, int tickThrottle) {
+        super("gregtech." + id, "gregtech." + id, isDominant);
+        AlleleManager.alleleRegistry.registerAllele(this, EnumBeeChromosome.EFFECT);
+        combinable = false;
+        this.tickThrottle = tickThrottle;
+    }
+
+    @Override
+    public boolean isCombinable() {
+        return combinable;
+    }
+
+    public GTAlleleEffect setIsCombinable(boolean canCombine) {
+        combinable = canCombine;
+        return this;
+    }
+
+    @Override
+    public abstract IEffectData validateStorage(IEffectData storedData);
+
+    @Override
+    public IEffectData doEffect(IBeeGenome genome, IEffectData storedData, IBeeHousing housing) {
+        int ticksPassed = storedData.getInteger(0);
+        if (ticksPassed >= this.tickThrottle) {
+            storedData = this.doEffectTickThrottled(genome, storedData, housing);
+        } else {
+            storedData.setInteger(0, ticksPassed + 1);
+        }
+        return storedData;
+    }
+
+    protected abstract IEffectData doEffectTickThrottled(IBeeGenome genome, IEffectData storedData,
+        IBeeHousing housing);
+
+    @Override
+    public IEffectData doFX(IBeeGenome genome, IEffectData storedData, IBeeHousing housing) {
+        return FORESTRY_BASE_EFFECT.doFX(genome, storedData, housing);
+    }
+}