package com.militarymod.entity; import com.militarymod.sound.ModSounds; import net.minecraft.entity.Entity; import net.minecraft.entity.EntityDimensions; import net.minecraft.entity.EntityType; import net.minecraft.entity.MovementType; import net.minecraft.entity.damage.DamageSource; import net.minecraft.entity.data.DataTracker; import net.minecraft.entity.data.TrackedData; import net.minecraft.entity.data.TrackedDataHandlerRegistry; import net.minecraft.nbt.NbtCompound; import net.minecraft.server.world.ServerWorld; import net.minecraft.sound.SoundCategory; import net.minecraft.util.math.Vec3d; import net.minecraft.world.World; import java.util.List; public class BomberEntity extends Entity { private static final TrackedData HEALTH = DataTracker.registerData(BomberEntity.class, TrackedDataHandlerRegistry.INTEGER); private static final TrackedData BOMBS = DataTracker.registerData(BomberEntity.class, TrackedDataHandlerRegistry.INTEGER); private static final TrackedData FUEL = DataTracker.registerData(BomberEntity.class, TrackedDataHandlerRegistry.FLOAT); private static final TrackedData BOMB_BAY_OPEN = DataTracker.registerData(BomberEntity.class, TrackedDataHandlerRegistry.BOOLEAN); public static final int MAX_HEALTH = 150; public static final int MAX_BOMBS = 8; public static final float MAX_FUEL = 1200.0f; private static final float MIN_SPEED = 0.4f; private static final float MAX_FLIGHT_SPEED = 1.2f; private static final float PITCH_RATE = 2.5f; private static final float YAW_RATE = 2.5f; private static final int BOMB_COOLDOWN_TICKS = 20; private float inputPitch = 0; private float inputYaw = 0; private float inputThrottle = 1.0f; private boolean inputDropBomb = false; private int bombCooldown = 0; public BomberEntity(EntityType type, World world) { super(type, world); setNoGravity(true); } @Override protected void initDataTracker(DataTracker.Builder builder) { builder.add(HEALTH, MAX_HEALTH); builder.add(BOMBS, MAX_BOMBS); builder.add(FUEL, MAX_FUEL); builder.add(BOMB_BAY_OPEN, false); } public int getHealth() { return dataTracker.get(HEALTH); } public int getBombs() { return dataTracker.get(BOMBS); } public float getFuel() { return dataTracker.get(FUEL); } public boolean isBombBayOpen() { return dataTracker.get(BOMB_BAY_OPEN); } public void setBomberHealth(int h) { dataTracker.set(HEALTH, Math.max(0, Math.min(MAX_HEALTH, h))); } public void setBombs(int b) { dataTracker.set(BOMBS, Math.max(0, Math.min(MAX_BOMBS, b))); } public void setFuel(float f) { dataTracker.set(FUEL, Math.max(0, Math.min(MAX_FUEL, f))); } public void handleSteerPacket(float pitch, float yaw, float throttle, boolean dropBomb) { this.inputPitch = Math.max(-1.0f, Math.min(1.0f, pitch)); this.inputYaw = Math.max(-1.0f, Math.min(1.0f, yaw)); this.inputThrottle = Math.max(0.5f, Math.min(1.5f, throttle)); this.inputDropBomb = dropBomb; } @Override public void tick() { super.tick(); if (bombCooldown > 0) bombCooldown--; if (!getWorld().isClient()) { float fuel = getFuel(); if (fuel <= 0) { setNoGravity(false); } else { setNoGravity(true); setFuel(fuel - 0.5f); setPitch(getPitch() + inputPitch * PITCH_RATE); setYaw(getYaw() + inputYaw * YAW_RATE); setPitch(Math.max(-60.0f, Math.min(60.0f, getPitch()))); float yawRad = (float) Math.toRadians(getYaw()); float pitchRad = (float) Math.toRadians(getPitch()); float speed = MIN_SPEED + (MAX_FLIGHT_SPEED - MIN_SPEED) * (inputThrottle - 0.5f) / 1.0f; speed = Math.max(MIN_SPEED, Math.min(MAX_FLIGHT_SPEED, speed)); double dx = -Math.sin(yawRad) * Math.cos(pitchRad) * speed; double dy = -Math.sin(pitchRad) * speed; double dz = Math.cos(yawRad) * Math.cos(pitchRad) * speed; setVelocity(dx, dy, dz); } move(MovementType.SELF, getVelocity()); if (inputDropBomb && bombCooldown <= 0 && getBombs() > 0) { dropBomb(); setBombs(getBombs() - 1); bombCooldown = BOMB_COOLDOWN_TICKS; dataTracker.set(BOMB_BAY_OPEN, true); } else if (bombCooldown < BOMB_COOLDOWN_TICKS - 5) { dataTracker.set(BOMB_BAY_OPEN, false); } if (getHealth() <= 0) { crashExplode(); discard(); } } } private void dropBomb() { if (!(getWorld() instanceof ServerWorld sw)) return; BombEntity bomb = new BombEntity(ModEntities.BOMB, getWorld()); bomb.setPosition(getX(), getY() - 1.0, getZ()); Vec3d vel = getVelocity(); bomb.setVelocity(vel.x, vel.y - 0.5, vel.z); sw.spawnEntity(bomb); sw.playSound(null, getX(), getY(), getZ(), ModSounds.BOMB_DROP, SoundCategory.HOSTILE, 3.0f, 1.0f); } private void crashExplode() { if (!(getWorld() instanceof ServerWorld sw)) return; sw.createExplosion(this, getX(), getY(), getZ(), 8.0f, World.ExplosionSourceType.TNT); sw.playSound(null, getX(), getY(), getZ(), ModSounds.EXPLOSION_LARGE, SoundCategory.HOSTILE, 5.0f, 0.7f); } @Override public boolean damage(ServerWorld world, DamageSource source, float amount) { if (world.isClient()) return false; setBomberHealth(getHealth() - (int) amount); return true; } @Override public boolean isPushable() { return false; } @Override protected boolean canAddPassenger(Entity passenger) { return getPassengerList().size() < 1; } @Override protected Vec3d getPassengerAttachmentPos(Entity passenger, EntityDimensions dimensions, float scaleFactor) { return new Vec3d(0.0, 0.8 * scaleFactor, 0.5); } @Override public Entity getControllingPassenger() { List passengers = getPassengerList(); return passengers.isEmpty() ? null : passengers.get(0); } @Override protected void readCustomDataFromNbt(NbtCompound nbt) { setBomberHealth(nbt.getInt("health")); setBombs(nbt.getInt("bombs")); setFuel(nbt.getFloat("fuel")); } @Override protected void writeCustomDataToNbt(NbtCompound nbt) { nbt.putInt("health", getHealth()); nbt.putInt("bombs", getBombs()); nbt.putFloat("fuel", getFuel()); } }