1

/* $Id$ */

     2

     3

/** @file tree_cmd.cpp */

     4

     5

#include "stdafx.h"

     6

#include "openttd.h"

     7

#include "bridge_map.h"

     8

#include "clear_map.h"

     9

#include "table/strings.h"

    10

#include "table/sprites.h"

    11

#include "table/tree_land.h"

    12

#include "functions.h"

    13

#include "map.h"

    14

#include "landscape.h"

    15

#include "tile.h"

    16

#include "tree_map.h"

    17

#include "viewport.h"

    18

#include "command.h"

    19

#include "town.h"

    20

#include "sound.h"

    21

#include "variables.h"

    22

#include "genworld.h"

    24

    25

/**

    26

 * List of tree placer algorithm.

    27

 *

    28

 * This enumeration defines all possible tree placer algorithm in the game.

    29

 */

    30

enum TreePlacer {

    31

	TP_NONE,     ///< No tree placer algorithm

    32

	TP_ORIGINAL, ///< The original algorithm

    33

	TP_IMPROVED, ///< A 'improved' algorithm

    34

};

    35

    36

/**

    37

 * Get a random TreeType for the given tile based on a given seed

    38

 *

    39

 * This function returns a random TreeType which can be placed on the given tile.

    40

 * The seed for randomness must be less or equal 256, use #GB on the value of Random()

    41

 * to get such a value.

    42

 *

    43

 * @param tile The tile to get a random TreeType from

    44

 * @param seed The seed for randomness, must be less or equal 256

    45

 * @return The random tree type

    46

 */

    47

static TreeType GetRandomTreeType(TileIndex tile, uint seed)

    48

{

    49

	switch (_opt.landscape) {

    50

		case LT_TEMPERATE:

    51

			return (TreeType)(seed * TREE_COUNT_TEMPERATE / 256 + TREE_TEMPERATE);

    52

    53

		case LT_ARCTIC:

    54

			return (TreeType)(seed * TREE_COUNT_SUB_ARCTIC / 256 + TREE_SUB_ARCTIC);

    55

    56

		case LT_TROPIC:

    57

			switch (GetTropicZone(tile)) {

    58

				case TROPICZONE_INVALID: return (TreeType)(seed * TREE_COUNT_SUB_TROPICAL / 256 + TREE_SUB_TROPICAL);

    59

				case TROPICZONE_DESERT:  return (TreeType)((seed > 12) ? TREE_INVALID : TREE_CACTUS);

    60

				default:                 return (TreeType)(seed * TREE_COUNT_RAINFOREST / 256 + TREE_RAINFOREST);

    61

			}

    62

    63

		default:

    64

			return (TreeType)(seed * TREE_COUNT_TOYLAND / 256 + TREE_TOYLAND);

    65

	}

    66

}

    67

    68

/**

    69

 * Make a random tree tile of the given tile

    70

 *

    71

 * Create a new tree-tile for the given tile. The second parameter is used for

    72

 * randomness like type and number of trees.

    73

 *

    74

 * @param tile The tile to make a tree-tile from

    75

 * @param r The randomness value from a Random() value

    76

 */

    77

static void PlaceTree(TileIndex tile, uint32 r)

    78

{

    79

	TreeType tree = GetRandomTreeType(tile, GB(r, 24, 8));

    80

    81

	if (tree != TREE_INVALID) {

    82

		MakeTree(tile, tree, GB(r, 22, 2), min(GB(r, 16, 3), 6), TREE_GROUND_GRASS, 0);

    83

    84

		/* above snowline? */

    85

		if (_opt.landscape == LT_ARCTIC && GetTileZ(tile) > GetSnowLine()) {

    86

			SetTreeGroundDensity(tile, TREE_GROUND_SNOW_DESERT, 3);

    87

			SetTreeCounter(tile, (TreeGround)GB(r, 24, 3));

    88

		} else {

    89

			SetTreeGroundDensity(tile, (TreeGround)GB(r, 28, 1), 3);

    90

			SetTreeCounter(tile, (TreeGround)GB(r, 24, 4));

    91

		}

    92

	}

    93

}

    94

    95

