3 files changed, 74 insertions, 0 deletions

diff --git a/src/game/g_buildable.c b/src/game/g_buildable.c
index 72d4d3c..37e3e66 100644
--- a/src/game/g_buildable.c
+++ b/src/game/g_buildable.c
@@ -391,6 +391,8 @@ Get the number of build points from a position.
 
 If sudden death has started, the returned value might be negative,
 but is never positive.
+
+Note: 'pos' can be NULL, in this case return the overall BP of the team.
 ==================
 */
 int G_GetBuildPoints( const vec3_t pos, team_t team )
diff --git a/src/game/g_local.h b/src/game/g_local.h
index 4f03aad..44c6996 100644
--- a/src/game/g_local.h
+++ b/src/game/g_local.h
@@ -670,6 +670,7 @@ typedef struct
   timeWarning_t     timelimitWarning;
   
   int               nextArmageddonKillTime;
+  int               nextNegativeBPCheck;
   spawnQueue_t      alienSpawnQueue;
   spawnQueue_t      humanSpawnQueue;
 
diff --git a/src/game/g_main.c b/src/game/g_main.c
index 3d9c52b..2a8fb13 100644
--- a/src/game/g_main.c
+++ b/src/game/g_main.c
@@ -366,6 +366,7 @@ void G_ShutdownGame( int restart );
 void CheckExitRules( void );
 void G_CountSpawns( void );
 void G_CalculateBuildPoints( void );
+void G_CheckForNegativeBuildPoints( void );
 
 /*
 ================
@@ -1353,6 +1354,75 @@ void G_CalculateBuildPoints( void )
 //  if( level.alienBuildPoints < 0 )
 //    level.alienBuildPoints = 0;
 }
+
+/*
+============
+G_CheckForNegativeBuildPoints
+
+Recalculate the quantity of building points available to the teams
+============
+*/
+void G_CheckForNegativeBuildPoints( void )
+{
+  static const int thinkduration = 1037;
+  int e;
+  gentity_t *ent;
+  const buildableAttributes_t *ba;
+  float dieprob1min;
+  float surviveprob1min;
+  float surviveprobcur;
+  int a_bps, h_bps;
+
+  if( level.nextNegativeBPCheck > level.time )
+    return;
+  level.nextNegativeBPCheck += thinkduration;
+
+  a_bps = G_GetBuildPoints( NULL, TEAM_ALIENS );
+  h_bps = G_GetBuildPoints( NULL, TEAM_HUMANS );
+
+  if( ( a_bps >= 0 ) && ( h_bps >= 0 ) )
+    return;
+
+  for( e = 0; e < level.num_entities; ++e ) {
+    ent = g_entities + e;
+
+    if (!ent->inuse)
+      continue;
+    ba = BG_Buildable( ent->s.modelindex );
+
+    // no gain from a building without BPs
+    if( ba->buildPoints <= 0 )
+      continue;
+
+    // TODO: Add a separate probability for each buildable.
+    // If it's 0, then they won't die at all, and the checks for building types
+    // won't be needed.
+    dieprob1min = 0.42f; /* hard-wired */
+    if( dieprob1min <= 0.0f )
+      continue;
+    surviveprob1min = 1.0f - dieprob1min;
+    surviveprob1min = MAX( 0.0f, surviveprob1min );
+
+    // check for buildings that must not die
+    switch( ent->s.modelindex ) {
+      case BA_H_REACTOR:
+      case BA_H_REFINERY:
+      case BA_A_OVERMIND:
+      case BA_A_CREEPCOLONY:
+        continue;
+    }
+    // TODO end
+
+    if( ( ( ent->buildableTeam == TEAM_ALIENS ) && ( a_bps < 0 ) ) ||
+        ( ( ent->buildableTeam == TEAM_HUMANS ) && ( h_bps < 0 ) ) )
+    {
+      surviveprobcur = pow( surviveprob1min, thinkduration / 60000.0f );
+      if( surviveprobcur * RAND_MAX < rand( ) )
+        G_Suicide( ent, MOD_NOBP );
+    }
+  }
+}
+
  /*
  ============
 G_HumanBuildPoints
@@ -2747,6 +2817,7 @@ void G_RunFrame( int levelTime )
 
   G_CountSpawns( );
   G_CalculateBuildPoints( );
+  G_CheckForNegativeBuildPoints( );
   G_CalculateStages( );
   G_SpawnClients( TEAM_ALIENS );
   G_SpawnClients( TEAM_HUMANS );