refresh.h

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/>.
 */
 
#ifndef REFRESH_H
#define REFRESH_H
 
#include "../cargo_type.h"
#include "../vehicle_base.h"
 
class LinkRefresher {
public:
  static void Run(Vehicle *v, bool allow_merge = true, bool is_full_loading = false);
 
protected:
  enum RefreshFlags {
    USE_NEXT,     
    HAS_CARGO,    
    WAS_REFIT,    
    RESET_REFIT,  
    IN_AUTOREFIT, 
  };
 
  struct RefitDesc {
    CargoID cargo;    
    uint16_t capacity;  
    uint16_t remaining; 
    RefitDesc(CargoID cargo, uint16_t capacity, uint16_t remaining) :
        cargo(cargo), capacity(capacity), remaining(remaining) {}
  };
 
  struct Hop {
    OrderID from;  
    OrderID to;    
    CargoID cargo; 
 
    Hop() {NOT_REACHED();}
 
    Hop(OrderID from, OrderID to, CargoID cargo) : from(from), to(to), cargo(cargo) {}
    bool operator<(const Hop &other) const;
  };
 
  typedef std::vector<RefitDesc> RefitList;
  typedef std::set<Hop> HopSet;
 
  Vehicle *vehicle;           
  uint capacities[NUM_CARGO]; 
  RefitList refit_capacities; 
  HopSet *seen_hops;          
  CargoID cargo;              
  bool allow_merge;           
  bool is_full_loading;       
 
  LinkRefresher(Vehicle *v, HopSet *seen_hops, bool allow_merge, bool is_full_loading);
 
  bool HandleRefit(CargoID refit_cargo);
  void ResetRefit();
  void RefreshStats(const Order *cur, const Order *next);
  const Order *PredictNextOrder(const Order *cur, const Order *next, uint8_t flags, uint num_hops = 0);
 
  void RefreshLinks(const Order *cur, const Order *next, uint8_t flags, uint num_hops = 0);
};
 
#endif /* REFRESH_H */