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

     9

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

    10

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

    11

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

    12

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

    13

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

    14

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

    15

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

    16

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

    17

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

    18

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

    19

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

    20

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

    21

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

    22

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

    23

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

    24

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

    25

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

    26

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

    27

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

    28

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

    29

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

    30

#include "default.h"

    31

    32

// remove some day perhaps?

    33

static uint _ai_service_interval;

    34

    35

typedef void AiStateAction(Player *p);

    36

    37

enum {

    38

	AIS_0                            =  0,

    39

	AIS_1                            =  1,

    40

	AIS_VEH_LOOP                     =  2,

    41

	AIS_VEH_CHECK_REPLACE_VEHICLE    =  3,

    42

	AIS_VEH_DO_REPLACE_VEHICLE       =  4,

    43

	AIS_WANT_NEW_ROUTE               =  5,

    44

	AIS_BUILD_DEFAULT_RAIL_BLOCKS    =  6,

    45

	AIS_BUILD_RAIL                   =  7,

    46

	AIS_BUILD_RAIL_VEH               =  8,

    47

	AIS_DELETE_RAIL_BLOCKS           =  9,

    48

	AIS_BUILD_DEFAULT_ROAD_BLOCKS    = 10,

    49

	AIS_BUILD_ROAD                   = 11,

    50

	AIS_BUILD_ROAD_VEHICLES          = 12,

    51

	AIS_DELETE_ROAD_BLOCKS           = 13,

    52

	AIS_AIRPORT_STUFF                = 14,

    53

	AIS_BUILD_DEFAULT_AIRPORT_BLOCKS = 15,

    54

	AIS_BUILD_AIRCRAFT_VEHICLES      = 16,

    55

	AIS_CHECK_SHIP_STUFF             = 17,

    56

	AIS_BUILD_DEFAULT_SHIP_BLOCKS    = 18,

    57

	AIS_DO_SHIP_STUFF                = 19,

    58

	AIS_SELL_VEHICLE                 = 20,

    59

	AIS_REMOVE_STATION               = 21,

    60

	AIS_REMOVE_TRACK                 = 22,

    61

	AIS_REMOVE_SINGLE_RAIL_TILE      = 23

    62

};

    63

    64

    65

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

    66

    67

static TrackBits GetRailTrackStatus(TileIndex tile)

    68

{

    69

	uint32 r = GetTileTrackStatus(tile, TRANSPORT_RAIL);

    70

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

    71

}

    72

    73

    74

static void AiCase0(Player *p)

    75

{

    76

	p->ai.state = AIS_REMOVE_TRACK;

    77

	p->ai.state_counter = 0;

    78

}

    79

    80

static void AiCase1(Player *p)

    81

{

    82

	p->ai.cur_veh = NULL;

    83

	p->ai.state = AIS_VEH_LOOP;

    84

}

    85

    86

static void AiStateVehLoop(Player *p)

    87

{

    88

	Vehicle *v;

    89

	uint index;

    90

    91

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

    92

    93

	FOR_ALL_VEHICLES_FROM(v, index) {

    94

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

    95

    96

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

    97

				v->type == VEH_ROAD ||

    98

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

    99

				v->type == VEH_SHIP) {

   100

			/* replace engine? */

   101

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

   102

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

   103

				p->ai.state = AIS_VEH_CHECK_REPLACE_VEHICLE;

   104

				p->ai.cur_veh = v;

   105

				return;

   106

			}

   107

   108

			/* not profitable? */

   109

			if (v->age >= 730 &&

   110

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

   111

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

   112

				p->ai.state_counter = 0;

   113

				p->ai.state = AIS_SELL_VEHICLE;

   114

				p->ai.cur_veh = v;

   115

				return;

   116

			}

   117

   118

			/* not reliable? */

   119

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

   120

						v->age != 0 &&

   121

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

   122

					)) {

   123

				p->ai.state = AIS_VEH_CHECK_REPLACE_VEHICLE;

   124

				p->ai.cur_veh = v;

   125

				return;

   126

			}

   127

		}

   128

	}

   129

   130

	p->ai.state = AIS_WANT_NEW_ROUTE;

   131

	p->ai.state_counter = 0;

   132

}

   133

   134

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

   135

{

   136

	EngineID best_veh_index = INVALID_ENGINE;

   137

	byte best_veh_score = 0;

   138

	int32 ret;

   139

	EngineID i;

   140

   141

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

   142

		const RailVehicleInfo *rvi = RailVehInfo(i);

   143

		const Engine* e = GetEngine(i);

   144

   145

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

   146

				rvi->railveh_type == RAILVEH_WAGON ||

   147

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

   148

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

   149

				e->reliability < 0x8A3D) {

   150

			continue;

   151

		}

   152

   153

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

   154

		if (!CmdFailed(ret) && ret <= money && rvi->ai_rank >= best_veh_score) {

   155

			best_veh_score = rvi->ai_rank;

   156

			best_veh_index = i;

   157

		}

   158

	}

   159

   160

	return best_veh_index;

   161

}

   162

   163

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

   164

