superb-warfare/src/main/java/com/atsuishio/superbwarfare/entity/vehicle/Mk42Entity.java
2025-07-16 00:27:29 +08:00

616 lines
24 KiB
Java

package com.atsuishio.superbwarfare.entity.vehicle;
import com.atsuishio.superbwarfare.Mod;
import com.atsuishio.superbwarfare.component.ModDataComponents;
import com.atsuishio.superbwarfare.config.server.ExplosionConfig;
import com.atsuishio.superbwarfare.config.server.VehicleConfig;
import com.atsuishio.superbwarfare.entity.vehicle.base.CannonEntity;
import com.atsuishio.superbwarfare.entity.vehicle.base.RemoteControllableTurret;
import com.atsuishio.superbwarfare.entity.vehicle.base.ThirdPersonCameraPosition;
import com.atsuishio.superbwarfare.entity.vehicle.base.VehicleEntity;
import com.atsuishio.superbwarfare.entity.vehicle.damage.DamageModifier;
import com.atsuishio.superbwarfare.entity.vehicle.weapon.CannonShellWeapon;
import com.atsuishio.superbwarfare.entity.vehicle.weapon.VehicleWeapon;
import com.atsuishio.superbwarfare.event.ClientMouseHandler;
import com.atsuishio.superbwarfare.init.ModDamageTypes;
import com.atsuishio.superbwarfare.init.ModItems;
import com.atsuishio.superbwarfare.init.ModSounds;
import com.atsuishio.superbwarfare.init.ModTags;
import com.atsuishio.superbwarfare.item.ArtilleryIndicator;
import com.atsuishio.superbwarfare.item.common.ammo.CannonShellItem;
import com.atsuishio.superbwarfare.network.message.receive.ShakeClientMessage;
import com.atsuishio.superbwarfare.tools.*;
import net.minecraft.ChatFormatting;
import net.minecraft.core.particles.ParticleTypes;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.network.chat.Component;
import net.minecraft.network.syncher.EntityDataAccessor;
import net.minecraft.network.syncher.EntityDataSerializers;
import net.minecraft.network.syncher.SynchedEntityData;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.server.level.ServerPlayer;
import net.minecraft.sounds.SoundSource;
import net.minecraft.util.Mth;
import net.minecraft.world.InteractionHand;
import net.minecraft.world.InteractionResult;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.entity.EntityType;
import net.minecraft.world.entity.MoverType;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.level.Explosion;
import net.minecraft.world.level.Level;
import net.minecraft.world.phys.Vec2;
import net.minecraft.world.phys.Vec3;
import net.neoforged.api.distmarker.Dist;
import net.neoforged.api.distmarker.OnlyIn;
import net.neoforged.neoforge.event.EventHooks;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.joml.Math;
import org.joml.Matrix4f;
import org.joml.Vector3f;
import org.joml.Vector4f;
import software.bernie.geckolib.animatable.GeoEntity;
import software.bernie.geckolib.animatable.instance.AnimatableInstanceCache;
import software.bernie.geckolib.animation.*;
import software.bernie.geckolib.util.GeckoLibUtil;
import static com.atsuishio.superbwarfare.tools.RangeTool.calculateLaunchVector;
public class Mk42Entity extends VehicleEntity implements GeoEntity, CannonEntity, RemoteControllableTurret {
public static final EntityDataAccessor<Integer> COOL_DOWN = SynchedEntityData.defineId(Mk42Entity.class, EntityDataSerializers.INT);
private final AnimatableInstanceCache cache = GeckoLibUtil.createInstanceCache(this);
public static final EntityDataAccessor<Float> PITCH = SynchedEntityData.defineId(Mk42Entity.class, EntityDataSerializers.FLOAT);
public static final EntityDataAccessor<Float> YAW = SynchedEntityData.defineId(Mk42Entity.class, EntityDataSerializers.FLOAT);
public static final EntityDataAccessor<Boolean> DEPRESSED = SynchedEntityData.defineId(Mk42Entity.class, EntityDataSerializers.BOOLEAN);
public static final EntityDataAccessor<Vector3f> TARGET_POS = SynchedEntityData.defineId(Mk42Entity.class, EntityDataSerializers.VECTOR3);
public static final EntityDataAccessor<Integer> RADIUS = SynchedEntityData.defineId(Mk42Entity.class, EntityDataSerializers.INT);
public Mk42Entity(EntityType<Mk42Entity> type, Level world) {
super(type, world);
}
@Override
protected void defineSynchedData(SynchedEntityData.Builder builder) {
super.defineSynchedData(builder);
builder.define(COOL_DOWN, 0)
.define(PITCH, 0F)
.define(YAW, 0F)
.define(DEPRESSED, false)
.define(TARGET_POS, new Vector3f())
.define(RADIUS, 0);
}
@Override
public VehicleWeapon[][] initWeapons() {
return new VehicleWeapon[][]{
new VehicleWeapon[]{
new CannonShellWeapon()
.hitDamage(VehicleConfig.MK42_AP_DAMAGE.get())
.explosionDamage(VehicleConfig.MK42_AP_EXPLOSION_DAMAGE.get())
.explosionRadius(VehicleConfig.MK42_AP_EXPLOSION_RADIUS.get().floatValue())
.durability(60)
.gravity(-projectileGravity())
.sound(ModSounds.CANNON_RELOAD.get())
.icon(Mod.loc("textures/screens/vehicle_weapon/ap_shell.png")),
new CannonShellWeapon()
.hitDamage(VehicleConfig.MK42_HE_DAMAGE.get())
.explosionDamage(VehicleConfig.MK42_HE_EXPLOSION_DAMAGE.get())
.explosionRadius(VehicleConfig.MK42_HE_EXPLOSION_RADIUS.get().floatValue())
.durability(1)
.fireProbability(0.18F)
.fireTime(2)
.gravity(-projectileGravity())
.sound(ModSounds.CANNON_RELOAD.get())
.icon(Mod.loc("textures/screens/vehicle_weapon/he_shell.png")),
}
};
}
@Override
public ThirdPersonCameraPosition getThirdPersonCameraPosition(int index) {
return new ThirdPersonCameraPosition(8 + ClientMouseHandler.custom3pDistanceLerp, 1, 0);
}
@Override
public void addAdditionalSaveData(CompoundTag compound) {
super.addAdditionalSaveData(compound);
compound.putInt("CoolDown", this.entityData.get(COOL_DOWN));
compound.putFloat("Pitch", this.entityData.get(PITCH));
compound.putFloat("Yaw", this.entityData.get(YAW));
compound.putBoolean("Depressed", this.entityData.get(DEPRESSED));
compound.putInt("Radius", this.entityData.get(RADIUS));
compound.putFloat("TargetX", this.entityData.get(TARGET_POS).x);
compound.putFloat("TargetY", this.entityData.get(TARGET_POS).y);
compound.putFloat("TargetZ", this.entityData.get(TARGET_POS).z);
}
@Override
protected void readAdditionalSaveData(CompoundTag compound) {
super.readAdditionalSaveData(compound);
this.entityData.set(COOL_DOWN, compound.getInt("CoolDown"));
this.entityData.set(PITCH, compound.getFloat("Pitch"));
this.entityData.set(YAW, compound.getFloat("Yaw"));
if (compound.contains("Depressed")) {
this.entityData.set(DEPRESSED, compound.getBoolean("Depressed"));
}
if (compound.contains("Radius")) {
this.entityData.set(RADIUS, compound.getInt("Radius"));
}
if (compound.contains("TargetX") && compound.contains("TargetY") && compound.contains("TargetZ")) {
this.entityData.set(TARGET_POS, new Vector3f(compound.getFloat("TargetX"), compound.getFloat("TargetX"), compound.getFloat("TargetZ")));
}
}
@Override
public @NotNull InteractionResult interact(Player player, @NotNull InteractionHand hand) {
ItemStack stack = player.getMainHandItem();
if (stack.getItem() instanceof ArtilleryIndicator indicator) {
return indicator.bind(stack, player, this);
}
if (stack.is(ModTags.Items.CROWBAR) && !player.isShiftKeyDown()) {
if (this.items.getFirst().getItem() instanceof CannonShellItem) {
var weaponType = this.items.getFirst().is(ModItems.AP_5_INCHES.get()) ? 0 : 1;
setWeaponIndex(0, weaponType);
vehicleShoot(player, 0);
}
return InteractionResult.SUCCESS;
}
if (stack.getItem() instanceof CannonShellItem) {
if (this.entityData.get(COOL_DOWN) == 0 && this.items.getFirst().isEmpty()) {
this.setItem(0, stack.copyWithCount(1));
if (!player.isCreative()) {
stack.shrink(1);
}
if (player instanceof ServerPlayer serverPlayer) {
SoundTool.playLocalSound(serverPlayer, ModSounds.CANNON_RELOAD.get(), 2, 1);
}
}
return InteractionResult.SUCCESS;
}
if (player.getMainHandItem().getItem() == ModItems.FIRING_PARAMETERS.get()) {
setTarget(player.getMainHandItem(), player);
}
if (player.getOffhandItem().getItem() == ModItems.FIRING_PARAMETERS.get()) {
setTarget(player.getMainHandItem(), player);
}
return super.interact(player, hand);
}
@Override
public void setTarget(ItemStack stack, Entity entity) {
var parameters = stack.get(ModDataComponents.FIRING_PARAMETERS);
if (parameters == null) return;
var pos = parameters.pos();
int targetX = pos.getX();
int targetY = pos.getY();
int targetZ = pos.getZ();
var isDepressed = parameters.isDepressed();
var radius = parameters.radius();
boolean canAim = true;
entityData.set(TARGET_POS, new Vector3f((float) targetX, (float) targetY, (float) targetZ));
entityData.set(DEPRESSED, isDepressed);
entityData.set(RADIUS, radius);
Vec3 randomPos = VectorTool.randomPos(new Vec3(entityData.get(TARGET_POS)), entityData.get(RADIUS));
Vec3 launchVector = calculateLaunchVector(getEyePosition(), randomPos, shootVelocity(), projectileGravity(), entityData.get(DEPRESSED));
Component component = Component.literal("");
Component location = Component.translatable("tips.superbwarfare.mortar.position", this.getDisplayName())
.append(Component.literal(" X:" + FormatTool.format0D(getX()) + " Y:" + FormatTool.format0D(getY()) + " Z:" + FormatTool.format0D(getZ()) + " "));
float angle = getXRot();
if (launchVector == null) {
canAim = false;
component = Component.translatable("tips.superbwarfare.mortar.out_of_range");
} else {
angle = (float) -getXRotFromVector(launchVector);
if (angle < -maxPitch() || angle > -minPitch()) {
canAim = false;
component = Component.translatable("tips.superbwarfare.mortar.warn", this.getDisplayName());
if (angle < -maxPitch()) {
component = Component.translatable("tips.superbwarfare.ballistics.warn");
}
}
}
if (canAim) {
this.look(randomPos);
entityData.set(PITCH, angle);
} else if (entity instanceof Player player) {
player.displayClientMessage(location.copy().append(component).withStyle(ChatFormatting.RED), false);
}
}
@Override
public void resetTarget() {
Vec3 randomPos = VectorTool.randomPos(new Vec3(entityData.get(TARGET_POS)), entityData.get(RADIUS));
Vec3 launchVector = calculateLaunchVector(getEyePosition(), randomPos, 15, projectileGravity(), entityData.get(DEPRESSED));
this.look(randomPos);
if (launchVector == null) {
return;
}
float angle = (float) -getXRotFromVector(launchVector);
if (angle < -85 || angle > 14.9) {
entityData.set(PITCH, angle);
}
}
@Override
public double minPitch() {
return -14.9;
}
@Override
public double maxPitch() {
return 85;
}
@Override
public double shootVelocity() {
return 15;
}
@Override
public float projectileGravity() {
return -0.1f;
}
@Override
public void look(Vec3 pTarget) {
Matrix4f transform = getVehicleFlatTransform(1);
Vector4f worldPosition = transformPosition(transform, 0f, 2.16f, 0.5175f);
Vec3 shootPos = new Vec3(worldPosition.x, worldPosition.y, worldPosition.z);
double d0 = pTarget.x - shootPos.x;
double d2 = pTarget.z - shootPos.z;
entityData.set(YAW, Mth.wrapDegrees((float) (Mth.atan2(d2, d0) * 57.2957763671875) - 90.0F));
}
@Override
public DamageModifier getDamageModifier() {
return super.getDamageModifier()
.custom((source, damage) -> getSourceAngle(source, 1f) * damage);
}
@Override
public @NotNull Vec3 getDeltaMovement() {
return new Vec3(0, Math.min(super.getDeltaMovement().y, 0), 0);
}
@Override
public void baseTick() {
super.baseTick();
if (this.entityData.get(COOL_DOWN) > 0) {
this.entityData.set(COOL_DOWN, this.entityData.get(COOL_DOWN) - 1);
}
this.move(MoverType.SELF, this.getDeltaMovement());
if (this.onGround()) {
this.setDeltaMovement(Vec3.ZERO);
} else {
this.setDeltaMovement(this.getDeltaMovement().add(0.0, -0.04, 0.0));
}
lowHealthWarning();
}
@Override
public void handleClientSync() {
if (isControlledByLocalInstance()) {
interpolationSteps = 0;
syncPacketPositionCodec(getX(), getY(), getZ());
}
if (interpolationSteps <= 0) {
return;
}
double interpolatedYaw = Mth.wrapDegrees(serverYRot - (double) getYRot());
setYRot(getYRot() + (float) interpolatedYaw / (float) interpolationSteps);
setXRot(getXRot() + (float) (serverXRot - (double) getXRot()) / (float) interpolationSteps);
setRot(getYRot(), getXRot());
}
@Override
public void lerpTo(double x, double y, double z, float yaw, float pitch, int interpolationSteps) {
serverYRot = yaw;
serverXRot = pitch;
this.interpolationSteps = 10;
}
@Override
public void destroy() {
if (level() instanceof ServerLevel) {
CustomExplosion explosion = new CustomExplosion(this.level(), this,
ModDamageTypes.causeCustomExplosionDamage(this.level().registryAccess(), getAttacker(), getAttacker()), 100f,
this.getX(), this.getY(), this.getZ(), 7f, ExplosionConfig.EXPLOSION_DESTROY.get() ? Explosion.BlockInteraction.DESTROY : Explosion.BlockInteraction.KEEP, true).setDamageMultiplier(1);
explosion.explode();
EventHooks.onExplosionStart(this.level(), explosion);
explosion.finalizeExplosion(false);
ParticleTool.spawnMediumExplosionParticles(this.level(), this.position());
}
explodePassengers();
super.destroy();
}
@Override
public void positionRider(@NotNull Entity passenger, @NotNull MoveFunction callback) {
if (!this.hasPassenger(passenger)) {
return;
}
Matrix4f transform = getVehicleFlatTransform(1);
float x = 0f;
float y = 2.3f;
float z = 0f;
Vector4f worldPosition = transformPosition(transform, x, y, z);
passenger.setPos(worldPosition.x, worldPosition.y, worldPosition.z);
callback.accept(passenger, worldPosition.x, worldPosition.y, worldPosition.z);
}
@Override
public Vec3 driverZoomPos(float ticks) {
Matrix4f transform = getVehicleFlatTransform(1);
Vector4f worldPosition = transformPosition(transform, 0f, 2.16f + 1.4f, 0.5175f);
return new Vec3(worldPosition.x, worldPosition.y, worldPosition.z);
}
@Override
public boolean canRemoteFire() {
return this.getItem(0).getItem() instanceof CannonShellItem && this.entityData.get(COOL_DOWN) == 0;
}
@Override
public void remoteFire(Player player) {
this.setWeaponIndex(0, this.getItem(0).is(ModItems.AP_5_INCHES.get()) ? 0 : 1);
this.vehicleShoot(player, 0);
}
@Override
public void vehicleShoot(Player player, int type) {
if (this.entityData.get(COOL_DOWN) > 0) return;
Level level = player.level();
if (level instanceof ServerLevel server) {
if (!InventoryTool.hasCreativeAmmoBox(player) && player == getFirstPassenger()) {
var ammo = getWeaponIndex(0) == 0 ? ModItems.AP_5_INCHES.get() : ModItems.HE_5_INCHES.get();
var ammoCount = InventoryTool.countItem(player.getInventory().items, ammo);
if (ammoCount <= 0) return;
InventoryTool.consumeItem(player.getInventory().items, ammo, 1);
}
if (getFirstPassenger() != player) {
this.clearContent();
}
var entityToSpawn = ((CannonShellWeapon) getWeapon(0)).create(player);
Matrix4f transform = getVehicleFlatTransform(1);
Vector4f worldPosition = transformPosition(transform, 0f, 2.16f, 0.5175f);
entityToSpawn.setPos(worldPosition.x, worldPosition.y, worldPosition.z);
entityToSpawn.shoot(getLookAngle().x, getLookAngle().y, getLookAngle().z, 15, 0.05f);
level.addFreshEntity(entityToSpawn);
if (player instanceof ServerPlayer serverPlayer) {
if (player == getFirstPassenger()) {
SoundTool.playLocalSound(serverPlayer, ModSounds.MK_42_FIRE_1P.get(), 2, 1);
SoundTool.playLocalSound(serverPlayer, ModSounds.CANNON_RELOAD.get(), 2, 1);
}
}
if (!this.level().isClientSide()) {
this.level().playSound(null, this.getX(), this.getY(), this.getZ(), ModSounds.MK_42_FIRE_3P.get(), SoundSource.PLAYERS, 24f, 1f);
this.level().playSound(null, this.getX(), this.getY(), this.getZ(), ModSounds.MK_42_FAR.get(), SoundSource.PLAYERS, 48f, 1f);
this.level().playSound(null, this.getX(), this.getY(), this.getZ(), ModSounds.MK_42_VERYFAR.get(), SoundSource.PLAYERS, 96f, 1f);
}
this.entityData.set(COOL_DOWN, 30);
server.sendParticles(ParticleTypes.CAMPFIRE_COSY_SMOKE,
this.getX() + 5 * this.getLookAngle().x,
this.getY(),
this.getZ() + 5 * this.getLookAngle().z,
100, 7, 0.02, 7, 0.005);
double x = worldPosition.x + 9 * this.getLookAngle().x;
double y = worldPosition.y + 9 * this.getLookAngle().y;
double z = worldPosition.z + 9 * this.getLookAngle().z;
server.sendParticles(ParticleTypes.CAMPFIRE_COSY_SMOKE, x, y, z, 10, 0.4, 0.4, 0.4, 0.0075);
server.sendParticles(ParticleTypes.CLOUD, x, y, z, 10, 0.4, 0.4, 0.4, 0.0075);
int count = 6;
for (float i = 9.5f; i < 16; i += .5f) {
server.sendParticles(ParticleTypes.CAMPFIRE_COSY_SMOKE,
this.getX() + i * this.getLookAngle().x,
this.getEyeY() + i * this.getLookAngle().y,
this.getZ() + i * this.getLookAngle().z,
Mth.clamp(count--, 1, 5), 0.15, 0.15, 0.15, 0.0025);
}
ShakeClientMessage.sendToNearbyPlayers(this, 20, 15, 15, 45);
resetTarget();
}
}
@Override
public void travel() {
Entity passenger = this.getFirstPassenger();
if (passenger != null) {
entityData.set(YAW, passenger.getYHeadRot());
entityData.set(PITCH, passenger.getXRot() - 2f);
}
float diffY = Mth.wrapDegrees(entityData.get(YAW) - this.getYRot());
float diffX = Mth.wrapDegrees(entityData.get(PITCH) - this.getXRot());
turretTurnSound(diffX, diffY, 0.95f);
this.setYRot(this.getYRot() + Mth.clamp(0.5f * diffY, -1.75f, 1.75f));
this.setXRot(Mth.clamp(this.getXRot() + Mth.clamp(0.5f * diffX, -3f, 3f), -85, 15f));
}
protected void clampRotation(Entity entity) {
float f = Mth.wrapDegrees(entity.getXRot());
float f1 = Mth.clamp(f, -85.0F, 17F);
entity.xRotO += f1 - f;
entity.setXRot(entity.getXRot() + f1 - f);
}
@Override
public void onPassengerTurned(@NotNull Entity entity) {
this.clampRotation(entity);
}
private PlayState movementPredicate(AnimationState<Mk42Entity> event) {
if (this.entityData.get(COOL_DOWN) > 0) {
return event.setAndContinue(RawAnimation.begin().thenPlay("animation.mk42.fire"));
}
return event.setAndContinue(RawAnimation.begin().thenLoop("animation.mk42.idle"));
}
@Override
public void registerControllers(AnimatableManager.ControllerRegistrar data) {
data.add(new AnimationController<>(this, "movement", 0, this::movementPredicate));
}
@Override
public AnimatableInstanceCache getAnimatableInstanceCache() {
return this.cache;
}
@Override
public int mainGunRpm(Player player) {
return 0;
}
@Override
public boolean canShoot(Player player) {
return true;
}
@Override
public int getAmmoCount(Player player) {
var ammo = getWeaponIndex(0) == 0 ? ModItems.AP_5_INCHES.get() : ModItems.HE_5_INCHES.get();
return InventoryTool.countItem(player.getInventory().items, ammo);
}
@Override
public boolean hidePassenger(Entity entity) {
return true;
}
@Override
public int zoomFov() {
return 5;
}
@Override
public int getWeaponHeat(Player player) {
return 0;
}
@Override
public Vec3 getBarrelVector(float pPartialTicks) {
if (getFirstPassenger() != null) {
return getFirstPassenger().getViewVector(pPartialTicks);
}
return super.getBarrelVector(pPartialTicks);
}
@Override
public ResourceLocation getVehicleIcon() {
return Mod.loc("textures/vehicle_icon/sherman_icon.png");
}
@Override
public double getSensitivity(double original, boolean zoom, int seatIndex, boolean isOnGround) {
return zoom ? 0.15 : 0.3;
}
@Override
public boolean isEnclosed(int index) {
return true;
}
@OnlyIn(Dist.CLIENT)
@Override
public @Nullable Vec2 getCameraRotation(float partialTicks, Player player, boolean zoom, boolean isFirstPerson) {
if (zoom || isFirstPerson) {
return new Vec2(Mth.lerp(partialTicks, player.yRotO, player.getYRot()), Mth.lerp(partialTicks, player.xRotO, player.getXRot()));
}
return super.getCameraRotation(partialTicks, player, false, false);
}
@OnlyIn(Dist.CLIENT)
@Override
public Vec3 getCameraPosition(float partialTicks, Player player, boolean zoom, boolean isFirstPerson) {
if (zoom || isFirstPerson) {
if (zoom) {
return new Vec3(this.driverZoomPos(partialTicks).x, this.driverZoomPos(partialTicks).y, this.driverZoomPos(partialTicks).z);
} else {
return new Vec3(Mth.lerp(partialTicks, player.xo, player.getX()), Mth.lerp(partialTicks, player.yo + player.getEyeHeight(), player.getEyeY()), Mth.lerp(partialTicks, player.zo, player.getZ()));
}
}
return super.getCameraPosition(partialTicks, player, false, false);
}
@OnlyIn(Dist.CLIENT)
public boolean useFixedCameraPos(Entity entity) {
return true;
}
@Override
public @Nullable ResourceLocation getVehicleItemIcon() {
return Mod.loc("textures/gui/vehicle/type/defense.png");
}
@Override
public int getContainerSize() {
return 1;
}
@Override
public int getMaxStackSize() {
return 1;
}
@Override
public void setChanged() {
// if (!entityData.get(INTELLIGENT)) {
// fire(null);
// }
}
@Override
public boolean stillValid(@NotNull Player player) {
return false;
}
@Override
public boolean canPlaceItem(int slot, @NotNull ItemStack stack) {
return super.canPlaceItem(slot, stack) && this.entityData.get(COOL_DOWN) == 0 && stack.getItem() instanceof CannonShellItem;
}
}