ActuallyAdditions/src/main/java/de/ellpeck/actuallyadditions/mod/tile/TileEntityLaserRelayEnergy.java
2019-05-02 03:10:29 -04:00

300 lines
12 KiB
Java

/*
* This file ("TileEntityLaserRelayEnergy.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://ellpeck.de/actaddlicense
* View the source code at https://github.com/Ellpeck/ActuallyAdditions
*
* © 2015-2017 Ellpeck
*/
package de.ellpeck.actuallyadditions.mod.tile;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import de.ellpeck.actuallyadditions.api.laser.IConnectionPair;
import de.ellpeck.actuallyadditions.api.laser.LaserType;
import de.ellpeck.actuallyadditions.api.laser.Network;
import de.ellpeck.actuallyadditions.mod.ActuallyAdditions;
import de.ellpeck.actuallyadditions.mod.config.values.ConfigBoolValues;
import de.ellpeck.actuallyadditions.mod.util.StringUtil;
import it.unimi.dsi.fastutil.objects.ObjectOpenHashSet;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.tileentity.TileEntity;
import net.minecraft.util.EnumFacing;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.MathHelper;
import net.minecraft.util.text.TextFormatting;
import net.minecraftforge.energy.CapabilityEnergy;
import net.minecraftforge.energy.IEnergyStorage;
import net.minecraftforge.fml.relauncher.Side;
import net.minecraftforge.fml.relauncher.SideOnly;
public class TileEntityLaserRelayEnergy extends TileEntityLaserRelay {
public static final int CAP = 1000;
public final ConcurrentHashMap<EnumFacing, TileEntity> receiversAround = new ConcurrentHashMap<>();
private final IEnergyStorage[] energyStorages = new IEnergyStorage[6];
private Mode mode = Mode.BOTH;
public TileEntityLaserRelayEnergy(String name) {
super(name, LaserType.ENERGY);
for (int i = 0; i < this.energyStorages.length; i++) {
final EnumFacing facing = EnumFacing.values()[i];
this.energyStorages[i] = new IEnergyStorage() {
@Override
public int receiveEnergy(int amount, boolean simulate) {
return TileEntityLaserRelayEnergy.this.transmitEnergy(facing, amount, simulate);
}
@Override
public int extractEnergy(int maxExtract, boolean simulate) {
return 0;
}
@Override
public int getEnergyStored() {
return 0;
}
@Override
public int getMaxEnergyStored() {
return TileEntityLaserRelayEnergy.this.getEnergyCap();
}
@Override
public boolean canExtract() {
return false;
}
@Override
public boolean canReceive() {
return true;
}
};
}
}
public TileEntityLaserRelayEnergy() {
this("laserRelay");
}
private int transmitEnergy(EnumFacing from, int maxTransmit, boolean simulate) {
int transmitted = 0;
if (maxTransmit > 0 && this.mode != Mode.OUTPUT_ONLY) {
Network network = this.getNetwork();
if (network != null) {
transmitted = this.transferEnergyToReceiverInNeed(from, network, maxTransmit, simulate);
}
}
return transmitted;
}
@Override
public IEnergyStorage getEnergyStorage(EnumFacing facing) {
return this.energyStorages[facing == null ? 0 : facing.ordinal()];
}
@Override
public boolean shouldSaveDataOnChangeOrWorldStart() {
return true;
}
@Override
public void saveDataOnChangeOrWorldStart() {
Map<EnumFacing, TileEntity> old = new HashMap<>(this.receiversAround);
boolean change = false;
this.receiversAround.clear();
for (EnumFacing side : EnumFacing.values()) {
BlockPos pos = this.getPos().offset(side);
if (this.world.isBlockLoaded(pos)) {
TileEntity tile = this.world.getTileEntity(pos);
if (tile != null && !(tile instanceof TileEntityLaserRelay)) {
if (tile.hasCapability(CapabilityEnergy.ENERGY, side.getOpposite())) {
this.receiversAround.put(side, tile);
TileEntity oldTile = old.get(side);
if (oldTile == null || !tile.equals(oldTile)) {
change = true;
}
}
}
}
}
if (change || old.size() != this.receiversAround.size()) {
Network network = this.getNetwork();
if (network != null) {
network.changeAmount++;
}
}
}
private int transferEnergyToReceiverInNeed(EnumFacing from, Network network, int maxTransfer, boolean simulate) {
int transmitted = 0;
//Keeps track of all the Laser Relays and Energy Acceptors that have been checked already to make nothing run multiple times
Set<BlockPos> alreadyChecked = new ObjectOpenHashSet<>();
Set<TileEntityLaserRelayEnergy> relaysThatWork = new ObjectOpenHashSet<>();
int totalReceiverAmount = 0;
for (IConnectionPair pair : network.connections) {
for (BlockPos relay : pair.getPositions()) {
if (relay != null && this.world.isBlockLoaded(relay) && !alreadyChecked.contains(relay)) {
alreadyChecked.add(relay);
TileEntity relayTile = this.world.getTileEntity(relay);
if (relayTile instanceof TileEntityLaserRelayEnergy) {
TileEntityLaserRelayEnergy theRelay = (TileEntityLaserRelayEnergy) relayTile;
if (theRelay.mode != Mode.INPUT_ONLY) {
boolean workedOnce = false;
for (EnumFacing facing : theRelay.receiversAround.keySet()) {
if (theRelay != this || facing != from) {
TileEntity tile = theRelay.receiversAround.get(facing);
EnumFacing opp = facing.getOpposite();
if (tile != null) {
if (tile.hasCapability(CapabilityEnergy.ENERGY, opp)) {
IEnergyStorage cap = tile.getCapability(CapabilityEnergy.ENERGY, opp);
if (cap != null && cap.receiveEnergy(maxTransfer, true) > 0) {
totalReceiverAmount++;
workedOnce = true;
}
}
}
}
}
if (workedOnce) {
relaysThatWork.add(theRelay);
}
}
}
}
}
}
if (totalReceiverAmount > 0 && !relaysThatWork.isEmpty()) {
int amountPer = maxTransfer / totalReceiverAmount;
if (amountPer <= 0) {
amountPer = maxTransfer;
}
for (TileEntityLaserRelayEnergy theRelay : relaysThatWork) {
double highestLoss = Math.max(theRelay.getLossPercentage(), this.getLossPercentage());
int lowestCap = Math.min(theRelay.getEnergyCap(), this.getEnergyCap());
for (Map.Entry<EnumFacing, TileEntity> receiver : theRelay.receiversAround.entrySet()) {
if (receiver != null) {
EnumFacing side = receiver.getKey();
EnumFacing opp = side.getOpposite();
TileEntity tile = receiver.getValue();
if (!alreadyChecked.contains(tile.getPos())) {
alreadyChecked.add(tile.getPos());
if (theRelay != this || side != from) {
if (tile.hasCapability(CapabilityEnergy.ENERGY, opp)) {
IEnergyStorage cap = tile.getCapability(CapabilityEnergy.ENERGY, opp);
if (cap != null) {
int theoreticalReceived = cap.receiveEnergy(Math.min(amountPer, lowestCap), true);
if (theoreticalReceived > 0) {
int deduct = this.calcDeduction(theoreticalReceived, highestLoss);
if (deduct >= theoreticalReceived) { //Happens with small numbers
deduct = 0;
}
transmitted += cap.receiveEnergy(theoreticalReceived - deduct, simulate);
transmitted += deduct;
}
}
}
//If everything that could be transmitted was transmitted
if (transmitted >= maxTransfer) { return transmitted; }
}
}
}
}
}
}
return transmitted;
}
private int calcDeduction(int theoreticalReceived, double highestLoss) {
return ConfigBoolValues.LASER_RELAY_LOSS.isEnabled() ? MathHelper.ceil(theoreticalReceived * (highestLoss / 100)) : 0;
}
public int getEnergyCap() {
return CAP;
}
public double getLossPercentage() {
return 5;
}
@Override
@SideOnly(Side.CLIENT)
public String getExtraDisplayString() {
return StringUtil.localize("info." + ActuallyAdditions.MODID + ".laserRelay.energy.extra") + ": " + TextFormatting.DARK_RED + StringUtil.localize(this.mode.name) + TextFormatting.RESET;
}
@Override
@SideOnly(Side.CLIENT)
public String getCompassDisplayString() {
return TextFormatting.GREEN + StringUtil.localize("info." + ActuallyAdditions.MODID + ".laserRelay.energy.display");
}
@Override
public void onCompassAction(EntityPlayer player) {
this.mode = this.mode.getNext();
}
@Override
public void writeSyncableNBT(NBTTagCompound compound, NBTType type) {
super.writeSyncableNBT(compound, type);
if (type != NBTType.SAVE_BLOCK) {
compound.setString("Mode", this.mode.toString());
}
}
@Override
public void readSyncableNBT(NBTTagCompound compound, NBTType type) {
super.readSyncableNBT(compound, type);
if (type != NBTType.SAVE_BLOCK) {
String modeStrg = compound.getString("Mode");
if (modeStrg != null && !modeStrg.isEmpty()) {
this.mode = Mode.valueOf(modeStrg);
}
}
}
public enum Mode {
BOTH("info." + ActuallyAdditions.MODID + ".laserRelay.mode.both"),
OUTPUT_ONLY("info." + ActuallyAdditions.MODID + ".laserRelay.mode.outputOnly"),
INPUT_ONLY("info." + ActuallyAdditions.MODID + ".laserRelay.mode.inputOnly");
public final String name;
Mode(String name) {
this.name = name;
}
public Mode getNext() {
int ordinal = this.ordinal() + 1;
if (ordinal >= values().length) {
ordinal = 0;
}
return values()[ordinal];
}
}
}