{

   165

	EngineID best_veh_index = INVALID_ENGINE;

   166

	int32 best_veh_rating = 0;

   167

	EngineID i = ROAD_ENGINES_INDEX;

   168

	EngineID end = i + NUM_ROAD_ENGINES;

   169

   170

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

   171

		const RoadVehicleInfo *rvi = RoadVehInfo(i);

   172

		const Engine* e = GetEngine(i);

   173

		int32 rating;

   174

		int32 ret;

   175

   176

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

   177

			continue;

   178

		}

   179

   180

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

   181

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

   182

   183

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

   184

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

   185

		if (rating <= best_veh_rating) continue;

   186

   187

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

   188

		if (CmdFailed(ret)) continue;

   189

   190

		/* Add the cost of refitting */

   191

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

   192

		if (ret > money) continue;

   193

   194

		best_veh_rating = rating;

   195

		best_veh_index = i;

   196

	}

   197

   198

	return best_veh_index;

   199

}

   200

   201

static EngineID AiChooseAircraftToBuild(int32 money, byte flag)

   202

{

   203

	EngineID best_veh_index = INVALID_ENGINE;

   204

	int32 best_veh_cost = 0;

   205

	EngineID i;

   206

   207

	for (i = AIRCRAFT_ENGINES_INDEX; i != AIRCRAFT_ENGINES_INDEX + NUM_AIRCRAFT_ENGINES; i++) {

   208

		const Engine* e = GetEngine(i);

   209

		int32 ret;

   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

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

   218

		if (!CmdFailed(ret) && ret <= money && ret >= best_veh_cost) {

   219

			best_veh_cost = ret;

   220

			best_veh_index = i;

   221

		}

   222

	}

   223

   224

	return best_veh_index;

   225

}

   226

   227

static int32 AiGetBasePrice(const Player* p)

   228

{

   229

	int32 base = _price.station_value;

   230

   231

	// adjust base price when more expensive vehicles are available

   232

	switch (p->ai.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

#if 0

   244

static EngineID AiChooseShipToBuild(byte cargo, int32 money)

   245

{

   246

	// XXX: not done

   247

	return INVALID_ENGINE;

   248

}

   249

#endif

   250

   251

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

   252

{

   253

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

   254

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

   255

}

   256

   257

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

   258

{

   259

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

   260

	return AiChooseAircraftToBuild(

   261

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

   262

	);

   263

}

   264

   265

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

   266

{

   267

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

   268

	const Vehicle* u = v;

   269

	int num = 0;

   270

   271

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

   272

		u = u->next;

   273

	}

   274

   275

	// XXX: check if a wagon

   276

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

   277

}

   278

   279

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

   280

{

   281

	error("!AiChooseShipToReplaceWith");

   282

   283

	/* maybe useless, but avoids compiler warning this way */

   284

	return INVALID_ENGINE;

   285

}

   286

   287

static void AiHandleGotoDepot(Player *p, int cmd)

   288

{

   289

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

   290

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

   291

   292

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

   293

		p->ai.state = AIS_VEH_DO_REPLACE_VEHICLE;

   294

		return;

   295

	}

   296

   297

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

   298

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

   299

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

   300

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

   301

	}

   302

}

   303

   304

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

   305

{

   306

	uint i;

   307

   308

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

   309

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

   310

			break;

   311

	}

   312

}

   313

   314

static void AiHandleReplaceTrain(Player *p)

   315

{

   316

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

   317

	BackuppedOrders orderbak[1];

   318

	EngineID veh;

   319

   320

	// wait until the vehicle reaches the depot.

   321

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

   322

		AiHandleGotoDepot(p, CMD_SEND_TRAIN_TO_DEPOT);

   323

		return;

   324

	}

   325

   326

	veh = AiChooseTrainToReplaceWith(p, v);

   327

	if (veh != INVALID_ENGINE) {

   328

		TileIndex tile;

   329

   330

		BackupVehicleOrders(v, orderbak);

   331

		tile = v->tile;

   332

   333

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

   334

				!CmdFailed(DoCommand(tile, veh, 0, DC_EXEC, CMD_BUILD_RAIL_VEHICLE))) {

   335

			VehicleID veh = _new_vehicle_id;

   336

			AiRestoreVehicleOrders(GetVehicle(veh), orderbak);

   337

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

   338

   339

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

   340

		}

   341

	}

   342

}

   343

   344

static void AiHandleReplaceRoadVeh(Player *p)

   345

