#ifndef SCHED_HARQ_H
#define SCHED_HARQ_H

#include <map>
#include "srslte/common/log.h"
#include "srslte/interfaces/sched_interface.h"

namespace srsenb {

class harq_proc 
{
public:
  void     config(uint32_t id, uint32_t max_retx, srslte::log* log_h);    
  void     set_max_retx(uint32_t max_retx); 
  void     reset();
  uint32_t get_id(); 
  bool     is_empty();
  
  void     new_retx(uint32_t tti, int *mcs, int *tbs);
  
  bool     get_ack();
  void     set_ack(bool ack);    
  
  uint32_t nof_tx();
  uint32_t nof_retx();
  uint32_t get_tti();
  bool     get_ndi();
  
protected:

  void     new_tx_common(uint32_t tti, int mcs, int tbs);
  bool     has_pending_retx_common(); 
  
  bool     ack;       
  bool     active; 
  bool     ndi; 
  uint32_t id;  
  uint32_t max_retx; 
  uint32_t n_rtx; 
  uint32_t tx_cnt;
  int      tti;
  int      last_mcs;
  int      last_tbs;
  
  srslte::log* log_h;

  private:
    bool ack_received;
}; 

class dl_harq_proc : public harq_proc
{
public:
  void     new_tx(uint32_t tti, int mcs, int tbs, uint32_t n_cce);
  uint32_t get_rbgmask();
  void     set_rbgmask(uint32_t new_mask); 
  bool     has_pending_retx(uint32_t tti);      
  int      get_tbs();
  uint32_t get_n_cce();
private:
  uint32_t rbgmask;     
  uint32_t nof_rbg; 
  uint32_t n_cce;
};

class ul_harq_proc : public harq_proc
{
public:
  
  typedef struct {
    uint32_t RB_start;
    uint32_t L;
  } ul_alloc_t;
  
  void       new_tx(uint32_t tti, int mcs, int tbs);
  
  ul_alloc_t get_alloc();
  void       set_alloc(ul_alloc_t alloc);
  void       same_alloc();
  bool       is_adaptive_retx(); 
  
  bool       has_pending_ack();
  uint32_t   get_pending_data(); 
  
  void       set_rar_mcs(uint32_t mcs);
  bool       get_rar_mcs(int* mcs);
  
private:
  ul_alloc_t allocation;       
  bool need_ack;
  int  pending_data;
  uint32_t rar_mcs;
  bool has_rar_mcs;
  bool is_adaptive;
  bool is_msg3;
};
  
}      


#endif