package com.gearexpansion.block.entity; import com.gearexpansion.GearExpansion; import com.gearexpansion.config.GearExpansionConfig; import net.minecraft.nbt.NbtCompound; import net.minecraft.nbt.NbtElement; import net.minecraft.nbt.NbtList; import net.minecraft.registry.RegistryWrapper; import net.minecraft.server.world.ServerWorld; import net.minecraft.util.math.BlockPos; import java.util.*; public class KineticNetwork { private final UUID id; private final Set members = new HashSet<>(); private final Map edges = new HashMap<>(); private double networkSpeed = 0.0; private double networkStress = 0.0; private boolean hasSource = false; private int sourceTeeth = 16; public record Edge(BlockPos a, BlockPos b, double ratio) {} public KineticNetwork(UUID id) { this.id = id; } public UUID getId() { return id; } public Set getMembers() { return Collections.unmodifiableSet(members); } public void addMember(BlockPos pos) { members.add(pos); rebuildEdges(); } public void removeMember(BlockPos pos) { members.remove(pos); edges.entrySet().removeIf(e -> e.getKey().equals(pos) || e.getValue().a().equals(pos) || e.getValue().b().equals(pos)); } public boolean contains(BlockPos pos) { return members.contains(pos); } private void rebuildEdges() { edges.clear(); List list = new ArrayList<>(members); for (int i = 0; i < list.size(); i++) { BlockPos a = list.get(i); for (int j = i + 1; j < list.size(); j++) { BlockPos b = list.get(j); double dist = distanceBetween(a, b); if (dist <= 4.0 && dist > 0) { edges.put(a, new Edge(a, b, 1.0)); } } } } // Helper: get teeth count for meshing from any block entity private static int getMeshTeeth(ServerWorld world, BlockPos pos) { if (world.getBlockEntity(pos) instanceof GearBlockEntity g) { return g.isDoubleGear() ? g.getLargeTeeth() : g.getTeeth(); } if (world.getBlockEntity(pos) instanceof DoubleGearBlockEntity d) { return d.getMeshTeeth(); } return 16; } // Helper: check if block entity is powered private static boolean isPowered(ServerWorld world, BlockPos pos) { if (world.getBlockEntity(pos) instanceof GearBlockEntity g) return g.isPowered(); if (world.getBlockEntity(pos) instanceof DoubleGearBlockEntity d) return d.isPowered(); return false; } // Helper: set speed on any gear BE private static void setSpeed(ServerWorld world, BlockPos pos, double speed) { if (world.getBlockEntity(pos) instanceof GearBlockEntity g) g.setSpeed(speed); if (world.getBlockEntity(pos) instanceof DoubleGearBlockEntity d) d.setSpeed(speed); } // Helper: get teeth for ratio calculation private static int getTeethForRatio(ServerWorld world, BlockPos pos) { if (world.getBlockEntity(pos) instanceof GearBlockEntity g) { return g.isDoubleGear() ? g.getLargeTeeth() : g.getTeeth(); } if (world.getBlockEntity(pos) instanceof DoubleGearBlockEntity d) { return d.getMeshTeeth(); } return 16; } public void tickNetwork(ServerWorld world) { if (members.isEmpty()) return; double sourceSpeed = GearExpansionConfig.INSTANCE.defaultRPM; int srcTeeth = 16; boolean foundSource = false; edges.clear(); List list = new ArrayList<>(members); for (int i = 0; i < list.size(); i++) { BlockPos a = list.get(i); int teethA = getMeshTeeth(world, a); if (teethA == 16 && !(world.getBlockEntity(a) instanceof GearBlockEntity) && !(world.getBlockEntity(a) instanceof DoubleGearBlockEntity)) { removeMember(a); continue; } if (isPowered(world, a) && !foundSource) { sourceSpeed = GearExpansionConfig.INSTANCE.defaultRPM; srcTeeth = teethA; foundSource = true; hasSource = true; sourceTeeth = srcTeeth; setSpeed(world, a, sourceSpeed); } for (int j = i + 1; j < list.size(); j++) { BlockPos b = list.get(j); int teethB = getMeshTeeth(world, b); if (canMesh(a, teethA, b, teethB)) { double ratio = (double) teethA / teethB; edges.put(a, new Edge(a, b, ratio)); edges.put(b, new Edge(b, a, 1.0 / ratio)); } } } hasSource = foundSource; if (!foundSource) { networkSpeed = 0.0; for (BlockPos pos : members) { setSpeed(world, pos, 0); } return; } // BFS propagation Set visited = new HashSet<>(); Queue queue = new LinkedList<>(); Map speedMap = new HashMap<>(); for (BlockPos pos : members) { if (isPowered(world, pos)) { queue.add(pos); visited.add(pos); speedMap.put(pos, sourceSpeed); } } while (!queue.isEmpty()) { BlockPos current = queue.poll(); double currentSpeed = speedMap.getOrDefault(current, 0.0); Edge edge = edges.get(current); if (edge == null) continue; BlockPos neighbor = edge.a().equals(current) ? edge.b() : edge.a(); if (!visited.contains(neighbor) && members.contains(neighbor)) { visited.add(neighbor); double ratio = (double) getTeethForRatio(world, current) / getTeethForRatio(world, neighbor); double neighborSpeed = currentSpeed * ratio; speedMap.put(neighbor, neighborSpeed); queue.add(neighbor); } } networkStress = 0; for (BlockPos pos : members) { double s = speedMap.getOrDefault(pos, 0.0); setSpeed(world, pos, s); int t = getTeethForRatio(world, pos); double stress = Math.abs(s) * (t / 16.0); if (world.getBlockEntity(pos) instanceof GearBlockEntity g) g.setStress(stress); if (world.getBlockEntity(pos) instanceof DoubleGearBlockEntity d) d.setStress(stress); networkStress += stress; } networkSpeed = speedMap.values().stream().mapToDouble(d -> d).average().orElse(0); } private static boolean canMesh(BlockPos a, int teethA, BlockPos b, int teethB) { double rA = teethA * 0.03125; double rB = teethB * 0.03125; double expectedDist = rA + rB; double actualDist = Math.sqrt(a.getSquaredDistance(b)); return Math.abs(actualDist - expectedDist) < 0.0625; } private static double distanceBetween(BlockPos a, BlockPos b) { return Math.sqrt(a.getSquaredDistance(b)); } public void writeNbt(NbtCompound nbt, RegistryWrapper.WrapperLookup registries) { nbt.putUuid("id", id); nbt.putDouble("speed", networkSpeed); nbt.putDouble("stress", networkStress); nbt.putBoolean("has_source", hasSource); nbt.putInt("source_teeth", sourceTeeth); NbtList memberList = new NbtList(); for (BlockPos pos : members) { NbtCompound posTag = new NbtCompound(); posTag.putInt("x", pos.getX()); posTag.putInt("y", pos.getY()); posTag.putInt("z", pos.getZ()); memberList.add(posTag); } nbt.put("members", memberList); } public static KineticNetwork readNbt(NbtCompound nbt, RegistryWrapper.WrapperLookup registries) { UUID id = nbt.getUuid("id"); KineticNetwork net = new KineticNetwork(id); net.networkSpeed = nbt.getDouble("speed", 0.0); net.networkStress = nbt.getDouble("stress", 0.0); net.hasSource = nbt.getBoolean("has_source", false); net.sourceTeeth = nbt.getInt("source_teeth", 16); NbtList memberList = nbt.getList("members", NbtElement.COMPOUND_TYPE); for (int i = 0; i < memberList.size(); i++) { NbtCompound posTag = memberList.getCompound(i); BlockPos pos = new BlockPos( posTag.getInt("x"), posTag.getInt("y"), posTag.getInt("z") ); net.members.add(pos); } net.rebuildEdges(); return net; } }