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 "default.h"

    30

    31

// remove some day perhaps?

    32

static uint _ai_service_interval;

    33

    34

typedef void AiStateAction(Player *p);

    35

    36

enum {

    37

	AIS_0                            =  0,

    38

	AIS_1                            =  1,

    39

	AIS_VEH_LOOP                     =  2,

    40

	AIS_VEH_CHECK_REPLACE_VEHICLE    =  3,

    41

	AIS_VEH_DO_REPLACE_VEHICLE       =  4,

    42

	AIS_WANT_NEW_ROUTE               =  5,

    43

	AIS_BUILD_DEFAULT_RAIL_BLOCKS    =  6,

    44

	AIS_BUILD_RAIL                   =  7,

    45

	AIS_BUILD_RAIL_VEH               =  8,

    46

	AIS_DELETE_RAIL_BLOCKS           =  9,

    47

	AIS_BUILD_DEFAULT_ROAD_BLOCKS    = 10,

    48

	AIS_BUILD_ROAD                   = 11,

    49

	AIS_BUILD_ROAD_VEHICLES          = 12,

    50

	AIS_DELETE_ROAD_BLOCKS           = 13,

    51

	AIS_AIRPORT_STUFF                = 14,

    52

	AIS_BUILD_DEFAULT_AIRPORT_BLOCKS = 15,

    53

	AIS_BUILD_AIRCRAFT_VEHICLES      = 16,

    54

	AIS_CHECK_SHIP_STUFF             = 17,

    55

	AIS_BUILD_DEFAULT_SHIP_BLOCKS    = 18,

    56

	AIS_DO_SHIP_STUFF                = 19,

    57

	AIS_SELL_VEHICLE                 = 20,

    58

	AIS_REMOVE_STATION               = 21,

    59

	AIS_REMOVE_TRACK                 = 22,

    60

	AIS_REMOVE_SINGLE_RAIL_TILE      = 23

    61

};

    62

    63

    64

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

    65

    66

static byte GetRailTrackStatus(TileIndex tile)

    67

{

    68

	uint32 r = GetTileTrackStatus(tile, TRANSPORT_RAIL);

    69

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

    70

}

    71

    72

    73

static void AiCase0(Player *p)

    74

{

    75

	p->ai.state = AIS_REMOVE_TRACK;

    76

	p->ai.state_counter = 0;

    77

}

    78

    79

static void AiCase1(Player *p)

    80

{

    81

	p->ai.cur_veh = NULL;

    82

	p->ai.state = AIS_VEH_LOOP;

    83

}

    84

    85

static void AiStateVehLoop(Player *p)

    86

{

    87

	Vehicle *v;

    88

	uint index;

    89

    90

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

    91

    92

	FOR_ALL_VEHICLES_FROM(v, index) {

    93

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

    94

    95

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

    96

				v->type == VEH_Road ||

    97

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

    98

				v->type == VEH_Ship) {

    99

			/* replace engine? */

   100

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

   101

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

   102

				p->ai.state = AIS_VEH_CHECK_REPLACE_VEHICLE;

   103

				p->ai.cur_veh = v;

   104

				return;

   105

			}

   106

   107

			/* not profitable? */

   108

			if (v->age >= 730 &&

   109

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

   110

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

   111

				p->ai.state_counter = 0;

   112

				p->ai.state = AIS_SELL_VEHICLE;

   113

				p->ai.cur_veh = v;

   114

				return;

   115

			}

   116

   117

			/* not reliable? */

   118

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

   119

						v->age != 0 &&

   120

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

   121

					)) {

   122

				p->ai.state = AIS_VEH_CHECK_REPLACE_VEHICLE;

   123

				p->ai.cur_veh = v;

   124

				return;

   125

			}

   126

		}

   127

	}

   128

   129

	p->ai.state = AIS_WANT_NEW_ROUTE;

   130

	p->ai.state_counter = 0;

   131

}

   132

   133

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

   134

{

   135

	EngineID best_veh_index = INVALID_ENGINE;

   136

	byte best_veh_score = 0;

   137

	int32 ret;

   138

	EngineID i;

   139

   140

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

   141

		const RailVehicleInfo *rvi = RailVehInfo(i);

   142

		const Engine* e = GetEngine(i);

   143

   144

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

   145

				rvi->flags & RVI_WAGON ||

   146

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

   147

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

   148

				e->reliability < 0x8A3D) {

   149

			continue;

   150

		}

   151

   152

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

   153

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

   154

			best_veh_score = rvi->ai_rank;

   155

			best_veh_index = i;

   156

		}

   157

	}

   158

   159

	return best_veh_index;

   160

}

   161

   162

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

   163

