1

#include "stdafx.h"

     2

#include "ttd.h"

     3

#include "debug.h"

     4

#include "map.h"

     5

#include "ai.h"

     6

#include "vehicle.h"

     7

     8

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

     9

	// 0 = vert

    10

	// 1 = horz

    11

	// 2 = dig up-left

    12

	// 3 = dig down-right

    13

	// 4 = dig down-left

    14

	// 5 = dig up-right

    15

    16

	int x1, x2, x3;

    17

	int y1, y2, y3;

    18

    19

	x1 = TileX(tile_a);

    20

	x2 = TileX(tile_b);

    21

	x3 = TileX(tile_c);

    22

    23

	y1 = TileY(tile_a);

    24

	y2 = TileY(tile_b);

    25

	y3 = TileY(tile_c);

    26

    27

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

    28

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

    29

	if (y2 > y1) {

    30

		if (x2 > x3) return 2;

    31

		else return 4;

    32

	}

    33

	if (x2 > x1) {

    34

		if (y2 > y3) return 2;

    35

		else return 5;

    36

	}

    37

	if (y1 > y2) {

    38

		if (x2 > x3) return 5;

    39

		else return 3;

    40

	}

    41

	if (x1 > x2) {

    42

		if (y2 > y3) return 4;

    43

		else return 3;

    44

	}

    45

    46

	return 0;

    47

}

    48

    49

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

    50

	int x1, x2, x3;

    51

	int y1, y2, y3;

    52

	int r;

    53

    54

	x1 = TileX(tile_a);

    55

	x2 = TileX(tile_b);

    56

	x3 = TileX(tile_c);

    57

    58

	y1 = TileY(tile_a);

    59

	y2 = TileY(tile_b);

    60

	y3 = TileY(tile_c);

    61

    62

	r = 0;

    63

    64

	if (x1 < x2) r += 8;

    65

	if (y1 < y2) r += 1;

    66

	if (x1 > x2) r += 2;

    67

	if (y1 > y2) r += 4;

    68

    69

	if (x2 < x3) r += 2;

    70

	if (y2 < y3) r += 4;

    71

	if (x2 > x3) r += 8;

    72

	if (y2 > y3) r += 1;

    73

    74

	return r;

    75

}

    76

    77

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

    78

int AiNew_GetDirection(uint tile_a, uint tile_b) {

    79

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

    80

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

    81

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

    82

	return 0;

    83

}

    84

    85

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

    86

//  and returns his flag

    87

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

    88

	int i;

    89

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

    90

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

    91

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

    92

		}

    93

	}

    94

    95

	// Not found :(

    96

	return 0;

    97

}

    98

    99

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

   100

	int i, new_id = -1;

   101

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

   102

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

   103

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

   104

			return true;

   105

		}

   106

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

   107

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

   108

			new_id = i;

   109

	}

   110

   111

	// Out of special_vehicle spots :s

   112

	if (new_id == -1) {

   113

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

   114

		return false;

   115

	}

   116

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

   117

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

   118

	return true;

   119

}