truelight@2381: /* $Id$ */
truelight@2381: 
rubidium@9111: /** @file shared.cpp Shared functions for the trolly AI. */
rubidium@9111: 
truelight@2381: #include "../../stdafx.h"
truelight@2381: #include "../../openttd.h"
truelight@2381: #include "../../debug.h"
rubidium@8139: #include "../../map_func.h"
rubidium@8144: #include "../../vehicle_base.h"
rubidium@10208: #include "../../company_base.h"
truelight@2381: #include "trolly.h"
truelight@2381: 
truelight@2381: int AiNew_GetRailDirection(TileIndex tile_a, TileIndex tile_b, TileIndex tile_c)
truelight@2381: {
truelight@2381: 	// 0 = vert
truelight@2381: 	// 1 = horz
truelight@2381: 	// 2 = dig up-left
truelight@2381: 	// 3 = dig down-right
truelight@2381: 	// 4 = dig down-left
truelight@2381: 	// 5 = dig up-right
truelight@2381: 
tron@4077: 	uint x1 = TileX(tile_a);
tron@4077: 	uint x2 = TileX(tile_b);
tron@4077: 	uint x3 = TileX(tile_c);
truelight@2381: 
tron@4077: 	uint y1 = TileY(tile_a);
tron@4077: 	uint y2 = TileY(tile_b);
tron@4077: 	uint y3 = TileY(tile_c);
truelight@2381: 
truelight@2381: 	if (y1 == y2 && y2 == y3) return 0;
truelight@2381: 	if (x1 == x2 && x2 == x3) return 1;
tron@4077: 	if (y2 > y1) return x2 > x3 ? 2 : 4;
tron@4077: 	if (x2 > x1) return y2 > y3 ? 2 : 5;
tron@4077: 	if (y1 > y2) return x2 > x3 ? 5 : 3;
tron@4077: 	if (x1 > x2) return y2 > y3 ? 4 : 3;
truelight@2381: 
truelight@2381: 	return 0;
truelight@2381: }
truelight@2381: 
truelight@2381: int AiNew_GetRoadDirection(TileIndex tile_a, TileIndex tile_b, TileIndex tile_c)
truelight@2381: {
truelight@2381: 	int x1, x2, x3;
truelight@2381: 	int y1, y2, y3;
truelight@2381: 	int r;
truelight@2381: 
truelight@2381: 	x1 = TileX(tile_a);
truelight@2381: 	x2 = TileX(tile_b);
truelight@2381: 	x3 = TileX(tile_c);
truelight@2381: 
truelight@2381: 	y1 = TileY(tile_a);
truelight@2381: 	y2 = TileY(tile_b);
truelight@2381: 	y3 = TileY(tile_c);
truelight@2381: 
truelight@2381: 	r = 0;
truelight@2381: 
truelight@2381: 	if (x1 < x2) r += 8;
truelight@2381: 	if (y1 < y2) r += 1;
truelight@2381: 	if (x1 > x2) r += 2;
truelight@2381: 	if (y1 > y2) r += 4;
truelight@2381: 
truelight@2381: 	if (x2 < x3) r += 2;
truelight@2381: 	if (y2 < y3) r += 4;
truelight@2381: 	if (x2 > x3) r += 8;
truelight@2381: 	if (y2 > y3) r += 1;
truelight@2381: 
truelight@2381: 	return r;
truelight@2381: }
truelight@2381: 
truelight@2381: // Get's the direction between 2 tiles seen from tile_a
tron@3644: DiagDirection AiNew_GetDirection(TileIndex tile_a, TileIndex tile_b)
truelight@2381: {
tron@3644: 	if (TileY(tile_a) < TileY(tile_b)) return DIAGDIR_SE;
tron@3644: 	if (TileY(tile_a) > TileY(tile_b)) return DIAGDIR_NW;
tron@3644: 	if (TileX(tile_a) < TileX(tile_b)) return DIAGDIR_SW;
tron@3644: 	return DIAGDIR_NE;
truelight@2381: }
truelight@2381: 
tron@4077: 
truelight@2381: // This functions looks up if this vehicle is special for this AI
truelight@2381: //  and returns his flag
rubidium@10207: uint AiNew_GetSpecialVehicleFlag(Company *c, Vehicle *v)
tron@4077: {
tron@4077: 	uint i;
tron@4077: 
tron@4077: 	for (i = 0; i < AI_MAX_SPECIAL_VEHICLES; i++) {
rubidium@10207: 		if (_companies_ainew[c->index].special_vehicles[i].veh_id == v->index) {
rubidium@10207: 			return _companies_ainew[c->index].special_vehicles[i].flag;
truelight@2381: 		}
truelight@2381: 	}
truelight@2381: 
truelight@2381: 	// Not found :(
truelight@2381: 	return 0;
truelight@2381: }
truelight@2381: 
tron@4077: 
rubidium@10207: bool AiNew_SetSpecialVehicleFlag(Company *c, Vehicle *v, uint flag)
tron@4077: {
tron@4077: 	int new_id = -1;
tron@4077: 	uint i;
tron@4077: 
tron@4077: 	for (i = 0; i < AI_MAX_SPECIAL_VEHICLES; i++) {
rubidium@10207: 		if (_companies_ainew[c->index].special_vehicles[i].veh_id == v->index) {
rubidium@10207: 			_companies_ainew[c->index].special_vehicles[i].flag |= flag;
truelight@2381: 			return true;
truelight@2381: 		}
tron@4077: 		if (new_id == -1 &&
rubidium@10207: 				_companies_ainew[c->index].special_vehicles[i].veh_id == 0 &&
rubidium@10207: 				_companies_ainew[c->index].special_vehicles[i].flag == 0) {
truelight@2381: 			new_id = i;
tron@4077: 		}
truelight@2381: 	}
truelight@2381: 
truelight@2381: 	// Out of special_vehicle spots :s
truelight@2381: 	if (new_id == -1) {
Darkvater@5380: 		DEBUG(ai, 1, "special_vehicles list is too small");
truelight@2381: 		return false;
truelight@2381: 	}
rubidium@10207: 	_companies_ainew[c->index].special_vehicles[new_id].veh_id = v->index;
rubidium@10207: 	_companies_ainew[c->index].special_vehicles[new_id].flag = flag;
truelight@2381: 	return true;
truelight@2381: }