/**

    96

 * Place some amount of trees around a given tile.

    97

 *

    98

 * This function adds some trees around a given tile. As this function use

    99

 * the Random() call it depends on the random how many trees are actually placed

   100

 * around the given tile.

   101

 *

   102

 * @param tile The center of the trees to add

   103

 */

   104

static void DoPlaceMoreTrees(TileIndex tile)

   105

{

   106

	uint i;

   107

   108

	for (i = 0; i < 1000; i++) {

   109

		uint32 r = Random();

   110

		int x = GB(r, 0, 5) - 16;

   111

		int y = GB(r, 8, 5) - 16;

   112

		uint dist = myabs(x) + myabs(y);

   113

		TileIndex cur_tile = TILE_MASK(tile + TileDiffXY(x, y));

   114

   115

		if (dist <= 13 &&

   116

				IsTileType(cur_tile, MP_CLEAR) &&

   117

				!IsBridgeAbove(cur_tile) &&

   118

				!IsClearGround(cur_tile, CLEAR_FIELDS) &&

   119

				!IsClearGround(cur_tile, CLEAR_ROCKS)) {

   120

			PlaceTree(cur_tile, r);

   121

		}

   122

	}

   123

}

   124

   125

/**

   126

 * Place more trees on the map.

   127

 *

   128

 * This function add more trees to the map.

   129

 */

   130

static void PlaceMoreTrees()

   131

{

   132

	uint i = ScaleByMapSize(GB(Random(), 0, 5) + 25);

   133

	do {

   134

		DoPlaceMoreTrees(RandomTile());

   135

	} while (--i);

   136

}

   137

   138

/**

   139

 * Place a tree at the same height as an existing tree.

   140

 *

   141

 * Add a new tree around the given tile which is at the same

   142

 * height or at some offset (2 units) of it.

   143

 *

   144

 * @param tile The base tile to add a new tree somewhere around

   145

 * @param height The height (like the one from the tile)

   146

 */

   147

void PlaceTreeAtSameHeight(TileIndex tile, uint height)

   148

{

   149

	uint i;

   150

   151

	for (i = 0; i < 1000; i++) {

   152

		uint32 r = Random();

   153

		int x = GB(r, 0, 5) - 16;

   154

		int y = GB(r, 8, 5) - 16;

   155

		TileIndex cur_tile = TILE_MASK(tile + TileDiffXY(x, y));

   156

   157

		/* Keep in range of the existing tree */

   158

		if (myabs(x) + myabs(y) > 16) continue;

   159

   160

		/* Clear tile, no farm-tiles or rocks */

   161

		if (!IsTileType(cur_tile, MP_CLEAR) ||

   162

				IsClearGround(cur_tile, CLEAR_FIELDS) ||

   163

				IsClearGround(cur_tile, CLEAR_ROCKS))

   164

			continue;

   165

   166

		/* Not too much height difference */

   167

		if (delta(GetTileZ(cur_tile), height) > 2) continue;

   168

   169

		/* Place one tree and quit */

   170

		PlaceTree(cur_tile, r);

   171

		break;

   172

	}

   173

}

   174

   175

/**

   176

 * Place some trees randomly

   177

 *

   178

 * This function just place some trees randomly on the map.

   179

 */

   180

void PlaceTreesRandomly()

   181

