/**
 *
 * \section COPYRIGHT
 *
 * Copyright 2013-2014 The libLTE Developers. See the
 * COPYRIGHT file at the top-level directory of this distribution.
 *
 * \section LICENSE
 *
 * This file is part of the libLTE library.
 *
 * libLTE is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version.
 *
 * libLTE 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 Lesser General Public License for more details.
 *
 * A copy of the GNU Lesser 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/.
 *
 */

#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <strings.h>
#include <stdlib.h>
#include <stdbool.h>
#include <assert.h>
#include <math.h>

#include "liblte/phy/phch/regs.h"
#include "liblte/phy/phch/pcfich.h"
#include "liblte/phy/common/base.h"
#include "liblte/phy/utils/bit.h"
#include "liblte/phy/utils/vector.h"
#include "liblte/phy/utils/debug.h"

// Table 5.3.4-1
static char cfi_table[4][PCFICH_CFI_LEN] = { { 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1,
    1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1 }, { 1, 0, 1,
    1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1,
    0, 1, 1, 0 }, { 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1,
    0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1 }, { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } // reserved
};

bool pcfich_exists(int nframe, int nslot) {
  return true;
}

/** Initializes the pcfich channel receiver */
int pcfich_init(pcfich_t *q, regs_t *regs, int cell_id, int nof_prb,
    int nof_ports, lte_cp_t cp) {
  int ret = -1;
  if (cell_id < 0) {
    return -1;
  }
  bzero(q, sizeof(pcfich_t));
  q->cell_id = cell_id;
  q->cp = cp;
  q->regs = regs;
  q->nof_prb = nof_prb;
  q->nof_ports = nof_ports;

  if (modem_table_std(&q->mod, LTE_QPSK, false)) {
    goto clean;
  }

  demod_hard_init(&q->demod);
  demod_hard_table_set(&q->demod, LTE_QPSK);

  for (int nsf = 0; nsf < NSUBFRAMES_X_FRAME; nsf++) {
    if (sequence_pcfich(&q->seq_pcfich[nsf], 2 * nsf, q->cell_id)) {
      goto clean;
    }
  }

  q->nof_symbols = PCFICH_RE;

  ret = 0;
  clean: if (ret == -1) {
    pcfich_free(q);
  }
  return ret;
}

void pcfich_free(pcfich_t *q) {
  for (int ns = 0; ns < NSUBFRAMES_X_FRAME; ns++) {
    sequence_free(&q->seq_pcfich[ns]);
  }
  modem_table_free(&q->mod);
}

/** Finds the CFI with minimum distance with the vector of received 32 bits.
 * Saves the CFI value in the cfi pointer and returns the distance.
 */
int pcfich_cfi_decode(char bits[PCFICH_CFI_LEN], int *cfi) {
  int i, j;
  int distance, index = -1;
  int min = 32;

  for (i = 0; i < 3; i++) {
    distance = 0;
    for (j = 0; j < PCFICH_CFI_LEN; j++) {
      distance += (bits[j] ^ cfi_table[i][j]);
    }
    DEBUG("CFI=%d, distance:%d\n", i, distance);
    if (distance < min) {
      min = distance;
      index = i;
    }
  }
  if (cfi) {
    *cfi = index + 1;
  }
  return min;

}

/** Encodes the CFI producing a vector of 32 bits.
 *  36.211 10.3 section 5.3.4
 */
int pcfich_cfi_encode(int cfi, char bits[PCFICH_CFI_LEN]) {
  if (cfi < 1 || cfi > 3) {
    fprintf(stderr, "Invalid CFI %d\n", cfi);
    return -1;
  }
  memcpy(bits, cfi_table[cfi - 1], PCFICH_CFI_LEN * sizeof(char));
  return 0;
}

/* Decodes the PCFICH channel and saves the CFI in the cfi pointer.
 *
 * Returns 1 if successfully decoded the CFI, 0 if not and -1 on error
 */
