/*
 * Licensed to the OpenAirInterface (OAI) Software Alliance under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The OpenAirInterface Software Alliance licenses this file to You under
 * the OAI Public License, Version 1.1  (the "License"); you may not use this file
 * except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.openairinterface.org/?page_id=698
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *-------------------------------------------------------------------------------
 * For more information about the OpenAirInterface (OAI) Software Alliance:
 *      contact@openairinterface.org
 */

/*! \file PHY/NR_TRANSPORT/srs_rx.c
 * \brief Top-level routines for getting the SRS physical channel
 * \date 2021
 * \version 1.0
 */

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>

#include "PHY/INIT/phy_init.h"
#include "PHY/impl_defs_nr.h"
#include "PHY/defs_nr_common.h"
#include "PHY/defs_gNB.h"
#include "PHY/CODING/nrSmallBlock/nr_small_block_defs.h"
#include "PHY/NR_UE_TRANSPORT/srs_modulation_nr.h"
#include "common/utils/LOG/log.h"

#include "nfapi/oai_integration/vendor_ext.h"

#include "T.h"

//#define SRS_DEBUG
#define SEND_BY_SOCKET

int g_dmrs_caps_idx = 0;
int g_dmrs_cnt  = 0;
nr_srs_cap_t g_srs_caps[2];

#if defined SEND_BY_SOCKET
    extern uint8_t *ric_send_buf;
  extern int      ric_send_len;
  extern sem_t   ric_send_sem;
#endif



NR_gNB_SRS_t *new_gNB_srs(void){
  NR_gNB_SRS_t *srs;
  srs = (NR_gNB_SRS_t *)malloc16(sizeof(NR_gNB_SRS_t));
  srs->active = 0;
  return (srs);
}

void free_gNB_srs(NR_gNB_SRS_t *srs)
{
  free_and_zero(srs);
}

int nr_find_srs(rnti_t rnti,
                frame_t frame,
                slot_t slot,
                PHY_VARS_gNB *gNB) {

  AssertFatal(gNB!=NULL,"gNB is null\n");
  int index = -1;

  for (int i=0; i<NUMBER_OF_NR_SRS_MAX; i++) {
    AssertFatal(gNB->srs[i]!=NULL,"gNB->srs[%d] is null\n",i);
    if ((gNB->srs[i]->active>0) &&
        (gNB->srs[i]->srs_pdu.rnti==rnti) &&
        (gNB->srs[i]->frame==frame) &&
        (gNB->srs[i]->slot==slot)) return(i);
    else if ((gNB->srs[i]->active == 0) && (index==-1)) index=i;
  }

  if (index==-1)
    LOG_E(PHY,"SRS list is full\n");

  return(index);
}

void nr_fill_srs(PHY_VARS_gNB *gNB,
                 frame_t frame,
                 slot_t slot,
                 nfapi_nr_srs_pdu_t *srs_pdu) {

  int id = nr_find_srs(srs_pdu->rnti,frame,slot,gNB);
  AssertFatal( (id>=0) && (id<NUMBER_OF_NR_SRS_MAX),
               "invalid id found for srs !!! rnti %04x id %d\n",srs_pdu->rnti,id);

  NR_gNB_SRS_t  *srs = gNB->srs[id];
  srs->frame = frame;
  srs->slot = slot;
  srs->active = 1;
  memcpy((void*)&srs->srs_pdu, (void*)srs_pdu, sizeof(nfapi_nr_srs_pdu_t));
}

