src/main/java/com/hbm/tileentity/machine/TileEntityPWRController.java - master - Belyash5/Flu

src/main/java/com/hbm/tileentity/machine/TileEntityPWRController.java
631 строка19 KB
BallOfEnergy
Merge remote-tracking branch 'upstream/master' into Optimization
27 окт 2024, 05:17
27 окт 2024, 05:17a226685
package com.hbm.tileentity.machine;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map.Entry;

import com.hbm.blocks.ModBlocks;
import com.hbm.handler.CompatHandler;
import com.hbm.interfaces.IControlReceiver;
import com.hbm.inventory.container.ContainerPWR;
import com.hbm.inventory.fluid.Fluids;
import com.hbm.inventory.fluid.tank.FluidTank;
import com.hbm.inventory.fluid.trait.FT_Heatable;
import com.hbm.inventory.fluid.trait.FT_PWRModerator;
import com.hbm.inventory.fluid.trait.FT_Heatable.HeatingStep;
import com.hbm.inventory.fluid.trait.FT_Heatable.HeatingType;
import com.hbm.inventory.gui.GUIPWR;
import com.hbm.items.ModItems;
import com.hbm.items.machine.ItemPWRFuel.EnumPWRFuel;
import com.hbm.main.MainRegistry;
import com.hbm.sound.AudioWrapper;
import com.hbm.tileentity.IGUIProvider;
import com.hbm.tileentity.TileEntityMachineBase;
import com.hbm.util.EnumUtil;
import com.hbm.util.fauxpointtwelve.BlockPos;

import api.hbm.fluid.IFluidStandardTransceiver;
import cpw.mods.fml.common.Optional;
import cpw.mods.fml.relauncher.Side;
import cpw.mods.fml.relauncher.SideOnly;
import io.netty.buffer.ByteBuf;
import li.cil.oc.api.machine.Arguments;
import li.cil.oc.api.machine.Callback;
import li.cil.oc.api.machine.Context;
import li.cil.oc.api.network.SimpleComponent;
import net.minecraft.block.Block;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.inventory.Container;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.util.MathHelper;
import net.minecraft.world.World;
import net.minecraftforge.common.util.ForgeDirection;

@Optional.InterfaceList({@Optional.Interface(iface = "li.cil.oc.api.network.SimpleComponent", modid = "OpenComputers")})
public class TileEntityPWRController extends TileEntityMachineBase implements IGUIProvider, IControlReceiver, SimpleComponent, IFluidStandardTransceiver, CompatHandler.OCComponent {

	public FluidTank[] tanks;
	public long coreHeat;
	public static final long coreHeatCapacityBase = 10_000_000;
	public long coreHeatCapacity = 10_000_000;
	public long hullHeat;
	public static final long hullHeatCapacityBase = 10_000_000;
	public double flux;

	public double rodLevel = 100;
	public double rodTarget = 100;

	public int typeLoaded;
	public int amountLoaded;
	public double progress;
	public double processTime;

	public int rodCount;
	public int connections;
	public int connectionsControlled;
	public int heatexCount;
	public int heatsinkCount;
	public int channelCount;
	public int sourceCount;

	public int unloadDelay = 0;
	public boolean assembled;

	private AudioWrapper audio;

	protected List<BlockPos> ports = new ArrayList();
	protected List<BlockPos> rods = new ArrayList();

	public TileEntityPWRController() {
		super(3);

		this.tanks = new FluidTank[2];
		this.tanks[0] = new FluidTank(Fluids.COOLANT, 128_000);
		this.tanks[1] = new FluidTank(Fluids.COOLANT_HOT, 128_000);
	}

