MineLittlePony/src/main/java/com/brohoof/minelittlepony/model/AbstractPonyModel.java

package com.brohoof.minelittlepony.model;

import static net.minecraft.client.renderer.GlStateManager.*;

import java.util.List;
import java.util.Random;

import com.brohoof.minelittlepony.PonyData;
import com.brohoof.minelittlepony.PonySize;
import com.brohoof.minelittlepony.model.part.IPonyPart;
import com.brohoof.minelittlepony.model.pony.ModelHumanPlayer;
import com.brohoof.minelittlepony.model.pony.ModelPlayerPony;
import com.brohoof.minelittlepony.renderer.PlaneRenderer;
import com.google.common.collect.Lists;

import net.minecraft.client.entity.AbstractClientPlayer;
import net.minecraft.client.model.ModelBase;
import net.minecraft.client.model.ModelPlayer;
import net.minecraft.client.model.ModelRenderer;
import net.minecraft.client.renderer.GlStateManager;
import net.minecraft.entity.Entity;
import net.minecraft.entity.player.EnumPlayerModelParts;
import net.minecraft.item.EnumAction;
import net.minecraft.item.ItemStack;
import net.minecraft.util.EnumHandSide;
import net.minecraft.util.math.MathHelper;

public abstract class AbstractPonyModel extends ModelPlayer {

    protected float scale = 0.0625F;

    public ModelRenderer steveLeftArm;
    public ModelRenderer steveRightArm;
    public ModelRenderer steveLeftArmwear;
    public ModelRenderer steveRightArmwear;

    public boolean isFlying;
    public boolean isSleeping;

    public PonyData metadata = new PonyData();

    protected List<IPonyPart> modelParts = Lists.newArrayList();

    public AbstractPonyModel(boolean arms) {
        super(0, arms);
        this.steveLeftArm = this.bipedLeftArm;
        this.steveRightArm = this.bipedRightArm;
        this.steveLeftArmwear = this.bipedLeftArmwear;
        this.steveRightArmwear = this.bipedLeftArmwear;
    }

    public final void init(float yOffset, float stretch) {
        this.initTextures();
        this.initPositions(yOffset, stretch);
        for (IPonyPart part : modelParts) {
            part.init(yOffset, stretch);
        }
    }

    protected void initTextures() {}

    protected void initPositions(float yOffset, float stretch) {}

    protected void animate(float move, float swing, float tick, float horz, float vert, Entity entityIn) {}

    protected void render() {}

    @Override
    public void render(Entity player, float Move, float Moveswing, float Loop, float Right, float Down, float Scale) {
        if (player instanceof AbstractClientPlayer) {
            setModelVisibilities((AbstractClientPlayer) player);
        }
        if (!doCancelRender()) {
            this.setRotationAngles(Move, Moveswing, Loop, Right, Down, Scale, player);
            GlStateManager.pushMatrix();
            this.render();
            GlStateManager.popMatrix();
            for (IPonyPart part : modelParts) {
                GlStateManager.pushMatrix();
                part.render(metadata, scale);
                GlStateManager.popMatrix();
            }
        } else {
            super.render(player, Move, Moveswing, Loop, Right, Down, Scale);
        }
    }

    @Override
    public void setRotationAngles(float Move, float Moveswing, float Loop, float Right, float Down, float Scale, Entity entityIn) {
        if (doCancelRender()) {
            super.setRotationAngles(Move, Moveswing, Loop, Right, Down, Scale, entityIn);
            return;
        }
        this.animate(Move, Moveswing, Loop, Right, Down, entityIn);
        for (IPonyPart part : modelParts) {
            part.animate(metadata, Move, Moveswing, Loop, Right, Down);
        }
        this.steveRightArm.rotateAngleX = MathHelper.cos(Move * 0.6662F + (float) Math.PI) * 2.0F * Moveswing * 0.5F;
        this.steveRightArm.rotateAngleY = 0;
        this.steveRightArm.rotateAngleZ = 0;
        this.steveLeftArm.rotateAngleX = MathHelper.cos(Move * 0.6662F) * 2.0F * Moveswing * 0.5F;
        this.steveLeftArm.rotateAngleY = 0;
        this.steveLeftArm.rotateAngleZ = 0;

        copyModelAngles(steveRightArm, steveRightArmwear);
        copyModelAngles(steveLeftArm, steveLeftArmwear);
    }

