/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set et sw=2 ts=2: */
/***************************************************************************
 *            test_init.cc
 *
 *  Mon Sep  2 14:02:16 CEST 2013
 *  Copyright 2013 Bent Bisballe Nyeng
 *  deva@aasimon.org
 ****************************************************************************/

/*
 *  This file is part of lrtp.
 *
 * lrtp 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 2.1 of the License, or (at your option) any later version.
 *
 * lrtp 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.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with lrtp; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */
#include <lrtp.h>

#include <stdio.h>

#include <string>
#include <vector>

#include <math.h>
#include <opus/opus.h>
#include <limits.h>
#include <ao/ao.h>

#define KEY "123456789012345678901234567890123456789012345678901234567890"
#define SSRC 1234567890

#define FS 8000

#define F1 440
#define AF1 0.3

#define F2 500
#define AF2 0.7

void dump(const char *title, const char *buf, size_t size)
{
  printf("%12s: ", title);
  for(int i = 0; i < size; i++) {
    if(i % 8 == 0) printf(" ");
    printf("%02x ", (unsigned char)*buf++);
  }
  printf("\n");
}

// Frame sizes based on frame type index:
static const int wb_frame_size[16] = {
  17, 23, 32, 36, 40, 46, 50, 58,
  60, 5, -1, -1, -1, -1, -1, 0
};

int main()
{
  std::vector<std::string> packets;
  unsigned int csrc = 42;
  
  int frame_type_index = 8;

  printf("========== Encode ==========\n");

  { // Encode
    struct lrtp_t *lrtp = lrtp_init(KEY, SSRC);

    struct lrtp_profile_t *profile =
      lrtp_create_profile(lrtp, PROFILE_AMRWB, csrc,
                          OPTION_AMRWB_FRAME_TYPE_INDEX, frame_type_index,
                          OPTION_END);

    char packet[16*1024];
    size_t packetsize = sizeof(packet);

    int cnt = 0;
    
    for(int i = 0; i < 10; i++) {

      size_t num_frames = 10;
      char frame[wb_frame_size[frame_type_index] * num_frames];
      size_t framesize = wb_frame_size[frame_type_index] * num_frames;

      for(int i = 0; i < framesize; i++) frame[i] = cnt++;

      int timestamp = 0;

      int ret = lrtp_enqueue_frame(profile, frame, framesize);
      while( (ret = lrtp_pack(lrtp, packet, sizeof(packet))) != 0) {
        std::string p;
        p.append(packet, ret);
        packets.push_back(p);
      }
    }

    lrtp_destroy_profile(lrtp, csrc);
    lrtp_close(lrtp);
  }

  printf("========== Decode ==========\n");

  { // Decode
    struct lrtp_t *lrtp = lrtp_init(KEY, SSRC);

    struct lrtp_profile_t *profile =
      lrtp_create_profile(lrtp, PROFILE_AMRWB, csrc,
                          OPTION_AMRWB_FRAME_TYPE_INDEX, frame_type_index,
                          OPTION_END);

    char frame[16*1024];
    size_t framesize = sizeof(frame);
    
    int cnt = 0;

    std::vector<std::string>::iterator i = packets.begin();
    while(i != packets.end()) {
      size_t packetsize = i->size();
      printf("unpack sz: %d\n", packetsize);
      const char *packet = i->data();
      unsigned int ts;

      framesize = sizeof(frame);

      lrtp_unpack(lrtp, packet, packetsize);
      int ret;
      while((ret = lrtp_dequeue_frame(lrtp, frame, framesize, &csrc, &ts)) != 0) {
        printf("Got %d bytes, csrc %d, ts: %d\n", ret, csrc, ts);
        
        printf("cnt:\n");
        for(int i = 0; i < ret; i++) {
          printf("%02x ", (unsigned char)frame[i]);
        }
        printf("\n");
      }

      i++;
    }

    lrtp_destroy_profile(lrtp, csrc);
    lrtp_close(lrtp);
  }

  return 0;
}