1

/* $Id$ */

     2

     3

#include "../../stdafx.h"

     4

#include "../../openttd.h"

     5

#include "../../aircraft.h"

     6

#include "../../bridge_map.h"

     7

#include "../../tile_cmd.h"

     8

#include "../../landscape.h"

     9

#include "../../rail_map.h"

    10

#include "../../road_map.h"

    11

#include "../../roadveh.h"

    12

#include "../../station_map.h"

    13

#include "../../tunnel_map.h"

    14

#include "../../command_func.h"

    15

#include "../../town.h"

    16

#include "../../industry.h"

    17

#include "../../pathfind.h"

    18

#include "../../airport.h"

    19

#include "../../variables.h"

    20

#include "../../bridge.h"

    21

#include "../../date_func.h"

    22

#include "../../tunnelbridge_map.h"

    23

#include "../../window_func.h"

    24

#include "../../vehicle_func.h"

    25

#include "../../functions.h"

    26

#include "../../saveload.h"

    27

#include "../../player_func.h"

    28

#include "../../player_base.h"

    29

#include "../../settings_type.h"

    30

#include "default.h"

    31

#include "../../tunnelbridge.h"

    32

#include "../../order_func.h"

    33

    34

#include "../../table/ai_rail.h"

    35

    36

// remove some day perhaps?

    37

static uint _ai_service_interval;

    38

PlayerAI _players_ai[MAX_PLAYERS];

    39

    40

typedef void AiStateAction(Player *p);

    41

    42

enum {

    43

	AIS_0                            =  0,

    44

	AIS_1                            =  1,

    45

	AIS_VEH_LOOP                     =  2,

    46

	AIS_VEH_CHECK_REPLACE_VEHICLE    =  3,

    47

	AIS_VEH_DO_REPLACE_VEHICLE       =  4,

    48

	AIS_WANT_NEW_ROUTE               =  5,

    49

	AIS_BUILD_DEFAULT_RAIL_BLOCKS    =  6,

    50

	AIS_BUILD_RAIL                   =  7,

    51

	AIS_BUILD_RAIL_VEH               =  8,

    52

	AIS_DELETE_RAIL_BLOCKS           =  9,

    53

	AIS_BUILD_DEFAULT_ROAD_BLOCKS    = 10,

    54

	AIS_BUILD_ROAD                   = 11,

    55

	AIS_BUILD_ROAD_VEHICLES          = 12,

    56

	AIS_DELETE_ROAD_BLOCKS           = 13,

    57

	AIS_AIRPORT_STUFF                = 14,

    58

	AIS_BUILD_DEFAULT_AIRPORT_BLOCKS = 15,

    59

	AIS_BUILD_AIRCRAFT_VEHICLES      = 16,

    60

	AIS_CHECK_SHIP_STUFF             = 17,

    61

	AIS_BUILD_DEFAULT_SHIP_BLOCKS    = 18,

    62

	AIS_DO_SHIP_STUFF                = 19,

    63

	AIS_SELL_VEHICLE                 = 20,

    64

	AIS_REMOVE_STATION               = 21,

    65

	AIS_REMOVE_TRACK                 = 22,

    66

	AIS_REMOVE_SINGLE_RAIL_TILE      = 23

    67

};

    68

    69

    70

static inline TrackBits GetRailTrackStatus(TileIndex tile)

    71

{

    72

	return TrackStatusToTrackBits(GetTileTrackStatus(tile, TRANSPORT_RAIL, 0));

    73

}

    74

    75

    76

static void AiCase0(Player *p)

    77

{

    78

	_players_ai[p->index].state = AIS_REMOVE_TRACK;

    79

	_players_ai[p->index].state_counter = 0;

    80

}

    81

    82

static void AiCase1(Player *p)

    83

{

    84

	_players_ai[p->index].cur_veh = NULL;

    85

	_players_ai[p->index].state = AIS_VEH_LOOP;

    86

}

    87

    88

static void AiStateVehLoop(Player *p)

    89

