1 files changed, 13 insertions, 9 deletions
diff --git a/src/T2DLL/T2DlgItemPopup.h b/src/T2DLL/T2DlgItemPopup.h
index 8b95cb2..f7d3247 100644
--- a/src/T2DLL/T2DlgItemPopup.h
+++ b/src/T2DLL/T2DlgItemPopup.h
@@ -1,16 +1,20 @@
 #pragma once
 #include "common.h"
+#include "T2DlgItem.h"
 
-class T2DlgItemPopup {
+class AFX_EXT_CLASS T2DlgItemPopup : public T2DlgItem {
 public:
-	T2DlgItemPopup(T2TowerDoc*, T2ImageObj*, CPalette*);
+	T2DlgItemPopup(T2TowerDoc* inDoc, T2ImageObj* inImageObj, CPalette* inPalette);
 	virtual ~T2DlgItemPopup();
+	void AppendItem(const CString& inName);
+
 protected:
-	virtual int OnT2DlgItemCreate(CREATESTRUCTA*);
-	virtual int OnT2DlgItemEraseBkgnd(CDC*);
-	virtual void OnT2DlgItemLButtonDown(unsigned int, CPoint);
-	virtual void OnT2DlgItemLButtonUp(unsigned int, CPoint);
-	virtual int OnCommand(unsigned int, long);
-public:
-	void AppendItem(const CString&);
+	virtual int OnT2DlgItemCreate(CREATESTRUCT* cs);
+	virtual BOOL OnT2DlgItemEraseBkgnd(CDC* pDC);
+	virtual void OnT2DlgItemLButtonDown(UINT inFlags, CPoint inPt);
+	virtual void OnT2DlgItemLButtonUp(UINT inFlags, CPoint inPt);
+	virtual BOOL OnCommand(WPARAM wParam, LPARAM lParam);
+
+	CMenu *mMenu;
+	BOOL mMenuClicked;
 };