/* $Id$ */

/** @file ai_engine.cpp handles the functions of the AIEngine class */

#include "ai_engine.hpp"
#include "ai_cargo.hpp"
#include "../../engine.h"
#include "../../player.h"
#include "../../strings_func.h"
#include "../../core/alloc_func.hpp"
#include "../../economy_func.h"

/* static */ bool AIEngine::IsValidEngine(EngineID engine_id)
{
	return ::IsEngineIndex(engine_id) && HasBit(GetEngine(engine_id)->player_avail, _current_player);
}

/* static */ char *AIEngine::GetName(EngineID engine_id)
{
	if (!AIEngine::IsValidEngine(engine_id)) return NULL;

	static const int len = 64;
	char *engine_name = MallocT<char>(len);

	GetString(engine_name, _engine_name_strings[engine_id], &engine_name[len - 1]);
	return engine_name;
}

/* static */ CargoID AIEngine::GetCargoType(EngineID engine_id)
{
	if (!AIEngine::IsValidEngine(engine_id)) return 0;

	switch (::GetEngine(engine_id)->type) {
		case VEH_ROAD: {
			const RoadVehicleInfo *vi = RoadVehInfo(engine_id);
			return vi->cargo_type;
		} break;

		case VEH_TRAIN: {
			const RailVehicleInfo *vi = RailVehInfo(engine_id);
			return vi->cargo_type;
		} break;

		case VEH_SHIP: {
			const ShipVehicleInfo *vi = ShipVehInfo(engine_id);
			return vi->cargo_type;
		} break;

		case VEH_AIRCRAFT: {
			return CT_PASSENGERS;
		} break;

		default: NOT_REACHED();
	}
}

/* static */ uint32 AIEngine::GetCapacity(EngineID engine_id)
{
	if (!AIEngine::IsValidEngine(engine_id)) return 0;

	switch (::GetEngine(engine_id)->type) {
		case VEH_ROAD: {
			const RoadVehicleInfo *vi = RoadVehInfo(engine_id);
			return vi->capacity;
		} break;

		case VEH_TRAIN: {
			const RailVehicleInfo *vi = RailVehInfo(engine_id);
			return vi->capacity;
		} break;

		case VEH_SHIP: {
			const ShipVehicleInfo *vi = ShipVehInfo(engine_id);
			return vi->capacity;
		} break;

		case VEH_AIRCRAFT: {
			const AircraftVehicleInfo *vi = AircraftVehInfo(engine_id);
			return vi->passenger_capacity;
		} break;

		default: NOT_REACHED();
	}
}

/* static */ uint32 AIEngine::GetReliability(EngineID engine_id)
{
	if (!AIEngine::IsValidEngine(engine_id)) return 0;

	return (::GetEngine(engine_id)->reliability * 100 >> 16);
}

/* static */ uint32 AIEngine::GetMaxSpeed(EngineID engine_id)
{
	if (!AIEngine::IsValidEngine(engine_id)) return 0;

	switch (::GetEngine(engine_id)->type) {
		case VEH_ROAD: {
			const RoadVehicleInfo *vi = RoadVehInfo(engine_id);
			return vi->max_speed;
		} break;

		case VEH_TRAIN: {
			const RailVehicleInfo *vi = RailVehInfo(engine_id);
			return vi->max_speed;
		} break;

		case VEH_SHIP: {
			const ShipVehicleInfo *vi = ShipVehInfo(engine_id);
			return vi->max_speed;
		} break;

		case VEH_AIRCRAFT: {
			const AircraftVehicleInfo *vi = AircraftVehInfo(engine_id);
			return vi->max_speed;
		} break;

		default: NOT_REACHED();
	}
}

/* static */ uint32 AIEngine::GetPrice(EngineID engine_id)
{
	if (!AIEngine::IsValidEngine(engine_id)) return 0;

	switch (::GetEngine(engine_id)->type) {
		case VEH_ROAD: {
			const RoadVehicleInfo *vi = RoadVehInfo(engine_id);
			return (_price.roadveh_base >> 3) * vi->base_cost >> 5;
		} break;

		case VEH_TRAIN: {
			const RailVehicleInfo *vi = RailVehInfo(engine_id);
			return (_price.build_railvehicle >> 3) * vi->base_cost >> 5;
		} break;

		case VEH_SHIP: {
			const ShipVehicleInfo *vi = ShipVehInfo(engine_id);
			return (_price.ship_base >> 3) * vi->base_cost >> 5;
		} break;

		case VEH_AIRCRAFT: {
			const AircraftVehicleInfo *vi = AircraftVehInfo(engine_id);
			return (_price.aircraft_base >> 3) * vi->base_cost >> 5;
		} break;

		default: NOT_REACHED();
	}
}

/* static */ AIVehicle::VehicleType AIEngine::GetVehicleType(EngineID engine_id)
{
	if (!AIEngine::IsValidEngine(engine_id)) return AIVehicle::VEHICLE_INVALID;

	switch (::GetEngine(engine_id)->type) {
		case VEH_ROAD:     return AIVehicle::VEHICLE_ROAD;
		case VEH_TRAIN:    return AIVehicle::VEHICLE_RAIL;
		case VEH_SHIP:     return AIVehicle::VEHICLE_WATER;
		case VEH_AIRCRAFT: return AIVehicle::VEHICLE_AIR;
		default: NOT_REACHED();
	}
}