{

    90

	Vehicle *v;

    91

	uint index;

    92

    93

	index = (_players_ai[p->index].cur_veh == NULL) ? 0 : _players_ai[p->index].cur_veh->index + 1;

    94

    95

	FOR_ALL_VEHICLES_FROM(v, index) {

    96

		if (v->owner != _current_player) continue;

    97

    98

		if ((v->type == VEH_TRAIN && v->subtype == 0) ||

    99

				v->type == VEH_ROAD ||

   100

				(v->type == VEH_AIRCRAFT && IsNormalAircraft(v)) ||

   101

				v->type == VEH_SHIP) {

   102

			/* replace engine? */

   103

			if (v->type == VEH_TRAIN && v->engine_type < 3 &&

   104

					(_price.build_railvehicle >> 3) < p->player_money) {

   105

				_players_ai[p->index].state = AIS_VEH_CHECK_REPLACE_VEHICLE;

   106

				_players_ai[p->index].cur_veh = v;

   107

				return;

   108

			}

   109

   110

			/* not profitable? */

   111

			if (v->age >= 730 &&

   112

					v->profit_last_year < _price.station_value * 5 * 256 &&

   113

					v->profit_this_year < _price.station_value * 5 * 256) {

   114

				_players_ai[p->index].state_counter = 0;

   115

				_players_ai[p->index].state = AIS_SELL_VEHICLE;

   116

				_players_ai[p->index].cur_veh = v;

   117

				return;

   118

			}

   119

   120

			/* not reliable? */

   121

			if (v->age >= v->max_age || (

   122

						v->age != 0 &&

   123

						GetEngine(v->engine_type)->reliability < 35389

   124

					)) {

   125

				_players_ai[p->index].state = AIS_VEH_CHECK_REPLACE_VEHICLE;

   126

				_players_ai[p->index].cur_veh = v;

   127

				return;

   128

			}

   129

		}

   130

	}

   131

   132

	_players_ai[p->index].state = AIS_WANT_NEW_ROUTE;

   133

	_players_ai[p->index].state_counter = 0;

   134

}

   135

   136

static EngineID AiChooseTrainToBuild(RailType railtype, Money money, byte flag, TileIndex tile)

   137

{

   138

	EngineID best_veh_index = INVALID_ENGINE;

   139

	byte best_veh_score = 0;

   140

	EngineID i;

   141

   142

	FOR_ALL_ENGINEIDS_OF_TYPE(i, VEH_TRAIN) {

   143

		const RailVehicleInfo *rvi = RailVehInfo(i);

   144

		const Engine* e = GetEngine(i);

   145

   146

		if (!IsCompatibleRail(rvi->railtype, railtype) ||

   147

				rvi->railveh_type == RAILVEH_WAGON ||

   148

				(rvi->railveh_type == RAILVEH_MULTIHEAD && flag & 1) ||

   149

				!HasBit(e->player_avail, _current_player) ||

   150

				e->reliability < 0x8A3D) {

   151

			continue;

   152

		}

   153

   154

		/* Don't choose an engine designated for passenger use for freight. */

   155

		if (rvi->ai_passenger_only != 0 && flag == 1) continue;

   156

   157

		CommandCost ret = DoCommand(tile, i, 0, 0, CMD_BUILD_RAIL_VEHICLE);

   158

		if (CmdSucceeded(ret) && ret.GetCost() <= money && rvi->ai_rank >= best_veh_score) {

   159

			best_veh_score = rvi->ai_rank;

   160

			best_veh_index = i;

   161

		}

   162

	}

   163

   164

	return best_veh_index;

   165

}

   166

   167

static EngineID AiChooseRoadVehToBuild(CargoID cargo, Money money, TileIndex tile)

   168

{

   169

	EngineID best_veh_index = INVALID_ENGINE;

   170

	int32 best_veh_rating = 0;

   171

	EngineID i;

   172

   173

	FOR_ALL_ENGINEIDS_OF_TYPE(i, VEH_ROAD) {

   174

		const RoadVehicleInfo *rvi = RoadVehInfo(i);

   175

		const Engine* e = GetEngine(i);

   176

   177

		if (!HasBit(e->player_avail, _current_player) || e->reliability < 0x8A3D) {

   178

			continue;

   179

		}

   180

   181

		/* Skip vehicles which can't take our cargo type */

   182

		if (rvi->cargo_type != cargo && !CanRefitTo(i, cargo)) continue;

   183

   184

		/* Rate and compare the engine by speed & capacity */

   185

		int rating = rvi->max_speed * rvi->capacity;

   186

		if (rating <= best_veh_rating) continue;

   187

   188

		CommandCost ret = DoCommand(tile, i, 0, 0, CMD_BUILD_ROAD_VEH);

   189

		if (CmdFailed(ret)) continue;

   190

   191

		/* Add the cost of refitting */

   192

		if (rvi->cargo_type != cargo) ret.AddCost(GetRefitCost(i));

   193

		if (ret.GetCost() > money) continue;

   194

   195

		best_veh_rating = rating;

   196

		best_veh_index = i;

   197

	}

   198

   199

	return best_veh_index;

   200

}

   201

   202

