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 "../../engine.h"

    15

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

    16

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

    17

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

    18

#include "../../station.h"

    19

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

    20

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

    21

#include "../../depot.h"

    22

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

    23

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

    24

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

    25

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

    26

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

    27

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

    28

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

    29

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

    30

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

    31

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

    32

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

    33

#include "default.h"

    35

    36

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

    37

    38

// remove some day perhaps?

    39

static uint _ai_service_interval;

    40

PlayerAI _players_ai[MAX_PLAYERS];

    41

    42

typedef void AiStateAction(Player *p);

    43

    44

enum {

    45

	AIS_0                            =  0,

    46

	AIS_1                            =  1,

    47

	AIS_VEH_LOOP                     =  2,

    48

	AIS_VEH_CHECK_REPLACE_VEHICLE    =  3,

    49

	AIS_VEH_DO_REPLACE_VEHICLE       =  4,

    50

	AIS_WANT_NEW_ROUTE               =  5,

    51

	AIS_BUILD_DEFAULT_RAIL_BLOCKS    =  6,

    52

	AIS_BUILD_RAIL                   =  7,

    53

	AIS_BUILD_RAIL_VEH               =  8,

    54

	AIS_DELETE_RAIL_BLOCKS           =  9,

    55

	AIS_BUILD_DEFAULT_ROAD_BLOCKS    = 10,

    56

	AIS_BUILD_ROAD                   = 11,

    57

	AIS_BUILD_ROAD_VEHICLES          = 12,

    58

	AIS_DELETE_ROAD_BLOCKS           = 13,

    59

	AIS_AIRPORT_STUFF                = 14,

    60

	AIS_BUILD_DEFAULT_AIRPORT_BLOCKS = 15,

    61

	AIS_BUILD_AIRCRAFT_VEHICLES      = 16,

    62

	AIS_CHECK_SHIP_STUFF             = 17,

    63

	AIS_BUILD_DEFAULT_SHIP_BLOCKS    = 18,

    64

	AIS_DO_SHIP_STUFF                = 19,

    65

	AIS_SELL_VEHICLE                 = 20,

    66

	AIS_REMOVE_STATION               = 21,

    67

	AIS_REMOVE_TRACK                 = 22,

    68

	AIS_REMOVE_SINGLE_RAIL_TILE      = 23

    69

};

    70

    71

    72

static TrackBits GetRailTrackStatus(TileIndex tile)

    73

{

    74

	uint32 r = GetTileTrackStatus(tile, TRANSPORT_RAIL, 0);

    75

	return (TrackBits)(byte) (r | r >> 8);

    76

}

    77

    78

    79

static void AiCase0(Player *p)

    80

{

    81

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

    82

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

    83

}

    84

    85

static void AiCase1(Player *p)

    86

{

    87

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

    88

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

    89

}

    90

    91

static void AiStateVehLoop(Player *p)

    92

{

    93

	Vehicle *v;

    94

	uint index;

    95

    96

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

    97

    98

	FOR_ALL_VEHICLES_FROM(v, index) {

    99

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

   100

   101

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

   102

				v->type == VEH_ROAD ||

   103

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

   104

				v->type == VEH_SHIP) {

   105

			/* replace engine? */

   106

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

   107

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

   108

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

   109

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

   110

				return;

   111

			}

   112

   113

			/* not profitable? */

   114

			if (v->age >= 730 &&

   115

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

   116

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

   117

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

   118

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

   119

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

   120

				return;

   121

			}

   122

   123

			/* not reliable? */

   124

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

   125

						v->age != 0 &&

   126

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

   127

					)) {

   128

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

   129

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

   130

				return;

   131

			}

   132

		}

   133

	}

   134

   135

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

   136

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

   137

}

   138

   139

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

   140

