/* $Id$ */

/** @file ai_vehicle.cpp handles the functions of the AIVehicle class */

#include "ai_vehicle.hpp"
#include "ai_cargo.hpp"
#include "../../command.h"
#include "../../depot.h"
#include "../../engine.h"
#include "../../player.h"
#include "../../vehicle.h"

/* static */ bool AIVehicle::IsValidEngine(EngineID engine_id)
{
	return ::IsEngineIndex(engine_id); // TODO: implement 'can I build this engine check'
}

/* static */ bool AIVehicle::IsValidVehicle(VehicleID vehicle_id)
{
	return ::IsValidVehicleID(vehicle_id) && ::GetVehicle(vehicle_id)->owner == _current_player;
}

EngineID AIVehicle::FindBestRoadVehicle(CargoID cargo, uint8 min_reliability)
{
	if (!AICargo::IsValidCargo(cargo) || min_reliability > 100) return INVALID_ENGINE;

	EngineID best_engine = INVALID_ENGINE;
	EngineID engine_id;

	FOR_ALL_ENGINEIDS_OF_TYPE(engine_id, VEH_ROAD) {
		/* Is the vehicle available for this player */
		if (IsEngineBuildable(engine_id, VEH_ROAD, _current_player) &&
				GetEngine(engine_id)->reliability * 100 >= min_reliability << 16 &&
				(RoadVehInfo(engine_id)->cargo_type == cargo || CanRefitTo(engine_id, cargo))) {
			best_engine = engine_id;
		}
	}

	return best_engine;
}

VehicleID AIVehicle::BuildVehicle(TileIndex depot, EngineID engine_id)
{
	if (!this->IsValidEngine(engine_id)) return false;

	/* Reset the internal NewVehicleID in case we are in TestMode */
	AIObject::SetNewVehicleID(0);

	bool ret;
	switch (::GetEngine(engine_id)->type) {
		case VEH_ROAD: ret = this->DoCommand(depot, engine_id, 0, CMD_BUILD_ROAD_VEH); break;
		default: NOT_REACHED(); return INVALID_VEHICLE; // TODO: implement trains, ships and aircraft
	}

	return ret ? AIObject::GetNewVehicleID() : INVALID_VEHICLE;
}

VehicleID AIVehicle::CloneVehicle(TileIndex depot, VehicleID vehicle_id, bool share_orders)
{
	if (!this->IsValidVehicle(vehicle_id)) return false;

	/* Reset the internal NewVehicleID in case we are in TestMode */
	AIObject::SetNewVehicleID(0);

	bool ret;
	switch (::GetVehicle(vehicle_id)->type) {
		case VEH_ROAD: ret = this->DoCommand(depot, vehicle_id, share_orders, CMD_CLONE_VEHICLE); break;
		default: return INVALID_VEHICLE; // TODO: implement trains, ships and aircraft
	}

	return ret ? AIObject::GetNewVehicleID() : INVALID_VEHICLE;
}

bool AIVehicle::RefitVehicle(VehicleID vehicle_id, CargoID cargo)
{
	if (!this->IsValidVehicle(vehicle_id) || !AICargo::IsValidCargo(cargo)) return false;

	switch (::GetVehicle(vehicle_id)->type) {
		case VEH_ROAD: return this->DoCommand(0, vehicle_id, cargo, CMD_REFIT_ROAD_VEH);
		default: return false; // TODO: implement trains, ships and aircraft
	}
}


bool AIVehicle::SellVehicle(VehicleID vehicle_id)
{
	if (!this->IsValidVehicle(vehicle_id)) return false;

	switch (::GetVehicle(vehicle_id)->type) {
		case VEH_ROAD: return this->DoCommand(0, vehicle_id, 0, CMD_SELL_ROAD_VEH);
		default: return false; // TODO: implement trains, ships and aircraft
	}
}

bool AIVehicle::SendVehicleToDepot(VehicleID vehicle_id)
{
	if (!this->IsValidVehicle(vehicle_id)) return false;

	switch (::GetVehicle(vehicle_id)->type) {
		case VEH_ROAD: return this->DoCommand(0, vehicle_id, 0, CMD_SEND_ROADVEH_TO_DEPOT);
		default: return false; // TODO: implement trains, ships and aircraft
	}
}

bool AIVehicle::StartStopVehicle(VehicleID vehicle_id)
{
	if (!this->IsValidVehicle(vehicle_id)) return false;

	switch (::GetVehicle(vehicle_id)->type) {
		case VEH_ROAD: return this->DoCommand(0, vehicle_id, 0, CMD_START_STOP_ROADVEH);
		default: return false; // TODO: implement trains, ships and aircraft
	}
}

bool AIVehicle::SkipVehicleOrder(VehicleID vehicle_id)
{
	if (!this->IsValidVehicle(vehicle_id)) return false;

	return this->DoCommand(0, vehicle_id, 0, CMD_SKIP_ORDER);
}

TileIndex AIVehicle::GetLocation(VehicleID vehicle_id)
{
	if (!AIVehicle::IsValidVehicle(vehicle_id)) return INVALID_TILE;

	return ::GetVehicle(vehicle_id)->tile;
}

EngineID AIVehicle::GetEngineType(VehicleID vehicle_id)
{
	if (!AIVehicle::IsValidVehicle(vehicle_id)) return INVALID_ENGINE;

	return ::GetVehicle(vehicle_id)->engine_type;
}

int32 AIVehicle::GetUnitNumber(VehicleID vehicle_id)
{
	if (!AIVehicle::IsValidVehicle(vehicle_id)) return -1;

	return ::GetVehicle(vehicle_id)->unitnumber;
}

int32 AIVehicle::GetAge(VehicleID vehicle_id)
{
	if (!AIVehicle::IsValidVehicle(vehicle_id)) return -1;

	return ::GetVehicle(vehicle_id)->age;
}

int32 AIVehicle::GetMaxAge(VehicleID vehicle_id)
{
	if (!AIVehicle::IsValidVehicle(vehicle_id)) return -1;

	return ::GetVehicle(vehicle_id)->max_age;
}

int32 AIVehicle::GetAgeLeft(VehicleID vehicle_id)
{
	if (!AIVehicle::IsValidVehicle(vehicle_id)) return -1;

	return ::GetVehicle(vehicle_id)->max_age - ::GetVehicle(vehicle_id)->age;
}

int32 AIVehicle::GetProfitThisYear(VehicleID vehicle_id)
{
	if (!AIVehicle::IsValidVehicle(vehicle_id)) return -1;

	return ::GetVehicle(vehicle_id)->profit_this_year;
}

int32 AIVehicle::GetProfitLastYear(VehicleID vehicle_id)
{
	if (!AIVehicle::IsValidVehicle(vehicle_id)) return -1;

	return ::GetVehicle(vehicle_id)->profit_last_year;
}