1

#include "stdafx.h"

     2

#include "ttd.h"

     3

#include "map.h"

     4

#include "ai.h"

     5

#include "vehicle.h"

     6

     7

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

     8

	// 0 = vert

     9

	// 1 = horz

    10

	// 2 = dig up-left

    11

	// 3 = dig down-right

    12

	// 4 = dig down-left

    13

	// 5 = dig up-right

    14

    15

	int x1, x2, x3;

    16

	int y1, y2, y3;

    17

    18

	x1 = TileX(tile_a);

    19

	x2 = TileX(tile_b);

    20

	x3 = TileX(tile_c);

    21

    22

	y1 = TileY(tile_a);

    23

	y2 = TileY(tile_b);

    24

	y3 = TileY(tile_c);

    25

    26

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

    27

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

    28

	if (y2 > y1) {

    29

		if (x2 > x3) return 2;

    30

		else return 4;

    31

	}

    32

	if (x2 > x1) {

    33

		if (y2 > y3) return 2;

    34

		else return 5;

    35

	}

    36

	if (y1 > y2) {

    37

		if (x2 > x3) return 5;

    38

		else return 3;

    39

	}

    40

	if (x1 > x2) {

    41

		if (y2 > y3) return 4;

    42

		else return 3;

    43

	}

    44

    45

	return 0;

    46

}

    47

    48

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

    49

	int x1, x2, x3;

    50

	int y1, y2, y3;

    51

	int r;

    52

    53

	x1 = TileX(tile_a);

    54

	x2 = TileX(tile_b);

    55

	x3 = TileX(tile_c);

    56

    57

	y1 = TileY(tile_a);

    58

	y2 = TileY(tile_b);

    59

	y3 = TileY(tile_c);

    60

    61

	r = 0;

    62

    63

	if (x1 < x2) r += 8;

    64

	if (y1 < y2) r += 1;

    65

	if (x1 > x2) r += 2;

    66

	if (y1 > y2) r += 4;

    67

    68

	if (x2 < x3) r += 2;

    69

	if (y2 < y3) r += 4;

    70

	if (x2 > x3) r += 8;

    71

	if (y2 > y3) r += 1;

    72

    73

	return r;

    74

}

    75

    76

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

    77

int AiNew_GetDirection(uint tile_a, uint tile_b) {

    78

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

    79

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

    80

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

    81

	return 0;

    82

}

    83

    84

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

    85

//  and returns his flag

    86

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

    87

	int i;

    88

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

    89

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

    90

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

    91

		}

    92

	}

    93

    94

	// Not found :(

    95

	return 0;

    96

}

    97

    98

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

    99

	int i, new_id = -1;

   100

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

   101

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

   102

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

   103

			return true;

   104

		}

   105

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

   106

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

   107

			new_id = i;

   108

	}

   109

   110

	// Out of special_vehicle spots :s

   111

	if (new_id == -1) {

   112

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

   113

		return false;

   114

	}

   115

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

   116

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

   117

	return true;

   118

}