{

   182

	uint i, j, ht;

   183

   184

	i = ScaleByMapSize(1000);

   185

	do {

   186

		uint32 r = Random();

   187

		TileIndex tile = RandomTileSeed(r);

   188

   189

		IncreaseGeneratingWorldProgress(GWP_TREE);

   190

   191

		if (IsTileType(tile, MP_CLEAR) &&

   192

				!IsBridgeAbove(tile) &&

   193

				!IsClearGround(tile, CLEAR_FIELDS) &&

   194

				!IsClearGround(tile, CLEAR_ROCKS)) {

   195

			PlaceTree(tile, r);

   196

			if (_patches.tree_placer != TP_IMPROVED) continue;

   197

   198

			/* Place a number of trees based on the tile height.

   199

			 *  This gives a cool effect of multiple trees close together.

   200

			 *  It is almost real life ;) */

   201

			ht = GetTileZ(tile);

   202

			/* The higher we get, the more trees we plant */

   203

			j = GetTileZ(tile) / TILE_HEIGHT * 2;

   204

			while (j--) {

   205

				/* Above snowline more trees! */

   206

				if (_opt.landscape == LT_ARCTIC && ht > GetSnowLine()) {

   207

					PlaceTreeAtSameHeight(tile, ht);

   208

					PlaceTreeAtSameHeight(tile, ht);

   209

				};

   210

   211

				PlaceTreeAtSameHeight(tile, ht);

   212

			}

   213

		}

   214

	} while (--i);

   215

   216

	/* place extra trees at rainforest area */

   217

	if (_opt.landscape == LT_TROPIC) {

   218

		i = ScaleByMapSize(15000);

   219

   220

		do {

   221

			uint32 r = Random();

   222

			TileIndex tile = RandomTileSeed(r);

   223

   224

			IncreaseGeneratingWorldProgress(GWP_TREE);

   225

   226

			if (IsTileType(tile, MP_CLEAR) &&

   227

					!IsBridgeAbove(tile) &&

   228

					!IsClearGround(tile, CLEAR_FIELDS) &&

   229

					GetTropicZone(tile) == TROPICZONE_RAINFOREST) {

   230

				PlaceTree(tile, r);

   231

			}

   232

		} while (--i);

   233

	}

   234

}

   235

   236

/**

   237

 * Place new trees.

   238

 *

   239

 * This function takes care of the selected tree placer algorithm and

   240

 * place randomly the trees for a new game.

   241

 */

   242

void GenerateTrees()

   243

{

   244

	uint i, total;

   245

   246

	if (_patches.tree_placer == TP_NONE) return;

   247

   248

	if (_opt.landscape != LT_TOYLAND) PlaceMoreTrees();

   249

   250

	switch (_patches.tree_placer) {

   251

		case TP_ORIGINAL: i = _opt.landscape == LT_ARCTIC ? 15 : 6; break;

   252

		case TP_IMPROVED: i = _opt.landscape == LT_ARCTIC ?  4 : 2; break;

   253

		default: NOT_REACHED(); return;

   254

	}

   255

   256

	total = ScaleByMapSize(1000);

   257

	if (_opt.landscape == LT_TROPIC) total += ScaleByMapSize(15000);

   258

	total *= i;

   259

	SetGeneratingWorldProgress(GWP_TREE, total);

   260

   261

	for (; i != 0; i--) {

   262

		PlaceTreesRandomly();

   263

	}

   264

}

   265

   266

/** Plant a tree.

   267

 * @param tile start tile of area-drag of tree plantation

   268

 * @param flags type of operation

   269

 * @param p1 tree type, -1 means random.

   270

 * @param p2 end tile of area-drag

   271

 */

   272

CommandCost CmdPlantTree(TileIndex tile, uint32 flags, uint32 p1, uint32 p2)

   273