	/** The initial creation of the reactor, does all the pre-calculation and whatnot */
	public void setup(HashMap<BlockPos, Block> partMap, HashMap<BlockPos, Block> rodMap) {

		rodCount = 0;
		connections = 0;
		connectionsControlled = 0;
		heatexCount = 0;
		channelCount = 0;
		heatsinkCount = 0;
		sourceCount = 0;
		ports.clear();
		rods.clear();

		int connectionsDouble = 0;
		int connectionsControlledDouble = 0;

		for(Entry<BlockPos, Block> entry : partMap.entrySet()) {
			Block block = entry.getValue();

			if(block == ModBlocks.pwr_fuel) rodCount++;
			if(block == ModBlocks.pwr_heatex) heatexCount++;
			if(block == ModBlocks.pwr_channel) channelCount++;
			if(block == ModBlocks.pwr_heatsink) heatsinkCount++;
			if(block == ModBlocks.pwr_neutron_source) sourceCount++;
			if(block == ModBlocks.pwr_port) ports.add(entry.getKey());
		}

		for(Entry<BlockPos, Block> entry : rodMap.entrySet()) {
			BlockPos fuelPos = entry.getKey();

			rods.add(fuelPos);

			for(ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS) {

				boolean controlled = false;

				for(int i = 1; i < 16; i++) {
					BlockPos checkPos = fuelPos.offset(dir, i);
					Block atPos = partMap.get(checkPos);
					if(atPos == null || atPos == ModBlocks.pwr_casing) break;
					if(atPos == ModBlocks.pwr_control) controlled = true;
					if(atPos == ModBlocks.pwr_fuel) {
						if(controlled) {
							connectionsControlledDouble++;
						} else {
							connectionsDouble++;
						}
						break;
					}
					if(atPos == ModBlocks.pwr_reflector) {
						if(controlled) {
							connectionsControlledDouble += 2;
						} else {
							connectionsDouble += 2;
						}
						break;
					}
				}
			}
		}

		connections = connectionsDouble / 2;
		connectionsControlled = connectionsControlledDouble / 2;
		heatsinkCount = Math.min(heatsinkCount, 80);

		//switching this to int64 because after 2127 heatsinks the capacity exceeds the int32 which is well within the 4000+ threshold we are working with. oops!
		this.coreHeatCapacity = this.coreHeatCapacityBase + this.heatsinkCount * (this.coreHeatCapacityBase / 20);
	}

	@Override
	public String getName() {
		return "container.pwrController";
	}