/**

   203

 * Choose aircraft to build.

   204

 * @param money current AI money

   205

 * @param forbidden forbidden flags - AIR_HELI = 0 (always allowed), AIR_CTOL = 1 (bit 0), AIR_FAST = 2 (bit 1)

   206

 * @return EngineID of aircraft to build

   207

 */

   208

static EngineID AiChooseAircraftToBuild(Money money, byte forbidden)

   209

{

   210

	EngineID best_veh_index = INVALID_ENGINE;

   211

	Money best_veh_cost = 0;

   212

	EngineID i;

   213

   214

	FOR_ALL_ENGINEIDS_OF_TYPE(i, VEH_AIRCRAFT) {

   215

		const Engine* e = GetEngine(i);

   216

   217

		if (!HasBit(e->player_avail, _current_player) || e->reliability < 0x8A3D) {

   218

			continue;

   219

		}

   220

   221

		if ((AircraftVehInfo(i)->subtype & forbidden) != 0) continue;

   222

   223

		CommandCost ret = DoCommand(0, i, 0, DC_QUERY_COST, CMD_BUILD_AIRCRAFT);

   224

		if (CmdSucceeded(ret) && ret.GetCost() <= money && ret.GetCost() >= best_veh_cost) {

   225

			best_veh_cost = ret.GetCost();

   226

			best_veh_index = i;

   227

		}

   228

	}

   229

   230

	return best_veh_index;

   231

}

   232

   233

static Money AiGetBasePrice(const Player* p)

   234

{

   235

	Money base = _price.station_value;

   236

   237

	// adjust base price when more expensive vehicles are available

   238

	switch (_players_ai[p->index].railtype_to_use) {

   239

		default: NOT_REACHED();

   240

		case RAILTYPE_RAIL:     break;

   241

		case RAILTYPE_ELECTRIC: break;

   242

		case RAILTYPE_MONO:     base = (base * 3) >> 1; break;

   243

		case RAILTYPE_MAGLEV:   base *= 2; break;

   244

	}

   245

   246

	return base;

   247

}

   248

   249

static EngineID AiChooseRoadVehToReplaceWith(const Player* p, const Vehicle* v)

   250

{

   251

	Money avail_money = p->player_money + v->value;

   252

	return AiChooseRoadVehToBuild(v->cargo_type, avail_money, v->tile);

   253

}

   254

   255

static EngineID AiChooseAircraftToReplaceWith(const Player* p, const Vehicle* v)

   256

{

   257

	Money avail_money = p->player_money + v->value;

   258

   259

	/* determine forbidden aircraft bits */

   260

	byte forbidden = 0;

   261

	const Order *o;

   262

   263

	FOR_VEHICLE_ORDERS(v, o) {

   264

		if (!o->IsValid()) continue;

   265

		if (!IsValidStationID(o->GetDestination())) continue;

   266

		const Station *st = GetStation(o->GetDestination());

   267

		if (!(st->facilities & FACIL_AIRPORT)) continue;

   268

   269

		AirportFTAClass::Flags flags = st->Airport()->flags;

   270

		if (!(flags & AirportFTAClass::AIRPLANES)) forbidden |= AIR_CTOL | AIR_FAST; // no planes for heliports / oil rigs

   271

		if (flags & AirportFTAClass::SHORT_STRIP) forbidden |= AIR_FAST; // no fast planes for small airports

   272

	}

   273

   274

	return AiChooseAircraftToBuild(

   275

		avail_money, forbidden

   276

	);

   277

}

   278

   279

static EngineID AiChooseTrainToReplaceWith(const Player* p, const Vehicle* v)

   280

{

   281

	Money avail_money = p->player_money + v->value;

   282

	const Vehicle* u = v;

   283

	int num = 0;

   284

   285

	while (++num, u->Next() != NULL) {

   286

		u = u->Next();

   287

	}

   288

   289

	// XXX: check if a wagon

   290

	return AiChooseTrainToBuild(v->u.rail.railtype, avail_money, 0, v->tile);

   291

}

   292

   293

static EngineID AiChooseShipToReplaceWith(const Player* p, const Vehicle* v)

   294

