/**
 * Copyright 2013-2021 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/.
 *
 */

/******************************************************************************
 *  File:         pss.h
 *
 *  Description:  Primary synchronization signal (PSS) generation and detection.
 *
 *                The srslte_pss_t object provides functions for fast
 *                computation of the crosscorrelation between the PSS and received
 *                signal and CFO estimation. Also, the function srslte_pss_tperiodic()
 *                is designed to be called periodically every subframe, taking
 *                care of the correct data alignment with respect to the PSS sequence.
 *
 *                The object is designed to work with signals sampled at 1.92 Mhz
 *                centered at the carrier frequency. Thus, downsampling is required
 *                if the signal is sampled at higher frequencies.
 *
 *  Reference:    3GPP TS 36.211 version 10.0.0 Release 10 Sec. 6.11.1
 *****************************************************************************/

#ifndef SRSLTE_PSS_H
#define SRSLTE_PSS_H

#include <stdbool.h>
#include <stdint.h>

#include "srslte/config.h"
#include "srslte/phy/common/phy_common.h"
#include "srslte/phy/utils/convolution.h"
#include "srslte/phy/utils/filter.h"

#define CONVOLUTION_FFT

#define SRSLTE_PSS_LEN 62
#define SRSLTE_PSS_RE (6 * 12)

/* PSS processing options */

#define SRSLTE_PSS_ACCUMULATE_ABS // If enabled, accumulates the correlation absolute value on consecutive calls to
                                  // srslte_pss_find_pss

#define SRSLTE_PSS_RETURN_PSR // If enabled returns peak to side-lobe ratio, otherwise returns absolute peak value

/* Low-level API */
typedef struct SRSLTE_API {

#ifdef CONVOLUTION_FFT
  srslte_conv_fft_cc_t conv_fft;
  srslte_filt_cc_t     filter;

#endif
  int decimate;

  uint32_t max_frame_size;
  uint32_t max_fft_size;

  uint32_t frame_size;
  uint32_t N_id_2;
  uint32_t fft_size;
  cf_t*    pss_signal_freq_full[3];

  cf_t* pss_signal_time[3];
  cf_t* pss_signal_time_scale[3];

  cf_t   pss_signal_freq[3][SRSLTE_PSS_LEN]; // One sequence for each N_id_2
  cf_t*  tmp_input;
  cf_t*  conv_output;
  float* conv_output_abs;
  float  ema_alpha;
  float* conv_output_avg;
  float  peak_value;

  bool              filter_pss_enable;
  srslte_dft_plan_t dftp_input;
  srslte_dft_plan_t idftp_input;
  cf_t              tmp_fft[SRSLTE_SYMBOL_SZ_MAX];
  cf_t              tmp_fft2[SRSLTE_SYMBOL_SZ_MAX];

  cf_t tmp_ce[SRSLTE_PSS_LEN];

  bool chest_on_filter;

} srslte_pss_t;

typedef enum { PSS_TX, PSS_RX } pss_direction_t;

/* Basic functionality */
SRSLTE_API int srslte_pss_init_fft(srslte_pss_t* q, uint32_t frame_size, uint32_t fft_size);

SRSLTE_API int srslte_pss_init_fft_offset(srslte_pss_t* q, uint32_t frame_size, uint32_t fft_size, int cfo_i);

SRSLTE_API int
srslte_pss_init_fft_offset_decim(srslte_pss_t* q, uint32_t frame_size, uint32_t fft_size, int cfo_i, int decimate);

SRSLTE_API int srslte_pss_resize(srslte_pss_t* q, uint32_t frame_size, uint32_t fft_size, int offset);

SRSLTE_API int srslte_pss_init(srslte_pss_t* q, uint32_t frame_size);

SRSLTE_API void srslte_pss_free(srslte_pss_t* q);

SRSLTE_API void srslte_pss_reset(srslte_pss_t* q);

SRSLTE_API void srslte_pss_filter_enable(srslte_pss_t* q, bool enable);

SRSLTE_API void srslte_pss_sic(srslte_pss_t* q, cf_t* input);

SRSLTE_API void srslte_pss_filter(srslte_pss_t* q, const cf_t* input, cf_t* output);

SRSLTE_API int srslte_pss_generate(cf_t* signal, uint32_t N_id_2);

SRSLTE_API void srslte_pss_get_slot(cf_t* slot, cf_t* pss_signal, uint32_t nof_prb, srslte_cp_t cp);

SRSLTE_API void srslte_pss_put_slot(cf_t* pss_signal, cf_t* slot, uint32_t nof_prb, srslte_cp_t cp);

SRSLTE_API void srslte_pss_set_ema_alpha(srslte_pss_t* q, float alpha);

SRSLTE_API int srslte_pss_set_N_id_2(srslte_pss_t* q, uint32_t N_id_2);

SRSLTE_API int srslte_pss_find_pss(srslte_pss_t* q, const cf_t* input, float* corr_peak_value);

SRSLTE_API int srslte_pss_chest(srslte_pss_t* q, const cf_t* input, cf_t ce[SRSLTE_PSS_LEN]);

SRSLTE_API float srslte_pss_cfo_compute(srslte_pss_t* q, const cf_t* pss_recv);

#endif // SRSLTE_PSS_H