/*******************************************************************************
    OpenAirInterface 
    Copyright(c) 1999 - 2014 Eurecom

    OpenAirInterface is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.


    OpenAirInterface 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 General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with OpenAirInterface.The full GNU General Public License is 
   included in this distribution in the file called "COPYING". If not, 
   see <http://www.gnu.org/licenses/>.

  Contact Information
  OpenAirInterface Admin: openair_admin@eurecom.fr
  OpenAirInterface Tech : openair_tech@eurecom.fr
  OpenAirInterface Dev  : openair4g-devel@eurecom.fr
  
  Address      : Eurecom, Campus SophiaTech, 450 Route des Chappes, CS 50193 - 06904 Biot Sophia Antipolis cedex, FRANCE

 *******************************************************************************/
#include <stdlib.h>
#include <stdio.h>
#include <fcntl.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <pthread.h>
#include <sys/ioctl.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <sys/mman.h>
//#include <sys/user.h>
//#include <errno.h>
#include <math.h>
#include <signal.h>
#include <strings.h>
#include "forms.h"
#include "lte_scope.h"

#include "PHY/types.h"
#include "PHY/defs.h"
#include "PHY/vars.h"
#ifdef CBMIMO1
#include "ARCH/CBMIMO1/DEVICE_DRIVER/cbmimo1_device.h"
#include "ARCH/CBMIMO1/DEVICE_DRIVER/defs.h"
#include "ARCH/COMMON/defs.h"
#endif
#ifdef PLATON
#include "daq.h"
#endif

//#include "PHY/CONFIG/vars.h"
//#include "MAC_INTERFACE/vars.h"
//#include "PHY/TOOLS/defs.h"


FD_lte_scope *form;

LTE_DL_FRAME_PARMS *frame_parms;
PHY_VARS_UE *PHY_vars_UE;

//short channel[2048];
//short channel_f[2048];
//char demod_data[2048];

short *channel[8],*channel_f[8],*rx_sig[8],**rx_sig_ptr;
short *rx_sig_f[8],**rx_sig_f_ptr;
int *sync_corr;
char *pbch_llr;
short *pbch_comp, *pdcch_comp;
short *dlsch_llr,*dlsch_comp;

int length,offset;
float avg=1;
int sach_flag; 

unsigned char nb_ant_tx, nb_ant_rx;