{

   141

	EngineID best_veh_index = INVALID_ENGINE;

   142

	byte best_veh_score = 0;

   143

	EngineID i;

   144

   145

	FOR_ALL_ENGINEIDS_OF_TYPE(i, VEH_TRAIN) {

   146

		const RailVehicleInfo *rvi = RailVehInfo(i);

   147

		const Engine* e = GetEngine(i);

   148

   149

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

   150

				rvi->railveh_type == RAILVEH_WAGON ||

   151

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

   152

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

   153

				e->reliability < 0x8A3D) {

   154

			continue;

   155

		}

   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

static EngineID AiChooseAircraftToBuild(Money money, byte flag)

   203

{

   204

	EngineID best_veh_index = INVALID_ENGINE;

   205

	Money best_veh_cost = 0;

   206

	EngineID i;

   207

   208

	FOR_ALL_ENGINEIDS_OF_TYPE(i, VEH_AIRCRAFT) {

   209

		const Engine* e = GetEngine(i);

   210

   211

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

   212

			continue;

   213

		}

   214

   215

		if ((AircraftVehInfo(i)->subtype & AIR_CTOL) != flag) continue;

   216

   217

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

   218

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

   219

			best_veh_cost = ret.GetCost();

   220

			best_veh_index = i;

   221

		}

   222

	}

   223

   224

	return best_veh_index;

   225

}

   226

   227

static Money AiGetBasePrice(const Player* p)

   228

{

   229

	Money base = _price.station_value;

   230

   231

	// adjust base price when more expensive vehicles are available

   232

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

   233

		default: NOT_REACHED();

   234

		case RAILTYPE_RAIL:     break;

   235

		case RAILTYPE_ELECTRIC: break;

   236

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

   237

		case RAILTYPE_MAGLEV:   base *= 2; break;

   238

	}

   239

   240

	return base;

   241

}

   242

   243

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

   244

{

   245

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

   246

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

   247

}

   248

   249

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

   250

{

   251

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

   252

	return AiChooseAircraftToBuild(

   253

		avail_money, AircraftVehInfo(v->engine_type)->subtype & AIR_CTOL

   254

	);

   255

}

   256

   257

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

   258

{

   259

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

   260

	const Vehicle* u = v;

   261

	int num = 0;

   262

   263

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

   264

		u = u->Next();

   265

	}

   266

   267

	// XXX: check if a wagon

   268

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

   269

}

   270

   271

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

   272

{

   273

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

   274

	return INVALID_ENGINE;

   275

}

   276

   277

static void AiHandleGotoDepot(Player *p, int cmd)

   278

{

   279

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

   280

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

   281

   282

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

   283

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

   284

		return;

   285

	}

   286

   287

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

   288

		_players_ai[p->index].cur_veh->current_order.type = OT_DUMMY;

   289

		_players_ai[p->index].cur_veh->current_order.flags = 0;

   290

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

   291

	}

   292

}

   293

   294

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

   295

{

   296

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

   297

   298

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

   299

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

   300

			break;

   301

	}

   302

}

   303

   304

static void AiHandleReplaceTrain(Player *p)

   305

{

   306

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

   307

	BackuppedOrders orderbak;

   308

	EngineID veh;

   309

   310

	// wait until the vehicle reaches the depot.

   311

	if (!IsTileDepotType(v->tile, TRANSPORT_RAIL) || v->u.rail.track != 0x80 || !(v->vehstatus&VS_STOPPED)) {

   312

		AiHandleGotoDepot(p, CMD_SEND_TRAIN_TO_DEPOT);

   313

		return;

   314

	}

   315

   316

	veh = AiChooseTrainToReplaceWith(p, v);

   317

	if (veh != INVALID_ENGINE) {

   318

		TileIndex tile;

   319

   320

		BackupVehicleOrders(v, &orderbak);

   321

		tile = v->tile;

   322

   323

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

   324

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

   325

			VehicleID veh = _new_vehicle_id;

   326

			AiRestoreVehicleOrders(GetVehicle(veh), &orderbak);

   327

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

   328

   329

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

   330

		}

   331

	}

   332

}

   333

   334

static void AiHandleReplaceRoadVeh(Player *p)

   335

{

   336

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

   337

	BackuppedOrders orderbak;

   338

	EngineID veh;

   339

   340

	if (!v->IsStoppedInDepot()) {

   341

		AiHandleGotoDepot(p, CMD_SEND_ROADVEH_TO_DEPOT);

   342

		return;

   343

	}

   344

   345

	veh = AiChooseRoadVehToReplaceWith(p, v);

   346

	if (veh != INVALID_ENGINE) {

   347

		TileIndex tile;

   348

   349

		BackupVehicleOrders(v, &orderbak);

   350

		tile = v->tile;

   351

   352

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

   353

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

   354

			VehicleID veh = _new_vehicle_id;

   355

   356

			AiRestoreVehicleOrders(GetVehicle(veh), &orderbak);

   357

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

   358

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

   359

		}

   360

	}

   361

}

   362

   363