	@Override
	public void updateEntity() {

		if(!worldObj.isRemote) {

			this.tanks[0].setType(2, slots);
			setupTanks();

			if(unloadDelay > 0) unloadDelay--;

			int chunkX = xCoord >> 4;
			int chunkZ = zCoord >> 4;

			//since fluid sources are often not within 1 chunk, we just do 2 chunks distance and call it a day
			if(!worldObj.getChunkProvider().chunkExists(chunkX, chunkZ) ||
					!worldObj.getChunkProvider().chunkExists(chunkX + 2, chunkZ + 2) ||
					!worldObj.getChunkProvider().chunkExists(chunkX + 2, chunkZ - 2) ||
					!worldObj.getChunkProvider().chunkExists(chunkX - 2, chunkZ + 2) ||
					!worldObj.getChunkProvider().chunkExists(chunkX - 2, chunkZ - 2)) {
				this.unloadDelay = 60;
			}

			if(this.assembled) {
				for(BlockPos pos : ports) {
					for(ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS) {
						BlockPos portPos = pos.offset(dir);

						if(tanks[1].getFill() > 0) this.sendFluid(tanks[1], worldObj, portPos.getX(), portPos.getY(), portPos.getZ(), dir);
						if(worldObj.getTotalWorldTime() % 20 == 0) this.trySubscribe(tanks[0].getTankType(), worldObj, portPos.getX(), portPos.getY(), portPos.getZ(), dir);
					}
				}

				//only perform fission if the area has been loaded for 40 ticks or more
				if(this.unloadDelay <= 0) {

					if((typeLoaded == -1 || amountLoaded <= 0) && slots[0] != null && slots[0].getItem() == ModItems.pwr_fuel) {
						typeLoaded = slots[0].getItemDamage();
						amountLoaded++;
						this.decrStackSize(0, 1);
						this.markChanged();
					} else if(slots[0] != null && slots[0].getItem() == ModItems.pwr_fuel && slots[0].getItemDamage() == typeLoaded && amountLoaded < rodCount){
						amountLoaded++;
						this.decrStackSize(0, 1);
						this.markChanged();
					}
					double diff = this.rodLevel - this.rodTarget;
					if(diff < 1 && diff > -1) this.rodLevel = this.rodTarget;
					if(this.rodTarget > this.rodLevel) this.rodLevel++;
					if(this.rodTarget < this.rodLevel) this.rodLevel--;

					int newFlux = this.sourceCount * 20;

					if(typeLoaded != -1 && amountLoaded > 0) {

						EnumPWRFuel fuel = EnumUtil.grabEnumSafely(EnumPWRFuel.class, typeLoaded);
						double usedRods = getTotalProcessMultiplier();
						double fluxPerRod = this.flux / this.rodCount;
						double outputPerRod = fuel.function.effonix(fluxPerRod);
						double totalOutput = outputPerRod * amountLoaded * usedRods;
						double totalHeatOutput = totalOutput * fuel.heatEmission;

						this.coreHeat += totalHeatOutput;
						newFlux += totalOutput;

						this.processTime = (int) fuel.yield;
						this.progress += totalOutput;

						if(this.progress >= this.processTime) {
							this.progress -= this.processTime;

							if(slots[1] == null) {
								slots[1] = new ItemStack(ModItems.pwr_fuel_hot, 1, typeLoaded);
							} else if(slots[1].getItem() == ModItems.pwr_fuel_hot && slots[1].getItemDamage() == typeLoaded && slots[1].stackSize < slots[1].getMaxStackSize()) {
								slots[1].stackSize++;
							}

							this.amountLoaded--;
							this.markChanged();
						}
					}

					if(this.amountLoaded <= 0) {
						this.typeLoaded = -1;
					}

					if(amountLoaded > rodCount) amountLoaded = rodCount;

					/* CORE COOLING */
					double coreCoolingApproachNum = getXOverE((double) this.heatexCount * 5 / (double) getRodCountForCoolant(), 2) / 2D;
					long averageCoreHeat = (this.coreHeat + this.hullHeat) / 2;
					this.coreHeat -= (coreHeat - averageCoreHeat) * coreCoolingApproachNum;
					this.hullHeat -= (hullHeat - averageCoreHeat) * coreCoolingApproachNum;

					updateCoolant();

					this.coreHeat *= 0.999D;
					this.hullHeat *= 0.999D;

					this.flux = newFlux;

					if(tanks[0].getTankType().hasTrait(FT_PWRModerator.class) && tanks[0].getFill() > 0) {
						this.flux *= tanks[0].getTankType().getTrait(FT_PWRModerator.class).getMultiplier();
					}

					if(this.coreHeat > this.coreHeatCapacity) {
						meltDown();
					}
				} else {
					this.hullHeat = 0;
					this.coreHeat = 0;
				}
			}

			this.networkPackNT(150);
		} else {

			if(amountLoaded > 0) {

				if(audio == null) {
					audio = createAudioLoop();
					audio.startSound();
				} else if(!audio.isPlaying()) {
					audio = rebootAudio(audio);
				}

				audio.updateVolume(getVolume(1F));
				audio.keepAlive();

			} else {

				if(audio != null) {
					audio.stopSound();
					audio = null;
				}
			}
		}
	}

	protected void meltDown() {

		worldObj.func_147480_a(xCoord, yCoord, zCoord, false);

		double x = 0;
		double y = 0;
		double z = 0;

		for(BlockPos pos : this.rods) {
			Block b = worldObj.getBlock(pos.getX(), pos.getY(), pos.getZ());
			b.breakBlock(worldObj, pos.getX(), pos.getY(), pos.getZ(), b, worldObj.getBlockMetadata(pos.getX(), pos.getY(), pos.getZ()));
			worldObj.setBlock(pos.getX(), pos.getY(), pos.getZ(), ModBlocks.corium_block, 5, 3);

			x += pos.getX() + 0.5;
			y += pos.getY() + 0.5;
			z += pos.getZ() + 0.5;
		}

		x /= rods.size();
		y /= rods.size();
		z /= rods.size();

		worldObj.newExplosion(null, x, y, z, 15F, true, true);
	}

