# Conflicts: # src/animated_tile.cpp # src/cargopacket.h # src/cheat_gui.cpp # src/company_cmd.cpp # src/company_gui.cpp # src/date.cpp # src/disaster_vehicle.cpp # src/dock_gui.cpp # src/economy.cpp # src/engine.cpp # src/error_gui.cpp # src/fontcache/spritefontcache.cpp # src/game/game_gui.cpp # src/game/game_text.cpp # src/gfx.cpp # src/graph_gui.cpp # src/highscore_gui.cpp # src/industry_cmd.cpp # src/lang/dutch.txt # src/lang/english_AU.txt # src/lang/english_US.txt # src/lang/finnish.txt # src/lang/french.txt # src/lang/italian.txt # src/lang/portuguese.txt # src/lang/russian.txt # src/lang/turkish.txt # src/lang/vietnamese.txt # src/main_gui.cpp # src/misc_gui.cpp # src/network/network_gui.cpp # src/network/network_server.cpp # src/newgrf.cpp # src/newgrf.h # src/newgrf_generic.cpp # src/news_gui.cpp # src/openttd.cpp # src/os/unix/unix.cpp # src/os/windows/font_win32.cpp # src/os/windows/win32.cpp # src/rail_gui.cpp # src/road_gui.cpp # src/saveload/afterload.cpp # src/saveload/misc_sl.cpp # src/saveload/oldloader_sl.cpp # src/saveload/saveload.cpp # src/saveload/saveload.h # src/script/script_gui.cpp # src/settings_table.cpp # src/signs_gui.cpp # src/smallmap_gui.cpp # src/smallmap_gui.h # src/spritecache.cpp # src/spritecache.h # src/spriteloader/grf.cpp # src/station_cmd.cpp # src/statusbar_gui.cpp # src/stdafx.h # src/strgen/strgen_base.cpp # src/subsidy.cpp # src/table/settings/difficulty_settings.ini # src/texteff.cpp # src/timetable_cmd.cpp # src/timetable_gui.cpp # src/toolbar_gui.cpp # src/town_cmd.cpp # src/town_gui.cpp # src/townname.cpp # src/vehicle.cpp # src/waypoint_cmd.cpp # src/widgets/dropdown.cpp # src/window.cpp
		
			
				
	
	
		
			159 lines
		
	
	
		
			6.0 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			159 lines
		
	
	
		
			6.0 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 cargomonitor.cpp Implementation of the cargo transport monitoring. */
 | 
						|
 | 
						|
#include "stdafx.h"
 | 
						|
#include "cargomonitor.h"
 | 
						|
#include "station_base.h"
 | 
						|
 | 
						|
#include "safeguards.h"
 | 
						|
 | 
						|
CargoMonitorMap _cargo_pickups;    ///< Map of monitored pick-ups   to the amount since last query/activation.
 | 
						|
CargoMonitorMap _cargo_deliveries; ///< Map of monitored deliveries to the amount since last query/activation.
 | 
						|
 | 
						|
/**
 | 
						|
 * Helper method for #ClearCargoPickupMonitoring and #ClearCargoDeliveryMonitoring.
 | 
						|
 * Clears all monitors that belong to the specified company or all if #INVALID_OWNER
 | 
						|
 * is specified as company.
 | 
						|
 * @param cargo_monitor_map reference to the cargo monitor map to operate on.
 | 
						|
 * @param company company to clear cargo monitors for or #INVALID_OWNER if all cargo monitors should be cleared.
 | 
						|
 */
 | 
						|
