1 files changed, 87 insertions, 0 deletions
diff --git a/src/UI/BeaconWindow.h b/src/UI/BeaconWindow.h
new file mode 100644
index 000000000..a2ec50f02
--- /dev/null
+++ b/src/UI/BeaconWindow.h
@@ -0,0 +1,87 @@
+
+// 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::DistributeStack(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::DistributeStack(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;
+};
+
+
+
+