src/main/java/com/keuin/ohmyvanillamc/mixins/rule/disableFishSchooling/SchoolingFishEntityMixin.java


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

package com.keuin.ohmyvanillamc.mixins.rule.disableFishSchooling;

import com.keuin.ohmyvanillamc.OmvmSettings;
import net.minecraft.entity.EntityType;
import net.minecraft.entity.ai.goal.FollowGroupLeaderGoal;
import net.minecraft.entity.passive.FishEntity;
import net.minecraft.entity.passive.SchoolingFishEntity;
import net.minecraft.world.World;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Overwrite;
import org.spongepowered.asm.mixin.Shadow;

@Mixin(SchoolingFishEntity.class)
public abstract class SchoolingFishEntityMixin extends FishEntity {

    public SchoolingFishEntityMixin(EntityType<? extends FishEntity> type, World world) {
        super(type, world);
    }

    @Shadow
    public abstract boolean hasLeader();

    @Shadow
    private SchoolingFishEntity leader;

    /**
     * @reason To disable SchoolingFish schooling.
     * @author trueKeuin
     */
    @Overwrite
    public void moveTowardLeader() {
        if (!OmvmSettings.disableFishSchooling) {
            if (this.hasLeader()) {
                this.getNavigation().startMovingTo(this.leader, 1.0D);
            }
        }
    }

    /**
     * @reason To disable SchoolingFish schooling.
     * @author trueKeuin
     */
    @Overwrite
    public void initGoals() {
        super.initGoals();
        if (!OmvmSettings.disableFishSchooling) {
            this.goalSelector.add(5, new FollowGroupLeaderGoal((SchoolingFishEntity) (Object) this));
        }
    }
}