path: root/src/UI/BeaconWindow.h

// BeaconWindow.h

// Representing the UI window for the beacon block





#pragma once

#include "Window.h"
#include "../Entities/Player.h"





class cBeaconWindow :
	public cWindow
{
	typedef cWindow super;

public:
	cBeaconWindow(int a_BlockX, int a_BlockY, int a_BlockZ, cBeaconEntity * a_Beacon) :
		cWindow(wtBeacon, "Beacon"),
		m_Beacon(a_Beacon)
	{
		m_SlotAreas.push_back(new cSlotAreaBeacon(m_Beacon, *this));
		m_SlotAreas.push_back(new cSlotAreaInventory(*this));
		m_SlotAreas.push_back(new cSlotAreaHotBar(*this));
	}


	cBeaconEntity * GetBeaconEntity(void) const { return m_Beacon; }

	
	virtual void DistributeStack(cItem & a_ItemStack, int a_Slot, cPlayer & a_Player, cSlotArea * a_ClickedArea, bool a_ShouldApply) override
	{
		cSlotAreas AreasInOrder;

		if (a_ClickedArea == m_SlotAreas[0])
		{
			// Beacon Area
			AreasInOrder.push_back(m_SlotAreas[2]);  /* Hotbar    */
			AreasInOrder.push_back(m_SlotAreas[1]);  /* Inventory */
			super::DistributeStackToAreas(a_ItemStack, a_Player, AreasInOrder, a_ShouldApply, true);
		}
		else
		{
			if (cSlotAreaBeacon::IsPlaceableItem(a_ItemStack.m_ItemType) && (a_ItemStack.m_ItemCount == 1))
			{
				AreasInOrder.push_back(m_SlotAreas[0]);  /* Beacon */
			}

			if (a_ClickedArea == m_SlotAreas[1])
			{
				// Inventory Area
				AreasInOrder.push_back(m_SlotAreas[2]);  /* Hotbar */
			}
			else
			{
				// Hotbar Area
				AreasInOrder.push_back(m_SlotAreas[1]);  /* Inventory */
			}
			super::DistributeStackToAreas(a_ItemStack, a_Player, AreasInOrder, a_ShouldApply, false);
		}
	}


	// cWindow Overrides:
	virtual void OpenedByPlayer(cPlayer & a_Player) override
	{
		super::OpenedByPlayer(a_Player);

		a_Player.GetClientHandle()->SendWindowProperty(*this, 0, m_Beacon->GetBeaconLevel());
		a_Player.GetClientHandle()->SendWindowProperty(*this, 1, m_Beacon->GetPrimaryEffect());
		a_Player.GetClientHandle()->SendWindowProperty(*this, 2, m_Beacon->GetSecondaryEffect());
	}

protected:
	cBeaconEntity * m_Beacon;
};