/**
 *
 * \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 SRSUE_DEMUX_H
#define SRSUE_DEMUX_H

#include "srsran/common/timers.h"
#include "srsran/interfaces/ue_mac_interfaces.h"
#include "srsran/interfaces/ue_rlc_interfaces.h"
#include "srsran/mac/pdu.h"
#include "srsran/mac/pdu_queue.h"
#include "srsran/srslog/srslog.h"

/* Logical Channel Demultiplexing and MAC CE dissassemble */

namespace srsue {

class rlc_interface_mac;
class phy_interface_mac_common;

class mac_interface_demux
{
public:
  virtual void reset_harq(uint32_t cc_idx)               = 0;
  virtual bool contention_resolution_id_rcv(uint64_t id) = 0;
};

class demux : public srsran::pdu_queue::process_callback
{
public:
  explicit demux(srslog::basic_logger& logger);
  void init(phy_interface_mac_common*            phy_h_,
            rlc_interface_mac*                   rlc,
            mac_interface_demux*                 mac,
            srsran::timer_handler::unique_timer* time_alignment_timer);
  void reset();

  bool     process_pdus();
  uint8_t* request_buffer(uint32_t len);
  uint8_t* request_buffer_bcch(uint32_t len);
  void     deallocate(uint8_t* payload_buffer_ptr);

  void push_pdu(uint8_t* buff, uint32_t nof_bytes, uint32_t tti);
  void push_pdu_bcch(uint8_t* buff, uint32_t nof_bytes);
  void push_pdu_mch(uint8_t* buff, uint32_t nof_bytes);
  void push_pdu_temp_crnti(uint8_t* buff, uint32_t nof_bytes);

  bool get_uecrid_successful();

  void process_pdu(uint8_t* pdu, uint32_t nof_bytes, srsran::pdu_queue::channel_t channel, int ul_nof_prbs);
  void mch_start_rx(uint32_t lcid);

private:
  const static int MAX_PDU_LEN      = 150 * 1024 / 8; // ~ 150 Mbps
  const static int MAX_BCCH_PDU_LEN = 1024;
  uint8_t          bcch_buffer[MAX_BCCH_PDU_LEN]; // BCCH PID has a dedicated buffer

  // args
  srslog::basic_logger&     logger;
  phy_interface_mac_common* phy_h = nullptr;
  rlc_interface_mac*        rlc   = nullptr;
  mac_interface_demux*      mac   = nullptr;

  srsran::sch_pdu mac_msg;
  srsran::mch_pdu mch_mac_msg;
  srsran::sch_pdu pending_mac_msg;
  uint8_t         mch_lcids[SRSRAN_N_MCH_LCIDS] = {};
  void            process_sch_pdu_rt(uint8_t* buff, uint32_t nof_bytes, uint32_t tti);
  void            process_sch_pdu(srsran::sch_pdu* pdu);
  void            process_mch_pdu(srsran::mch_pdu* pdu);
  bool            process_ce(srsran::sch_subh* subheader, uint32_t tti);
  void            parse_ta_cmd(srsran::sch_subh* subh, uint32_t tti);

  bool is_uecrid_successful = false;

  srsran::timer_handler::unique_timer* time_alignment_timer = nullptr;

  // Buffer of PDUs
  srsran::pdu_queue pdus;
};

} // namespace srsue

#endif // SRSUE_DEMUX_H