#include "../../Platform/stdafx.h"
#include "../../Headers/net.minecraft.world.entity.ai.control.h"
#include "../../Headers/net.minecraft.world.entity.ai.navigation.h"
#include "../../Headers/net.minecraft.world.entity.animal.h"
#include "../../Headers/net.minecraft.world.level.h"
#include "../../Headers/net.minecraft.world.phys.h"
#include "../../Util/BasicTypeContainers.h"
#include "BreedGoal.h"
#include "../../Entities/Mobs/ExperienceOrb.h"

#include "../../Stats/GenericStats.h"

BreedGoal::BreedGoal(Animal* animal, double speedModifier) {
    partner = std::weak_ptr<Animal>();
    loveTime = 0;

    this->animal = animal;
    this->level = animal->level;
    this->speedModifier = speedModifier;
    setRequiredControlFlags(Control::MoveControlFlag |
                            Control::LookControlFlag);
}

bool BreedGoal::canUse() {
    if (!animal->isInLove()) return false;
    partner = std::weak_ptr<Animal>(getFreePartner());
    return partner.lock() != NULL;
}

bool BreedGoal::canContinueToUse() {
    return partner.lock() != NULL && partner.lock()->isAlive() &&
           partner.lock()->isInLove() && loveTime < 20 * 3;
}

void BreedGoal::stop() {
    partner = std::weak_ptr<Animal>();
    loveTime = 0;
}

void BreedGoal::tick() {
    animal->getLookControl()->setLookAt(partner.lock(), 10,
                                        animal->getMaxHeadXRot());
    animal->getNavigation()->moveTo(partner.lock(), speedModifier);
    ++loveTime;
    if (loveTime >= 20 * 3 && animal->distanceToSqr(partner.lock()) < 3 * 3)
        breed();
}

std::shared_ptr<Animal> BreedGoal::getFreePartner() {
    float r = 8;
    std::vector<std::shared_ptr<Entity> >* others =
        level->getEntitiesOfClass(typeid(*animal), animal->bb->grow(r, r, r));
    double dist = Double::MAX_VALUE;
    std::shared_ptr<Animal> partner = nullptr;
    for (AUTO_VAR(it, others->begin()); it != others->end(); ++it) {
        std::shared_ptr<Animal> p = std::dynamic_pointer_cast<Animal>(*it);
        if (animal->canMate(p) && animal->distanceToSqr(p) < dist) {
            partner = p;
            dist = animal->distanceToSqr(p);
        }
    }
    delete others;
    return partner;
}

void BreedGoal::breed() {
    std::shared_ptr<AgableMob> offspring =
        animal->getBreedOffspring(partner.lock());
    animal->setDespawnProtected();
    partner.lock()->setDespawnProtected();
    if (offspring == NULL) {
        // This will be NULL if we've hit our limits for spawning any particular
        // type of animal... reset things as normally as we can, without
        // actually producing any offspring
        animal->resetLove();
        partner.lock()->resetLove();
        return;
    }

    std::shared_ptr<Player> loveCause = animal->getLoveCause();
    if (loveCause == NULL && partner.lock()->getLoveCause() != NULL) {
        loveCause = partner.lock()->getLoveCause();
    }

    if (loveCause != NULL) {
        // Record mob bred stat.
        loveCause->awardStat(
            GenericStats::breedEntity(offspring->GetType()),
            GenericStats::param_breedEntity(offspring->GetType()));

        if (animal->GetType() == eTYPE_COW) {
            // loveCause->awardStat(Achievements.breedCow);
        }
    }

    animal->setAge(5 * 60 * 20);
    partner.lock()->setAge(5 * 60 * 20);
    animal->resetLove();
    partner.lock()->resetLove();
    offspring->setAge(AgableMob::BABY_START_AGE);
    offspring->moveTo(animal->x, animal->y, animal->z, 0, 0);
    offspring->setDespawnProtected();
    level->addEntity(offspring);

    Random* random = animal->getRandom();
    for (int i = 0; i < 7; i++) {
        double xa = random->nextGaussian() * 0.02;
        double ya = random->nextGaussian() * 0.02;
        double za = random->nextGaussian() * 0.02;
        level->addParticle(
            eParticleType_heart,
            animal->x + random->nextFloat() * animal->bbWidth * 2 -
                animal->bbWidth,
            animal->y + .5f + random->nextFloat() * animal->bbHeight,
            animal->z + random->nextFloat() * animal->bbWidth * 2 -
                animal->bbWidth,
            xa, ya, za);
    }
    // 4J-PB - Fix for 106869- Customer Encountered: TU12: Content: Gameplay:
    // Breeding animals does not give any Experience Orbs.
    level->addEntity(std::shared_ptr<ExperienceOrb>(new ExperienceOrb(
        level, animal->x, animal->y, animal->z, random->nextInt(7) + 1)));
}