#include "stdafx.h"
#include "net.minecraft.world.entity.ai.control.h"
#include "net.minecraft.world.entity.h"
#include "LeapAtTargetGoal.h"
#include "Player.h"

LeapAtTargetGoal::LeapAtTargetGoal(Mob *mob, float yd)
{
	target = weak_ptr<LivingEntity>();

	this->mob = mob;
	this->yd = yd;
	setRequiredControlFlags(Control::JumpControlFlag | Control::MoveControlFlag);
}

bool LeapAtTargetGoal::canUse()
{
	target = weak_ptr<LivingEntity>(mob->getTarget());
	if (target.lock() == nullptr) return false;
	double d = mob->distanceToSqr(target.lock());
	if (d < 2 * 2 || d > 4 * 4) return false;
	if (!mob->onGround) return false;
	if (mob->getRandom()->nextInt(5) != 0) return false;
	return true;
}

bool LeapAtTargetGoal::canContinueToUse()
{
    shared_ptr<LivingEntity> mobTarget = target.lock();
    if (mobTarget == nullptr) return false;
    if (!mobTarget->isAlive()) return false;
    
    if (mobTarget->instanceof(eTYPE_PLAYER)) {
        shared_ptr<Player> player = dynamic_pointer_cast<Player>(mobTarget);
        if (player->abilities.invulnerable) return false;
    }
    
	return !mob->onGround;
}

void LeapAtTargetGoal::start()
{
	// TODO: move to control?
	double xdd = target.lock()->x - mob->x;
	double zdd = target.lock()->z - mob->z;
	float dd = sqrt(xdd * xdd + zdd * zdd);
	mob->xd += (xdd / dd * 0.5f) * 0.8f + mob->xd * 0.2f;
	mob->zd += (zdd / dd * 0.5f) * 0.8f + mob->zd * 0.2f;
	mob->yd = yd;
}