package net.minecraft.client.renderer.entity;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.model.ZombieModel;
import net.minecraft.client.model.geom.ModelLayerLocation;
import net.minecraft.client.model.geom.ModelLayers;
import net.minecraft.client.renderer.entity.state.ZombieRenderState;
import net.minecraft.world.entity.monster.Zombie;

@Environment(EnvType.CLIENT)
public class ZombieRenderer extends AbstractZombieRenderer<Zombie, ZombieRenderState, ZombieModel<ZombieRenderState>> {
	public ZombieRenderer(EntityRendererProvider.Context context) {
		this(
			context,
			ModelLayers.ZOMBIE,
			ModelLayers.ZOMBIE_BABY,
			ModelLayers.ZOMBIE_INNER_ARMOR,
			ModelLayers.ZOMBIE_OUTER_ARMOR,
			ModelLayers.ZOMBIE_BABY_INNER_ARMOR,
			ModelLayers.ZOMBIE_BABY_OUTER_ARMOR
		);
	}

	public ZombieRenderState createRenderState() {
		return new ZombieRenderState();
	}

	public ZombieRenderer(
		EntityRendererProvider.Context context,
		ModelLayerLocation adultModel,
		ModelLayerLocation babyModel,
		ModelLayerLocation innerModel,
		ModelLayerLocation outerModel,
		ModelLayerLocation innerModelBaby,
		ModelLayerLocation outerModelBaby
	) {
		super(
			context,
			new ZombieModel<>(context.bakeLayer(adultModel)),
			new ZombieModel<>(context.bakeLayer(babyModel)),
			new ZombieModel<>(context.bakeLayer(innerModel)),
			new ZombieModel<>(context.bakeLayer(outerModel)),
			new ZombieModel<>(context.bakeLayer(innerModelBaby)),
			new ZombieModel<>(context.bakeLayer(outerModelBaby))
		);
	}
}