nr-mimo-matrices.h

// Copyright (c) 2024 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
//
// SPDX-License-Identifier: GPL-2.0-only
 
#ifndef NR_MIMO_MATRICES_H
#define NR_MIMO_MATRICES_H
 
#include <ns3/matrix-array.h>
#include <ns3/spectrum-value.h>
 
namespace ns3
{
 
class NrIntfNormChanMat;
class NrSinrMatrix;
 
class NrCovMat : public ComplexMatrixArray
{
  public:
    NrCovMat() = default;
    NrCovMat(ComplexMatrixArray arr)
        : ComplexMatrixArray(arr){};
 
    virtual void AddInterferenceSignal(const ComplexMatrixArray& rhs);
 
    virtual void SubtractInterferenceSignal(const ComplexMatrixArray& rhs);
 
    virtual NrIntfNormChanMat CalcIntfNormChannel(const ComplexMatrixArray& chanMat) const;
 
  private:
    virtual NrIntfNormChanMat CalcIntfNormChannelMimo(const ComplexMatrixArray& chanMat) const;
};
 
class NrIntfNormChanMat : public ComplexMatrixArray
{
  public:
    NrIntfNormChanMat() = default;
    NrIntfNormChanMat(ComplexMatrixArray arr)
        : ComplexMatrixArray(arr){};
 
    virtual NrSinrMatrix ComputeSinrForPrecoding(const ComplexMatrixArray& precMats) const;
 
  private:
    virtual ComplexMatrixArray ComputeMse(const ComplexMatrixArray& precMats) const;
 
    virtual ComplexMatrixArray ComputeMseMimo(const ComplexMatrixArray& precMats) const;
};
 
class NrSinrMatrix : public DoubleMatrixArray
{
  public:
    NrSinrMatrix() = default;
    NrSinrMatrix(DoubleMatrixArray arr)
        : DoubleMatrixArray(arr){};
    NrSinrMatrix(const std::valarray<double>& values)
        : DoubleMatrixArray(values){};
    NrSinrMatrix(uint8_t rank, size_t nRbs)
        : DoubleMatrixArray(rank, nRbs){};
 
    uint8_t GetRank() const;
    size_t GetNumRbs() const;
    SpectrumValue GetVectorizedSpecVal() const;
};
 
} // namespace ns3
 
#endif // NR_MIMO_MATRICES_H