IndustrialWires/src/main/java/malte0811/industrialWires/blocks/controlpanel/TileEntityRSPanelConn.java

/*
 * This file is part of Industrial Wires.
 * Copyright (C) 2016-2017 malte0811
 *
 * Industrial Wires is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Industrial Wires is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Industrial Wires.  If not, see <http://www.gnu.org/licenses/>.
 */

package malte0811.industrialWires.blocks.controlpanel;

import blusunrize.immersiveengineering.api.TargetingInfo;
import blusunrize.immersiveengineering.api.energy.wires.IImmersiveConnectable;
import blusunrize.immersiveengineering.api.energy.wires.ImmersiveNetHandler;
import blusunrize.immersiveengineering.api.energy.wires.TileEntityImmersiveConnectable;
import blusunrize.immersiveengineering.api.energy.wires.WireType;
import blusunrize.immersiveengineering.api.energy.wires.redstone.IRedstoneConnector;
import blusunrize.immersiveengineering.api.energy.wires.redstone.RedstoneWireNetwork;
import blusunrize.immersiveengineering.common.blocks.IEBlockInterfaces;
import malte0811.industrialWires.blocks.IBlockBoundsIW;
import malte0811.industrialWires.blocks.INetGUI;
import malte0811.industrialWires.controlpanel.PanelComponent;
import malte0811.industrialWires.controlpanel.PanelUtils;
import malte0811.industrialWires.controlpanel.PropertyComponents;
import malte0811.industrialWires.util.TriConsumer;
import net.minecraft.block.state.IBlockState;
import net.minecraft.entity.EntityLivingBase;
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.ITickable;
import net.minecraft.util.math.AxisAlignedBB;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.Vec3d;
import net.minecraft.world.World;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.lang.ref.WeakReference;
import java.util.*;
import java.util.function.Consumer;

public class TileEntityRSPanelConn extends TileEntityImmersiveConnectable implements IRedstoneConnector, ITickable, INetGUI, IEBlockInterfaces.IDirectionalTile, IBlockBoundsIW {
	private byte[] out = new byte[16];
	private boolean dirty = true;
	private byte[] oldInput = new byte[16];
	private Set<Consumer<byte[]>> changeListeners = new HashSet<>();
	private Set<TileEntityPanel> connectedPanels = new HashSet<>();
	private EnumFacing facing = EnumFacing.NORTH;
	@Nonnull
	private RedstoneWireNetwork network = new RedstoneWireNetwork().add(this);
	private boolean hasConn = false;
	private int id;

	{
		for (int i = 0; i < 16; i++) {
			oldInput[i] = -1;
		}
	}

	private boolean loaded = false;

	@Override
	public void update() {
		if (hasWorld() && !world.isRemote) {
			if (!loaded) {
				loaded = true;
				// completely reload the network
				network.removeFromNetwork(null);
				List<BlockPos> parts = PanelUtils.discoverPanelParts(world, pos, 100);
				for (BlockPos bp : parts) {
					TileEntity te = world.getTileEntity(bp);
					if (te instanceof TileEntityPanel) {
						registerPanel(((TileEntityPanel) te));
					}
				}
			}
			if (dirty) {
				network.updateValues();
				dirty = false;
			}
		}
	}

	@Override
	public void writeCustomNBT(NBTTagCompound out, boolean updatePacket) {
		super.writeCustomNBT(out, updatePacket);
		out.setByteArray("out", this.out);
		out.setBoolean("hasConn", hasConn);
		out.setInteger("rsId", id);
		out.setInteger("facing", facing.getIndex());
	}

	@Override
	public void readCustomNBT(NBTTagCompound in, boolean updatePacket) {
		super.readCustomNBT(in, updatePacket);
		out = in.getByteArray("out");
		hasConn = in.getBoolean("hasConn");
		id = in.getInteger("rsId");
		facing = EnumFacing.VALUES[in.getInteger("facing")];
		aabb = null;
	}

