/*
 * Copyright 2013-2019 Software Radio Systems Limited
 *
 * This file is part of srsLTE.
 *
 * srsLTE is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version.
 *
 * srsLTE is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * A copy of the GNU Affero General Public License can be found in
 * the LICENSE file in the top-level directory of this distribution
 * and at http://www.gnu.org/licenses/.
 *
 */

#ifndef SRSGNB_MAC_NR_H
#define SRSGNB_MAC_NR_H

#include "srslte/common/logmap.h"
#include "srslte/common/mac_nr_pcap.h"
#include "srslte/common/mac_nr_pdu.h"

#include "srsenb/hdr/stack/enb_stack_base.h"
#include "srslte/interfaces/enb_metrics_interface.h"
#include "srslte/interfaces/gnb_interfaces.h"

namespace srsenb {

struct mac_nr_args_t {
  srsenb::pcap_args_t pcap;

  // params for the dummy user
  srsenb::sched_interface::sched_args_t sched;
  uint16_t                              rnti;

  // Add args
  std::string log_level;
  uint32_t    log_hex_limit;
  uint32_t    tb_size = 64;
};

class mac_nr : public mac_interface_phy_nr, public mac_interface_rrc_nr, public mac_interface_rlc_nr
{
public:
  mac_nr(srslte::logger* logger_);
  ~mac_nr();

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

  void get_metrics(srsenb::mac_metrics_t* metrics);

  // MAC interface for RRC
  int cell_cfg(srsenb::sched_interface::cell_cfg_t* cell_cfg);
  int read_pdu_bcch_bch(uint8_t* payload);

  // MAC interface for RLC
  // TODO:
  int rlc_buffer_state(uint16_t rnti, uint32_t lc_id, uint32_t tx_queue, uint32_t retx_queue) { return 0; }

  // Interface for PHY
  int sf_indication(const uint32_t tti);

private:
  void get_dl_config(const uint32_t                               tti,
                     phy_interface_stack_nr::dl_config_request_t& config_request,
                     phy_interface_stack_nr::tx_request_t&        tx_request);

  // Interaction with PHY
  phy_interface_stack_nr* phy_h = nullptr;
  rlc_interface_mac_nr*   rlc_h = nullptr;
  rrc_interface_mac_nr*   rrc_h = nullptr;

  std::unique_ptr<srslte::mac_nr_pcap> pcap = nullptr;
  srslte::log_ref                      log_h;
  srslte::byte_buffer_pool*            pool = nullptr;
  mac_nr_args_t                        args = {};

  bool started = false;

  srsenb::sched_interface::cell_cfg_t cfg = {};

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

  // UE-specific buffer
  srslte::mac_nr_sch_pdu                    ue_tx_pdu;
  std::vector<srslte::unique_byte_buffer_t> ue_tx_buffer;

  srslte::unique_byte_buffer_t ue_rlc_buffer;
};

} // namespace srsenb

#endif // SRSGNB_MAC_NR_H