#include "stdafx.h"
#include "com.mojang.nbt.h"
#include "net.minecraft.world.h"
#include "net.minecraft.world.level.h"
#include "TileEntity.h"
#include "net.minecraft.world.entity.item.h"
#include "net.minecraft.world.entity.player.h"
#include "net.minecraft.world.item.h"
#include "net.minecraft.world.inventory.h"
#include "net.minecraft.world.level.tile.h"
#include "net.minecraft.world.phys.h"
#include "Facing.h" // needed for direction-dependent chest logic
#include "ChestPair.h"
#include "ChestTileEntity.h"
#include "ContainerOpenPacket.h"
#include "SoundTypes.h"

namespace
{
	enum class ChestAxis
	{
		X,
		Z,
	};

	struct NeighborSelectionContext
	{
		Level *level;
		int chestType;
		int selfX;
		int selfY;
		int selfZ;
		shared_ptr<ChestTileEntity> selfEntity;
		shared_ptr<ChestTileEntity> cachedN;
		shared_ptr<ChestTileEntity> cachedE;
		shared_ptr<ChestTileEntity> cachedW;
		shared_ptr<ChestTileEntity> cachedS;
	};

	static bool isSameChestType(Level *level, int chestType, int x, int y, int z)
	{
		Tile *tile = Tile::tiles[level->getTile(x, y, z)];
		if (tile == nullptr || !(dynamic_cast<ChestTile *>(tile) != nullptr)) return false;
		return static_cast<ChestTile *>(tile)->type == chestType;
	}

	static bool isRightCandidateFirst(Level *level, int chestX, int chestY, int chestZ, ChestAxis axis, int firstX, int firstZ)
	{
		int data = level->getData(chestX, chestY, chestZ);

		if (axis == ChestAxis::X)
		{
			int rightX = chestX + 1;
			if (data == Facing::SOUTH) rightX = chestX - 1;
			else if (data == Facing::NORTH) rightX = chestX + 1;
			return firstX == rightX;
		}

		int rightZ = chestZ + 1;
		if (data == Facing::WEST) rightZ = chestZ - 1;
		else if (data == Facing::EAST) rightZ = chestZ + 1;
		return firstZ == rightZ;
	}

	static ChestAxis getChestAxis(Level *level, int chestType, int chestX, int chestY, int chestZ)
	{
		int data = level->getData(chestX, chestY, chestZ);
		if (data == Facing::NORTH || data == Facing::SOUTH) return ChestAxis::X;
		if (data == Facing::WEST || data == Facing::EAST) return ChestAxis::Z;

		bool hasX = isSameChestType(level, chestType, chestX - 1, chestY, chestZ) || isSameChestType(level, chestType, chestX + 1, chestY, chestZ);
		bool hasZ = isSameChestType(level, chestType, chestX, chestY, chestZ - 1) || isSameChestType(level, chestType, chestX, chestY, chestZ + 1);
		if (hasZ && !hasX) return ChestAxis::Z;

		return ChestAxis::X;
	}

	static bool hasOtherChestOnAxis(Level *level, int chestType, int chestX, int chestY, int chestZ, ChestAxis axis, int exceptX, int exceptZ)
	{
		if (axis == ChestAxis::X)
		{
			if (isSameChestType(level, chestType, chestX - 1, chestY, chestZ) && !(chestX - 1 == exceptX && chestZ == exceptZ)) return true;
			if (isSameChestType(level, chestType, chestX + 1, chestY, chestZ) && !(chestX + 1 == exceptX && chestZ == exceptZ)) return true;
			return false;
		}

		if (isSameChestType(level, chestType, chestX, chestY, chestZ - 1) && !(chestX == exceptX && chestZ - 1 == exceptZ)) return true;
		if (isSameChestType(level, chestType, chestX, chestY, chestZ + 1) && !(chestX == exceptX && chestZ + 1 == exceptZ)) return true;
		return false;
	}