    protected void setModelVisibilities(AbstractClientPlayer clientPlayer) {
        ModelPlayer modelplayer = this;

        if (clientPlayer.isSpectator()) {
            modelplayer.setInvisible(false);
            modelplayer.bipedHead.showModel = true;
            modelplayer.bipedHeadwear.showModel = true;
        } else {
            modelplayer.setInvisible(true);
            modelplayer.bipedHeadwear.showModel = clientPlayer.isWearing(EnumPlayerModelParts.HAT);
            modelplayer.bipedBodyWear.showModel = clientPlayer.isWearing(EnumPlayerModelParts.JACKET);
            modelplayer.bipedLeftLegwear.showModel = clientPlayer.isWearing(EnumPlayerModelParts.LEFT_PANTS_LEG);
            modelplayer.bipedRightLegwear.showModel = clientPlayer.isWearing(EnumPlayerModelParts.RIGHT_PANTS_LEG);
            modelplayer.bipedLeftArmwear.showModel = clientPlayer.isWearing(EnumPlayerModelParts.LEFT_SLEEVE);
            modelplayer.bipedRightArmwear.showModel = clientPlayer.isWearing(EnumPlayerModelParts.RIGHT_SLEEVE);
            modelplayer.isSneak = clientPlayer.isSneaking();

            ItemStack main = clientPlayer.getHeldItemMainhand();
            ArmPose mainPose;
            if (main == null) {
                mainPose = ArmPose.EMPTY;
            } else {
                mainPose = ArmPose.ITEM;

                if (clientPlayer.getItemInUseCount() > 0) {
                    EnumAction enumaction = main.getItemUseAction();

                    if (enumaction == EnumAction.BLOCK) {
                        mainPose = ArmPose.BLOCK;
                    } else if (enumaction == EnumAction.BOW) {
                        mainPose = ArmPose.BOW_AND_ARROW;
                    }
                }
            }

            ItemStack off = clientPlayer.getHeldItemOffhand();
            ArmPose offPose;
            if (off == null) {
                offPose = ArmPose.EMPTY;
            } else {
                offPose = ArmPose.ITEM;

                if (clientPlayer.getItemInUseCount() > 0) {
                    EnumAction enumaction = off.getItemUseAction();

                    if (enumaction == EnumAction.BLOCK) {
                        offPose = ArmPose.BLOCK;
                    } else if (enumaction == EnumAction.BOW) {
                        offPose = ArmPose.BOW_AND_ARROW;
                    }
                }
            }

            if (clientPlayer.getPrimaryHand() == EnumHandSide.RIGHT) {
                modelplayer.rightArmPose = mainPose;
                modelplayer.leftArmPose = offPose;
            } else {
                modelplayer.leftArmPose = mainPose;
                modelplayer.rightArmPose = offPose;
            }
        }
    }

    protected boolean doCancelRender() {
        return false;
    }

    protected ArmPose getMainArmPose(Entity entity) {
        EnumHandSide mainHand = this.getMainHand(entity);
        return mainHand == EnumHandSide.RIGHT ? this.rightArmPose : this.leftArmPose;
    }

    public static void setRotationPoint(ModelRenderer aRenderer, float setX, float setY, float setZ) {
        aRenderer.rotationPointX = setX;
        aRenderer.rotationPointY = setY;
        aRenderer.rotationPointZ = setZ;
    }

    public static void setRotationPoint(PlaneRenderer aPlaneRenderer, float setX, float setY, float setZ) {
        aPlaneRenderer.rotationPointX = setX;
        aPlaneRenderer.rotationPointY = setY;
        aPlaneRenderer.rotationPointZ = setZ;
    }

    public static void shiftRotationPoint(PlaneRenderer aPlaneRenderer, float shiftX, float shiftY, float shiftZ) {
        aPlaneRenderer.rotationPointX += shiftX;
        aPlaneRenderer.rotationPointY += shiftY;
        aPlaneRenderer.rotationPointZ += shiftZ;
    }

    public static void shiftRotationPoint(ModelRenderer aRenderer, float shiftX, float shiftY, float shiftZ) {
        aRenderer.rotationPointX += shiftX;
        aRenderer.rotationPointY += shiftY;
        aRenderer.rotationPointZ += shiftZ;
    }