int nr_get_srs_signal(PHY_VARS_gNB *gNB,
                      frame_t frame,
                      slot_t slot,
                      nfapi_nr_srs_pdu_t *srs_pdu,
                      nr_srs_info_t *nr_srs_info,
                      int32_t srs_received_signal[][gNB->frame_parms.ofdm_symbol_size*(1<<srs_pdu->num_symbols)]) {

#ifdef SRS_DEBUG
  LOG_I(NR_PHY,"Calling %s function\n", __FUNCTION__);
#endif

  int32_t **rxdataF = gNB->common_vars.rxdataF;
  const NR_DL_FRAME_PARMS *frame_parms = &gNB->frame_parms;

  const uint16_t n_symbols = (slot&3)*frame_parms->symbols_per_slot;                    // number of symbols until this slot
  const uint8_t l0 = frame_parms->symbols_per_slot - 1 - srs_pdu->time_start_position;  // starting symbol in this slot
  const uint64_t symbol_offset = (n_symbols+l0)*frame_parms->ofdm_symbol_size;
  const uint64_t subcarrier_offset = frame_parms->first_carrier_offset + srs_pdu->bwp_start*NR_NB_SC_PER_RB;

  const uint8_t N_ap = 1<<srs_pdu->num_ant_ports;
  const uint8_t N_symb_SRS = 1<<srs_pdu->num_symbols;
  const uint8_t K_TC = 2<<srs_pdu->comb_size;
  const uint16_t M_sc_b_SRS = srs_bandwidth_config[srs_pdu->config_index][srs_pdu->bandwidth_index][0] * NR_NB_SC_PER_RB/K_TC;

  int32_t *rx_signal;
  bool no_srs_signal = true;
  for (int ant = 0; ant < frame_parms->nb_antennas_rx; ant++) {

    memset(srs_received_signal[ant], 0, frame_parms->ofdm_symbol_size*sizeof(int32_t));
    rx_signal = &rxdataF[ant][symbol_offset];

    for (int p_index = 0; p_index < N_ap; p_index++) {

#ifdef SRS_DEBUG
      LOG_I(NR_PHY,"===== UE port %d --> gNB Rx antenna %i =====\n", p_index, ant);
#endif

      for (int l_line = 0; l_line < N_symb_SRS; l_line++) {

#ifdef SRS_DEBUG
        LOG_I(NR_PHY,":::::::: OFDM symbol %d ::::::::\n", l0+l_line);
#endif

        uint16_t subcarrier = subcarrier_offset + nr_srs_info->k_0_p[p_index][l_line];
        if (subcarrier>frame_parms->ofdm_symbol_size) {
          subcarrier -= frame_parms->ofdm_symbol_size;
        }
        uint16_t l_line_offset = l_line*frame_parms->ofdm_symbol_size;

        for (int k = 0; k < M_sc_b_SRS; k++) {

          srs_received_signal[ant][l_line_offset+subcarrier] = rx_signal[l_line_offset+subcarrier];

          if (rx_signal[l_line_offset+subcarrier] != 0) {
            no_srs_signal = false;
          }

#ifdef SRS_DEBUG
          int subcarrier_log = subcarrier-subcarrier_offset;
          if(subcarrier_log < 0) {
            subcarrier_log = subcarrier_log + frame_parms->ofdm_symbol_size;
          }
          if(subcarrier_log%12 == 0) {
            LOG_I(NR_PHY,"------------ %d ------------\n", subcarrier_log/12);
          }
          LOG_I(NR_PHY,"(%i)  \t%i\t%i\n",
                subcarrier_log,
                (int16_t)(srs_received_signal[ant][l_line_offset+subcarrier]&0xFFFF),
                (int16_t)((srs_received_signal[ant][l_line_offset+subcarrier]>>16)&0xFFFF));
#endif

          // Subcarrier increment
          subcarrier += K_TC;
          if (subcarrier >= frame_parms->ofdm_symbol_size) {
            subcarrier=subcarrier-frame_parms->ofdm_symbol_size;
          }

        } // for (int k = 0; k < M_sc_b_SRS; k++)
      } // for (int l_line = 0; l_line < N_symb_SRS; l_line++)
    } // for (int p_index = 0; p_index < N_ap; p_index++)
  } // for (int ant = 0; ant < frame_parms->nb_antennas_rx; ant++)

  if (no_srs_signal) {
    LOG_W(NR_PHY, "No SRS signal\n");
    return -1;
  } else {
    return 0;
  }
}

