1

/* $Id$ */

     2

     3

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

     4

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

     5

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

     6

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

     7

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

     8

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

     9

#include "../../map.h"

    10

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

    11

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

    12

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

    13

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

    14

#include "../../tile.h"

    15

#include "../../player.h"

    16

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

    17

#include "../../vehicle.h"

    18

#include "../../engine.h"

    19

#include "../../command.h"

    20

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

    21

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

    22

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

    23

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

    24

#include "../../economy.h"

    25

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

    26

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

    27

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

    28

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

    29

#include "../../date.h"

    30

#include "../../helpers.hpp"

    31

#include "default.h"

    32

    33

// remove some day perhaps?

    34

static uint _ai_service_interval;

    35

    36

typedef void AiStateAction(Player *p);

    37

    38

enum {

    39

	AIS_0                            =  0,

    40

	AIS_1                            =  1,

    41

	AIS_VEH_LOOP                     =  2,

    42

	AIS_VEH_CHECK_REPLACE_VEHICLE    =  3,

    43

	AIS_VEH_DO_REPLACE_VEHICLE       =  4,

    44

	AIS_WANT_NEW_ROUTE               =  5,

    45

	AIS_BUILD_DEFAULT_RAIL_BLOCKS    =  6,

    46

	AIS_BUILD_RAIL                   =  7,

    47

	AIS_BUILD_RAIL_VEH               =  8,

    48

	AIS_DELETE_RAIL_BLOCKS           =  9,

    49

	AIS_BUILD_DEFAULT_ROAD_BLOCKS    = 10,

    50

	AIS_BUILD_ROAD                   = 11,

    51

	AIS_BUILD_ROAD_VEHICLES          = 12,

    52

	AIS_DELETE_ROAD_BLOCKS           = 13,

    53

	AIS_AIRPORT_STUFF                = 14,

    54

	AIS_BUILD_DEFAULT_AIRPORT_BLOCKS = 15,

    55

	AIS_BUILD_AIRCRAFT_VEHICLES      = 16,

    56

	AIS_CHECK_SHIP_STUFF             = 17,

    57

	AIS_BUILD_DEFAULT_SHIP_BLOCKS    = 18,

    58

	AIS_DO_SHIP_STUFF                = 19,

    59

	AIS_SELL_VEHICLE                 = 20,

    60

	AIS_REMOVE_STATION               = 21,

    61

	AIS_REMOVE_TRACK                 = 22,

    62

	AIS_REMOVE_SINGLE_RAIL_TILE      = 23

    63

};

    64

    65

    66

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

    67

    68

static TrackBits GetRailTrackStatus(TileIndex tile)

    69

{

    70

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

    71

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

    72

}

    73

    74

    75

static void AiCase0(Player *p)

    76

{

    77

	p->ai.state = AIS_REMOVE_TRACK;

    78

	p->ai.state_counter = 0;

    79

}

    80

    81

static void AiCase1(Player *p)

    82

{

    83

	p->ai.cur_veh = NULL;

    84

	p->ai.state = AIS_VEH_LOOP;

    85

}

    86

    87

static void AiStateVehLoop(Player *p)

    88

{

    89

	Vehicle *v;

    90

	uint index;

    91

    92

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

    93

    94

	FOR_ALL_VEHICLES_FROM(v, index) {

    95

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

    96

    97

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

    98

				v->type == VEH_ROAD ||

    99

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

   100

				v->type == VEH_SHIP) {

   101

			/* replace engine? */

   102

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

   103

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

   104

				p->ai.state = AIS_VEH_CHECK_REPLACE_VEHICLE;

   105

				p->ai.cur_veh = v;

   106

				return;

   107

			}

   108

   109

			/* not profitable? */

   110

			if (v->age >= 730 &&

   111

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

   112

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

   113

				p->ai.state_counter = 0;

   114

				p->ai.state = AIS_SELL_VEHICLE;

   115

				p->ai.cur_veh = v;

   116

				return;

   117

			}

   118

   119

			/* not reliable? */

   120

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

   121

						v->age != 0 &&

   122

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

   123

					)) {

   124

				p->ai.state = AIS_VEH_CHECK_REPLACE_VEHICLE;

   125

				p->ai.cur_veh = v;

   126

				return;

   127

			}

   128

		}

   129

	}

   130

   131

	p->ai.state = AIS_WANT_NEW_ROUTE;

   132

	p->ai.state_counter = 0;

   133

}

   134

   135

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

   136

