118 lines
4.5 KiB
Java
118 lines
4.5 KiB
Java
package net.minecraft.client.renderer.item;
|
|
|
|
import com.google.common.base.Suppliers;
|
|
import com.mojang.serialization.MapCodec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.Set;
|
|
import java.util.function.Supplier;
|
|
import net.fabricmc.api.EnvType;
|
|
import net.fabricmc.api.Environment;
|
|
import net.minecraft.client.color.item.ItemTintSource;
|
|
import net.minecraft.client.color.item.ItemTintSources;
|
|
import net.minecraft.client.multiplayer.ClientLevel;
|
|
import net.minecraft.client.renderer.ItemBlockRenderTypes;
|
|
import net.minecraft.client.renderer.block.model.BakedQuad;
|
|
import net.minecraft.client.renderer.block.model.FaceBakery;
|
|
import net.minecraft.client.renderer.block.model.TextureSlots;
|
|
import net.minecraft.client.resources.model.BlockModelRotation;
|
|
import net.minecraft.client.resources.model.ModelBaker;
|
|
import net.minecraft.client.resources.model.ResolvableModel;
|
|
import net.minecraft.client.resources.model.ResolvedModel;
|
|
import net.minecraft.resources.ResourceLocation;
|
|
import net.minecraft.tags.ItemTags;
|
|
import net.minecraft.world.entity.LivingEntity;
|
|
import net.minecraft.world.item.ItemDisplayContext;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.item.Items;
|
|
import org.jetbrains.annotations.Nullable;
|
|
import org.joml.Vector3f;
|
|
|
|
@Environment(EnvType.CLIENT)
|
|
public class BlockModelWrapper implements ItemModel {
|
|
private final List<ItemTintSource> tints;
|
|
private final List<BakedQuad> quads;
|
|
private final Supplier<Vector3f[]> extents;
|
|
private final ModelRenderProperties properties;
|
|
|
|
public BlockModelWrapper(List<ItemTintSource> tints, List<BakedQuad> quads, ModelRenderProperties properties) {
|
|
this.tints = tints;
|
|
this.quads = quads;
|
|
this.properties = properties;
|
|
this.extents = Suppliers.memoize(() -> computeExtents(this.quads));
|
|
}
|
|
|
|
public static Vector3f[] computeExtents(List<BakedQuad> quads) {
|
|
Set<Vector3f> set = new HashSet();
|
|
|
|
for (BakedQuad bakedQuad : quads) {
|
|
FaceBakery.extractPositions(bakedQuad.vertices(), set::add);
|
|
}
|
|
|
|
return (Vector3f[])set.toArray(Vector3f[]::new);
|
|
}
|
|
|
|
@Override
|
|
public void update(
|
|
ItemStackRenderState renderState,
|
|
ItemStack stack,
|
|
ItemModelResolver itemModelResolver,
|
|
ItemDisplayContext displayContext,
|
|
@Nullable ClientLevel level,
|
|
@Nullable LivingEntity entity,
|
|
int seed
|
|
) {
|
|
ItemStackRenderState.LayerRenderState layerRenderState = renderState.newLayer();
|
|
if (stack.hasFoil()) {
|
|
layerRenderState.setFoilType(hasSpecialAnimatedTexture(stack) ? ItemStackRenderState.FoilType.SPECIAL : ItemStackRenderState.FoilType.STANDARD);
|
|
}
|
|
|
|
int i = this.tints.size();
|
|
int[] is = layerRenderState.prepareTintLayers(i);
|
|
|
|
for (int j = 0; j < i; j++) {
|
|
is[j] = ((ItemTintSource)this.tints.get(j)).calculate(stack, level, entity);
|
|
}
|
|
|
|
layerRenderState.setExtents(this.extents);
|
|
layerRenderState.setRenderType(ItemBlockRenderTypes.getRenderType(stack));
|
|
this.properties.applyToLayer(layerRenderState, displayContext);
|
|
layerRenderState.prepareQuadList().addAll(this.quads);
|
|
}
|
|
|
|
private static boolean hasSpecialAnimatedTexture(ItemStack stack) {
|
|
return stack.is(ItemTags.COMPASSES) || stack.is(Items.CLOCK);
|
|
}
|
|
|
|
@Environment(EnvType.CLIENT)
|
|
public record Unbaked(ResourceLocation model, List<ItemTintSource> tints) implements ItemModel.Unbaked {
|
|
public static final MapCodec<BlockModelWrapper.Unbaked> MAP_CODEC = RecordCodecBuilder.mapCodec(
|
|
instance -> instance.group(
|
|
ResourceLocation.CODEC.fieldOf("model").forGetter(BlockModelWrapper.Unbaked::model),
|
|
ItemTintSources.CODEC.listOf().optionalFieldOf("tints", List.of()).forGetter(BlockModelWrapper.Unbaked::tints)
|
|
)
|
|
.apply(instance, BlockModelWrapper.Unbaked::new)
|
|
);
|
|
|
|
@Override
|
|
public void resolveDependencies(ResolvableModel.Resolver resolver) {
|
|
resolver.markDependency(this.model);
|
|
}
|
|
|
|
@Override
|
|
public ItemModel bake(ItemModel.BakingContext context) {
|
|
ModelBaker modelBaker = context.blockModelBaker();
|
|
ResolvedModel resolvedModel = modelBaker.getModel(this.model);
|
|
TextureSlots textureSlots = resolvedModel.getTopTextureSlots();
|
|
List<BakedQuad> list = resolvedModel.bakeTopGeometry(textureSlots, modelBaker, BlockModelRotation.X0_Y0).getAll();
|
|
ModelRenderProperties modelRenderProperties = ModelRenderProperties.fromResolvedModel(modelBaker, resolvedModel, textureSlots);
|
|
return new BlockModelWrapper(this.tints, list, modelRenderProperties);
|
|
}
|
|
|
|
@Override
|
|
public MapCodec<BlockModelWrapper.Unbaked> type() {
|
|
return MAP_CODEC;
|
|
}
|
|
}
|
|
}
|