1

/* $Id$ */

     2

     3

/** @file default.cpp The original AI. */

     4

     5

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

     6

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

     7

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

     8

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

     9

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

    10

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

    11

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

    12

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

    13

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

    14

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

    15

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

    16

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

    17

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

    18

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

    19

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

    20

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

    21

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

    22

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

    23

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

    24

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

    25

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

    26

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

    27

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

    28

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

    29

#include "../../company_func.h"

    30

#include "../../company_base.h"

    31

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

    32

#include "default.h"

    33

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

    34

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

    35

    36

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

    37

    38

// remove some day perhaps?

    39

static uint _ai_service_interval;

    40

CompanyAI _companies_ai[MAX_COMPANIES];

    41

    42

typedef void AiStateAction(Company *c);

    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 inline TrackBits GetRailTrackStatus(TileIndex tile)

    73

{

    74

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

    75

}

    76

    77

    78

static void AiCase0(Company *c)

    79

{

    80

	_companies_ai[c->index].state = AIS_REMOVE_TRACK;

    81

	_companies_ai[c->index].state_counter = 0;

    82

}

    83

    84

static void AiCase1(Company *c)

    85

{

    86

	_companies_ai[c->index].cur_veh = NULL;

    87

	_companies_ai[c->index].state = AIS_VEH_LOOP;

    88

}

    89

    90

static void AiStateVehLoop(Company *c)

    91

{

    92

	Vehicle *v;

    93

	uint index;

    94

    95

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

    96

    97

	FOR_ALL_VEHICLES_FROM(v, index) {

    98

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

    99

   100

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

   101

				v->type == VEH_ROAD ||

   102

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

   103

				v->type == VEH_SHIP) {

   104

			/* replace engine? */

   105

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

   106

					(_price.build_railvehicle >> 3) < c->money) {

   107

				_companies_ai[c->index].state = AIS_VEH_CHECK_REPLACE_VEHICLE;

   108

				_companies_ai[c->index].cur_veh = v;

   109

				return;

   110

			}

   111

   112

			/* not profitable? */

   113

			if (v->age >= 730 &&

   114

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

   115

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

   116

				_companies_ai[c->index].state_counter = 0;

   117

				_companies_ai[c->index].state = AIS_SELL_VEHICLE;

   118

				_companies_ai[c->index].cur_veh = v;

   119

				return;

   120

			}

   121

   122

			/* not reliable? */

   123

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

   124

						v->age != 0 &&

   125

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

   126

					)) {

   127

				_companies_ai[c->index].state = AIS_VEH_CHECK_REPLACE_VEHICLE;

   128

				_companies_ai[c->index].cur_veh = v;

   129

				return;

   130

			}

   131

		}

   132

	}

   133

   134

	_companies_ai[c->index].state = AIS_WANT_NEW_ROUTE;

   135

	_companies_ai[c->index].state_counter = 0;

   136

}

   137

   138

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

   139

{

   140

	EngineID best_veh_index = INVALID_ENGINE;

   141

	byte best_veh_score = 0;

   142

	const Engine *e;

   143

   144

	FOR_ALL_ENGINES_OF_TYPE(e, VEH_TRAIN) {

   145

		EngineID i = e->index;

   146

		const RailVehicleInfo *rvi = &e->u.rail;

   147

   148

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

   149

				rvi->railveh_type == RAILVEH_WAGON ||

   150

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

   151

				!HasBit(e->company_avail, _current_company) ||

   152

				e->reliability < 0x8A3D) {

   153

			continue;

   154

		}

   155

   156

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

   157

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

   158

   159

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

   160

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

   161

			best_veh_score = rvi->ai_rank;

   162

			best_veh_index = i;

   163

		}

   164

	}

   165

   166

	return best_veh_index;

   167

}

   168

   169

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

   170

{

   171

	EngineID best_veh_index = INVALID_ENGINE;

   172

	int32 best_veh_rating = 0;

   173

	const Engine *e;

   174

   175

	FOR_ALL_ENGINES_OF_TYPE(e, VEH_ROAD) {

   176

		EngineID i = e->index;

   177

		const RoadVehicleInfo *rvi = &e->u.road;

   178

   179

		if (!HasBit(e->company_avail, _current_company) || e->reliability < 0x8A3D) {

   180

			continue;

   181

		}

   182

   183

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

   184

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

   185

   186

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

   187

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

   188

		if (rating <= best_veh_rating) continue;

   189

   190

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

   191

		if (CmdFailed(ret)) continue;

   192

   193

		/* Add the cost of refitting */

   194

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

   195

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

   196

   197

		best_veh_rating = rating;

   198

		best_veh_index = i;

   199

	}

   200

   201

	return best_veh_index;

   202

}

   203

   204

