#include "../../Platform/stdafx.h"
#include "HorseRenderer.h"
#include "MobRenderer.h"
#include "EntityRenderDispatcher.h"
#include "../../../Minecraft.World/Headers/net.minecraft.world.entity.animal.h"

ResourceLocation HorseRenderer::HORSE_LOCATION =
    ResourceLocation(TN_MOB_HORSE_WHITE);
ResourceLocation HorseRenderer::HORSE_MULE_LOCATION =
    ResourceLocation(TN_MOB_MULE);
ResourceLocation HorseRenderer::HORSE_DONKEY_LOCATION =
    ResourceLocation(TN_MOB_DONKEY);
ResourceLocation HorseRenderer::HORSE_ZOMBIE_LOCATION =
    ResourceLocation(TN_MOB_HORSE_ZOMBIE);
ResourceLocation HorseRenderer::HORSE_SKELETON_LOCATION =
    ResourceLocation(TN_MOB_HORSE_SKELETON);

std::map<std::wstring, ResourceLocation*> HorseRenderer::LAYERED_LOCATION_CACHE;

HorseRenderer::HorseRenderer(Model* model, float f) : MobRenderer(model, f) {}

void HorseRenderer::adjustHeight(std::shared_ptr<PathfinderMob> mob,
                                 float FHeight) {
    glTranslatef(0.0F, FHeight, 0.0F);
}

void HorseRenderer::scale(std::shared_ptr<LivingEntity> entityliving, float f) {
    float sizeFactor = 1.0f;

    int type = std::dynamic_pointer_cast<EntityHorse>(entityliving)->getType();
    if (type == EntityHorse::TYPE_DONKEY) {
        sizeFactor *= 0.87F;
    } else if (type == EntityHorse::TYPE_MULE) {
        sizeFactor *= 0.92F;
    }
    glScalef(sizeFactor, sizeFactor, sizeFactor);
    MobRenderer::scale(entityliving, f);
}

void HorseRenderer::renderModel(std::shared_ptr<LivingEntity> mob, float wp,
                                float ws, float bob, float headRotMinusBodyRot,
                                float headRotx, float scale) {
    if (mob->isInvisible()) {
        model->setupAnim(wp, ws, bob, headRotMinusBodyRot, headRotx, scale,
                         mob);
    } else {
        EntityRenderer::bindTexture(mob);
        model->render(mob, wp, ws, bob, headRotMinusBodyRot, headRotx, scale,
                      true);
        // Ensure that any extra layers of texturing are disabled after
        // rendering this horse
        RenderManager.TextureBind(-1);
    }
}

void HorseRenderer::bindTexture(ResourceLocation* location) {
    // Set up (potentially) multiple texture layers for the horse
    entityRenderDispatcher->textures->bindTextureLayers(location);
}

ResourceLocation* HorseRenderer::getTextureLocation(
    std::shared_ptr<Entity> entity) {
    std::shared_ptr<EntityHorse> horse =
        std::dynamic_pointer_cast<EntityHorse>(entity);

    if (!horse->hasLayeredTextures()) {
        switch (horse->getType()) {
            default:
            case EntityHorse::TYPE_HORSE:
                return &HORSE_LOCATION;
            case EntityHorse::TYPE_MULE:
                return &HORSE_MULE_LOCATION;
            case EntityHorse::TYPE_DONKEY:
                return &HORSE_DONKEY_LOCATION;
            case EntityHorse::TYPE_UNDEAD:
                return &HORSE_ZOMBIE_LOCATION;
            case EntityHorse::TYPE_SKELETON:
                return &HORSE_SKELETON_LOCATION;
        }
    }

    return getOrCreateLayeredTextureLocation(horse);
}

ResourceLocation* HorseRenderer::getOrCreateLayeredTextureLocation(
    std::shared_ptr<EntityHorse> horse) {
    std::wstring textureName = horse->getLayeredTextureHashName();

    AUTO_VAR(it, LAYERED_LOCATION_CACHE.find(textureName));

    ResourceLocation* location;
    if (it != LAYERED_LOCATION_CACHE.end()) {
        location = it->second;
    } else {
        LAYERED_LOCATION_CACHE[textureName] =
            new ResourceLocation(horse->getLayeredTextureLayers());

        it = LAYERED_LOCATION_CACHE.find(textureName);
        location = it->second;
    }

    return location;
}