{

   274

	StringID msg = INVALID_STRING_ID;

   275

	CommandCost cost;

   276

	int ex;

   277

	int ey;

   278

	int sx, sy, x, y;

   279

   280

	if (p2 >= MapSize()) return CMD_ERROR;

   281

	/* Check the tree type. It can be random or some valid value within the current climate */

   282

	if (p1 != (uint)-1 && p1 - _tree_base_by_landscape[_opt.landscape] >= _tree_count_by_landscape[_opt.landscape]) return CMD_ERROR;

   283

   284

	SET_EXPENSES_TYPE(EXPENSES_OTHER);

   285

   286

	// make sure sx,sy are smaller than ex,ey

   287

	ex = TileX(tile);

   288

	ey = TileY(tile);

   289

	sx = TileX(p2);

   290

	sy = TileY(p2);

   291

	if (ex < sx) Swap(ex, sx);

   292

	if (ey < sy) Swap(ey, sy);

   293

   294

	for (x = sx; x <= ex; x++) {

   295

		for (y = sy; y <= ey; y++) {

   296

			TileIndex tile = TileXY(x, y);

   297

   298

			switch (GetTileType(tile)) {

   299

				case MP_TREES:

   300

					/* no more space for trees? */

   301

					if (_game_mode != GM_EDITOR && GetTreeCount(tile) == 3) {

   302

						msg = STR_2803_TREE_ALREADY_HERE;

   303

						continue;

   304

					}

   305

   306

					if (flags & DC_EXEC) {

   307

						AddTreeCount(tile, 1);

   308

						MarkTileDirtyByTile(tile);

   309

					}

   310

					/* 2x as expensive to add more trees to an existing tile */

   311

					cost.AddCost(_price.build_trees * 2);

   312

					break;

   313

   314

				case MP_WATER:

   315

					msg = STR_3807_CAN_T_BUILD_ON_WATER;

   316

					continue;

   317

					break;

   318

   319

				case MP_CLEAR:

   320

					if (!IsTileOwner(tile, OWNER_NONE) ||

   321

							IsBridgeAbove(tile)) {

   322

						msg = STR_2804_SITE_UNSUITABLE;

   323

						continue;

   324

					}

   325

   326

					switch (GetClearGround(tile)) {

   327

						case CLEAR_FIELDS: cost.AddCost(_price.clear_3); break;

   328

						case CLEAR_ROCKS:  cost.AddCost(_price.clear_2); break;

   329

						default: break;

   330

					}

   331

   332

					if (flags & DC_EXEC) {

   333

						TreeType treetype;

   334

						uint growth;

   335

   336

						if (_game_mode != GM_EDITOR && IsValidPlayer(_current_player)) {

   337

							Town *t = ClosestTownFromTile(tile, _patches.dist_local_authority);

   338

							if (t != NULL)

   339

								ChangeTownRating(t, RATING_TREE_UP_STEP, RATING_TREE_MAXIMUM);

   340

						}

   341

   342

						treetype = (TreeType)p1;

   343

						if (treetype == TREE_INVALID) {

   344

							treetype = GetRandomTreeType(tile, GB(Random(), 24, 8));

   345

							if (treetype == TREE_INVALID) treetype = TREE_CACTUS;

   346

						}

   347

   348

						growth = _game_mode == GM_EDITOR ? 3 : 0;

   349

						switch (GetClearGround(tile)) {

   350

							case CLEAR_ROUGH:  MakeTree(tile, treetype, 0, growth, TREE_GROUND_ROUGH, 3); break;

   351

							case CLEAR_SNOW:

   352

							case CLEAR_DESERT: MakeTree(tile, treetype, 0, growth, TREE_GROUND_SNOW_DESERT, GetClearDensity(tile)); break;

   353

							default:           MakeTree(tile, treetype, 0, growth, TREE_GROUND_GRASS, GetClearDensity(tile)); break;

   354

						}

   355

						MarkTileDirtyByTile(tile);

   356

   357

						if (_game_mode == GM_EDITOR && IS_INT_INSIDE(treetype, TREE_RAINFOREST, TREE_CACTUS))

   358

							SetTropicZone(tile, TROPICZONE_RAINFOREST);

   359

					}

   360

					cost.AddCost(_price.build_trees);

   361

					break;

   362

   363

				default:

   364

					msg = STR_2804_SITE_UNSUITABLE;

   365

					break;

   366

			}

   367

		}

   368

	}

   369

   370

	if (cost.GetCost() == 0) {

   371

		return_cmd_error(msg);

   372

	} else {

   373

		return cost;

   374

	}

   375

}

   376

   377

struct TreeListEnt {

   378

	SpriteID image;

   379

	SpriteID pal;

