rubidium@6872: /* $Id$ */
rubidium@6872: 
rubidium@6872: /** @file terraform_cmd.cpp Commands related to terraforming. */
rubidium@6872: 
rubidium@6872: #include "stdafx.h"
rubidium@6872: #include "openttd.h"
rubidium@6872: #include "strings_type.h"
rubidium@6872: #include "command_func.h"
rubidium@6872: #include "tile_map.h"
rubidium@6872: #include "tunnel_map.h"
rubidium@6872: #include "bridge_map.h"
rubidium@6872: #include "variables.h"
rubidium@6872: #include "functions.h"
rubidium@6872: #include "economy_func.h"
rubidium@6872: 
rubidium@6872: #include "table/strings.h"
rubidium@6872: 
rubidium@6872: /*
rubidium@6872:  * In one terraforming command all four corners of a initial tile can be raised/lowered (though this is not available to the player).
rubidium@6872:  * The maximal amount of height modifications is archieved when raising a complete flat land from sea level to MAX_TILE_HEIGHT or vice versa.
rubidium@6872:  * This affects all corners with a manhatten distance smaller than MAX_TILE_HEIGHT to one of the initial 4 corners.
rubidium@6872:  * Their maximal amount is computed to 4 * \sum_{i=1}^{h_max} i  =  2 * h_max * (h_max + 1).
rubidium@6872:  */
rubidium@6872: static const int TERRAFORMER_MODHEIGHT_SIZE = 2 * MAX_TILE_HEIGHT * (MAX_TILE_HEIGHT + 1);
rubidium@6872: 
rubidium@6872: /*
rubidium@6872:  * The maximal amount of affected tiles (i.e. the tiles that incident with one of the corners above, is computed similiar to
rubidium@6872:  * 1 + 4 * \sum_{i=1}^{h_max} (i+1)  =  1 + 2 * h_max + (h_max + 3).
rubidium@6872:  */
rubidium@6872: static const int TERRAFORMER_TILE_TABLE_SIZE = 1 + 2 * MAX_TILE_HEIGHT * (MAX_TILE_HEIGHT + 3);
rubidium@6872: 
rubidium@6872: struct TerraformerHeightMod {
rubidium@6872: 	TileIndex tile;   ///< Referenced tile.
rubidium@6872: 	byte height;      ///< New TileHeight (height of north corner) of the tile.
rubidium@6872: };
rubidium@6872: 
rubidium@6872: struct TerraformerState {
richk@10184: 	int modheight_count;  ///< amount of entries in "modheight".
richk@10184: 	int tile_table_count; ///< amount of entries in "tile_table".
rubidium@6872: 
rubidium@6872: 	/**
rubidium@6872: 	 * Dirty tiles, i.e.\ at least one corner changed.
rubidium@6872: 	 *
rubidium@6872: 	 * This array contains the tiles which are or will be marked as dirty.
rubidium@6872: 	 *
rubidium@6872: 	 * @ingroup dirty
rubidium@6872: 	 */
rubidium@6872: 	TileIndex tile_table[TERRAFORMER_TILE_TABLE_SIZE];
rubidium@6872: 	TerraformerHeightMod modheight[TERRAFORMER_MODHEIGHT_SIZE];  ///< Height modifications.
rubidium@6872: };
rubidium@6872: 
richk@10184: TileIndex _terraform_err_tile; ///< first tile we couldn't terraform
rubidium@6872: 
rubidium@6872: /**
rubidium@6872:  * Gets the TileHeight (height of north corner) of a tile as of current terraforming progress.
rubidium@6872:  *
rubidium@6872:  * @param ts TerraformerState.
rubidium@6872:  * @param tile Tile.
rubidium@6872:  * @return TileHeight.
rubidium@6872:  */
richk@10184: static int TerraformGetHeightOfTile(const TerraformerState *ts, TileIndex tile)
rubidium@6872: {
richk@10184: 	const TerraformerHeightMod *mod = ts->modheight;
rubidium@6872: 
richk@10184: 	for (int count = ts->modheight_count; count != 0; count--, mod++) {
rubidium@6872: 		if (mod->tile == tile) return mod->height;
rubidium@6872: 	}
rubidium@6872: 
rubidium@6872: 	/* TileHeight unchanged so far, read value from map. */
rubidium@6872: 	return TileHeight(tile);
rubidium@6872: }
rubidium@6872: 
rubidium@6872: /**
rubidium@6872:  * Stores the TileHeight (height of north corner) of a tile in a TerraformerState.
rubidium@6872:  *
rubidium@6872:  * @param ts TerraformerState.
rubidium@6872:  * @param tile Tile.
rubidium@6872:  * @param height New TileHeight.
rubidium@6872:  */
rubidium@6872: static void TerraformSetHeightOfTile(TerraformerState *ts, TileIndex tile, int height)
rubidium@6872: {
rubidium@6872: 	/* Find tile in the "modheight" table.
rubidium@6872: 	 * Note: In a normal user-terraform command the tile will not be found in the "modheight" table.
rubidium@6872: 	 *       But during house- or industry-construction multiple corners can be terraformed at once. */
rubidium@6872: 	TerraformerHeightMod *mod = ts->modheight;
rubidium@6872: 	int count = ts->modheight_count;
richk@10184: 
rubidium@6872: 	while ((count > 0) && (mod->tile != tile)) {
rubidium@6872: 		mod++;
rubidium@6872: 		count--;
rubidium@6872: 	}
rubidium@6872: 
rubidium@6872: 	/* New entry? */
rubidium@6872: 	if (count == 0) {
rubidium@6872: 		assert(ts->modheight_count < TERRAFORMER_MODHEIGHT_SIZE);
rubidium@6872: 		ts->modheight_count++;
rubidium@6872: 	}
rubidium@6872: 
rubidium@6872: 	/* Finally store the new value */
rubidium@6872: 	mod->tile = tile;
rubidium@6872: 	mod->height = (byte)height;
rubidium@6872: }
rubidium@6872: 
rubidium@6872: /**
rubidium@6872:  * Adds a tile to the "tile_table" in a TerraformerState.
rubidium@6872:  *
rubidium@6872:  * @param ts TerraformerState.
rubidium@6872:  * @param tile Tile.
rubidium@6872:  * @ingroup dirty
rubidium@6872:  */
rubidium@6872: static void TerraformAddDirtyTile(TerraformerState *ts, TileIndex tile)
rubidium@6872: {
richk@10184: 	int count = ts->tile_table_count;
rubidium@6872: 
richk@10184: 	for (TileIndex *t = ts->tile_table; count != 0; count--, t++) {
rubidium@6872: 		if (*t == tile) return;
rubidium@6872: 	}
rubidium@6872: 
rubidium@6872: 	assert(ts->tile_table_count < TERRAFORMER_TILE_TABLE_SIZE);
rubidium@6872: 
rubidium@6872: 	ts->tile_table[ts->tile_table_count++] = tile;
rubidium@6872: }
rubidium@6872: 
rubidium@6872: /**
rubidium@6872:  * Adds all tiles that incident with the north corner of a specific tile to the "tile_table" in a TerraformerState.
rubidium@6872:  *
rubidium@6872:  * @param ts TerraformerState.
rubidium@6872:  * @param tile Tile.
rubidium@6872:  * @ingroup dirty
rubidium@6872:  */
rubidium@6872: static void TerraformAddDirtyTileAround(TerraformerState *ts, TileIndex tile)
rubidium@6872: {
rubidium@6872: 	TerraformAddDirtyTile(ts, tile + TileDiffXY( 0, -1));
rubidium@6872: 	TerraformAddDirtyTile(ts, tile + TileDiffXY(-1, -1));
rubidium@6872: 	TerraformAddDirtyTile(ts, tile + TileDiffXY(-1,  0));
rubidium@6872: 	TerraformAddDirtyTile(ts, tile);
rubidium@6872: }
rubidium@6872: 
rubidium@6872: /**
rubidium@6872:  * Terraform the north corner of a tile to a specific height.
rubidium@6872:  *
rubidium@6872:  * @param ts TerraformerState.
rubidium@6872:  * @param tile Tile.
rubidium@6872:  * @param height Aimed height.
rubidium@6872:  * @param return Error code or cost.
rubidium@6872:  */
rubidium@6872: static CommandCost TerraformTileHeight(TerraformerState *ts, TileIndex tile, int height)
rubidium@6872: {
rubidium@6872: 	assert(tile < MapSize());
rubidium@6872: 
rubidium@6872: 	/* Check range of destination height */
rubidium@6872: 	if (height < 0) return_cmd_error(STR_1003_ALREADY_AT_SEA_LEVEL);
rubidium@6872: 	if (height > MAX_TILE_HEIGHT) return_cmd_error(STR_1004_TOO_HIGH);
rubidium@6872: 
rubidium@6872: 	/*
rubidium@6872: 	 * Check if the terraforming has any effect.
rubidium@6872: 	 * This can only be true, if multiple corners of the start-tile are terraformed (i.e. the terraforming is done by towns/industries etc.).
rubidium@6872: 	 * In this case the terraforming should fail. (Don't know why.)
rubidium@6872: 	 */
rubidium@6872: 	if (height == TerraformGetHeightOfTile(ts, tile)) return CMD_ERROR;
rubidium@6872: 
rubidium@6872: 	/* Check "too close to edge of map" */
rubidium@6872: 	uint x = TileX(tile);
rubidium@6872: 	uint y = TileY(tile);
rubidium@6872: 	if ((x <= 1) || (y <= 1) || (x >= MapMaxX() - 1) || (y >= MapMaxY() - 1)) {
rubidium@6872: 		/*
rubidium@6872: 		 * Determine a sensible error tile
rubidium@6872: 		 * Note: If x and y are both zero this will disable the error tile. (Tile 0 cannot be highlighted :( )
rubidium@6872: 		 */
rubidium@6872: 		if ((x == 1) && (y != 0)) x = 0;
rubidium@6872: 		if ((y == 1) && (x != 0)) y = 0;
rubidium@6872: 		_terraform_err_tile = TileXY(x, y);
rubidium@6872: 		return_cmd_error(STR_0002_TOO_CLOSE_TO_EDGE_OF_MAP);
rubidium@6872: 	}
rubidium@6872: 
rubidium@6872: 	/* Mark incident tiles, that are involved in the terraforming */
rubidium@6872: 	TerraformAddDirtyTileAround(ts, tile);
rubidium@6872: 
rubidium@6872: 	/* Store the height modification */
rubidium@6872: 	TerraformSetHeightOfTile(ts, tile, height);
rubidium@6872: 
richk@10184: 	CommandCost total_cost(EXPENSES_CONSTRUCTION);
richk@10184: 
rubidium@6872: 	/* Increment cost */
rubidium@6872: 	total_cost.AddCost(_price.terraform);
rubidium@6872: 
rubidium@6872: 	/* Recurse to neighboured corners if height difference is larger than 1 */
rubidium@6872: 	{
rubidium@6872: 		const TileIndexDiffC *ttm;
rubidium@6872: 
rubidium@6872: 		static const TileIndexDiffC _terraform_tilepos[] = {
rubidium@6872: 			{ 1,  0}, // move to tile in SE
rubidium@6872: 			{-2,  0}, // undo last move, and move to tile in NW
rubidium@6872: 			{ 1,  1}, // undo last move, and move to tile in SW
rubidium@6872: 			{ 0, -2}  // undo last move, and move to tile in NE
rubidium@6872: 		};
rubidium@6872: 
rubidium@6872: 		for (ttm = _terraform_tilepos; ttm != endof(_terraform_tilepos); ttm++) {
rubidium@6872: 			tile += ToTileIndexDiff(*ttm);
rubidium@6872: 
rubidium@6872: 			/* Get TileHeight of neighboured tile as of current terraform progress */
rubidium@6872: 			int r = TerraformGetHeightOfTile(ts, tile);
rubidium@6872: 			int height_diff = height - r;
rubidium@6872: 
rubidium@6872: 			/* Is the height difference to the neighboured corner greater than 1? */
rubidium@6872: 			if (abs(height_diff) > 1) {
rubidium@6872: 				/* Terraform the neighboured corner. The resulting height difference should be 1. */
rubidium@6872: 				height_diff += (height_diff < 0 ? 1 : -1);
rubidium@6872: 				CommandCost cost = TerraformTileHeight(ts, tile, r + height_diff);
rubidium@6872: 				if (CmdFailed(cost)) return cost;
rubidium@6872: 				total_cost.AddCost(cost);
rubidium@6872: 			}
rubidium@6872: 		}
rubidium@6872: 	}
rubidium@6872: 
rubidium@6872: 	return total_cost;
rubidium@6872: }
rubidium@6872: 
rubidium@6872: /** Terraform land
rubidium@6872:  * @param tile tile to terraform
rubidium@6872:  * @param flags for this command type
rubidium@6872:  * @param p1 corners to terraform (SLOPE_xxx)
rubidium@6872:  * @param p2 direction; eg up (non-zero) or down (zero)
rubidium@6872:  * @return error or cost of terraforming
rubidium@6872:  */
rubidium@6872: CommandCost CmdTerraformLand(TileIndex tile, uint32 flags, uint32 p1, uint32 p2)
rubidium@6872: {
richk@10184: 	/* Make an extra check for map-bounds cause we add tiles to the originating tile */
richk@10184: 	if (tile + TileDiffXY(1, 1) >= MapSize()) return CMD_ERROR;
richk@10184: 
richk@10184: 	_terraform_err_tile = INVALID_TILE;
richk@10184: 
rubidium@6872: 	CommandCost total_cost(EXPENSES_CONSTRUCTION);
rubidium@6872: 	int direction = (p2 != 0 ? 1 : -1);
richk@10184: 	TerraformerState ts;
rubidium@6872: 
rubidium@6872: 	ts.modheight_count = ts.tile_table_count = 0;
rubidium@6872: 
rubidium@6872: 	/* Compute the costs and the terraforming result in a model of the landscape */
rubidium@6872: 	if ((p1 & SLOPE_W) != 0) {
rubidium@6872: 		TileIndex t = tile + TileDiffXY(1, 0);
rubidium@6872: 		CommandCost cost = TerraformTileHeight(&ts, t, TileHeight(t) + direction);
rubidium@6872: 		if (CmdFailed(cost)) return cost;
rubidium@6872: 		total_cost.AddCost(cost);
rubidium@6872: 	}
rubidium@6872: 
rubidium@6872: 	if ((p1 & SLOPE_S) != 0) {
rubidium@6872: 		TileIndex t = tile + TileDiffXY(1, 1);
rubidium@6872: 		CommandCost cost = TerraformTileHeight(&ts, t, TileHeight(t) + direction);
rubidium@6872: 		if (CmdFailed(cost)) return cost;
rubidium@6872: 		total_cost.AddCost(cost);
rubidium@6872: 	}
rubidium@6872: 
rubidium@6872: 	if ((p1 & SLOPE_E) != 0) {
rubidium@6872: 		TileIndex t = tile + TileDiffXY(0, 1);
rubidium@6872: 		CommandCost cost = TerraformTileHeight(&ts, t, TileHeight(t) + direction);
rubidium@6872: 		if (CmdFailed(cost)) return cost;
rubidium@6872: 		total_cost.AddCost(cost);
rubidium@6872: 	}
rubidium@6872: 
rubidium@6872: 	if ((p1 & SLOPE_N) != 0) {
rubidium@6872: 		TileIndex t = tile + TileDiffXY(0, 0);
rubidium@6872: 		CommandCost cost = TerraformTileHeight(&ts, t, TileHeight(t) + direction);
rubidium@6872: 		if (CmdFailed(cost)) return cost;
rubidium@6872: 		total_cost.AddCost(cost);
rubidium@6872: 	}
rubidium@6872: 
rubidium@6872: 	/* Check if the terraforming is valid wrt. tunnels, bridges and objects on the surface */
rubidium@6872: 	{
rubidium@6872: 		TileIndex *ti = ts.tile_table;
rubidium@6872: 
richk@10184: 		for (int count = ts.tile_table_count; count != 0; count--, ti++) {
rubidium@6872: 			TileIndex tile = *ti;
rubidium@6872: 
rubidium@6872: 			/* Find new heights of tile corners */
rubidium@6872: 			uint z_N = TerraformGetHeightOfTile(&ts, tile + TileDiffXY(0, 0));
rubidium@6872: 			uint z_W = TerraformGetHeightOfTile(&ts, tile + TileDiffXY(1, 0));
rubidium@6872: 			uint z_S = TerraformGetHeightOfTile(&ts, tile + TileDiffXY(1, 1));
rubidium@6872: 			uint z_E = TerraformGetHeightOfTile(&ts, tile + TileDiffXY(0, 1));
rubidium@6872: 
rubidium@6872: 			/* Find min and max height of tile */
rubidium@6872: 			uint z_min = min(min(z_N, z_W), min(z_S, z_E));
rubidium@6872: 			uint z_max = max(max(z_N, z_W), max(z_S, z_E));
rubidium@6872: 
rubidium@6872: 			/* Compute tile slope */
richk@10184: 			Slope tileh = (z_max > z_min + 1 ? SLOPE_STEEP : SLOPE_FLAT);
richk@10184: 			if (z_W > z_min) tileh |= SLOPE_W;
richk@10184: 			if (z_S > z_min) tileh |= SLOPE_S;
richk@10184: 			if (z_E > z_min) tileh |= SLOPE_E;
richk@10184: 			if (z_N > z_min) tileh |= SLOPE_N;
rubidium@6872: 
rubidium@6872: 			/* Check if bridge would take damage */
rubidium@6872: 			if (direction == 1 && MayHaveBridgeAbove(tile) && IsBridgeAbove(tile) &&
rubidium@6872: 					GetBridgeHeight(GetSouthernBridgeEnd(tile)) <= z_max * TILE_HEIGHT) {
rubidium@6872: 				_terraform_err_tile = tile; // highlight the tile under the bridge
rubidium@6872: 				return_cmd_error(STR_5007_MUST_DEMOLISH_BRIDGE_FIRST);
rubidium@6872: 			}
rubidium@6872: 			/* Check if tunnel would take damage */
rubidium@6872: 			if (direction == -1 && IsTunnelInWay(tile, z_min * TILE_HEIGHT)) {
rubidium@6872: 				_terraform_err_tile = tile; // highlight the tile above the tunnel
rubidium@6872: 				return_cmd_error(STR_1002_EXCAVATION_WOULD_DAMAGE);
rubidium@6872: 			}
rubidium@6872: 			/* Check tiletype-specific things, and add extra-cost */
richk@10731: 			const bool curr_gen = _generating_world;
richk@10731: 			if (_game_mode == GM_EDITOR) _generating_world = true; // used to create green terraformed land
richk@10184: 			CommandCost cost = _tile_type_procs[GetTileType(tile)]->terraform_tile_proc(tile, flags | DC_AUTO, z_min * TILE_HEIGHT, tileh);
richk@10731: 			_generating_world = curr_gen;
rubidium@6872: 			if (CmdFailed(cost)) {
rubidium@6872: 				_terraform_err_tile = tile;
rubidium@6872: 				return cost;
rubidium@6872: 			}
rubidium@6872: 			total_cost.AddCost(cost);
rubidium@6872: 		}
rubidium@6872: 	}
rubidium@6872: 
rubidium@6872: 	if (flags & DC_EXEC) {
rubidium@6872: 		/* change the height */
rubidium@6872: 		{
rubidium@6872: 			int count;
rubidium@6872: 			TerraformerHeightMod *mod;
rubidium@6872: 
rubidium@6872: 			mod = ts.modheight;
rubidium@6872: 			for (count = ts.modheight_count; count != 0; count--, mod++) {
rubidium@6872: 				TileIndex til = mod->tile;
rubidium@6872: 
rubidium@6872: 				SetTileHeight(til, mod->height);
rubidium@6872: 			}
rubidium@6872: 		}
rubidium@6872: 
rubidium@6872: 		/* finally mark the dirty tiles dirty */
rubidium@6872: 		{
rubidium@6872: 			int count;
rubidium@6872: 			TileIndex *ti = ts.tile_table;
rubidium@6872: 			for (count = ts.tile_table_count; count != 0; count--, ti++) {
rubidium@6872: 				MarkTileDirtyByTile(*ti);
rubidium@6872: 			}
rubidium@6872: 		}
rubidium@6872: 	}
rubidium@6872: 	return total_cost;
rubidium@6872: }
rubidium@6872: 
rubidium@6872: 
rubidium@6872: /** Levels a selected (rectangle) area of land
rubidium@6872:  * @param tile end tile of area-drag
rubidium@6872:  * @param flags for this command type
rubidium@6872:  * @param p1 start tile of area drag
rubidium@6872:  * @param p2 height difference; eg raise (+1), lower (-1) or level (0)
rubidium@6872:  * @return  error or cost of terraforming
rubidium@6872:  */
rubidium@6872: CommandCost CmdLevelLand(TileIndex tile, uint32 flags, uint32 p1, uint32 p2)
rubidium@6872: {
rubidium@6872: 	if (p1 >= MapSize()) return CMD_ERROR;
rubidium@6872: 
rubidium@6872: 	/* remember level height */
richk@10184: 	uint oldh = TileHeight(p1);
rubidium@6872: 
rubidium@6872: 	/* compute new height */
richk@10184: 	uint h = oldh + p2;
rubidium@6872: 
rubidium@6872: 	/* Check range of destination height */
rubidium@6872: 	if (h > MAX_TILE_HEIGHT) return_cmd_error((oldh == 0) ? STR_1003_ALREADY_AT_SEA_LEVEL : STR_1004_TOO_HIGH);
rubidium@6872: 
rubidium@6872: 	/* make sure sx,sy are smaller than ex,ey */
richk@10184: 	int ex = TileX(tile);
richk@10184: 	int ey = TileY(tile);
richk@10184: 	int sx = TileX(p1);
richk@10184: 	int sy = TileY(p1);
rubidium@6872: 	if (ex < sx) Swap(ex, sx);
rubidium@6872: 	if (ey < sy) Swap(ey, sy);
rubidium@6872: 	tile = TileXY(sx, sy);
rubidium@6872: 
richk@10184: 	int size_x = ex - sx + 1;
richk@10184: 	int size_y = ey - sy + 1;
rubidium@6872: 
richk@10184: 	Money money = GetAvailableMoneyForCommand();
richk@10184: 	CommandCost cost(EXPENSES_CONSTRUCTION);
rubidium@6872: 
rubidium@6872: 	BEGIN_TILE_LOOP(tile2, size_x, size_y, tile) {
richk@10184: 		uint curh = TileHeight(tile2);
rubidium@6872: 		while (curh != h) {
richk@10184: 			CommandCost ret = DoCommand(tile2, SLOPE_N, (curh > h) ? 0 : 1, flags & ~DC_EXEC, CMD_TERRAFORM_LAND);
rubidium@6872: 			if (CmdFailed(ret)) break;
rubidium@6872: 
rubidium@6872: 			if (flags & DC_EXEC) {
richk@10184: 				money -= ret.GetCost();
richk@10184: 				if (money < 0) {
rubidium@6872: 					_additional_cash_required = ret.GetCost();
rubidium@6872: 					return cost;
rubidium@6872: 				}
rubidium@6872: 				DoCommand(tile2, SLOPE_N, (curh > h) ? 0 : 1, flags, CMD_TERRAFORM_LAND);
rubidium@6872: 			}
rubidium@6872: 
rubidium@6872: 			cost.AddCost(ret);
rubidium@6872: 			curh += (curh > h) ? -1 : 1;
rubidium@6872: 		}
rubidium@6872: 	} END_TILE_LOOP(tile2, size_x, size_y, tile)
rubidium@6872: 
rubidium@6872: 	return (cost.GetCost() == 0) ? CMD_ERROR : cost;
rubidium@6872: }