/**

   205

 * Choose aircraft to build.

   206

 * @param money current AI money

   207

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

   208

 * @return EngineID of aircraft to build

   209

 */

   210

static EngineID AiChooseAircraftToBuild(Money money, byte forbidden)

   211

{

   212

	EngineID best_veh_index = INVALID_ENGINE;

   213

	Money best_veh_cost = 0;

   214

	const Engine *e;

   215

   216

	FOR_ALL_ENGINES_OF_TYPE(e, VEH_AIRCRAFT) {

   217

		EngineID i = e->index;

   218

		const AircraftVehicleInfo *avi = &e->u.air;

   219

   220

		if (!HasBit(e->company_avail, _current_company) || e->reliability < 0x8A3D) {

   221

			continue;

   222

		}

   223

   224

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

   225

   226

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

   227

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

   228

			best_veh_cost = ret.GetCost();

   229

			best_veh_index = i;

   230

		}

   231

	}

   232

   233

	return best_veh_index;

   234

}

   235

   236

static Money AiGetBasePrice(const Company *c)

   237

{

   238

	Money base = _price.station_value;

   239

   240

	// adjust base price when more expensive vehicles are available

   241

	switch (_companies_ai[c->index].railtype_to_use) {

   242

		default: NOT_REACHED();

   243

		case RAILTYPE_RAIL:     break;

   244

		case RAILTYPE_ELECTRIC: break;

   245

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

   246

		case RAILTYPE_MAGLEV:   base *= 2; break;

   247

	}

   248

   249

	return base;

   250

}

   251

   252

static EngineID AiChooseRoadVehToReplaceWith(const Company *c, const Vehicle *v)

   253

{

   254

	Money avail_money = c->money + v->value;

   255

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

   256

}

   257

   258

static EngineID AiChooseAircraftToReplaceWith(const Company *c, const Vehicle *v)

   259

{

   260

	Money avail_money = c->money + v->value;

   261

   262

	/* determine forbidden aircraft bits */

   263

	byte forbidden = 0;

   264

	const Order *o;

   265

   266

	FOR_VEHICLE_ORDERS(v, o) {

   267

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

   268

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

   269

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

   270

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

   271

   272

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

   273

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

   274

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

   275

	}

   276

   277

	return AiChooseAircraftToBuild(

   278

		avail_money, forbidden

   279

	);

   280

}

   281

   282

static EngineID AiChooseTrainToReplaceWith(const Company *c, const Vehicle *v)

   283

{

   284

	Money avail_money = c->money + v->value;

   285

	const Vehicle *u = v;

   286

	int num = 0;

   287

   288

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

   289

		u = u->Next();

   290

	}

   291

   292

	// XXX: check if a wagon

   293

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

   294

}

   295

   297

{

   298

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

   299

	return INVALID_ENGINE;

   300

}

   301

   302

static void AiHandleGotoDepot(Company *c, int cmd)

   303

{

   304

	if (!_companies_ai[c->index].cur_veh->current_order.IsType(OT_GOTO_DEPOT))

   305

		DoCommand(0, _companies_ai[c->index].cur_veh->index, 0, DC_EXEC, cmd);

   306

   307

	if (++_companies_ai[c->index].state_counter <= 1387) {

   308

		_companies_ai[c->index].state = AIS_VEH_DO_REPLACE_VEHICLE;

   309

		return;

   310

	}

   311

   312

	if (_companies_ai[c->index].cur_veh->current_order.IsType(OT_GOTO_DEPOT)) {

   313

		_companies_ai[c->index].cur_veh->current_order.MakeDummy();

   314

		InvalidateWindow(WC_VEHICLE_VIEW, _companies_ai[c->index].cur_veh->index);

   315

	}

   316

}

   317

   318

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

   319

{

   320

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

   321

   322

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

   323

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

   324

			break;

   325

	}

   326

}

   327

   328

static void AiHandleReplaceTrain(Company *c)

   329

{

   330

	const Vehicle* v = _companies_ai[c->index].cur_veh;

   331

	BackuppedOrders orderbak;

   332

	EngineID veh;

   333

   334

	// wait until the vehicle reaches the depot.

   335

	if (!IsRailDepotTile(v->tile) || v->u.rail.track != TRACK_BIT_DEPOT || !(v->vehstatus & VS_STOPPED)) {

   336

		AiHandleGotoDepot(c, CMD_SEND_TRAIN_TO_DEPOT);

   337

		return;

   338

	}

   339

   340

	veh = AiChooseTrainToReplaceWith(c, v);

   341

	if (veh != INVALID_ENGINE) {

   342

		TileIndex tile;

   343

   344

		BackupVehicleOrders(v, &orderbak);

   345

		tile = v->tile;

   346

   347

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

   348

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

   349

			VehicleID veh = _new_vehicle_id;

   350

			AiRestoreVehicleOrders(GetVehicle(veh), &orderbak);

   351

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

   352

   353

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

   354

		}

   355

	}

   356

}

   357

   358