   380

	byte x, y;

   381

};

   382

   383

static void DrawTile_Trees(TileInfo *ti)

   384

{

   385

	const PalSpriteID *s;

   386

	const TreePos* d;

   387

	byte z;

   388

   389

	switch (GetTreeGround(ti->tile)) {

   390

		case TREE_GROUND_GRASS: DrawClearLandTile(ti, GetTreeDensity(ti->tile)); break;

   391

		case TREE_GROUND_ROUGH: DrawHillyLandTile(ti); break;

   392

		default: DrawGroundSprite(_tree_sprites_1[GetTreeDensity(ti->tile)] + _tileh_to_sprite[ti->tileh], PAL_NONE); break;

   393

	}

   394

   395

	DrawClearLandFence(ti);

   396

   397

	z = ti->z;

   398

	if (ti->tileh != SLOPE_FLAT) {

   399

		z += 4;

   400

		if (IsSteepSlope(ti->tileh)) z += 4;

   401

	}

   402

   403

	{

   404

		uint16 tmp = ti->x;

   405

		uint index;

   406

   407

		tmp = ROR(tmp, 2);

   408

		tmp -= ti->y;

   409

		tmp = ROR(tmp, 3);

   410

		tmp -= ti->x;

   411

		tmp = ROR(tmp, 1);

   412

		tmp += ti->y;

   413

   414

		d = _tree_layout_xy[GB(tmp, 4, 2)];

   415

   416

		index = GB(tmp, 6, 2) + (GetTreeType(ti->tile) << 2);

   417

   418

		/* different tree styles above one of the grounds */

   419

		if (GetTreeGround(ti->tile) == TREE_GROUND_SNOW_DESERT &&

   420

				GetTreeDensity(ti->tile) >= 2 &&

   421

				IS_INT_INSIDE(index, TREE_SUB_ARCTIC << 2, TREE_RAINFOREST << 2)) {

   422

			index += 164 - (TREE_SUB_ARCTIC << 2);

   423

		}

   424

   425

		assert(index < lengthof(_tree_layout_sprite));

   426

		s = _tree_layout_sprite[index];

   427

	}

   428

   429

	StartSpriteCombine();

   430

   432

		TreeListEnt te[4];

   433

		uint i;

   434

   435

		/* put the trees to draw in a list */

   436

		i = GetTreeCount(ti->tile) + 1;

   437

		do {

   438

			SpriteID image = s[0].sprite + (--i == 0 ? GetTreeGrowth(ti->tile) : 3);

   439

			SpriteID pal = s[0].pal;

   440

   441

			te[i].image = image;

   442

			te[i].pal   = pal;

   443

			te[i].x = d->x;

   444

			te[i].y = d->y;

   445

			s++;

   446

			d++;

   447

		} while (i);

   448

   449

		/* draw them in a sorted way */

   450

		for (;;) {

   451

			byte min = 0xFF;

   452

			TreeListEnt *tep = NULL;

   453

   454

			i = GetTreeCount(ti->tile) + 1;

   455

			do {

   456

				if (te[--i].image != 0 && te[i].x + te[i].y < min) {

   457

					min = te[i].x + te[i].y;

   458

					tep = &te[i];

   459

				}

   460

			} while (i);

   461

   462

			if (tep == NULL) break;

   463

   465

			tep->image = 0;

   466

		}

   467

	}

   468

   469

	EndSpriteCombine();

   470

}

   471

   472

   473

static uint GetSlopeZ_Trees(TileIndex tile, uint x, uint y)

   474

{

   475

	uint z;

   476

	Slope tileh = GetTileSlope(tile, &z);

   477

   478

	return z + GetPartialZ(x & 0xF, y & 0xF, tileh);

   479

}

   480

   481

static Foundation GetFoundation_Trees(TileIndex tile, Slope tileh)

   482

{

   483

	return FOUNDATION_NONE;

   484

}

   485

   486

static CommandCost ClearTile_Trees(TileIndex tile, byte flags)