1

#include "stdafx.h"

     2

#include "ttd.h"

     3

#include "ai.h"

     4

#include "vehicle.h"

     5

     6

int AiNew_GetRailDirection(uint tile_a, uint tile_b, uint tile_c) {

     7

	// 0 = vert

     8

	// 1 = horz

     9

	// 2 = dig up-left

    10

	// 3 = dig down-right

    11

	// 4 = dig down-left

    12

	// 5 = dig up-right

    13

    14

	int x1, x2, x3;

    15

	int y1, y2, y3;

    16

    17

	x1 = GET_TILE_X(tile_a);

    18

	x2 = GET_TILE_X(tile_b);

    19

	x3 = GET_TILE_X(tile_c);

    20

    21

	y1 = GET_TILE_Y(tile_a);

    22

	y2 = GET_TILE_Y(tile_b);

    23

	y3 = GET_TILE_Y(tile_c);

    24

    25

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

    26

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

    27

	if (y2 > y1) {

    28

		if (x2 > x3) return 2;

    29

		else return 4;

    30

	}

    31

	if (x2 > x1) {

    32

		if (y2 > y3) return 2;

    33

		else return 5;

    34

	}

    35

	if (y1 > y2) {

    36

		if (x2 > x3) return 5;

    37

		else return 3;

    38

	}

    39

	if (x1 > x2) {

    40

		if (y2 > y3) return 4;

    41

		else return 3;

    42

	}

    43

    44

	return 0;

    45

}

    46

    47

int AiNew_GetRoadDirection(uint tile_a, uint tile_b, uint tile_c) {

    48

	int x1, x2, x3;

    49

	int y1, y2, y3;

    50

	int r;

    51

    52

	x1 = GET_TILE_X(tile_a);

    53

	x2 = GET_TILE_X(tile_b);

    54

	x3 = GET_TILE_X(tile_c);

    55

    56

	y1 = GET_TILE_Y(tile_a);

    57

	y2 = GET_TILE_Y(tile_b);

    58

	y3 = GET_TILE_Y(tile_c);

    59

    60

	r = 0;

    61

    62

	if (x1 < x2) r += 8;

    63

	if (y1 < y2) r += 1;

    64

	if (x1 > x2) r += 2;

    65

	if (y1 > y2) r += 4;

    66

    67

	if (x2 < x3) r += 2;

    68

	if (y2 < y3) r += 4;

    69

	if (x2 > x3) r += 8;

    70

	if (y2 > y3) r += 1;

    71

    72

	return r;

    73

}

    74

    75

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

    76

int AiNew_GetDirection(uint tile_a, uint tile_b) {

    77

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

    78

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

    79

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

    80

	return 0;

    81

}

    82

    83

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

    84

//  and returns his flag

    85

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

    86

	int i;

    87

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

    88

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

    89

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

    90

		}

    91

	}

    92

    93

	// Not found :(

    94

	return 0;

    95

}

    96

    97

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

    98

	int i, new_id = -1;

    99

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

   100

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

   101

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

   102

			return true;

   103

		}

   104

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

   105

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

   106

			new_id = i;

   107

	}

   108

   109

	// Out of special_vehicle spots :s

   110

	if (new_id == -1) {

   111

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

   112

		return false;

   113

	}

   114

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

   115

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

   116

	return true;

   117

}