577 lines
16 KiB
Java
577 lines
16 KiB
Java
/*
|
|
* This file is part of ComputerCraft - http://www.computercraft.info
|
|
* Copyright Daniel Ratcliffe, 2011-2021. Do not distribute without permission.
|
|
* Send enquiries to dratcliffe@gmail.com
|
|
*/
|
|
package dan200.computercraft.shared.turtle.blocks;
|
|
|
|
import com.mojang.authlib.GameProfile;
|
|
import dan200.computercraft.ComputerCraft;
|
|
import dan200.computercraft.api.peripheral.IPeripheral;
|
|
import dan200.computercraft.api.turtle.ITurtleAccess;
|
|
import dan200.computercraft.api.turtle.ITurtleUpgrade;
|
|
import dan200.computercraft.api.turtle.TurtleSide;
|
|
import dan200.computercraft.core.computer.ComputerSide;
|
|
import dan200.computercraft.shared.common.TileGeneric;
|
|
import dan200.computercraft.shared.computer.blocks.ComputerPeripheral;
|
|
import dan200.computercraft.shared.computer.blocks.ComputerProxy;
|
|
import dan200.computercraft.shared.computer.blocks.TileComputerBase;
|
|
import dan200.computercraft.shared.computer.core.ComputerFamily;
|
|
import dan200.computercraft.shared.computer.core.ComputerState;
|
|
import dan200.computercraft.shared.computer.core.ServerComputer;
|
|
import dan200.computercraft.shared.turtle.apis.TurtleAPI;
|
|
import dan200.computercraft.shared.turtle.core.TurtleBrain;
|
|
import dan200.computercraft.shared.turtle.inventory.ContainerTurtle;
|
|
import dan200.computercraft.shared.util.*;
|
|
import net.minecraft.entity.player.PlayerEntity;
|
|
import net.minecraft.entity.player.PlayerInventory;
|
|
import net.minecraft.inventory.container.Container;
|
|
import net.minecraft.item.DyeColor;
|
|
import net.minecraft.item.DyeItem;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraft.item.Items;
|
|
import net.minecraft.nbt.CompoundNBT;
|
|
import net.minecraft.nbt.ListNBT;
|
|
import net.minecraft.tileentity.TileEntityType;
|
|
import net.minecraft.util.*;
|
|
import net.minecraft.util.math.BlockPos;
|
|
import net.minecraft.util.math.BlockRayTraceResult;
|
|
import net.minecraft.util.math.Vec3d;
|
|
import net.minecraftforge.common.capabilities.Capability;
|
|
import net.minecraftforge.common.util.Constants;
|
|
import net.minecraftforge.common.util.LazyOptional;
|
|
import net.minecraftforge.items.IItemHandlerModifiable;
|
|
import net.minecraftforge.items.wrapper.InvWrapper;
|
|
|
|
import javax.annotation.Nonnull;
|
|
import javax.annotation.Nullable;
|
|
import java.util.Collections;
|
|
|
|
import static dan200.computercraft.shared.Capabilities.CAPABILITY_PERIPHERAL;
|
|
import static net.minecraftforge.items.CapabilityItemHandler.ITEM_HANDLER_CAPABILITY;
|
|
|
|
public class TileTurtle extends TileComputerBase implements ITurtleTile, DefaultInventory
|
|
{
|
|
public static final int INVENTORY_SIZE = 16;
|
|
public static final int INVENTORY_WIDTH = 4;
|
|
public static final int INVENTORY_HEIGHT = 4;
|
|
|
|
enum MoveState
|
|
{
|
|
NOT_MOVED,
|
|
IN_PROGRESS,
|
|
MOVED
|
|
}
|
|
|
|
private final NonNullList<ItemStack> inventory = NonNullList.withSize( INVENTORY_SIZE, ItemStack.EMPTY );
|
|
private final NonNullList<ItemStack> previousInventory = NonNullList.withSize( INVENTORY_SIZE, ItemStack.EMPTY );
|
|
private final IItemHandlerModifiable itemHandler = new InvWrapper( this );
|
|
private LazyOptional<IItemHandlerModifiable> itemHandlerCap;
|
|
private boolean inventoryChanged = false;
|
|
private TurtleBrain brain = new TurtleBrain( this );
|
|
private MoveState moveState = MoveState.NOT_MOVED;
|
|
private LazyOptional<IPeripheral> peripheral;
|
|
|
|
public TileTurtle( TileEntityType<? extends TileGeneric> type, ComputerFamily family )
|
|
{
|
|
super( type, family );
|
|
}
|
|
|
|
private boolean hasMoved()
|
|
{
|
|
return moveState == MoveState.MOVED;
|
|
}
|
|
|
|
@Override
|
|
protected ServerComputer createComputer( int instanceID, int id )
|
|
{
|
|
ServerComputer computer = new ServerComputer(
|
|
getLevel(), id, label, instanceID, getFamily(),
|
|
ComputerCraft.turtleTermWidth, ComputerCraft.turtleTermHeight
|
|
);
|
|
computer.setPosition( getBlockPos() );
|
|
computer.addAPI( new TurtleAPI( computer.getAPIEnvironment(), getAccess() ) );
|
|
brain.setupComputer( computer );
|
|
return computer;
|
|
}
|
|
|
|
public ComputerProxy createProxy()
|
|
{
|
|
return brain.getProxy();
|
|
}
|
|
|
|
@Override
|
|
public void destroy()
|
|
{
|
|
if( !hasMoved() )
|
|
{
|
|
// Stop computer
|
|
super.destroy();
|
|
|
|
// Drop contents
|
|
if( !getLevel().isClientSide )
|
|
{
|
|
int size = getContainerSize();
|
|
for( int i = 0; i < size; i++ )
|
|
{
|
|
ItemStack stack = getItem( i );
|
|
if( !stack.isEmpty() )
|
|
{
|
|
WorldUtil.dropItemStack( stack, getLevel(), getBlockPos() );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Just turn off any redstone we had on
|
|
for( Direction dir : DirectionUtil.FACINGS )
|
|
{
|
|
RedstoneUtil.propagateRedstoneOutput( getLevel(), getBlockPos(), dir );
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected void unload()
|
|
{
|
|
if( !hasMoved() )
|
|
{
|
|
super.unload();
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected void invalidateCaps()
|
|
{
|
|
super.invalidateCaps();
|
|
itemHandlerCap = CapabilityUtil.invalidate( itemHandlerCap );
|
|
peripheral = CapabilityUtil.invalidate( peripheral );
|
|
}
|
|
|
|
@Nonnull
|
|
@Override
|
|
public ActionResultType onActivate( PlayerEntity player, Hand hand, BlockRayTraceResult hit )
|
|
{
|
|
// Apply dye
|
|
ItemStack currentItem = player.getItemInHand( hand );
|
|
if( !currentItem.isEmpty() )
|
|
{
|
|
if( currentItem.getItem() instanceof DyeItem )
|
|
{
|
|
// Dye to change turtle colour
|
|
if( !getLevel().isClientSide )
|
|
{
|
|
DyeColor dye = ((DyeItem) currentItem.getItem()).getDyeColor();
|
|
if( brain.getDyeColour() != dye )
|
|
{
|
|
brain.setDyeColour( dye );
|
|
if( !player.isCreative() )
|
|
{
|
|
currentItem.shrink( 1 );
|
|
}
|
|
}
|
|
}
|
|
return ActionResultType.SUCCESS;
|
|
}
|
|
else if( currentItem.getItem() == Items.WATER_BUCKET && brain.getColour() != -1 )
|
|
{
|
|
// Water to remove turtle colour
|
|
if( !getLevel().isClientSide )
|
|
{
|
|
if( brain.getColour() != -1 )
|
|
{
|
|
brain.setColour( -1 );
|
|
if( !player.isCreative() )
|
|
{
|
|
player.setItemInHand( hand, new ItemStack( Items.BUCKET ) );
|
|
player.inventory.setChanged();
|
|
}
|
|
}
|
|
}
|
|
return ActionResultType.SUCCESS;
|
|
}
|
|
}
|
|
|
|
// Open GUI or whatever
|
|
return super.onActivate( player, hand, hit );
|
|
}
|
|
|
|
@Override
|
|
protected boolean canNameWithTag( PlayerEntity player )
|
|
{
|
|
return true;
|
|
}
|
|
|
|
@Override
|
|
protected double getInteractRange( PlayerEntity player )
|
|
{
|
|
return 12.0;
|
|
}
|
|
|
|
@Override
|
|
public void tick()
|
|
{
|
|
super.tick();
|
|
brain.update();
|
|
if( !getLevel().isClientSide && inventoryChanged )
|
|
{
|
|
ServerComputer computer = getServerComputer();
|
|
if( computer != null ) computer.queueEvent( "turtle_inventory" );
|
|
|
|
inventoryChanged = false;
|
|
for( int n = 0; n < getContainerSize(); n++ )
|
|
{
|
|
previousInventory.set( n, getItem( n ).copy() );
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected void updateBlockState( ComputerState newState )
|
|
{
|
|
}
|
|
|
|
@Override
|
|
public void onNeighbourChange( @Nonnull BlockPos neighbour )
|
|
{
|
|
if( moveState == MoveState.NOT_MOVED ) super.onNeighbourChange( neighbour );
|
|
}
|
|
|
|
@Override
|
|
public void onNeighbourTileEntityChange( @Nonnull BlockPos neighbour )
|
|
{
|
|
if( moveState == MoveState.NOT_MOVED ) super.onNeighbourTileEntityChange( neighbour );
|
|
}
|
|
|
|
public void notifyMoveStart()
|
|
{
|
|
if( moveState == MoveState.NOT_MOVED ) moveState = MoveState.IN_PROGRESS;
|
|
}
|
|
|
|
public void notifyMoveEnd()
|
|
{
|
|
// MoveState.MOVED is final
|
|
if( moveState == MoveState.IN_PROGRESS ) moveState = MoveState.NOT_MOVED;
|
|
}
|
|
|
|
@Override
|
|
public void load( @Nonnull CompoundNBT nbt )
|
|
{
|
|
super.load( nbt );
|
|
|
|
// Read inventory
|
|
ListNBT nbttaglist = nbt.getList( "Items", Constants.NBT.TAG_COMPOUND );
|
|
inventory.clear();
|
|
previousInventory.clear();
|
|
for( int i = 0; i < nbttaglist.size(); i++ )
|
|
{
|
|
CompoundNBT tag = nbttaglist.getCompound( i );
|
|
int slot = tag.getByte( "Slot" ) & 0xff;
|
|
if( slot < getContainerSize() )
|
|
{
|
|
inventory.set( slot, ItemStack.of( tag ) );
|
|
previousInventory.set( slot, inventory.get( slot ).copy() );
|
|
}
|
|
}
|
|
|
|
// Read state
|
|
brain.readFromNBT( nbt );
|
|
}
|
|
|
|
@Nonnull
|
|
@Override
|
|
public CompoundNBT save( @Nonnull CompoundNBT nbt )
|
|
{
|
|
// Write inventory
|
|
ListNBT nbttaglist = new ListNBT();
|
|
for( int i = 0; i < INVENTORY_SIZE; i++ )
|
|
{
|
|
if( !inventory.get( i ).isEmpty() )
|
|
{
|
|
CompoundNBT tag = new CompoundNBT();
|
|
tag.putByte( "Slot", (byte) i );
|
|
inventory.get( i ).save( tag );
|
|
nbttaglist.add( tag );
|
|
}
|
|
}
|
|
nbt.put( "Items", nbttaglist );
|
|
|
|
// Write brain
|
|
nbt = brain.writeToNBT( nbt );
|
|
|
|
return super.save( nbt );
|
|
}
|
|
|
|
@Override
|
|
protected boolean isPeripheralBlockedOnSide( ComputerSide localSide )
|
|
{
|
|
return hasPeripheralUpgradeOnSide( localSide );
|
|
}
|
|
|
|
// IDirectionalTile
|
|
|
|
@Override
|
|
public Direction getDirection()
|
|
{
|
|
return getBlockState().getValue( BlockTurtle.FACING );
|
|
}
|
|
|
|
public void setDirection( Direction dir )
|
|
{
|
|
if( dir.getAxis() == Direction.Axis.Y ) dir = Direction.NORTH;
|
|
level.setBlockAndUpdate( worldPosition, getBlockState().setValue( BlockTurtle.FACING, dir ) );
|
|
updateOutput();
|
|
updateInput();
|
|
onTileEntityChange();
|
|
}
|
|
|
|
// ITurtleTile
|
|
|
|
@Override
|
|
public ITurtleUpgrade getUpgrade( TurtleSide side )
|
|
{
|
|
return brain.getUpgrade( side );
|
|
}
|
|
|
|
@Override
|
|
public int getColour()
|
|
{
|
|
return brain.getColour();
|
|
}
|
|
|
|
@Override
|
|
public ResourceLocation getOverlay()
|
|
{
|
|
return brain.getOverlay();
|
|
}
|
|
|
|
@Override
|
|
public ITurtleAccess getAccess()
|
|
{
|
|
return brain;
|
|
}
|
|
|
|
@Override
|
|
public Vec3d getRenderOffset( float f )
|
|
{
|
|
return brain.getRenderOffset( f );
|
|
}
|
|
|
|
@Override
|
|
public float getRenderYaw( float f )
|
|
{
|
|
return brain.getVisualYaw( f );
|
|
}
|
|
|
|
@Override
|
|
public float getToolRenderAngle( TurtleSide side, float f )
|
|
{
|
|
return brain.getToolRenderAngle( side, f );
|
|
}
|
|
|
|
void setOwningPlayer( GameProfile player )
|
|
{
|
|
brain.setOwningPlayer( player );
|
|
setChanged();
|
|
}
|
|
|
|
// IInventory
|
|
|
|
@Override
|
|
public int getContainerSize()
|
|
{
|
|
return INVENTORY_SIZE;
|
|
}
|
|
|
|
@Override
|
|
public boolean isEmpty()
|
|
{
|
|
for( ItemStack stack : inventory )
|
|
{
|
|
if( !stack.isEmpty() ) return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
@Nonnull
|
|
@Override
|
|
public ItemStack getItem( int slot )
|
|
{
|
|
return slot >= 0 && slot < INVENTORY_SIZE ? inventory.get( slot ) : ItemStack.EMPTY;
|
|
}
|
|
|
|
@Nonnull
|
|
@Override
|
|
public ItemStack removeItemNoUpdate( int slot )
|
|
{
|
|
ItemStack result = getItem( slot );
|
|
setItem( slot, ItemStack.EMPTY );
|
|
return result;
|
|
}
|
|
|
|
@Nonnull
|
|
@Override
|
|
public ItemStack removeItem( int slot, int count )
|
|
{
|
|
if( count == 0 ) return ItemStack.EMPTY;
|
|
|
|
ItemStack stack = getItem( slot );
|
|
if( stack.isEmpty() ) return ItemStack.EMPTY;
|
|
|
|
if( stack.getCount() <= count )
|
|
{
|
|
setItem( slot, ItemStack.EMPTY );
|
|
return stack;
|
|
}
|
|
|
|
ItemStack part = stack.split( count );
|
|
onInventoryDefinitelyChanged();
|
|
return part;
|
|
}
|
|
|
|
@Override
|
|
public void setItem( int i, @Nonnull ItemStack stack )
|
|
{
|
|
if( i >= 0 && i < INVENTORY_SIZE && !InventoryUtil.areItemsEqual( stack, inventory.get( i ) ) )
|
|
{
|
|
inventory.set( i, stack );
|
|
onInventoryDefinitelyChanged();
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void clearContent()
|
|
{
|
|
boolean changed = false;
|
|
for( int i = 0; i < INVENTORY_SIZE; i++ )
|
|
{
|
|
if( !inventory.get( i ).isEmpty() )
|
|
{
|
|
inventory.set( i, ItemStack.EMPTY );
|
|
changed = true;
|
|
}
|
|
}
|
|
|
|
if( changed ) onInventoryDefinitelyChanged();
|
|
}
|
|
|
|
@Override
|
|
public void setChanged()
|
|
{
|
|
super.setChanged();
|
|
if( !inventoryChanged )
|
|
{
|
|
for( int n = 0; n < getContainerSize(); n++ )
|
|
{
|
|
if( !ItemStack.matches( getItem( n ), previousInventory.get( n ) ) )
|
|
{
|
|
inventoryChanged = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public boolean stillValid( @Nonnull PlayerEntity player )
|
|
{
|
|
return isUsable( player, false );
|
|
}
|
|
|
|
private void onInventoryDefinitelyChanged()
|
|
{
|
|
super.setChanged();
|
|
inventoryChanged = true;
|
|
}
|
|
|
|
public void onTileEntityChange()
|
|
{
|
|
super.setChanged();
|
|
}
|
|
|
|
// Networking stuff
|
|
|
|
@Override
|
|
protected void writeDescription( @Nonnull CompoundNBT nbt )
|
|
{
|
|
super.writeDescription( nbt );
|
|
brain.writeDescription( nbt );
|
|
}
|
|
|
|
@Override
|
|
protected void readDescription( @Nonnull CompoundNBT nbt )
|
|
{
|
|
super.readDescription( nbt );
|
|
brain.readDescription( nbt );
|
|
}
|
|
|
|
// Privates
|
|
|
|
private boolean hasPeripheralUpgradeOnSide( ComputerSide side )
|
|
{
|
|
ITurtleUpgrade upgrade;
|
|
switch( side )
|
|
{
|
|
case RIGHT:
|
|
upgrade = getUpgrade( TurtleSide.RIGHT );
|
|
break;
|
|
case LEFT:
|
|
upgrade = getUpgrade( TurtleSide.LEFT );
|
|
break;
|
|
default:
|
|
return false;
|
|
}
|
|
return upgrade != null && upgrade.getType().isPeripheral();
|
|
}
|
|
|
|
public void transferStateFrom( TileTurtle copy )
|
|
{
|
|
super.transferStateFrom( copy );
|
|
Collections.copy( inventory, copy.inventory );
|
|
Collections.copy( previousInventory, copy.previousInventory );
|
|
inventoryChanged = copy.inventoryChanged;
|
|
brain = copy.brain;
|
|
brain.setOwner( this );
|
|
|
|
// Mark the other turtle as having moved, and so its peripheral is dead.
|
|
copy.moveState = MoveState.MOVED;
|
|
copy.peripheral = CapabilityUtil.invalidate( copy.peripheral );
|
|
}
|
|
|
|
public IItemHandlerModifiable getItemHandler()
|
|
{
|
|
return itemHandler;
|
|
}
|
|
|
|
@Nonnull
|
|
@Override
|
|
public <T> LazyOptional<T> getCapability( @Nonnull Capability<T> cap, @Nullable Direction side )
|
|
{
|
|
if( cap == ITEM_HANDLER_CAPABILITY )
|
|
{
|
|
if( itemHandlerCap == null ) itemHandlerCap = LazyOptional.of( () -> new InvWrapper( this ) );
|
|
return itemHandlerCap.cast();
|
|
}
|
|
|
|
if( cap == CAPABILITY_PERIPHERAL )
|
|
{
|
|
if( hasMoved() ) return LazyOptional.empty();
|
|
if( peripheral == null )
|
|
{
|
|
peripheral = LazyOptional.of( () -> new ComputerPeripheral( "turtle", createProxy() ) );
|
|
}
|
|
return peripheral.cast();
|
|
}
|
|
|
|
return super.getCapability( cap, side );
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
public Container createMenu( int id, @Nonnull PlayerInventory inventory, @Nonnull PlayerEntity player )
|
|
{
|
|
return new ContainerTurtle( id, inventory, brain );
|
|
}
|
|
}
|