# Conflicts: # .github/workflows/ci-build.yml # .github/workflows/codeql.yml # .github/workflows/release-linux.yml # .github/workflows/release-macos.yml # .github/workflows/release-windows.yml # src/map.cpp # src/music/midifile.hpp # src/order_cmd.cpp # src/order_gui.cpp # src/pathfinder/yapf/yapf_rail.cpp # src/pbs.cpp # src/saveload/afterload.cpp # src/saveload/saveload.h # src/saveload/vehicle_sl.cpp # src/script/api/script_text.cpp # src/ship_cmd.cpp # src/train_cmd.cpp # src/vehicle.cpp # src/vehicle_base.h # src/vehicle_func.h # src/vehicle_gui.cpp # src/vehicle_gui_base.h # src/viewport.cpp # src/waypoint_cmd.cpp
		
			
				
	
	
		
			1122 lines
		
	
	
		
			33 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			1122 lines
		
	
	
		
			33 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
 * This file is part of OpenTTD.
 | 
						|
 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
 | 
						|
 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 | 
						|
 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
 | 
						|
 */
 | 
						|
 | 
						|
/** @file group_cmd.cpp Handling of the engine groups */
 | 
						|
 | 
						|
#include "stdafx.h"
 | 
						|
#include "cmd_helper.h"
 | 
						|
#include "command_func.h"
 | 
						|
#include "train.h"
 | 
						|
#include "vehiclelist.h"
 | 
						|
#include "vehicle_func.h"
 | 
						|
#include "autoreplace_base.h"
 | 
						|
#include "autoreplace_func.h"
 | 
						|
#include "base_station_base.h"
 | 
						|
#include "string_func.h"
 | 
						|
#include "company_func.h"
 | 
						|
#include "core/pool_func.hpp"
 | 
						|
#include "order_backup.h"
 | 
						|
#include "tbtr_template_vehicle.h"
 | 
						|
#include "tracerestrict.h"
 | 
						|
 | 
						|
#include "table/strings.h"
 | 
						|
 | 
						|
#include "safeguards.h"
 | 
						|
#include "strings_func.h"
 | 
						|
#include "town.h"
 | 
						|
#include "townname_func.h"
 | 
						|
 | 
						|
GroupID _new_group_id;
 | 
						|
 | 
						|
GroupPool _group_pool("Group");
 | 
						|
INSTANTIATE_POOL_METHODS(Group)
 | 
						|
 | 
						|
/**
 | 
						|
 * Clear all caches.
 | 
						|
 */
 | 
						|