{

   346

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

   347

	BackuppedOrders orderbak[1];

   348

	EngineID veh;

   349

   350

	if (!IsRoadVehInDepotStopped(v)) {

   351

		AiHandleGotoDepot(p, CMD_SEND_ROADVEH_TO_DEPOT);

   352

		return;

   353

	}

   354

   355

	veh = AiChooseRoadVehToReplaceWith(p, v);

   356

	if (veh != INVALID_ENGINE) {

   357

		TileIndex tile;

   358

   359

		BackupVehicleOrders(v, orderbak);

   360

		tile = v->tile;

   361

   362

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

   363

				!CmdFailed(DoCommand(tile, veh, 0, DC_EXEC, CMD_BUILD_ROAD_VEH))) {

   364

			VehicleID veh = _new_vehicle_id;

   365

   366

			AiRestoreVehicleOrders(GetVehicle(veh), orderbak);

   367

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

   368

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

   369

		}

   370

	}

   371

}

   372

   373

static void AiHandleReplaceAircraft(Player *p)

   374

{

   375

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

   376

	BackuppedOrders orderbak[1];

   377

	EngineID veh;

   378

   379

	if (!IsAircraftInHangarStopped(v)) {

   380

		AiHandleGotoDepot(p, CMD_SEND_AIRCRAFT_TO_HANGAR);

   381

		return;

   382

	}

   383

   384

	veh = AiChooseAircraftToReplaceWith(p, v);

   385

	if (veh != INVALID_ENGINE) {

   386

		TileIndex tile;

   387

   388

		BackupVehicleOrders(v, orderbak);

   389

		tile = v->tile;

   390

   391

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

   392

				!CmdFailed(DoCommand(tile, veh, 0, DC_EXEC, CMD_BUILD_AIRCRAFT))) {

   393

			VehicleID veh = _new_vehicle_id;

   394

			AiRestoreVehicleOrders(GetVehicle(veh), orderbak);

   395

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

   396

   397

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

   398

		}

   399

	}

   400

}

   401

   402

static void AiHandleReplaceShip(Player *p)

   403

{

   404

	error("!AiHandleReplaceShip");

   405

}

   406

   407

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

   408

   409

static CheckReplaceProc* const _veh_check_replace_proc[] = {

   410

	AiChooseTrainToReplaceWith,

   411

	AiChooseRoadVehToReplaceWith,

   412

	AiChooseShipToReplaceWith,

   413

	AiChooseAircraftToReplaceWith,

   414

};

   415

   416

typedef void DoReplaceProc(Player *p);

   417

static DoReplaceProc* const _veh_do_replace_proc[] = {

   418

	AiHandleReplaceTrain,

   419

	AiHandleReplaceRoadVeh,

   420

	AiHandleReplaceShip,

   421

	AiHandleReplaceAircraft

   422

};

   423

   424

static void AiStateCheckReplaceVehicle(Player *p)

   425

{

   426

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

   427

   428

	if (!IsValidVehicle(v) ||

   429

			v->owner != _current_player ||

   430

			v->type > VEH_SHIP ||

   431

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

   432

		p->ai.state = AIS_VEH_LOOP;

   433

	} else {

   434

		p->ai.state_counter = 0;

   435

		p->ai.state = AIS_VEH_DO_REPLACE_VEHICLE;

   436

	}

   437

}

   438

   439

static void AiStateDoReplaceVehicle(Player *p)

   440

{

   441

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

   442

   443

	p->ai.state = AIS_VEH_LOOP;

   444

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

   445

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

   446

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

   447

}

   448

   449

struct FoundRoute {

   450

	int distance;

   451

	CargoID cargo;

   452

	void *from;

   453

	void *to;

   454

};

   455

   456

static Town *AiFindRandomTown()

   457

{

   458

	return GetRandomTown();

   459

}

   460

   461

static Industry *AiFindRandomIndustry()

   462

{

   463

	int num = RandomRange(GetMaxIndustryIndex());

   464

	if (IsValidIndustry(GetIndustry(num))) return GetIndustry(num);

   465

   466

	return NULL;

   467

}

   468

   469

static void AiFindSubsidyIndustryRoute(FoundRoute *fr)

   470

{

   471

	uint i;

   472

	CargoID cargo;

   473

	const Subsidy* s;

   474

	Industry* from;

   475

	TileIndex to_xy;

   476

   477

	// initially error

   478

	fr->distance = -1;

   479

   480

	// Randomize subsidy index..

   481

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

   482

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

   483

   484

	s = &_subsidies[i];

   485

   486

	// Don't want passengers or mail

   487

	cargo = s->cargo_type;

   488

	if (cargo == CT_INVALID ||

   489

			cargo == CT_PASSENGERS ||

   490

			cargo == CT_MAIL ||

   491

			s->age > 7) {

   492

		return;

   493

	}

   494

	fr->cargo = cargo;

   495

   496

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

   497