diff options
Diffstat (limited to 'src/main/java/moe/nea/beacons/mixin/BeaconMixin.java')
| -rw-r--r-- | src/main/java/moe/nea/beacons/mixin/BeaconMixin.java | 94 |
1 files changed, 94 insertions, 0 deletions
diff --git a/src/main/java/moe/nea/beacons/mixin/BeaconMixin.java b/src/main/java/moe/nea/beacons/mixin/BeaconMixin.java new file mode 100644 index 0000000..4602208 --- /dev/null +++ b/src/main/java/moe/nea/beacons/mixin/BeaconMixin.java @@ -0,0 +1,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; + } + +} |