int nr_cap_srs_signal(PHY_VARS_gNB *gNB,
                      frame_t frame,
                      slot_t slot,
                      nfapi_nr_srs_pdu_t *srs_config_pdu,
                      nr_srs_info_t *nr_srs_info)
{
 int32_t **rxdataF = gNB->common_vars.rxdataF;
  const NR_DL_FRAME_PARMS *frame_parms = &gNB->frame_parms;

  const uint16_t n_symbols = (slot&3)*frame_parms->symbols_per_slot;                    // number of symbols until this slot
  const uint8_t l0 = frame_parms->symbols_per_slot - 1 - srs_config_pdu->time_start_position;  // starting symbol in this slot
  const uint64_t symbol_offset = (n_symbols+l0)*frame_parms->ofdm_symbol_size;
  nr_srs_cap_t *ns_srs_cap = (nr_srs_cap_t *)&g_srs_caps[g_dmrs_caps_idx];
  ns_srs_cap->num_ant_ports = srs_config_pdu->num_ant_ports;       
  ns_srs_cap->num_symbols = srs_config_pdu->num_symbols;        
  ns_srs_cap->num_repetitions = srs_config_pdu->num_repetitions;       
  ns_srs_cap->time_start_position = srs_config_pdu->time_start_position;  
  ns_srs_cap->config_index = srs_config_pdu->config_index;         
  ns_srs_cap->sequence_id = srs_config_pdu->sequence_id;          
  ns_srs_cap->bandwidth_index = srs_config_pdu->bandwidth_index;      
  ns_srs_cap->comb_size = srs_config_pdu->comb_size;             
  ns_srs_cap->comb_offset = srs_config_pdu->comb_offset;         
  ns_srs_cap->cyclic_shift = srs_config_pdu->cyclic_shift;        
  ns_srs_cap->frequency_position = srs_config_pdu->frequency_position;    
  ns_srs_cap->frequency_shift = srs_config_pdu->frequency_shift;        
  ns_srs_cap->frequency_hopping = srs_config_pdu->frequency_hopping;      
  ns_srs_cap->group_or_sequence_hopping = srs_config_pdu->group_or_sequence_hopping;
  ns_srs_cap->resource_type = srs_config_pdu->resource_type;             
  ns_srs_cap->t_srs = srs_config_pdu->t_srs;                    
  ns_srs_cap->t_offset = srs_config_pdu->t_offset;  

  uint8_t N_ap = 1<<srs_config_pdu->num_ant_ports;            // Number of antenna port for transmission
  uint8_t N_symb_SRS = 1<<srs_config_pdu->num_symbols;
  uint8_t n_SRS_cs = srs_config_pdu->cyclic_shift;
  uint8_t n_SRS_cs_max = srs_max_number_cs[srs_config_pdu->comb_size];
  for (int p_index = 0; p_index < N_ap; p_index++) 
  {  
    ns_srs_cap->n_SRS_cs_i[p_index] = (n_SRS_cs +  (n_SRS_cs_max * (SRS_antenna_port[p_index] - 1000)/N_ap))%n_SRS_cs_max;
    for (int l_line = 0; l_line < N_symb_SRS; l_line++)
      ns_srs_cap->k_0_p[p_index][l_line] = nr_srs_info->k_0_p[p_index][l_line];
  }
  int len = frame_parms->N_RB_DL * 6;
  int32_t *dst = ns_srs_cap->dmrsData;
  int32_t *src;
  int ant_num = 0;
  for (int ant = 0; ant < 4; ant++) 
  {
     if (rxdataF[ant]!=NULL)
     {
      for (int l_line = 0; l_line < N_symb_SRS; l_line++)
      {
        src = (int32_t *)&rxdataF[ant][symbol_offset + l_line * gNB->frame_parms.ofdm_symbol_size];
        memcpy(dst, &src[gNB->frame_parms.ofdm_symbol_size - len], len*4);
        dst += len;
        memcpy(dst, &src[0], len*4);
        dst += len;
      }
      ant_num++;
      }
  }
  
#ifdef SEND_BY_SOCKET
  //ric_send_buf = ns_srs_cap;
  //ric_send_len = 40*2+273*12*ant_num*N_symb_SRS*4;//for test
  ric_send_buf = ns_srs_cap->dmrsData;
  ric_send_len = 273*12*ant_num*N_symb_SRS*4;//for test
  sem_post(&ric_send_sem);
  LOG_I(PHY,"sem_post(&ric_send_sem)  frame %d %d, buf%p, size %d, port %d, ant %d config %d\n\n\n",ric_send_buf,ric_send_len, ns_srs_cap->num_ant_ports, ant_num, ns_srs_cap->config_index);
  g_dmrs_cnt++;
  
#else
  FILE *output_fd = NULL;
  output_fd = fopen("srs_data.am","w");
  if (output_fd==NULL){
   printf("Error opening \n");
  }

  if (output_fd) {
    LOG_I(PHY, "log srs data, ant %d, symbs %d, port %d, size %d, config %d\n", frame_parms->nb_antennas_rx, N_symb_SRS, N_ap, sizeof(nr_srs_cap_t)/2, ns_srs_cap->config_index);
    fwrite(&(ns_srs_cap->num_ant_ports),sizeof(int16_t),38+273*12*frame_parms->nb_antennas_rx*N_symb_SRS*2,output_fd);
  fclose(output_fd);
  }
#endif

  g_dmrs_caps_idx = (g_dmrs_caps_idx+1) % 2;

}