1 files changed, 86 insertions, 13 deletions

diff --git a/source/cWindowOwner.h b/source/cWindowOwner.h
index ef08bbe44..6c9a15a45 100644
--- a/source/cWindowOwner.h
+++ b/source/cWindowOwner.h
@@ -2,6 +2,7 @@
 #pragma once
 
 #include "cBlockEntity.h"
+#include "cEntity.h"
 
 
 
@@ -14,29 +15,101 @@ class cWindow;
 
 
 /**
-Implements the base behavior expected from a class that can handle UI windows for block entities.
+Base class for the behavior expected from a class that can handle UI windows for block entities.
 */
 class cWindowOwner
 {
 public:
-	cWindowOwner() : m_Window( NULL ) {}
-	void CloseWindow() { m_Window = NULL; }
-	void OpenWindow( cWindow* a_Window ) { m_Window = a_Window; }
+	cWindowOwner() :
+		m_Window(NULL)
+	{
+	}
+	
+	void CloseWindow(void)
+	{
+		m_Window = NULL;
+	}
+	
+	void OpenWindow(cWindow * a_Window)
+	{
+		m_Window = a_Window;
+	}
+
+	cWindow * GetWindow(void) const
+	{
+		return m_Window;
+	}
+
+	/// Returns the block position at which the element owning the window is
+	virtual void GetBlockPos(int & a_BlockX, int & a_BlockY, int & a_BlockZ) = 0;
+	
+private:
+	cWindow * m_Window;
+} ;
+
+
+
+
+
+/**
+Window owner that is associated with a block entity (chest, furnace, ...)
+*/
+class cBlockEntityWindowOwner :
+	public cWindowOwner
+{
+public:
+	cBlockEntityWindowOwner(void) :
+		m_BlockEntity(NULL)
+	{
+	}
+	
+	void SetBlockEntity(cBlockEntity * a_BlockEntity)
+	{
+		m_BlockEntity = a_BlockEntity;
+	}
+	
+	virtual void GetBlockPos(int & a_BlockX, int & a_BlockY, int & a_BlockZ) override
+	{
+		a_BlockX = m_BlockEntity->GetPosX();
+		a_BlockY = m_BlockEntity->GetPosY();
+		a_BlockZ = m_BlockEntity->GetPosZ();
+	}
+	
+private:
+	cBlockEntity * m_BlockEntity;
+} ;
+
+
 
-	cWindow* GetWindow() { return m_Window; }
 
-	void SetEntity(cBlockEntity * a_Entity) { m_Entity = a_Entity; }
-	void GetBlockPos(int & a_BlockX, int & a_BlockY, int & a_BlockZ)
+
+/**
+Window owner that is associated with an entity (chest minecart)
+*/
+class cEntityWindowOwner :
+	public cWindowOwner
+{
+public:
+	cEntityWindowOwner(void) :
+		m_Entity(NULL)
+	{
+	}
+	
+	void SetEntity(cEntity * a_Entity)
+	{
+		m_Entity = a_Entity;
+	}
+
+	virtual void GetBlockPos(int & a_BlockX, int & a_BlockY, int & a_BlockZ) override
 	{
-		a_BlockX = m_Entity->GetPosX();
-		a_BlockY = m_Entity->GetPosY();
-		a_BlockZ = m_Entity->GetPosZ();
+		a_BlockX = (int)(m_Entity->GetPosX());
+		a_BlockY = (int)(m_Entity->GetPosY());
+		a_BlockZ = (int)(m_Entity->GetPosZ());
 	}
 	
 private:
-	cWindow      * m_Window;
-	cBlockEntity * m_Entity;
-};
+	cEntity * m_Entity;
+} ;