Function GetCompanyRailTypes

src/rail.cpp:135–159

* Get the rail types the given company can build. * @param company the company to get the rail types for. * @param introduces If true, include rail types introduced by other rail types * @return the rail types. */

 * @return the rail types.
 */
RailTypes GetCompanyRailTypes(CompanyID company, bool introduces)
{
	RailTypes rts{};

	for (const Engine *e : Engine::IterateType(VEH_TRAIN)) {
		const EngineInfo *ei = &e->info;

		if (ei->climates.Test(_settings_game.game_creation.landscape) &&
				(e->company_avail.Test(company) || TimerGameCalendar::date >= e->intro_date + CalendarTime::DAYS_IN_YEAR)) {
			const RailVehicleInfo *rvi = &e->VehInfo<RailVehicleInfo>();

			if (rvi->railveh_type != RAILVEH_WAGON) {
				assert(rvi->railtypes.Any());
				if (introduces) {
					rts.Set(GetAllIntroducesRailTypes(rvi->railtypes));
				} else {
					rts.Set(rvi->railtypes);
				}
			}
		}
	}

	if (introduces) return AddDateIntroducedRailTypes(rts, TimerGameCalendar::date);
	return rts;
}

/**
 * Get list of rail types, regardless of company availability.