/**
 *
 * \section COPYRIGHT
 *
 * Copyright 2013-2020 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 SRSLTE_CC_WORKER_H
#define SRSLTE_CC_WORKER_H

#include "phy_common.h"
#include "srslte/interfaces/ue_interfaces.h"
#include "srslte/srslte.h"

namespace srsue {

class cc_worker
{
public:
  cc_worker(uint32_t cc_idx, uint32_t max_prb, phy_common* phy, srslte::log* log);
  ~cc_worker();

  /* Functions used by main PHY thread */
  cf_t*    get_rx_buffer(uint32_t antenna_idx);
  cf_t*    get_tx_buffer(uint32_t antenna_idx);
  uint32_t get_buffer_len();

  void  set_tti(uint32_t tti);
  void  set_cfo_unlocked(float cfo);
  float get_ref_cfo() const;

  // Functions to set configuration.
  // Warning: all these functions are unlocked and must be called while the worker is not processing data
  void reset_cell_unlocked();
  bool set_cell_unlocked(srslte_cell_t cell_);
  void set_tdd_config_unlocked(srslte_tdd_config_t config);
  void set_config_unlocked(srslte::phy_cfg_t& phy_cfg);
  void upd_config_dci_unlocked(srslte_dci_cfg_t& dci_cfg);
  void set_crnti_unlocked(uint16_t rnti);
  void enable_pregen_signals_unlocked(bool enabled);

  void set_uci_periodic_cqi(srslte_uci_data_t* uci_data);

  bool work_dl_regular();
  bool work_dl_mbsfn(srslte_mbsfn_cfg_t mbsfn_cfg);
  bool work_ul(srslte_uci_data_t* uci_data);

  int read_ce_abs(float* ce_abs, uint32_t tx_antenna, uint32_t rx_antenna);
  int read_pdsch_d(cf_t* pdsch_d);

  void update_measurements(std::vector<rrc_interface_phy_lte::phy_meas_t>& serving_cells,
                           cf_t*                                           rssi_power_buffer = nullptr);

private:
  void reset();

  void dl_phy_to_mac_grant(srslte_pdsch_grant_t*                  phy_grant,
                           srslte_dci_dl_t*                       dl_dci,
                           mac_interface_phy_lte::mac_grant_dl_t* mac_grant);
  void ul_phy_to_mac_grant(srslte_pusch_grant_t*                  phy_grant,
                           srslte_dci_ul_t*                       ul_dci,
                           uint32_t                               pid,
                           bool                                   ul_grant_available,
                           mac_interface_phy_lte::mac_grant_ul_t* mac_grant);

  /* Methods for DL... */
  int decode_pdcch_ul();
  int decode_pdcch_dl();

  void decode_phich();
  int  decode_pdsch(srslte_pdsch_ack_resource_t            ack_resource,
                    mac_interface_phy_lte::tb_action_dl_t* action,
                    bool                                   acks[SRSLTE_MAX_CODEWORDS]);
  int  decode_pmch(mac_interface_phy_lte::tb_action_dl_t* action, srslte_mbsfn_cfg_t* mbsfn_cfg);

  /* Methods for UL */
  bool     encode_uplink(mac_interface_phy_lte::tb_action_ul_t* action, srslte_uci_data_t* uci_data);
  void     set_uci_sr(srslte_uci_data_t* uci_data);
  void     set_uci_aperiodic_cqi(srslte_uci_data_t* uci_data);
  void     set_uci_ack(srslte_uci_data_t* uci_data, bool is_grant_available, uint32_t dai_ul, bool is_pusch_available);
  uint32_t get_wideband_cqi();

  /* Common objects */
  phy_common*  phy   = nullptr;
  srslte::log* log_h = nullptr;

  srslte_cell_t      cell      = {};
  srslte_dl_sf_cfg_t sf_cfg_dl = {};
  srslte_ul_sf_cfg_t sf_cfg_ul = {};

  uint32_t cc_idx                             = 0;
  bool     pregen_enabled                     = false;
  bool     cell_initiated                     = false;
  cf_t*    signal_buffer_rx[SRSLTE_MAX_PORTS] = {};
  cf_t*    signal_buffer_tx[SRSLTE_MAX_PORTS] = {};
  uint32_t signal_buffer_max_samples          = 0;

  /* Objects for DL */
  srslte_ue_dl_t     ue_dl     = {};
  srslte_ue_dl_cfg_t ue_dl_cfg = {};
  srslte_pmch_cfg_t  pmch_cfg  = {};

  srslte_chest_dl_cfg_t chest_mbsfn_cfg   = {};
  srslte_chest_dl_cfg_t chest_default_cfg = {};

  /* Objects for UL */
  srslte_ue_ul_t     ue_ul     = {};
  srslte_ue_ul_cfg_t ue_ul_cfg = {};
};

} // namespace srsue

#endif // SRSLTE_CC_WORKER_H