RlcRetransmissionBuffer.h Source File¶

Simu5G: RlcRetransmissionBuffer.h Source File
Simu5G
//
//                  Simu5G
//
// Authors: Esteban Egea Lopez (Universidad Politecnica de Cartagena)
//
// This file is part of a software released under the license included in file
// "license.pdf". Please read LICENSE and README files before using it.
// The above files and the present reference are part of the software itself,
// and cannot be removed from it.
//
 
#ifndef STACK_RLC_AM_BUFFER_RLCRETRANSMISSIONBUFFER_H_
#define STACK_RLC_AM_BUFFER_RLCRETRANSMISSIONBUFFER_H_
 
#include <omnetpp.h>
 
#include <map>
#include <set>
 
namespace simu5g {
struct RetxTask {
    uint32_t sn;
    uint32_t soStart;
    uint32_t soEnd;
    bool isWholeUnit;
 
    bool operator<(const RetxTask &other) const {
        if (sn != other.sn) return sn < other.sn;
        if (soStart != other.soStart) return soStart < other.soStart;
        return soEnd < other.soEnd;
    }
};
 
class RlcRetransmissionBuffer
{
  public:
    RlcRetransmissionBuffer(uint32_t threshold);
    virtual ~RlcRetransmissionBuffer();
    virtual void beginRetxRound() {
        for (auto &[sn, state] : retxCounters_)
            state.incrementedInCurrentRound = false;
    }
    virtual bool addNack(uint32_t sn, bool isWhole, uint32_t start, uint32_t end);
    virtual bool getNextRetxTask(RetxTask &outTask) {
        if (pendingRetx_.empty())
            return false;
        outTask = *pendingRetx_.begin();
        return true;
    }
    virtual uint64_t getRetxPendingBytes();
    const std::set<RetxTask>& getPendingRetx() const { return pendingRetx_; }
    void markRetransmitted(const RetxTask &task) { pendingRetx_.erase(task); }
    virtual void clearSn(uint32_t sn);
 
  protected:
    struct RetxState {
        uint32_t retxCount = 0;
        bool incrementedInCurrentRound = false;
    };
 
    std::map<uint32_t, RetxState> retxCounters_;
    std::set<RetxTask> pendingRetx_;
    uint32_t maxRetxThreshold_;
};
 
} // namespace simu5g
 
#endif // STACK_RLC_AM_BUFFER_RLCRETRANSMISSIONBUFFER_H_