1 files changed, 28 insertions, 4 deletions

diff --git a/src/Entities/FireworkEntity.h b/src/Entities/FireworkEntity.h
index 20f18b6dc..300ec571e 100644
--- a/src/Entities/FireworkEntity.h
+++ b/src/Entities/FireworkEntity.h
@@ -1,6 +1,11 @@
-//
-//  FireworkEntity.h
-//
+
+// FireworkEntity.h
+
+// Declares the cFireworkEntity class representing the flying firework rocket
+
+
+
+
 
 #pragma once
 
@@ -24,7 +29,22 @@ public:
 	CLASS_PROTODEF(cFireworkEntity)
 	
 	cFireworkEntity(cEntity * a_Creator, double a_X, double a_Y, double a_Z, const cItem & a_Item);
+
+	// tolua_begin
+
+	/** Returns the item used to create the rocket (has all the firework effects on it) */
 	const cItem & GetItem(void) const { return m_FireworkItem; }
+
+	/** Sets the item that is used to create the rocket (has all the firework effects on it) */
+	void SetItem(const cItem & a_Item) { m_FireworkItem = a_Item; }
+
+	/** Returns the number of ticks left until the firework explosion. */
+	int GetTicksToExplosion(void) const { return m_TicksToExplosion; }
+
+	/** Sets the number of ticks left until the firework explosion. */
+	void SetTicksToExplosion(int a_TicksToExplosion) { m_TicksToExplosion = a_TicksToExplosion; }
+
+	// tolua_end
 	
 protected:
 	
@@ -34,7 +54,11 @@ protected:
 	
 private:
 	
-	int m_ExplodeTimer;
+	int m_TicksToExplosion;
 	cItem m_FireworkItem;
 	
 };  // tolua_export
+
+
+
+