/*
 * Copyright 2013-2020 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 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();
  void reset();

  bool set_cell(srslte_cell_t cell);

  /* 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(float cfo);
  float get_ref_cfo() const;

  void set_tdd_config(srslte_tdd_config_t config);
  void set_config(srslte::phy_cfg_t& phy_cfg);
  void upd_config_dci(srslte_dci_cfg_t& dci_cfg);
  void set_crnti(uint16_t rnti);
  void enable_pregen_signals(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();

private:
  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 = {};

  // Metrics
  dl_metrics_t dl_metrics = {};
  ul_metrics_t ul_metrics = {};

  // Mutex, for protecting what matters most: ue_ul, ue_ul_cfg, ue_dl, ue_dl_cfg, cell, pmch_cfg
  std::mutex mutex;
};

} // namespace srsue

#endif // SRSLTE_CC_WORKER_H