ActuallyAdditions/src/main/java/ellpeck/actuallyadditions/misc/LaserRelayConnectionHandler.java

/*
 * 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
 *
 * <EFBFBD> 2015 Ellpeck
 */

package ellpeck.actuallyadditions.misc;

import cofh.api.energy.IEnergyReceiver;
import ellpeck.actuallyadditions.config.values.ConfigIntValues;
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.minecraftforge.common.util.ForgeDirection;

import java.util.ArrayList;

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<ArrayList<ConnectionPair>> networks = new ArrayList<ArrayList<ConnectionPair>>();

    public static LaserRelayConnectionHandler getInstance(){
        return instance;
    }

    public static void setInstance(LaserRelayConnectionHandler i){
        instance = i;
    }

    public void writeNetworkToNBT(ArrayList<ConnectionPair> 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<ConnectionPair> readNetworkFromNBT(NBTTagCompound tag, String name){
        NBTTagCompound compound = tag.getCompoundTag(name);

        int networkSize = compound.getInteger("NetworkSize");

        ArrayList<ConnectionPair> network = new ArrayList<ConnectionPair>();
        for(int pair = 0; pair < networkSize; pair++){
            network.add(ConnectionPair.readFromNBT(compound, "Pair"+pair));
        }

        return network;
    }

    /**
     * Gets all Connections for a Relay
     */
    public ArrayList<ConnectionPair> getConnectionsFor(WorldPos relay){
        ArrayList<ConnectionPair> allPairs = new ArrayList<ConnectionPair>();
        ArrayList<ArrayList<ConnectionPair>> theNetworks = new ArrayList<ArrayList<ConnectionPair>>(this.networks);
        for(ArrayList<ConnectionPair> aNetwork : theNetworks){
            ArrayList<ConnectionPair> pairs = new ArrayList<ConnectionPair>(aNetwork);
            for(ConnectionPair pair : pairs){
                if(pair.contains(relay)){
                    allPairs.add(pair);
                }
            }
        }
        return allPairs;
    }

    /**
     * Removes a Relay from its Network
     */
    public void removeRelayFromNetwork(WorldPos relay){
        ArrayList<ConnectionPair> network = this.getNetworkFor(relay);
        if(network != null){
            //Setup new network (so that splitting a network will cause it to break into two)
            this.networks.remove(network);
            ArrayList<ConnectionPair> pairs = new ArrayList<ConnectionPair>(network);
            for(ConnectionPair pair : pairs){
                if(!pair.contains(relay)){
                    this.addConnection(pair.firstRelay, pair.secondRelay);
                }
            }
        }
        WorldData.makeDirty();
    }

    /**
     * Gets a Network for a Relay
     */
    public ArrayList<ConnectionPair> getNetworkFor(WorldPos relay){
        ArrayList<ArrayList<ConnectionPair>> theNetworks = new ArrayList<ArrayList<ConnectionPair>>(this.networks);
        for(ArrayList<ConnectionPair> aNetwork : theNetworks){
            ArrayList<ConnectionPair> pairs = new ArrayList<ConnectionPair>(aNetwork);
            for(ConnectionPair pair : pairs){
                if(pair.contains(relay)){
                    return aNetwork;
                }
            }
        }
        return null;
    }

    /**
     * Adds a new connection between two relays
     * (Puts it into the correct network!)
     */
    public boolean addConnection(WorldPos firstRelay, WorldPos secondRelay){
        int distance = (int)firstRelay.toVec().distanceTo(secondRelay.toVec());
        if(distance > ConfigIntValues.LASER_RELAY_MAX_DISTANCE.getValue() || firstRelay.isEqual(secondRelay) || firstRelay.getWorld() != secondRelay.getWorld()){
            return false;
        }

        ArrayList<ConnectionPair> firstNetwork = this.getNetworkFor(firstRelay);
        ArrayList<ConnectionPair> secondNetwork = this.getNetworkFor(secondRelay);

        //No Network exists
        if(firstNetwork == null && secondNetwork == null){
            firstNetwork = new ArrayList<ConnectionPair>();
            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;
    }

    /**
     * Merges two networks together
     * (Actually puts everything from the second network into the first one and removes the second one)
     */
    public void mergeNetworks(ArrayList<ConnectionPair> firstNetwork, ArrayList<ConnectionPair> secondNetwork){
        ArrayList<ConnectionPair> pairs = new ArrayList<ConnectionPair>(secondNetwork);
        for(ConnectionPair secondPair : pairs){
            firstNetwork.add(secondPair);
        }
        this.networks.remove(secondNetwork);
        WorldData.makeDirty();
        //System.out.println("Merged Two Networks!");
    }

    public int transferEnergyToReceiverInNeed(ArrayList<ConnectionPair> network, int maxTransfer, boolean simulate){
        int transmitted = 0;
        //Go through all of the connections in the network
        ArrayList<ConnectionPair> pairs = new ArrayList<ConnectionPair>(network);
        for(ConnectionPair pair : pairs){
            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 (with the energy loss!)
                                int theoreticalReceived = ((IEnergyReceiver)tile).receiveEnergy(side.getOpposite(), maxTransfer-transmitted, true);
                                //The amount of energy lost during a transfer
                                int deduct = (int)(theoreticalReceived*((double)ConfigIntValues.LASER_RELAY_LOSS.getValue()/100));

                                transmitted+=((IEnergyReceiver)tile).receiveEnergy(side.getOpposite(), theoreticalReceived-deduct, simulate);
                                transmitted+=deduct;
                            }
                        }
                    }
                }
            }
        }
        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 static ConnectionPair readFromNBT(NBTTagCompound compound, String name){
            WorldPos[] pos = new WorldPos[2];
            for(int i = 0; i < pos.length; 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(compound.getInteger("world"+name+i), anX, aY, aZ);
            }
            return new ConnectionPair(pos[0], pos[1]);
        }

        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.getWorldID());
                compound.setInteger("x"+name+i, relay.getX());
                compound.setInteger("y"+name+i, relay.getY());
                compound.setInteger("z"+name+i, relay.getZ());
            }
        }
    }
}