void GroupStatistics::Clear()
 | 
						|
{
 | 
						|
	this->num_vehicle = 0;
 | 
						|
	this->profit_last_year = 0;
 | 
						|
	this->num_vehicle_min_age = 0;
 | 
						|
	this->profit_last_year_min_age = 0;
 | 
						|
 | 
						|
	/* This is also called when NewGRF change. So the number of engines might have changed. Reset. */
 | 
						|
	this->num_engines.clear();
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get number of vehicles of a specific engine ID.
 | 
						|
 * @param engine Engine ID.
 | 
						|
 * @returns number of vehicles of this engine ID.
 | 
						|
 */
 | 
						|
uint16_t GroupStatistics::GetNumEngines(EngineID engine) const
 | 
						|
{
 | 
						|
	auto found = this->num_engines.find(engine);
 | 
						|
	if (found != std::end(this->num_engines)) return found->second;
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Returns the GroupStatistics for a specific group.
 | 
						|
 * @param company Owner of the group.
 | 
						|
 * @param id_g    GroupID of the group.
 | 
						|
 * @param type    VehicleType of the vehicles in the group.
 | 
						|
 * @return Statistics for the group.
 | 
						|
 */
 | 
						|
/* static */ GroupStatistics &GroupStatistics::Get(CompanyID company, GroupID id_g, VehicleType type)
 | 
						|
{
 | 
						|
	if (Group::IsValidID(id_g)) {
 | 
						|
		Group *g = Group::Get(id_g);
 | 
						|
		assert(g->owner == company);
 | 
						|
		assert(g->vehicle_type == type);
 | 
						|
		return g->statistics;
 | 
						|
	}
 | 
						|
 | 
						|
	if (IsDefaultGroupID(id_g)) return Company::Get(company)->group_default[type];
 | 
						|
	if (IsAllGroupID(id_g)) return Company::Get(company)->group_all[type];
 | 
						|
 | 
						|
	NOT_REACHED();
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Returns the GroupStatistic for the group of a vehicle.
 | 
						|
 * @param v Vehicle.
 | 
						|
 * @return GroupStatistics for the group of the vehicle.
 | 
						|
 */
 | 
						|
/* static */ GroupStatistics &GroupStatistics::Get(const Vehicle *v)
 | 
						|
{
 | 
						|
	return GroupStatistics::Get(v->owner, v->group_id, v->type);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Returns the GroupStatistic for the ALL_GROUPO of a vehicle type.
 | 
						|
 * @param v Vehicle.
 | 
						|
 * @return GroupStatistics for the ALL_GROUP of the vehicle type.
 | 
						|
 */
 | 
						|
/* static */ GroupStatistics &GroupStatistics::GetAllGroup(const Vehicle *v)
 | 
						|
{
 | 
						|
	return GroupStatistics::Get(v->owner, ALL_GROUP, v->type);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Update all caches after loading a game, changing NewGRF, etc.
 | 
						|
 */
 | 
						|
/* static */ void GroupStatistics::UpdateAfterLoad()
 | 
						|
{
 | 
						|
	/* Set up the engine count for all companies */
 | 
						|
	for (Company *c : Company::Iterate()) {
 | 
						|
		for (VehicleType type = VEH_BEGIN; type < VEH_COMPANY_END; type++) {
 | 
						|
			c->group_all[type].Clear();
 | 
						|
			c->group_default[type].Clear();
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	/* Recalculate */
 | 
						|
	for (Group *g : Group::Iterate()) {
 | 
						|
		g->statistics.Clear();
 | 
						|
	}
 | 
						|
 | 
						|
	for (const Vehicle *v : Vehicle::Iterate()) {
 | 
						|
		if (!v->IsEngineCountable()) continue;
 | 
						|
 | 
						|
		GroupStatistics::CountEngine(v, 1);
 | 
						|
		if (v->IsPrimaryVehicle()) GroupStatistics::CountVehicle(v, 1);
 | 
						|
	}
 | 
						|
 | 
						|
	for (const Company *c : Company::Iterate()) {
 | 
						|
		GroupStatistics::UpdateAutoreplace(c->index);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Update num_vehicle when adding or removing a vehicle.
 | 
						|
 * @param v Vehicle to count.
 | 
						|
 * @param delta +1 to add, -1 to remove.
 | 
						|
 */
 | 
						|
/* static */ void GroupStatistics::CountVehicle(const Vehicle *v, int delta)
 | 
						|
{
 | 
						|
	/* make virtual trains group-neutral */
 | 
						|
	if (HasBit(v->subtype, GVSF_VIRTUAL)) return;
 | 
						|
 | 
						|
	assert(delta == 1 || delta == -1);
 | 
						|
 | 
						|
	GroupStatistics &stats_all = GroupStatistics::GetAllGroup(v);
 | 
						|
	GroupStatistics &stats = GroupStatistics::Get(v);
 | 
						|
 | 
						|
	stats_all.num_vehicle += delta;
 | 
						|
	stats_all.profit_last_year += v->GetDisplayProfitLastYear() * delta;
 | 
						|
	stats.num_vehicle += delta;
 | 
						|
	stats.profit_last_year += v->GetDisplayProfitLastYear() * delta;
 | 
						|
 | 
						|
	if (v->economy_age > VEHICLE_PROFIT_MIN_AGE) {
 | 
						|
		stats_all.num_vehicle_min_age += delta;
 | 
						|
		stats_all.profit_last_year_min_age += v->GetDisplayProfitLastYear() * delta;
 | 
						|
		stats.num_vehicle_min_age += delta;
 | 
						|
		stats.profit_last_year_min_age += v->GetDisplayProfitLastYear() * delta;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Update num_engines when adding/removing an engine.
 | 
						|
 * @param v Engine to count.
 | 
						|
 * @param delta +1 to add, -1 to remove.
 | 
						|
 */
 | 
						|
/* static */ void GroupStatistics::CountEngine(const Vehicle *v, int delta)
 | 
						|
{
 | 
						|
	/* make virtual trains group-neutral */
 | 
						|
	if (HasBit(v->subtype, GVSF_VIRTUAL)) return;
 | 
						|
 | 
						|
	assert(delta == 1 || delta == -1);
 | 
						|
	GroupStatistics::GetAllGroup(v).num_engines[v->engine_type] += delta;
 | 
						|
	GroupStatistics::Get(v).num_engines[v->engine_type] += delta;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Add a vehicle's last year profit to the profit sum of its group.
 | 
						|
 */
 | 
						|
/* static */ void GroupStatistics::AddProfitLastYear(const Vehicle *v)
 | 
						|
{
 | 
						|
	GroupStatistics &stats_all = GroupStatistics::GetAllGroup(v);
 | 
						|
	GroupStatistics &stats = GroupStatistics::Get(v);
 | 
						|
 | 
						|
	stats_all.profit_last_year += v->GetDisplayProfitLastYear();
 | 
						|
	stats.profit_last_year += v->GetDisplayProfitLastYear();
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Add a vehicle to the profit sum of its group.
 | 
						|
 */
 | 
						|
/* static */ void GroupStatistics::VehicleReachedMinAge(const Vehicle *v)
 | 
						|
{
 | 
						|
	GroupStatistics &stats_all = GroupStatistics::GetAllGroup(v);
 | 
						|
	GroupStatistics &stats = GroupStatistics::Get(v);
 | 
						|
 | 
						|
	stats_all.num_vehicle_min_age++;
 | 
						|
	stats_all.profit_last_year_min_age += v->GetDisplayProfitLastYear();
 | 
						|
	stats.num_vehicle_min_age++;
 | 
						|
	stats.profit_last_year_min_age += v->GetDisplayProfitLastYear();
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Recompute the profits for all groups.
 | 
						|
 */
 | 
						|
/* static */ void GroupStatistics::UpdateProfits()
 | 
						|
{
 | 
						|
	/* Set up the engine count for all companies */
 | 
						|
	for (Company *c : Company::Iterate()) {
 | 
						|
		for (VehicleType type = VEH_BEGIN; type < VEH_COMPANY_END; type++) {
 | 
						|
			c->group_all[type].ClearProfits();
 | 
						|
			c->group_default[type].ClearProfits();
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	/* Recalculate */
 | 
						|
	for (Group *g : Group::Iterate()) {
 | 
						|
		g->statistics.ClearProfits();
 | 
						|
	}
 | 
						|
 | 
						|
	for (const Vehicle *v : Vehicle::IterateFrontOnly()) {
 | 
						|
		if (v->IsPrimaryVehicle() && !HasBit(v->subtype, GVSF_VIRTUAL)) {
 | 
						|
			GroupStatistics::AddProfitLastYear(v);
 | 
						|
			if (v->economy_age > VEHICLE_PROFIT_MIN_AGE) GroupStatistics::VehicleReachedMinAge(v);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Update autoreplace_defined and autoreplace_finished of all statistics of a company.
 | 
						|
 * @param company Company to update statistics for.
 | 
						|
 */
 | 
						|
/* static */ void GroupStatistics::UpdateAutoreplace(CompanyID company)
 | 
						|
{
 | 
						|
	/* Set up the engine count for all companies */
 | 
						|
	Company *c = Company::Get(company);
 | 
						|
	for (VehicleType type = VEH_BEGIN; type < VEH_COMPANY_END; type++) {
 | 
						|
		c->group_all[type].ClearAutoreplace();
 | 
						|
		c->group_default[type].ClearAutoreplace();
 | 
						|
	}
 | 
						|
 | 
						|
	/* Recalculate */
 | 
						|
	for (Group *g : Group::Iterate()) {
 | 
						|
		if (g->owner != company) continue;
 | 
						|
		g->statistics.ClearAutoreplace();
 | 
						|
	}
 | 
						|
 | 
						|
	for (EngineRenewList erl = c->engine_renew_list; erl != nullptr; erl = erl->next) {
 | 
						|
		const Engine *e = Engine::Get(erl->from);
 | 
						|
		GroupStatistics &stats = GroupStatistics::Get(company, erl->group_id, e->type);
 | 
						|
		if (!stats.autoreplace_defined) {
 | 
						|
			stats.autoreplace_defined = true;
 | 
						|
			stats.autoreplace_finished = true;
 | 
						|
		}
 | 
						|
		if (GetGroupNumEngines(company, erl->group_id, erl->from) > 0) stats.autoreplace_finished = false;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Update the num engines of a groupID. Decrease the old one and increase the new one
 | 
						|
 * @note called in SetTrainGroupID and UpdateTrainGroupID
 | 
						|
 * @param v     Vehicle we have to update
 | 
						|
 * @param old_g index of the old group
 | 
						|
 * @param new_g index of the new group
 | 
						|
 */
 | 
						|
static inline void UpdateNumEngineGroup(const Vehicle *v, GroupID old_g, GroupID new_g)
 | 
						|
{
 | 
						|
	if (old_g != new_g) {
 | 
						|
		/* Decrease the num engines in the old group */
 | 
						|
		GroupStatistics::Get(v->owner, old_g, v->type).num_engines[v->engine_type]--;
 | 
						|
 | 
						|
		/* Increase the num engines in the new group */
 | 
						|
		GroupStatistics::Get(v->owner, new_g, v->type).num_engines[v->engine_type]++;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
const Livery *GetParentLivery(const Group *g)
 | 
						|
{
 | 
						|
	if (g->parent == INVALID_GROUP) {
 | 
						|
		const Company *c = Company::Get(g->owner);
 | 
						|
		return &c->livery[LS_DEFAULT];
 | 
						|
	}
 | 
						|
 | 
						|
	const Group *pg = Group::Get(g->parent);
 | 
						|
	return &pg->livery;
 | 
						|
}
 | 
						|
 | 
						|
static inline bool IsGroupDescendantOfGroupID(const Group *g, const GroupID top_gid, const Owner owner)
 | 
						|
{
 | 
						|
	if (g->owner != owner) return false;
 | 
						|
 | 
						|
	while (true) {
 | 
						|
		if (g->parent == top_gid) return true;
 | 
						|
		if (g->parent == INVALID_GROUP) return false;
 | 
						|
		g = Group::Get(g->parent);
 | 
						|
	}
 | 
						|
 | 
						|
	NOT_REACHED();
 | 
						|
}
 | 
						|
 | 
						|
static inline bool IsGroupDescendantOfGroup(const Group *g, const Group *top)
 | 
						|
{
 | 
						|
	return IsGroupDescendantOfGroupID(g, top->index, top->owner);
 | 
						|
}
 | 
						|
 | 
						|
static inline bool IsGroupIDDescendantOfGroupID(const GroupID gid, const GroupID top_gid, const Owner owner)
 | 
						|
{
 | 
						|
	if (IsTopLevelGroupID(gid) || gid == INVALID_GROUP) return false;
 | 
						|
 | 
						|
	return IsGroupDescendantOfGroupID(Group::Get(gid), top_gid, owner);
 | 
						|
}
 | 
						|
 | 
						|
template <typename F>
 | 
						|
void IterateDescendantsOfGroup(const Group *top, F func)
 | 
						|
{
 | 
						|
	for (Group *cg : Group::Iterate()) {
 | 
						|
		if (IsGroupDescendantOfGroup(cg, top)) {
 | 
						|
			func(cg);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
template <typename F>
 | 
						|
void IterateDescendantsOfGroup(GroupID id_top, F func)
 | 
						|
{
 | 
						|
	const Group *top = Group::GetIfValid(id_top);
 | 
						|
	if (top != nullptr) IterateDescendantsOfGroup<F>(top, func);
 | 
						|
}
 | 
						|
 | 
						|
static void PropagateChildLiveryResetVehicleCache(const Group *g)
 | 
						|
{
 | 
						|
	/* Company colour data is indirectly cached. */
 | 
						|
	for (Vehicle *v : Vehicle::IterateFrontOnly()) {
 | 
						|
		if (v->IsPrimaryVehicle() && (v->group_id == g->index || IsGroupIDDescendantOfGroupID(v->group_id, g->index, g->owner))) {
 | 
						|
			for (Vehicle *u = v; u != nullptr; u = u->Next()) {
 | 
						|
				u->colourmap = PAL_NONE;
 | 
						|
				u->InvalidateNewGRFCache();
 | 
						|
				u->InvalidateImageCache();
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void PropagateChildLivery(const GroupID top_gid, const Owner owner, const Livery &top_livery)
 | 
						|
{
 | 
						|
	for (Group *g : Group::Iterate()) {
 | 
						|
		if (g->owner != owner) continue;
 | 
						|
 | 
						|
		Livery livery = g->livery;
 | 
						|
 | 
						|
		const Group *pg = g;
 | 
						|
		bool is_descendant = (g->index == top_gid);
 | 
						|
		while (!is_descendant) {
 | 
						|
			if (pg->parent == top_gid) {
 | 
						|
				is_descendant = true;
 | 
						|
				break;
 | 
						|
			}
 | 
						|
			if (pg->parent == INVALID_GROUP) break;
 | 
						|
			pg = Group::Get(pg->parent);
 | 
						|
			if (!HasBit(livery.in_use, 0)) livery.colour1 = pg->livery.colour1;
 | 
						|
			if (!HasBit(livery.in_use, 1)) livery.colour2 = pg->livery.colour2;
 | 
						|
			livery.in_use |= pg->livery.in_use;
 | 
						|
		}
 | 
						|
		if (is_descendant) {
 | 
						|
			if (!HasBit(livery.in_use, 0)) livery.colour1 = top_livery.colour1;
 | 
						|
			if (!HasBit(livery.in_use, 1)) livery.colour2 = top_livery.colour2;
 | 
						|
			g->livery.colour1 = livery.colour1;
 | 
						|
			g->livery.colour2 = livery.colour2;
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Propagate a livery change to a group's children, and optionally update cached vehicle colourmaps.
 | 
						|
 * @param g Group to propagate colours to children.
 | 
						|
 * @param reset_cache Reset colourmap of vehicles in this group.
 | 
						|
 */
 | 
						|
static void PropagateChildLivery(const Group *g, bool reset_cache)
 | 
						|
{
 | 
						|
	PropagateChildLivery(g->index, g->owner, g->livery);
 | 
						|
	if (reset_cache) PropagateChildLiveryResetVehicleCache(g);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Update group liveries for a company. This is called when the LS_DEFAULT scheme is changed, to update groups with
 | 
						|
 * colours set to default.
 | 
						|
 * @param c Company to update.
 | 
						|
 */
 | 
						|
void UpdateCompanyGroupLiveries(const Company *c)
 | 
						|
{
 | 
						|
	PropagateChildLivery(INVALID_GROUP, c->index, c->livery[LS_DEFAULT]);
 | 
						|
}
 | 
						|
 | 
						|
Group::Group(Owner owner)
 | 
						|
{
 | 
						|
	this->owner = owner;
 | 
						|
	this->folded = false;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
 * Create a new vehicle group.
 | 
						|
 * @param tile unused
 | 
						|
 * @param flags type of operation
 | 
						|
 * @param p1   vehicle type
 | 
						|
 * @param p2   parent groupid
 | 
						|
 * @param text unused
 | 
						|
 * @return the cost of this operation or an error
 | 
						|
 */
 | 
						|
CommandCost CmdCreateGroup(TileIndex tile, DoCommandFlag flags, uint32_t p1, uint32_t p2, const char *text)
 | 
						|
{
 | 
						|
	VehicleType vt = Extract<VehicleType, 0, 3>(p1);
 | 
						|
	if (!IsCompanyBuildableVehicleType(vt)) return CMD_ERROR;
 | 
						|
 | 
						|
	if (!Group::CanAllocateItem()) return CMD_ERROR;
 | 
						|
 | 
						|
	const Group *pg = Group::GetIfValid(GB(p2, 0, 16));
 | 
						|
	if (pg != nullptr) {
 | 
						|
		if (pg->owner != _current_company) return CMD_ERROR;
 | 
						|
		if (pg->vehicle_type != vt) return CMD_ERROR;
 | 
						|
	}
 | 
						|
 | 
						|
	if (flags & DC_EXEC) {
 | 
						|
		Group *g = new Group(_current_company);
 | 
						|
		g->vehicle_type = vt;
 | 
						|
		g->parent = INVALID_GROUP;
 | 
						|
 | 
						|
		if (pg == nullptr) {
 | 
						|
			const Company *c = Company::Get(_current_company);
 | 
						|
			g->livery.colour1 = c->livery[LS_DEFAULT].colour1;
 | 
						|
			g->livery.colour2 = c->livery[LS_DEFAULT].colour2;
 | 
						|
			if (c->settings.renew_keep_length) SetBit(g->flags, GroupFlags::GF_REPLACE_WAGON_REMOVAL);
 | 
						|
		} else {
 | 
						|
			g->parent = pg->index;
 | 
						|
			g->livery.colour1 = pg->livery.colour1;
 | 
						|
			g->livery.colour2 = pg->livery.colour2;
 | 
						|
			g->flags = pg->flags;
 | 
						|
			if (vt == VEH_TRAIN) ReindexTemplateReplacementsRecursive();
 | 
						|
		}
 | 
						|
 | 
						|
		_new_group_id = g->index;
 | 
						|
 | 
						|
		InvalidateWindowData(GetWindowClassForVehicleType(vt), VehicleListIdentifier(VL_GROUP_LIST, vt, _current_company).Pack());
 | 
						|
		InvalidateWindowData(WC_COMPANY_COLOUR, g->owner, g->vehicle_type);
 | 
						|
		InvalidateWindowData(WC_TEMPLATEGUI_MAIN, 0, 0, 0);
 | 
						|
	}
 | 
						|
 | 
						|
	return CommandCost();
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
 * Add all vehicles in the given group to the default group and then deletes the group.
 | 
						|
 * @param tile unused
 | 
						|
 * @param flags type of operation
 | 
						|
 * @param p1   index of array group
 | 
						|
 *      - p1 bit 0-15 : GroupID
 | 
						|
 * @param p2   unused
 | 
						|
 * @param text unused
 | 
						|
 * @return the cost of this operation or an error
 | 
						|
 */
 | 
						|
CommandCost CmdDeleteGroup(TileIndex tile, DoCommandFlag flags, uint32_t p1, uint32_t p2, const char *text)
 | 
						|
{
 | 
						|
	Group *g = Group::GetIfValid(p1);
 | 
						|
	if (g == nullptr || g->owner != _current_company) return CMD_ERROR;
 | 
						|
 | 
						|
	/* Remove all vehicles from the group */
 | 
						|
	DoCommand(0, p1, 0, flags, CMD_REMOVE_ALL_VEHICLES_GROUP);
 | 
						|
 | 
						|
	/* Delete sub-groups */
 | 
						|
	for (const Group *gp : Group::Iterate()) {
 | 
						|
		if (gp->parent == g->index) {
 | 
						|
			DoCommand(0, gp->index, 0, flags, CMD_DELETE_GROUP);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (flags & DC_EXEC) {
 | 
						|
		/* Update backupped orders if needed */
 | 
						|
		OrderBackup::ClearGroup(g->index);
 | 
						|
 | 
						|
		/* If we set an autoreplace for the group we delete, remove it. */
 | 
						|
		if (_current_company < MAX_COMPANIES) {
 | 
						|
			Company *c;
 | 
						|
 | 
						|
			c = Company::Get(_current_company);
 | 
						|
			for (EngineRenew *er : EngineRenew::Iterate()) {
 | 
						|
				if (er->group_id == g->index) RemoveEngineReplacementForCompany(c, er->from, g->index, flags);
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		VehicleType vt = g->vehicle_type;
 | 
						|
 | 
						|
		/* Delete all template replacements using the just deleted group */
 | 
						|
		DeleteTemplateReplacementsByGroupID(g);
 | 
						|
 | 
						|
		/* notify tracerestrict that group is about to be deleted */
 | 
						|
		TraceRestrictRemoveGroupID(g->index);
 | 
						|
 | 
						|
		/* Delete the Replace Vehicle Windows */
 | 
						|
		CloseWindowById(WC_REPLACE_VEHICLE, g->vehicle_type);
 | 
						|
		delete g;
 | 
						|
 | 
						|
		InvalidateWindowData(GetWindowClassForVehicleType(vt), VehicleListIdentifier(VL_GROUP_LIST, vt, _current_company).Pack());
 | 
						|
		InvalidateWindowData(WC_COMPANY_COLOUR, _current_company, vt);
 | 
						|
		InvalidateWindowData(WC_TEMPLATEGUI_MAIN, 0, 0, 0);
 | 
						|
	}
 | 
						|
 | 
						|
	return CommandCost();
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Alter a group
 | 
						|
 * @param tile unused
 | 
						|
 * @param flags type of operation
 | 
						|
 * @param p1   index of array group
 | 
						|
 *   - p1 bit 0-15 : GroupID
 | 
						|
 *   - p1 bit 16: 0 - Rename grouop
 | 
						|
 *                1 - Set group parent
 | 
						|
 * @param p2   parent group index
 | 
						|
 * @param text the new name or an empty string when resetting to the default
 | 
						|
 * @return the cost of this operation or an error
 | 
						|
 */
 | 
						|
CommandCost CmdAlterGroup(TileIndex tile, DoCommandFlag flags, uint32_t p1, uint32_t p2, const char *text)
 | 
						|
{
 | 
						|
	Group *g = Group::GetIfValid(GB(p1, 0, 16));
 | 
						|
	if (g == nullptr || g->owner != _current_company) return CMD_ERROR;
 | 
						|
 | 
						|
	if (!HasBit(p1, 16)) {
 | 
						|
		/* Rename group */
 | 
						|
		bool reset = StrEmpty(text);
 | 
						|
 | 
						|
		if (!reset) {
 | 
						|
			if (Utf8StringLength(text) >= MAX_LENGTH_GROUP_NAME_CHARS) return CMD_ERROR;
 | 
						|
		}
 | 
						|
 | 
						|
		if (flags & DC_EXEC) {
 | 
						|
			/* Assign the new one */
 | 
						|
			if (reset) {
 | 
						|
				g->name.clear();
 | 
						|
			} else {
 | 
						|
				g->name = text;
 | 
						|
			}
 | 
						|
		}
 | 
						|
	} else {
 | 
						|
		/* Set group parent */
 | 
						|
		const Group *pg = Group::GetIfValid(GB(p2, 0, 16));
 | 
						|
 | 
						|
		if (pg != nullptr) {
 | 
						|
			if (pg->owner != _current_company) return CMD_ERROR;
 | 
						|
			if (pg->vehicle_type != g->vehicle_type) return CMD_ERROR;
 | 
						|
 | 
						|
			/* Ensure request parent isn't child of group.
 | 
						|
			 * This is the only place that infinite loops are prevented. */
 | 
						|
			if (GroupIsInGroup(pg->index, g->index)) return_cmd_error(STR_ERROR_GROUP_CAN_T_SET_PARENT_RECURSION);
 | 
						|
		}
 | 
						|
 | 
						|
		if (flags & DC_EXEC) {
 | 
						|
			g->parent = (pg == nullptr) ? INVALID_GROUP : pg->index;
 | 
						|
			GroupStatistics::UpdateAutoreplace(g->owner);
 | 
						|
			if (g->vehicle_type == VEH_TRAIN) ReindexTemplateReplacementsRecursive();
 | 
						|
 | 
						|
			if (!HasBit(g->livery.in_use, 0) || !HasBit(g->livery.in_use, 1)) {
 | 
						|
				/* Update livery with new parent's colours if either colour is default. */
 | 
						|
				const Livery *livery = GetParentLivery(g);
 | 
						|
				if (!HasBit(g->livery.in_use, 0)) g->livery.colour1 = livery->colour1;
 | 
						|
				if (!HasBit(g->livery.in_use, 1)) g->livery.colour2 = livery->colour2;
 | 
						|
 | 
						|
				PropagateChildLivery(g, true);
 | 
						|
				MarkWholeScreenDirty();
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (flags & DC_EXEC) {
 | 
						|
		InvalidateWindowData(WC_REPLACE_VEHICLE, g->vehicle_type, 1);
 | 
						|
		InvalidateWindowData(GetWindowClassForVehicleType(g->vehicle_type), VehicleListIdentifier(VL_GROUP_LIST, g->vehicle_type, _current_company).Pack());
 | 
						|
		InvalidateWindowData(WC_COMPANY_COLOUR, g->owner, g->vehicle_type);
 | 
						|
		InvalidateWindowClassesData(WC_VEHICLE_VIEW);
 | 
						|
		InvalidateWindowClassesData(WC_VEHICLE_DETAILS);
 | 
						|
		InvalidateWindowData(WC_TEMPLATEGUI_MAIN, 0, 0, 0);
 | 
						|
	}
 | 
						|
 | 
						|
	return CommandCost();
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Create a new vehicle group.
 | 
						|
 * @param tile unused
 | 
						|
 * @param flags type of operation
 | 
						|
 * @param p1 packed VehicleListIdentifier
 | 
						|
 * @param p2 bitmask
 | 
						|
 *   - bit 0-7 Cargo filter
 | 
						|
 * @param text the new name or an empty string when setting to the default
 | 
						|
 * @return the cost of this operation or an error
 | 
						|
 */
 | 
						|
CommandCost CmdCreateGroupFromList(TileIndex tile, DoCommandFlag flags, uint32_t p1, uint32_t p2, const char *text)
 | 
						|
{
 | 
						|
	VehicleListIdentifier vli;
 | 
						|
	VehicleList list;
 | 
						|
	if (!vli.UnpackIfValid(p1)) return CMD_ERROR;
 | 
						|
	if (!IsCompanyBuildableVehicleType(vli.vtype)) return CMD_ERROR;
 | 
						|
	if (!GenerateVehicleSortList(&list, vli, GB(p2, 0, 8))) return CMD_ERROR;
 | 
						|
 | 
						|
	CommandCost ret = DoCommand(tile, vli.vtype, INVALID_GROUP, flags, CMD_CREATE_GROUP);
 | 
						|
	if (ret.Failed()) return ret;
 | 
						|
 | 
						|
	if (!StrEmpty(text)) {
 | 
						|
		if (Utf8StringLength(text) >= MAX_LENGTH_GROUP_NAME_CHARS) return CMD_ERROR;
 | 
						|
	}
 | 
						|
 | 
						|
	if (flags & DC_EXEC) {
 | 
						|
		Group *g = Group::GetIfValid(_new_group_id);
 | 
						|
		if (g == nullptr || g->owner != _current_company) return CMD_ERROR;
 | 
						|
 | 
						|
		if (!StrEmpty(text)) {
 | 
						|
			DoCommand(tile, g->index, 0, flags, CMD_ALTER_GROUP, text);
 | 
						|
		}
 | 
						|
 | 
						|
		for (uint i = 0; i < list.size(); i++) {
 | 
						|
			const Vehicle *v = list[i];
 | 
						|
 | 
						|
			/* Just try and don't care if some vehicle's can't be added. */
 | 
						|
			DoCommand(tile, g->index, v->index, flags, CMD_ADD_VEHICLE_GROUP);
 | 
						|
		}
 | 
						|
 | 
						|
		MarkWholeScreenDirty();
 | 
						|
	}
 | 
						|
 | 
						|
	return CommandCost();
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
 * Do add a vehicle to a group.
 | 
						|
 * @param v Vehicle to add.
 | 
						|
 * @param new_g Group to add to.
 | 
						|
 */
 | 
						|
static void AddVehicleToGroup(Vehicle *v, GroupID new_g)
 | 
						|
{
 | 
						|
	GroupStatistics::CountVehicle(v, -1);
 | 
						|
 | 
						|
	switch (v->type) {
 | 
						|
		default: NOT_REACHED();
 | 
						|
		case VEH_TRAIN:
 | 
						|
			SetTrainGroupID(Train::From(v), new_g);
 | 
						|
			break;
 | 
						|
 | 
						|
		case VEH_ROAD:
 | 
						|
		case VEH_SHIP:
 | 
						|
		case VEH_AIRCRAFT:
 | 
						|
			if (v->IsEngineCountable()) UpdateNumEngineGroup(v, v->group_id, new_g);
 | 
						|
			v->group_id = new_g;
 | 
						|
			for (Vehicle *u = v; u != nullptr; u = u->Next()) {
 | 
						|
				u->colourmap = PAL_NONE;
 | 
						|
				u->InvalidateNewGRFCache();
 | 
						|
				u->InvalidateImageCache();
 | 
						|
				u->UpdateViewport(true);
 | 
						|
			}
 | 
						|
			break;
 | 
						|
	}
 | 
						|
 | 
						|
	GroupStatistics::CountVehicle(v, 1);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Add a vehicle to a group
 | 
						|
 * @param tile unused
 | 
						|
 * @param flags type of operation
 | 
						|
 * @param p1   index of array group
 | 
						|
 *   - p1 bit 0-15 : GroupID
 | 
						|
 * @param p2   vehicle to add to a group
 | 
						|
 *   - p2 bit 0-19 : VehicleID
 | 
						|
 *   - p2 bit   31 : Add shared vehicles as well.
 | 
						|
 * @param text unused
 | 
						|
 * @return the cost of this operation or an error
 | 
						|
 */
 | 
						|
CommandCost CmdAddVehicleGroup(TileIndex tile, DoCommandFlag flags, uint32_t p1, uint32_t p2, const char *text)
 | 
						|
{
 | 
						|
	Vehicle *v = Vehicle::GetIfValid(GB(p2, 0, 20));
 | 
						|
	GroupID new_g = p1;
 | 
						|
 | 
						|
	if (v == nullptr || (!Group::IsValidID(new_g) && !IsDefaultGroupID(new_g) && new_g != NEW_GROUP)) return CMD_ERROR;
 | 
						|
 | 
						|
	if (Group::IsValidID(new_g)) {
 | 
						|
		Group *g = Group::Get(new_g);
 | 
						|
		if (g->owner != _current_company || g->vehicle_type != v->type) return CMD_ERROR;
 | 
						|
	}
 | 
						|
 | 
						|
	if (v->owner != _current_company || !v->IsPrimaryVehicle()) return CMD_ERROR;
 | 
						|
 | 
						|
	if (new_g == NEW_GROUP) {
 | 
						|
		/* Create new group. */
 | 
						|
		CommandCost ret = CmdCreateGroup(0, flags, v->type, INVALID_GROUP, nullptr);
 | 
						|
		if (ret.Failed()) return ret;
 | 
						|
 | 
						|
		new_g = _new_group_id;
 | 
						|
	}
 | 
						|
 | 
						|
	if (flags & DC_EXEC) {
 | 
						|
		AddVehicleToGroup(v, new_g);
 | 
						|
 | 
						|
		if (HasBit(p2, 31)) {
 | 
						|
			/* Add vehicles in the shared order list as well. */
 | 
						|
			for (Vehicle *v2 = v->FirstShared(); v2 != nullptr; v2 = v2->NextShared()) {
 | 
						|
				if (v2->group_id != new_g) AddVehicleToGroup(v2, new_g);
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		GroupStatistics::UpdateAutoreplace(v->owner);
 | 
						|
 | 
						|
		/* Update the Replace Vehicle Windows */
 | 
						|
		SetWindowDirty(WC_REPLACE_VEHICLE, v->type);
 | 
						|
		SetWindowDirty(WC_VEHICLE_DEPOT, v->tile);
 | 
						|
		SetWindowDirty(WC_VEHICLE_VIEW, v->index);
 | 
						|
		SetWindowDirty(WC_VEHICLE_DETAILS, v->index);
 | 
						|
		InvalidateWindowData(GetWindowClassForVehicleType(v->type), VehicleListIdentifier(VL_GROUP_LIST, v->type, _current_company).Pack());
 | 
						|
		InvalidateWindowData(WC_VEHICLE_VIEW, v->index);
 | 
						|
		InvalidateWindowData(WC_VEHICLE_DETAILS, v->index);
 | 
						|
	}
 | 
						|
 | 
						|
	return CommandCost();
 | 
						|
}
 | 
						|
 | 
						|
static Town* GetTownFromDestination(const DestinationID destination)
 | 
						|
{
 | 
						|
	Town* town = nullptr;
 | 
						|
 | 
						|
	BaseStation *st = BaseStation::GetIfValid(destination);
 | 
						|
	if (st != nullptr) {
 | 
						|
		town = st->town;
 | 
						|
	}
 | 
						|
 | 
						|
	return town;
 | 
						|
}
 | 
						|
 | 
						|
static void GetAutoGroupMostRelevantTowns(const Vehicle *vehicle, Town* &from, Town* &to)
 | 
						|
{
 | 
						|
	std::vector<Town*> unique_destinations;
 | 
						|
 | 
						|
	const int num = vehicle->GetNumOrders();
 | 
						|
 | 
						|
	for (int x = 0; x < num; x++)
 | 
						|
	{
 | 
						|
		Order *order = vehicle->GetOrder(x);
 | 
						|
 | 
						|
		if (order->GetType() != OT_GOTO_STATION) continue;
 | 
						|
 | 
						|
		const DestinationID dest = order->GetDestination();
 | 
						|
		Town *town = GetTownFromDestination(dest);
 | 
						|
 | 
						|
		if (town != nullptr && unique_destinations.end() == std::find(unique_destinations.begin(), unique_destinations.end(), town))
 | 
						|
		{
 | 
						|
			unique_destinations.push_back(town);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (unique_destinations.empty()) return;
 | 
						|
 | 
						|
	from = unique_destinations[0];
 | 
						|
 | 
						|
	if (unique_destinations.size() > 1) {
 | 
						|
		to = unique_destinations[unique_destinations.size() - 1];
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static CargoTypes GetVehicleCargoList(const Vehicle *vehicle)
 | 
						|
{
 | 
						|
	CargoTypes cargoes = 0;
 | 
						|
 | 
						|
	for (const Vehicle *u = vehicle; u != nullptr; u = u->Next()) {
 | 
						|
		if (u->cargo_cap == 0) continue;
 | 
						|
 | 
						|
		SetBit(cargoes, u->cargo_type);
 | 
						|
	}
 | 
						|
	return cargoes;
 | 
						|
}
 | 
						|
 | 
						|
std::string GenerateAutoNameForVehicleGroup(const Vehicle *v)
 | 
						|
{
 | 
						|
	Town *town_from = nullptr;
 | 
						|
	Town *town_to = nullptr;
 | 
						|
 | 
						|
	GetAutoGroupMostRelevantTowns(v, town_from, town_to);
 | 
						|
	if (town_from == nullptr) return "";
 | 
						|
 | 
						|
	CargoTypes cargoes = GetVehicleCargoList(v);
 | 
						|
 | 
						|
	StringID str;
 | 
						|
	if (town_from == town_to || town_to == nullptr) {
 | 
						|
		SetDParam(0, town_from->index);
 | 
						|
		SetDParam(1, (cargoes != 0) ? STR_VEHICLE_AUTO_GROUP_CARGO_LIST : STR_EMPTY);
 | 
						|
		SetDParam(2, cargoes);
 | 
						|
		str = STR_VEHICLE_AUTO_GROUP_LOCAL_ROUTE;
 | 
						|
	} else {
 | 
						|
		SetDParam(0, town_from->index);
 | 
						|
		SetDParam(1, town_to->index);
 | 
						|
		SetDParam(2, (cargoes != 0) ? STR_VEHICLE_AUTO_GROUP_CARGO_LIST : STR_EMPTY);
 | 
						|
		SetDParam(3, cargoes);
 | 
						|
		str = STR_VEHICLE_AUTO_GROUP_ROUTE;
 | 
						|
	}
 | 
						|
	return GetString(str);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Add all shared vehicles of all vehicles from a group
 | 
						|
 * @param tile unused
 | 
						|
 * @param flags type of operation
 | 
						|
 * @param p1   index of group array
 | 
						|
 *  - p1 bit 0-15 : GroupID
 | 
						|
 * @param p2   type of vehicles
 | 
						|
 * @param text unused
 | 
						|
 * @return the cost of this operation or an error
 | 
						|
 */
 | 
						|
CommandCost CmdAddSharedVehicleGroup(TileIndex tile, DoCommandFlag flags, uint32_t p1, uint32_t p2, const char *text)
 | 
						|
{
 | 
						|
	VehicleType type = Extract<VehicleType, 0, 3>(p2);
 | 
						|
	GroupID id_g = p1;
 | 
						|
	if (!Group::IsValidID(id_g) || !IsCompanyBuildableVehicleType(type)) return CMD_ERROR;
 | 
						|
 | 
						|
	if (flags & DC_EXEC) {
 | 
						|
		/* Find the first front engine which belong to the group id_g
 | 
						|
		 * then add all shared vehicles of this front engine to the group id_g */
 | 
						|
		for (const Vehicle *v : Vehicle::IterateTypeFrontOnly(type)) {
 | 
						|
			if (v->IsPrimaryVehicle()) {
 | 
						|
				if (v->group_id != id_g) continue;
 | 
						|
 | 
						|
				/* For each shared vehicles add it to the group */
 | 
						|
				for (Vehicle *v2 = v->FirstShared(); v2 != nullptr; v2 = v2->NextShared()) {
 | 
						|
					if (v2->group_id != id_g) DoCommand(tile, id_g, v2->index, flags, CMD_ADD_VEHICLE_GROUP, text);
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		InvalidateWindowData(GetWindowClassForVehicleType(type), VehicleListIdentifier(VL_GROUP_LIST, type, _current_company).Pack());
 | 
						|
	}
 | 
						|
 | 
						|
	return CommandCost();
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
 * Remove all vehicles from a group
 | 
						|
 * @param tile unused
 | 
						|
 * @param flags type of operation
 | 
						|
 * @param p1   index of group array
 | 
						|
 * - p1 bit 0-15 : GroupID
 | 
						|
 * @param p2   unused
 | 
						|
 * @param text unused
 | 
						|
 * @return the cost of this operation or an error
 | 
						|
 */
 | 
						|
CommandCost CmdRemoveAllVehiclesGroup(TileIndex tile, DoCommandFlag flags, uint32_t p1, uint32_t p2, const char *text)
 | 
						|
{
 | 
						|
	GroupID old_g = p1;
 | 
						|
	Group *g = Group::GetIfValid(old_g);
 | 
						|
 | 
						|
	if (g == nullptr || g->owner != _current_company) return CMD_ERROR;
 | 
						|
 | 
						|
	if (flags & DC_EXEC) {
 | 
						|
		/* Find each Vehicle that belongs to the group old_g and add it to the default group */
 | 
						|
		for (const Vehicle *v : Vehicle::IterateFrontOnly()) {
 | 
						|
			if (v->IsPrimaryVehicle()) {
 | 
						|
				if (v->group_id != old_g) continue;
 | 
						|
 | 
						|
				/* Add The Vehicle to the default group */
 | 
						|
				DoCommand(tile, DEFAULT_GROUP, v->index, flags, CMD_ADD_VEHICLE_GROUP, text);
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		InvalidateWindowData(GetWindowClassForVehicleType(g->vehicle_type), VehicleListIdentifier(VL_GROUP_LIST, g->vehicle_type, _current_company).Pack());
 | 
						|
	}
 | 
						|
 | 
						|
	return CommandCost();
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Set the livery for a vehicle group.
 | 
						|
 * @param tile      Unused.
 | 
						|
 * @param flags     Command flags.
 | 
						|
 * @param p1
 | 
						|
 * - p1 bit  0-15   Group ID.
 | 
						|
 * @param p2
 | 
						|
 * - p2 bit  8      Set secondary instead of primary colour
 | 
						|
 * - p2 bit 16-23   Colour.
 | 
						|
 */
 | 
						|
CommandCost CmdSetGroupLivery(TileIndex tile, DoCommandFlag flags, uint32_t p1, uint32_t p2, const char *text)
 | 
						|
{
 | 
						|
	Group *g = Group::GetIfValid(p1);
 | 
						|
	bool primary = !HasBit(p2, 8);
 | 
						|
	Colours colour = Extract<Colours, 16, 8>(p2);
 | 
						|
 | 
						|
	if (g == nullptr || g->owner != _current_company) return CMD_ERROR;
 | 
						|
 | 
						|
	if (colour >= COLOUR_END && colour != INVALID_COLOUR) return CMD_ERROR;
 | 
						|
 | 
						|
	if (flags & DC_EXEC) {
 | 
						|
		if (primary) {
 | 
						|
			SB(g->livery.in_use, 0, 1, colour != INVALID_COLOUR);
 | 
						|
			if (colour == INVALID_COLOUR) colour = GetParentLivery(g)->colour1;
 | 
						|
			g->livery.colour1 = colour;
 | 
						|
		} else {
 | 
						|
			SB(g->livery.in_use, 1, 1, colour != INVALID_COLOUR);
 | 
						|
			if (colour == INVALID_COLOUR) colour = GetParentLivery(g)->colour2;
 | 
						|
			g->livery.colour2 = colour;
 | 
						|
		}
 | 
						|
 | 
						|
		PropagateChildLivery(g, true);
 | 
						|
		MarkWholeScreenDirty();
 | 
						|
	}
 | 
						|
 | 
						|
	return CommandCost();
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Set group flag for a group and its sub-groups.
 | 
						|
 * @param g initial group.
 | 
						|
 * @param set 1 to set or 0 to clear protection.
 | 
						|
 */
 | 
						|
static void SetGroupFlag(Group *g, GroupFlags flag, bool set, bool children)
 | 
						|
{
 | 
						|
	if (set) {
 | 
						|
		SetBit(g->flags, flag);
 | 
						|
	} else {
 | 
						|
		ClrBit(g->flags, flag);
 | 
						|
	}
 | 
						|
 | 
						|
	if (!children) return;
 | 
						|
 | 
						|
	IterateDescendantsOfGroup(g, [&](Group *pg) {
 | 
						|
		SetGroupFlag(pg, flag, set, false);
 | 
						|
	});
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * (Un)set group flag from a group
 | 
						|
 * @param tile unused
 | 
						|
 * @param flags type of operation
 | 
						|
 * @param p1   index of group array
 | 
						|
 * - p1 bit 0-15  : GroupID
 | 
						|
 * - p1 bit 16-18 : Flag to set, by value not bit.
 | 
						|
 * @param p2
 | 
						|
 * - p2 bit 0    : 1 to set or 0 to clear protection.
 | 
						|
 * - p2 bit 1    : 1 to apply to sub-groups.
 | 
						|
 * @param text unused
 | 
						|
 * @return the cost of this operation or an error
 | 
						|
 */
 | 
						|
CommandCost CmdSetGroupFlag(TileIndex tile, DoCommandFlag flags, uint32_t p1, uint32_t p2, const char *text)
 | 
						|
{
 | 
						|
	Group *g = Group::GetIfValid(GB(p1, 0, 16));
 | 
						|
	if (g == nullptr || g->owner != _current_company) return CMD_ERROR;
 | 
						|
 | 
						|
	/* GroupFlags are stored in as an 8 bit bitfield but passed here by value,
 | 
						|
	 * so 3 bits is sufficient to cover each possible value. */
 | 
						|
	GroupFlags flag = (GroupFlags)GB(p1, 16, 3);
 | 
						|
	if (flag >= GroupFlags::GF_END) return CMD_ERROR;
 | 
						|
 | 
						|
	if (flags & DC_EXEC) {
 | 
						|
		SetGroupFlag(g, flag, HasBit(p2, 0), HasBit(p2, 1));
 | 
						|
 | 
						|
		SetWindowDirty(GetWindowClassForVehicleType(g->vehicle_type), VehicleListIdentifier(VL_GROUP_LIST, g->vehicle_type, _current_company).Pack());
 | 
						|
		InvalidateWindowData(WC_REPLACE_VEHICLE, g->vehicle_type);
 | 
						|
	}
 | 
						|
 | 
						|
	return CommandCost();
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Affect the groupID of a train to new_g.
 | 
						|
 * @note called in CmdAddVehicleGroup and CmdMoveRailVehicle
 | 
						|
 * @param v     First vehicle of the chain.
 | 
						|
 * @param new_g index of array group
 | 
						|
 */
 | 
						|
void SetTrainGroupID(Train *v, GroupID new_g)
 | 
						|
{
 | 
						|
	if (!Group::IsValidID(new_g) && !IsDefaultGroupID(new_g)) return;
 | 
						|
 | 
						|
	assert(v->IsFrontEngine() || IsDefaultGroupID(new_g));
 | 
						|
 | 
						|
	for (Vehicle *u = v; u != nullptr; u = u->Next()) {
 | 
						|
		if (u->IsEngineCountable()) UpdateNumEngineGroup(u, u->group_id, new_g);
 | 
						|
 | 
						|
		u->group_id = new_g;
 | 
						|
		u->colourmap = PAL_NONE;
 | 
						|
		u->InvalidateNewGRFCache();
 | 
						|
		u->InvalidateImageCache();
 | 
						|
		u->UpdateViewport(true);
 | 
						|
	}
 | 
						|
 | 
						|
	/* Update the Replace Vehicle Windows */
 | 
						|
	GroupStatistics::UpdateAutoreplace(v->owner);
 | 
						|
	SetWindowDirty(WC_REPLACE_VEHICLE, VEH_TRAIN);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
 * Recalculates the groupID of a train. Should be called each time a vehicle is added
 | 
						|
 * to/removed from the chain,.
 | 
						|
 * @note this needs to be called too for 'wagon chains' (in the depot, without an engine)
 | 
						|
 * @note Called in CmdBuildRailVehicle, CmdBuildRailWagon, CmdMoveRailVehicle, CmdSellRailWagon
 | 
						|
 * @param v First vehicle of the chain.
 | 
						|
 */
 | 
						|
void UpdateTrainGroupID(Train *v)
 | 
						|
{
 | 
						|
	assert(v->IsFrontEngine() || v->IsFreeWagon());
 | 
						|
 | 
						|
	GroupID new_g = v->IsFrontEngine() ? v->group_id : (GroupID)DEFAULT_GROUP;
 | 
						|
	for (Vehicle *u = v; u != nullptr; u = u->Next()) {
 | 
						|
		if (u->IsEngineCountable()) UpdateNumEngineGroup(u, u->group_id, new_g);
 | 
						|
 | 
						|
		u->group_id = new_g;
 | 
						|
		u->colourmap = PAL_NONE;
 | 
						|
		u->InvalidateNewGRFCache();
 | 
						|
		u->InvalidateImageCache();
 | 
						|
	}
 | 
						|
 | 
						|
	/* Update the Replace Vehicle Windows */
 | 
						|
	GroupStatistics::UpdateAutoreplace(v->owner);
 | 
						|
	SetWindowDirty(WC_REPLACE_VEHICLE, VEH_TRAIN);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get the number of engines with EngineID id_e in the group with GroupID
 | 
						|
 * id_g and its sub-groups.
 | 
						|
 * @param company The company the group belongs to
 | 
						|
 * @param id_g The GroupID of the group used
 | 
						|
 * @param id_e The EngineID of the engine to count
 | 
						|
 * @return The number of engines with EngineID id_e in the group
 | 
						|
 */
 | 
						|
uint GetGroupNumEngines(CompanyID company, GroupID id_g, EngineID id_e)
 | 
						|
{
 | 
						|
	uint count = 0;
 | 
						|
	const Engine *e = Engine::Get(id_e);
 | 
						|
	IterateDescendantsOfGroup(id_g, [&](Group *g) {
 | 
						|
		count += GroupStatistics::Get(company, g->index, e->type).GetNumEngines(id_e);
 | 
						|
	});
 | 
						|
	return count + GroupStatistics::Get(company, id_g, e->type).GetNumEngines(id_e);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get the number of vehicles in the group with GroupID
 | 
						|
 * id_g and its sub-groups.
 | 
						|
 * @param company The company the group belongs to
 | 
						|
 * @param id_g The GroupID of the group used
 | 
						|
 * @param type The vehicle type of the group
 | 
						|
 * @return The number of vehicles in the group
 | 
						|
 */
 | 
						|
uint GetGroupNumVehicle(CompanyID company, GroupID id_g, VehicleType type)
 | 
						|
{
 | 
						|
	uint count = 0;
 | 
						|
	IterateDescendantsOfGroup(id_g, [&](Group *g) {
 | 
						|
		count += GroupStatistics::Get(company, g->index, type).num_vehicle;
 | 
						|
	});
 | 
						|
	return count + GroupStatistics::Get(company, id_g, type).num_vehicle;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get the number of vehicles above profit minimum age in the group with GroupID
 | 
						|
 * id_g and its sub-groups.
 | 
						|
 * @param company The company the group belongs to
 | 
						|
 * @param id_g The GroupID of the group used
 | 
						|
 * @param type The vehicle type of the group
 | 
						|
 * @return The number of vehicles above profit minimum age in the group
 | 
						|
 */
 | 
						|
uint GetGroupNumVehicleMinAge(CompanyID company, GroupID id_g, VehicleType type)
 | 
						|
{
 | 
						|
	uint count = 0;
 | 
						|
	IterateDescendantsOfGroup(id_g, [&](Group *g) {
 | 
						|
		count += GroupStatistics::Get(company, g->index, type).num_vehicle_min_age;
 | 
						|
	});
 | 
						|
	return count + GroupStatistics::Get(company, id_g, type).num_vehicle_min_age;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get last year's profit of vehicles above minimum age
 | 
						|
 * for the group with GroupID id_g and its sub-groups.
 | 
						|
 * @param company The company the group belongs to
 | 
						|
 * @param id_g The GroupID of the group used
 | 
						|
 * @param type The vehicle type of the group
 | 
						|
 * @return Last year's profit of vehicles above minimum age for the group
 | 
						|
 */
 | 
						|
Money GetGroupProfitLastYearMinAge(CompanyID company, GroupID id_g, VehicleType type)
 | 
						|
{
 | 
						|
	Money sum = 0;
 | 
						|
	IterateDescendantsOfGroup(id_g, [&](Group *g) {
 | 
						|
		sum += GroupStatistics::Get(company, g->index, type).profit_last_year_min_age;
 | 
						|
	});
 | 
						|
	return sum + GroupStatistics::Get(company, id_g, type).profit_last_year_min_age;
 | 
						|
}
 | 
						|
 | 
						|
void RemoveAllGroupsForCompany(const CompanyID company)
 | 
						|
{
 | 
						|
	ReindexTemplateReplacementsRecursiveGuard guard;
 | 
						|
 | 
						|
	for (Group *g : Group::Iterate()) {
 | 
						|
		if (company == g->owner) {
 | 
						|
			DeleteTemplateReplacementsByGroupID(g);
 | 
						|
			delete g;
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
 * Test if GroupID group is a descendant of (or is) GroupID search
 | 
						|
 * @param search The GroupID to search in
 | 
						|
 * @param group The GroupID to search for
 | 
						|
 * @return True iff group is search or a descendant of search
 | 
						|
 */
 | 
						|
bool GroupIsInGroup(GroupID search, GroupID group)
 | 
						|
{
 | 
						|
	if (!Group::IsValidID(search)) return search == group;
 | 
						|
 | 
						|
	do {
 | 
						|
		if (search == group) return true;
 | 
						|
		search = Group::Get(search)->parent;
 | 
						|
	} while (search != INVALID_GROUP);
 | 
						|
 | 
						|
	return false;
 | 
						|
}
 |