1

/* $Id$ */

     2

     3

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

     4

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

     5

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

     6

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

     7

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

     8

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

     9

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

    11

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

    12

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

    13

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

    14

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

    15

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

    16

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

    17

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

    18

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

    19

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

    20

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

    21

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

    22

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

    23

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

    24

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

    25

#include "default.h"

    26

    27

// remove some day perhaps?

    28

static Player *_cur_ai_player;

    29

static uint _ai_service_interval;

    30

    31

typedef void AiStateAction(Player *p);

    32

    33

enum {

    34

	AIS_0 = 0,

    35

	AIS_1 = 1,

    36

	AIS_VEH_LOOP = 2,

    37

	AIS_VEH_CHECK_REPLACE_VEHICLE = 3,

    38

	AIS_VEH_DO_REPLACE_VEHICLE = 4,

    39

	AIS_WANT_NEW_ROUTE = 5,

    40

	AIS_BUILD_DEFAULT_RAIL_BLOCKS = 6,

    41

	AIS_BUILD_RAIL = 7,

    42

	AIS_BUILD_RAIL_VEH = 8,

    43

	AIS_DELETE_RAIL_BLOCKS = 9,

    44

	AIS_BUILD_DEFAULT_ROAD_BLOCKS = 10,

    45

	AIS_BUILD_ROAD = 11,

    46

	AIS_BUILD_ROAD_VEHICLES = 12,

    47

	AIS_DELETE_ROAD_BLOCKS = 13,

    48

	AIS_AIRPORT_STUFF = 14,

    49

	AIS_BUILD_DEFAULT_AIRPORT_BLOCKS = 15,

    50

	AIS_BUILD_AIRCRAFT_VEHICLES = 16,

    51

	AIS_CHECK_SHIP_STUFF = 17,

    52

	AIS_BUILD_DEFAULT_SHIP_BLOCKS = 18,

    53

	AIS_DO_SHIP_STUFF = 19,

    54

	AIS_SELL_VEHICLE = 20,

    55

	AIS_REMOVE_STATION = 21,

    56

	AIS_REMOVE_TRACK = 22,

    57

	AIS_REMOVE_SINGLE_RAIL_TILE = 23

    58

};

    59

    60

    61

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

    62

    63

static byte GetRailTrackStatus(TileIndex tile)

    64

{

    65

	uint32 r = GetTileTrackStatus(tile, TRANSPORT_RAIL);

    66

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

    67

}

    68

    69

    70

static void AiCase0(Player *p)

    71

{

    72

	p->ai.state = AIS_REMOVE_TRACK;

    73

	p->ai.state_counter = 0;

    74

}

    75

    76

static void AiCase1(Player *p)

    77

{

    78

	p->ai.cur_veh = NULL;

    79

	p->ai.state = AIS_VEH_LOOP;

    80

}

    81

    82

static void AiStateVehLoop(Player *p)

    83

{

    84

	Vehicle *v;

    85

	uint index;

    86

    87

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

    88

    89

	FOR_ALL_VEHICLES_FROM(v, index) {

    90

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

    91

    92

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

    93

				v->type == VEH_Road ||

    94

				(v->type == VEH_Aircraft && v->subtype <= 2) ||

    95

				v->type == VEH_Ship) {

    96

			/* replace engine? */

    97

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

    98

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

    99

				p->ai.state = AIS_VEH_CHECK_REPLACE_VEHICLE;

   100

				p->ai.cur_veh = v;

   101

				return;

   102

			}

   103

   104

			/* not profitable? */

   105

			if (v->age >= 730 &&

   106

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

   107

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

   108

				p->ai.state_counter = 0;

   109

				p->ai.state = AIS_SELL_VEHICLE;

   110

				p->ai.cur_veh = v;

   111

				return;

   112

			}

   113

   114

			/* not reliable? */

   115

			if ((v->age != 0 &&

   116

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

   117

					v->age >= v->max_age) {

   118

				p->ai.state = AIS_VEH_CHECK_REPLACE_VEHICLE;

   119

				p->ai.cur_veh = v;

   120

				return;

   121

			}

   122

		}

   123

	}

   124

   125

	p->ai.state = AIS_WANT_NEW_ROUTE;

   126

	p->ai.state_counter = 0;

   127

}

   128

   129

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

   130

{

   131

	EngineID best_veh_index = INVALID_ENGINE;

   132

	byte best_veh_score = 0;

   133

	int32 ret;

   134

	EngineID i;

   135

   136

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

   137

		const RailVehicleInfo *rvi = RailVehInfo(i);

   138

		const Engine* e = GetEngine(i);

   139

   140

		if (e->railtype != railtype ||

   141

				rvi->flags & RVI_WAGON ||

   142

				(rvi->flags & RVI_MULTIHEAD && flag & 1) ||

   143

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

   144

				e->reliability < 0x8A3D) {

   145

			continue;

   146

		}

   147

   148

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

   149

		if (!CmdFailed(ret) && ret <= money &&

   150

				rvi->ai_rank >= best_veh_score) {

   151

			best_veh_score = rvi->ai_rank;

   152

			best_veh_index = i;

   153

		}

   154

	}

   155

   156

	return best_veh_index;

   157

}

   158

   159

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

   160