int pcfich_decode(pcfich_t *q, cf_t *slot_symbols, cf_t *ce[MAX_PORTS_CTRL],
    int nsubframe, int *cfi, int *distance) {
  int dist;

  /* Set pointers for layermapping & precoding */
  int i;
  cf_t *x[MAX_LAYERS];
  cf_t *ce_precoding[MAX_PORTS];

  if (nsubframe < 0 || nsubframe > NSUBFRAMES_X_FRAME) {
    fprintf(stderr, "Invalid nslot %d\n", nsubframe);
    return -1;
  }

  /* number of layers equals number of ports */
  for (i = 0; i < MAX_PORTS_CTRL; i++) {
    x[i] = q->pcfich_x[i];
  }
  for (i = 0; i < MAX_PORTS; i++) {
    ce_precoding[i] = q->ce[i];
  }

  /* extract symbols */
  if (q->nof_symbols
      != regs_pcfich_get(q->regs, slot_symbols, q->pcfich_symbols[0])) {
    fprintf(stderr, "There was an error getting the PCFICH symbols\n");
    return -1;
  }

  /* extract channel estimates */
  for (i = 0; i < q->nof_ports; i++) {
    if (q->nof_symbols != regs_pcfich_get(q->regs, ce[i], q->ce[i])) {
      fprintf(stderr, "There was an error getting the PCFICH symbols\n");
      return -1;
    }
  }

  /* in control channels, only diversity is supported */
  if (q->nof_ports == 1) {
    /* no need for layer demapping */
    predecoding_single_zf(q->pcfich_symbols[0], q->ce[0], q->pcfich_d,
        q->nof_symbols);
  } else {
    predecoding_diversity_zf(q->pcfich_symbols[0], ce_precoding, x,
        q->nof_ports, q->nof_symbols);
    layerdemap_diversity(x, q->pcfich_d, q->nof_ports,
        q->nof_symbols / q->nof_ports);
  }

  /* demodulate symbols */
  demod_hard_demodulate(&q->demod, q->pcfich_d, q->data, q->nof_symbols);

  /* Scramble with the sequence for slot nslot */
  scrambling_b(&q->seq_pcfich[nsubframe], q->data);

  /* decode CFI */
  dist = pcfich_cfi_decode(q->data, cfi);
  if (distance) {
    *distance = dist;
  }
  if (dist < PCFICH_MAX_DISTANCE) {
    return 1;
  } else {
    return 0;
  }
}

/** Encodes CFI and maps symbols to the slot
 */
int pcfich_encode(pcfich_t *q, int cfi, cf_t *slot_symbols[MAX_PORTS_CTRL],
    int nsubframe) {
  int i;

  if (nsubframe < 0 || nsubframe > NSUBFRAMES_X_FRAME) {
    fprintf(stderr, "Invalid nslot %d\n", nsubframe);
    return -1;
  }

  /* Set pointers for layermapping & precoding */
  cf_t *x[MAX_LAYERS];
  cf_t *symbols_precoding[MAX_PORTS];

  /* number of layers equals number of ports */
  for (i = 0; i < q->nof_ports; i++) {
    x[i] = q->pcfich_x[i];
  }
  for (i = 0; i < MAX_PORTS; i++) {
    symbols_precoding[i] = q->pcfich_symbols[i];
  }

  /* pack MIB */
  pcfich_cfi_encode(cfi, q->data);

  /* scramble for slot sequence nslot */
  scrambling_b(&q->seq_pcfich[nsubframe], q->data);

  mod_modulate(&q->mod, q->data, q->pcfich_d, PCFICH_CFI_LEN);

  /* layer mapping & precoding */
  if (q->nof_ports > 1) {
    layermap_diversity(q->pcfich_d, x, q->nof_ports, q->nof_symbols);
    precoding_diversity(x, symbols_precoding, q->nof_ports,
        q->nof_symbols / q->nof_ports);
  } else {
    memcpy(q->pcfich_symbols[0], q->pcfich_d, q->nof_symbols * sizeof(cf_t));
  }

  /* mapping to resource elements */
  for (i = 0; i < q->nof_ports; i++) {
    if (regs_pcfich_put(q->regs, q->pcfich_symbols[i], slot_symbols[i]) < 0) {
      fprintf(stderr, "Error putting PCHICH resource elements\n");
      return -1;
    }
  }

  return 0;
}