(svn r24186) -Feature: [NewGRF] Callback to set industry production level on construction. (andythenorth)

3 files changed, 25 insertions, 5 deletions

diff --git a/src/industry_cmd.cpp b/src/industry_cmd.cpp
index f331f3a21..e5329e15b 100644
--- a/src/industry_cmd.cpp
+++ b/src/industry_cmd.cpp
@@ -1655,8 +1655,6 @@ static void DoCreateNewIndustry(Industry *i, TileIndex tile, IndustryType type,
 	i->last_month_transported[1] = 0;
 	i->was_cargo_delivered = false;
 	i->last_prod_year = _cur_year;
-	i->last_month_production[0] = i->production_rate[0] * 8;
-	i->last_month_production[1] = i->production_rate[1] * 8;
 	i->founder = founder;
 
 	i->construction_date = _date;
@@ -1668,12 +1666,29 @@ static void DoCreateNewIndustry(Industry *i, TileIndex tile, IndustryType type,
 	 * else, chosen layout + 1 */
 	i->selected_layout = layout + 1;
 
-	if (!_generating_world) i->last_month_production[0] = i->last_month_production[1] = 0;
-
 	i->prod_level = PRODLEVEL_DEFAULT;
 
 	/* Call callbacks after the regular fields got initialised. */
 
+	if (HasBit(indspec->callback_mask, CBM_IND_PROD_CHANGE_BUILD)) {
+		uint16 res = GetIndustryCallback(CBID_INDUSTRY_PROD_CHANGE_BUILD, 0, Random(), i, type, INVALID_TILE);
+		if (res != CALLBACK_FAILED) {
+			if (res < PRODLEVEL_MINIMUM || res > PRODLEVEL_MAXIMUM) {
+				ErrorUnknownCallbackResult(indspec->grf_prop.grffile->grfid, CBID_INDUSTRY_PROD_CHANGE_BUILD, res);
+			} else {
+				i->prod_level = res;
+				i->RecomputeProductionMultipliers();
+			}
+		}
+	}
+
+	if (_generating_world) {
+		i->last_month_production[0] = i->production_rate[0] * 8;
+		i->last_month_production[1] = i->production_rate[1] * 8;
+	} else {
+		i->last_month_production[0] = i->last_month_production[1] = 0;
+	}
+
 	if (HasBit(indspec->callback_mask, CBM_IND_DECIDE_COLOUR)) {
 		uint16 res = GetIndustryCallback(CBID_INDUSTRY_DECIDE_COLOUR, 0, 0, i, type, INVALID_TILE);
 		if (res != CALLBACK_FAILED) {
@@ -2741,7 +2756,7 @@ bool IndustrySpec::UsesSmoothEconomy() const
 {
 	return _settings_game.economy.smooth_economy &&
 		!(HasBit(this->callback_mask, CBM_IND_PRODUCTION_256_TICKS) || HasBit(this->callback_mask, CBM_IND_PRODUCTION_CARGO_ARRIVAL)) && // production callbacks
-		!(HasBit(this->callback_mask, CBM_IND_MONTHLYPROD_CHANGE) || HasBit(this->callback_mask, CBM_IND_PRODUCTION_CHANGE));            // production change callbacks
+		!(HasBit(this->callback_mask, CBM_IND_MONTHLYPROD_CHANGE) || HasBit(this->callback_mask, CBM_IND_PRODUCTION_CHANGE) || HasBit(this->callback_mask, CBM_IND_PROD_CHANGE_BUILD)); // production change callbacks
 }
 
 static CommandCost TerraformTile_Industry(TileIndex tile, DoCommandFlag flags, int z_new, Slope tileh_new)
diff --git a/src/newgrf_callbacks.h b/src/newgrf_callbacks.h
index 546ba630d..e96d06803 100644
--- a/src/newgrf_callbacks.h
+++ b/src/newgrf_callbacks.h
@@ -275,6 +275,9 @@ enum CallbackID {
 
 	/** Called to determine the cost factor for refitting a vehicle. */
 	CBID_VEHICLE_REFIT_COST              = 0x15E, // 15 bit callback
+
+	/** Called when industry is built to set initial production level. */
+	CBID_INDUSTRY_PROD_CHANGE_BUILD      = 0x15F, // 15 bit callback
 };
 
 /**
@@ -355,6 +358,7 @@ enum IndustryCallbackMask {
 	CBM_IND_DECIDE_COLOUR             = 11, ///< give a custom colour to newly build industries
 	CBM_IND_INPUT_CARGO_TYPES         = 12, ///< customize the cargoes the industry requires
 	CBM_IND_OUTPUT_CARGO_TYPES        = 13, ///< customize the cargoes the industry produces
+	CBM_IND_PROD_CHANGE_BUILD         = 14, ///< initialise production level on construction
 };
 
 /**
diff --git a/src/table/newgrf_debug_data.h b/src/table/newgrf_debug_data.h
index 040a275b1..3ea83176d 100644
--- a/src/table/newgrf_debug_data.h
+++ b/src/table/newgrf_debug_data.h
@@ -265,6 +265,7 @@ static const NICallback _nic_industries[] = {
 	NICI(CBID_INDUSTRY_DECIDE_COLOUR,        CBM_IND_DECIDE_COLOUR),
 	NICI(CBID_INDUSTRY_INPUT_CARGO_TYPES,    CBM_IND_INPUT_CARGO_TYPES),
 	NICI(CBID_INDUSTRY_OUTPUT_CARGO_TYPES,   CBM_IND_OUTPUT_CARGO_TYPES),
+	NICI(CBID_INDUSTRY_PROD_CHANGE_BUILD,    CBM_IND_PROD_CHANGE_BUILD),
 	NIC_END()
 };