@@ -31,13 +31,13 @@
uint GetEngineListHeight(VehicleType type);
void DrawEngineList(VehicleType type, int x, int r, int y, const GUIEngineList *eng_list, uint16 min, uint16 max, EngineID selected_id, bool show_count, GroupID selected_group);
static int CDECL EngineNumberSorter(const EngineID *a, const EngineID *b)
{
int r = ListPositionOfEngine(*a) - ListPositionOfEngine(*b);
int r = Engine::Get(*a)->list_position - Engine::Get(*b)->list_position;
return r;
}
/**
* Rebuild the left autoreplace list if an engine is removed or added
@@ -96,13 +96,13 @@ static CargoID _last_filter_criteria[] =
* @param *a first engine to compare
* @param *b second engine to compare
* @return for descending order: returns < 0 if a < b and > 0 for a > b. Vice versa for ascending order and 0 for equal
*/
return _internal_sort_order ? -r : r;
* Determines order of engines by introduction date
@@ -1097,13 +1097,13 @@ struct BuildVehicleWindow : Window {
if (eid == this->sel_engine) sel_id = eid;
this->sel_engine = sel_id;
/* make engines first, and then wagons, sorted by ListPositionOfEngine() */
/* make engines first, and then wagons, sorted by selected sort_criteria */
_internal_sort_order = false;
EngList_Sort(&this->eng_list, TrainEnginesThenWagonsSorter);
/* and then sort engines */
_internal_sort_order = this->descending_sort_order;
EngList_SortPartial(&this->eng_list, _sorter[0][this->sort_criteria], 0, num_engines);
@@ -1176,90 +1176,104 @@ void TriggerVehicle(Vehicle *v, VehicleT
v->InvalidateNewGRFCacheOfChain();
DoTriggerVehicle(v, trigger, 0, true);
/* Functions for changing the order of vehicle purchase lists
* This is currently only implemented for rail vehicles. */
* Get the list position of an engine.
* Used when sorting a list of engines.
* @param engine ID of the engine.
* @return The list position of the engine.
uint ListPositionOfEngine(EngineID engine)
const Engine *e = Engine::Get(engine);
/* Crude sorting to group by GRF ID */
return (e->GetGRFID() * 256) + e->list_position;
/* Functions for changing the order of vehicle purchase lists */
struct ListOrderChange {
EngineID engine;
EngineID target;
uint target; ///< local ID
};
static SmallVector<ListOrderChange, 16> _list_order_changes;
void AlterVehicleListOrder(EngineID engine, EngineID target)
* Record a vehicle ListOrderChange.
* @param engine Engine to move
* @param target Local engine ID to move \a engine in front of
* @note All sorting is done later in CommitVehicleListOrderChanges
void AlterVehicleListOrder(EngineID engine, uint target)
/* Add the list order change to a queue */
ListOrderChange *loc = _list_order_changes.Append();
loc->engine = engine;
loc->target = target;
* Comparator function to sort engines via scope-GRFID and local ID.
* @param a left side
* @param b right side
* @return comparison result
static int CDECL EnginePreSort(const EngineID *a, const EngineID *b)
const EngineIDMapping *id_a = _engine_mngr.Get(*a);
const EngineIDMapping *id_b = _engine_mngr.Get(*b);
/* 1. Sort by engine type */
if (id_a->type != id_b->type) return (int)id_a->type - (int)id_b->type;
/* 2. Sort by scope-GRFID */
if (id_a->grfid != id_b->grfid) return id_a->grfid < id_b->grfid ? -1 : 1;
/* 3. Sort by local ID */
return (int)id_a->internal_id - (int)id_b->internal_id;
* Deternine default engine sorting and execute recorded ListOrderChanges from AlterVehicleListOrder.
void CommitVehicleListOrderChanges()
/* List position to Engine map */
typedef SmallMap<uint16, Engine *, 16> ListPositionMap;
ListPositionMap lptr_map;
/* Pre-sort engines by scope-grfid and local index */
SmallVector<EngineID, 16> ordering;
Engine *e;
FOR_ALL_ENGINES(e) {
*ordering.Append() = e->index;
QSortT(ordering.Begin(), ordering.Length(), EnginePreSort);
/* Apply Insertion-Sort opeations */
const ListOrderChange *end = _list_order_changes.End();
for (const ListOrderChange *it = _list_order_changes.Begin(); it != end; ++it) {
EngineID engine = it->engine;
EngineID target = it->target;
EngineID source = it->engine;
uint local_target = it->target;
const EngineIDMapping *id_source = _engine_mngr.Get(source);
if (id_source->internal_id == local_target) continue;
if (engine == target) continue;
EngineID target = _engine_mngr.GetID(id_source->type, local_target, id_source->grfid);
if (target == INVALID_ENGINE) continue;
int source_index = ordering.FindIndex(source);
int target_index = ordering.FindIndex(target);
Engine *source_e = Engine::Get(engine);
Engine *target_e = NULL;
assert(source_index >= 0 && target_index >= 0);
assert(source_index != target_index);
/* Populate map with current list positions */
FOR_ALL_ENGINES_OF_TYPE(e, source_e->type) {
if (!_settings_game.vehicle.dynamic_engines || e->GetGRF() == source_e->GetGRF()) {
if (e->grf_prop.local_id == target) target_e = e;
lptr_map[e->list_position] = e;
EngineID *list = ordering.Begin();
if (source_index < target_index) {
--target_index;
for (int i = source_index; i < target_index; ++i) list[i] = list[i + 1];
list[target_index] = source;
} else {
for (int i = source_index; i > target_index; --i) list[i] = list[i - 1];
/* std::map sorted by default, SmallMap does not */
lptr_map.SortByKey();
/* Get the target position, if it exists */
if (target_e != NULL) {
uint16 target_position = target_e->list_position;
bool moving = false;
const ListPositionMap::Pair *end = lptr_map.End();
for (ListPositionMap::Pair *it = lptr_map.Begin(); it != end; ++it) {
if (it->first == target_position) moving = true;
if (moving) it->second->list_position++;
/* Store final sort-order */
const EngineID *idend = ordering.End();
uint index = 0;
for (const EngineID *it = ordering.Begin(); it != idend; ++it, ++index) {
Engine::Get(*it)->list_position = index;
source_e->list_position = target_position;
lptr_map.Clear();
/* Clear out the queue */
_list_order_changes.Reset();
* Resolve an engine's spec and such so we can get a variable.
* @param ro The resolver object to fill.
* @param index The vehicle to get the data from.
@@ -60,13 +60,12 @@ enum VehicleTrigger {
VEHICLE_TRIGGER_CALLBACK_32 = 0x10,
void TriggerVehicle(Vehicle *veh, VehicleTrigger trigger);
void UnloadWagonOverrides(Engine *e);
uint ListPositionOfEngine(EngineID engine);
void AlterVehicleListOrder(EngineID engine, EngineID target);
void AlterVehicleListOrder(EngineID engine, uint target);
void CommitVehicleListOrderChanges();
EngineID GetNewEngineID(const GRFFile *file, VehicleType type, uint16 internal_id);
#endif /* NEWGRF_ENGINE_H */
Status change: