Qtscript: Add eventStructureDemolish.

An event that triggers only when a completely built structures starts to be demolished. Refs ticket 4790.
Warzone2100 · Aug 27, 2018 · 349e73f · 349e73f
1 parent d02157d
commit 349e73f
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 1 deletion.
diff --git a/src/qtscript.cpp b/src/qtscript.cpp
@@ -1338,6 +1338,32 @@ bool triggerEventStructBuilt(STRUCTURE *psStruct, DROID *psDroid)
 	return true;
 }
 
+//__ ## eventStructureDemolish(structure[, droid])
+//__
+//__ An event that is run every time a structure begins to be demolished. This does
+//__ not trigger again if the structure is partially demolished.
+//__
+bool triggerEventStructDemolish(STRUCTURE *psStruct, DROID *psDroid)
+{
+	ASSERT(scriptsReady, "Scripts not initialized yet");
+	for (auto *engine : scripts)
+	{
+		int player = engine->globalObject().property("me").toInt32();
+		bool receiveAll = engine->globalObject().property("isReceivingAllEvents").toBool();
+		if (player == psStruct->player || receiveAll)
+		{
+			QScriptValueList args;
+			args += convStructure(psStruct, engine);
+			if (psDroid)
+			{
+				args += convDroid(psDroid, engine);
+			}
+			callFunction(engine, "eventStructureDemolish", args);
+		}
+	}
+	return true;
+}
+
 //__ ## eventStructureReady(structure)
 //__
 //__ An event that is run every time a structure is ready to perform some
@@ -1893,4 +1919,3 @@ void to_json(nlohmann::json& j, const QVariant& value) {
 			}
 	}
 }
-
diff --git a/src/qtscript.h b/src/qtscript.h
@@ -121,6 +121,7 @@ bool triggerEventDroidBuilt(DROID *psDroid, STRUCTURE *psFactory);
 bool triggerEventAttacked(BASE_OBJECT *psVictim, BASE_OBJECT *psAttacker, int lastHit);
 bool triggerEventResearched(RESEARCH *psResearch, STRUCTURE *psStruct, int player);
 bool triggerEventStructBuilt(STRUCTURE *psStruct, DROID *psDroid);
+bool triggerEventStructDemolish(STRUCTURE *psStruct, DROID *psDroid);
 bool triggerEventDroidIdle(DROID *psDroid);
 bool triggerEventDestroyed(BASE_OBJECT *psVictim);
 bool triggerEventStructureReady(STRUCTURE *psStruct);

diff --git a/src/structure.cpp b/src/structure.cpp
@@ -842,6 +842,14 @@ void structureBuild(STRUCTURE *psStruct, DROID *psDroid, int buildPoints, int bu
 		if (prevStatus == SS_BUILT)
 		{
 			// Starting to demolish.
+			if (psDroid)
+			{
+				triggerEventStructDemolish(psStruct, psDroid);
+			}
+			else
+			{
+				triggerEventStructDemolish(psStruct, nullptr);
+			}
 			switch (psStruct->pStructureType->type)
 			{
 			case REF_POWER_GEN: