/*
 * Copyright 2013-2019 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:         sss.h
 *
 *  Description:  Secondary synchronization signal (SSS) generation and detection.
 *
 *
 *  Reference:    3GPP TS 36.211 version 10.0.0 Release 10 Sec. 6.11.2
 *****************************************************************************/

#ifndef SRSLTE_SSS_H
#define SRSLTE_SSS_H

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

#include "srslte/config.h"
#include "srslte/phy/common/phy_common.h"
#include "srslte/phy/dft/dft.h"


#define SRSLTE_SSS_N      31
#define SRSLTE_SSS_LEN    2*SRSLTE_SSS_N

typedef struct SRSLTE_API {
  int z1[SRSLTE_SSS_N][SRSLTE_SSS_N];
  int c[2][SRSLTE_SSS_N];
  int s[SRSLTE_SSS_N][SRSLTE_SSS_N];
}srslte_sss_tables_t;

/* Allocate 32 complex to make it multiple of 32-byte AVX instructions alignment requirement.
 * Should use srslte_vec_malloc() to make it platform agnostic.
 */
typedef struct SRSLTE_API {
  float z1[SRSLTE_SSS_N][SRSLTE_SSS_N];
  float c[2][SRSLTE_SSS_N];
  float s[SRSLTE_SSS_N][SRSLTE_SSS_N];
  float sd[SRSLTE_SSS_N][SRSLTE_SSS_N-1];
}srslte_sss_fc_tables_t;


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

  srslte_dft_plan_t dftp_input;
  
  uint32_t fft_size;
  uint32_t max_fft_size;

  float corr_peak_threshold;
  uint32_t symbol_sz;
  uint32_t subframe_sz;
  uint32_t N_id_2;
  
  uint32_t N_id_1_table[30][30];
  srslte_sss_fc_tables_t fc_tables[3]; // one for each N_id_2

  float corr_output_m0[SRSLTE_SSS_N];
  float corr_output_m1[SRSLTE_SSS_N];

}srslte_sss_t;


/* Basic functionality */
SRSLTE_API int srslte_sss_init(srslte_sss_t *q,
                                     uint32_t fft_size);

SRSLTE_API int srslte_sss_resize(srslte_sss_t *q,
                                        uint32_t fft_size); 

SRSLTE_API void srslte_sss_free(srslte_sss_t *q);

SRSLTE_API void srslte_sss_generate(float *signal0, 
                                    float *signal5, 
                                    uint32_t cell_id);

SRSLTE_API void srslte_sss_put_slot(float *sss, 
                                    cf_t *symbol, 
                                    uint32_t nof_prb, 
                                    srslte_cp_t cp);

SRSLTE_API void srslte_sss_put_symbol(float* sss, cf_t* symbol, uint32_t nof_prb);

SRSLTE_API int srslte_sss_set_N_id_2(srslte_sss_t* q, uint32_t N_id_2);

SRSLTE_API int srslte_sss_m0m1_partial(srslte_sss_t *q,
                                             const cf_t *input,
                                             uint32_t M, 
                                             cf_t ce[2*SRSLTE_SSS_N],
                                             uint32_t *m0, 
                                             float *m0_value, 
                                             uint32_t *m1, 
                                             float *m1_value);

SRSLTE_API int srslte_sss_m0m1_diff_coh(srslte_sss_t *q,
                                              const cf_t *input,
                                              cf_t ce[2*SRSLTE_SSS_N],
                                              uint32_t *m0, 
                                              float *m0_value, 
                                              uint32_t *m1, 
                                              float *m1_value);

SRSLTE_API int srslte_sss_m0m1_diff(srslte_sss_t *q,
                                          const cf_t *input,
                                          uint32_t *m0, 
                                          float *m0_value, 
                                          uint32_t *m1, 
                                          float *m1_value);


SRSLTE_API uint32_t srslte_sss_subframe(uint32_t m0,
                                              uint32_t m1);

SRSLTE_API int srslte_sss_N_id_1(srslte_sss_t* q, uint32_t m0, uint32_t m1, float corr);

SRSLTE_API int srslte_sss_frame(srslte_sss_t *q,
                                      cf_t *input, 
                                      uint32_t *subframe_idx, 
                                      uint32_t *N_id_1);

SRSLTE_API void srslte_sss_set_threshold(srslte_sss_t *q,
                                               float threshold);

SRSLTE_API void srslte_sss_set_symbol_sz(srslte_sss_t *q,
                                               uint32_t symbol_sz);

SRSLTE_API void srslte_sss_set_subframe_sz(srslte_sss_t *q,
                                                 uint32_t subframe_sz);

#endif // SRSLTE_SSS_H