    public void transform(BodyPart part) {
        if (this instanceof ModelHumanPlayer)
            return;
        if (this.isRiding) {
            translate(0.0F, -0.6F, -0.2F);
        }

        if (this.isSleeping) {
            rotate(90.0F, 0.0F, 1.0F, 0.0F);
            rotate(270.0F, 0.0F, 0.0F, 1.0F);
            rotate(90.0F, 0.0F, 1.0F, 0.0F);
            rotate(180.0F, 0.0F, 0.0F, 1.0F);
            rotate(180.0F, 0.0F, 1.0F, 0.0F);
        }

        if (this.metadata.getSize() == PonySize.FOAL || isChild) {
            if (this.isSneak && !this.isFlying) {
                translate(0.0F, -0.12F, 0.0F);
            }

            if (this.isSleeping) {
                translate(0.0F, -1.0F, 0.25F);
            }
            if (this.isRiding) {
                translate(0, -.1, 0);
            }
            switch (part) {
            case NECK:
            case HEAD:
                translate(0.0F, 0.76F, 0.0F);
                scale(0.9F, 0.9F, 0.9F);
                if (part == BodyPart.HEAD)
                    break;
                if (this.isSneak && !this.isFlying) {
                    translate(0.0F, -0.01F, 0.15F);
                }
                break;
            case BODY:
            case TAIL:
                translate(0.0F, 0.76F, -0.04F);
                scale(0.6F, 0.6F, 0.6F);
                break;
            case LEGS:
                translate(0.0F, 0.89F, 0.0F);
                scale(0.6F, 0.41F, 0.6F);
                if (this.isSneak && !this.isFlying) {
                    translate(0.0F, 0.12F, 0.0F);
                }

                if (this instanceof ModelPlayerPony && ((ModelPlayerPony) this).rainboom) {
                    translate(0.0F, -0.08F, 0.0F);
                }

                break;
            }

        } else if (this.metadata.getSize() == PonySize.LARGE) {
            if (this.isSleeping) {
                translate(0.0F, -0.47F, 0.2F);
            }

            switch (part) {
            case HEAD:

                translate(0.0F, -0.17F, -0.04F);
                if (this.isSleeping) {
                    translate(0.0F, 0.0F, -0.1F);
                }

                if (this.isSneak && !this.isFlying) {
                    translate(0.0F, 0.15F, 0.0F);
                }

                break;
            case NECK:
                translate(0.0F, -0.15F, -0.07F);
                if (this.isSneak && !this.isFlying) {
                    translate(0.0F, 0.0F, -0.05F);
                }

                break;
            case BODY:
                translate(0.0F, -0.2F, -0.04F);
                scale(1.15F, 1.2F, 1.2F);
                break;
            case TAIL:
                translate(0.0F, -0.2F, 0.08F);
                break;
            case LEGS:
                translate(0.0F, -0.14F, 0.0F);
                scale(1.15F, 1.12F, 1.15F);
                break;
            }
        } else if (this.metadata.getSize() == PonySize.TALL) {
            if (this.isSleeping) {
                translate(0.0F, -0.43F, 0.25F);
            }

            switch (part) {
            case HEAD:
                translate(0.0F, -0.15F, 0.01F);
                if (this.isSneak && !this.isFlying) {
                    translate(0.0F, 0.05F, 0.0F);
                }
                break;
            case NECK:
                translate(0.0F, -0.19F, -0.01F);
                scale(1.0F, 1.1F, 1.0F);
                if (this.isSneak && !this.isFlying) {
                    translate(0.0F, -0.06F, -0.04F);
                }
                break;
            case BODY:
            case TAIL:
                translate(0.0F, -0.1F, 0.0F);
                scale(1.0F, 1.0F, 1.0F);
                break;
            case LEGS:
                translate(0.0F, -0.25F, 0.03F);
                scale(1.0F, 1.18F, 1.0F);
                if (this instanceof ModelPlayerPony && ((ModelPlayerPony) this).rainboom) {
                    translate(0.0F, 0.05F, 0.0F);
                }
                break;
            }
        } else {
            if (this.isSleeping) {
                translate(0.0F, -0.535F, 0.25F);
            }
        }
    }

    @Override
    public void setModelAttributes(ModelBase model) {
        super.setModelAttributes(model);
        if (model instanceof AbstractPonyModel) {
            AbstractPonyModel pony = (AbstractPonyModel) model;
            this.isFlying = pony.isFlying;
            this.isSleeping = pony.isSleeping;
        }
    }

    @Override
    public ModelRenderer getRandomModelBox(Random rand) {
        // empty lists cause problems
        ModelRenderer mr = null;
        do {
            // try until it's not
            mr = super.getRandomModelBox(rand);
        } while (mr.cubeList.isEmpty());
        return mr;
    }
}