{

   161

	EngineID best_veh_index = INVALID_ENGINE;

   162

	int32 best_veh_cost = 0;

   163

	int32 ret;

   164

   165

	EngineID i = _cargoc.ai_roadveh_start[cargo];

   166

	EngineID end = i + _cargoc.ai_roadveh_count[cargo];

   167

   168

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

   169

		const Engine* e = GetEngine(i);

   170

   171

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

   172

			continue;

   173

   174

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

   175

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

   176

			best_veh_cost = ret;

   177

			best_veh_index = i;

   178

		}

   179

	}

   180

   181

	return best_veh_index;

   182

}

   183

   184

static EngineID AiChooseAircraftToBuild(int32 money, byte flag)

   185

{

   186

	EngineID best_veh_index = INVALID_ENGINE;

   187

	int32 best_veh_cost = 0;

   188

	int32 ret;

   189

   190

	EngineID i = AIRCRAFT_ENGINES_INDEX;

   191

	EngineID end = i + NUM_AIRCRAFT_ENGINES;

   192

   193

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

   194

		const Engine* e = GetEngine(i);

   195

   196

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

   197

			continue;

   198

   199

		if (flag&1) {

   200

			if (i<253) continue;

   201

		} else {

   202

			if (i>=253) continue;

   203

		}

   204

   205

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

   206

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

   207

			best_veh_cost = ret;

   208

			best_veh_index = i;

   209

		}

   210

	}

   211

   212

	return best_veh_index;

   213

}

   214

   215

static int32 AiGetBasePrice(Player *p)

   216

{

   217

	int32 base = _price.station_value;

   218

   219

	// adjust base price when more expensive vehicles are available

   220

	switch (p->ai.railtype_to_use) {

   221

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

   222

		case 2: base *= 2; break;

   223

	}

   224

   225

	return base;

   226

}

   227

   228

#if 0

   229

static EngineID AiChooseShipToBuild(byte cargo, int32 money)

   230

{

   231

	// XXX: not done

   232

	return INVALID_ENGINE;

   233

}

   234

#endif

   235

   236

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

   237

{

   238

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

   239

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

   240

}

   241

   242

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

   243

{

   244

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

   245

	return AiChooseAircraftToBuild(avail_money, v->engine_type>=253?1:0);

   246

}

   247

   248

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

   249

{

   250

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

   251

	int num=0;

   252

	Vehicle *u = v;

   253

   254

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

   255

		u = u->next;

   256

	}

   257

   258

	// XXX: check if a wagon

   259

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

   260

}

   261

   262

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

   263

{

   264

	error("!AiChooseShipToReplaceWith");

   265

   266

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

   267

	return INVALID_ENGINE;

   268

}

   269

   270

static void AiHandleGotoDepot(Player *p, int cmd)

   271

{

   272

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

   273

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

   274

   275

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

   276

		p->ai.state = AIS_VEH_DO_REPLACE_VEHICLE;

   277

		return;

   278

	}

   279

   280

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

   281

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

   282

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

   283

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

   284

	}

   285

}

   286

   287

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

   288

{

   289

	uint i;

   290

   291

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

   292

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

   293

			break;

   294

	}

   295

}

   296

   297

static void AiHandleReplaceTrain(Player *p)

   298

{

   299

	Vehicle *v = p->ai.cur_veh;

   300

	BackuppedOrders orderbak[1];

   301

	EngineID veh;

   302

   303

	// wait until the vehicle reaches the depot.

   304

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

   305

		AiHandleGotoDepot(p, CMD_TRAIN_GOTO_DEPOT);

   306

		return;

   307

	}

   308

   309

	veh = AiChooseTrainToReplaceWith(p, v);

   310

	if (veh != INVALID_ENGINE) {

   311

		TileIndex tile;

   312

   313

		BackupVehicleOrders(v, orderbak);

   314

		tile = v->tile;

   315

   316

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

   317

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

   318

			VehicleID veh = _new_train_id;

   319

			AiRestoreVehicleOrders(GetVehicle(veh), orderbak);

   320

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

   321

   322

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

   323

		}

   324

	}

   325

}

   326

   327

static void AiHandleReplaceRoadVeh(Player *p)

   328

{

   329

	Vehicle *v = p->ai.cur_veh;

   330

	BackuppedOrders orderbak[1];

   331

	EngineID veh;

   332

   333

	if (!IsTileDepotType(v->tile, TRANSPORT_ROAD) || v->u.road.state != 254 || !(v->vehstatus&VS_STOPPED)) {

   334

		AiHandleGotoDepot(p, CMD_SEND_ROADVEH_TO_DEPOT);

   335

		return;

   336

	}

   337

   338

	veh = AiChooseRoadVehToReplaceWith(p, v);

   339

	if (veh != INVALID_ENGINE) {

   340

		TileIndex tile;

   341

   342

		BackupVehicleOrders(v, orderbak);

   343

		tile = v->tile;

   344

   345

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

   346

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

   347

			VehicleID veh = _new_roadveh_id;

   348

			AiRestoreVehicleOrders(GetVehicle(veh), orderbak);

   349

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

   350

   351

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

   352

		}

   353

	}

   354

}

   355

   356

