1 files changed, 25 insertions, 6 deletions
diff --git a/src/T2DLL/CPiledMultiBarChartView.h b/src/T2DLL/CPiledMultiBarChartView.h
index 4a54d23..4beb180 100644
--- a/src/T2DLL/CPiledMultiBarChartView.h
+++ b/src/T2DLL/CPiledMultiBarChartView.h
@@ -1,14 +1,33 @@
 #pragma once
 #include "common.h"
 
-class CPiledMultiBarChartView {
+class AFX_EXT_CLASS CPiledMultiBarChartView : public CWnd {
 public:
-	static CPiledMultiBarChartView* CreateCPiledMultiBarChartViewStream(T2Archive*);
-	CPiledMultiBarChartView(T2Archive*);
+	struct GraphData {
+		int piles, bars;
+		int *data;
+	};
+
+	static CPiledMultiBarChartView* CreateCPiledMultiBarChartViewStream(T2Archive* inStream);
+	CPiledMultiBarChartView(T2Archive* inStream);
 	virtual ~CPiledMultiBarChartView();
-	void SetGraphData(const CPiledMultiBarChartView::GraphData*);
-	void SetPixelPerValue(int, int);
-	void SetColorOfPiles(int, const unsigned long*, int);
+	void SetGraphData(const GraphData* inData);
+	void SetPixelPerValue(int inPixelPerValue, BOOL inRefresh);
+	void SetColorOfPiles(int inCount, const COLORREF* inColors, BOOL inRefresh);
 	void Refresh();
 	void DrawSelf();
+
+protected:
+	BOOL mSkipDrawing;
+	int mPiles;
+	int mBars;
+	int mBarWidth;
+	int mBarInterval;
+	int mXOffset;
+	int mLines;
+	int mLineInterval;
+	int mLineLength;
+	int mPixelPerValue;
+	COLORREF *mColorOfPiles;
+	const GraphData *mGraphData;
 };