	static bool useCandidateForAxis(Level *level, int chestType, int chestX, int chestY, int chestZ, ChestAxis axis, shared_ptr<ChestTileEntity> candidate, int &outX, int &outZ)
	{
		if (candidate == nullptr) return false;

		int candidateX = candidate->x;
		int candidateZ = candidate->z;
		bool isAdjacent = (candidateX == chestX - 1 && candidateZ == chestZ) || (candidateX == chestX + 1 && candidateZ == chestZ) || (candidateX == chestX && candidateZ == chestZ - 1) || (candidateX == chestX && candidateZ == chestZ + 1);
		if (!isAdjacent) return false;
		if (axis == ChestAxis::X && candidateZ != chestZ) return false;
		if (axis == ChestAxis::Z && candidateX != chestX) return false;
		if (!isSameChestType(level, chestType, candidateX, chestY, candidateZ)) return false;

		outX = candidateX;
		outZ = candidateZ;
		return true;
	}

	static bool getCachedPartnerOnAxis(const NeighborSelectionContext &context, int chestX, int chestY, int chestZ, ChestAxis axis, int &outX, int &outZ)
	{
		if (chestX == context.selfX && chestY == context.selfY && chestZ == context.selfZ)
		{
			if (useCandidateForAxis(context.level, context.chestType, chestX, chestY, chestZ, axis, context.cachedN, outX, outZ)) return true;
			if (useCandidateForAxis(context.level, context.chestType, chestX, chestY, chestZ, axis, context.cachedE, outX, outZ)) return true;
			if (useCandidateForAxis(context.level, context.chestType, chestX, chestY, chestZ, axis, context.cachedW, outX, outZ)) return true;
			if (useCandidateForAxis(context.level, context.chestType, chestX, chestY, chestZ, axis, context.cachedS, outX, outZ)) return true;
			return false;
		}

		shared_ptr<ChestTileEntity> chestEntity = dynamic_pointer_cast<ChestTileEntity>(context.level->getTileEntity(chestX, chestY, chestZ));
		if (chestEntity == nullptr) return false;
		chestEntity->checkNeighbors();

		if (useCandidateForAxis(context.level, context.chestType, chestX, chestY, chestZ, axis, chestEntity->n.lock(), outX, outZ)) return true;
		if (useCandidateForAxis(context.level, context.chestType, chestX, chestY, chestZ, axis, chestEntity->e.lock(), outX, outZ)) return true;
		if (useCandidateForAxis(context.level, context.chestType, chestX, chestY, chestZ, axis, chestEntity->w.lock(), outX, outZ)) return true;
		if (useCandidateForAxis(context.level, context.chestType, chestX, chestY, chestZ, axis, chestEntity->s.lock(), outX, outZ)) return true;

		return false;
	}

	static bool candidateCommittedElsewhere(const NeighborSelectionContext &context, int chestX, int chestY, int chestZ, int candidateX, int candidateZ)
	{
		int partnerX = candidateX;
		int partnerZ = candidateZ;
		ChestAxis candidateAxis = getChestAxis(context.level, context.chestType, candidateX, chestY, candidateZ);
		if (!getCachedPartnerOnAxis(context, candidateX, chestY, candidateZ, candidateAxis, partnerX, partnerZ)) return false;
		return !(partnerX == chestX && partnerZ == chestZ);
	}

	static ChestPair::CandidateDecision buildCandidateDecision(const NeighborSelectionContext &context, int chestX, int chestY, int chestZ, ChestAxis axis, int selfData, int candidateX, int candidateZ)
	{
		ChestPair::CandidateDecision decision = { false, false, false, false };

		int partnerX = candidateX;
		int partnerZ = candidateZ;
		ChestAxis candidateAxis = getChestAxis(context.level, context.chestType, candidateX, chestY, candidateZ);
		bool hasPartner = getCachedPartnerOnAxis(context, candidateX, chestY, candidateZ, candidateAxis, partnerX, partnerZ);
		decision.prefersThisChest = hasPartner && partnerX == chestX && partnerZ == chestZ;
		decision.committedElsewhere = hasPartner && !decision.prefersThisChest;
		decision.matchesFacingData = selfData != 0 && context.level->getData(candidateX, chestY, candidateZ) == selfData;
		decision.available = !hasOtherChestOnAxis(context.level, context.chestType, candidateX, chestY, candidateZ, axis, chestX, chestZ);

		return decision;
	}