{

   295

	/* Ships are not implemented in this (broken) AI */

   296

	return INVALID_ENGINE;

   297

}

   298

   299

static void AiHandleGotoDepot(Player *p, int cmd)

   300

{

   301

	if (!_players_ai[p->index].cur_veh->current_order.IsType(OT_GOTO_DEPOT))

   302

		DoCommand(0, _players_ai[p->index].cur_veh->index, 0, DC_EXEC, cmd);

   303

   304

	if (++_players_ai[p->index].state_counter <= 1387) {

   305

		_players_ai[p->index].state = AIS_VEH_DO_REPLACE_VEHICLE;

   306

		return;

   307

	}

   308

   309

	if (_players_ai[p->index].cur_veh->current_order.IsType(OT_GOTO_DEPOT)) {

   310

		_players_ai[p->index].cur_veh->current_order.MakeDummy();

   311

		InvalidateWindow(WC_VEHICLE_VIEW, _players_ai[p->index].cur_veh->index);

   312

	}

   313

}

   314

   315

static void AiRestoreVehicleOrders(Vehicle *v, BackuppedOrders *bak)

   316

{

   317

	if (bak->order == NULL) return;

   318

   319

	for (uint i = 0; !bak->order[i].IsType(OT_NOTHING); i++) {

   320

		if (!DoCommandP(0, v->index + (i << 16), bak->order[i].Pack(), NULL, CMD_INSERT_ORDER | CMD_NO_TEST_IF_IN_NETWORK))

   321

			break;

   322

	}

   323

}

   324

   325

static void AiHandleReplaceTrain(Player *p)

   326

{

   327

	const Vehicle* v = _players_ai[p->index].cur_veh;

   328

	BackuppedOrders orderbak;

   329

	EngineID veh;

   330

   331

	// wait until the vehicle reaches the depot.

   333

		AiHandleGotoDepot(p, CMD_SEND_TRAIN_TO_DEPOT);

   334

		return;

   335

	}

   336

   337

	veh = AiChooseTrainToReplaceWith(p, v);

   338

	if (veh != INVALID_ENGINE) {

   339

		TileIndex tile;

   340

   341

		BackupVehicleOrders(v, &orderbak);

   342

		tile = v->tile;

   343

   344

		if (CmdSucceeded(DoCommand(0, v->index, 2, DC_EXEC, CMD_SELL_RAIL_WAGON)) &&

   345

				CmdSucceeded(DoCommand(tile, veh, 0, DC_EXEC, CMD_BUILD_RAIL_VEHICLE))) {

   346

			VehicleID veh = _new_vehicle_id;

   347

			AiRestoreVehicleOrders(GetVehicle(veh), &orderbak);

   348

			DoCommand(0, veh, 0, DC_EXEC, CMD_START_STOP_TRAIN);

   349

   350

			DoCommand(0, veh, _ai_service_interval, DC_EXEC, CMD_CHANGE_SERVICE_INT);

   351

		}

   352

	}

   353

}

   354

   355

static void AiHandleReplaceRoadVeh(Player *p)

   356

{

   357

	const Vehicle* v = _players_ai[p->index].cur_veh;

   358

	BackuppedOrders orderbak;

   359

	EngineID veh;

   360

   361

	if (!v->IsStoppedInDepot()) {

   362

		AiHandleGotoDepot(p, CMD_SEND_ROADVEH_TO_DEPOT);

   363

		return;

   364

	}

   365

   366

	veh = AiChooseRoadVehToReplaceWith(p, v);

   367

	if (veh != INVALID_ENGINE) {

   368

		TileIndex tile;

   369

   370

		BackupVehicleOrders(v, &orderbak);

   371

		tile = v->tile;

   372

   373

		if (CmdSucceeded(DoCommand(0, v->index, 0, DC_EXEC, CMD_SELL_ROAD_VEH)) &&

   374

				CmdSucceeded(DoCommand(tile, veh, 0, DC_EXEC, CMD_BUILD_ROAD_VEH))) {

   375

			VehicleID veh = _new_vehicle_id;

   376

   377

			AiRestoreVehicleOrders(GetVehicle(veh), &orderbak);

   378

			DoCommand(0, veh, 0, DC_EXEC, CMD_START_STOP_ROADVEH);

   379

			DoCommand(0, veh, _ai_service_interval, DC_EXEC, CMD_CHANGE_SERVICE_INT);

   380

		}

   381

	}

   382

}

   383

   384

static void AiHandleReplaceAircraft(Player *p)

   385

