001package net.minecraft.entity.ai;
002
003import java.util.Iterator;
004import java.util.List;
005import java.util.Random;
006import net.minecraft.entity.EntityAgeable;
007import net.minecraft.entity.item.EntityXPOrb;
008import net.minecraft.entity.passive.EntityAnimal;
009import net.minecraft.world.World;
010
011public class EntityAIMate extends EntityAIBase
012{
013    private EntityAnimal theAnimal;
014    World theWorld;
015    private EntityAnimal targetMate;
016
017    /**
018     * Delay preventing a baby from spawning immediately when two mate-able animals find each other.
019     */
020    int spawnBabyDelay = 0;
021
022    /** The speed the creature moves at during mating behavior. */
023    float moveSpeed;
024
025    public EntityAIMate(EntityAnimal par1EntityAnimal, float par2)
026    {
027        this.theAnimal = par1EntityAnimal;
028        this.theWorld = par1EntityAnimal.worldObj;
029        this.moveSpeed = par2;
030        this.setMutexBits(3);
031    }
032
033    /**
034     * Returns whether the EntityAIBase should begin execution.
035     */
036    public boolean shouldExecute()
037    {
038        if (!this.theAnimal.isInLove())
039        {
040            return false;
041        }
042        else
043        {
044            this.targetMate = this.getNearbyMate();
045            return this.targetMate != null;
046        }
047    }
048
049    /**
050     * Returns whether an in-progress EntityAIBase should continue executing
051     */
052    public boolean continueExecuting()
053    {
054        return this.targetMate.isEntityAlive() && this.targetMate.isInLove() && this.spawnBabyDelay < 60;
055    }
056
057    /**
058     * Resets the task
059     */
060    public void resetTask()
061    {
062        this.targetMate = null;
063        this.spawnBabyDelay = 0;
064    }
065
066    /**
067     * Updates the task
068     */
069    public void updateTask()
070    {
071        this.theAnimal.getLookHelper().setLookPositionWithEntity(this.targetMate, 10.0F, (float)this.theAnimal.getVerticalFaceSpeed());
072        this.theAnimal.getNavigator().tryMoveToEntityLiving(this.targetMate, this.moveSpeed);
073        ++this.spawnBabyDelay;
074
075        if (this.spawnBabyDelay >= 60 && this.theAnimal.getDistanceSqToEntity(this.targetMate) < 9.0D)
076        {
077            this.spawnBaby();
078        }
079    }
080
081    /**
082     * Loops through nearby animals and finds another animal of the same type that can be mated with. Returns the first
083     * valid mate found.
084     */
085    private EntityAnimal getNearbyMate()
086    {
087        float f = 8.0F;
088        List list = this.theWorld.getEntitiesWithinAABB(this.theAnimal.getClass(), this.theAnimal.boundingBox.expand((double)f, (double)f, (double)f));
089        double d0 = Double.MAX_VALUE;
090        EntityAnimal entityanimal = null;
091        Iterator iterator = list.iterator();
092
093        while (iterator.hasNext())
094        {
095            EntityAnimal entityanimal1 = (EntityAnimal)iterator.next();
096
097            if (this.theAnimal.canMateWith(entityanimal1) && this.theAnimal.getDistanceSqToEntity(entityanimal1) < d0)
098            {
099                entityanimal = entityanimal1;
100                d0 = this.theAnimal.getDistanceSqToEntity(entityanimal1);
101            }
102        }
103
104        return entityanimal;
105    }
106
107    /**
108     * Spawns a baby animal of the same type.
109     */
110    private void spawnBaby()
111    {
112        EntityAgeable entityageable = this.theAnimal.createChild(this.targetMate);
113
114        if (entityageable != null)
115        {
116            this.theAnimal.setGrowingAge(6000);
117            this.targetMate.setGrowingAge(6000);
118            this.theAnimal.resetInLove();
119            this.targetMate.resetInLove();
120            entityageable.setGrowingAge(-24000);
121            entityageable.setLocationAndAngles(this.theAnimal.posX, this.theAnimal.posY, this.theAnimal.posZ, 0.0F, 0.0F);
122            this.theWorld.spawnEntityInWorld(entityageable);
123            Random random = this.theAnimal.getRNG();
124
125            for (int i = 0; i < 7; ++i)
126            {
127                double d0 = random.nextGaussian() * 0.02D;
128                double d1 = random.nextGaussian() * 0.02D;
129                double d2 = random.nextGaussian() * 0.02D;
130                this.theWorld.spawnParticle("heart", this.theAnimal.posX + (double)(random.nextFloat() * this.theAnimal.width * 2.0F) - (double)this.theAnimal.width, this.theAnimal.posY + 0.5D + (double)(random.nextFloat() * this.theAnimal.height), this.theAnimal.posZ + (double)(random.nextFloat() * this.theAnimal.width * 2.0F) - (double)this.theAnimal.width, d0, d1, d2);
131            }
132
133            this.theWorld.spawnEntityInWorld(new EntityXPOrb(this.theWorld, this.theAnimal.posX, this.theAnimal.posY, this.theAnimal.posZ, random.nextInt(7) + 1));
134        }
135    }
136}