-
Notifications
You must be signed in to change notification settings - Fork 390
/
IETemplateMultiblock.java
142 lines (126 loc) · 4.5 KB
/
IETemplateMultiblock.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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
/*
* BluSunrize
* Copyright (c) 2017
*
* This code is licensed under "Blu's License of Common Sense"
* Details can be found in the license file in the root folder of this project
*/
package blusunrize.immersiveengineering.common.blocks.multiblocks;
import blusunrize.immersiveengineering.api.IEProperties;
import blusunrize.immersiveengineering.api.multiblocks.ClientMultiblocks.MultiblockManualData;
import blusunrize.immersiveengineering.api.multiblocks.TemplateMultiblock;
import blusunrize.immersiveengineering.client.utils.BasicClientProperties;
import blusunrize.immersiveengineering.common.blocks.generic.MultiblockPartBlockEntity;
import blusunrize.immersiveengineering.common.register.IEBlocks;
import blusunrize.immersiveengineering.common.util.IELogger;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableMap;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.core.Vec3i;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.entity.BlockEntity;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.chunk.LevelChunk;
import net.minecraft.world.level.levelgen.structure.templatesystem.StructureTemplate;
import net.minecraft.world.level.levelgen.structure.templatesystem.StructureTemplate.StructureBlockInfo;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.function.Consumer;
public abstract class IETemplateMultiblock extends TemplateMultiblock
{
private final IEBlocks.BlockEntry<?> baseState;
public IETemplateMultiblock(
ResourceLocation loc, BlockPos masterFromOrigin, BlockPos triggerFromOrigin, BlockPos size,
IEBlocks.BlockEntry<?> baseState
)
{
super(loc, masterFromOrigin, triggerFromOrigin, size, ImmutableMap.of());
this.baseState = baseState;
}
@Override
protected void replaceStructureBlock(StructureBlockInfo info, Level world, BlockPos actualPos, boolean mirrored, Direction clickDirection, Vec3i offsetFromMaster)
{
BlockState newState = baseState.get().defaultBlockState();
if(!offsetFromMaster.equals(Vec3i.ZERO))
newState = newState.setValue(IEProperties.MULTIBLOCKSLAVE, true);
final BlockState oldState = world.getBlockState(actualPos);
world.setBlock(actualPos, newState, 0);
BlockEntity curr = world.getBlockEntity(actualPos);
if(curr instanceof MultiblockPartBlockEntity<?> tile)
{
tile.formed = true;
tile.offsetToMaster = new BlockPos(offsetFromMaster);
tile.posInMultiblock = info.pos;
if(newState.hasProperty(IEProperties.MIRRORED))
tile.setMirrored(mirrored);
tile.setFacing(transformDirection(clickDirection.getOpposite()));
tile.setChanged();
world.blockEvent(actualPos, world.getBlockState(actualPos).getBlock(), 255, 0);
}
else
IELogger.logger.error("Expected MB TE at {} during placement", actualPos);
final LevelChunk chunk = world.getChunkAt(actualPos);
world.markAndNotifyBlock(actualPos, chunk, oldState, newState, Block.UPDATE_ALL, 512);
}
public Direction transformDirection(Direction original)
{
return original;
}
public Direction untransformDirection(Direction transformed)
{
return transformed;
}
public BlockPos multiblockToModelPos(BlockPos posInMultiblock)
{
return posInMultiblock.subtract(masterFromOrigin);
}
@Override
public Vec3i getSize(@Nullable Level world)
{
return size;
}
@Nonnull
@Override
protected StructureTemplate getTemplate(@Nullable Level world)
{
StructureTemplate result = super.getTemplate(world);
Preconditions.checkState(
result.getSize().equals(size),
"Wrong template size for multiblock %s, template size: %s",
getTemplateLocation(), result.getSize()
);
return result;
}
@Override
protected void prepareBlockForDisassembly(Level world, BlockPos pos)
{
BlockEntity be = world.getBlockEntity(pos);
if(be instanceof MultiblockPartBlockEntity<?> multiblockBE)
multiblockBE.formed = false;
else if(be!=null)
IELogger.logger.error("Expected multiblock TE at {}, got {}", pos, be);
}
@Override
public void initializeClient(Consumer<MultiblockManualData> consumer)
{
consumer.accept(new BasicClientProperties(this));
}
public ResourceLocation getBlockName()
{
return baseState.getId();
}
@Override
public Component getDisplayName()
{
return baseState.get().getName();
}
@Override
public Block getBlock()
{
return baseState.get();
}
}