/*
 *  Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

/* kenny.c  - Main function for the iSAC coder */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

#ifdef WIN32
#include "windows.h"
#ifndef CLOCKS_PER_SEC
#define CLOCKS_PER_SEC 1000
#endif
#endif

#include <ctype.h>
#include <math.h>

/* include API */
#include "isac.h"
#include "utility.h"
#include "webrtc/rtc_base/format_macros.h"

/* max number of samples per frame (= 60 ms frame) */
#define MAX_FRAMESAMPLES_SWB 1920
/* number of samples per 10ms frame */
#define FRAMESAMPLES_SWB_10ms 320
#define FRAMESAMPLES_WB_10ms 160

/* sampling frequency (Hz) */
#define FS_SWB 32000
#define FS_WB 16000

unsigned long framecnt = 0;

int main(int argc, char* argv[]) {
  //--- File IO ----
  FILE* inp;
  FILE* outp;
  char inname[500];
  char outname[500];

  /* Runtime statistics */
  double rate;
  double rateRCU;
  size_t totalbits = 0;
  unsigned long totalBitsRCU = 0;
  unsigned long totalsmpls = 0;

  int32_t bottleneck = 39;
  int frameSize = 30; /* ms */
  int16_t codingMode = 1;
  int16_t shortdata[FRAMESAMPLES_SWB_10ms];
  int16_t decoded[MAX_FRAMESAMPLES_SWB];
  int16_t speechType[1];
  int16_t payloadLimit;
  int32_t rateLimit;
  ISACStruct* ISAC_main_inst;

  size_t stream_len = 0;
  int declen = 0;
  int16_t err;
  int cur_framesmpls;
  int endfile;
#ifdef WIN32
  double length_file;
  double runtime;
  char outDrive[10];
  char outPath[500];
  char outPrefix[500];
  char outSuffix[500];
  char bitrateFileName[500];
  FILE* bitrateFile;
  double starttime;
  double rateLB = 0;
  double rateUB = 0;
#endif
  FILE* histFile;
  FILE* averageFile;
  int sampFreqKHz;
  int samplesIn10Ms;
  size_t maxStreamLen = 0;
  char histFileName[500];
  char averageFileName[500];
  unsigned int hist[600];
  double tmpSumStreamLen = 0;
  unsigned int packetCntr = 0;
  unsigned int lostPacketCntr = 0;
  uint8_t payload[1200];
  uint8_t payloadRCU[1200];
  uint16_t packetLossPercent = 0;
  int16_t rcuStreamLen = 0;
  int onlyEncode;
  int onlyDecode;

  BottleNeckModel packetData;
  packetData.arrival_time = 0;
  packetData.sample_count = 0;
  packetData.rtp_number = 0;
  memset(hist, 0, sizeof(hist));

  /* handling wrong input arguments in the command line */
  if (argc < 5) {
    int size;
    WebRtcIsac_AssignSize(&size);

    printf("\n\nWrong number of arguments or flag values.\n\n");

    printf("Usage:\n\n");
    printf("%s infile outfile -bn bottleneck [options]\n\n", argv[0]);
    printf("with:\n");
    printf("-I.............. indicates encoding in instantaneous mode.\n");
    printf("-bn bottleneck.. the value of the bottleneck in bit/sec, e.g.\n");
    printf("                 39742, in instantaneous (channel-independent)\n");
    printf("                 mode.\n\n");
    printf("infile.......... Normal speech input file\n\n");
    printf("outfile......... Speech output file\n\n");
    printf("OPTIONS\n");
    printf("-------\n");
    printf("-fs sampFreq.... sampling frequency of codec 16 or 32 (default)\n");
    printf("                 kHz.\n");
    printf("-plim payloadLim payload limit in bytes, default is the maximum\n");
    printf("                 possible.\n");
    printf("-rlim rateLim... rate limit in bits/sec, default is the maximum\n");
    printf("                 possible.\n");
    printf("-h file......... record histogram and *append* to 'file'.\n");
    printf("-ave file....... record average rate of 3 sec intervales and\n");
    printf("                 *append* to 'file'.\n");
    printf("-ploss.......... packet-loss percentage.\n");
    printf("-enc............ do only encoding and store the bit-stream\n");
    printf("-dec............ the input file is a bit-stream, decode it.\n\n");
    printf("Example usage:\n\n");
    printf("%s speechIn.pcm speechOut.pcm -B 40000 -fs 32\n\n", argv[0]);
    printf("structure size %d bytes\n", size);

    exit(0);
  }

  /* Get Bottleneck value */
  bottleneck = readParamInt(argc, argv, "-bn", 50000);
  fprintf(stderr, "\nfixed bottleneck rate of %d bits/s\n\n", bottleneck);

  /* Get Input and Output files */
  sscanf(argv[1], "%s", inname);
  sscanf(argv[2], "%s", outname);
  codingMode = readSwitch(argc, argv, "-I");
  sampFreqKHz = (int16_t)readParamInt(argc, argv, "-fs", 32);
  if (readParamString(argc, argv, "-h", histFileName, 500) > 0) {
    histFile = fopen(histFileName, "a");
    if (histFile == NULL) {
      printf("cannot open hist file %s", histFileName);
      exit(0);
    }
  } else {
    // NO recording of hitstogram
    histFile = NULL;
  }

  packetLossPercent = readParamInt(argc, argv, "-ploss", 0);

  if (readParamString(argc, argv, "-ave", averageFileName, 500) > 0) {
    averageFile = fopen(averageFileName, "a");
    if (averageFile == NULL) {
      printf("cannot open file to write rate %s", averageFileName);
      exit(0);
    }
  } else {
    averageFile = NULL;
  }

  onlyEncode = readSwitch(argc, argv, "-enc");
  onlyDecode = readSwitch(argc, argv, "-dec");

  switch (sampFreqKHz) {
    case 16: {
      samplesIn10Ms = 160;
      break;
    }
    case 32: {
      samplesIn10Ms = 320;
      break;
    }
    default:
      printf("A sampling frequency of %d kHz is not supported, valid values are"
             " 8 and 16.\n", sampFreqKHz);
      exit(-1);
  }
  payloadLimit = (int16_t)readParamInt(argc, argv, "-plim", 400);
  rateLimit = readParamInt(argc, argv, "-rlim", 106800);

  if ((inp = fopen(inname, "rb")) == NULL) {
    printf("  iSAC: Cannot read file %s.\n", inname);
    exit(1);
  }
  if ((outp = fopen(outname, "wb")) == NULL) {
    printf("  iSAC: Cannot write file %s.\n", outname);
    exit(1);
  }

#ifdef WIN32
  _splitpath(outname, outDrive, outPath, outPrefix, outSuffix);
  _makepath(bitrateFileName, outDrive, outPath, "bitrate", ".txt");

  bitrateFile = fopen(bitrateFileName, "a");
  fprintf(bitrateFile, "%%  %s  \n", inname);
#endif

  printf("\n");
  printf("Input.................... %s\n", inname);
  printf("Output................... %s\n", outname);
  printf("Encoding Mode............ %s\n",
         (codingMode == 1) ? "Channel-Independent" : "Channel-Adaptive");
  printf("Bottleneck............... %d bits/sec\n", bottleneck);
  printf("Packet-loss Percentage... %d\n", packetLossPercent);
  printf("\n");

#ifdef WIN32
  starttime = clock() / (double)CLOCKS_PER_SEC; /* Runtime statistics */
#endif

  /* Initialize the ISAC and BN structs */
  err = WebRtcIsac_Create(&ISAC_main_inst);

  WebRtcIsac_SetEncSampRate(ISAC_main_inst, sampFreqKHz * 1000);
  WebRtcIsac_SetDecSampRate(ISAC_main_inst, sampFreqKHz >= 32 ? 32000 : 16000);
  /* Error check */
  if (err < 0) {
    fprintf(stderr, "\n\n Error in create.\n\n");
    exit(EXIT_FAILURE);
  }

  framecnt = 0;
  endfile = 0;

  /* Initialize encoder and decoder */
  if (WebRtcIsac_EncoderInit(ISAC_main_inst, codingMode) < 0) {
    printf("cannot initialize encoder\n");
    return -1;
  }
  WebRtcIsac_DecoderInit(ISAC_main_inst);

  if (codingMode == 1) {
    if (WebRtcIsac_Control(ISAC_main_inst, bottleneck, frameSize) < 0) {
      printf("cannot set bottleneck\n");
      return -1;
    }
  } else {
    if (WebRtcIsac_ControlBwe(ISAC_main_inst, 15000, 30, 1) < 0) {
      printf("cannot configure BWE\n");
      return -1;
    }
  }

  if (WebRtcIsac_SetMaxPayloadSize(ISAC_main_inst, payloadLimit) < 0) {
    printf("cannot set maximum payload size %d.\n", payloadLimit);
    return -1;
  }

  if (rateLimit < 106800) {
    if (WebRtcIsac_SetMaxRate(ISAC_main_inst, rateLimit) < 0) {
      printf("cannot set the maximum rate %d.\n", rateLimit);
      return -1;
    }
  }

  while (endfile == 0) {
    fprintf(stderr, "  \rframe = %7li", framecnt);

    //============== Readind from the file and encoding =================
    cur_framesmpls = 0;
    stream_len = 0;

    if (onlyDecode) {
      uint8_t auxUW8;
      if (fread(&auxUW8, sizeof(uint8_t), 1, inp) < 1) {
        break;
      }
      stream_len = auxUW8 << 8;
      if (fread(&auxUW8, sizeof(uint8_t), 1, inp) < 1) {
        break;
      }
      stream_len |= auxUW8;
      if (fread(payload, 1, stream_len, inp) < stream_len) {
        printf("last payload is corrupted\n");
        break;
      }
    } else {
      while (stream_len == 0) {
        int stream_len_int;

        // Read 10 ms speech block
        endfile = readframe(shortdata, inp, samplesIn10Ms);
        if (endfile) {
          break;
        }
        cur_framesmpls += samplesIn10Ms;

        //-------- iSAC encoding ---------
        stream_len_int = WebRtcIsac_Encode(ISAC_main_inst, shortdata, payload);

        if (stream_len_int < 0) {
          // exit if returned with error
          // errType=WebRtcIsac_GetErrorCode(ISAC_main_inst);
          fprintf(stderr, "\nError in encoder\n");
          getc(stdin);
          exit(EXIT_FAILURE);
        }
        stream_len = (size_t)stream_len_int;
      }
      //===================================================================
      if (endfile) {
        break;
      }

      rcuStreamLen = WebRtcIsac_GetRedPayload(ISAC_main_inst, payloadRCU);
      if (rcuStreamLen < 0) {
        fprintf(stderr, "\nError getting RED payload\n");
        getc(stdin);
        exit(EXIT_FAILURE);
      }

      get_arrival_time(cur_framesmpls, stream_len, bottleneck, &packetData,
                       sampFreqKHz * 1000, sampFreqKHz * 1000);
      if (WebRtcIsac_UpdateBwEstimate(
              ISAC_main_inst, payload, stream_len, packetData.rtp_number,
              packetData.sample_count, packetData.arrival_time) < 0) {
        printf(" BWE Error at client\n");
        return -1;
      }
    }

    if (endfile) {
      break;
    }

    maxStreamLen = (stream_len > maxStreamLen) ? stream_len : maxStreamLen;
    packetCntr++;

    hist[stream_len]++;
    if (averageFile != NULL) {
      tmpSumStreamLen += stream_len;
      if (packetCntr == 100) {
        // kbps
        fprintf(averageFile, "%8.3f ",
                tmpSumStreamLen * 8.0 / (30.0 * packetCntr));
        packetCntr = 0;
        tmpSumStreamLen = 0;
      }
    }

    if (onlyEncode) {
      uint8_t auxUW8;
      auxUW8 = (uint8_t)(((stream_len & 0x7F00) >> 8) & 0xFF);
      if (fwrite(&auxUW8, sizeof(uint8_t), 1, outp) != 1) {
        return -1;
      }

      auxUW8 = (uint8_t)(stream_len & 0xFF);
      if (fwrite(&auxUW8, sizeof(uint8_t), 1, outp) != 1) {
        return -1;
      }
      if (fwrite(payload, 1, stream_len, outp) != stream_len) {
        return -1;
      }
    } else {
      //======================= iSAC decoding ===========================

      if ((rand() % 100) < packetLossPercent) {
        declen = WebRtcIsac_DecodeRcu(ISAC_main_inst, payloadRCU,
                                      (size_t)rcuStreamLen, decoded,
                                      speechType);
        lostPacketCntr++;
      } else {
        declen = WebRtcIsac_Decode(ISAC_main_inst, payload, stream_len, decoded,
                                   speechType);
      }
      if (declen <= 0) {
        // errType=WebRtcIsac_GetErrorCode(ISAC_main_inst);
        fprintf(stderr, "\nError in decoder.\n");
        getc(stdin);
        exit(1);
      }

      // Write decoded speech frame to file
      if (fwrite(decoded, sizeof(int16_t), declen, outp) != (size_t)declen) {
        return -1;
      }
      cur_framesmpls = declen;
    }
    // Update Statistics
    framecnt++;
    totalsmpls += cur_framesmpls;
    if (stream_len > 0) {
      totalbits += 8 * stream_len;
    }
    if (rcuStreamLen > 0) {
      totalBitsRCU += 8 * rcuStreamLen;
    }
  }

  rate = ((double)totalbits * (sampFreqKHz)) / (double)totalsmpls;
  rateRCU = ((double)totalBitsRCU * (sampFreqKHz)) / (double)totalsmpls;

  printf("\n\n");
  printf("Sampling Rate............... %d kHz\n", sampFreqKHz);
  printf("Payload Limit............... %d bytes \n", payloadLimit);
  printf("Rate Limit.................. %d bits/sec \n", rateLimit);

#ifdef WIN32
  fprintf(bitrateFile, "%d  %10u     %d     %6.3f  %6.3f    %6.3f\n",
          sampFreqKHz, framecnt, bottleneck, rateLB, rateUB, rate);
  fclose(bitrateFile);
#endif  // WIN32

  printf("\n");
  printf("Measured bit-rate........... %0.3f kbps\n", rate);
  printf("Measured RCU bit-ratre...... %0.3f kbps\n", rateRCU);
  printf("Maximum bit-rate/payloadsize %0.3f / %" PRIuS "\n",
         maxStreamLen * 8 / 0.03, maxStreamLen);
  printf("Measured packet-loss........ %0.1f%% \n",
         100.0f * (float)lostPacketCntr / (float)packetCntr);

  printf("\n");

/* Runtime statistics */
#ifdef WIN32
  runtime = (double)(clock() / (double)CLOCKS_PER_SEC - starttime);
  length_file = ((double)framecnt * (double)declen / (sampFreqKHz * 1000));
  printf("Length of speech file....... %.1f s\n", length_file);
  printf("Time to run iSAC............ %.2f s (%.2f %% of realtime)\n\n",
         runtime, (100 * runtime / length_file));
#endif
  printf("\n\n_______________________________________________\n");

  if (histFile != NULL) {
    int n;
    for (n = 0; n < 600; n++) {
      fprintf(histFile, "%6d ", hist[n]);
    }
    fprintf(histFile, "\n");
    fclose(histFile);
  }
  if (averageFile != NULL) {
    if (packetCntr > 0) {
      fprintf(averageFile, "%8.3f ",
              tmpSumStreamLen * 8.0 / (30.0 * packetCntr));
    }
    fprintf(averageFile, "\n");
    fclose(averageFile);
  }

  fclose(inp);
  fclose(outp);

  WebRtcIsac_Free(ISAC_main_inst);

  exit(0);
}