1 files changed, 17 insertions, 0 deletions
diff --git a/src/T2GraphWindow.h b/src/T2GraphWindow.h
index fcaf1ff..4286e73 100644
--- a/src/T2GraphWindow.h
+++ b/src/T2GraphWindow.h
@@ -1,3 +1,20 @@
 #pragma once
 #include "common.h"
 
+class T2GraphWindow : public CFrameWnd {
+    DECLARE_DYNCREATE(T2GraphWindow)
+    DECLARE_MESSAGE_MAP()
+
+public:
+    T2GraphWindow();
+    virtual ~T2GraphWindow();
+    void SetDocument(T2TowerDoc *inDoc);
+
+protected:
+    BOOL OnEraseBkgnd(CDC *pDC);
+    void OnHScroll(UINT nSBCode, UINT nPos, CScrollBar *pScrollBar);
+    void OnSize(UINT nType, int cx, int cy);
+    BOOL OnQueryNewPalette();
+
+    T2TowerDoc *mDocument;
+};