4 * This file is part of OpenTTD.
5 * 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.
6 * 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.
7 * 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/>.
10 /** @file cargopacket.cpp Implementation of the cargo packets. */
13 #include "station_base.h"
14 #include "core/pool_func.hpp"
15 #include "core/random_func.hpp"
16 #include "economy_base.h"
17 #include "cargoaction.h"
18 #include "order_type.h"
22 #include "safeguards.h"
24 /* Initialize the cargopacket-pool */
25 CargoPacketPool
_cargopacket_pool("CargoPacket");
26 INSTANTIATE_POOL_METHODS(CargoPacket
)
29 * Create a new packet for savegame loading.
31 CargoPacket::CargoPacket()
33 this->source_type
= ST_INDUSTRY
;
34 this->source_id
= INVALID_SOURCE
;
38 * Creates a new cargo packet.
39 * @param source Source station of the packet.
40 * @param source_xy Source location of the packet.
41 * @param count Number of cargo entities to put in this packet.
42 * @param source_type 'Type' of source the packet comes from (for subsidies).
43 * @param source_id Actual source of the packet (for subsidies).
45 * @note We have to zero memory ourselves here because we are using a 'new'
46 * that, in contrary to all other pools, does not memset to 0.
48 CargoPacket::CargoPacket(StationID source
, TileIndex source_xy
, uint16 count
, SourceType source_type
, SourceID source_id
) :
58 this->source_type
= source_type
;
62 * Creates a new cargo packet. Initializes the fields that cannot be changed later.
63 * Used when loading or splitting packets.
64 * @param count Number of cargo entities to put in this packet.
65 * @param days_in_transit Number of days the cargo has been in transit.
66 * @param source Station the cargo was initially loaded.
67 * @param source_xy Station location the cargo was initially loaded.
68 * @param loaded_at_xy Location the cargo was loaded last.
69 * @param feeder_share Feeder share the packet has already accumulated.
70 * @param source_type 'Type' of source the packet comes from (for subsidies).
71 * @param source_id Actual source of the packet (for subsidies).
72 * @note We have to zero memory ourselves here because we are using a 'new'
73 * that, in contrary to all other pools, does not memset to 0.
75 CargoPacket::CargoPacket(uint16 count
, byte days_in_transit
, StationID source
, TileIndex source_xy
, TileIndex loaded_at_xy
, Money feeder_share
, SourceType source_type
, SourceID source_id
) :
76 feeder_share(feeder_share
),
78 days_in_transit(days_in_transit
),
82 loaded_at_xy(loaded_at_xy
)
85 this->source_type
= source_type
;
89 * Split this packet in two and return the split off part.
90 * @param new_size Size of the split part.
91 * @return Split off part, or nullptr if no packet could be allocated!
93 CargoPacket
*CargoPacket::Split(uint new_size
)
95 if (!CargoPacket::CanAllocateItem()) return nullptr;
97 Money fs
= this->FeederShare(new_size
);
98 CargoPacket
*cp_new
= new CargoPacket(new_size
, this->days_in_transit
, this->source
, this->source_xy
, this->loaded_at_xy
, fs
, this->source_type
, this->source_id
);
99 this->feeder_share
-= fs
;
100 this->count
-= new_size
;
105 * Merge another packet into this one.
106 * @param cp Packet to be merged in.
108 void CargoPacket::Merge(CargoPacket
*cp
)
110 this->count
+= cp
->count
;
111 this->feeder_share
+= cp
->feeder_share
;
116 * Reduce the packet by the given amount and remove the feeder share.
117 * @param count Amount to be removed.
119 void CargoPacket::Reduce(uint count
)
121 assert(count
< this->count
);
122 this->feeder_share
-= this->FeederShare(count
);
123 this->count
-= count
;
127 * Invalidates (sets source_id to INVALID_SOURCE) all cargo packets from given source.
128 * @param src_type Type of source.
129 * @param src Index of source.
131 /* static */ void CargoPacket::InvalidateAllFrom(SourceType src_type
, SourceID src
)
134 FOR_ALL_CARGOPACKETS(cp
) {
135 if (cp
->source_type
== src_type
&& cp
->source_id
== src
) cp
->source_id
= INVALID_SOURCE
;
140 * Invalidates (sets source to INVALID_STATION) all cargo packets from given station.
141 * @param sid Station that gets removed.
143 /* static */ void CargoPacket::InvalidateAllFrom(StationID sid
)
146 FOR_ALL_CARGOPACKETS(cp
) {
147 if (cp
->source
== sid
) cp
->source
= INVALID_STATION
;
153 * Cargo list implementation
158 * Destroy the cargolist ("frees" all cargo packets).
160 template <class Tinst
, class Tcont
>
161 CargoList
<Tinst
, Tcont
>::~CargoList()
163 for (Iterator
it(this->packets
.begin()); it
!= this->packets
.end(); ++it
) {
169 * Empty the cargo list, but don't free the cargo packets;
170 * the cargo packets are cleaned by CargoPacket's CleanPool.
172 template <class Tinst
, class Tcont
>
173 void CargoList
<Tinst
, Tcont
>::OnCleanPool()
175 this->packets
.clear();
179 * Update the cached values to reflect the removal of this packet or part of it.
180 * Decreases count and days_in_transit.
181 * @param cp Packet to be removed from cache.
182 * @param count Amount of cargo from the given packet to be removed.
184 template <class Tinst
, class Tcont
>
185 void CargoList
<Tinst
, Tcont
>::RemoveFromCache(const CargoPacket
*cp
, uint count
)
187 assert(count
<= cp
->count
);
188 this->count
-= count
;
189 this->cargo_days_in_transit
-= cp
->days_in_transit
* count
;
193 * Update the cache to reflect adding of this packet.
194 * Increases count and days_in_transit.
195 * @param cp New packet to be inserted.
197 template <class Tinst
, class Tcont
>
198 void CargoList
<Tinst
, Tcont
>::AddToCache(const CargoPacket
*cp
)
200 this->count
+= cp
->count
;
201 this->cargo_days_in_transit
+= cp
->days_in_transit
* cp
->count
;
204 /** Invalidates the cached data and rebuilds it. */
205 template <class Tinst
, class Tcont
>
206 void CargoList
<Tinst
, Tcont
>::InvalidateCache()
209 this->cargo_days_in_transit
= 0;
211 for (ConstIterator
it(this->packets
.begin()); it
!= this->packets
.end(); it
++) {
212 static_cast<Tinst
*>(this)->AddToCache(*it
);
217 * Tries to merge the second packet into the first and return if that was
219 * @param icp Packet to be merged into.
220 * @param cp Packet to be eliminated.
221 * @return If the packets could be merged.
223 template <class Tinst
, class Tcont
>
224 /* static */ bool CargoList
<Tinst
, Tcont
>::TryMerge(CargoPacket
*icp
, CargoPacket
*cp
)
226 if (Tinst::AreMergable(icp
, cp
) &&
227 icp
->count
+ cp
->count
<= CargoPacket::MAX_COUNT
) {
237 * Vehicle cargo list implementation.
242 * Appends the given cargo packet. Tries to merge it with another one in the
243 * packets list. If no fitting packet is found, appends it. You can only append
244 * packets to the ranges of packets designated for keeping or loading.
245 * Furthermore if there are already packets reserved for loading you cannot
246 * directly add packets to the "keep" list. You first have to load the reserved
248 * @warning After appending this packet may not exist anymore!
249 * @note Do not use the cargo packet anymore after it has been appended to this CargoList!
250 * @param cp Cargo packet to add.
251 * @param action Either MTA_KEEP if you want to add the packet directly or MTA_LOAD
252 * if you want to reserve it first.
254 * @pre action == MTA_LOAD || (action == MTA_KEEP && this->designation_counts[MTA_LOAD] == 0)
256 void VehicleCargoList::Append(CargoPacket
*cp
, MoveToAction action
)
258 assert(cp
!= nullptr);
259 assert(action
== MTA_LOAD
||
260 (action
== MTA_KEEP
&& this->action_counts
[MTA_LOAD
] == 0));
261 this->AddToMeta(cp
, action
);
263 if (this->count
== cp
->count
) {
264 this->packets
.push_back(cp
);
268 uint sum
= cp
->count
;
269 for (ReverseIterator
it(this->packets
.rbegin()); it
!= this->packets
.rend(); it
++) {
270 CargoPacket
*icp
= *it
;
271 if (VehicleCargoList::TryMerge(icp
, cp
)) return;
273 if (sum
>= this->action_counts
[action
]) {
274 this->packets
.push_back(cp
);
283 * Shifts cargo from the front of the packet list and applies some action to it.
284 * @tparam Taction Action class or function to be used. It should define
285 * "bool operator()(CargoPacket *)". If true is returned the
286 * cargo packet will be removed from the list. Otherwise it
287 * will be kept and the loop will be aborted.
288 * @param action Action instance to be applied.
290 template<class Taction
>
291 void VehicleCargoList::ShiftCargo(Taction action
)
293 std::vector
<CargoPacket
*> packets_to_erase
;
294 std::vector
<CargoPacket
*> packets_copy(this->packets
.begin(), this->packets
.end());
296 auto it
= packets_copy
.begin();
297 while (it
!= packets_copy
.end() && action
.MaxMove() > 0) {
298 CargoPacket
*cp
= *it
;
300 packets_to_erase
.push_back(*it
);
307 for (auto cp
: packets_to_erase
) {
308 this->packets
.erase(std::find(this->packets
.begin(), this->packets
.end(), cp
));
313 * Pops cargo from the back of the packet list and applies some action to it.
314 * @tparam Taction Action class or function to be used. It should define
315 * "bool operator()(CargoPacket *)". If true is returned the
316 * cargo packet will be removed from the list. Otherwise it
317 * will be kept and the loop will be aborted.
318 * @param action Action instance to be applied.
320 template<class Taction
>
321 void VehicleCargoList::PopCargo(Taction action
)
323 if (this->packets
.empty()) return;
324 for (auto it
= this->packets
.end(); it
!= this->packets
.begin();) {
325 if (action
.MaxMove() <= 0) break;
327 CargoPacket
*cp
= *it
;
329 it
= this->packets
.erase(it
);
337 * Update the cached values to reflect the removal of this packet or part of it.
338 * Decreases count, feeder share and days_in_transit.
339 * @param cp Packet to be removed from cache.
340 * @param count Amount of cargo from the given packet to be removed.
342 void VehicleCargoList::RemoveFromCache(const CargoPacket
*cp
, uint count
)
344 this->feeder_share
-= cp
->FeederShare(count
);
345 this->Parent::RemoveFromCache(cp
, count
);
349 * Update the cache to reflect adding of this packet.
350 * Increases count, feeder share and days_in_transit.
351 * @param cp New packet to be inserted.
353 void VehicleCargoList::AddToCache(const CargoPacket
*cp
)
355 this->feeder_share
+= cp
->feeder_share
;
356 this->Parent::AddToCache(cp
);
360 * Removes a packet or part of it from the metadata.
361 * @param cp Packet to be removed.
362 * @param action MoveToAction of the packet (for updating the counts).
363 * @param count Amount of cargo to be removed.
365 void VehicleCargoList::RemoveFromMeta(const CargoPacket
*cp
, MoveToAction action
, uint count
)
367 assert(count
<= this->action_counts
[action
]);
368 this->AssertCountConsistency();
369 this->RemoveFromCache(cp
, count
);
370 this->action_counts
[action
] -= count
;
371 this->AssertCountConsistency();
375 * Adds a packet to the metadata.
376 * @param cp Packet to be added.
377 * @param action MoveToAction of the packet.
379 void VehicleCargoList::AddToMeta(const CargoPacket
*cp
, MoveToAction action
)
381 this->AssertCountConsistency();
382 this->AddToCache(cp
);
383 this->action_counts
[action
] += cp
->count
;
384 this->AssertCountConsistency();
388 * Ages the all cargo in this list.
390 void VehicleCargoList::AgeCargo()
392 for (ConstIterator
it(this->packets
.begin()); it
!= this->packets
.end(); it
++) {
393 CargoPacket
*cp
= *it
;
394 /* If we're at the maximum, then we can't increase no more. */
395 if (cp
->days_in_transit
== 0xFF) continue;
397 cp
->days_in_transit
++;
398 this->cargo_days_in_transit
+= cp
->count
;
403 * Sets loaded_at_xy to the current station for all cargo to be transfered.
404 * This is done when stopping or skipping while the vehicle is unloading. In
405 * that case the vehicle will get part of its transfer credits early and it may
406 * get more transfer credits than it's entitled to.
407 * @param xy New loaded_at_xy for the cargo.
409 void VehicleCargoList::SetTransferLoadPlace(TileIndex xy
)
412 for (Iterator it
= this->packets
.begin(); sum
< this->action_counts
[MTA_TRANSFER
]; ++it
) {
413 CargoPacket
*cp
= *it
;
414 cp
->loaded_at_xy
= xy
;
420 * Choose action to be performed with the given cargo packet.
421 * @param cp The packet.
422 * @param cargo_next Next hop the cargo wants to pass.
423 * @param current_station Current station of the vehicle carrying the cargo.
424 * @param accepted If the cargo is accepted at the current station.
425 * @param next_station Next station(s) the vehicle may stop at.
426 * @return MoveToAction to be performed.
428 /* static */ VehicleCargoList::MoveToAction
VehicleCargoList::ChooseAction(const CargoPacket
*cp
, StationID cargo_next
,
429 StationID current_station
, bool accepted
, StationIDStack next_station
)
431 if (cargo_next
== INVALID_STATION
) {
432 return (accepted
&& cp
->source
!= current_station
) ? MTA_DELIVER
: MTA_KEEP
;
433 } else if (cargo_next
== current_station
) {
435 } else if (next_station
.Contains(cargo_next
)) {
443 * Stages cargo for unloading. The cargo is sorted so that packets to be
444 * transferred, delivered or kept are in consecutive chunks in the list. At the
445 * same time the designation_counts are updated to reflect the size of those
447 * @param accepted If the cargo will be accepted at the station.
448 * @param current_station ID of the station.
449 * @param next_station ID of the station the vehicle will go to next.
450 * @param order_flags OrderUnloadFlags that will apply to the unload operation.
451 * @param ge GoodsEntry for getting the flows.
452 * @param payment Payment object for registering transfers.
453 * return If any cargo will be unloaded.
455 bool VehicleCargoList::Stage(bool accepted
, StationID current_station
, StationIDStack next_station
, uint8 order_flags
, const GoodsEntry
*ge
, CargoPayment
*payment
)
457 this->AssertCountConsistency();
458 assert(this->action_counts
[MTA_LOAD
] == 0);
459 this->action_counts
[MTA_TRANSFER
] = this->action_counts
[MTA_DELIVER
] = this->action_counts
[MTA_KEEP
] = 0;
460 Iterator it
= this->packets
.begin();
462 CargoPacketList transfer_deliver
;
463 std::vector
<CargoPacket
*> keep
;
465 bool force_keep
= (order_flags
& OUFB_NO_UNLOAD
) != 0;
466 bool force_unload
= (order_flags
& OUFB_UNLOAD
) != 0;
467 bool force_transfer
= (order_flags
& (OUFB_TRANSFER
| OUFB_UNLOAD
)) != 0;
468 assert(this->count
> 0 || it
== this->packets
.end());
469 while (sum
< this->count
) {
470 CargoPacket
*cp
= *it
;
472 it
= this->packets
.erase(it
);
473 StationID cargo_next
= INVALID_STATION
;
474 MoveToAction action
= MTA_LOAD
;
477 } else if (force_unload
&& accepted
&& cp
->source
!= current_station
) {
478 action
= MTA_DELIVER
;
479 } else if (force_transfer
) {
480 action
= MTA_TRANSFER
;
481 /* We cannot send the cargo to any of the possible next hops and
482 * also not to the current station. */
483 FlowStatMap::const_iterator
flow_it(ge
->flows
.find(cp
->source
));
484 if (flow_it
== ge
->flows
.end()) {
485 cargo_next
= INVALID_STATION
;
487 FlowStat new_shares
= flow_it
->second
;
488 new_shares
.ChangeShare(current_station
, INT_MIN
);
489 StationIDStack excluded
= next_station
;
490 while (!excluded
.IsEmpty() && !new_shares
.GetShares()->empty()) {
491 new_shares
.ChangeShare(excluded
.Pop(), INT_MIN
);
493 if (new_shares
.GetShares()->empty()) {
494 cargo_next
= INVALID_STATION
;
496 cargo_next
= new_shares
.GetVia();
500 /* Rewrite an invalid source station to some random other one to
501 * avoid keeping the cargo in the vehicle forever. */
502 if (cp
->source
== INVALID_STATION
&& !ge
->flows
.empty()) {
503 cp
->source
= ge
->flows
.begin()->first
;
505 bool restricted
= false;
506 FlowStatMap::const_iterator
flow_it(ge
->flows
.find(cp
->source
));
507 if (flow_it
== ge
->flows
.end()) {
508 cargo_next
= INVALID_STATION
;
510 cargo_next
= flow_it
->second
.GetViaWithRestricted(restricted
);
512 action
= VehicleCargoList::ChooseAction(cp
, cargo_next
, current_station
, accepted
, next_station
);
513 if (restricted
&& action
== MTA_TRANSFER
) {
514 /* If the flow is restricted we can't transfer to it. Choose an
515 * unrestricted one instead. */
516 cargo_next
= flow_it
->second
.GetVia();
517 action
= VehicleCargoList::ChooseAction(cp
, cargo_next
, current_station
, accepted
, next_station
);
526 transfer_deliver
.push_back(cp
);
529 transfer_deliver
.push_front(cp
);
530 /* Add feeder share here to allow reusing field for next station. */
531 share
= payment
->PayTransfer(cp
, cp
->count
);
532 cp
->AddFeederShare(share
);
533 this->feeder_share
+= share
;
534 cp
->next_station
= cargo_next
;
539 this->action_counts
[action
] += cp
->count
;
542 assert(this->packets
.empty());
543 this->packets
= std::move(transfer_deliver
);
544 this->packets
.insert(this->packets
.end(), keep
.begin(), keep
.end());
545 this->AssertCountConsistency();
546 this->InvalidateCache();
547 return this->action_counts
[MTA_DELIVER
] > 0 || this->action_counts
[MTA_TRANSFER
] > 0;
550 /** Invalidates the cached data and rebuild it. */
551 void VehicleCargoList::InvalidateCache()
553 this->feeder_share
= 0;
554 this->Parent::InvalidateCache();
558 * Moves some cargo from one designation to another. You can only move
559 * between adjacent designations. E.g. you can keep cargo that was previously
560 * reserved (MTA_LOAD), but you can't reserve cargo that's marked as to be
561 * delivered. Furthermore, as this method doesn't change the actual packets,
562 * you cannot move cargo from or to MTA_TRANSFER. You need a specialized
563 * template method for that.
564 * @tparam from Previous designation of cargo.
565 * @tparam to New designation of cargo.
566 * @param max_move Maximum amount of cargo to reassign.
567 * @return Amount of cargo actually reassigned.
569 template<VehicleCargoList::MoveToAction Tfrom
, VehicleCargoList::MoveToAction Tto
>
570 uint
VehicleCargoList::Reassign(uint max_move
, TileOrStationID
)
572 assert_tcompile(Tfrom
!= MTA_TRANSFER
&& Tto
!= MTA_TRANSFER
);
573 assert_tcompile(Tfrom
- Tto
== 1 || Tto
- Tfrom
== 1);
574 max_move
= min(this->action_counts
[Tfrom
], max_move
);
575 this->action_counts
[Tfrom
] -= max_move
;
576 this->action_counts
[Tto
] += max_move
;
581 * Reassign cargo from MTA_DELIVER to MTA_TRANSFER and take care of the next
582 * station the cargo wants to visit.
583 * @param max_move Maximum amount of cargo to reassign.
584 * @param next_station Station to record as next hop in the reassigned packets.
585 * @return Amount of cargo actually reassigned.
588 uint
VehicleCargoList::Reassign
<VehicleCargoList::MTA_DELIVER
, VehicleCargoList::MTA_TRANSFER
>(uint max_move
, TileOrStationID next_station
)
590 max_move
= min(this->action_counts
[MTA_DELIVER
], max_move
);
594 std::list
<CargoPacket
*> packets_to_insert
;
596 for (auto it(this->packets
.begin()); sum
< this->action_counts
[MTA_TRANSFER
] + max_move
;) {
597 CargoPacket
* cp
= *it
++;
599 if (sum
<= this->action_counts
[MTA_TRANSFER
]) continue;
600 if (sum
> this->action_counts
[MTA_TRANSFER
] + max_move
) {
601 CargoPacket
* cp_split
= cp
->Split(sum
- this->action_counts
[MTA_TRANSFER
] + max_move
);
602 sum
-= cp_split
->Count();
603 packets_to_insert
.push_back(cp_split
);
605 cp
->next_station
= next_station
;
608 for (auto packet
: packets_to_insert
) {
609 this->packets
.push_back(packet
);
612 this->action_counts
[MTA_DELIVER
] -= max_move
;
613 this->action_counts
[MTA_TRANSFER
] += max_move
;
619 * Returns reserved cargo to the station and removes it from the cache.
620 * @param max_move Maximum amount of cargo to move.
621 * @param dest Station the cargo is returned to.
622 * @param ID of next the station the cargo wants to go next.
623 * @return Amount of cargo actually returned.
625 uint
VehicleCargoList::Return(uint max_move
, StationCargoList
*dest
, StationID next
)
627 max_move
= min(this->action_counts
[MTA_LOAD
], max_move
);
628 this->PopCargo(CargoReturn(this, dest
, max_move
, next
));
633 * Shifts cargo between two vehicles.
634 * @param dest Other vehicle's cargo list.
635 * @param max_move Maximum amount of cargo to be moved.
636 * @return Amount of cargo actually moved.
638 uint
VehicleCargoList::Shift(uint max_move
, VehicleCargoList
*dest
)
640 max_move
= min(this->count
, max_move
);
641 this->PopCargo(CargoShift(this, dest
, max_move
));
646 * Unloads cargo at the given station. Deliver or transfer, depending on the
647 * ranges defined by designation_counts.
648 * @param dest StationCargoList to add transferred cargo to.
649 * @param max_move Maximum amount of cargo to move.
650 * @param payment Payment object to register payments in.
651 * @return Amount of cargo actually unloaded.
653 uint
VehicleCargoList::Unload(uint max_move
, StationCargoList
*dest
, CargoPayment
*payment
)
656 if (this->action_counts
[MTA_TRANSFER
] > 0) {
657 uint move
= min(this->action_counts
[MTA_TRANSFER
], max_move
);
658 this->ShiftCargo(CargoTransfer(this, dest
, move
));
661 if (this->action_counts
[MTA_TRANSFER
] == 0 && this->action_counts
[MTA_DELIVER
] > 0 && moved
< max_move
) {
662 uint move
= min(this->action_counts
[MTA_DELIVER
], max_move
- moved
);
663 this->ShiftCargo(CargoDelivery(this, move
, payment
));
670 * Truncates the cargo in this list to the given amount. It leaves the
671 * first cargo entities and removes max_move from the back of the list.
672 * @param max_move Maximum amount of entities to be removed from the list.
673 * @return Amount of entities actually moved.
675 uint
VehicleCargoList::Truncate(uint max_move
)
677 max_move
= min(this->count
, max_move
);
678 if (max_move
> this->ActionCount(MTA_KEEP
)) this->KeepAll();
679 this->PopCargo(CargoRemoval
<VehicleCargoList
>(this, max_move
));
684 * Routes packets with station "avoid" as next hop to a different place.
685 * @param max_move Maximum amount of cargo to move.
686 * @param dest List to prepend the cargo to.
687 * @param avoid Station to exclude from routing and current next hop of packets to reroute.
688 * @param avoid2 Additional station to exclude from routing.
689 * @oaram ge GoodsEntry to get the routing info from.
691 uint
VehicleCargoList::Reroute(uint max_move
, VehicleCargoList
*dest
, StationID avoid
, StationID avoid2
, const GoodsEntry
*ge
)
693 max_move
= min(this->action_counts
[MTA_TRANSFER
], max_move
);
694 this->ShiftCargo(VehicleCargoReroute(this, dest
, max_move
, avoid
, avoid2
, ge
));
700 * Station cargo list implementation.
705 * Appends the given cargo packet to the range of packets with the same next station
706 * @warning After appending this packet may not exist anymore!
707 * @note Do not use the cargo packet anymore after it has been appended to this CargoList!
708 * @param next the next hop
709 * @param cp the cargo packet to add
712 void StationCargoList::Append(CargoPacket
*cp
, StationID next
)
714 assert(cp
!= nullptr);
715 this->AddToCache(cp
);
717 StationCargoPacketMap::List
&list
= this->packets
[next
];
718 for (StationCargoPacketMap::List::reverse_iterator
it(list
.rbegin());
719 it
!= list
.rend(); it
++) {
720 if (StationCargoList::TryMerge(*it
, cp
)) return;
723 /* The packet could not be merged with another one */
728 * Shifts cargo from the front of the packet list for a specific station and
729 * applies some action to it.
730 * @tparam Taction Action class or function to be used. It should define
731 * "bool operator()(CargoPacket *)". If true is returned the
732 * cargo packet will be removed from the list. Otherwise it
733 * will be kept and the loop will be aborted.
734 * @param action Action instance to be applied.
735 * @param next Next hop the cargo wants to visit.
736 * @return True if all packets with the given next hop have been removed,
739 template <class Taction
>
740 bool StationCargoList::ShiftCargo(Taction
&action
, StationID next
)
742 std::pair
<Iterator
, Iterator
> range(this->packets
.equal_range(next
));
743 for (Iterator
it(range
.first
); it
!= range
.second
&& it
.GetKey() == next
;) {
744 if (action
.MaxMove() == 0) return false;
745 CargoPacket
*cp
= *it
;
747 it
= this->packets
.erase(it
);
756 * Shifts cargo from the front of the packet list for a specific station and
757 * and optional also from the list for "any station", then applies some action
759 * @tparam Taction Action class or function to be used. It should define
760 * "bool operator()(CargoPacket *)". If true is returned the
761 * cargo packet will be removed from the list. Otherwise it
762 * will be kept and the loop will be aborted.
763 * @param action Action instance to be applied.
764 * @param next Next hop the cargo wants to visit.
765 * @param include_invalid If cargo from the INVALID_STATION list should be
767 * @return Amount of cargo actually moved.
769 template <class Taction
>
770 uint
StationCargoList::ShiftCargo(Taction action
, StationIDStack next
, bool include_invalid
)
772 uint max_move
= action
.MaxMove();
773 while (!next
.IsEmpty()) {
774 this->ShiftCargo(action
, next
.Pop());
775 if (action
.MaxMove() == 0) break;
777 if (include_invalid
&& action
.MaxMove() > 0) {
778 this->ShiftCargo(action
, INVALID_STATION
);
780 return max_move
- action
.MaxMove();
784 * Truncates where each destination loses roughly the same percentage of its
785 * cargo. This is done by randomizing the selection of packets to be removed.
786 * Optionally count the cargo by origin station.
787 * @param max_move Maximum amount of cargo to remove.
788 * @param cargo_per_source Container for counting the cargo by origin.
789 * @return Amount of cargo actually moved.
791 uint
StationCargoList::Truncate(uint max_move
, StationCargoAmountMap
*cargo_per_source
)
793 max_move
= min(max_move
, this->count
);
794 uint prev_count
= this->count
;
797 bool do_count
= cargo_per_source
!= nullptr;
798 while (max_move
> moved
) {
799 for (Iterator
it(this->packets
.begin()); it
!= this->packets
.end();) {
800 CargoPacket
*cp
= *it
;
801 if (prev_count
> max_move
&& RandomRange(prev_count
) < prev_count
- max_move
) {
802 if (do_count
&& loop
== 0) {
803 (*cargo_per_source
)[cp
->source
] += cp
->count
;
808 uint diff
= max_move
- moved
;
809 if (cp
->count
> diff
) {
811 this->RemoveFromCache(cp
, diff
);
816 if (do_count
) (*cargo_per_source
)[cp
->source
] -= diff
;
819 if (do_count
) (*cargo_per_source
)[cp
->source
] += cp
->count
;
823 it
= this->packets
.erase(it
);
824 if (do_count
&& loop
> 0) {
825 (*cargo_per_source
)[cp
->source
] -= cp
->count
;
828 this->RemoveFromCache(cp
, cp
->count
);
838 * Reserves cargo for loading onto the vehicle.
839 * @param max_move Maximum amount of cargo to reserve.
840 * @param dest VehicleCargoList to reserve for.
841 * @param load_place Tile index of the current station.
842 * @param next_station Next station(s) the loading vehicle will visit.
843 * @return Amount of cargo actually reserved.
845 uint
StationCargoList::Reserve(uint max_move
, VehicleCargoList
*dest
, TileIndex load_place
, StationIDStack next_station
)
847 return this->ShiftCargo(CargoReservation(this, dest
, max_move
, load_place
), next_station
, true);
851 * Loads cargo onto a vehicle. If the vehicle has reserved cargo load that.
852 * Otherwise load cargo from the station.
853 * @param max_move Amount of cargo to load.
854 * @param dest Vehicle cargo list where the cargo resides.
855 * @param load_place The new loaded_at_xy to be assigned to packets being moved.
856 * @param next_station Next station(s) the loading vehicle will visit.
857 * @return Amount of cargo actually loaded.
858 * @note Vehicles may or may not reserve, depending on their orders. The two
859 * modes of loading are exclusive, though. If cargo is reserved we don't
860 * need to load unreserved cargo.
862 uint
StationCargoList::Load(uint max_move
, VehicleCargoList
*dest
, TileIndex load_place
, StationIDStack next_station
)
864 uint move
= min(dest
->ActionCount(VehicleCargoList::MTA_LOAD
), max_move
);
866 this->reserved_count
-= move
;
867 dest
->Reassign
<VehicleCargoList::MTA_LOAD
, VehicleCargoList::MTA_KEEP
>(move
);
870 return this->ShiftCargo(CargoLoad(this, dest
, max_move
, load_place
), next_station
, true);
875 * Routes packets with station "avoid" as next hop to a different place.
876 * @param max_move Maximum amount of cargo to move.
877 * @param dest List to append the cargo to.
878 * @param avoid Station to exclude from routing and current next hop of packets to reroute.
879 * @param avoid2 Additional station to exclude from routing.
880 * @oaram ge GoodsEntry to get the routing info from.
882 uint
StationCargoList::Reroute(uint max_move
, StationCargoList
*dest
, StationID avoid
, StationID avoid2
, const GoodsEntry
*ge
)
884 return this->ShiftCargo(StationCargoReroute(this, dest
, max_move
, avoid
, avoid2
, ge
), avoid
, false);
888 * We have to instantiate everything we want to be usable.
890 template class CargoList
<VehicleCargoList
, CargoPacketList
>;
891 template class CargoList
<StationCargoList
, StationCargoPacketMap
>;
892 template uint
VehicleCargoList::Reassign
<VehicleCargoList::MTA_DELIVER
, VehicleCargoList::MTA_KEEP
>(uint
, TileOrStationID
);