	struct IsCommittedElsewhereSelector
	{
		const NeighborSelectionContext &context;
		int chestX;
		int chestY;
		int chestZ;

		bool operator()(int candidateX, int candidateZ) const
		{
			return candidateCommittedElsewhere(context, chestX, chestY, chestZ, candidateX, candidateZ);
		}
	};

	struct BuildCandidateDecisionSelector
	{
		const NeighborSelectionContext &context;
		int chestX;
		int chestY;
		int chestZ;
		ChestAxis axis;
		int selfData;

		ChestPair::CandidateDecision operator()(int candidateX, int candidateZ) const
		{
			return buildCandidateDecision(context, chestX, chestY, chestZ, axis, selfData, candidateX, candidateZ);
		}
	};

	static bool getPreferredNeighbor(const NeighborSelectionContext &context, int chestX, int chestY, int chestZ, int &outX, int &outZ)
	{
		ChestAxis axis = getChestAxis(context.level, context.chestType, chestX, chestY, chestZ);

		int firstX = chestX;
		int firstZ = chestZ;
		int secondX = chestX;
		int secondZ = chestZ;

		if (axis == ChestAxis::X)
		{
			firstX = chestX - 1;
			secondX = chestX + 1;
		}
		else
		{
			firstZ = chestZ - 1;
			secondZ = chestZ + 1;
		}

		bool hasFirst = isSameChestType(context.level, context.chestType, firstX, chestY, firstZ);
		bool hasSecond = isSameChestType(context.level, context.chestType, secondX, chestY, secondZ);

		bool hasPlacementIntent = false;
		int intendedPartnerX = chestX;
		int intendedPartnerY = chestY;
		int intendedPartnerZ = chestZ;
		bool strictPlacementIntent = false;
		shared_ptr<ChestTileEntity> preferenceEntity = chestX == context.selfX && chestY == context.selfY && chestZ == context.selfZ
			? context.selfEntity
			: dynamic_pointer_cast<ChestTileEntity>(context.level->getTileEntity(chestX, chestY, chestZ));
		if (preferenceEntity != nullptr)
		{
			hasPlacementIntent = preferenceEntity->getPlacementPartnerIntent(intendedPartnerX, intendedPartnerY, intendedPartnerZ, strictPlacementIntent);
			if (hasPlacementIntent && !isSameChestType(context.level, context.chestType, intendedPartnerX, intendedPartnerY, intendedPartnerZ))
			{
				if (strictPlacementIntent) return false;
				preferenceEntity->clearPlacementPartnerIntent();
				hasPlacementIntent = false;
			}
		}

		int cachedPartnerX = chestX;
		int cachedPartnerZ = chestZ;
		bool hasCachedPartner = getCachedPartnerOnAxis(context, chestX, chestY, chestZ, axis, cachedPartnerX, cachedPartnerZ);

		bool rightIsFirst = isRightCandidateFirst(context.level, chestX, chestY, chestZ, axis, firstX, firstZ);
		int selfData = context.level->getData(chestX, chestY, chestZ);

		ChestPair::PreferredNeighborInput input =
		{
			firstX,
			firstZ,
			secondX,
			secondZ,
			hasFirst,
			hasSecond,
			hasPlacementIntent,
			intendedPartnerX,
			intendedPartnerY,
			intendedPartnerZ,
			chestY,
			strictPlacementIntent,
			hasCachedPartner,
			cachedPartnerX,
			cachedPartnerZ,
			rightIsFirst,
		};

		IsCommittedElsewhereSelector isCommittedElsewhere = { context, chestX, chestY, chestZ };
		BuildCandidateDecisionSelector candidateDecisionSelector = { context, chestX, chestY, chestZ, axis, selfData };

		return ChestPair::choosePreferredNeighbor(
			input,
			isCommittedElsewhere,
			candidateDecisionSelector,
			outX,
			outZ);
	}

