/* * This file ("LaserRelayConnectionHandler.java") is part of the Actually Additions Mod for Minecraft. * It is created and owned by Ellpeck and distributed * under the Actually Additions License to be found at * http://github.com/Ellpeck/ActuallyAdditions/blob/master/README.md * View the source code at https://github.com/Ellpeck/ActuallyAdditions * * © 2015 Ellpeck */ package ellpeck.actuallyadditions.misc; import cofh.api.energy.IEnergyReceiver; import ellpeck.actuallyadditions.tile.TileEntityLaserRelay; import ellpeck.actuallyadditions.util.WorldPos; import ellpeck.actuallyadditions.util.WorldUtil; import net.minecraft.nbt.NBTTagCompound; import net.minecraft.tileentity.TileEntity; import net.minecraft.util.Vec3; import net.minecraft.world.World; import net.minecraftforge.common.DimensionManager; import net.minecraftforge.common.util.ForgeDirection; import java.util.ArrayList; import java.util.Iterator; public class LaserRelayConnectionHandler{ private static LaserRelayConnectionHandler instance; /** * An ArrayList of all of the networks a world has * (Every place contains an ArrayList of ConnectionPairs, that is a single network!) */ public ArrayList> networks = new ArrayList>(); public static LaserRelayConnectionHandler getInstance(){ return instance; } public static void setInstance(LaserRelayConnectionHandler i){ instance = i; } public void writeNetworkToNBT(ArrayList network, NBTTagCompound tag, String name){ NBTTagCompound compound = new NBTTagCompound(); compound.setInteger("NetworkSize", network.size()); for(int pair = 0; pair < network.size(); pair++){ network.get(pair).writeToNBT(compound, "Pair"+pair); } tag.setTag(name, compound); } public ArrayList readNetworkFromNBT(NBTTagCompound tag, String name){ NBTTagCompound compound = tag.getCompoundTag(name); int networkSize = compound.getInteger("NetworkSize"); ArrayList network = new ArrayList(); for(int pair = 0; pair < networkSize; pair++){ network.add(ConnectionPair.readFromNBT(compound, "Pair"+pair)); } return network; } /** * Gets a Network for a Relay */ public ArrayList getNetworkFor(WorldPos relay){ for(ArrayList aNetwork : this.networks){ for(ConnectionPair pair : aNetwork){ if(pair.contains(relay)){ return aNetwork; } } } return null; } /** * Gets all Connections for a Relay */ public ArrayList getConnectionsFor(WorldPos relay){ ArrayList allPairs = new ArrayList(); for(ArrayList aNetwork : this.networks){ for(ConnectionPair pair : aNetwork){ if(pair.contains(relay)){ allPairs.add(pair); } } } return allPairs; } /** * Adds a new connection between two relays * (Puts it into the correct network!) */ public boolean addConnection(WorldPos firstRelay, WorldPos secondRelay){ int distance = (int)Vec3.createVectorHelper(firstRelay.getX(), firstRelay.getY(), firstRelay.getZ()).distanceTo(Vec3.createVectorHelper(secondRelay.getX(), secondRelay.getY(), secondRelay.getZ())); if(distance > 15 || firstRelay.isEqual(secondRelay) || firstRelay.getWorld() != secondRelay.getWorld()){ return false; } ArrayList firstNetwork = this.getNetworkFor(firstRelay); ArrayList secondNetwork = this.getNetworkFor(secondRelay); //No Network exists if(firstNetwork == null && secondNetwork == null){ firstNetwork = new ArrayList(); this.networks.add(firstNetwork); firstNetwork.add(new ConnectionPair(firstRelay, secondRelay)); } //The same Network else if(firstNetwork == secondNetwork){ return false; } //Both relays have networks else if(firstNetwork != null && secondNetwork != null){ this.mergeNetworks(firstNetwork, secondNetwork); firstNetwork.add(new ConnectionPair(firstRelay, secondRelay)); } //Only first network exists else if(firstNetwork != null){ firstNetwork.add(new ConnectionPair(firstRelay, secondRelay)); } //Only second network exists else if(secondNetwork != null){ secondNetwork.add(new ConnectionPair(firstRelay, secondRelay)); } WorldData.makeDirty(); //System.out.println("Connected "+firstRelay.toString()+" to "+secondRelay.toString()); //System.out.println(firstNetwork == null ? secondNetwork.toString() : firstNetwork.toString()); return true; } /** * Removes a Relay from its Network */ public void removeRelayFromNetwork(WorldPos relay){ ArrayList network = this.getNetworkFor(relay); if(network != null){ //Remove the relay from the network Iterator iterator = network.iterator(); while(iterator.hasNext()){ ConnectionPair next = iterator.next(); if(next.contains(relay)){ iterator.remove(); //System.out.println("Removed "+relay.toString()+" from Network "+network.toString()); } } //Setup new network (so that splitting a network will cause it to break into two) this.networks.remove(network); for(ConnectionPair pair : network){ this.addConnection(pair.firstRelay, pair.secondRelay); } } WorldData.makeDirty(); } /** * Merges two networks together * (Actually puts everything from the second network into the first one and removes the second one) */ public void mergeNetworks(ArrayList firstNetwork, ArrayList secondNetwork){ for(ConnectionPair secondPair : secondNetwork){ firstNetwork.add(secondPair); } this.networks.remove(secondNetwork); WorldData.makeDirty(); //System.out.println("Merged Two Networks!"); } public int transferEnergyToReceiverInNeed(ArrayList network, int maxTransfer, boolean simulate){ int transmitted = 0; //Go through all of the connections in the network for(ConnectionPair pair : network){ WorldPos[] relays = new WorldPos[]{pair.firstRelay, pair.secondRelay}; //Go through both relays in the connection for(WorldPos relay : relays){ if(relay != null){ //Get every side of the relay for(int i = 0; i <= 5; i++){ ForgeDirection side = ForgeDirection.getOrientation(i); //Get the TileEntity at the side TileEntity tile = WorldUtil.getTileEntityFromSide(side, relay.getWorld(), relay.getX(), relay.getY(), relay.getZ()); if(tile instanceof IEnergyReceiver && !(tile instanceof TileEntityLaserRelay)){ IEnergyReceiver receiver = (IEnergyReceiver)tile; if(receiver.canConnectEnergy(side.getOpposite())){ //Transfer the energy transmitted += ((IEnergyReceiver)tile).receiveEnergy(side.getOpposite(), maxTransfer-transmitted, simulate); } } } } } } return transmitted; } public static class ConnectionPair{ public WorldPos firstRelay; public WorldPos secondRelay; public ConnectionPair(WorldPos firstRelay, WorldPos secondRelay){ this.firstRelay = firstRelay; this.secondRelay = secondRelay; } public boolean contains(WorldPos relay){ return (this.firstRelay != null && this.firstRelay.isEqual(relay)) || (this.secondRelay != null && this.secondRelay.isEqual(relay)); } @Override public String toString(){ return (this.firstRelay == null ? "-" : this.firstRelay.toString())+" | "+(this.secondRelay == null ? "-" : this.secondRelay.toString()); } public void writeToNBT(NBTTagCompound compound, String name){ for(int i = 0; i < 2; i++){ WorldPos relay = i == 0 ? this.firstRelay : this.secondRelay; compound.setInteger("world"+name+i, relay.getWorld().provider.dimensionId); compound.setInteger("x"+name+i, relay.getX()); compound.setInteger("y"+name+i, relay.getY()); compound.setInteger("z"+name+i, relay.getZ()); } } public static ConnectionPair readFromNBT(NBTTagCompound compound, String name){ WorldPos[] pos = new WorldPos[2]; for(int i = 0; i < pos.length; i++){ World aWorld = DimensionManager.getWorld(compound.getInteger("world"+name+i)); int anX = compound.getInteger("x"+name+i); int aY = compound.getInteger("y"+name+i); int aZ = compound.getInteger("z"+name+i); pos[i] = new WorldPos(aWorld, anX, aY, aZ); } return new ConnectionPair(pos[0], pos[1]); } } }