1 files changed, 13 insertions, 8 deletions
diff --git a/src/T2DLL/T2DlgItemTab.h b/src/T2DLL/T2DlgItemTab.h
index 234cd0a..f728837 100644
--- a/src/T2DLL/T2DlgItemTab.h
+++ b/src/T2DLL/T2DlgItemTab.h
@@ -1,15 +1,20 @@
 #pragma once
 #include "common.h"
+#include "T2DlgItem.h"
 
-class T2DlgItemTab {
+class CustomTabControl;
+
+class AFX_EXT_CLASS T2DlgItemTab : public T2DlgItem {
 public:
-	T2DlgItemTab(T2TowerDoc*, T2ImageObj*, CPalette*);
+	T2DlgItemTab(T2TowerDoc* inDoc, T2ImageObj* inImageObj, CPalette* inPalette);
 	virtual ~T2DlgItemTab();
-	virtual int OnT2DlgItemCreate(CREATESTRUCTA*);
-	virtual void SetFont(HFONT);
-	virtual void CreateSubItem(void*);
+	virtual int OnT2DlgItemCreate(CREATESTRUCT* cs);
+	virtual void SetFont(HFONT inFont);
+	virtual void CreateSubItem(void* inData);
+	virtual LRESULT WindowProc(UINT message, WPARAM wParam, LPARAM lParam);
+
 protected:
-	virtual int OnT2DlgItemEraseBkgnd(CDC*);
-public:
-	virtual long WindowProc(unsigned int, unsigned int, long);
+	virtual BOOL OnT2DlgItemEraseBkgnd(CDC* pDC);
+
+	CustomTabControl *mSubItem;
 };