cn.nukkit.entity.ai.evaluator.ConditionalProbabilityEvaluator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of powernukkitx Show documentation

Make Nukkit great again!

There is a newer version: 1.20.40-r1

package cn.nukkit.entity.ai.evaluator;

import cn.nukkit.api.PowerNukkitXOnly;
import cn.nukkit.api.Since;
import cn.nukkit.entity.Entity;
import cn.nukkit.entity.EntityIntelligent;

import java.util.concurrent.ThreadLocalRandom;
import java.util.function.Function;

@PowerNukkitXOnly
@Since("1.19.30-r1")
public class ConditionalProbabilityEvaluator extends ProbabilityEvaluator {

    int probability2;
    Function condition;

    public ConditionalProbabilityEvaluator(int probability1, int probability2, Function condition, int total) {
        super(probability1, total);
        this.condition = condition;
        this.probability2 = probability2;
    }

    @Override
    public boolean evaluate(EntityIntelligent entity) {
        if (condition.apply(entity)) {
            return ThreadLocalRandom.current().nextInt(total) < probability2;
        } else return ThreadLocalRandom.current().nextInt(total) < probability;
    }
}