Create animation controllers for my entity

Hi, i made a entity and if i put my animations in "setRotationAngles" the animation plays when the entity moves, but i want to make separated animations for when the entity died, moved, jumped, idle, etc, like if(entityIn.WALKING or entityIn.IS_IDLE or entityIn.DIED)

@Override
    public void setRotationAngles(T entityIn, float limbSwing, float limbSwingAmount, float ageInTicks, float netHeadYaw, float headPitch) {
        if(entityIn.isWalking) //an example
		{
			//my animation for walking
			this.root.rotateAngleX = MathHelper.cos(limbSwing * 0.6662F) * 0.3F * limbSwingAmount;
		}
		if(entityIn.isIdle)
		{
			//my animation for idle
			this.root.rotateAngleX = MathHelper.cos(limbSwing * 0.6662F) * 0.3F * limbSwingAmount;
		}
    }