1 files changed, 10 insertions, 5 deletions
diff --git a/src/T2DLL/T2TreasureDialog.h b/src/T2DLL/T2TreasureDialog.h
index 8156f23..b0fb3fe 100644
--- a/src/T2DLL/T2TreasureDialog.h
+++ b/src/T2DLL/T2TreasureDialog.h
@@ -1,17 +1,22 @@
 #pragma once
 #include "common.h"
+#include "T2Dialog.h"
 
-class T2TreasureDialog {
+class AFX_EXT_CLASS T2TreasureDialog : public T2Dialog {
 public:
 	T2TreasureDialog();
 	virtual ~T2TreasureDialog();
+	void SetTreasureInfo(T2TowerDoc* inTowerDoc, T2TreasureStatus* inStatus);
+	static void ShowTreasureDialog(T2TowerDoc* inTowerDoc, T2TreasureStatus* inStatus);
+
 protected:
 	virtual void OnT2Create();
-	virtual void CreateDlgItem(CTokenizer&, T2Dialog::T2DialogDef&);
+	virtual void CreateDlgItem(CTokenizer& inTokenizer, T2DialogDef& inDef);
 	virtual void OnT2Destroy();
 	virtual void ShowSelf();
 	virtual void HideSelf();
-public:
-	static void ShowTreasureDialog(T2TowerDoc*, T2TreasureStatus*);
-	void SetTreasureInfo(T2TowerDoc*, T2TreasureStatus*);
+
+	T2TreasurePane *mTreasurePane;
+	T2DlgItem *mTextPane;
+	char mText[256];
 };