1 files changed, 8 insertions, 9 deletions
diff --git a/src/T2DLL/T2DlgItemStressGage.h b/src/T2DLL/T2DlgItemStressGage.h
index 8e90fa0..9240f30 100644
--- a/src/T2DLL/T2DlgItemStressGage.h
+++ b/src/T2DLL/T2DlgItemStressGage.h
@@ -1,21 +1,20 @@
 #pragma once
 #include "common.h"
+#include "T2DlgItemGageBase.h"
 
-class T2DlgItemStressGage {
+class AFX_EXT_CLASS T2DlgItemStressGage : public T2DlgItemGageBase {
 public:
-	T2DlgItemStressGage(T2TowerDoc*, T2ImageObj*, CPalette*);
+	T2DlgItemStressGage(T2TowerDoc* towerDoc, T2ImageObj* imageObj, CPalette* palette);
+	void SetPeople(T2People*);
+
 protected:
 	virtual int GetMinValue();
 	virtual int GetMaxValue();
 	virtual int GetValue();
 	virtual int GetBlueValue();
 	virtual int GetYellowValue();
-public:
-	void SetPeople(T2People*);
-protected:
-	virtual unsigned long GetGageColor(int);
-	virtual int IsDrawInterior();
+	virtual COLORREF GetGageColor(int value);
+	virtual BOOL IsDrawInterior();
 
-public:
-	virtual ~T2DlgItemStressGage() {}
+	T2People *mPeople;
 };