	static bool getMutualNeighbor(const NeighborSelectionContext &context, int chestX, int chestY, int chestZ, int &outX, int &outZ)
	{
		int neighborX = chestX;
		int neighborZ = chestZ;
		if (!getPreferredNeighbor(context, chestX, chestY, chestZ, neighborX, neighborZ)) return false;

		int backX = neighborX;
		int backZ = neighborZ;
		if (!getPreferredNeighbor(context, neighborX, chestY, neighborZ, backX, backZ)) return false;
		if (backX != chestX || backZ != chestZ) return false;

		outX = neighborX;
		outZ = neighborZ;
		return true;
	}
}

int ChestTileEntity::getContainerType()
{
	if (isBonusChest)	return ContainerOpenPacket::BONUS_CHEST;
	else				return ContainerOpenPacket::CONTAINER;
}

void ChestTileEntity::_init(bool isBonusChest)
{
	items = new ItemInstanceArray(9 * 4);

	hasCheckedNeighbors = false;
	hasPlacementPartnerIntent = false;
	placementPartnerX = 0;
	placementPartnerY = 0;
	placementPartnerZ = 0;
	placementPartnerStrict = false;
	this->isBonusChest = isBonusChest;

	openness = 0.0f;
	oOpenness = 0.0f;
	openCount = 0;
	tickInterval = 0;

	type = -1;
	name = L"";
}

ChestTileEntity::ChestTileEntity(bool isBonusChest/* = false*/) : TileEntity()
{
	_init(isBonusChest);
}

ChestTileEntity::ChestTileEntity(int type, bool isBonusChest/* = false*/) : TileEntity()
{
	_init(isBonusChest);

	this->type = type;
}

ChestTileEntity::~ChestTileEntity()
{
	delete[] items->data;
	delete items;
}

unsigned int ChestTileEntity::getContainerSize()
{
	return 9 * 3;
}

shared_ptr<ItemInstance> ChestTileEntity::getItem(unsigned int slot)
{
	return items->data[slot];
}

shared_ptr<ItemInstance> ChestTileEntity::removeItem(unsigned int slot, int count)
{
	if (items->data[slot] != nullptr)
	{
		if (items->data[slot]->count <= count)
		{
			shared_ptr<ItemInstance> item = items->data[slot];
			items->data[slot] = nullptr;
			setChanged();
			// 4J Stu - Fix for duplication glitch
			if(item->count <= 0) return nullptr;
			return item;
		}
		else
		{
			shared_ptr<ItemInstance> i = items->data[slot]->remove(count);
			if (items->data[slot]->count == 0) items->data[slot] = nullptr;
			setChanged();
			// 4J Stu - Fix for duplication glitch
			if(i->count <= 0) return nullptr;
			return i;
		}
	}
	return nullptr;
}

shared_ptr<ItemInstance> ChestTileEntity::removeItemNoUpdate(int slot)
{
	if (items->data[slot] != nullptr)
	{
		shared_ptr<ItemInstance> item = items->data[slot];
		items->data[slot] = nullptr;
		return item;
	}
	return nullptr;
}

void ChestTileEntity::setItem(unsigned int slot, shared_ptr<ItemInstance> item)
{
	items->data[slot] = item;
	if (item != nullptr && item->count > getMaxStackSize()) item->count = getMaxStackSize();
	this->setChanged();
}

wstring ChestTileEntity::getName()
{
	return hasCustomName() ? name : app.GetString(IDS_TILE_CHEST);
}

wstring ChestTileEntity::getCustomName()
{
	return hasCustomName() ? name : L"";
}

bool ChestTileEntity::hasCustomName()
{
	return !name.empty();
}

void ChestTileEntity::setCustomName(const wstring &name)
{
	this->name = name;
}

