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

#include "mac_nr_interfaces.h"
#include "srsran/common/block_queue.h"
#include "srsran/interfaces/ue_nr_interfaces.h"
#include "srsran/interfaces/ue_rlc_interfaces.h"

namespace srsue {

/**
 * @brief Logical Channel Demultiplexing and MAC CE dissassemble according to TS 38.321
 *
 * Currently only SDU handling for SCH PDU processing is implemented.
 * Downlink CE are parsed but not handled.
 *
 * PDUs can be pushed by multiple HARQ processes in parallel.
 * Handling of the PDUs is done from Stack thread which reads the enqueued PDUs
 * from the thread-safe queue.
 */
class demux_nr : public demux_interface_harq_nr
{
public:
  demux_nr(srslog::basic_logger& logger_);
  ~demux_nr();

  int32_t init(rlc_interface_mac* rlc_, phy_interface_mac_nr* phy_);

  void process_pdus(); /// Called by MAC to process received PDUs

  // HARQ interface
  void     push_bcch(srsran::unique_byte_buffer_t pdu);
  void     push_pdu(srsran::unique_byte_buffer_t pdu, uint32_t tti);
  void     push_pdu_temp_crnti(srsran::unique_byte_buffer_t pdu, uint32_t tti);
  uint64_t get_received_crueid();

private:
  // internal helpers
  void handle_pdu(srsran::mac_sch_pdu_nr& pdu_buffer, srsran::unique_byte_buffer_t pdu);

  srslog::basic_logger& logger;
  rlc_interface_mac*    rlc = nullptr;
  phy_interface_mac_nr* phy = nullptr;

  uint64_t received_crueid = 0;

  ///< currently only DCH & BCH PDUs supported (add PCH, etc)
  srsran::block_queue<srsran::unique_byte_buffer_t> pdu_queue;
  srsran::block_queue<srsran::unique_byte_buffer_t> bcch_queue;

  srsran::mac_sch_pdu_nr rx_pdu;
  srsran::mac_sch_pdu_nr rx_pdu_tcrnti;
};

} // namespace srsue

#endif // SRSRAN_DEMUX_NR_H