rubidium@6643: /* $Id$ */
rubidium@6643: 
rubidium@6643: /** @file group.h */
rubidium@6643: 
rubidium@6643: #ifndef GROUP_H
rubidium@6643: #define GROUP_H
rubidium@6643: 
rubidium@6643: #include "oldpool.h"
rubidium@6643: 
rubidium@6643: enum {
rubidium@7258: 	ALL_GROUP     = 0xFFFD,
rubidium@6643: 	DEFAULT_GROUP = 0xFFFE,
rubidium@6643: 	INVALID_GROUP = 0xFFFF,
rubidium@6643: };
rubidium@6643: 
rubidium@7382: struct Group;
rubidium@7382: DECLARE_OLD_POOL(Group, Group, 5, 2047)
rubidium@7382: 
rubidium@7382: struct Group : PoolItem<Group, GroupID, &_Group_pool> {
rubidium@6643: 	StringID string_id;                     ///< Group Name
rubidium@6643: 
rubidium@6643: 	uint16 num_vehicle;                     ///< Number of vehicles wich belong to the group
rubidium@6643: 	PlayerID owner;                         ///< Group Owner
rubidium@6643: 	VehicleTypeByte vehicle_type;           ///< Vehicle type of the group
rubidium@6643: 
rubidium@6643: 	bool replace_protection;                ///< If set to true, the global autoreplace have no effect on the group
rubidium@6643: 	uint16 num_engines[TOTAL_NUM_ENGINES];  ///< Caches the number of engines of each type the player owns (no need to save this)
rubidium@7382: 
rubidium@7382: 	Group(StringID str = STR_NULL);
rubidium@7382: 	virtual ~Group();
rubidium@7382: 
rubidium@7382: 	bool IsValid() const;
rubidium@6643: };
rubidium@6643: 
rubidium@6643: 
rubidium@6643: static inline bool IsValidGroupID(GroupID index)
rubidium@6643: {
rubidium@7382: 	return index < GetGroupPoolSize() && GetGroup(index)->IsValid();
rubidium@6643: }
rubidium@6643: 
rubidium@6643: static inline bool IsDefaultGroupID(GroupID index)
rubidium@6643: {
rubidium@7258: 	return index == DEFAULT_GROUP;
rubidium@7258: }
rubidium@7258: 
rubidium@7258: /**
rubidium@7258:  * Checks if a GroupID stands for all vehicles of a player
rubidium@7258:  * @param id_g The GroupID to check
rubidium@7258:  * @return true is id_g is identical to ALL_GROUP
rubidium@7258:  */
rubidium@7258: static inline bool IsAllGroupID(GroupID id_g)
rubidium@7258: {
rubidium@7258: 	return id_g == ALL_GROUP;
rubidium@6643: }
rubidium@6643: 
rubidium@7382: #define FOR_ALL_GROUPS_FROM(g, start) for (g = GetGroup(start); g != NULL; g = (g->index + 1U < GetGroupPoolSize()) ? GetGroup(g->index + 1) : NULL) if (g->IsValid())
rubidium@6643: #define FOR_ALL_GROUPS(g) FOR_ALL_GROUPS_FROM(g, 0)
rubidium@6643: 
rubidium@6643: /**
rubidium@6643:  * Get the current size of the GroupPool
rubidium@6643:  */
rubidium@6643: static inline uint GetGroupArraySize(void)
rubidium@6643: {
rubidium@6643: 	const Group *g;
rubidium@6643: 	uint num = 0;
rubidium@6643: 
rubidium@6643: 	FOR_ALL_GROUPS(g) num++;
rubidium@6643: 
rubidium@6643: 	return num;
rubidium@6643: }
rubidium@6643: 
rubidium@7258: /**
rubidium@7258:  * Get the number of engines with EngineID id_e in the group with GroupID
rubidium@7258:  * id_g
rubidium@7258:  * @param id_g The GroupID of the group used
rubidium@7258:  * @param id_e The EngineID of the engine to count
rubidium@7258:  * @return The number of engines with EngineID id_e in the group
rubidium@7258:  */
rubidium@7481: static inline uint GetGroupNumEngines(PlayerID p, GroupID id_g, EngineID id_e)
rubidium@7258: {
rubidium@7258: 	if (IsValidGroupID(id_g)) return GetGroup(id_g)->num_engines[id_e];
rubidium@7258: 
rubidium@7481: 	uint num = GetPlayer(p)->num_engines[id_e];
rubidium@7258: 	if (!IsDefaultGroupID(id_g)) return num;
rubidium@7258: 
rubidium@7258: 	const Group *g;
rubidium@7481: 	FOR_ALL_GROUPS(g) {
rubidium@7481: 		if (g->owner == p) num -= g->num_engines[id_e];
rubidium@7481: 	}
rubidium@7258: 	return num;
rubidium@7258: }
rubidium@7258: 
rubidium@6643: static inline void IncreaseGroupNumVehicle(GroupID id_g)
rubidium@6643: {
rubidium@6643: 	if (IsValidGroupID(id_g)) GetGroup(id_g)->num_vehicle++;
rubidium@6643: }
rubidium@6643: 
rubidium@6643: static inline void DecreaseGroupNumVehicle(GroupID id_g)
rubidium@6643: {
rubidium@6643: 	if (IsValidGroupID(id_g)) GetGroup(id_g)->num_vehicle--;
rubidium@6643: }
rubidium@6643: 
rubidium@6643: 
rubidium@6643: void InitializeGroup();
rubidium@6643: void SetTrainGroupID(Vehicle *v, GroupID grp);
rubidium@6643: void UpdateTrainGroupID(Vehicle *v);
rubidium@6643: void RemoveVehicleFromGroup(const Vehicle *v);
rubidium@6643: void RemoveAllGroupsForPlayer(const Player *p);
rubidium@6643: 
rubidium@6643: #endif /* GROUP_H */