nr-mac-harq-vector.h

// Copyright (c) 2019 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
//
// SPDX-License-Identifier: GPL-2.0-only
 
#pragma once
 
#include "nr-mac-harq-process.h"
 
#include <unordered_map>
 
namespace ns3
{
 
class NrMacHarqVector : private std::unordered_map<uint8_t, HarqProcess>
{
  public:
    friend std::ostream& operator<<(std::ostream& os, const NrMacHarqVector& item);
    typedef typename std::unordered_map<uint8_t, HarqProcess>::iterator iterator;
    typedef typename std::unordered_map<uint8_t, HarqProcess>::const_iterator const_iterator;
 
    NrMacHarqVector() = default;
 
    void SetMaxSize(uint8_t size)
    {
        m_maxSize = size;
        reserve(size);
        for (auto i = 0; i < size; ++i)
        {
            emplace(i, HarqProcess());
        }
    }
 
    bool Erase(uint8_t id);
 
    bool Insert(uint8_t* id, const HarqProcess& element);
 
    const iterator Find(uint8_t key)
    {
        return this->find(key);
    }
 
    const iterator Begin()
    {
        return this->begin();
    }
 
    const iterator End()
    {
        return this->end();
    }
 
    const_iterator CBegin()
    {
        return this->cbegin();
    }
 
    const_iterator CEnd()
    {
        return this->cend();
    }
 
    bool Exist(uint8_t id) const
    {
        // could be id > max_size?
        return find(id) != end();
    }
 
    HarqProcess& Get(uint8_t id)
    {
        NS_ASSERT(Exist(id));
        return find(id)->second;
    }
 
    const HarqProcess& Get(uint8_t id) const
    {
        NS_ASSERT(Exist(id));
        return find(id)->second;
    }
 
    uint8_t FirstAvailableId() const
    {
        for (const auto& it : *this)
        {
            if (!it.second.m_active)
            {
                return it.first;
            }
        }
        return 255;
    }
 
    bool CanInsert() const
    {
        return Size() < m_maxSize;
    }
 
    uint32_t Size() const
    {
        return m_usedSize;
    }
 
  private:
    uint8_t m_maxSize{0};  
    uint8_t m_usedSize{0}; 
};
 
std::ostream& operator<<(std::ostream& os, const HarqProcess& item);
 
} // namespace ns3