mirror of
https://github.com/Sollace/Unicopia.git
synced 2024-11-28 23:48:00 +01:00
81 lines
3.4 KiB
Java
81 lines
3.4 KiB
Java
package com.minelittlepony.unicopia.mixin;
|
|
|
|
import org.spongepowered.asm.mixin.Mixin;
|
|
import org.spongepowered.asm.mixin.injection.At;
|
|
import org.spongepowered.asm.mixin.injection.Inject;
|
|
import org.spongepowered.asm.mixin.injection.ModifyVariable;
|
|
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
|
|
import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
|
|
|
|
import com.minelittlepony.unicopia.ducks.PonyContainer;
|
|
import com.minelittlepony.unicopia.entity.Ponylike;
|
|
import com.minelittlepony.unicopia.entity.player.Pony;
|
|
import com.minelittlepony.unicopia.entity.player.PlayerImpl;
|
|
import com.mojang.datafixers.util.Either;
|
|
|
|
import net.minecraft.entity.EntityDimensions;
|
|
import net.minecraft.entity.EntityPose;
|
|
import net.minecraft.entity.ItemEntity;
|
|
import net.minecraft.entity.LivingEntity;
|
|
import net.minecraft.entity.player.PlayerEntity;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraft.util.Unit;
|
|
import net.minecraft.util.math.BlockPos;
|
|
import net.minecraft.world.GameMode;
|
|
|
|
@Mixin(PlayerEntity.class)
|
|
abstract class MixinPlayerEntity extends LivingEntity implements PonyContainer<Pony> {
|
|
private MixinPlayerEntity() { super(null, null); }
|
|
|
|
@Override
|
|
public Ponylike create() {
|
|
return new PlayerImpl((PlayerEntity)(Object)this);
|
|
}
|
|
|
|
@ModifyVariable(method = "handleFallDamage(FF)Z",
|
|
at = @At("HEAD"),
|
|
ordinal = 0,
|
|
argsOnly = true)
|
|
private float onHandleFallDamage(float distance) {
|
|
return get().onImpact(distance);
|
|
}
|
|
|
|
@Inject(method = "trySleep(Lnet/minecraft/util/math/BlockPos;)Lcom/mojang/datafixers/util/Either;",
|
|
at = @At("HEAD"),
|
|
cancellable = true)
|
|
private void onTrySleep(BlockPos pos, CallbackInfoReturnable<Either<PlayerEntity.SleepFailureReason, Unit>> info) {
|
|
if (!world.isClient) {
|
|
Either<PlayerEntity.SleepFailureReason, Unit> result = get().trySleep(pos);
|
|
|
|
result.ifLeft(reason -> info.setReturnValue(result));
|
|
}
|
|
}
|
|
|
|
@Inject(method = "dropItem(Lnet/minecraft/item/ItemStack;ZZ)Lnet/minecraft/entity/ItemEntity;",
|
|
at = @At("HEAD"))
|
|
private void onDropItem(ItemStack itemStack_1, boolean a, boolean b, CallbackInfoReturnable<ItemEntity> info) {
|
|
PonyContainer.of(info.getReturnValue()).ifPresent(o -> {
|
|
o.get().setSpecies(get().getSpecies());
|
|
});
|
|
}
|
|
|
|
@Inject(method = "setGameMode(Lnet/minecraft/world/GameMode;)V",
|
|
at = @At("RETURN"))
|
|
private void onSetGameMode(GameMode mode, CallbackInfo info) {
|
|
get().setSpecies(get().getSpecies());
|
|
}
|
|
|
|
@Inject(method = "getActiveEyeHeight(Lnet/minecraft/entity/EntityPose;Lnet/minecraft/entity/EntityDimensions;)F",
|
|
at = @At("RETURN"),
|
|
cancellable = true)
|
|
private void onGetActiveEyeHeight(EntityPose pose, EntityDimensions dimensions, CallbackInfoReturnable<Float> info) {
|
|
info.setReturnValue(get().getGravity().getDimensions().getActiveEyeHeight(info.getReturnValue()));
|
|
}
|
|
|
|
@Inject(method = "getDimensions(Lnet/minecraft/entity/EntityPose;)Lnet/minecraft/entity/EntityDimensions;",
|
|
at = @At("RETURN"),
|
|
cancellable = true)
|
|
public void onGetDimensions(EntityPose pose, CallbackInfoReturnable<EntityDimensions> info) {
|
|
info.setReturnValue(get().getGravity().getDimensions().getDimensions(pose, info.getReturnValue()));
|
|
}
|
|
}
|