	@Override
	public AudioWrapper createAudioLoop() {
		return MainRegistry.proxy.getLoopedSound("hbm:block.reactorLoop", xCoord, yCoord, zCoord, 1F, 10F, 1.0F, 20);
	}

	@Override
	public void onChunkUnload() {

		if(audio != null) {
			audio.stopSound();
			audio = null;
		}
	}

	@Override
	public void invalidate() {

		super.invalidate();

		if(audio != null) {
			audio.stopSound();
			audio = null;
		}
	}

	protected void updateCoolant() {

		FT_Heatable trait = tanks[0].getTankType().getTrait(FT_Heatable.class);
		if(trait == null || trait.getEfficiency(HeatingType.PWR) <= 0) return;

		double coolingEff = (double) this.channelCount / (double) getRodCountForCoolant() * 0.1D; //10% cooling if numbers match
		if(coolingEff > 1D) coolingEff = 1D;

		//no use in trying to convert everythin to long since the internal tanks would never even support operation like that, just cap the heat cycle count to prevent overflows in the math
		int heatToUse = (int) Math.min(Math.min(this.hullHeat, (long) (this.hullHeat * coolingEff * trait.getEfficiency(HeatingType.PWR))), 2_000_000_000);
		HeatingStep step = trait.getFirstStep();
		int coolCycles = tanks[0].getFill() / step.amountReq;
		int hotCycles = (tanks[1].getMaxFill() - tanks[1].getFill()) / step.amountProduced;
		int heatCycles = heatToUse / step.heatReq;
		int cycles = Math.min(coolCycles, Math.min(hotCycles, heatCycles));

		this.hullHeat -= step.heatReq * cycles;
		this.tanks[0].setFill(tanks[0].getFill() - step.amountReq * cycles);
		this.tanks[1].setFill(tanks[1].getFill() + step.amountProduced * cycles);
	}

	protected int getRodCountForCoolant() {
		return this.rodCount + (int) Math.ceil(this.heatsinkCount / 4D);
	}

	@Override
	public void serialize(ByteBuf buf) {
		super.serialize(buf);
		buf.writeInt(this.rodCount);
		buf.writeLong(this.coreHeat);
		buf.writeLong(this.hullHeat);
		buf.writeDouble(this.flux);
		buf.writeDouble(this.processTime);
		buf.writeDouble(this.progress);
		buf.writeInt(this.typeLoaded);
		buf.writeInt(this.amountLoaded);
		buf.writeDouble(this.rodLevel);
		buf.writeDouble(this.rodTarget);
		buf.writeLong(this.coreHeatCapacity);
		tanks[0].serialize(buf);
		tanks[1].serialize(buf);
	}

	@Override
	public void deserialize(ByteBuf buf) {
		super.deserialize(buf);
		this.rodCount = buf.readInt();
		this.coreHeat = buf.readLong();
		this.hullHeat = buf.readLong();
		this.flux = buf.readDouble();
		this.processTime = buf.readDouble();
		this.progress = buf.readDouble();
		this.typeLoaded = buf.readInt();
		this.amountLoaded = buf.readInt();
		this.rodLevel = buf.readDouble();
		this.rodTarget = buf.readDouble();
		this.coreHeatCapacity = buf.readLong();
		tanks[0].deserialize(buf);
		tanks[1].deserialize(buf);
	}

	protected void setupTanks() {

		FT_Heatable trait = tanks[0].getTankType().getTrait(FT_Heatable.class);

		if(trait == null || trait.getEfficiency(HeatingType.PWR) <= 0) {
			tanks[0].setTankType(Fluids.NONE);
			tanks[1].setTankType(Fluids.NONE);
			return;
		}

		tanks[1].setTankType(trait.getFirstStep().typeProduced);
	}

