120 lines
3.6 KiB
C
120 lines
3.6 KiB
C
/* $Id$ */
|
|
|
|
/*
|
|
* 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.h Base class for groups and group functions. */
|
|
|
|
#ifndef GROUP_H
|
|
#define GROUP_H
|
|
|
|
#include "group_type.h"
|
|
#include "core/pool_type.hpp"
|
|
#include "company_type.h"
|
|
#include "vehicle_type.h"
|
|
#include "engine_type.h"
|
|
|
|
typedef Pool<Group, GroupID, 16, 64000> GroupPool;
|
|
extern GroupPool _group_pool; ///< Pool of groups.
|
|
|
|
/** Statistics and caches on the vehicles in a group. */
|
|
struct GroupStatistics {
|
|
uint16 num_vehicle; ///< Number of vehicles.
|
|
uint16 *num_engines; ///< Caches the number of engines of each type the company owns.
|
|
|
|
bool autoreplace_defined; ///< Are any autoreplace rules set?
|
|
bool autoreplace_finished; ///< Have all autoreplacement finished?
|
|
|
|
uint16 num_profit_vehicle; ///< Number of vehicles considered for profit statistics;
|
|
Money profit_last_year; ///< Sum of profits for all vehicles.
|
|
|
|
GroupStatistics();
|
|
~GroupStatistics();
|
|
|
|
void Clear();
|
|
|
|
void ClearProfits()
|
|
{
|
|
this->num_profit_vehicle = 0;
|
|
this->profit_last_year = 0;
|
|
}
|
|
|
|
void ClearAutoreplace()
|
|
{
|
|
this->autoreplace_defined = false;
|
|
this->autoreplace_finished = false;
|
|
}
|
|
|
|
static GroupStatistics &Get(CompanyID company, GroupID id_g, VehicleType type);
|
|
static GroupStatistics &Get(const Vehicle *v);
|
|
static GroupStatistics &GetAllGroup(const Vehicle *v);
|
|
|
|
static void CountVehicle(const Vehicle *v, int delta);
|
|
static void CountEngine(const Vehicle *v, int delta);
|
|
static void VehicleReachedProfitAge(const Vehicle *v);
|
|
|
|
static void UpdateProfits();
|
|
static void UpdateAfterLoad();
|
|
static void UpdateAutoreplace(CompanyID company);
|
|
};
|
|
|
|
/** Group data. */
|
|
struct Group : GroupPool::PoolItem<&_group_pool> {
|
|
char *name; ///< Group Name
|
|
OwnerByte owner; ///< Group Owner
|
|
VehicleTypeByte vehicle_type; ///< Vehicle type of the group
|
|
|
|
bool replace_protection; ///< If set to true, the global autoreplace have no effect on the group
|
|
GroupStatistics statistics; ///< NOSAVE: Statistics and caches on the vehicles in the group.
|
|
|
|
Group(CompanyID owner = INVALID_COMPANY);
|
|
~Group();
|
|
};
|
|
|
|
|
|
static inline bool IsDefaultGroupID(GroupID index)
|
|
{
|
|
return index == DEFAULT_GROUP;
|
|
}
|
|
|
|
/**
|
|
* Checks if a GroupID stands for all vehicles of a company
|
|
* @param id_g The GroupID to check
|
|
* @return true is id_g is identical to ALL_GROUP
|
|
*/
|
|
static inline bool IsAllGroupID(GroupID id_g)
|
|
{
|
|
return id_g == ALL_GROUP;
|
|
}
|
|
|
|
#define FOR_ALL_GROUPS_FROM(var, start) FOR_ALL_ITEMS_FROM(Group, group_index, var, start)
|
|
#define FOR_ALL_GROUPS(var) FOR_ALL_GROUPS_FROM(var, 0)
|
|
|
|
/**
|
|
* Get the current size of the GroupPool
|
|
*/
|
|
static inline uint GetGroupArraySize()
|
|
{
|
|
const Group *g;
|
|
uint num = 0;
|
|
|
|
FOR_ALL_GROUPS(g) num++;
|
|
|
|
return num;
|
|
}
|
|
|
|
uint GetGroupNumEngines(CompanyID company, GroupID id_g, EngineID id_e);
|
|
|
|
void SetTrainGroupID(Train *v, GroupID grp);
|
|
void UpdateTrainGroupID(Train *v);
|
|
void RemoveVehicleFromGroup(const Vehicle *v);
|
|
void RemoveAllGroupsForCompany(const CompanyID company);
|
|
|
|
extern GroupID _new_group_id;
|
|
|
|
#endif /* GROUP_H */
|