/**
 * Copyright 2013-2022 Software Radio Systems Limited
 *
 * This file is part of srsRAN.
 *
 * srsRAN 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.
 *
 * srsRAN 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/.
 *
 */

#include <strings.h>

#include "srsran/phy/phch/mib_sl.h"
#include "srsran/phy/utils/bit.h"

int srsran_mib_sl_init(srsran_mib_sl_t* q, srsran_sl_tm_t tm)
{
  if (tm == SRSRAN_SIDELINK_TM1 || tm == SRSRAN_SIDELINK_TM2) {
    q->mib_sl_len = SRSRAN_MIB_SL_LEN;
  } else if (tm == SRSRAN_SIDELINK_TM3 || tm == SRSRAN_SIDELINK_TM4) {
    q->mib_sl_len = SRSRAN_MIB_SL_V2X_LEN;
  } else {
    return SRSRAN_ERROR;
  }

  q->sl_bandwidth_r12           = 0;
  q->direct_frame_number_r12    = 0;
  q->direct_subframe_number_r12 = 0;
  q->in_coverage_r12            = false;
  q->tdd_config_sl_r12          = 0;

  return SRSRAN_SUCCESS;
}

int srsran_mib_sl_set(srsran_mib_sl_t* q,
                      uint32_t         nof_prb,
                      uint32_t         tdd_config,
                      uint32_t         direct_frame_number,
                      uint32_t         direct_subframe_number,
                      bool             in_coverage)
{
  int ret = SRSRAN_ERROR_INVALID_INPUTS;
  if (q != NULL) {
    switch (nof_prb) {
      case 6:
        q->sl_bandwidth_r12 = 0;
        break;
      case 15:
        q->sl_bandwidth_r12 = 1;
        break;
      case 25:
        q->sl_bandwidth_r12 = 2;
        break;
      case 50:
        q->sl_bandwidth_r12 = 3;
        break;
      case 75:
        q->sl_bandwidth_r12 = 4;
        break;
      case 100:
        q->sl_bandwidth_r12 = 5;
        break;
      default:
        printf("Invalid bandwidth\n");
        return SRSRAN_ERROR;
    }
    q->tdd_config_sl_r12          = tdd_config;
    q->direct_frame_number_r12    = direct_frame_number;
    q->direct_subframe_number_r12 = direct_subframe_number;
    q->in_coverage_r12            = in_coverage;

    ret = SRSRAN_SUCCESS;
  }

  return ret;
}

void srsran_mib_sl_pack(srsran_mib_sl_t* q, uint8_t* msg)
{
  bzero(msg, sizeof(uint8_t) * SRSRAN_MIB_SL_MAX_LEN);
  srsran_bit_unpack(q->sl_bandwidth_r12, &msg, 3);
  srsran_bit_unpack(q->tdd_config_sl_r12, &msg, 3);
  srsran_bit_unpack(q->direct_frame_number_r12, &msg, 10);
  srsran_bit_unpack(q->direct_subframe_number_r12, &msg, 4);
  srsran_bit_unpack((uint32_t)q->in_coverage_r12, &msg, 1);
}

void srsran_mib_sl_unpack(srsran_mib_sl_t* q, uint8_t* msg)
{
  q->sl_bandwidth_r12           = srsran_bit_pack(&msg, 3);
  q->tdd_config_sl_r12          = srsran_bit_pack(&msg, 3);
  q->direct_frame_number_r12    = srsran_bit_pack(&msg, 10);
  q->direct_subframe_number_r12 = srsran_bit_pack(&msg, 4);
  q->in_coverage_r12            = (bool)srsran_bit_pack(&msg, 1);
}

void srsran_mib_sl_printf(FILE* f, srsran_mib_sl_t* q)
{
  fprintf(f, " - Bandwidth:              %i\n", q->sl_bandwidth_r12);
  fprintf(f, " - Direct Frame Number:    %i\n", q->direct_frame_number_r12);
  fprintf(f, " - Direct Subframe Number: %i\n", q->direct_subframe_number_r12);
  fprintf(f, " - TDD config:             %i\n", q->tdd_config_sl_r12);
  fprintf(f, " - In coverage:            %s\n", q->in_coverage_r12 ? "yes" : "no");
}

void srsran_mib_sl_free(srsran_mib_sl_t* q)
{
  if (q != NULL) {
    bzero(q, sizeof(srsran_mib_sl_t));
  }
}