/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set et sw=2 ts=2: */
/***************************************************************************
 *            test_jpeg.cc
 *
 *  Wed Dec 18 08:59:12 CET 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

class test_jpeg_class : public CppUnit::TestFixture
{
  CPPUNIT_TEST_SUITE(test_jpeg_class);
	CPPUNIT_TEST(test_jpeg);
	CPPUNIT_TEST_SUITE_END();

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

	void test_jpeg() {

    std::vector<std::string> packets;
    unsigned int csrc = 42;

    { // 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_JPEG, csrc,
                                  OPTION_END);
      CPPUNIT_ASSERT_EQUAL(0, x);
      //char num = 0;

      char packet[16*1024];


      for(int n = 1; n < 5; n++) {
        char filename[32];
        sprintf(filename, "test%d.jpg", n);

        //printf("----------- Image %d -----------\n", n);
        
        FILE *fp = fopen(filename, "r");
        CPPUNIT_ASSERT(fp);
        CPPUNIT_ASSERT(fseek(fp, 0, SEEK_END) != -1);
        int imagesize = ftell(fp);
        CPPUNIT_ASSERT(fseek(fp, 0, SEEK_SET) != -1);
        char *image = (char*)malloc(imagesize);
        CPPUNIT_ASSERT(fread(image, 1, imagesize, fp) == (size_t)imagesize);
        fclose(fp);

        int ret = 0;
        ret = lrtp_enqueue_frame(lrtp, csrc, image, imagesize, n, LRTP_COPY);
        CPPUNIT_ASSERT_EQUAL(0, ret);

        while( (ret = lrtp_pack(lrtp, packet, sizeof(packet))) > 0) {
          std::string p;
          p.append(packet, ret);
          packets.push_back(p);
          //dump("pkg", packet, ret);
        }
        CPPUNIT_ASSERT_EQUAL(ret, 0);

        free(image);
      }
      
      status = lrtp_destroy_profile(lrtp, csrc);
      CPPUNIT_ASSERT_EQUAL(status, LRTP_OK);
      
      status = lrtp_close(lrtp);
      CPPUNIT_ASSERT_EQUAL(status, LRTP_OK);
    }

    //printf("\nPackets: %d\n", packets.size());

    { // Decode
      //FILE *fp = fopen("output.jpg", "w");

      // Write SIO and JFIF from original image:
      //      fwrite(image, 2 + 16, 1, fp);

      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_JPEG, csrc,
                                  OPTION_END);
      CPPUNIT_ASSERT_EQUAL(0, x);

      char frame[16*1024];
      size_t framesize = sizeof(frame);
      
      //int cnt = 0;
      
      //      int num = 2 + 16; // Skip SOI and JFIF in comparison

      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);
          
          //fwrite(frame, ret, 1, fp);
          /*
          int err = 0;
          for(int i = 0; i < ret; i++) {
            err += abs(frame[i] - image[num++]);
          }
          CPPUNIT_ASSERT_EQUAL(0, err);
          */
          //dump("pkg", frame, ret);
        }
        
        i++;
      }
      
      //      CPPUNIT_ASSERT_EQUAL((int)imagesize, num);

      status = lrtp_destroy_profile(lrtp, csrc);
      CPPUNIT_ASSERT_EQUAL(status, LRTP_OK);

      status = lrtp_close(lrtp);
      CPPUNIT_ASSERT_EQUAL(status, LRTP_OK);

      //fclose(fp);
    }

    //free(image);
	}
};

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