/**
 *
 * \section COPYRIGHT
 *
 * Copyright 2013-2021 Software Radio Systems Limited
 *
 * By using this file, you agree to the terms and conditions set
 * forth in the LICENSE file which can be found at the top level of
 * the distribution.
 *
 */

#ifndef SRSENB_MAC_NR_H
#define SRSENB_MAC_NR_H

#include "srsran/common/block_queue.h"
#include "srsran/common/mac_pcap.h"

#include "srsenb/hdr/common/rnti_pool.h"
#include "srsenb/hdr/stack/enb_stack_base.h"
#include "srsenb/hdr/stack/mac/nr/ue_nr.h"
#include "srsran/common/task_scheduler.h"
#include "srsran/interfaces/enb_metrics_interface.h"
#include "srsran/interfaces/enb_rlc_interfaces.h"
#include "srsran/interfaces/gnb_mac_interfaces.h"

namespace srsenb {

struct mac_nr_args_t {
  srsran::phy_cfg_nr_t             phy_base_cfg = {};
  int                              fixed_dl_mcs = -1;
  int                              fixed_ul_mcs = -1;
  sched_nr_interface::sched_args_t sched_cfg    = {};
  srsenb::pcap_args_t              pcap;
};

class sched_nr;

class mac_nr final : public mac_interface_phy_nr, public mac_interface_rrc_nr, public mac_interface_rlc_nr
{
public:
  explicit mac_nr(srsran::task_sched_handle task_sched_);
  ~mac_nr();

  int  init(const mac_nr_args_t&    args_,
            phy_interface_stack_nr* phy,
            stack_interface_mac*    stack_,
            rlc_interface_mac*      rlc_,
            rrc_interface_mac_nr*   rrc_);
  void stop();

  /// Called from metrics thread.
  void get_metrics(srsenb::mac_metrics_t& metrics);

  // MAC interface for RRC
  int      cell_cfg(const std::vector<srsenb::sched_nr_interface::cell_cfg_t>& nr_cells) override;
  uint16_t reserve_rnti(uint32_t enb_cc_idx, const sched_nr_interface::ue_cfg_t& uecfg) override;
  int      read_pdu_bcch_bch(uint8_t* payload);
  int      ue_cfg(uint16_t rnti, const sched_nr_interface::ue_cfg_t& ue_cfg) override;
  int      remove_ue(uint16_t rnti) override;

  // MAC interface for RLC
  int rlc_buffer_state(uint16_t rnti, uint32_t lcid, uint32_t tx_queue, uint32_t retx_queue) override;

  // Interface for PHY
  int  slot_indication(const srsran_slot_cfg_t& slot_cfg) override;
  int  get_dl_sched(const srsran_slot_cfg_t& slot_cfg, dl_sched_t& dl_sched) override;
  int  get_ul_sched(const srsran_slot_cfg_t& slot_cfg, ul_sched_t& ul_sched) override;
  int  pucch_info(const srsran_slot_cfg_t& slot_cfg, const pucch_info_t& pucch_info) override;
  int  pusch_info(const srsran_slot_cfg_t& slot_cfg, pusch_info_t& pusch_info) override;
  void rach_detected(const rach_info_t& rach_info) override;

  // Test interface
  void ul_bsr(uint16_t rnti, uint32_t lcid, uint32_t bsr);

private:
  uint16_t add_ue_(uint32_t enb_cc_idx);
  uint16_t alloc_ue(uint32_t enb_cc_idx);

  // internal misc helpers
  bool is_rnti_valid_nolock(uint16_t rnti);
  bool is_rnti_active_nolock(uint16_t rnti);

  // handle UCI data from either PUCCH or PUSCH
  bool handle_uci_data(const uint16_t rnti, const srsran_uci_cfg_nr_t& cfg, const srsran_uci_value_nr_t& value);

  // PDU processing
  int handle_pdu(srsran::unique_byte_buffer_t pdu);

  // Metrics processing
  void get_metrics_nolock(srsenb::mac_metrics_t& metrics);

  // Encoding
  srsran::byte_buffer_t*       assemble_rar(srsran::const_span<sched_nr_interface::msg3_grant_t> grants);
  srsran::unique_byte_buffer_t rar_pdu_buffer = nullptr;

  // Interaction with other components
  phy_interface_stack_nr* phy   = nullptr;
  stack_interface_mac*    stack = nullptr;
  rlc_interface_mac*      rlc   = nullptr;
  rrc_interface_mac_nr*   rrc   = nullptr;

  // args
  srsran::task_sched_handle             task_sched;
  srsran::task_multiqueue::queue_handle stack_task_queue;

  std::unique_ptr<srsran::mac_pcap> pcap = nullptr;
  mac_nr_args_t                     args = {};
  srslog::basic_logger&             logger;

  std::atomic<bool> started = {false};

  const static uint32_t                       NUMEROLOGY_IDX = 0; /// only 15kHz supported at this stage
  std::unique_ptr<srsenb::sched_nr>           sched;
  std::vector<sched_nr_interface::cell_cfg_t> cell_config;

  // Map of active UEs
  pthread_rwlock_t                                                              rwmutex    = {};
  static const uint16_t                                                         FIRST_RNTI = 0x4601;
  srsran::static_circular_map<uint16_t, std::unique_ptr<ue_nr>, SRSENB_MAX_UES> ue_db;

  std::atomic<uint16_t> ue_counter{0};

  // BCH buffers
  struct sib_info_t {
    uint32_t                     index;
    uint32_t                     periodicity;
    srsran::unique_byte_buffer_t payload;
  };
  std::vector<sib_info_t>      bcch_dlsch_payload;
  srsran::unique_byte_buffer_t bcch_bch_payload = nullptr;

  // Number of rach preambles detected for a CC
  std::vector<uint32_t> detected_rachs;
};

} // namespace srsenb

#endif // SRSENB_MAC_NR_H