nr-mimo-chunk-processor.h

// Copyright (c) 2024 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
//
// SPDX-License-Identifier: GPL-2.0-only
 
#ifndef NR_MIMO_CHUNK_PROCESSOR_H
#define NR_MIMO_CHUNK_PROCESSOR_H
 
#include "nr-mimo-matrices.h"
 
#include "ns3/nstime.h"
#include "ns3/object.h"
#include "ns3/ptr.h"
#include "ns3/spectrum-value.h"
 
namespace ns3
{
 
struct MimoSinrChunk
{
    NrSinrMatrix mimoSinr; 
    uint16_t rnti{0};      
    Time dur;              
};
 
struct MimoSignalChunk
{
    ComplexMatrixArray chanSpct; 
    NrCovMat interfNoiseCov;     
    uint16_t rnti{0};            
    Time dur;                    
};
 
using MimoSinrChunksCb = Callback<void, const std::vector<MimoSinrChunk>&>;
using MimoSignalChunksCb = Callback<void, const std::vector<MimoSignalChunk>&>;
 
class NrMimoChunkProcessor : public SimpleRefCount<NrMimoChunkProcessor>
{
  public:
    void AddCallback(MimoSinrChunksCb cb);
 
    void AddCallback(MimoSignalChunksCb cb);
 
    void Start();
 
    void EvaluateChunk(const MimoSinrChunk& mimoSinr);
 
    void EvaluateChunk(const MimoSignalChunk& mimoSignal);
 
    void End();
 
  private:
    std::vector<MimoSinrChunk> m_mimoSinrChunks;     
    std::vector<MimoSignalChunk> m_mimoSignalChunks; 
 
    std::vector<MimoSinrChunksCb> m_sinrChunksCbs;     
    std::vector<MimoSignalChunksCb> m_signalChunksCbs; 
};
} // namespace ns3
 
#endif // NR_MIMO_CHUNK_PROCESSOR_H