/*
 * Decompiled with CFR 0.152.
 */
package net.minecraft.client.model;

import java.util.Set;
import net.minecraft.client.model.BabyModelTransform;
import net.minecraft.client.model.QuadrupedModel;
import net.minecraft.client.model.geom.ModelPart;
import net.minecraft.client.model.geom.PartPose;
import net.minecraft.client.model.geom.builders.CubeListBuilder;
import net.minecraft.client.model.geom.builders.LayerDefinition;
import net.minecraft.client.model.geom.builders.MeshDefinition;
import net.minecraft.client.model.geom.builders.MeshTransformer;
import net.minecraft.client.model.geom.builders.PartDefinition;
import net.minecraft.client.renderer.entity.state.PandaRenderState;
import net.minecraft.util.Mth;

public class PandaModel
extends QuadrupedModel<PandaRenderState> {
    public static final MeshTransformer BABY_TRANSFORMER = new BabyModelTransform(true, 23.0f, 4.8f, 2.7f, 3.0f, 49.0f, Set.of("head"));

    public PandaModel(ModelPart root) {
        super(root);
    }

    public static LayerDefinition createBodyLayer() {
        MeshDefinition mesh = new MeshDefinition();
        PartDefinition root = mesh.getRoot();
        root.addOrReplaceChild("head", CubeListBuilder.create().texOffs(0, 6).addBox(-6.5f, -5.0f, -4.0f, 13.0f, 10.0f, 9.0f).texOffs(45, 16).addBox("nose", -3.5f, 0.0f, -6.0f, 7.0f, 5.0f, 2.0f).texOffs(52, 25).addBox("left_ear", 3.5f, -8.0f, -1.0f, 5.0f, 4.0f, 1.0f).texOffs(52, 25).addBox("right_ear", -8.5f, -8.0f, -1.0f, 5.0f, 4.0f, 1.0f), PartPose.offset(0.0f, 11.5f, -17.0f));
        root.addOrReplaceChild("body", CubeListBuilder.create().texOffs(0, 25).addBox(-9.5f, -13.0f, -6.5f, 19.0f, 26.0f, 13.0f), PartPose.offsetAndRotation(0.0f, 10.0f, 0.0f, 1.5707964f, 0.0f, 0.0f));
        int legH = 9;
        int legW = 6;
        CubeListBuilder leg = CubeListBuilder.create().texOffs(40, 0).addBox(-3.0f, 0.0f, -3.0f, 6.0f, 9.0f, 6.0f);
        root.addOrReplaceChild("right_hind_leg", leg, PartPose.offset(-5.5f, 15.0f, 9.0f));
        root.addOrReplaceChild("left_hind_leg", leg, PartPose.offset(5.5f, 15.0f, 9.0f));
        root.addOrReplaceChild("right_front_leg", leg, PartPose.offset(-5.5f, 15.0f, -9.0f));
        root.addOrReplaceChild("left_front_leg", leg, PartPose.offset(5.5f, 15.0f, -9.0f));
        return LayerDefinition.create(mesh, 64, 64);
    }

    @Override
    public void setupAnim(PandaRenderState state) {
        super.setupAnim(state);
        if (state.isUnhappy) {
            this.head.yRot = 0.35f * Mth.sin(0.6f * state.ageInTicks);
            this.head.zRot = 0.35f * Mth.sin(0.6f * state.ageInTicks);
            this.rightFrontLeg.xRot = -0.75f * Mth.sin(0.3f * state.ageInTicks);
            this.leftFrontLeg.xRot = 0.75f * Mth.sin(0.3f * state.ageInTicks);
        } else {
            this.head.zRot = 0.0f;
        }
        if (state.isSneezing) {
            if (state.sneezeTime < 15) {
                this.head.xRot = -0.7853982f * (float)state.sneezeTime / 14.0f;
            } else if (state.sneezeTime < 20) {
                float internalSneezePos = (state.sneezeTime - 15) / 5;
                this.head.xRot = -0.7853982f + 0.7853982f * internalSneezePos;
            }
        }
        if (state.sitAmount > 0.0f) {
            this.body.xRot = Mth.rotLerpRad(state.sitAmount, this.body.xRot, 1.7407963f);
            this.head.xRot = Mth.rotLerpRad(state.sitAmount, this.head.xRot, 1.5707964f);
            this.rightFrontLeg.zRot = -0.27079642f;
            this.leftFrontLeg.zRot = 0.27079642f;
            this.rightHindLeg.zRot = 0.5707964f;
            this.leftHindLeg.zRot = -0.5707964f;
            if (state.isEating) {
                this.head.xRot = 1.5707964f + 0.2f * Mth.sin(state.ageInTicks * 0.6f);
                this.rightFrontLeg.xRot = -0.4f - 0.2f * Mth.sin(state.ageInTicks * 0.6f);
                this.leftFrontLeg.xRot = -0.4f - 0.2f * Mth.sin(state.ageInTicks * 0.6f);
            }
            if (state.isScared) {
                this.head.xRot = 2.1707964f;
                this.rightFrontLeg.xRot = -0.9f;
                this.leftFrontLeg.xRot = -0.9f;
            }
        } else {
            this.rightHindLeg.zRot = 0.0f;
            this.leftHindLeg.zRot = 0.0f;
            this.rightFrontLeg.zRot = 0.0f;
            this.leftFrontLeg.zRot = 0.0f;
        }
        if (state.lieOnBackAmount > 0.0f) {
            this.rightHindLeg.xRot = -0.6f * Mth.sin(state.ageInTicks * 0.15f);
            this.leftHindLeg.xRot = 0.6f * Mth.sin(state.ageInTicks * 0.15f);
            this.rightFrontLeg.xRot = 0.3f * Mth.sin(state.ageInTicks * 0.25f);
            this.leftFrontLeg.xRot = -0.3f * Mth.sin(state.ageInTicks * 0.25f);
            this.head.xRot = Mth.rotLerpRad(state.lieOnBackAmount, this.head.xRot, 1.5707964f);
        }
        if (state.rollAmount > 0.0f) {
            this.head.xRot = Mth.rotLerpRad(state.rollAmount, this.head.xRot, 2.0561945f);
            this.rightHindLeg.xRot = -0.5f * Mth.sin(state.ageInTicks * 0.5f);
            this.leftHindLeg.xRot = 0.5f * Mth.sin(state.ageInTicks * 0.5f);
            this.rightFrontLeg.xRot = 0.5f * Mth.sin(state.ageInTicks * 0.5f);
            this.leftFrontLeg.xRot = -0.5f * Mth.sin(state.ageInTicks * 0.5f);
        }
    }
}