void ChestTileEntity::load(CompoundTag *base)
{
	TileEntity::load(base);
	ListTag<CompoundTag> *inventoryList = (ListTag<CompoundTag> *) base->getList(L"Items");
	if( items )
	{
		delete [] items->data;
		delete items;
	}
	items = new ItemInstanceArray(getContainerSize());
	if (base->contains(L"CustomName")) name = base->getString(L"CustomName");
	for (int i = 0; i < inventoryList->size(); i++)
	{
		CompoundTag *tag = inventoryList->get(i);
		unsigned int slot = tag->getByte(L"Slot") & 0xff;
		if (slot >= 0 && slot < items->length) (*items)[slot] = ItemInstance::fromTag(tag);
	}
	isBonusChest = base->getBoolean(L"bonus");
}

void ChestTileEntity::save(CompoundTag *base)
{
	TileEntity::save(base);
	ListTag<CompoundTag> *listTag = new ListTag<CompoundTag>;

	for (unsigned int i = 0; i < items->length; i++)
	{
		if (items->data[i] != nullptr)
		{
			CompoundTag *tag = new CompoundTag();
			tag->putByte(L"Slot", static_cast<byte>(i));
			items->data[i]->save(tag);
			listTag->add(tag);
		}
	}
	base->put(L"Items", listTag);
	if (hasCustomName()) base->putString(L"CustomName", name);
	base->putBoolean(L"bonus", isBonusChest);
}

int ChestTileEntity::getMaxStackSize() const
{
	return Container::LARGE_MAX_STACK_SIZE;
}

bool ChestTileEntity::stillValid(shared_ptr<Player> player)
{
	if (level->getTileEntity(x, y, z) != shared_from_this() ) return false;
	if (player->distanceToSqr(x + 0.5, y + 0.5, z + 0.5) > 8 * 8) return false;
	return true;
}

void ChestTileEntity::setChanged()
{
	TileEntity::setChanged();
}

void ChestTileEntity::clearCache()
{
	TileEntity::clearCache();
	hasCheckedNeighbors = false;
}

void ChestTileEntity::setPlacementPartnerIntent(int partnerX, int partnerY, int partnerZ, bool strict)
{
	hasPlacementPartnerIntent = true;
	placementPartnerX = partnerX;
	placementPartnerY = partnerY;
	placementPartnerZ = partnerZ;
	placementPartnerStrict = strict;
	hasCheckedNeighbors = false;
}

void ChestTileEntity::clearPlacementPartnerIntent()
{
	hasPlacementPartnerIntent = false;
	placementPartnerX = 0;
	placementPartnerY = 0;
	placementPartnerZ = 0;
	placementPartnerStrict = false;
	hasCheckedNeighbors = false;
}

bool ChestTileEntity::getPlacementPartnerIntent(int &partnerX, int &partnerY, int &partnerZ, bool &strict) const
{
	if (!hasPlacementPartnerIntent) return false;

	partnerX = placementPartnerX;
	partnerY = placementPartnerY;
	partnerZ = placementPartnerZ;
	strict = placementPartnerStrict;
	return true;
}

void ChestTileEntity::heyImYourNeighbor(shared_ptr<ChestTileEntity> neighbor, int from)
{
	if (neighbor->isRemoved())
	{
		hasCheckedNeighbors = false;
	}
	else if (hasCheckedNeighbors)
	{
		switch (from)
		{
		case Direction::NORTH:
			if (n.lock() != neighbor) hasCheckedNeighbors = false;
			break;
		case Direction::SOUTH:
			if (s.lock() != neighbor) hasCheckedNeighbors = false;
			break;
		case Direction::EAST:
			if (e.lock() != neighbor) hasCheckedNeighbors = false;
			break;
		case Direction::WEST:
			if (w.lock() != neighbor) hasCheckedNeighbors = false;
			break;
		}
	}
}