{

   164

	EngineID best_veh_index = INVALID_ENGINE;

   165

	int32 best_veh_rating = 0;

   166

	EngineID i = ROAD_ENGINES_INDEX;

   167

	EngineID end = i + NUM_ROAD_ENGINES;

   168

   169

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

   170

		const RoadVehicleInfo *rvi = RoadVehInfo(i);

   171

		const Engine* e = GetEngine(i);

   172

		int32 rating;

   173

		int32 ret;

   174

   175

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

   176

			continue;

   177

		}

   178

   179

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

   180

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

   181

   182

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

   183

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

   184

		if (rating <= best_veh_rating) continue;

   185

   186

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

   187

		if (CmdFailed(ret)) continue;

   188

   189

		/* Add the cost of refitting */

   190

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

   191

		if (ret > money) continue;

   192

   193

		best_veh_rating = rating;

   194

		best_veh_index = i;

   195

	}

   196

   197

	return best_veh_index;

   198

}

   199

   200

static EngineID AiChooseAircraftToBuild(int32 money, byte flag)

   201

{

   202

	EngineID best_veh_index = INVALID_ENGINE;

   203

	int32 best_veh_cost = 0;

   204

	EngineID i;

   205

   206

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

   207

		const Engine* e = GetEngine(i);

   208

		int32 ret;

   209

   210

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

   211

			continue;

   212

		}

   213

   214

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

   215

   216

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

   217

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

   218

			best_veh_cost = ret;

   219

			best_veh_index = i;

   220

		}

   221

	}

   222

   223

	return best_veh_index;

   224

}

   225

   226

static int32 AiGetBasePrice(const Player* p)

   227

{

   228

	int32 base = _price.station_value;

   229

   230

	// adjust base price when more expensive vehicles are available

   231

	switch (p->ai.railtype_to_use) {

   232

		default: NOT_REACHED();

   233

		case RAILTYPE_RAIL:     break;

   234

		case RAILTYPE_ELECTRIC: break;

   235

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

   236

		case RAILTYPE_MAGLEV:   base *= 2; break;

   237

	}

   238

   239

	return base;

   240

}

   241

   242

#if 0

   243

static EngineID AiChooseShipToBuild(byte cargo, int32 money)

   244

{

   245

	// XXX: not done

   246

	return INVALID_ENGINE;

   247

}

   248

#endif

   249

   250

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

   251

{

   252

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

   253

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

   254

}

   255

   256

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

   257

{

   258

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

   259

	return AiChooseAircraftToBuild(

   260

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

   261

	);

   262

}

   263

   264

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

   265

{

   266

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

   267

	const Vehicle* u = v;

   268

	int num = 0;

   269

   270

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

   271

		u = u->next;

   272

	}

   273

   274

	// XXX: check if a wagon

   275

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

   276

}

   277

   278

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

   279

{

   280

	error("!AiChooseShipToReplaceWith");

   281

   282

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

   283

	return INVALID_ENGINE;

   284

}

   285

   286

static void AiHandleGotoDepot(Player *p, int cmd)

   287

{

   288

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

   289

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

   290

   291

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

   292

		p->ai.state = AIS_VEH_DO_REPLACE_VEHICLE;

   293

		return;

   294

	}

   295

   296

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

   297

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

   298

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

   299

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

   300

	}

   301

}

   302

   303

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

   304

{

   305

	uint i;

   306

   307

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

   308

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

   309

			break;

   310

	}

   311

}

   312

   313

static void AiHandleReplaceTrain(Player *p)

   314

{

   315

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

   316

	BackuppedOrders orderbak[1];

   317

	EngineID veh;

   318

   319

	// wait until the vehicle reaches the depot.

   320

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

   321

		AiHandleGotoDepot(p, CMD_SEND_TRAIN_TO_DEPOT);

   322

		return;

   323

	}

   324

   325

	veh = AiChooseTrainToReplaceWith(p, v);

   326

	if (veh != INVALID_ENGINE) {

   327

		TileIndex tile;

   328

   329

		BackupVehicleOrders(v, orderbak);

   330

		tile = v->tile;

   331

   332

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

   333

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

   334

			VehicleID veh = _new_vehicle_id;

   335

			AiRestoreVehicleOrders(GetVehicle(veh), orderbak);

   336

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

   337

   338

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

   339

		}

   340

	}

   341

}

   342

   343

static void AiHandleReplaceRoadVeh(Player *p)

   344