static void AiHandleReplaceAircraft(Player *p)

   364

{

   365

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

   366

	BackuppedOrders orderbak;

   367

	EngineID veh;

   368

   369

	if (!v->IsStoppedInDepot()) {

   370

		AiHandleGotoDepot(p, CMD_SEND_AIRCRAFT_TO_HANGAR);

   371

		return;

   372

	}

   373

   374

	veh = AiChooseAircraftToReplaceWith(p, v);

   375

	if (veh != INVALID_ENGINE) {

   376

		TileIndex tile;

   377

   378

		BackupVehicleOrders(v, &orderbak);

   379

		tile = v->tile;

   380

   381

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

   382

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

   383

			VehicleID veh = _new_vehicle_id;

   384

			AiRestoreVehicleOrders(GetVehicle(veh), &orderbak);

   385

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

   386

   387

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

   388

		}

   389

	}

   390

}

   391

   392

static void AiHandleReplaceShip(Player *p)

   393

{

   394

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

   395

}

   396

   397

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

   398

   399

static CheckReplaceProc* const _veh_check_replace_proc[] = {

   400

	AiChooseTrainToReplaceWith,

   401

	AiChooseRoadVehToReplaceWith,

   402

	AiChooseShipToReplaceWith,

   403

	AiChooseAircraftToReplaceWith,

   404

};

   405

   406

typedef void DoReplaceProc(Player *p);

   407

static DoReplaceProc* const _veh_do_replace_proc[] = {

   408

	AiHandleReplaceTrain,

   409

	AiHandleReplaceRoadVeh,

   410

	AiHandleReplaceShip,

   411

	AiHandleReplaceAircraft

   412

};

   413

   414

static void AiStateCheckReplaceVehicle(Player *p)

   415

{

   416

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

   417

   418

	if (!v->IsValid() ||

   419

			v->owner != _current_player ||

   420

			v->type > VEH_SHIP ||

   421

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

   422

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

   423

	} else {

   424

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

   425

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

   426

	}

   427

}

   428

   429

static void AiStateDoReplaceVehicle(Player *p)

   430

{

   431

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

   432

   433

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

   434

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

   435

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

   436

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

   437

}

   438

   439

struct FoundRoute {

   440

	int distance;

   441

	CargoID cargo;

   442

	void *from;

   443

	void *to;

   444

};

   445

   446

static Town *AiFindRandomTown()

   447

{

   448

	return GetRandomTown();

   449

}

   450

   451

static Industry *AiFindRandomIndustry()

   452

{

   453

	int num = RandomRange(GetMaxIndustryIndex());

   454

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

   455

   456

	return NULL;

   457

}

   458

   459

static void AiFindSubsidyIndustryRoute(FoundRoute *fr)

   460

{

   461

	uint i;

   462

	CargoID cargo;

   463

	const Subsidy* s;

   464

	Industry* from;

   465

	TileIndex to_xy;

   466

   467

	// initially error

   468

	fr->distance = -1;

   469

   470

	// Randomize subsidy index..

   471

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

   472

	if (i >= lengthof(_subsidies)) return;

   473

   474

	s = &_subsidies[i];

   475

   476

	// Don't want passengers or mail

   477

	cargo = s->cargo_type;

   478

	if (cargo == CT_INVALID ||

   479

			cargo == CT_PASSENGERS ||

   480

			cargo == CT_MAIL ||

   481

			s->age > 7) {

   482

		return;

   483

	}

   484

	fr->cargo = cargo;

   485

   486

	fr->from = from = GetIndustry(s->from);

   487

   488

	if (cargo == CT_GOODS || cargo == CT_FOOD) {

   489

		Town* to_tow = GetTown(s->to);

   490

   491

		if (to_tow->population < (cargo == CT_FOOD ? 200U : 900U)) return; // error

   492

		fr->to = to_tow;

   493

		to_xy = to_tow->xy;

   494

	} else {

   495

		Industry* to_ind = GetIndustry(s->to);