void ChestTileEntity::checkNeighbors()
{
	if (hasCheckedNeighbors) return;

	// get cached neighbors
	shared_ptr<ChestTileEntity> cachedN = n.lock();
	shared_ptr<ChestTileEntity> cachedE = e.lock();
	shared_ptr<ChestTileEntity> cachedW = w.lock();
	shared_ptr<ChestTileEntity> cachedS = s.lock();

	hasCheckedNeighbors = true;
	n = weak_ptr<ChestTileEntity>();
	e = weak_ptr<ChestTileEntity>();
	w = weak_ptr<ChestTileEntity>();
	s = weak_ptr<ChestTileEntity>();

	NeighborSelectionContext neighborContext =
	{
		level,
		getType(),
		x,
		y,
		z,
		dynamic_pointer_cast<ChestTileEntity>(shared_from_this()),
		cachedN,
		cachedE,
		cachedW,
		cachedS,
	};

	int partnerX = x;
	int partnerZ = z;
	if (getMutualNeighbor(neighborContext, x, y, z, partnerX, partnerZ))
	{
		shared_ptr<ChestTileEntity> partner = dynamic_pointer_cast<ChestTileEntity>(level->getTileEntity(partnerX, y, partnerZ));
		if (partner != nullptr)
		{
			if (partnerZ < z) n = partner;
			else if (partnerZ > z) s = partner;
			else if (partnerX < x) w = partner;
			else if (partnerX > x) e = partner;
		}
	}

	shared_ptr<ChestTileEntity> cteThis = dynamic_pointer_cast<ChestTileEntity>(shared_from_this());
	if (n.lock() != nullptr) n.lock()->heyImYourNeighbor(cteThis, Direction::SOUTH);
	if (s.lock() != nullptr) s.lock()->heyImYourNeighbor(cteThis, Direction::NORTH);
	if (e.lock() != nullptr) e.lock()->heyImYourNeighbor(cteThis, Direction::WEST);
	if (w.lock() != nullptr) w.lock()->heyImYourNeighbor(cteThis, Direction::EAST);
}

bool ChestTileEntity::isSameChest(int x, int y, int z)
{
	Tile *tile = Tile::tiles[level->getTile(x, y, z)];
	if (tile == nullptr || !(dynamic_cast<ChestTile *>(tile) != nullptr)) return false;
	return static_cast<ChestTile *>(tile)->type == getType();
}

void ChestTileEntity::tick()
{
	TileEntity::tick();
	checkNeighbors();

	++tickInterval;
	if (!level->isClientSide && openCount != 0 && (tickInterval + x + y + z) % (SharedConstants::TICKS_PER_SECOND * 10) == 0)
	{
		//            level.tileEvent(x, y, z, Tile.chest.id, ChestTile.EVENT_SET_OPEN_COUNT, openCount);

		openCount = 0;

		float range = 5;
		vector<shared_ptr<Entity> > *players = level->getEntitiesOfClass(typeid(Player), AABB::newTemp(x - range, y - range, z - range, x + 1 + range, y + 1 + range, z + 1 + range));
		if ( players )
		{
			for (auto& it : *players)
			{
				shared_ptr<Player> player = dynamic_pointer_cast<Player>(it);

				ContainerMenu* containerMenu = dynamic_cast<ContainerMenu*>(player->containerMenu);
				if (containerMenu)
				{
					shared_ptr<Container> container = containerMenu->getContainer();
					shared_ptr<Container> thisContainer = dynamic_pointer_cast<Container>(shared_from_this());
					shared_ptr<CompoundContainer> compoundContainer = dynamic_pointer_cast<CompoundContainer>(container);
					if ((container == thisContainer) || (compoundContainer != nullptr && compoundContainer->contains(thisContainer)))
					{
						openCount++;
					}
				}
			}
			delete players;
		}
	}

	oOpenness = openness;

	float speed = 0.10f;
	if (openCount > 0 && openness == 0)
	{
		if (n.lock() == nullptr && w.lock() == nullptr)
		{
			double xc = x + 0.5;
			double zc = z + 0.5;
			if (s.lock() != nullptr) zc += 0.5;
			if (e.lock() != nullptr) xc += 0.5;

			// 4J-PB - Seems the chest open volume is much louder than other sounds from user reports. We'll tone it down a bit
			level->playSound(xc, y + 0.5, zc, eSoundType_RANDOM_CHEST_OPEN, 0.2f, level->random->nextFloat() * 0.1f + 0.9f);
		}
	}
	if ((openCount == 0 && openness > 0) || (openCount > 0 && openness < 1))
	{
		float oldOpen = openness;
		if (openCount > 0) openness += speed;
		else openness -= speed;
		if (openness > 1)
		{
			openness = 1;
		}
		float lim = 0.5f;
		if (openness < lim && oldOpen >= lim)
		{
			// Fix for #64546 - Customer Encountered: TU7: Chests placed by the Player are closing too fast.
			//openness = 0;
			if (n.lock() == nullptr && w.lock() == nullptr)
			{
				double xc = x + 0.5;
				double zc = z + 0.5;
				if (s.lock() != nullptr) zc += 0.5;
				if (e.lock() != nullptr) xc += 0.5;

				// 4J-PB - Seems the chest open volume is much louder than other sounds from user reports. We'll tone it down a bit
				level->playSound(xc, y + 0.5, zc, eSoundType_RANDOM_CHEST_CLOSE, 0.2f, level->random->nextFloat() * 0.1f + 0.9f);
			}
		}
		if (openness < 0)
		{
			openness = 0;
		}
	}

}