void lte_scope_idle_callback(void) {

  int i,j,ind,k,s;

  float Re,Im;
  float mag_sig[NB_ANTENNAS_RX*NB_ANTENNAS_TX*NUMBER_OF_OFDM_CARRIERS*NUMBER_OF_OFDM_SYMBOLS_PER_SLOT],
    sig_time[NB_ANTENNAS_RX*NB_ANTENNAS_TX*NUMBER_OF_OFDM_CARRIERS*NUMBER_OF_OFDM_SYMBOLS_PER_SLOT],
    sig2[FRAME_LENGTH_COMPLEX_SAMPLES],
    time2[FRAME_LENGTH_COMPLEX_SAMPLES],
    I[25*12*12], Q[25*12*12],
    //llr[8*(3*8*6144+12)],
    //llr_time[8*(3*8*6144+12)];
    llr[25*12*4*7],
    llr_time[25*12*4*7];

  /*
    mag_h[NB_ANTENNAS_RX*NUMBER_OF_OFDM_CARRIERS*NUMBER_OF_OFDM_SYMBOLS_PER_SLOT*8],
    mag_sig2[NUMBER_OF_OFDM_CARRIERS*NUMBER_OF_OFDM_SYMBOLS_PER_SLOT*8],
    scat_sig_re[NUMBER_OF_OFDM_CARRIERS*NUMBER_OF_OFDM_SYMBOLS_PER_SLOT*8],
    scat_sig_im[NUMBER_OF_OFDM_CARRIERS*NUMBER_OF_OFDM_SYMBOLS_PER_SLOT*8],
    scat_sig_re2[NUMBER_OF_OFDM_CARRIERS*NUMBER_OF_OFDM_SYMBOLS_PER_SLOT*8],
    scat_sig_im2[NUMBER_OF_OFDM_CARRIERS*NUMBER_OF_OFDM_SYMBOLS_PER_SLOT*8];
  */
  float cum_avg;
  
  // Channel frequency response
  cum_avg = 0;
  ind = 0;
  for (j=0; j<4; j++) { 
  for (i=0;i<nb_ant_rx;i++) {
    //for (j=0; j<1; j++) { 
    //for (i=0;i<1;i++) {
      
      //for (k=0;k<frame_parms->symbols_per_tti*NUMBER_OF_OFDM_CARRIERS;k++){
      for (k=0;k<NUMBER_OF_OFDM_CARRIERS;k++){
	sig_time[ind] = (float)ind;
	Re = (float)(channel_f[(j<<1)+i][2*k]);
	Im = (float)(channel_f[(j<<1)+i][2*k+1]);
	//mag_sig[ind] = (short) rand(); 
	mag_sig[ind] = (short)10*log10(1.0+((double)Re*Re + (double)Im*Im)); 
	cum_avg += (short)sqrt((double)Re*Re + (double)Im*Im) ;
	ind++;
      }
      //      ind+=NUMBER_OF_OFDM_CARRIERS/4; // spacing for visualization
    }
  }

  avg = cum_avg/NUMBER_OF_USEFUL_CARRIERS;

  //fl_set_xyplot_ybounds(form->channel_f,30,70);
  fl_set_xyplot_data(form->channel_f,sig_time,mag_sig,ind,"","","");

  // channel_t_re = sync_corr
  for (i=0; i<FRAME_LENGTH_COMPLEX_SAMPLES; i++)  {
    sig2[i] = (float) (sync_corr[i]);
    time2[i] = (float) i;
  }

  //fl_set_xyplot_ybounds(form->channel_t_im,10,90);
  fl_set_xyplot_data(form->channel_t_im,time2,sig2,FRAME_LENGTH_COMPLEX_SAMPLES,"","","");

  /*
  // channel time resonse
  cum_avg = 0;
  ind = 0;
  for (k=0;k<1;k++){
    for (j=0;j<1;j++) {
      
      for (i=0;i<frame_parms->ofdm_symbol_size;i++){
	sig_time[ind] = (float)ind;
	Re = (float)(channel[k+2*j][2*i]);
	Im = (float)(channel[k+2*j][2*i+1]);
	//mag_sig[ind] = (short) rand(); 
	mag_sig[ind] = (short)10*log10(1.0+((double)Re*Re + (double)Im*Im)); 
	cum_avg += (short)sqrt((double)Re*Re + (double)Im*Im) ;
	ind++;
      }
    }
  }
  
  //fl_set_xyplot_ybounds(form->channel_t_im,10,90);
  fl_set_xyplot_data(form->channel_t_im,sig_time,mag_sig,ind,"","","");
  */

  // channel_t_re = rx_sig_f[0]
  for (i=0; i<FRAME_LENGTH_COMPLEX_SAMPLES; i++)  {
    //for (i=0; i<NUMBER_OF_OFDM_CARRIERS*frame_parms->symbols_per_tti/2; i++)  {
    sig2[i] = 10*log10(1.0+(double) ((rx_sig[0][2*i])*(rx_sig[0][2*i])+(rx_sig[0][2*i+1])*(rx_sig[0][2*i+1])));
    time2[i] = (float) i;
  } 

  //fl_set_xyplot_ybounds(form->channel_t_re,10,90);
  //fl_set_xyplot_data(form->channel_t_re,&time2[512*12*9],&sig2[512*12*9],512*12,"","","");
  fl_set_xyplot_data(form->channel_t_re,time2,sig2,FRAME_LENGTH_COMPLEX_SAMPLES,"","","");
  

  // channel_t_im = rx_sig[1]
  if (nb_ant_rx>1) {
    for (i=0; i<FRAME_LENGTH_COMPLEX_SAMPLES; i++)  {
      //for (i=0; i<NUMBER_OF_OFDM_CARRIERS*frame_parms->symbols_per_tti/2; i++)  {
      sig2[i] = 10*log10(1.0+(double) ((rx_sig[1][2*i])*(rx_sig[1][2*i])+(rx_sig[1][2*i+1])*(rx_sig[1][2*i+1])));
      time2[i] = (float) i;
    }

    //fl_set_xyplot_ybounds(form->channel_t_im,0,100);
    //fl_set_xyplot_data(form->channel_t_im,&time2[512*12*9],&sig2[512*12*9],512*12,"","","");
    fl_set_xyplot_data(form->channel_t_im,time2,sig2,FRAME_LENGTH_COMPLEX_SAMPLES_NO_PREFIX,"","","");
  }

  // PBCH LLR
  j=0;
  for(i=0;i<1920;i++) {
    llr[j] = (float) pbch_llr[i];
    llr_time[j] = (float) j;
    /*
    if (i==63)
      i=127;
    else if (i==191)
      i=319;
    */
    j++;
  }

  fl_set_xyplot_data(form->decoder_input,llr_time,llr,1920,"","","");
  //fl_set_xyplot_ybounds(form->decoder_input,-100,100);

  // PBCH I/Q
  j=0;
  for(i=0;i<12*12;i++) {
    I[j] = pbch_comp[2*i];
    Q[j] = pbch_comp[2*i+1];
    j++;
    /*
    if (i==47)
      i=96;
    else if (i==191)
      i=239;
    */
  }

  fl_set_xyplot_data(form->scatter_plot,I,Q,12*12,"","","");
  //fl_set_xyplot_xbounds(form->scatter_plot,-100,100);
  //fl_set_xyplot_ybounds(form->scatter_plot,-100,100);

  // PDCCH I/Q
  j=0;
  for(i=0;i<12*25*3;i++) {
    I[j] = pdcch_comp[2*i];
    Q[j] = pdcch_comp[2*i+1];
    j++;
    /*
    if (i==47)
      i=96;
    else if (i==191)
      i=239;
    */
  }

  fl_set_xyplot_data(form->scatter_plot1,I,Q,12*25*3,"","","");
  //fl_set_xyplot_xbounds(form->scatter_plot,-100,100);
  //fl_set_xyplot_ybounds(form->scatter_plot,-100,100);

  // DLSCH LLR
  for(i=0;i<12*12*7*2;i++) {
    llr[i] = (float) dlsch_llr[i];
    llr_time[i] = (float) i;
  }

  fl_set_xyplot_data(form->demod_out,llr_time,llr,12*12*7*2,"","","");
  //  fl_set_xyplot_data(form->demod_out,time2,llr,25*12*4,"","","");
  fl_set_xyplot_ybounds(form->demod_out,-500,500);

  // DLSCH I/Q
  j=0;
  for (s=2;s<12;s++) {
    for(i=0;i<12*12;i++) {
      I[j] = dlsch_comp[(2*25*12*s)+2*i];
      Q[j] = dlsch_comp[(2*25*12*s)+2*i+1];
      j++;
    }
    if (s==2)
      s=3;
    else if (s==5)
      s=6;
    else if (s==8)
      s=9;
  }

  fl_set_xyplot_data(form->scatter_plot2,I,Q,j,"","","");
  //fl_set_xyplot_xbounds(form->scatter_plot2,-100,100);
  //fl_set_xyplot_ybounds(form->scatter_plot2,-100,100);

  usleep(500000);
}
//-----------------------------------------------------------------------------
do_scope(){

//-----------------------------------------------------------------------------
  char ch;

  fl_set_idle_callback(lte_scope_idle_callback, NULL);
  fl_do_forms() ;      /* SIGSCOPE */



}

