beamforming-helper-base.h

// Copyright (c) 2020 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
//
// SPDX-License-Identifier: GPL-2.0-only
 
#include <ns3/beamforming-vector.h>
#include <ns3/object-factory.h>
#include <ns3/object.h>
#include <ns3/vector.h>
 
#ifndef SRC_NR_HELPER_BEAMFORMING_HELPER_BASE_H_
#define SRC_NR_HELPER_BEAMFORMING_HELPER_BASE_H_
 
namespace ns3
{
 
class NrSpectrumPhy;
class NrGnbNetDevice;
class NrUeNetDevice;
 
class BeamformingHelperBase : public Object
{
  public:
    BeamformingHelperBase();
    ~BeamformingHelperBase() override;
 
    static TypeId GetTypeId();
 
    virtual void AddBeamformingTask(const Ptr<NrGnbNetDevice>& gNbDev,
                                    const Ptr<NrUeNetDevice>& ueDev) = 0;
 
    virtual void SetBeamformingMethod(const TypeId& beamformingMethod) = 0;
 
    void SetBeamformingAlgorithmAttribute(const std::string& n, const AttributeValue& v);
 
  protected:
    virtual void RunTask(const Ptr<NrSpectrumPhy>& gnbSpectrumPhy,
                         const Ptr<NrSpectrumPhy>& ueSpectrumPhy) const;
 
    virtual BeamformingVectorPair GetBeamformingVectors(
        const Ptr<NrSpectrumPhy>& gnbSpectrumPhy,
        const Ptr<NrSpectrumPhy>& ueSpectrumPhy) const = 0;
 
    typedef std::pair<Ptr<NrSpectrumPhy>, Ptr<NrSpectrumPhy>> SpectrumPhyPair;
 
    ObjectFactory
        m_algorithmFactory; 
};
 
}; // namespace ns3
 
#endif /* SRC_NR_HELPER_BEAMFORMING_HELPER_BASE_H_ */