	public double getTotalProcessMultiplier() {
		double totalConnections = this.connections + this.connectionsControlled * (1D - (this.rodLevel / 100D));
		double connectionsEff = connectinFunc(totalConnections);
		return connectionsEff;
	}

	public double connectinFunc(double connections) {
		return connections / 10D * (1D - getXOverE(connections, 300D)) + connections / 150D * getXOverE(connections, 300D);
	}

	public double getXOverE(double x, double d) {
		return 1 - Math.pow(Math.E, -x / d);
	}

	@Override
	public boolean isItemValidForSlot(int slot, ItemStack stack) {
		if(slot == 0) return stack.getItem() == ModItems.pwr_fuel;
		return false;
	}

	@Override
	public int[] getAccessibleSlotsFromSide(int side) {
		return new int[] {0, 1};
	}

	@Override
	public boolean canExtractItem(int slot, ItemStack itemStack, int side) {
		return slot == 1;
	}

	@Override
	public void readFromNBT(NBTTagCompound nbt) {
		super.readFromNBT(nbt);

		tanks[0].readFromNBT(nbt, "t0");
		tanks[1].readFromNBT(nbt, "t1");

		this.assembled = nbt.getBoolean("assembled");
		this.coreHeat = Math.max(nbt.getInteger("coreHeat"), nbt.getLong("coreHeatL"));
		this.hullHeat = Math.max(nbt.getInteger("hullHeat"), nbt.getLong("hullHeatL"));
		this.flux = nbt.getDouble("flux");
		this.rodLevel = nbt.getDouble("rodLevel");
		this.rodTarget = nbt.getDouble("rodTarget");
		this.typeLoaded = nbt.getInteger("typeLoaded");
		this.amountLoaded = nbt.getInteger("amountLoaded");
		this.progress = nbt.getDouble("progress");
		this.processTime = nbt.getDouble("processTime");
		this.coreHeatCapacity = Math.max(nbt.getInteger("coreHeatCapacity"), nbt.getLong("coreHeatCapacityL"));
		if(this.coreHeatCapacity < this.coreHeatCapacityBase) this.coreHeatCapacity = this.coreHeatCapacityBase;

		this.rodCount = nbt.getInteger("rodCount");
		this.connections = nbt.getInteger("connections");
		this.connectionsControlled = nbt.getInteger("connectionsControlled");
		this.heatexCount = nbt.getInteger("heatexCount");
		this.channelCount = nbt.getInteger("channelCount");
		this.sourceCount = nbt.getInteger("sourceCount");
		this.heatsinkCount = nbt.getInteger("heatsinkCount");

		ports.clear();
		int portCount = nbt.getInteger("portCount");
		for(int i = 0; i < portCount; i++) {
			int[] port = nbt.getIntArray("p" + i);
			ports.add(new BlockPos(port[0], port[1], port[2]));
		}

		rods.clear();
		int rodCount = nbt.getInteger("rodCount");
		for(int i = 0; i < rodCount; i++) {
			if(nbt.hasKey("r" + i)) {
				int[] port = nbt.getIntArray("r" + i);
				rods.add(new BlockPos(port[0], port[1], port[2]));
			}
		}
	}