static void ClearCargoMonitoring(CargoMonitorMap &cargo_monitor_map, CompanyID company = INVALID_OWNER)
 | 
						|
{
 | 
						|
	if (company == INVALID_OWNER) {
 | 
						|
		cargo_monitor_map.clear();
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	for (CargoMonitorMap::iterator it = cargo_monitor_map.begin(); it != cargo_monitor_map.end();) {
 | 
						|
		if (DecodeMonitorCompany(it->first) == company) {
 | 
						|
			it = cargo_monitor_map.erase(it);
 | 
						|
		} else {
 | 
						|
			++it;
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Clear all pick-up cargo monitors.
 | 
						|
 * @param company clear all pick-up monitors for this company or if #INVALID_OWNER
 | 
						|
 * is passed, all pick-up monitors are cleared regardless of company.
 | 
						|
 */
 | 
						|
void ClearCargoPickupMonitoring(CompanyID company)
 | 
						|
{
 | 
						|
	ClearCargoMonitoring(_cargo_pickups, company);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Clear all delivery cargo monitors.
 | 
						|
 * @param company clear all delivery monitors for this company or if #INVALID_OWNER
 | 
						|
 * is passed, all delivery monitors are cleared regardless of company.
 | 
						|
 */
 | 
						|
void ClearCargoDeliveryMonitoring(CompanyID company)
 | 
						|
{
 | 
						|
	ClearCargoMonitoring(_cargo_deliveries, company);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get and reset the amount associated with a cargo monitor.
 | 
						|
 * @param[in,out] monitor_map Monitoring map to search (and reset for the queried entry).
 | 
						|
 * @param monitor Cargo monitor to query/reset.
 | 
						|
 * @param keep_monitoring After returning from this call, continue monitoring.
 | 
						|
 * @return Amount collected since last query/activation for the monitored combination.
 | 
						|
 */
 | 
						|
static int32 GetAmount(CargoMonitorMap &monitor_map, CargoMonitorID monitor, bool keep_monitoring)
 | 
						|
{
 | 
						|
	CargoMonitorMap::iterator iter = monitor_map.find(monitor);
 | 
						|
	if (iter == monitor_map.end()) {
 | 
						|
		if (keep_monitoring) {
 | 
						|
			std::pair<CargoMonitorID, uint32> p(monitor, 0);
 | 
						|
			monitor_map.insert(p);
 | 
						|
		}
 | 
						|
		return 0;
 | 
						|
	} else {
 | 
						|
		int32 result = iter->second;
 | 
						|
		iter->second = 0;
 | 
						|
		if (!keep_monitoring) monitor_map.erase(iter);
 | 
						|
		return result;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get the amount of cargo delivered for the given cargo monitor since activation or last query.
 | 
						|
 * @param monitor Cargo monitor to query.
 | 
						|
 * @param keep_monitoring After returning from this call, continue monitoring.
 | 
						|
 * @return Amount of delivered cargo for the monitored combination.
 | 
						|
 */
 | 
						|
int32 GetDeliveryAmount(CargoMonitorID monitor, bool keep_monitoring)
 | 
						|
{
 | 
						|
	return GetAmount(_cargo_deliveries, monitor, keep_monitoring);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Get the amount of cargo picked up for the given cargo monitor since activation or last query.
 | 
						|
 * @param monitor Monitoring number to query.
 | 
						|
 * @param keep_monitoring After returning from this call, continue monitoring.
 | 
						|
 * @return Amount of picked up cargo for the monitored combination.
 | 
						|
 * @note Cargo pick up is counted on final delivery, to prevent users getting credit for picking up cargo without delivering it.
 | 
						|
 */
 | 
						|
int32 GetPickupAmount(CargoMonitorID monitor, bool keep_monitoring)
 | 
						|
{
 | 
						|
	return GetAmount(_cargo_pickups, monitor, keep_monitoring);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Cargo was delivered to its final destination, update the pickup and delivery maps.
 | 
						|
 * @param cargo_type type of cargo.
 | 
						|
 * @param company company delivering the cargo.
 | 
						|
 * @param amount Amount of cargo delivered.
 | 
						|
 * @param src_type type of \a src.
 | 
						|
 * @param src index of source.
 | 
						|
 * @param st station where the cargo is delivered to.
 | 
						|
 * @param dest industry index where the cargo is delivered to.
 | 
						|
 */
 | 
						|
void AddCargoDelivery(CargoID cargo_type, CompanyID company, uint32 amount, SourceType src_type, SourceID src, const Station *st, IndustryID dest)
 | 
						|
{
 | 
						|
	if (amount == 0) return;
 | 
						|
 | 
						|
	if (src != INVALID_SOURCE) {
 | 
						|
		/* Handle pickup update. */
 | 
						|
		switch (src_type) {
 | 
						|
			case SourceType::Industry: {
 | 
						|
				CargoMonitorID num = EncodeCargoIndustryMonitor(company, cargo_type, src);
 | 
						|
				CargoMonitorMap::iterator iter = _cargo_pickups.find(num);
 | 
						|
				if (iter != _cargo_pickups.end()) iter->second += amount;
 | 
						|
				break;
 | 
						|
			}
 | 
						|
			case SourceType::Town: {
 | 
						|
				CargoMonitorID num = EncodeCargoTownMonitor(company, cargo_type, src);
 | 
						|
				CargoMonitorMap::iterator iter = _cargo_pickups.find(num);
 | 
						|
				if (iter != _cargo_pickups.end()) iter->second += amount;
 | 
						|
				break;
 | 
						|
			}
 | 
						|
			default: break;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	/* Handle delivery.
 | 
						|
	 * Note that delivery in the right area is sufficient to prevent trouble with neighbouring industries or houses. */
 | 
						|
 | 
						|
	/* Town delivery. */
 | 
						|
	CargoMonitorID num = EncodeCargoTownMonitor(company, cargo_type, st->town->index);
 | 
						|
	CargoMonitorMap::iterator iter = _cargo_deliveries.find(num);
 | 
						|
	if (iter != _cargo_deliveries.end()) iter->second += amount;
 | 
						|
 | 
						|
	/* Industry delivery. */
 | 
						|
	for (const auto &i : st->industries_near) {
 | 
						|
		if (i.industry->index != dest) continue;
 | 
						|
		CargoMonitorID num = EncodeCargoIndustryMonitor(company, cargo_type, i.industry->index);
 | 
						|
		CargoMonitorMap::iterator iter = _cargo_deliveries.find(num);
 | 
						|
		if (iter != _cargo_deliveries.end()) iter->second += amount;
 | 
						|
	}
 | 
						|
}
 | 
						|
 |