/
DisguiseBakedModel.java
63 lines (54 loc) · 2.65 KB
/
DisguiseBakedModel.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package slimeknights.tmechworks.client.model;
import net.minecraft.block.BlockState;
import net.minecraft.block.DirectionalBlock;
import net.minecraft.client.Minecraft;
import net.minecraft.client.renderer.BlockModelShapes;
import net.minecraft.client.renderer.model.BakedQuad;
import net.minecraft.client.renderer.model.IBakedModel;
import net.minecraft.item.BlockItem;
import net.minecraft.item.BlockItemUseContext;
import net.minecraft.item.ItemStack;
import net.minecraft.item.ItemUseContext;
import net.minecraft.util.Direction;
import net.minecraft.util.math.BlockPos;
import net.minecraft.world.IEnviromentBlockReader;
import net.minecraftforge.client.MinecraftForgeClient;
import net.minecraftforge.client.model.BakedModelWrapper;
import net.minecraftforge.client.model.data.IModelData;
import net.minecraftforge.client.model.data.ModelProperty;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.Collections;
import java.util.List;
import java.util.Random;
public class DisguiseBakedModel extends BakedModelWrapper<IBakedModel> {
public static final ModelProperty<ItemStack> DISGUISE = new ModelProperty<>();
public DisguiseBakedModel(IBakedModel originalModel) {
super(originalModel);
}
@Nonnull
@Override
public List<BakedQuad> getQuads(@Nullable BlockState state, @Nullable Direction side, @Nonnull Random rand, @Nonnull IModelData extraData) {
if(extraData.hasProperty(DISGUISE)) {
ItemStack disguise = extraData.getData(DISGUISE);
if(disguise != null && disguise.getItem() instanceof BlockItem) {
BlockItem disguiseItem = (BlockItem) disguise.getItem();
BlockState disguiseState = disguiseItem.getBlock().getDefaultState();
if(disguiseState.has(DirectionalBlock.FACING))
disguiseState = disguiseState.with(DirectionalBlock.FACING, state.get(DirectionalBlock.FACING));
if(disguiseState.canRenderInLayer(MinecraftForgeClient.getRenderLayer())) {
IBakedModel model = Minecraft.getInstance().getBlockRendererDispatcher().getBlockModelShapes().getModel(disguiseState);
return model.getQuads(disguiseState, side, rand, extraData);
} else {
return Collections.emptyList();
}
}
}
return super.getQuads(state, side, rand, extraData);
}
@Nonnull
@Override
public IModelData getModelData(@Nonnull IEnviromentBlockReader world, @Nonnull BlockPos pos, @Nonnull BlockState state, @Nonnull IModelData tileData) {
return super.getModelData(world, pos, state, tileData);
}
}