{

   137

	EngineID best_veh_index = INVALID_ENGINE;

   138

	byte best_veh_score = 0;

   139

	CommandCost ret;

   140

	EngineID i;

   141

   142

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

   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

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

   155

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

   156

			best_veh_score = rvi->ai_rank;

   157

			best_veh_index = i;

   158

		}

   159

	}

   160

   161

	return best_veh_index;

   162

}

   163

   164

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

   165

{

   166

	EngineID best_veh_index = INVALID_ENGINE;

   167

	int32 best_veh_rating = 0;

   168

	EngineID i = ROAD_ENGINES_INDEX;

   169

	EngineID end = i + NUM_ROAD_ENGINES;

   170

   171

	for (; i != end; i++) {

   172

		const RoadVehicleInfo *rvi = RoadVehInfo(i);

   173

		const Engine* e = GetEngine(i);

   174

		int32 rating;

   175

		CommandCost ret;

   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

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

   186

		if (rating <= best_veh_rating) continue;

   187

   188

		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 (i = AIRCRAFT_ENGINES_INDEX; i != AIRCRAFT_ENGINES_INDEX + NUM_AIRCRAFT_ENGINES; i++) {

   209

		const Engine* e = GetEngine(i);

   210

		CommandCost ret;

   211

   212

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

   213

			continue;

   214

		}

   215

   216

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

   217

   218

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

   219

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

   220

			best_veh_cost = ret.GetCost();

   221

			best_veh_index = i;

   222

		}

   223

	}

   224

   225

	return best_veh_index;

   226

}

   227

   228

static Money AiGetBasePrice(const Player* p)

   229

{

   230

	Money base = _price.station_value;

   231

   232

	// adjust base price when more expensive vehicles are available

   233

	switch (p->ai.railtype_to_use) {

   234

		default: NOT_REACHED();

   235

		case RAILTYPE_RAIL:     break;

   236

		case RAILTYPE_ELECTRIC: break;

   237

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

   238

		case RAILTYPE_MAGLEV:   base *= 2; break;

   239

	}

   240

   241

	return base;

   242

}

   243

   244

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

   245

{

   246

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

   247

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

   248

}

   249

   250

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

   251

{

   252

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

   253

	return AiChooseAircraftToBuild(

   254

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

   255

	);

   256

}

   257

   258

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

   259

{

   260

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

   261

	const Vehicle* u = v;

   262

	int num = 0;

   263

   264

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

   265

		u = u->next;

   266

	}

   267

   268

	// XXX: check if a wagon

   269

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

   270

}

   271

   272

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

   273

{

   274

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

   275

	return INVALID_ENGINE;

   276

}

   277

   278

static void AiHandleGotoDepot(Player *p, int cmd)

   279

{

   280

	if (p->ai.cur_veh->current_order.type != OT_GOTO_DEPOT)

   281

		DoCommand(0, p->ai.cur_veh->index, 0, DC_EXEC, cmd);

   282

   283

	if (++p->ai.state_counter <= 1387) {

   284

		p->ai.state = AIS_VEH_DO_REPLACE_VEHICLE;

   285

		return;

   286

	}

   287

   288

	if (p->ai.cur_veh->current_order.type == OT_GOTO_DEPOT) {

   289

		p->ai.cur_veh->current_order.type = OT_DUMMY;

   290

		p->ai.cur_veh->current_order.flags = 0;

   291

		InvalidateWindow(WC_VEHICLE_VIEW, p->ai.cur_veh->index);

   292

	}

   293

}

   294

   295

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

   296

{

   297

	uint i;

   298

   299

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

   300

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

   301

			break;

   302

	}

   303

}

   304

   305

static void AiHandleReplaceTrain(Player *p)

   306

{

   307

	const Vehicle* v = p->ai.cur_veh;

   308

	BackuppedOrders orderbak[1];

   309

	EngineID veh;

   310

   311

	// wait until the vehicle reaches the depot.

   312

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

   313

		AiHandleGotoDepot(p, CMD_SEND_TRAIN_TO_DEPOT);

   314

		return;

   315

	}

   316

   317

	veh = AiChooseTrainToReplaceWith(p, v);

   318

	if (veh != INVALID_ENGINE) {

   319

		TileIndex tile;

   320

   321

		BackupVehicleOrders(v, orderbak);

   322

		tile = v->tile;

   323

   324

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

   325

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

   326

			VehicleID veh = _new_vehicle_id;

   327

			AiRestoreVehicleOrders(GetVehicle(veh), orderbak);

   328

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

   329

   330

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

   331

		}

   332

	}

   333

}

   334

   335

static void AiHandleReplaceRoadVeh(Player *p)

   336