static void AiHandleReplaceAircraft(Player *p)

   357

{

   358

	Vehicle *v = p->ai.cur_veh;

   359

	BackuppedOrders orderbak[1];

   360

	EngineID veh;

   361

   362

	if (!IsAircraftHangarTile(v->tile) && !(v->vehstatus&VS_STOPPED)) {

   363

		AiHandleGotoDepot(p, CMD_SEND_AIRCRAFT_TO_HANGAR);

   364

		return;

   365

	}

   366

   367

	veh = AiChooseAircraftToReplaceWith(p, v);

   368

	if (veh != INVALID_ENGINE) {

   369

		TileIndex tile;

   370

   371

		BackupVehicleOrders(v, orderbak);

   372

		tile = v->tile;

   373

   374

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

   375

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

   376

			VehicleID veh = _new_aircraft_id;

   377

			AiRestoreVehicleOrders(GetVehicle(veh), orderbak);

   378

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

   379

   380

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

   381

		}

   382

	}

   383

}

   384

   385

static void AiHandleReplaceShip(Player *p)

   386

{

   387

	error("!AiHandleReplaceShip");

   388

}

   389

   390

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

   391

   392

static CheckReplaceProc* const _veh_check_replace_proc[] = {

   393

	AiChooseTrainToReplaceWith,

   394

	AiChooseRoadVehToReplaceWith,

   395

	AiChooseShipToReplaceWith,

   396

	AiChooseAircraftToReplaceWith,

   397

};

   398

   399

typedef void DoReplaceProc(Player *p);

   400

static DoReplaceProc* const _veh_do_replace_proc[] = {

   401

	AiHandleReplaceTrain,

   402

	AiHandleReplaceRoadVeh,

   403

	AiHandleReplaceShip,

   404

	AiHandleReplaceAircraft

   405

};

   406

   407

static void AiStateCheckReplaceVehicle(Player *p)

   408

{

   409

	Vehicle *v = p->ai.cur_veh;

   410

   411

	if (v->type == 0 ||

   412

			v->owner != _current_player ||

   413

			v->type > VEH_Ship ||

   414

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

   415

		p->ai.state = AIS_VEH_LOOP;

   416

	} else {

   417

		p->ai.state_counter = 0;

   418

		p->ai.state = AIS_VEH_DO_REPLACE_VEHICLE;

   419

	}

   420

}

   421

   422

static void AiStateDoReplaceVehicle(Player *p)

   423

{

   424

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

   425

   426

	p->ai.state = AIS_VEH_LOOP;

   427

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

   428

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

   429

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

   430

}

   431

   432

typedef struct FoundRoute {

   433

	int distance;

   434

	byte cargo;

   435

	void *from;

   436

	void *to;

   437

} FoundRoute;

   438

   439

static Town *AiFindRandomTown(void)

   440

{

   441

	Town *t = GetTown(RandomRange(_total_towns));

   442

	return (t->xy != 0) ? t : NULL;

   443

}

   444

   445

static Industry *AiFindRandomIndustry(void)

   446

{

   447

	Industry *i = GetIndustry(RandomRange(_total_industries));

   448

	return (i->xy != 0) ? i : NULL;

   449

}

   450

   451

static void AiFindSubsidyIndustryRoute(FoundRoute *fr)

   452

{

   453

	uint i;

   454

	byte cargo;

   455

	Subsidy *s;

   456

	Industry *from, *to_ind;

   457

	Town *to_tow;

   458

	TileIndex to_xy;

   459

   460

	// initially error

   461

	fr->distance = -1;

   462

   463

	// Randomize subsidy index..

   464

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

   465

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

   466

   467

	s = &_subsidies[i];

   468

   469

	// Don't want passengers or mail

   470

	cargo = s->cargo_type;

   471

	if (cargo == 0xFF || cargo == CT_PASSENGERS || cargo == CT_MAIL || s->age > 7)

   472

		return;

   473

	fr->cargo = cargo;

   474

   475

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

   476

   477

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

   478

		to_tow = GetTown(s->to);

   479

		if (to_tow->population < (uint32)(cargo == CT_FOOD ? 200 : 900))

   480

			return; // error

   481

		fr->to = to_tow;

   482

		to_xy = to_tow->xy;

   483

	} else {

   484

		to_ind = GetIndustry(s->to);

   485

		fr->to = to_ind;

   486

		to_xy = to_ind->xy;

   487

	}

   488

   489

	fr->distance = DistanceManhattan(from->xy, to_xy);

   490

}

   491

   492

static void AiFindSubsidyPassengerRoute(FoundRoute *fr)

   493

{

   494

	uint i;

   495

	Subsidy *s;

   496

	Town *from,*to;

   497

   498

	// initially error

   499

	fr->distance = -1;

   500

   501

	// Randomize subsidy index..

   502

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

   503

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

   504

   505

	s = &_subsidies[i];

   506