package net.minecraft.world.entity.ai.behavior;

import com.google.common.collect.ImmutableMap;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.util.Mth;
import net.minecraft.util.RandomSource;
import net.minecraft.util.valueproviders.IntProvider;
import net.minecraft.world.entity.Mob;
import net.minecraft.world.entity.ai.memory.MemoryModuleType;
import net.minecraft.world.entity.ai.memory.MemoryStatus;
import net.minecraft.world.phys.Vec3;

public class RandomLookAround extends Behavior<Mob> {
	private final IntProvider interval;
	private final float maxYaw;
	private final float minPitch;
	private final float pitchRange;

	public RandomLookAround(final IntProvider interval, final float maxYaw, final float minPitch, final float maxPitch) {
		super(ImmutableMap.of(MemoryModuleType.LOOK_TARGET, MemoryStatus.VALUE_ABSENT, MemoryModuleType.GAZE_COOLDOWN_TICKS, MemoryStatus.VALUE_ABSENT));
		if (minPitch > maxPitch) {
			throw new IllegalArgumentException("Minimum pitch is larger than maximum pitch! " + minPitch + " > " + maxPitch);
		} else {
			this.interval = interval;
			this.maxYaw = maxYaw;
			this.minPitch = minPitch;
			this.pitchRange = maxPitch - minPitch;
		}
	}

	protected void start(final ServerLevel level, final Mob body, final long timestamp) {
		RandomSource random = body.getRandom();
		float pitch = Mth.clamp(random.nextFloat() * this.pitchRange + this.minPitch, -90.0F, 90.0F);
		float rotation = Mth.wrapDegrees(body.getYRot() + 2.0F * random.nextFloat() * this.maxYaw - this.maxYaw);
		Vec3 newLookVec = Vec3.directionFromRotation(pitch, rotation);
		body.getBrain().setMemory(MemoryModuleType.LOOK_TARGET, new BlockPosTracker(body.getEyePosition().add(newLookVec)));
		body.getBrain().setMemory(MemoryModuleType.GAZE_COOLDOWN_TICKS, this.interval.sample(random));
	}
}