/*
 * SPDX-License-Identifier: LicenseRef-CSSL-1.0
 */
#ifndef __NR_CHAN_MODEL_H__
#define __NR_CHAN_MODEL_H__

#include "common/platform_types.h"
#include "openair2/NR_PHY_INTERFACE/NR_IF_Module.h"

#define NR_NUM_MCS 29
#define NR_NUM_SINR 372
#define NUM_BLER_COL 13
#define NUM_NFAPI_SLOT 20
#define NR_NUM_LAYER 1
#define MAX_NR_CHAN_PARAMS 256

typedef struct NR_UL_TIME_ALIGNMENT NR_UL_TIME_ALIGNMENT_t;

typedef struct {
  float sinr;
  float rsrp;
  float rsrq;
  uint8_t source;
  uint8_t pmi;
  uint8_t ri;
  uint8_t cqi;
  uint8_t area_code;
} nr_channel_status;

typedef struct {
  uint8_t slot;
  uint16_t rnti[MAX_NR_CHAN_PARAMS];
  uint8_t mcs[MAX_NR_CHAN_PARAMS];
  uint8_t rvIndex[MAX_NR_CHAN_PARAMS];
  float sinr;
  uint16_t num_pdus;
  bool drop_flag[MAX_NR_CHAN_PARAMS];
  bool latest;
  uint8_t area_code;
} slot_rnti_mcs_t;

typedef struct {
  uint16_t length;
  float bler_table[NR_NUM_SINR][NUM_BLER_COL];
} nr_bler_struct;

extern nr_bler_struct nr_bler_data[NR_NUM_MCS];
extern nr_bler_struct nr_mimo_bler_data[NR_NUM_MCS];

void read_channel_param(const nfapi_nr_dl_tti_pdsch_pdu_rel15_t *pdu, int sf, int index);
float get_bler_val(uint8_t mcs, int sinr);
bool should_drop_transport_block(int slot, uint16_t rnti);
bool is_channel_modeling(void);
int get_mcs_from_sinr(nr_bler_struct *nr_bler_data, float sinr);
void init_nr_bler_table(const char *env_string);

#endif