static void AiHandleReplaceRoadVeh(Company *c)

   359

{

   360

	const Vehicle* v = _companies_ai[c->index].cur_veh;

   361

	BackuppedOrders orderbak;

   362

	EngineID veh;

   363

   364

	if (!v->IsStoppedInDepot()) {

   365

		AiHandleGotoDepot(c, CMD_SEND_ROADVEH_TO_DEPOT);

   366

		return;

   367

	}

   368

   369

	veh = AiChooseRoadVehToReplaceWith(c, v);

   370

	if (veh != INVALID_ENGINE) {

   371

		TileIndex tile;

   372

   373

		BackupVehicleOrders(v, &orderbak);

   374

		tile = v->tile;

   375

   376

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

   377

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

   378

			VehicleID veh = _new_vehicle_id;

   379

   380

			AiRestoreVehicleOrders(GetVehicle(veh), &orderbak);

   381

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

   382

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

   383

		}

   384

	}

   385

}

   386

   387

static void AiHandleReplaceAircraft(Company *c)

   388

{

   389

	const Vehicle* v = _companies_ai[c->index].cur_veh;

   390

	BackuppedOrders orderbak;

   391

	EngineID veh;

   392

   393

	if (!v->IsStoppedInDepot()) {

   394

		AiHandleGotoDepot(c, CMD_SEND_AIRCRAFT_TO_HANGAR);

   395

		return;

   396

	}

   397

   398

	veh = AiChooseAircraftToReplaceWith(c, v);

   399

	if (veh != INVALID_ENGINE) {

   400

		TileIndex tile;

   401

   402

		BackupVehicleOrders(v, &orderbak);

   403

		tile = v->tile;

   404

   405

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

   406

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

   407

			VehicleID veh = _new_vehicle_id;

   408

			AiRestoreVehicleOrders(GetVehicle(veh), &orderbak);

   409

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

   410

   411

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

   412

		}

   413

	}

   414

}

   415

   416

static void AiHandleReplaceShip(Company *c)

   417

{

   418

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

   419

}

   420

   421

typedef EngineID CheckReplaceProc(const Company *c, const Vehicle* v);

   422

   423

static CheckReplaceProc* const _veh_check_replace_proc[] = {

   424

	AiChooseTrainToReplaceWith,

   425

	AiChooseRoadVehToReplaceWith,

   426

	AiChooseShipToReplaceWith,

   427

	AiChooseAircraftToReplaceWith,

   428

};

   429

   430

typedef void DoReplaceProc(Company *c);

   431

static DoReplaceProc* const _veh_do_replace_proc[] = {

   432

	AiHandleReplaceTrain,

   433

	AiHandleReplaceRoadVeh,

   434

	AiHandleReplaceShip,

   435

	AiHandleReplaceAircraft

   436

};

   437

   438

static void AiStateCheckReplaceVehicle(Company *c)

   439

{

   440

	const Vehicle* v = _companies_ai[c->index].cur_veh;

   441

   442

	if (!v->IsValid() ||

   443

			v->owner != _current_company ||

   444

			v->type > VEH_SHIP ||

   445

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

   446

		_companies_ai[c->index].state = AIS_VEH_LOOP;

   447

	} else {

   448

		_companies_ai[c->index].state_counter = 0;

   449

		_companies_ai[c->index].state = AIS_VEH_DO_REPLACE_VEHICLE;

   450

	}

   451

}

   452

   453

static void AiStateDoReplaceVehicle(Company *c)

   454

{

   455

	const Vehicle* v = _companies_ai[c->index].cur_veh;

   456

   457

	_companies_ai[c->index].state = AIS_VEH_LOOP;

   458

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

   459

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

   460

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

   461

}

   462

   463

struct FoundRoute {

   464

	int distance;

   465

	CargoID cargo;

   466

	void *from;

   467

	void *to;

   468

};

   469

   470

static Town *AiFindRandomTown()

   471

{

   472

	return GetRandomTown();

   473

}

   474

   475

static Industry *AiFindRandomIndustry()

   476

{

   477

	int num = RandomRange(GetMaxIndustryIndex());

   478

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

   479

   480

	return NULL;

   481

}

   482

   483

static void AiFindSubsidyIndustryRoute(FoundRoute *fr)

   484

{

   485

	uint i;

   486

	CargoID cargo;

   487

	const Subsidy* s;

   488

	Industry* from;

   489

	TileIndex to_xy;

   490

   491

	// initially error

   492