286 lines
10 KiB
Java
286 lines
10 KiB
Java
package ru.betterend.mixin.common;
|
|
|
|
import net.minecraft.sounds.SoundEvent;
|
|
import net.minecraft.tags.EntityTypeTags;
|
|
import net.minecraft.util.Mth;
|
|
import net.minecraft.world.damagesource.DamageSource;
|
|
import net.minecraft.world.effect.MobEffect;
|
|
import net.minecraft.world.effect.MobEffectInstance;
|
|
import net.minecraft.world.effect.MobEffects;
|
|
import net.minecraft.world.entity.Entity;
|
|
import net.minecraft.world.entity.EntityType;
|
|
import net.minecraft.world.entity.EquipmentSlot;
|
|
import net.minecraft.world.entity.LivingEntity;
|
|
import net.minecraft.world.entity.MoverType;
|
|
import net.minecraft.world.entity.ai.attributes.AttributeMap;
|
|
import net.minecraft.world.entity.ai.attributes.AttributeModifier;
|
|
import net.minecraft.world.entity.ai.attributes.AttributeSupplier;
|
|
import net.minecraft.world.entity.ai.attributes.Attributes;
|
|
import net.minecraft.world.entity.animal.FlyingAnimal;
|
|
import net.minecraft.world.item.ElytraItem;
|
|
import net.minecraft.world.item.Item;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.level.Level;
|
|
import net.minecraft.world.phys.AABB;
|
|
import net.minecraft.world.phys.Vec3;
|
|
import org.spongepowered.asm.mixin.Mixin;
|
|
import org.spongepowered.asm.mixin.Shadow;
|
|
import org.spongepowered.asm.mixin.injection.At;
|
|
import org.spongepowered.asm.mixin.injection.At.Shift;
|
|
import org.spongepowered.asm.mixin.injection.Inject;
|
|
import org.spongepowered.asm.mixin.injection.ModifyArg;
|
|
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
|
|
import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
|
|
import ru.betterend.BetterEnd;
|
|
import ru.betterend.interfaces.FallFlyingItem;
|
|
import ru.betterend.interfaces.MobEffectApplier;
|
|
import ru.betterend.item.CrystaliteArmor;
|
|
import ru.betterend.registry.EndAttributes;
|
|
|
|
import java.util.Collection;
|
|
|
|
@Mixin(value=LivingEntity.class, priority=200)
|
|
public abstract class LivingEntityMixin extends Entity {
|
|
|
|
public LivingEntityMixin(EntityType<?> entityType, Level level) {
|
|
super(entityType, level);
|
|
}
|
|
|
|
@Shadow
|
|
protected int fallFlyTicks;
|
|
|
|
@Shadow
|
|
public abstract boolean hasEffect(MobEffect mobEffect);
|
|
|
|
@Shadow
|
|
public abstract ItemStack getItemBySlot(EquipmentSlot equipmentSlot);
|
|
|
|
@Shadow
|
|
public abstract void calculateEntityAnimation(LivingEntity livingEntity, boolean b);
|
|
|
|
@Shadow
|
|
protected abstract SoundEvent getFallDamageSound(int i);
|
|
|
|
@Shadow
|
|
public abstract boolean isFallFlying();
|
|
|
|
@Shadow
|
|
public abstract AttributeMap getAttributes();
|
|
|
|
private Entity lastAttacker;
|
|
|
|
@Inject(method = "createLivingAttributes", at = @At("RETURN"), cancellable = true)
|
|
private static void be_addLivingAttributes(CallbackInfoReturnable<AttributeSupplier.Builder> info) {
|
|
EndAttributes.addLivingEntityAttributes(info.getReturnValue());
|
|
}
|
|
|
|
@Inject(method = "tickEffects", at = @At("HEAD"))
|
|
protected void be_applyEffects(CallbackInfo info) {
|
|
if (!level.isClientSide()) {
|
|
LivingEntity owner = LivingEntity.class.cast(this);
|
|
if (CrystaliteArmor.hasFullSet(owner)) {
|
|
CrystaliteArmor.applySetEffect(owner);
|
|
}
|
|
getArmorSlots().forEach(itemStack -> {
|
|
if (itemStack.getItem() instanceof MobEffectApplier) {
|
|
((MobEffectApplier) itemStack.getItem()).applyEffect(owner);
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
@Inject(method = "canBeAffected", at = @At("HEAD"), cancellable = true)
|
|
public void be_canBeAffected(MobEffectInstance mobEffectInstance, CallbackInfoReturnable<Boolean> info) {
|
|
try {
|
|
if (mobEffectInstance.getEffect() == MobEffects.BLINDNESS && getAttributes().getValue(EndAttributes.BLINDNESS_RESISTANCE) > 0.0) {
|
|
info.setReturnValue(false);
|
|
}
|
|
}
|
|
catch (Exception ex) {
|
|
BetterEnd.LOGGER.warning("Blindness resistance attribute haven't been registered.");
|
|
}
|
|
}
|
|
|
|
@Inject(method = "hurt", at = @At("HEAD"))
|
|
public void be_hurt(DamageSource source, float amount, CallbackInfoReturnable<Boolean> info) {
|
|
this.lastAttacker = source.getEntity();
|
|
}
|
|
|
|
@ModifyArg(method = "hurt", at = @At(value = "INVOKE", target = "Lnet/minecraft/world/entity/LivingEntity;knockback(DDD)V"), index = 0)
|
|
private double be_increaseKnockback(double value, double x, double z) {
|
|
if (lastAttacker != null && lastAttacker instanceof LivingEntity) {
|
|
LivingEntity attacker = (LivingEntity) lastAttacker;
|
|
value += this.be_getKnockback(attacker.getMainHandItem().getItem());
|
|
}
|
|
return value;
|
|
}
|
|
|
|
// FlyFallingLib (part of Origin) redirected the call to updateFallFlying,
|
|
// so we inject our code before the actual call and cancel the execution if the player is still
|
|
// flying. That means we have to replicate all vanilla code that happens after the call to
|
|
// updateFallFlying. We do this in vanillaAfterUpdateFallFlying
|
|
@Inject(method="aiStep", cancellable = true, at=@At(value="INVOKE", target="Lnet/minecraft/world/entity/LivingEntity;updateFallFlying()V"))
|
|
private void be_updateFallFlying_originFix(CallbackInfo info) {
|
|
//run be_updateFallFlying instead
|
|
if (!BetterEnd.RUNS_FALL_FLYING_LIB) return;
|
|
|
|
if (be_updateFallFlyingCommon()) {
|
|
vanillaAfterUpdateFallFlying();
|
|
info.cancel();
|
|
}
|
|
}
|
|
|
|
@Inject(method = "updateFallFlying", at = @At("HEAD"), cancellable = true)
|
|
private void be_updateFallFlying(CallbackInfo info) {
|
|
//run be_updateFallFlying_originFix instead?
|
|
if (BetterEnd.RUNS_FALL_FLYING_LIB) return;
|
|
if (be_updateFallFlyingCommon()) {
|
|
info.cancel();
|
|
}
|
|
}
|
|
|
|
private boolean be_updateFallFlyingCommon() {
|
|
ItemStack itemStack = getItemBySlot(EquipmentSlot.CHEST);
|
|
if (!level.isClientSide && itemStack.getItem() instanceof FallFlyingItem) {
|
|
boolean isFlying = getSharedFlag(7);
|
|
if (isFlying && !onGround && !isPassenger() && !hasEffect(MobEffects.LEVITATION)) {
|
|
if (ElytraItem.isFlyEnabled(itemStack)) {
|
|
if ((fallFlyTicks + 1) % 20 == 0) {
|
|
itemStack.hurtAndBreak(
|
|
1,
|
|
LivingEntity.class.cast(this),
|
|
livingEntity -> livingEntity.broadcastBreakEvent(EquipmentSlot.CHEST)
|
|
);
|
|
}
|
|
isFlying = true;
|
|
}
|
|
else {
|
|
isFlying = false;
|
|
}
|
|
}
|
|
else {
|
|
isFlying = false;
|
|
}
|
|
setSharedFlag(7, isFlying);
|
|
return isFlying;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
@Shadow protected abstract void removeFrost();
|
|
@Shadow protected abstract void tryAddFrost();
|
|
@Shadow protected abstract void pushEntities();
|
|
@Shadow protected abstract void checkAutoSpinAttack(AABB aABB, AABB aABB2);
|
|
@Shadow protected int autoSpinAttackTicks;
|
|
|
|
private void vanillaAfterUpdateFallFlying(){
|
|
LivingEntity self = (LivingEntity)(Object)this;
|
|
AABB aABB = this.getBoundingBox();
|
|
self.travel(new Vec3(self.xxa, self.yya, self.zza));
|
|
this.level.getProfiler().pop();
|
|
this.level.getProfiler().push("freezing");
|
|
boolean bl2 = this.getType().is(EntityTypeTags.FREEZE_HURTS_EXTRA_TYPES);
|
|
int o;
|
|
if (!this.level.isClientSide && !self.isDeadOrDying()) {
|
|
o = this.getTicksFrozen();
|
|
if (this.isInPowderSnow && this.canFreeze()) {
|
|
this.setTicksFrozen(Math.min(this.getTicksRequiredToFreeze(), o + 1));
|
|
} else {
|
|
this.setTicksFrozen(Math.max(0, o - 2));
|
|
}
|
|
}
|
|
|
|
this.removeFrost();
|
|
this.tryAddFrost();
|
|
if (!this.level.isClientSide && this.tickCount % 40 == 0 && this.isFullyFrozen() && this.canFreeze()) {
|
|
o = bl2 ? 5 : 1;
|
|
this.hurt(DamageSource.FREEZE, (float)o);
|
|
}
|
|
|
|
this.level.getProfiler().pop();
|
|
this.level.getProfiler().push("push");
|
|
if (this.autoSpinAttackTicks > 0) {
|
|
--this.autoSpinAttackTicks;
|
|
this.checkAutoSpinAttack(aABB, this.getBoundingBox());
|
|
}
|
|
|
|
this.pushEntities();
|
|
this.level.getProfiler().pop();
|
|
if (!this.level.isClientSide && self.isSensitiveToWater() && this.isInWaterRainOrBubble()) {
|
|
this.hurt(DamageSource.DROWN, 1.0F);
|
|
}
|
|
}
|
|
|
|
|
|
|
|
@Inject(method = "travel", at = @At(value = "INVOKE", target = "Lnet/minecraft/world/entity/LivingEntity;isFallFlying()Z", shift = Shift.AFTER), cancellable = true)
|
|
public void be_travel(Vec3 vec3, CallbackInfo info) {
|
|
ItemStack itemStack = getItemBySlot(EquipmentSlot.CHEST);
|
|
if (isFallFlying() && itemStack.getItem() instanceof FallFlyingItem) {
|
|
Vec3 moveDelta = getDeltaMovement();
|
|
if (moveDelta.y > -0.5D) {
|
|
fallDistance = 1.0F;
|
|
}
|
|
|
|
Vec3 lookAngle = getLookAngle();
|
|
double d = 0.08D;
|
|
float rotX = getXRot() * 0.017453292F;
|
|
double k = Math.sqrt(lookAngle.x * lookAngle.x + lookAngle.z * lookAngle.z);
|
|
double l = moveDelta.horizontalDistance();
|
|
double lookLen = lookAngle.length();
|
|
float n = Mth.cos(rotX);
|
|
n = (float) (n * n * Math.min(1.0D, lookLen / 0.4D));
|
|
moveDelta = getDeltaMovement().add(0.0D, d * (-1.0D + (double) n * 0.75D), 0.0D);
|
|
double coef;
|
|
if (moveDelta.y < 0.0D && k > 0.0D) {
|
|
coef = moveDelta.y * -0.1D * (double) n;
|
|
moveDelta = moveDelta.add(lookAngle.x * coef / k, coef, lookAngle.z * coef / k);
|
|
}
|
|
|
|
if (rotX < 0.0F && k > 0.0D) {
|
|
coef = l * (double) (-Mth.sin(rotX)) * 0.04D;
|
|
moveDelta = moveDelta.add(-lookAngle.x * coef / k, coef * 3.2D, -lookAngle.z * coef / k);
|
|
}
|
|
|
|
if (k > 0.0D) {
|
|
moveDelta = moveDelta.add(
|
|
(lookAngle.x / k * l - moveDelta.x) * 0.1D,
|
|
0.0D,
|
|
(lookAngle.z / k * l - moveDelta.z) * 0.1D
|
|
);
|
|
}
|
|
moveDelta = moveDelta.multiply(0.9900000095367432D, 0.9800000190734863D, 0.9900000095367432D);
|
|
double movementFactor = ((FallFlyingItem) itemStack.getItem()).getMovementFactor();
|
|
moveDelta = moveDelta.multiply(movementFactor, 1.0D, movementFactor);
|
|
setDeltaMovement(moveDelta);
|
|
move(MoverType.SELF, moveDelta);
|
|
if (!level.isClientSide) {
|
|
if (horizontalCollision) {
|
|
coef = moveDelta.horizontalDistance();
|
|
double r = l - coef;
|
|
float dmg = (float) (r * 10.0D - 3.0D);
|
|
if (dmg > 0.0F) {
|
|
playSound(getFallDamageSound((int) dmg), 1.0F, 1.0F);
|
|
hurt(DamageSource.FLY_INTO_WALL, dmg);
|
|
}
|
|
}
|
|
if (onGround) {
|
|
setSharedFlag(7, false);
|
|
}
|
|
}
|
|
|
|
calculateEntityAnimation(LivingEntity.class.cast(this), this instanceof FlyingAnimal);
|
|
info.cancel();
|
|
}
|
|
}
|
|
|
|
private double be_getKnockback(Item tool) {
|
|
if (tool == null) return 0.0D;
|
|
Collection<AttributeModifier> modifiers = tool.getDefaultAttributeModifiers(EquipmentSlot.MAINHAND)
|
|
.get(Attributes.ATTACK_KNOCKBACK);
|
|
if (modifiers.size() > 0) {
|
|
return modifiers.iterator().next().getAmount();
|
|
}
|
|
return 0.0D;
|
|
}
|
|
}
|