	private final Map<PCWrapper, byte[]> outputs = new HashMap<>();
	private TriConsumer<Integer, Byte, PanelComponent> rsOut = (channel, value, pc) -> {
		PCWrapper wrapper = new PCWrapper(pc);
		if (!outputs.containsKey(wrapper)) {
			outputs.put(wrapper, new byte[16]);
		}
		if (outputs.get(wrapper)[channel] != value) {
			outputs.get(wrapper)[channel] = value;
			byte max = 0;
			Iterator<Map.Entry<PCWrapper, byte[]>> it = outputs.entrySet().iterator();
			while (it.hasNext()) {
				Map.Entry<PCWrapper, byte[]> curr = it.next();
				if (curr.getKey().pc.get() == null) {
					it.remove();
					continue;
				}
				if (curr.getValue()[channel] > max) {
					max = curr.getValue()[channel];
				}
			}
			dirty = true;
			out[channel] = max;
		}
	};

	private class PCWrapper {
		@Nonnull
		private final WeakReference<PanelComponent> pc;

		public PCWrapper(@Nonnull PanelComponent pc) {
			this.pc = new WeakReference<>(pc);
		}

		@Override
		public boolean equals(Object o) {
			if (this == o) return true;
			if (o == null || getClass() != o.getClass()) return false;

			PCWrapper pcWrapper = (PCWrapper) o;

			return pcWrapper.pc.get() == pc.get();
		}

		@Override
		public int hashCode() {
			return System.identityHashCode(pc.get());
		}
	}

	public void registerPanel(TileEntityPanel panel) {
		PropertyComponents.PanelRenderProperties p = panel.getComponents();
		for (PanelComponent pc : p) {
			Consumer<byte[]> listener = pc.getRSInputHandler(id, panel);
			if (listener != null) {
				changeListeners.add(listener);
				listener.accept(network.channelValues);
			}
			pc.registerRSOutput(id, rsOut);
		}
		panel.registerRS(this);
		connectedPanels.add(panel);
	}

	public void unregisterPanel(TileEntityPanel panel, boolean remove, boolean callPanel) {
		out = new byte[16];
		PropertyComponents.PanelRenderProperties p = panel.getComponents();
		for (PanelComponent pc : p) {
			Consumer<byte[]> listener = pc.getRSInputHandler(id, panel);
			if (listener != null) {
				listener.accept(new byte[16]);
				changeListeners.remove(listener);
			}
			pc.unregisterRSOutput(id, rsOut);
			outputs.remove(new PCWrapper(pc));
		}
		if (callPanel) {
			panel.unregisterRS(this);
		}
		if (remove) {
			connectedPanels.remove(panel);
		}
		for (TileEntityPanel te : connectedPanels) {
			for (PanelComponent pc : te.getComponents()) {
				pc.registerRSOutput(id, rsOut);
			}
		}
		network.updateValues();
	}

	@Override
	public void setNetwork(@Nonnull RedstoneWireNetwork net) {
		network = net;
	}

	@Nonnull
	@Override
	public RedstoneWireNetwork getNetwork() {
		return network;
	}

	@Override
	public void onChange() {
		if (!Arrays.equals(oldInput, network.channelValues)) {
			oldInput = Arrays.copyOf(network.channelValues, 16);
			for (Consumer<byte[]> c : changeListeners) {
				c.accept(oldInput);
			}
		}
	}

	@Override
	public void updateInput(byte[] currIn) {
		for (int i = 0; i < 16; i++) {
			currIn[i] = (byte) Math.max(currIn[i], out[i]);
		}
	}

	@Override
	public BlockPos getConnectionMaster(@Nullable WireType wire, TargetingInfo target) {
		return pos;
	}

	@Override
	public boolean canConnectCable(WireType wire, TargetingInfo targetingInfo) {
		return wire == WireType.REDSTONE && !hasConn;
	}

	@Override
	public void connectCable(WireType wireType, TargetingInfo targetingInfo, IImmersiveConnectable other) {
		hasConn = true;
		if (other instanceof IRedstoneConnector && ((IRedstoneConnector) other).getNetwork() != network) {
			network.mergeNetwork(((IRedstoneConnector) other).getNetwork());
		}
	}

	@Override
	public WireType getCableLimiter(TargetingInfo targetingInfo) {
		return WireType.REDSTONE;
	}

	@Override
	public boolean allowEnergyToPass(ImmersiveNetHandler.Connection connection) {
		return false;
	}

