(svn r15027) -Merge: tomatos and bananas left to be, here is NoAI for all to see.
NoAI is an API (a framework) to build your own AIs in. See: http://wiki.openttd.org/wiki/index.php/AI:Main_Page With many thanks to: - glx and Rubidium for their syncing, feedback and hard work - Yexo for his feedback, patches, and AIs which tested the system very deep - Morloth for his feedback and patches - TJIP for hosting a challenge which kept NoAI on track - All AI authors for testing our AI API, and all other people who helped in one way or another -Remove: all old AIs and their cheats/hacks
This commit is contained in:
truebrain
295
src/ai/api/ai_engine.cpp
Normal file
295
src/ai/api/ai_engine.cpp
Normal file
@@ -0,0 +1,295 @@
|
||||
/* $Id$ */
|
||||
|
||||
/** @file ai_engine.cpp Implementation of AIEngine. */
|
||||
|
||||
#include "ai_engine.hpp"
|
||||
#include "ai_cargo.hpp"
|
||||
#include "ai_rail.hpp"
|
||||
#include "../../openttd.h"
|
||||
#include "../../company_func.h"
|
||||
#include "../../strings_func.h"
|
||||
#include "../../roadveh.h"
|
||||
#include "../../train.h"
|
||||
#include "../../ship.h"
|
||||
#include "../../aircraft.h"
|
||||
#include "../../vehicle_func.h"
|
||||
#include "../../core/alloc_func.hpp"
|
||||
#include "../../economy_func.h"
|
||||
#include "../../core/bitmath_func.hpp"
|
||||
#include "../../settings_type.h"
|
||||
#include "../../articulated_vehicles.h"
|
||||
#include "table/strings.h"
|
||||
|
||||
/* static */ bool AIEngine::IsValidEngine(EngineID engine_id)
|
||||
{
|
||||
return ::IsEngineIndex(engine_id) && HasBit(::GetEngine(engine_id)->company_avail, _current_company);
|
||||
}
|
||||
|
||||
/* static */ const char *AIEngine::GetName(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return NULL;
|
||||
|
||||
static const int len = 64;
|
||||
char *engine_name = MallocT<char>(len);
|
||||
|
||||
::SetDParam(0, engine_id);
|
||||
::GetString(engine_name, STR_ENGINE_NAME, &engine_name[len - 1]);
|
||||
return engine_name;
|
||||
}
|
||||
|
||||
/* static */ CargoID AIEngine::GetCargoType(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return CT_INVALID;
|
||||
|
||||
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 */ bool AIEngine::CanRefitCargo(EngineID engine_id, CargoID cargo_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return false;
|
||||
if (!AICargo::IsValidCargo(cargo_id)) return false;
|
||||
|
||||
if (GetCargoType(engine_id) == cargo_id) return true;
|
||||
if (cargo_id == CT_MAIL && ::GetEngine(engine_id)->type == VEH_AIRCRAFT) return true;
|
||||
if (::GetEngine(engine_id)->type == VEH_SHIP && !ShipVehInfo(engine_id)->refittable) return false;
|
||||
return ::CanRefitTo(engine_id, cargo_id);
|
||||
}
|
||||
|
||||
/* static */ bool AIEngine::CanPullCargo(EngineID engine_id, CargoID cargo_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return false;
|
||||
if (GetVehicleType(engine_id) != AIVehicle::VEHICLE_RAIL) return false;
|
||||
if (!AICargo::IsValidCargo(cargo_id)) return false;
|
||||
|
||||
return (::RailVehInfo(engine_id)->ai_passenger_only != 1) || AICargo::HasCargoClass(cargo_id, AICargo::CC_PASSENGERS);
|
||||
}
|
||||
|
||||
|
||||
/* static */ int32 AIEngine::GetCapacity(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return -1;
|
||||
|
||||
switch (::GetEngine(engine_id)->type) {
|
||||
case VEH_ROAD:
|
||||
case VEH_TRAIN: {
|
||||
uint16 *capacities = GetCapacityOfArticulatedParts(engine_id, ::GetEngine(engine_id)->type);
|
||||
for (CargoID c = 0; c < NUM_CARGO; c++) {
|
||||
if (capacities[c] == 0) continue;
|
||||
return capacities[c];
|
||||
}
|
||||
return -1;
|
||||
} 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 */ int32 AIEngine::GetReliability(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return -1;
|
||||
|
||||
return (::GetEngine(engine_id)->reliability * 100 >> 16);
|
||||
}
|
||||
|
||||
/* static */ int32 AIEngine::GetMaxSpeed(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return -1;
|
||||
|
||||
switch (::GetEngine(engine_id)->type) {
|
||||
case VEH_ROAD: {
|
||||
const RoadVehicleInfo *vi = ::RoadVehInfo(engine_id);
|
||||
/* Internal speeds are km/h * 2 */
|
||||
return vi->max_speed / 2;
|
||||
} break;
|
||||
|
||||
case VEH_TRAIN: {
|
||||
const RailVehicleInfo *vi = ::RailVehInfo(engine_id);
|
||||
return vi->max_speed;
|
||||
} break;
|
||||
|
||||
case VEH_SHIP: {
|
||||
const ShipVehicleInfo *vi = ::ShipVehInfo(engine_id);
|
||||
/* Internal speeds are km/h * 2 */
|
||||
return vi->max_speed / 2;
|
||||
} break;
|
||||
|
||||
case VEH_AIRCRAFT: {
|
||||
const AircraftVehicleInfo *vi = ::AircraftVehInfo(engine_id);
|
||||
return vi->max_speed / _settings_game.vehicle.plane_speed;
|
||||
} break;
|
||||
|
||||
default: NOT_REACHED();
|
||||
}
|
||||
}
|
||||
|
||||
/* static */ Money AIEngine::GetPrice(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return -1;
|
||||
|
||||
switch (::GetEngine(engine_id)->type) {
|
||||
case VEH_ROAD: {
|
||||
const RoadVehicleInfo *vi = ::RoadVehInfo(engine_id);
|
||||
return (_price.roadveh_base >> 3) * vi->cost_factor >> 5;
|
||||
} break;
|
||||
|
||||
case VEH_TRAIN: {
|
||||
const RailVehicleInfo *vi = ::RailVehInfo(engine_id);
|
||||
return (_price.build_railvehicle >> 3) * vi->cost_factor >> 5;
|
||||
} break;
|
||||
|
||||
case VEH_SHIP: {
|
||||
const ShipVehicleInfo *vi = ::ShipVehInfo(engine_id);
|
||||
return (_price.ship_base >> 3) * vi->cost_factor >> 5;
|
||||
} break;
|
||||
|
||||
case VEH_AIRCRAFT: {
|
||||
const AircraftVehicleInfo *vi = ::AircraftVehInfo(engine_id);
|
||||
return (_price.aircraft_base >> 3) * vi->cost_factor >> 5;
|
||||
} break;
|
||||
|
||||
default: NOT_REACHED();
|
||||
}
|
||||
}
|
||||
|
||||
/* static */ int32 AIEngine::GetMaxAge(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return -1;
|
||||
|
||||
return ::GetEngine(engine_id)->lifelength * 366;
|
||||
}
|
||||
|
||||
/* static */ Money AIEngine::GetRunningCost(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return -1;
|
||||
|
||||
/* We need to create an instance in order to obtain GetRunningCost.
|
||||
* This means we temporary allocate a vehicle in the pool, but
|
||||
* there is no other way.. */
|
||||
Vehicle *vehicle;
|
||||
switch (::GetEngine(engine_id)->type) {
|
||||
case VEH_ROAD: {
|
||||
vehicle = new RoadVehicle();
|
||||
} break;
|
||||
|
||||
case VEH_TRAIN: {
|
||||
vehicle = new Train();
|
||||
} break;
|
||||
|
||||
case VEH_SHIP: {
|
||||
vehicle = new Ship();
|
||||
} break;
|
||||
|
||||
case VEH_AIRCRAFT: {
|
||||
vehicle = new Aircraft();
|
||||
} break;
|
||||
|
||||
default: NOT_REACHED();
|
||||
}
|
||||
|
||||
vehicle->engine_type = engine_id;
|
||||
Money runningCost = vehicle->GetRunningCost();
|
||||
delete vehicle;
|
||||
return runningCost >> 8;
|
||||
}
|
||||
|
||||
/* static */ AIVehicle::VehicleType AIEngine::GetVehicleType(EngineID engine_id)
|
||||
{
|
||||
if (!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();
|
||||
}
|
||||
}
|
||||
|
||||
/* static */ bool AIEngine::IsWagon(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return false;
|
||||
if (GetVehicleType(engine_id) != AIVehicle::VEHICLE_RAIL) return false;
|
||||
|
||||
return ::RailVehInfo(engine_id)->power == 0;
|
||||
}
|
||||
|
||||
/* static */ bool AIEngine::CanRunOnRail(EngineID engine_id, AIRail::RailType track_rail_type)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return false;
|
||||
if (GetVehicleType(engine_id) != AIVehicle::VEHICLE_RAIL) return false;
|
||||
if (!AIRail::IsRailTypeAvailable(track_rail_type)) return false;
|
||||
|
||||
return ::IsCompatibleRail((::RailType)::RailVehInfo(engine_id)->railtype, (::RailType)track_rail_type);
|
||||
}
|
||||
|
||||
/* static */ bool AIEngine::HasPowerOnRail(EngineID engine_id, AIRail::RailType track_rail_type)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return false;
|
||||
if (GetVehicleType(engine_id) != AIVehicle::VEHICLE_RAIL) return false;
|
||||
if (!AIRail::IsRailTypeAvailable(track_rail_type)) return false;
|
||||
|
||||
return ::HasPowerOnRail((::RailType)::RailVehInfo(engine_id)->railtype, (::RailType)track_rail_type);
|
||||
}
|
||||
|
||||
/* static */ AIRoad::RoadType AIEngine::GetRoadType(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return AIRoad::ROADTYPE_INVALID;
|
||||
if (GetVehicleType(engine_id) != AIVehicle::VEHICLE_ROAD) return AIRoad::ROADTYPE_INVALID;
|
||||
|
||||
return HasBit(::EngInfo(engine_id)->misc_flags, EF_ROAD_TRAM) ? AIRoad::ROADTYPE_TRAM : AIRoad::ROADTYPE_ROAD;
|
||||
}
|
||||
|
||||
/* static */ AIRail::RailType AIEngine::GetRailType(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return AIRail::RAILTYPE_INVALID;
|
||||
if (GetVehicleType(engine_id) != AIVehicle::VEHICLE_RAIL) return AIRail::RAILTYPE_INVALID;
|
||||
|
||||
return (AIRail::RailType)(uint)::RailVehInfo(engine_id)->railtype;
|
||||
}
|
||||
|
||||
/* static */ bool AIEngine::IsArticulated(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return false;
|
||||
if (GetVehicleType(engine_id) != AIVehicle::VEHICLE_ROAD && GetVehicleType(engine_id) != AIVehicle::VEHICLE_RAIL) return false;
|
||||
|
||||
return CountArticulatedParts(engine_id, true) != 0;
|
||||
}
|
||||
|
||||
/* static */ AIAirport::PlaneType AIEngine::GetPlaneType(EngineID engine_id)
|
||||
{
|
||||
if (!IsValidEngine(engine_id)) return AIAirport::PT_INVALID;
|
||||
if (GetVehicleType(engine_id) != AIVehicle::VEHICLE_AIR) return AIAirport::PT_INVALID;
|
||||
|
||||
return (AIAirport::PlaneType)::AircraftVehInfo(engine_id)->subtype;
|
||||
}
|
||||
Reference in New Issue
Block a user