RedstoneSwitch.java

package crazypants.enderio.conduit.redstone;

import java.util.List;
import java.util.Set;

import net.minecraft.client.renderer.texture.IIconRegister;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.tileentity.TileEntity;
import net.minecraft.util.IIcon;
import net.minecraftforge.common.util.ForgeDirection;

import com.enderio.core.client.render.BoundingBox;
import com.enderio.core.client.render.IconUtil;
import com.enderio.core.common.util.BlockCoord;
import com.enderio.core.common.util.DyeColor;
import com.enderio.core.common.vecmath.Vector3d;

import cpw.mods.fml.relauncher.Side;
import cpw.mods.fml.relauncher.SideOnly;
import crazypants.enderio.EnderIO;
import crazypants.enderio.conduit.RaytraceResult;
import crazypants.enderio.conduit.geom.CollidableComponent;
import crazypants.enderio.conduit.geom.ConduitGeometryUtil;
import crazypants.enderio.conduit.geom.Offset;

public class RedstoneSwitch extends RedstoneConduit {

  static final String SWITCH_TAG = "switch";
  public static final String SWITHC_ICON_OFF_KEY = "enderio:redstoneConduitSwitchOff";
  public static final String SWITCH_ICON_ON_KEY = "enderio:redstoneConduitSwitchOn";

  private boolean isOn;

  @SideOnly(Side.CLIENT)
  public static void initIcons() {
    IconUtil.addIconProvider(new IconUtil.IIconProvider() {

      @Override
      public void registerIcons(IIconRegister register) {
        ICONS.put(RedstoneSwitch.SWITHC_ICON_OFF_KEY, register.registerIcon(SWITHC_ICON_OFF_KEY));
        ICONS.put(RedstoneSwitch.SWITCH_ICON_ON_KEY, register.registerIcon(SWITCH_ICON_ON_KEY));
      }

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

    });
  }

  @Override
  public ItemStack createItem() {
    return new ItemStack(EnderIO.itemRedstoneConduit, 1, 1);
  }

  @Override
  public int isProvidingStrongPower(ForgeDirection toDirection) {
    if(network == null || !network.isNetworkEnabled()) {
      return 0;
    }
    return isOn ? 15 : 0;
  }

  @Override
  public void writeToNBT(NBTTagCompound nbtRoot) {
    super.writeToNBT(nbtRoot);
    nbtRoot.setBoolean("switchOn", isOn);
  }

  @Override
  public void readFromNBT(NBTTagCompound nbtRoot, short nbtVersion) {
    super.readFromNBT(nbtRoot, nbtVersion);
    isOn = nbtRoot.getBoolean("switchOn");
  }

  IIcon getSwitchIcon() {
    return isOn ? ICONS.get(SWITCH_ICON_ON_KEY) : ICONS.get(SWITHC_ICON_OFF_KEY);
  }

  @Override
  public IIcon getTextureForState(CollidableComponent component) {
    if(SWITCH_TAG.equals(component.data)) {
      return isOn ? ICONS.get(SWITCH_ICON_ON_KEY) : ICONS.get(SWITHC_ICON_OFF_KEY);
    }
    return super.getTextureForState(component);
  }

  @Override
  public List<CollidableComponent> getCollidableComponents() {
    if(collidables != null && !collidablesDirty) {
      return collidables;
    }

    Offset o = getBundle().getOffset(getBaseConduitType(), ForgeDirection.UNKNOWN);
    Vector3d trans = ConduitGeometryUtil.instance.getTranslation(ForgeDirection.UNKNOWN, o);

    List<CollidableComponent> result = super.getCollidableComponents();
    BoundingBox[] aabb = RedstoneSwitchBounds.getInstance().getAABB();

    for (BoundingBox bb : aabb) {
      result.add(new CollidableComponent(IRedstoneConduit.class, bb.translate(trans), ForgeDirection.UNKNOWN, SWITCH_TAG));
    }

    return result;
  }

  @Override
  public boolean onBlockActivated(EntityPlayer player, RaytraceResult res, List<RaytraceResult> all) {
    if(res != null && res.component != null && res.component.data != null && res.component.data.equals(SWITCH_TAG)) {
      toggleSwitch();
      return true;
    }
    return false;
  }

  private void toggleSwitch() {
    isOn = !isOn;
    if(network == null) {
      return;
    }
    TileEntity te = bundle.getEntity();
    Signal signal = new Signal(te.xCoord, te.yCoord, te.zCoord, ForgeDirection.UNKNOWN, 15, DyeColor.RED);
    if(isOn) {
      network.addSignal(signal);
    } else {
      network.removeSignal(signal);
    }
  }

  @Override
  public Set<Signal> getNetworkInputs() {
    Set<Signal> res = super.getNetworkInputs();
    if(isOn) {
      BlockCoord loc = getLocation();
      Signal signal = new Signal(loc.x, loc.y, loc.z, ForgeDirection.UNKNOWN, 15, DyeColor.RED);
      res.add(signal);
    }
    return res;
  }
}