cargomonitor.cpp

Go to the documentation of this file.

/*
 * 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/>.
 */
 
#include "stdafx.h"
#include "cargomonitor.h"
#include "station_base.h"
 
#include "safeguards.h"
 
CargoMonitorMap _cargo_pickups;    
CargoMonitorMap _cargo_deliveries; 
 
static void ClearCargoMonitoring(CargoMonitorMap &cargo_monitor_map, CompanyID company = INVALID_OWNER)
{
  if (company == INVALID_OWNER) {
    cargo_monitor_map.clear();
    return;
  }
 
  for (auto it = cargo_monitor_map.begin(); it != cargo_monitor_map.end(); /* nothing */) {
    if (DecodeMonitorCompany(it->first) == company) {
      it = cargo_monitor_map.erase(it);
    } else {
      ++it;
    }
  }
}
 
void ClearCargoPickupMonitoring(CompanyID company)
{
  ClearCargoMonitoring(_cargo_pickups, company);
}
 
void ClearCargoDeliveryMonitoring(CompanyID company)
{
  ClearCargoMonitoring(_cargo_deliveries, company);
}
 
static int32_t GetAmount(CargoMonitorMap &monitor_map, CargoMonitorID monitor, bool keep_monitoring)
{
  CargoMonitorMap::iterator iter = monitor_map.find(monitor);
  if (iter == monitor_map.end()) {
    if (keep_monitoring) {
      monitor_map.emplace(monitor, 0);
    }
    return 0;
  } else {
    int32_t result = iter->second;
    iter->second = 0;
    if (!keep_monitoring) monitor_map.erase(iter);
    return result;
  }
}
 
int32_t GetDeliveryAmount(CargoMonitorID monitor, bool keep_monitoring)
{
  return GetAmount(_cargo_deliveries, monitor, keep_monitoring);
}
 
int32_t GetPickupAmount(CargoMonitorID monitor, bool keep_monitoring)
{
  return GetAmount(_cargo_pickups, monitor, keep_monitoring);
}
 
void AddCargoDelivery(CargoType cargo_type, CompanyID company, uint32_t amount, Source src, const Station *st, IndustryID dest)
{
  if (amount == 0) return;
 
  if (src.IsValid()) {
    /* Handle pickup update. */
    switch (src.type) {
      case SourceType::Industry: {
        CargoMonitorID num = EncodeCargoIndustryMonitor(company, cargo_type, src.ToIndustryID());
        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.ToTownID());
        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;
  }
}