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
|
package moe.nea.beacons.mixin;
import com.llamalad7.mixinextras.injector.wrapoperation.Operation;
import com.llamalad7.mixinextras.injector.wrapoperation.WrapOperation;
import com.llamalad7.mixinextras.sugar.Local;
import moe.nea.beacons.BeaconBeamSectionExtra;
import moe.nea.beacons.BeaconBlockEntityExtra;
import moe.nea.beacons.BeaconRangeExtender;
import moe.nea.beacons.BeaconRangeIncreaser;
import net.minecraft.core.BlockPos;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.block.entity.BeaconBlockEntity;
import net.minecraft.world.level.block.state.BlockState;
import org.objectweb.asm.Opcodes;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Shadow;
import org.spongepowered.asm.mixin.Unique;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.ModifyArg;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
import java.util.List;
@Mixin(BeaconBlockEntity.class)
public class BeaconMixin implements BeaconBlockEntityExtra {
@Shadow
private List<BeaconBlockEntity.BeaconBeamSection> checkingBeamSections;
@Unique
private int rangeExtenders;
@Override
public int getRangeExtender_nea() {
return rangeExtenders;
}
@Override
public void setRangeExtender_nea(int rangeExtender) {
this.rangeExtenders = rangeExtender;
}
@Override
public List<BeaconBlockEntity.BeaconBeamSection> getCheckingBeamSections_nea() {
return this.checkingBeamSections;
}
@Inject(method = "tick", at = @At(value = "FIELD", target = "Lnet/minecraft/world/level/block/entity/BeaconBlockEntity;beamSections:Ljava/util/List;", opcode = Opcodes.ASTORE))
private static void saveRangeExtender(Level level, BlockPos blockPos, BlockState blockState, BeaconBlockEntity beaconBlockEntity, CallbackInfo ci) {
var extra = BeaconBlockEntityExtra.cast(beaconBlockEntity);
var accExtraRange = 0;
for (var beaconBeamSection : extra.getCheckingBeamSections_nea()) {
var extraSection = BeaconBeamSectionExtra.cast(beaconBeamSection);
accExtraRange += extraSection.getRangeExtenders_nea();
}
extra.setRangeExtender_nea(accExtraRange);
}
@WrapOperation(method = "tick", at = @At(value = "INVOKE", target = "Lnet/minecraft/core/BlockPos;above()Lnet/minecraft/core/BlockPos;"))
private static BlockPos checkForRangeExtender(
BlockPos oldBlockPos, Operation<BlockPos> original, @Local BeaconBlockEntity.BeaconBeamSection lastSection,
@Local(argsOnly = true) Level level) {
var newPos = original.call(oldBlockPos);
var blockState = level.getBlockState(oldBlockPos);
if (blockState.getBlock() instanceof BeaconRangeExtender extender && lastSection != null) {
var lastSectionExtra = BeaconBeamSectionExtra.cast(lastSection);
lastSectionExtra.setRangeExtenders_nea(
extender.extraBeaconRange_nea() + lastSectionExtra.getRangeExtenders_nea()
);
}
return newPos;
}
@Inject(method = "tick", at = @At(value = "INVOKE", target = "Lnet/minecraft/world/level/block/entity/BeaconBlockEntity;applyEffects(Lnet/minecraft/world/level/Level;Lnet/minecraft/core/BlockPos;ILnet/minecraft/core/Holder;Lnet/minecraft/core/Holder;)V"))
private static void applyExtraEffects(Level level, BlockPos blockPos, BlockState blockState, BeaconBlockEntity beaconBlockEntity, CallbackInfo ci) {
var extra = BeaconBlockEntityExtra.cast(beaconBlockEntity);
lastRangeExtend = extra.getRangeExtender_nea();
}
@Unique
private static int lastRangeExtend = -1;
@ModifyArg(method = "applyEffects", at = @At(value = "INVOKE", target = "Lnet/minecraft/world/phys/AABB;inflate(D)Lnet/minecraft/world/phys/AABB;"))
private static double increaseRange(double d) {
if (lastRangeExtend < 0) {
BeaconRangeIncreaser.LOGGER.error("Could not get extended range from beacon range increase mod (which is awesome, btw)");
return d;
}
d += lastRangeExtend;
lastRangeExtend = -1;
return d;
}
}
|
