1

/* $Id$ */

     2

     3

/** @file shared.cpp Shared functions for the trolly AI. */

     4

     5

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

     6

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

     7

#include "../../debug.h"

     8

#include "../../map_func.h"

     9

#include "../../vehicle_base.h"

    10

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

    11

#include "trolly.h"

    12

    13

int AiNew_GetRailDirection(TileIndex tile_a, TileIndex tile_b, TileIndex tile_c)

    14

{

    15

	// 0 = vert

    16

	// 1 = horz

    17

	// 2 = dig up-left

    18

	// 3 = dig down-right

    19

	// 4 = dig down-left

    20

	// 5 = dig up-right

    21

    22

	uint x1 = TileX(tile_a);

    23

	uint x2 = TileX(tile_b);

    24

	uint x3 = TileX(tile_c);

    25

    26

	uint y1 = TileY(tile_a);

    27

	uint y2 = TileY(tile_b);

    28

	uint y3 = TileY(tile_c);

    29

    30

	if (y1 == y2 && y2 == y3) return 0;

    31

	if (x1 == x2 && x2 == x3) return 1;

    32

	if (y2 > y1) return x2 > x3 ? 2 : 4;

    33

	if (x2 > x1) return y2 > y3 ? 2 : 5;

    34

	if (y1 > y2) return x2 > x3 ? 5 : 3;

    35

	if (x1 > x2) return y2 > y3 ? 4 : 3;

    36

    37

	return 0;

    38

}

    39

    40

int AiNew_GetRoadDirection(TileIndex tile_a, TileIndex tile_b, TileIndex tile_c)

    41

{

    42

	int x1, x2, x3;

    43

	int y1, y2, y3;

    44

	int r;

    45

    46

	x1 = TileX(tile_a);

    47

	x2 = TileX(tile_b);

    48

	x3 = TileX(tile_c);

    49

    50

	y1 = TileY(tile_a);

    51

	y2 = TileY(tile_b);

    52

	y3 = TileY(tile_c);

    53

    54

	r = 0;

    55

    56

	if (x1 < x2) r += 8;

    57

	if (y1 < y2) r += 1;

    58

	if (x1 > x2) r += 2;

    59

	if (y1 > y2) r += 4;

    60

    61

	if (x2 < x3) r += 2;

    62

	if (y2 < y3) r += 4;

    63

	if (x2 > x3) r += 8;

    64

	if (y2 > y3) r += 1;

    65

    66

	return r;

    67

}

    68

    69

// Get's the direction between 2 tiles seen from tile_a

    70

DiagDirection AiNew_GetDirection(TileIndex tile_a, TileIndex tile_b)

    71

{

    72

	if (TileY(tile_a) < TileY(tile_b)) return DIAGDIR_SE;

    73

	if (TileY(tile_a) > TileY(tile_b)) return DIAGDIR_NW;

    74

	if (TileX(tile_a) < TileX(tile_b)) return DIAGDIR_SW;

    75

	return DIAGDIR_NE;

    76

}

    77

    78

    79

// This functions looks up if this vehicle is special for this AI

    80

//  and returns his flag

    81

uint AiNew_GetSpecialVehicleFlag(Company *c, Vehicle *v)

    82

{

    83

	uint i;

    84

    85

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

    86

		if (_companies_ainew[c->index].special_vehicles[i].veh_id == v->index) {

    87

			return _companies_ainew[c->index].special_vehicles[i].flag;

    88

		}

    89

	}

    90

    91

	// Not found :(

    92

	return 0;

    93

}

    94

    95

    96

bool AiNew_SetSpecialVehicleFlag(Company *c, Vehicle *v, uint flag)

    97

{

    98

	int new_id = -1;

    99

	uint i;

   100

   101

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

   102

		if (_companies_ainew[c->index].special_vehicles[i].veh_id == v->index) {

   103

			_companies_ainew[c->index].special_vehicles[i].flag |= flag;

   104

			return true;

   105

		}

   106

		if (new_id == -1 &&

   107

				_companies_ainew[c->index].special_vehicles[i].veh_id == 0 &&

   108

				_companies_ainew[c->index].special_vehicles[i].flag == 0) {

   109

			new_id = i;

   110

		}

   111

	}

   112

   113

	// Out of special_vehicle spots :s

   114

	if (new_id == -1) {

   115

		DEBUG(ai, 1, "special_vehicles list is too small");

   116

		return false;

   117

	}

   118

	_companies_ainew[c->index].special_vehicles[new_id].veh_id = v->index;

   119

	_companies_ainew[c->index].special_vehicles[new_id].flag = flag;

   120

	return true;

   121

}