	@Override
	public void removeCable(ImmersiveNetHandler.Connection connection) {
		hasConn = false;
		network.removeFromNetwork(this);
		this.markDirty();
		if (world != null) {
			IBlockState state = world.getBlockState(pos);
			world.notifyBlockUpdate(pos, state, state, 3);
		}
	}

	@Override
	public Vec3d getRaytraceOffset(IImmersiveConnectable other) {
		EnumFacing side = facing.getOpposite();
		return new Vec3d(.5 + side.getFrontOffsetX() * .0625, .5 + side.getFrontOffsetY() * .0625, .5 + side.getFrontOffsetZ() * .0625);
	}

	@Override
	public Vec3d getConnectionOffset(ImmersiveNetHandler.Connection connection) {
		EnumFacing side = facing.getOpposite();
		double conRadius = connection.cableType.getRenderDiameter() / 2;
		return new Vec3d(.5 - conRadius * side.getFrontOffsetX(), .5 - conRadius * side.getFrontOffsetY(), .5 - conRadius * side.getFrontOffsetZ());
	}

	@Override
	public void onChunkUnload() {
		super.onChunkUnload();
		for (TileEntityPanel panel : connectedPanels) {
			unregisterPanel(panel, false, true);
		}
	}

	@Override
	public void invalidate() {
		super.invalidate();
		for (TileEntityPanel panel : connectedPanels) {
			unregisterPanel(panel, false, true);
		}
	}

	@Override
	public void onChange(NBTTagCompound nbt, EntityPlayer p) {
		if (nbt.hasKey("rsId")) {
			List<BlockPos> parts = PanelUtils.discoverPanelParts(world, pos, 100);
			List<TileEntityPanel> tes = new ArrayList<>(parts.size());
			for (BlockPos bp : parts) {
				TileEntity te = world.getTileEntity(bp);
				if (te instanceof TileEntityPanel) {
					tes.add((TileEntityPanel) te);
					unregisterPanel((TileEntityPanel) te, true, true);
				}
			}
			id = nbt.getInteger("rsId");
			out = new byte[16];
			for (TileEntityPanel panel : tes) {
				registerPanel(panel);
			}
			network.updateValues();
			IBlockState state = world.getBlockState(pos);
			world.notifyBlockUpdate(pos, state, state, 3);
			world.addBlockEvent(pos, state.getBlock(), 255, 0);
		}
	}

	@Override
	public World getConnectorWorld() {
		return world;
	}

	public int getRsId() {
		return id;
	}

	@Nonnull
	@Override
	public EnumFacing getFacing() {
		return facing;
	}

	@Override
	public void setFacing(@Nonnull EnumFacing facing) {
		this.facing = facing;
	}

	@Override
	public int getFacingLimitation() {
		return 0;
	}

	@Override
	public boolean mirrorFacingOnPlacement(@Nonnull EntityLivingBase placer) {
		return true;
	}

	@Override
	public boolean canHammerRotate(@Nonnull EnumFacing side, float hitX, float hitY, float hitZ, @Nonnull EntityLivingBase entity) {
		return false;
	}

	@Override
	public boolean canRotate(@Nonnull EnumFacing axis) {
		return false;
	}

	private AxisAlignedBB aabb;

	@Override
	public AxisAlignedBB getBoundingBox() {
		if (aabb == null) {
			double h = 9 / 16D;
			switch (facing) {
			case DOWN:
				aabb = new AxisAlignedBB(0, 0, 0, 1, h, 1);
				break;
			case UP:
				aabb = new AxisAlignedBB(0, 1 - h, 0, 1, 1, 1);
				break;
			case NORTH:
				aabb = new AxisAlignedBB(0, 0, 0, 1, 1, h);
				break;
			case SOUTH:
				aabb = new AxisAlignedBB(0, 0, 1 - h, 1, 1, 1);
				break;
			case WEST:
				aabb = new AxisAlignedBB(0, 0, 0, h, 1, 1);
				break;
			case EAST:
				aabb = new AxisAlignedBB(1 - h, 0, 0, 1, 1, 1);
				break;
			}
		}
		return aabb;
	}
}