{

   345

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

   346

	BackuppedOrders orderbak[1];

   347

	EngineID veh;

   348

   349

	if (!IsRoadVehInDepotStopped(v)) {

   350

		AiHandleGotoDepot(p, CMD_SEND_ROADVEH_TO_DEPOT);

   351

		return;

   352

	}

   353

   354

	veh = AiChooseRoadVehToReplaceWith(p, v);

   355

	if (veh != INVALID_ENGINE) {

   356

		TileIndex tile;

   357

   358

		BackupVehicleOrders(v, orderbak);

   359

		tile = v->tile;

   360

   361

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

   362

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

   363

			VehicleID veh = _new_vehicle_id;

   364

   365

			AiRestoreVehicleOrders(GetVehicle(veh), orderbak);

   366

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

   367

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

   368

		}

   369

	}

   370

}

   371

   372

static void AiHandleReplaceAircraft(Player *p)

   373

{

   374

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

   375

	BackuppedOrders orderbak[1];

   376

	EngineID veh;

   377

   378

	if (!IsAircraftInHangarStopped(v)) {

   379

		AiHandleGotoDepot(p, CMD_SEND_AIRCRAFT_TO_HANGAR);

   380

		return;

   381

	}

   382

   383

	veh = AiChooseAircraftToReplaceWith(p, v);

   384

	if (veh != INVALID_ENGINE) {

   385

		TileIndex tile;

   386

   387

		BackupVehicleOrders(v, orderbak);

   388

		tile = v->tile;

   389

   390

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

   391

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

   392

			VehicleID veh = _new_vehicle_id;

   393

			AiRestoreVehicleOrders(GetVehicle(veh), orderbak);

   394

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

   395

   396

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

   397

		}

   398

	}

   399

}

   400

   401

static void AiHandleReplaceShip(Player *p)

   402

{

   403

	error("!AiHandleReplaceShip");

   404

}

   405

   406

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

   407

   408

static CheckReplaceProc* const _veh_check_replace_proc[] = {

   409

	AiChooseTrainToReplaceWith,

   410

	AiChooseRoadVehToReplaceWith,

   411

	AiChooseShipToReplaceWith,

   412

	AiChooseAircraftToReplaceWith,

   413

};

   414

   415

typedef void DoReplaceProc(Player *p);

   416

static DoReplaceProc* const _veh_do_replace_proc[] = {

   417

	AiHandleReplaceTrain,

   418

	AiHandleReplaceRoadVeh,

   419

	AiHandleReplaceShip,

   420

	AiHandleReplaceAircraft

   421

};

   422

   423

static void AiStateCheckReplaceVehicle(Player *p)

   424

{

   425

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

   426

   427

	if (!IsValidVehicle(v) ||

   428

			v->owner != _current_player ||

   429

			v->type > VEH_Ship ||

   430

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

   431

		p->ai.state = AIS_VEH_LOOP;

   432

	} else {

   433

		p->ai.state_counter = 0;

   434

		p->ai.state = AIS_VEH_DO_REPLACE_VEHICLE;

   435

	}

   436

}

   437

   438

static void AiStateDoReplaceVehicle(Player *p)

   439

{

   440

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

   441

   442

	p->ai.state = AIS_VEH_LOOP;

   443

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

   444

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

   445

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

   446

}

   447

   448

typedef struct FoundRoute {

   449

	int distance;

   450

	CargoID cargo;

   451

	void *from;

   452

	void *to;

   453

} FoundRoute;

   454

   455

static Town *AiFindRandomTown(void)

   456

{

   457

	return GetRandomTown();

   458

}

   459

   460

static Industry *AiFindRandomIndustry(void)

   461

{

   462

	return GetRandomIndustry();

   463

}

   464

   465

static void AiFindSubsidyIndustryRoute(FoundRoute *fr)

   466

{

   467

	uint i;

   468

	CargoID cargo;

   469

	const Subsidy* s;

   470

	Industry* from;

   471

	TileIndex to_xy;

   472

   473

	// initially error

   474

	fr->distance = -1;

   475

   476

	// Randomize subsidy index..

   477

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

   478

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

   479

   480

	s = &_subsidies[i];

   481

   482

	// Don't want passengers or mail

   483

	cargo = s->cargo_type;

   484

	if (cargo == CT_INVALID ||

   485

			cargo == CT_PASSENGERS ||

   486

			cargo == CT_MAIL ||

   487

			s->age > 7) {

   488

		return;

   489

	}

   490

	fr->cargo = cargo;

   491

   492

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

   493

   494

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

   495

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

   496

   497

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

   498

		fr->to = to_tow;

   499

		to_xy = to_tow->xy;

   500

	} else {

   501

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

   502

   503

		fr->to = to_ind;

   504

		to_xy = to_ind->xy;

   505