MineLittlePony/src/main/java/com/minelittlepony/renderer/layer/LayerPonyElytra.java

package com.minelittlepony.renderer.layer;

import com.minelittlepony.ducks.IRenderPony;
import com.minelittlepony.model.AbstractPonyModel;
import com.minelittlepony.model.BodyPart;
import com.minelittlepony.model.ModelPonyElytra;
import com.minelittlepony.model.pony.ModelHumanPlayer;
import net.minecraft.client.entity.AbstractClientPlayer;
import net.minecraft.client.renderer.GlStateManager;
import net.minecraft.client.renderer.entity.RenderPlayer;
import net.minecraft.client.renderer.entity.layers.LayerArmorBase;
import net.minecraft.client.renderer.entity.layers.LayerElytra;
import net.minecraft.client.renderer.entity.layers.LayerRenderer;
import net.minecraft.entity.player.EnumPlayerModelParts;
import net.minecraft.init.Items;
import net.minecraft.inventory.EntityEquipmentSlot;
import net.minecraft.item.ItemStack;
import net.minecraft.util.ResourceLocation;

public class LayerPonyElytra implements LayerRenderer<AbstractClientPlayer> {

    private static final ResourceLocation TEXTURE_ELYTRA = new ResourceLocation("textures/entity/elytra.png");
    private RenderPlayer renderPlayer;
    private ModelPonyElytra modelElytra = new ModelPonyElytra();

    private LayerElytra elytra;

    public LayerPonyElytra(RenderPlayer rp) {
        this.renderPlayer = rp;
        this.elytra = new LayerElytra(rp);
    }

    @Override
    public void doRenderLayer(AbstractClientPlayer entity, float swing, float swingAmount, float ticks, float age, float yaw, float head, float scale) {

        AbstractPonyModel model = ((IRenderPony) this.renderPlayer).getPony().getModel();
        if (model instanceof ModelHumanPlayer) {
            this.elytra.doRenderLayer(entity, swing, swingAmount, ticks, age, yaw, head, scale);
            return;
        }

        ItemStack itemstack = entity.getItemStackFromSlot(EntityEquipmentSlot.CHEST);

        if (itemstack != null && itemstack.getItem() == Items.ELYTRA) {
            GlStateManager.color(1.0F, 1.0F, 1.0F, 1.0F);

            if (entity.isPlayerInfoSet() && entity.getLocationElytra() != null) {
                this.renderPlayer.bindTexture(entity.getLocationElytra());
            } else if (entity.hasPlayerInfo() && entity.getLocationCape() != null && entity.isWearing(EnumPlayerModelParts.CAPE)) {
                this.renderPlayer.bindTexture(entity.getLocationCape());
            } else {
                this.renderPlayer.bindTexture(TEXTURE_ELYTRA);
            }

            GlStateManager.pushMatrix();
            GlStateManager.translate(0.0F, 0.25F, 0.125F);
            model.transform(BodyPart.BODY);
            this.modelElytra.setRotationAngles(swing, swingAmount, age, yaw, head, scale, entity);
            this.modelElytra.render(entity, swing, swingAmount, age, yaw, head, scale);

            if (itemstack.isItemEnchanted()) {
                LayerArmorBase.renderEnchantedGlint(this.renderPlayer, entity, this.modelElytra, swing, swingAmount, ticks, age, yaw, head, scale);
            }

            GlStateManager.popMatrix();
        }
    }

    @Override
    public boolean shouldCombineTextures() {
        return false;
    }

}