1 files changed, 26 insertions, 8 deletions
diff --git a/src/T2DLL/T2FireWork.h b/src/T2DLL/T2FireWork.h
index e9e5215..37b3fc0 100644
--- a/src/T2DLL/T2FireWork.h
+++ b/src/T2DLL/T2FireWork.h
@@ -1,19 +1,37 @@
 #pragma once
 #include "common.h"
 
-class T2FireWork {
+struct FIREWORK_STARINFO {
+	float m0;
+	float m4;
+	float m8;
+	float mC;
+	float m10;
+	float m14;
+	float m18;
+	float m1C;
+	unsigned short mFlags;
+	float m24;
+};
+
+class AFX_EXT_CLASS T2FireWork {
 public:
-	T2FireWork(int, int, CPalette*);
+	T2FireWork(int inWidth, int inHeight, CPalette* inPalette);
 	~T2FireWork();
 	T2BitImage* GetBitImage();
 	void Start();
 	void Stop();
-	int IsDisposable();
-	int Idle();
+	BOOL IsDisposable();
+	BOOL Idle();
 protected:
-	int CalcStarData(FIREWORK_STARINFO*, float, int*, int*, int*, int*);
-	void PutDot(int, int, int, unsigned char);
+	BOOL CalcStarData(FIREWORK_STARINFO* inInfo, float inT, int* outX, int* outY, int* outPixel, int* outSize);
+	void PutDot(int inX, int inY, int inSize, unsigned char inCol);
 
-public:
-	T2FireWork& operator=(const T2FireWork&) {}
+	BOOL mIsActive;
+	T2BitImage *mImage;
+	int mRowSize;
+	RECT mRect;
+	unsigned char mPalettes[5][8];
+	FIREWORK_STARINFO *mStarInfo[20];
+	BOOL mUsed[20];
 };