package com.medievalrts.regiment; import com.medievalrts.entity.soldier.Faction; import com.medievalrts.entity.soldier.SoldierEntity; import net.minecraft.nbt.NbtCompound; import net.minecraft.nbt.NbtElement; import net.minecraft.nbt.NbtList; import net.minecraft.server.world.ServerWorld; import net.minecraft.util.math.Vec3d; import java.util.ArrayList; import java.util.List; import java.util.UUID; public class Regiment { private final UUID id; private String name; private final List unitIds; private Formation formation; private int color; private Faction faction; public Regiment(String name, int color, Faction faction) { this.id = UUID.randomUUID(); this.name = name; this.unitIds = new ArrayList<>(); this.formation = Formation.LINE; this.color = color; this.faction = faction; } public Regiment(UUID id, String name, int color, Faction faction) { this.id = id; this.name = name; this.unitIds = new ArrayList<>(); this.formation = Formation.LINE; this.color = color; this.faction = faction; } public UUID getId() { return id; } public String getName() { return name; } public void setName(String name) { this.name = name; } public List getUnitIds() { return unitIds; } public void addUnit(SoldierEntity soldier) { unitIds.add(soldier.getUuid()); soldier.setRegiment(this); soldier.setFaction(faction); soldier.setFormationIndex(unitIds.size() - 1); } public void removeUnit(UUID unitId) { unitIds.remove(unitId); } public Formation getFormation() { return formation; } public void setFormation(Formation formation) { this.formation = formation; } public int getColor() { return color; } public void setColor(int color) { this.color = color; } public Faction getFaction() { return faction; } public void moveTo(ServerWorld world, Vec3d target) { List soldiers = getAliveSoldiers(world); List positions = formation.calculatePositions(target, soldiers.size()); for (int i = 0; i < soldiers.size(); i++) { if (i < positions.size()) { soldiers.get(i).setMoveTarget(positions.get(i)); } } } public List getAliveSoldiers(ServerWorld world) { List soldiers = new ArrayList<>(); List toRemove = new ArrayList<>(); for (UUID unitId : unitIds) { if (world.getEntity(unitId) instanceof SoldierEntity soldier && soldier.isAlive()) { soldiers.add(soldier); } else { toRemove.add(unitId); } } unitIds.removeAll(toRemove); return soldiers; } public NbtCompound toNbt() { NbtCompound nbt = new NbtCompound(); nbt.putUuid("Id", id); nbt.putString("Name", name); nbt.putString("Formation", formation.name()); nbt.putInt("Color", color); nbt.putString("Faction", faction.name()); NbtList unitList = new NbtList(); for (UUID unitId : unitIds) { NbtCompound unitNbt = new NbtCompound(); unitNbt.putUuid("UnitId", unitId); unitList.add(unitNbt); } nbt.put("Units", unitList); return nbt; } public static Regiment fromNbt(NbtCompound nbt) { UUID id = nbt.getUuid("Id"); String name = nbt.getString("Name"); int color = nbt.getInt("Color"); Faction faction; try { faction = Faction.valueOf(nbt.getString("Faction")); } catch (IllegalArgumentException e) { faction = Faction.FRIENDLY; } Regiment regiment = new Regiment(id, name, color, faction); try { regiment.formation = Formation.valueOf(nbt.getString("Formation")); } catch (IllegalArgumentException e) { regiment.formation = Formation.LINE; } NbtList unitList = nbt.getList("Units", NbtElement.COMPOUND_TYPE); for (int i = 0; i < unitList.size(); i++) { NbtCompound unitNbt = unitList.getCompound(i); regiment.unitIds.add(unitNbt.getUuid("UnitId")); } return regiment; } }