/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set et sw=2 ts=2: */
/***************************************************************************
 *            test_amrwb.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 <cppunit/extensions/HelperMacros.h>

#include <lrtp.h>

#include <stdio.h>

#include <string>
#include <vector>

#define KEY "123456789012345678901234567890123456789012345678901234567890"
#define SSRC 1234567890

#define PKG_SIZE (16*1024)
#define NUM_PKGS 10

// 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
};

class test_amrwb_class : public CppUnit::TestFixture
{
  CPPUNIT_TEST_SUITE(test_amrwb_class);
	CPPUNIT_TEST(test_amrwb_0);
	CPPUNIT_TEST(test_amrwb_1);
	CPPUNIT_TEST(test_amrwb_2);
	CPPUNIT_TEST(test_amrwb_3);
	CPPUNIT_TEST(test_amrwb_4);
	CPPUNIT_TEST(test_amrwb_5);
	CPPUNIT_TEST(test_amrwb_6);
	CPPUNIT_TEST(test_amrwb_7);
	CPPUNIT_TEST(test_amrwb_8);
	CPPUNIT_TEST_SUITE_END();

public:
	void setUp() {}
	void tearDown() {}

  void test_amrwb(int frame_type_index) {
    std::vector<std::string> packets;
    unsigned int csrc = 42;
  
    // int sent = 0;

    size_t num_frames = NUM_PKGS;
    size_t framesize = wb_frame_size[frame_type_index] * num_frames;

    { // Encode
      lrtp_status_t status;
      struct lrtp_t *lrtp = lrtp_init(&status, KEY, SSRC);
      CPPUNIT_ASSERT_EQUAL(status, LRTP_OK);

      int x =
        lrtp_create_profile(lrtp, PROFILE_AMRWB, csrc,
                            OPTION_AMRWB_FRAME_TYPE_INDEX, frame_type_index,
                            OPTION_END);
      CPPUNIT_ASSERT_EQUAL(0, x);

      char packet[PKG_SIZE];
      //size_t packetsize = sizeof(packet);

      char cnt = 0;
    
      int timestamp = 0;

      for(int i = 0; i < NUM_PKGS; i++) {
        char frame[wb_frame_size[frame_type_index] * num_frames];

        for(size_t j = 0; j < framesize; j++) frame[j] = cnt++;

        int ret = lrtp_enqueue_frame(lrtp, csrc, frame, framesize, timestamp++,
                                     LRTP_COPY);
        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);
    }

    { // Decode
      lrtp_status_t status;
      struct lrtp_t *lrtp = lrtp_init(&status, KEY, SSRC);
      CPPUNIT_ASSERT_EQUAL(status, LRTP_OK);

      int x =
        lrtp_create_profile(lrtp, PROFILE_AMRWB, csrc,
                            OPTION_AMRWB_FRAME_TYPE_INDEX, frame_type_index,
                            OPTION_END);
      CPPUNIT_ASSERT_EQUAL(0, x);

      char frame[16*1024];
    
      char cnt = 0;
      int total = 0;

      std::vector<std::string>::iterator i = packets.begin();
      while(i != packets.end()) {
        size_t packetsize = i->size();
        const char *packet = i->data();
        unsigned int ts;

        lrtp_unpack(lrtp, packet, packetsize);
        int ret;
        while((ret = lrtp_dequeue_frame(lrtp, frame, sizeof(frame), &csrc, &ts))
              != 0) {
          int err = 0;
          for(int i = 0; i < ret; i++) {
            err += abs(frame[i] - cnt);
            cnt++;
            total++;
          }
          CPPUNIT_ASSERT_EQUAL(0, err);
        }

        i++;
      }

      CPPUNIT_ASSERT_EQUAL((int)(framesize * NUM_PKGS), total);

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

  void test_amrwb_0() { test_amrwb(0); }
  void test_amrwb_1() { test_amrwb(1); }
  void test_amrwb_2() { test_amrwb(2); }
  void test_amrwb_3() { test_amrwb(3); }
  void test_amrwb_4() { test_amrwb(4); }
  void test_amrwb_5() { test_amrwb(5); }
  void test_amrwb_6() { test_amrwb(6); }
  void test_amrwb_7() { test_amrwb(7); }
  void test_amrwb_8() { test_amrwb(8); }
};

// Registers the fixture into the 'registry'
CPPUNIT_TEST_SUITE_REGISTRATION(test_amrwb_class);