//-----------------------------------------------------------------------------
int main(int argc, char *argv[]) {
//-----------------------------------------------------------------------------

  int openair_fd,i,j;
  unsigned int mem_base;
  unsigned int first_symbol;
  char title[20];

  LTE_UE_DLSCH     *lte_ue_dlsch;
  LTE_UE_PDCCH     *lte_ue_pdcch;
  LTE_UE_PBCH      *lte_ue_pbch;
  unsigned int     bigphys_top;

  //PHY_config = malloc(sizeof(PHY_CONFIG));
  //PHY_vars = malloc(sizeof(PHY_VARS));

  frame_parms = malloc(sizeof(LTE_DL_FRAME_PARMS));
  PHY_vars_UE = malloc(sizeof(PHY_VARS_UE));

  printf("Opening /dev/openair0\n");
  if ((openair_fd = open("/dev/openair0", O_RDONLY)) <0) {
    fprintf(stderr,"Error %d opening /dev/openair0\n",openair_fd);
    exit(-1);
  }

  printf("Getting PHY_config ...\n");
  ioctl(openair_fd,openair_GET_CONFIG,frame_parms);
  dump_frame_parms(frame_parms);

  printf("Getting PHY_vars ...\n");
  ioctl(openair_fd,openair_GET_VARS,PHY_vars_UE);

  if (PHY_vars_UE->lte_ue_common_vars.txdataF == NULL) {
    printf("Problem getting PHY_vars. Is UE configured?\n");
    exit(-1);
  }

  printf("Getting BIGPHYS_top ...\n");
  ioctl(openair_fd,openair_GET_BIGPHYSTOP,(void *)&bigphys_top);

  printf("Bigphys_top = %p\n",bigphys_top);
  printf("TX_DMA_BUFFER = %p\n",PHY_vars_UE->lte_ue_common_vars.txdataF);
  printf("RX_DMA_BUFFER = %p\n",PHY_vars_UE->lte_ue_common_vars.rxdata);
  printf("PHY_vars->lte_ue_common_vars.dl_ch_estimates[0] = %p\n",PHY_vars_UE->lte_ue_common_vars.dl_ch_estimates[0]);
  printf("PHY_vars->lte_ue_common_vars.sync_corr = %p\n",PHY_vars_UE->lte_ue_common_vars.sync_corr);
  printf("PHY_vars->lte_ue_pbch_vars[0] = %p\n",PHY_vars_UE->lte_ue_pbch_vars[0]);
  printf("PHY_vars->lte_ue_pdcch_vars[0] = %p\n",PHY_vars_UE->lte_ue_pdcch_vars[0]);
  printf("PHY_vars->lte_ue_dlsch_vars[0] = %p\n",PHY_vars_UE->lte_ue_dlsch_vars[0]);

  printf("NUMBER_OF_OFDM_CARRIERS = %d\n",NUMBER_OF_OFDM_CARRIERS);

  nb_ant_tx = PHY_vars_UE->lte_frame_parms.nb_antennas_tx;
  nb_ant_rx = PHY_vars_UE->lte_frame_parms.nb_antennas_rx;
  printf("(TX, RX) ANTENNAS = %d, %d\n",nb_ant_tx,nb_ant_rx);
  
  mem_base = (unsigned int) mmap(0,
				 BIGPHYS_NUMPAGES*4096,
				 PROT_READ,
				 MAP_PRIVATE,
				 openair_fd,
				 0);

  if (mem_base != -1)
    msg("MEM base= %x\n",mem_base);
  else
    msg("Could not map physical memory\n");

  //  txdata0 = (int *)(mem_base + PHY_vars_UE->lte_ue_common_vars.txdataF) - 
  for (i=0;i<nb_ant_rx;i++) {
    for (j=0; j<4; j++) { 

      channel_f[(j<<1)+i] = (short*)(mem_base + 
				     (unsigned int)PHY_vars_UE->lte_ue_common_vars.dl_ch_estimates[0] + 
				     8*sizeof(int*) + 
				     (j+i*4)*(frame_parms->symbols_per_tti*sizeof(int)*frame_parms->ofdm_symbol_size) - 
				     bigphys_top);

      channel[(j<<1)+i] = (short*)(mem_base + 
				   (unsigned int)PHY_vars_UE->lte_ue_common_vars.dl_ch_estimates_time + 
				   8*sizeof(int*) + 
				   (j+i*4)*(frame_parms->symbols_per_tti*sizeof(int)*frame_parms->ofdm_symbol_size) - 
				   bigphys_top);
    }

    // fix this for visualizing more than 1 antenna!!!
    rx_sig_ptr = (short **)(mem_base + 
			    (unsigned int)PHY_vars_UE->lte_ue_common_vars.rxdata -
			    bigphys_top);
    rx_sig[i] = (short *)(mem_base + 
			  (unsigned int)rx_sig_ptr[i] -
			  bigphys_top); 

    rx_sig_f_ptr = (short **)(mem_base + 
			    (unsigned int)PHY_vars_UE->lte_ue_common_vars.rxdataF -
			    bigphys_top);
    rx_sig_f[i] = (short *)(mem_base + 
			  (unsigned int)rx_sig_f_ptr[i] -
			  bigphys_top); 
  }

  sync_corr = (int*)(mem_base + 
		     (unsigned int)PHY_vars_UE->lte_ue_common_vars.sync_corr - 
		     bigphys_top);
  printf("sync_corr = %p\n", sync_corr);

  // only if UE
  lte_ue_pbch = (LTE_UE_PBCH *) (mem_base + 
				 (unsigned int)PHY_vars_UE->lte_ue_pbch_vars[0] - 
				 bigphys_top);

  printf("lte_ue_pbch (kernel) = %p\n",PHY_vars_UE->lte_ue_pbch_vars[0]);
  printf("lte_ue_pbch (local)= %p\n",lte_ue_pbch);

  pbch_comp = (short*)(mem_base + 
		       (unsigned int)lte_ue_pbch->rxdataF_comp + 
		       8*sizeof(int*) +
		       8*sizeof(short*) -// because there is dl_ch_estimates_ext allocated in between
		       bigphys_top);


  pbch_llr = (char*) (mem_base + 
		       (unsigned int)lte_ue_pbch->llr - 
		       bigphys_top);
  
  printf("pbch_comp = %p\n",pbch_comp);
  printf("pbch_llr= %p\n",pbch_llr);

  lte_ue_pdcch = (LTE_UE_PDCCH *) (mem_base + 
				   (unsigned int)PHY_vars_UE->lte_ue_pdcch_vars[0] - 
				   bigphys_top);

  printf("lte_ue_pdcch (kernel) = %p\n",PHY_vars_UE->lte_ue_pdcch_vars[0]);
  printf("lte_ue_pdcch (local)= %p\n",lte_ue_pdcch);

  pdcch_comp = (short*)(mem_base + 
			(unsigned int)lte_ue_pdcch->rxdataF_comp + 
			8*sizeof(int*) -
			bigphys_top);

  printf("pdcch_comp = %p\n",pdcch_comp);


  lte_ue_dlsch = (LTE_UE_DLSCH *) (mem_base + 
				   (unsigned int)PHY_vars_UE->lte_ue_dlsch_vars[0] - 
				   bigphys_top);

  printf("lte_ue_dlsch (kernel)= %p\n",PHY_vars_UE->lte_ue_dlsch_vars[0]);
  printf("lte_ue_dlsch (local) = %p\n",lte_ue_dlsch);

  dlsch_comp = (short*)(mem_base + 
			(unsigned int)lte_ue_dlsch->rxdataF_comp + 
			8*sizeof(int*) - 
			bigphys_top);
  
  dlsch_llr = (short*) (mem_base + 
			(unsigned int)lte_ue_dlsch->llr[0] - 
			bigphys_top);

  printf("dlsch_comp = %p\n",dlsch_comp);
  printf("dlsch_llr = %p\n",dlsch_llr);
  
  sprintf(title, "LTE SCOPE"),

  fl_initialize(&argc, argv, title, 0, 0);    /* SIGSCOPE */
  form = create_form_lte_scope();                 /* SIGSCOPE */
  fl_show_form(form->lte_scope,FL_PLACE_HOTSPOT,FL_FULLBORDER,title);   /* SIGSCOPE */

  do_scope();
  sleep(5);
  close(openair_fd);
  return(0);
}