rubidium@8786: /* $Id$ */ rubidium@8786: rubidium@8786: /** @file engine.h */ rubidium@8786: rubidium@8786: #ifndef ENGINE_H rubidium@8786: #define ENGINE_H rubidium@8786: rubidium@8786: #include "engine_type.h" rubidium@8786: rubidium@8786: void SetupEngines(); rubidium@8786: void StartupEngines(); rubidium@8786: rubidium@8786: rubidium@8786: void DrawTrainEngine(int x, int y, EngineID engine, SpriteID pal); rubidium@8786: void DrawRoadVehEngine(int x, int y, EngineID engine, SpriteID pal); rubidium@8786: void DrawShipEngine(int x, int y, EngineID engine, SpriteID pal); rubidium@8786: void DrawAircraftEngine(int x, int y, EngineID engine, SpriteID pal); rubidium@8786: rubidium@8786: void LoadCustomEngineNames(); rubidium@8786: void DeleteCustomEngineNames(); rubidium@8786: rubidium@8786: bool IsEngineBuildable(EngineID engine, VehicleType type, PlayerID player); rubidium@8786: CargoID GetEngineCargoType(EngineID engine); rubidium@8786: rubidium@8786: static inline EngineID GetFirstEngineOfType(VehicleType type) rubidium@8786: { rubidium@8786: const EngineID start[] = {0, ROAD_ENGINES_INDEX, SHIP_ENGINES_INDEX, AIRCRAFT_ENGINES_INDEX}; rubidium@8786: rubidium@8786: return start[type]; rubidium@8786: } rubidium@8786: rubidium@8786: static inline EngineID GetLastEngineOfType(VehicleType type) rubidium@8786: { rubidium@8786: const EngineID end[] = { rubidium@8786: NUM_TRAIN_ENGINES, rubidium@8786: ROAD_ENGINES_INDEX + NUM_ROAD_ENGINES, rubidium@8786: SHIP_ENGINES_INDEX + NUM_SHIP_ENGINES, rubidium@8786: AIRCRAFT_ENGINES_INDEX + NUM_AIRCRAFT_ENGINES}; rubidium@8786: rubidium@8786: return end[type]; rubidium@8786: } rubidium@8786: rubidium@8786: extern Engine _engines[TOTAL_NUM_ENGINES]; rubidium@8786: #define FOR_ALL_ENGINES(e) for (e = _engines; e != endof(_engines); e++) rubidium@8786: #define FOR_ALL_ENGINEIDS_OF_TYPE(e, type) for (e = GetFirstEngineOfType(type); e != GetLastEngineOfType(type); e++) rubidium@8786: rubidium@8786: rubidium@8786: static inline Engine* GetEngine(EngineID i) rubidium@8786: { rubidium@8786: assert(i < lengthof(_engines)); rubidium@8786: return &_engines[i]; rubidium@8786: } rubidium@8786: rubidium@8786: static inline bool IsEngineIndex(uint index) rubidium@8786: { rubidium@8786: return index < TOTAL_NUM_ENGINES; rubidium@8786: } rubidium@8786: rubidium@8786: /* Access Vehicle Data */ rubidium@8786: extern const EngineInfo _orig_engine_info[TOTAL_NUM_ENGINES]; rubidium@8786: extern const RailVehicleInfo _orig_rail_vehicle_info[NUM_TRAIN_ENGINES]; rubidium@8786: extern const ShipVehicleInfo _orig_ship_vehicle_info[NUM_SHIP_ENGINES]; rubidium@8786: extern const AircraftVehicleInfo _orig_aircraft_vehicle_info[NUM_AIRCRAFT_ENGINES]; rubidium@8786: extern const RoadVehicleInfo _orig_road_vehicle_info[NUM_ROAD_ENGINES]; rubidium@8786: rubidium@8786: extern EngineInfo _engine_info[TOTAL_NUM_ENGINES]; rubidium@8786: extern RailVehicleInfo _rail_vehicle_info[NUM_TRAIN_ENGINES]; rubidium@8786: extern ShipVehicleInfo _ship_vehicle_info[NUM_SHIP_ENGINES]; rubidium@8786: extern AircraftVehicleInfo _aircraft_vehicle_info[NUM_AIRCRAFT_ENGINES]; rubidium@8786: extern RoadVehicleInfo _road_vehicle_info[NUM_ROAD_ENGINES]; rubidium@8786: rubidium@8786: static inline const EngineInfo *EngInfo(EngineID e) rubidium@8786: { rubidium@8786: assert(e < lengthof(_engine_info)); rubidium@8786: return &_engine_info[e]; rubidium@8786: } rubidium@8786: rubidium@8786: static inline const RailVehicleInfo* RailVehInfo(EngineID e) rubidium@8786: { rubidium@8786: assert(e < lengthof(_rail_vehicle_info)); rubidium@8786: return &_rail_vehicle_info[e]; rubidium@8786: } rubidium@8786: rubidium@8786: static inline const ShipVehicleInfo* ShipVehInfo(EngineID e) rubidium@8786: { rubidium@8786: assert(e >= SHIP_ENGINES_INDEX && e < SHIP_ENGINES_INDEX + lengthof(_ship_vehicle_info)); rubidium@8786: return &_ship_vehicle_info[e - SHIP_ENGINES_INDEX]; rubidium@8786: } rubidium@8786: rubidium@8786: static inline const AircraftVehicleInfo* AircraftVehInfo(EngineID e) rubidium@8786: { rubidium@8786: assert(e >= AIRCRAFT_ENGINES_INDEX && e < AIRCRAFT_ENGINES_INDEX + lengthof(_aircraft_vehicle_info)); rubidium@8786: return &_aircraft_vehicle_info[e - AIRCRAFT_ENGINES_INDEX]; rubidium@8786: } rubidium@8786: rubidium@8786: static inline const RoadVehicleInfo* RoadVehInfo(EngineID e) rubidium@8786: { rubidium@8786: assert(e >= ROAD_ENGINES_INDEX && e < ROAD_ENGINES_INDEX + lengthof(_road_vehicle_info)); rubidium@8786: return &_road_vehicle_info[e - ROAD_ENGINES_INDEX]; rubidium@8786: } rubidium@8786: rubidium@8786: /* Engine list manipulators - current implementation is only C wrapper of CBlobT class (helpers.cpp) */ rubidium@8786: void EngList_Create(EngineList *el); ///< Creates engine list rubidium@8786: void EngList_Destroy(EngineList *el); ///< Deallocate and destroy engine list rubidium@8786: uint EngList_Count(const EngineList *el); ///< Returns number of items in the engine list rubidium@8786: void EngList_Add(EngineList *el, EngineID eid); ///< Append one item at the end of engine list rubidium@8786: EngineID* EngList_Items(EngineList *el); ///< Returns engine list items as C array rubidium@8786: void EngList_RemoveAll(EngineList *el); ///< Removes all items from engine list rubidium@8786: typedef int CDECL EngList_SortTypeFunction(const void*, const void*); ///< argument type for EngList_Sort() rubidium@8786: void EngList_Sort(EngineList *el, EngList_SortTypeFunction compare); ///< qsort of the engine list rubidium@8786: void EngList_SortPartial(EngineList *el, EngList_SortTypeFunction compare, uint begin, uint num_items); ///< qsort of specified portion of the engine list rubidium@8786: rubidium@8786: #endif /* ENGINE_H */