{

   337

	const Vehicle* v = p->ai.cur_veh;

   338

	BackuppedOrders orderbak[1];

   339

	EngineID veh;

   340

   342

		AiHandleGotoDepot(p, CMD_SEND_ROADVEH_TO_DEPOT);

   343

		return;

   344

	}

   345

   346

	veh = AiChooseRoadVehToReplaceWith(p, v);

   347

	if (veh != INVALID_ENGINE) {

   348

		TileIndex tile;

   349

   350

		BackupVehicleOrders(v, orderbak);

   351

		tile = v->tile;

   352

   353

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

   354

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

   355

			VehicleID veh = _new_vehicle_id;

   356

   357

			AiRestoreVehicleOrders(GetVehicle(veh), orderbak);

   358

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

   359

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

   360

		}

   361

	}

   362

}

   363

   364

static void AiHandleReplaceAircraft(Player *p)

   365

{

   366

	const Vehicle* v = p->ai.cur_veh;

   367

	BackuppedOrders orderbak[1];

   368

	EngineID veh;

   369

   371

		AiHandleGotoDepot(p, CMD_SEND_AIRCRAFT_TO_HANGAR);

   372

		return;

   373

	}

   374

   375

	veh = AiChooseAircraftToReplaceWith(p, v);

   376

	if (veh != INVALID_ENGINE) {

   377

		TileIndex tile;

   378

   379

		BackupVehicleOrders(v, orderbak);

   380

		tile = v->tile;

   381

   382

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

   383

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

   384

			VehicleID veh = _new_vehicle_id;

   385

			AiRestoreVehicleOrders(GetVehicle(veh), orderbak);

   386

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

   387

   388

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

   389

		}

   390

	}

   391

}

   392

   393

static void AiHandleReplaceShip(Player *p)

   394

{

   395

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

   396

}

   397

   398

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

   399

   400

static CheckReplaceProc* const _veh_check_replace_proc[] = {

   401

	AiChooseTrainToReplaceWith,

   402

	AiChooseRoadVehToReplaceWith,

   403

	AiChooseShipToReplaceWith,

   404

	AiChooseAircraftToReplaceWith,

   405

};

   406

   407

typedef void DoReplaceProc(Player *p);

   408

static DoReplaceProc* const _veh_do_replace_proc[] = {

   409

	AiHandleReplaceTrain,

   410

	AiHandleReplaceRoadVeh,

   411

	AiHandleReplaceShip,

   412

	AiHandleReplaceAircraft

   413

};

   414

   415

static void AiStateCheckReplaceVehicle(Player *p)

   416

{

   417

	const Vehicle* v = p->ai.cur_veh;

   418

   419

	if (!v->IsValid() ||

   420

			v->owner != _current_player ||

   421

			v->type > VEH_SHIP ||

   422

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

   423

		p->ai.state = AIS_VEH_LOOP;

   424

	} else {

   425

		p->ai.state_counter = 0;

   426

		p->ai.state = AIS_VEH_DO_REPLACE_VEHICLE;

   427

	}

   428

}

   429

   430

static void AiStateDoReplaceVehicle(Player *p)

   431

{

   432

	const Vehicle* v = p->ai.cur_veh;

   433

   434

	p->ai.state = AIS_VEH_LOOP;

   435

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

   436

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

   437

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

   438

}

   439

   440

struct FoundRoute {

   441

	int distance;

   442

	CargoID cargo;

   443

	void *from;

   444

	void *to;

   445

};

   446

   447

static Town *AiFindRandomTown()

   448

{

   449

	return GetRandomTown();

   450

}

   451

   452

static Industry *AiFindRandomIndustry()

   453

{

   454

	int num = RandomRange(GetMaxIndustryIndex());

   455

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

   456

   457

	return NULL;

   458

}

   459

   460

static void AiFindSubsidyIndustryRoute(FoundRoute *fr)

   461

{

   462

	uint i;

   463

	CargoID cargo;

   464

	const Subsidy* s;

   465

	Industry* from;

   466

	TileIndex to_xy;

   467

   468

	// initially error

   469

	fr->distance = -1;

   470

   471

	// Randomize subsidy index..

   472

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

   473

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

   474

   475

	s = &_subsidies[i];

   476

   477

	// Don't want passengers or mail

   478

	cargo = s->cargo_type;

   479

	if (cargo == CT_INVALID ||

   480

			cargo == CT_PASSENGERS ||

   481

			cargo == CT_MAIL ||

   482

			s->age > 7) {

   483

		return;

   484

	}

   485

	fr->cargo = cargo;

   486

   487

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

   488

   489

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

   490

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

   491

   492

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

   493

		fr->to = to_tow;

   494

		to_xy = to_tow->xy;

   495

	} else {