agora-doxy/html/dobeamweights_8h_source.html

#ifndef DOBEAMWEIGHTS_H_

#define DOBEAMWEIGHTS_H_


#include <memory>


#include "armadillo"

#include "common_typedef_sdk.h"

#include "config.h"

#include "doer.h"

#include "mat_logger.h"

#include "memory_manage.h"

#include "message.h"

#include "phy_stats.h"

#include "stats.h"


class DoBeamWeights : public Doer {

 public:

  DoBeamWeights(

      Config* in_config, int tid,

      PtrGrid<kFrameWnd, kMaxUEs, complex_float>& csi_buffers,

      Table<complex_float>& calib_dl_buffer,

      Table<complex_float>& calib_ul_buffer,

      Table<complex_float>& calib_dl_msum_buffer,

      Table<complex_float>& calib_ul_msum_buffer,

      PtrGrid<kFrameWnd, kMaxDataSCs, complex_float>& ul_beam_matrices_,

      PtrGrid<kFrameWnd, kMaxDataSCs, complex_float>& dl_beam_matrices_,

      PhyStats* in_phy_stats, Stats* stats_manager);

  ~DoBeamWeights() override;


  EventData Launch(size_t tag) override;


 private:

  void ComputePartialCsiBeams(size_t tag);


  float ComputePrecoder(size_t frame_id, size_t cur_sc_id,

                        const arma::cx_fmat& mat_csi,

                        const arma::cx_fvec& calib_sc_vec, const float noise,

                        complex_float* ul_beam_mem, complex_float* dl_beam_mem);

  void ComputeCalib(size_t frame_id, size_t sc_id, arma::cx_fvec& calib_sc_vec);

  void ComputeFullCsiBeams(size_t tag);


  void Predict(size_t offset);


  PtrGrid<kFrameWnd, kMaxUEs, complex_float>& csi_buffers_;

  complex_float* pred_csi_buffer_;


  //Should be read only (Set by FFT and read by Zf)

  Table<complex_float>& calib_dl_buffer_;

  Table<complex_float>& calib_ul_buffer_;


  //Shared by all doZf objects

  Table<complex_float>& calib_dl_msum_buffer_;

  Table<complex_float>& calib_ul_msum_buffer_;

  PtrGrid<kFrameWnd, kMaxDataSCs, complex_float>& ul_beam_matrices_;

  PtrGrid<kFrameWnd, kMaxDataSCs, complex_float>& dl_beam_matrices_;

  DurationStat* duration_stat_;


  complex_float* csi_gather_buffer_;  // Intermediate buffer to gather CSI

  // Intermediate buffer to gather reciprical calibration data vector

  complex_float* calib_gather_buffer_;

  std::unique_ptr<arma::cx_fvec> calib_sc_vec_ptr_;


  PhyStats* phy_stats_;

  arma::uvec ext_ref_id_;

  size_t num_ext_ref_;

};


#endif  // DOBEAMWEIGHTS_H_