	@Override
	public void writeToNBT(NBTTagCompound nbt) {
		super.writeToNBT(nbt);

		tanks[0].writeToNBT(nbt, "t0");
		tanks[1].writeToNBT(nbt, "t1");

		nbt.setBoolean("assembled", assembled);
		nbt.setLong("coreHeatL", coreHeat);
		nbt.setLong("hullHeatL", hullHeat);
		nbt.setDouble("flux", flux);
		nbt.setDouble("rodLevel", rodLevel);
		nbt.setDouble("rodTarget", rodTarget);
		nbt.setInteger("typeLoaded", typeLoaded);
		nbt.setInteger("amountLoaded", amountLoaded);
		nbt.setDouble("progress", progress);
		nbt.setDouble("processTime", processTime);
		nbt.setLong("coreHeatCapacityL", coreHeatCapacity);

		nbt.setInteger("rodCount", rodCount);
		nbt.setInteger("connections", connections);
		nbt.setInteger("connectionsControlled", connectionsControlled);
		nbt.setInteger("heatexCount", heatexCount);
		nbt.setInteger("channelCount", channelCount);
		nbt.setInteger("sourceCount", sourceCount);
		nbt.setInteger("heatsinkCount", heatsinkCount);

		nbt.setInteger("portCount", ports.size());
		for(int i = 0; i < ports.size(); i++) {
			BlockPos pos = ports.get(i);
			nbt.setIntArray("p" + i, new int[] { pos.getX(), pos.getY(), pos.getZ() });
		}

		nbt.setInteger("rodCount", rods.size());
		for(int i = 0; i < rods.size(); i++) {
			BlockPos pos = rods.get(i);
			nbt.setIntArray("r" + i, new int[] { pos.getX(), pos.getY(), pos.getZ() });
		}
	}

	@Override
	public boolean hasPermission(EntityPlayer player) {
		return this.isUseableByPlayer(player);
	}

	@Override
	public void receiveControl(NBTTagCompound data) {

		if(data.hasKey("control")) {
			this.rodTarget = MathHelper.clamp_int(data.getInteger("control"), 0, 100);
			this.markChanged();
		}
	}


	// do some opencomputer stuff
	@Override
	@Optional.Method(modid = "OpenComputers")
	public String getComponentName() {
		return "ntm_pwr_control";
	}

	@Callback(direct = true)
	@Optional.Method(modid = "OpenComputers")
	public Object[] getHeat(Context context, Arguments args) {
		return new Object[] {coreHeat, hullHeat};
	}

	@Callback(direct = true)
	@Optional.Method(modid = "OpenComputers")
	public Object[] getFlux(Context context, Arguments args) {
		return new Object[] {flux};
	}

	@Callback(direct = true)
	@Optional.Method(modid = "OpenComputers")
	public Object[] getLevel(Context context, Arguments args) {
		return new Object[] {rodTarget, rodLevel};
	}

	@Callback(direct = true)
	@Optional.Method(modid = "OpenComputers")
	public Object[] getCoolantInfo(Context context, Arguments args) {
		return new Object[] {tanks[0].getFill(), tanks[0].getMaxFill(), tanks[1].getFill(), tanks[1].getMaxFill()};
	}

	@Callback(direct = true)
	@Optional.Method(modid = "OpenComputers")
	public Object[] getFuelInfo(Context context, Arguments args) {
		return new Object[] {amountLoaded, progress, processTime};
	}

	@Callback(direct = true)
	@Optional.Method(modid = "OpenComputers")
	public Object[] getInfo(Context context, Arguments args) {
		return new Object[] {coreHeat, hullHeat, flux, rodTarget, rodLevel, amountLoaded, progress, processTime, tanks[0].getFill(), tanks[0].getMaxFill(), tanks[1].getFill(), tanks[1].getMaxFill()};
	}

	@Callback(direct = true, limit = 4)
	@Optional.Method(modid = "OpenComputers")
	public Object[] setLevel(Context context, Arguments args) {
		rodTarget = MathHelper.clamp_double(args.checkDouble(0), 0, 100);
		this.markChanged();
		return new Object[] {true};
	}

	@Override
	public Container provideContainer(int ID, EntityPlayer player, World world, int x, int y, int z) {
		return new ContainerPWR(player.inventory, this);
	}

	@Override
	@SideOnly(Side.CLIENT)
	public Object provideGUI(int ID, EntityPlayer player, World world, int x, int y, int z) {
		return new GUIPWR(player.inventory, this);
	}

	@Override
	public FluidTank[] getAllTanks() {
		return tanks;
	}

	@Override
	public FluidTank[] getSendingTanks() {
		return new FluidTank[] { tanks[1] };
	}

	@Override
	public FluidTank[] getReceivingTanks() {
		return new FluidTank[] { tanks[0] };
	}
}