OpenMiner/source/client/gui/InventoryWidget.cpp

/*
 * =====================================================================================
 *
 *  OpenMiner
 *
 *  Copyright (C) 2018-2020 Unarelith, Quentin Bazin <openminer@unarelith.net>
 *  Copyright (C) 2019-2020 the OpenMiner contributors (see CONTRIBUTORS.md)
 *
 *  This file is part of OpenMiner.
 *
 *  OpenMiner is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public
 *  License as published by the Free Software Foundation; either
 *  version 2.1 of the License, or (at your option) any later version.
 *
 *  OpenMiner 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
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with OpenMiner; if not, write to the Free Software Foundation, Inc.,
 *  51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
 *
 * =====================================================================================
 */
#include "ClientCommandHandler.hpp"
#include "InventoryWidget.hpp"

void InventoryWidget::init(Inventory &inventory, u16 offset, u16 size) {
	m_inventory = &inventory;

	m_offset = offset;
	m_size = size > 0 ? size : inventory.width() * inventory.height();

	loadItemWidgets(m_offset, m_size);

	m_width = inventory.width() * 18;
	m_height = inventory.height() * 18;

	m_inventoryWidth = inventory.width();
	m_inventoryHeight = inventory.height();
}

void InventoryWidget::scroll(float scrolling) {
	if (m_itemWidgets.size() < m_size)
		return;

	u16 offset = m_offset + floor((m_inventory->height() - m_size / m_inventory->width()) * scrolling) * m_inventory->width();
	u16 size = m_size;

	if (offset + size > m_inventory->width() * m_inventory->height())
		size = m_inventory->width() * m_inventory->height() - offset;

	loadItemWidgets(offset, size, m_lastSearch);
}

void InventoryWidget::onEvent(const SDL_Event &event) {
	if (event.type == SDL_MOUSEMOTION) {
		m_currentItemWidget = nullptr;
		for (std::size_t i = 0 ; i < m_itemWidgets.size() ; ++i) {
			if (m_itemWidgets[i].isPointInWidget(event.motion.x, event.motion.y)) {
				m_currentItemWidget = &m_itemWidgets[i];

				m_selectedItemBackground.setPosition(1 + (i % m_inventoryWidth) * 18, 1 + (i / m_inventoryWidth) * 18, 0);
			}
		}
	}
}

void InventoryWidget::update() {
	bool hasChanged = false;

	for (auto &it : m_itemWidgets) {
		it.update();

		hasChanged = hasChanged || it.hasChanged();
		it.setChanged(false);
	}

	if (hasChanged)
		sendUpdatePacket();
}

void InventoryWidget::applySearch(const std::string &search) {
	if (search != m_lastSearch) {
		loadItemWidgets(m_offset, m_size, search);
		m_lastSearch = search;
	}
}

bool InventoryWidget::sendItemStackToDest(const ItemWidget *itemStack, AbstractInventoryWidget *dest) {
	if (dest->doItemMatchFilter(itemStack->stack().item())) {
		ItemStack stackRet = dest->receiveItemStack(itemStack, this);
		if (stackRet.amount() != itemStack->stack().amount()) {
			if (dest != this && stackRet.amount() == 0)
				m_inventory->clearStack(itemStack->x(), itemStack->y());

			update();
			sendUpdatePacket();
			return true;
		}
	}

	return false;
}

ItemStack InventoryWidget::receiveItemStack(const ItemWidget *itemStack, AbstractInventoryWidget *src) {
	const ItemStack &stack = itemStack->stack();
	if (src == this)
		m_inventory->clearStack(itemStack->x(), itemStack->y());

	ItemStack stackRet = m_inventory->addStack(stack.item().stringID(), stack.amount(), m_offset, m_size);

	if (stackRet.amount() != stack.amount())
		sendUpdatePacket();

	return stackRet;
}

void InventoryWidget::sendUpdatePacket() {
	if (m_inventory->inBlock()) {
		m_client.sendBlockInvUpdate(*m_inventory);
	}
	else {
		m_client.sendPlayerInvUpdate();
	}
}

void InventoryWidget::draw(gk::RenderTarget &target, gk::RenderStates states) const {
	states.transform *= getTransform();

	for (std::size_t i = 0 ; i < m_itemWidgets.size() ; ++i) {
		target.draw(m_itemWidgets[i], states);
	}

	if (m_currentItemWidget)
		target.draw(m_selectedItemBackground, states);
}

void InventoryWidget::loadItemWidgets(u16 offset, u16 size, std::string search) {
	m_itemWidgets.clear();

	u16 itemCounter = 0;
	for (u16 i = 0 ; itemCounter < size ; ++i) {
		u16 x = (i + offset) % m_inventory->width();
		u16 y = (i + offset) / m_inventory->width();
		if (x >= m_inventory->width() || y >= m_inventory->height())
			break;

		std::string label = m_inventory->getStack(x, y).item().label();
		std::transform(label.begin(), label.end(), label.begin(), [](unsigned char c) { return std::tolower(c); });
		std::transform(search.begin(), search.end(), search.begin(), [](unsigned char c) { return std::tolower(c); });
		if (search.empty() || label.find(search) != std::string::npos) {
			m_itemWidgets.emplace_back(*m_inventory, x, y, this);

			ItemWidget &widget = m_itemWidgets.back();
			widget.update();
			widget.setPosition((itemCounter % m_inventory->width()) * 18, (itemCounter / m_inventory->width()) * 18, 0);

			itemCounter++;
		}
	}
}