1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "debug.h"

     6

#include "map.h"

     7

#include "ai_new.h"

     8

#include "vehicle.h"

     9

    10

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

    11

{

    12

	// 0 = vert

    13

	// 1 = horz

    14

	// 2 = dig up-left

    15

	// 3 = dig down-right

    16

	// 4 = dig down-left

    17

	// 5 = dig up-right

    18

    19

	int x1, x2, x3;

    20

	int y1, y2, y3;

    21

    22

	x1 = TileX(tile_a);

    23

	x2 = TileX(tile_b);

    24

	x3 = TileX(tile_c);

    25

    26

	y1 = TileY(tile_a);

    27

	y2 = TileY(tile_b);

    28

	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) {

    33

		if (x2 > x3) return 2;

    34

		else return 4;

    35

	}

    36

	if (x2 > x1) {

    37

		if (y2 > y3) return 2;

    38

		else return 5;

    39

	}

    40

	if (y1 > y2) {

    41

		if (x2 > x3) return 5;

    42

		else return 3;

    43

	}

    44

	if (x1 > x2) {

    45

		if (y2 > y3) return 4;

    46

		else return 3;

    47

	}

    48

    49

	return 0;

    50

}

    51

    52

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

    53

{

    54

	int x1, x2, x3;

    55

	int y1, y2, y3;

    56

	int r;

    57

    58

	x1 = TileX(tile_a);

    59

	x2 = TileX(tile_b);

    60

	x3 = TileX(tile_c);

    61

    62

	y1 = TileY(tile_a);

    63

	y2 = TileY(tile_b);

    64

	y3 = TileY(tile_c);

    65

    66

	r = 0;

    67

    68

	if (x1 < x2) r += 8;

    69

	if (y1 < y2) r += 1;

    70

	if (x1 > x2) r += 2;

    71

	if (y1 > y2) r += 4;

    72

    73

	if (x2 < x3) r += 2;

    74

	if (y2 < y3) r += 4;

    75

	if (x2 > x3) r += 8;

    76

	if (y2 > y3) r += 1;

    77

    78

	return r;

    79

}

    80

    81

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

    82

int AiNew_GetDirection(TileIndex tile_a, TileIndex tile_b)

    83

{

    84

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

    85

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

    86

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

    87

	return 0;

    88

}

    89

    90

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

    91

//  and returns his flag

    92

uint AiNew_GetSpecialVehicleFlag(Player *p, Vehicle *v) {

    93

	int i;

    94

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

    95

		if (p->ainew.special_vehicles[i].veh_id == v->index) {

    96

			return p->ainew.special_vehicles[i].flag;

    97

		}

    98

	}

    99

   100

	// Not found :(

   101

	return 0;

   102

}

   103

   104

bool AiNew_SetSpecialVehicleFlag(Player *p, Vehicle *v, uint flag) {

   105

	int i, new_id = -1;

   106

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

   107

		if (p->ainew.special_vehicles[i].veh_id == v->index) {

   108

			p->ainew.special_vehicles[i].flag |= flag;

   109

			return true;

   110

		}

   111

		if (new_id == -1 && p->ainew.special_vehicles[i].veh_id == 0 &&

   112

			p->ainew.special_vehicles[i].flag == 0)

   113

			new_id = i;

   114

	}

   115

   116

	// Out of special_vehicle spots :s

   117

	if (new_id == -1) {

   118

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

   119

		return false;

   120

	}

   121

	p->ainew.special_vehicles[new_id].veh_id = v->index;

   122

	p->ainew.special_vehicles[new_id].flag = flag;

   123

	return true;

   124

}