aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/gregtech/common/bees/GTBeeMutation.java
blob: 871b25f41f5a82754ac735d9b63bbfc860d14caf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package gregtech.common.bees;

import java.lang.reflect.Field;
import java.util.List;

import net.minecraft.util.ChunkCoordinates;
import net.minecraft.world.World;

import org.apache.commons.lang3.reflect.FieldUtils;

import forestry.api.apiculture.BeeManager;
import forestry.api.apiculture.IAlleleBeeSpecies;
import forestry.api.apiculture.IBeeGenome;
import forestry.api.apiculture.IBeeHousing;
import forestry.api.apiculture.IBeeModifier;
import forestry.api.core.IClimateProvider;
import forestry.api.genetics.IAllele;
import forestry.api.genetics.IGenome;
import forestry.api.genetics.IMutationCondition;
import forestry.apiculture.genetics.BeeMutation;
import forestry.core.genetics.mutations.Mutation;

public class GTBeeMutation extends BeeMutation {

    private final float split;

    public GTBeeMutation(IAlleleBeeSpecies bee0, IAlleleBeeSpecies bee1, IAllele[] result, int chance, float split) {
        super(bee0, bee1, result, chance);
        this.split = split;
        BeeManager.beeRoot.registerMutation(this);
    }

    @Override
    public float getBaseChance() {
        return super.getBaseChance() / split;
    }

    @Override
    public float getChance(IBeeHousing housing, IAlleleBeeSpecies allele0, IAlleleBeeSpecies allele1,
        IBeeGenome genome0, IBeeGenome genome1) {
        World world = housing != null ? housing.getWorld() : null;
        ChunkCoordinates housingCoordinates = housing != null ? housing.getCoordinates() : null;
        int x = housingCoordinates != null ? housingCoordinates.posX : 0;
        int y = housingCoordinates != null ? housingCoordinates.posY : 0;
        int z = housingCoordinates != null ? housingCoordinates.posZ : 0;

        float processedChance = getBasicChance(world, x, y, z, allele0, allele1, genome0, genome1, housing);

        if (processedChance <= 0f) {
            return 0f;
        }

        IBeeModifier beeHousingModifier = BeeManager.beeRoot.createBeeHousingModifier(housing);
        IBeeModifier beeModeModifier = BeeManager.beeRoot.getBeekeepingMode(world)
            .getBeeModifier();

        processedChance *= beeHousingModifier.getMutationModifier(genome0, genome1, processedChance);
        processedChance *= beeModeModifier.getMutationModifier(genome0, genome1, processedChance);

        return processedChance;
    }

    @SuppressWarnings("unchecked")
    private float getBasicChance(World world, int x, int y, int z, IAllele allele0, IAllele allele1, IGenome genome0,
        IGenome genome1, IClimateProvider climate) {
        float mutationChance = this.getBaseChance();
        List<IMutationCondition> mutationConditions = null;
        Field f = FieldUtils.getDeclaredField(Mutation.class, "mutationConditions", true);
        if (f == null) f = FieldUtils.getField(Mutation.class, "mutationConditions", true);
        if (f == null) return mutationChance;
        try {
            mutationConditions = f.get(this) instanceof List ? (List<IMutationCondition>) f.get(this) : null;
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }

        if (mutationConditions != null) for (IMutationCondition mutationCondition : mutationConditions) {
            mutationChance *= mutationCondition.getChance(world, x, y, z, allele0, allele1, genome0, genome1, climate);
            if (mutationChance == 0) {
                return 0;
            }
        }
        return mutationChance;
    }
}