# HG changeset patch # User rubidium # Date 2011-01-19 16:25:00 # Node ID d51196d5afa12094749e35aa4cf6c36ce95ada59 # Parent 93f76317eea000fde1f0d7a81054c340eaf9c63a (svn r21849) -Codechange: move merging/splitting of cargopackets into a helper function (fonsinchen) diff --git a/src/cargopacket.cpp b/src/cargopacket.cpp --- a/src/cargopacket.cpp +++ b/src/cargopacket.cpp @@ -86,6 +86,31 @@ CargoPacket::CargoPacket(uint16 count, b } /** + * Split this packet in two and return the split off part. + * @param new_size Size of the remaining part. + * @return Split off part. + */ +FORCEINLINE CargoPacket *CargoPacket::Split(uint new_size) +{ + Money fs = this->feeder_share * new_size / static_cast(this->count); + 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); + this->feeder_share -= fs; + this->count -= new_size; + return cp_new; +} + +/** + * Merge another packet into this one. + * @param cp Packet to be merged in. + */ +FORCEINLINE void CargoPacket::Merge(CargoPacket *cp) +{ + this->count += cp->count; + this->feeder_share += cp->feeder_share; + delete cp; +} + +/** * Invalidates (sets source_id to INVALID_SOURCE) all cargo packets from given source. * @param src_type Type of source. * @param src Index of source. @@ -168,10 +193,7 @@ void CargoList::Append(CargoPacke for (List::reverse_iterator it(this->packets.rbegin()); it != this->packets.rend(); it++) { CargoPacket *icp = *it; if (Tinst::AreMergable(icp, cp) && icp->count + cp->count <= CargoPacket::MAX_COUNT) { - icp->count += cp->count; - icp->feeder_share += cp->feeder_share; - - delete cp; + icp->Merge(cp); return; } } @@ -291,16 +313,15 @@ bool CargoList::MoveTo(Tother_ins cp->count = left; } else { /* But... the rest needs package splitting. */ - Money fs = cp->feeder_share * max_move / static_cast(cp->count); - cp->feeder_share -= fs; - cp->count -= max_move; + CargoPacket *cp_new = cp->Split(cp->count - max_move); - CargoPacket *cp_new = new CargoPacket(max_move, cp->days_in_transit, cp->source, cp->source_xy, (mta == MTA_CARGO_LOAD) ? data : cp->loaded_at_xy, fs, cp->source_type, cp->source_id); static_cast(this)->RemoveFromCache(cp_new); // this reflects the changes in cp. if (mta == MTA_TRANSFER) { /* Add the feeder share before inserting in dest. */ cp_new->feeder_share += payment->PayTransfer(cp_new, max_move); + } else if (mta == MTA_CARGO_LOAD) { + cp_new->loaded_at_xy = data; } dest->Append(cp_new); diff --git a/src/cargopacket.h b/src/cargopacket.h --- a/src/cargopacket.h +++ b/src/cargopacket.h @@ -62,10 +62,12 @@ public: /** Destroy the packet. */ ~CargoPacket() { } + CargoPacket *Split(uint new_size); + void Merge(CargoPacket *cp); /** * Gets the number of 'items' in this packet. - * @return the item count. + * @return Item count. */ FORCEINLINE uint16 Count() const {