{

   386

	const Vehicle* v = _players_ai[p->index].cur_veh;

   387

	BackuppedOrders orderbak;

   388

	EngineID veh;

   389

   390

	if (!v->IsStoppedInDepot()) {

   391

		AiHandleGotoDepot(p, CMD_SEND_AIRCRAFT_TO_HANGAR);

   392

		return;

   393

	}

   394

   395

	veh = AiChooseAircraftToReplaceWith(p, v);

   396

	if (veh != INVALID_ENGINE) {

   397

		TileIndex tile;

   398

   399

		BackupVehicleOrders(v, &orderbak);

   400

		tile = v->tile;

   401

   402

		if (CmdSucceeded(DoCommand(0, v->index, 0, DC_EXEC, CMD_SELL_AIRCRAFT)) &&

   403

				CmdSucceeded(DoCommand(tile, veh, 0, DC_EXEC, CMD_BUILD_AIRCRAFT))) {

   404

			VehicleID veh = _new_vehicle_id;

   405

			AiRestoreVehicleOrders(GetVehicle(veh), &orderbak);

   406

			DoCommand(0, veh, 0, DC_EXEC, CMD_START_STOP_AIRCRAFT);

   407

   408

			DoCommand(0, veh, _ai_service_interval, DC_EXEC, CMD_CHANGE_SERVICE_INT);

   409

		}

   410

	}

   411

}

   412

   413

static void AiHandleReplaceShip(Player *p)

   414

{

   415

	/* Ships are not implemented in this (broken) AI */

   416

}

   417

   418

typedef EngineID CheckReplaceProc(const Player* p, const Vehicle* v);

   419

   420

static CheckReplaceProc* const _veh_check_replace_proc[] = {

   421

	AiChooseTrainToReplaceWith,

   422

	AiChooseRoadVehToReplaceWith,

   423

	AiChooseShipToReplaceWith,

   424

	AiChooseAircraftToReplaceWith,

   425

};

   426

   427

typedef void DoReplaceProc(Player *p);

   428

static DoReplaceProc* const _veh_do_replace_proc[] = {

   429

	AiHandleReplaceTrain,

   430

	AiHandleReplaceRoadVeh,

   431

	AiHandleReplaceShip,

   432

	AiHandleReplaceAircraft

   433

};

   434

   435

static void AiStateCheckReplaceVehicle(Player *p)

   436

{

   437

	const Vehicle* v = _players_ai[p->index].cur_veh;

   438

   439

	if (!v->IsValid() ||

   440

			v->owner != _current_player ||

   441

			v->type > VEH_SHIP ||

   442

			_veh_check_replace_proc[v->type - VEH_TRAIN](p, v) == INVALID_ENGINE) {

   443

		_players_ai[p->index].state = AIS_VEH_LOOP;

   444

	} else {

   445

		_players_ai[p->index].state_counter = 0;

   446

		_players_ai[p->index].state = AIS_VEH_DO_REPLACE_VEHICLE;

   447

	}

   448

}

   449

   450

static void AiStateDoReplaceVehicle(Player *p)

   451

{

   452

	const Vehicle* v = _players_ai[p->index].cur_veh;

   453

   454

	_players_ai[p->index].state = AIS_VEH_LOOP;

   455

	// vehicle is not owned by the player anymore, something went very wrong.

   456

	if (!v->IsValid() || v->owner != _current_player) return;

   457

	_veh_do_replace_proc[v->type - VEH_TRAIN](p);

   458

}

   459

   460

struct FoundRoute {

   461

	int distance;

   462

	CargoID cargo;

   463

	void *from;

   464

	void *to;

   465

};

   466

   467

static Town *AiFindRandomTown()

   468

{

   469

	return GetRandomTown();

   470

}

   471

   472

static Industry *AiFindRandomIndustry()

   473

{

   474

	int num = RandomRange(GetMaxIndustryIndex());

   475

	if (IsValidIndustryID(num)) return GetIndustry(num);

   476

   477

	return NULL;

   478

}

   479

   480

static void AiFindSubsidyIndustryRoute(FoundRoute *fr)

   481

{

   482

	uint i;

   483

	CargoID cargo;

   484

	const Subsidy* s;

   485

	Industry* from;

   486

	TileIndex to_xy;

   487

   488

	// initially error

   489

	fr->distance = -1;

   490

   491

	// Randomize subsidy index..

   492

	i = RandomRange(lengthof(_subsidies) * 3);