bool ChestTileEntity::triggerEvent(int b0, int b1)
{
	if (b0 == ChestTile::EVENT_SET_OPEN_COUNT)
	{
		openCount = b1;
		return true;
	}
	return TileEntity::triggerEvent(b0, b1);
}

void ChestTileEntity::startOpen()
{
	if (openCount < 0)
	{
		openCount = 0;
	}
	openCount++;
	level->tileEvent(x, y, z, getTile()->id, ChestTile::EVENT_SET_OPEN_COUNT, openCount);
	level->updateNeighborsAt(x, y, z, getTile()->id);
	level->updateNeighborsAt(x, y - 1, z, getTile()->id);
}

void ChestTileEntity::stopOpen()
{
	if (getTile() == nullptr || !( dynamic_cast<ChestTile *>( getTile() ) != nullptr)) return;
	openCount--;
	level->tileEvent(x, y, z, getTile()->id, ChestTile::EVENT_SET_OPEN_COUNT, openCount);
	level->updateNeighborsAt(x, y, z, getTile()->id);
	level->updateNeighborsAt(x, y - 1, z, getTile()->id);
}

bool ChestTileEntity::canPlaceItem(int slot, shared_ptr<ItemInstance> item)
{
	return true;
}

void ChestTileEntity::setRemoved()
{
	TileEntity::setRemoved();
	clearCache();
	checkNeighbors();
}

int ChestTileEntity::getType()
{
	if (type == -1)
	{
		if (level != nullptr && dynamic_cast<ChestTile *>( getTile() ) != nullptr)
		{
			type = static_cast<ChestTile *>(getTile())->type;
		}
		else
		{
			return ChestTile::TYPE_BASIC;
		}
	}

	return type;
}

// 4J Added
shared_ptr<TileEntity> ChestTileEntity::clone()
{
	shared_ptr<ChestTileEntity> result = std::make_shared<ChestTileEntity>();
	TileEntity::clone(result);

	for (unsigned int i = 0; i < items->length; i++)
	{
		if (items->data[i] != nullptr)
		{
			result->items->data[i] = ItemInstance::clone(items->data[i]);
		}
	}
	result->hasPlacementPartnerIntent = hasPlacementPartnerIntent;
	result->placementPartnerX = placementPartnerX;
	result->placementPartnerY = placementPartnerY;
	result->placementPartnerZ = placementPartnerZ;
	result->placementPartnerStrict = placementPartnerStrict;
	return result;
}