1 files changed, 20 insertions, 4 deletions
diff --git a/src/T2DLL/CPieChartView.h b/src/T2DLL/CPieChartView.h
index 8980203..0150e59 100644
--- a/src/T2DLL/CPieChartView.h
+++ b/src/T2DLL/CPieChartView.h
@@ -1,11 +1,27 @@
 #pragma once
 #include "common.h"
 
-class CPieChartView {
+class AFX_EXT_CLASS CPieChartView : public CWnd {
 public:
-	static CPieChartView* CreateCPieChartViewStream(T2Archive*);
-	CPieChartView(T2Archive*);
+    struct PieSlice {
+        unsigned int value;
+        COLORREF color;
+    };
+
+    struct Pie {
+        unsigned int totalValue;
+        unsigned int sliceCount;
+        PieSlice *slices;
+    };
+
+	static CPieChartView* CreateCPieChartViewStream(T2Archive* inStream);
+	CPieChartView(T2Archive* inStream);
 	virtual ~CPieChartView();
 	void DrawSelf();
-	void CutPie(const CPieChartView::Pie*);
+	void CutPie(const Pie* inPie);
+
+protected:
+    int mTextTraits;
+    int mBorderWidth;
+    const Pie *mPie;
 };