From 0f900a2b4d7707dddeddadd62ac5a109aed93e7e Mon Sep 17 00:00:00 2001
From: Bent Bisballe Nyeng <deva@aasimon.org>
Date: Mon, 6 Jan 2014 08:07:56 +0100
Subject: Fix round robin frame selector. Add l16 and jpeg profiles. Add
 timestamp to lrtp_enqueue_frame.

---
 test/test_jpeg.cc | 165 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 165 insertions(+)
 create mode 100644 test/test_jpeg.cc

(limited to 'test/test_jpeg.cc')

diff --git a/test/test_jpeg.cc b/test/test_jpeg.cc
new file mode 100644
index 0000000..894ae3a
--- /dev/null
+++ b/test/test_jpeg.cc
@@ -0,0 +1,165 @@
+/* -*- 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
+      struct lrtp_t *lrtp = lrtp_init(KEY, SSRC);
+      
+      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);
+        fseek(fp, 0, SEEK_END);
+        int imagesize = ftell(fp);
+        fseek(fp, 0, SEEK_SET);
+        char *image = (char*)malloc(imagesize);
+        fread(image, imagesize, 1, fp);
+        fclose(fp);
+
+        int ret = 0;
+        ret = lrtp_enqueue_frame(lrtp, csrc, image, imagesize, n);
+        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);
+        }
+
+        free(image);
+      }
+      
+      lrtp_destroy_profile(lrtp, csrc);
+      
+      lrtp_close(lrtp);
+    }
+
+    //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);
+
+      struct lrtp_t *lrtp = lrtp_init(KEY, SSRC);
+      
+      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);
+
+      lrtp_destroy_profile(lrtp, csrc);
+      lrtp_close(lrtp);
+
+      //fclose(fp);
+    }
+
+    //free(image);
+	}
+};
+
+// Registers the fixture into the 'registry'
+CPPUNIT_TEST_SUITE_REGISTRATION(test_jpeg_class);
+
+
-- 
cgit v1.2.3