1 files changed, 14 insertions, 9 deletions
diff --git a/src/T2DLL/T2DlgItemAllPurpose.h b/src/T2DLL/T2DlgItemAllPurpose.h
index 3797e11..6aef4c6 100644
--- a/src/T2DLL/T2DlgItemAllPurpose.h
+++ b/src/T2DLL/T2DlgItemAllPurpose.h
@@ -1,17 +1,22 @@
 #pragma once
 #include "common.h"
+#include "T2DlgItem.h"
 
-class T2DlgItemAllPurpose {
+class AFX_EXT_CLASS T2DlgItemAllPurpose : public T2DlgItem {
 public:
-	T2DlgItemAllPurpose(T2TowerDoc*, T2ImageObj*, CPalette*);
+	T2DlgItemAllPurpose(T2TowerDoc* inDoc, T2ImageObj* inImageObj, CPalette* inPalette);
 	virtual ~T2DlgItemAllPurpose();
-protected:
-	virtual int OnT2DlgItemEraseBkgnd(CDC*);
-	virtual void OnT2DlgItemLButtonDown(unsigned int, CPoint);
-	virtual void OnT2DlgItemLButtonUp(unsigned int, CPoint);
-	virtual void OnT2DlgItemRButtonDown(unsigned int, CPoint);
-	virtual void OnT2DlgItemRButtonUp(unsigned int, CPoint);
-public:
 	CPoint GetUpPoint();
 	CPoint GetDownPoint();
+
+protected:
+	virtual BOOL OnT2DlgItemEraseBkgnd(CDC*);
+	virtual void OnT2DlgItemLButtonDown(UINT, CPoint);
+	virtual void OnT2DlgItemLButtonUp(UINT, CPoint);
+	virtual void OnT2DlgItemRButtonDown(UINT, CPoint);
+	virtual void OnT2DlgItemRButtonUp(UINT, CPoint);
+
+	CPoint *mDownPoint;
+	CPoint *mUpPoint;
+	int mButton;
 };