summaryrefslogtreecommitdiff
path: root/tests/pjsua/tools/cmp_wav.c
diff options
context:
space:
mode:
Diffstat (limited to 'tests/pjsua/tools/cmp_wav.c')
-rw-r--r--tests/pjsua/tools/cmp_wav.c262
1 files changed, 262 insertions, 0 deletions
diff --git a/tests/pjsua/tools/cmp_wav.c b/tests/pjsua/tools/cmp_wav.c
new file mode 100644
index 0000000..e3c864c
--- /dev/null
+++ b/tests/pjsua/tools/cmp_wav.c
@@ -0,0 +1,262 @@
+/* $Id: cmp_wav.c 3553 2011-05-05 06:14:19Z nanang $ */
+/*
+ * Copyright (C) 2008-2011 Teluu Inc. (http://www.teluu.com)
+ * Copyright (C) 2003-2008 Benny Prijono <benny@prijono.org>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ */
+#include <pjmedia.h>
+#include <pjlib-util.h>
+#include <pjlib.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+#define app_perror(a,b,c) printf("%s: %s (%d)", a, b, c)
+
+
+/* For logging purpose. */
+#define THIS_FILE "cmp_wav.c"
+#define BYTES_PER_FRAME 512
+
+static const char *desc =
+" FILE \n"
+" \n"
+" cmp_wav.c \n"
+" \n"
+" PURPOSE \n"
+" \n"
+" Compare two WAV files. \n"
+" \n"
+" USAGE \n"
+" \n"
+" cmp_wav ORIGINAL_WAV DEGRADED_WAV [TIME] [DETAIL] \n"
+" \n"
+" ORIGINAL_WAV The original WAV file as reference. \n"
+" DEGRADED_WAV The degraded WAV file. \n"
+" TIME Compare only some part of the files \n"
+" (in ms, since the beginning). \n"
+" Specify 0 (default) to compare the whole time. \n"
+" DETAIL Show detail result, 1 or 0 (default=0, means no)\n"
+" \n"
+" Both files must have same clock rate and must contain \n"
+" uncompressed (i.e. 16bit) PCM. \n";
+
+
+/* Sum of multiplication of corresponding samples in buf1 & buf2 */
+double sum_mult_sig(pj_int16_t *buf1, pj_int16_t *buf2, unsigned nsamples)
+{
+ double mag = 0;
+
+ while (nsamples--)
+ mag += (double)*buf1++ * (double)*buf2++;
+
+ return mag;
+}
+
+
+/*
+ * main()
+ */
+int main(int argc, char *argv[])
+{
+ pj_caching_pool cp;
+ pjmedia_endpt *med_endpt;
+ pj_pool_t *pool;
+ pjmedia_port *file_ori_port;
+ pjmedia_port *file_deg_port;
+ pj_status_t status;
+ unsigned first_nsamples = 0;
+ unsigned samples_compared = 0;
+
+ char buf1[BYTES_PER_FRAME];
+ char buf2[BYTES_PER_FRAME];
+
+ double ref_mag = 0;
+ double deg_mag = 0;
+ double mix_mag = 0;
+
+ int detail = 0;
+ int res_deg, res_mix, res_overall;
+
+ if (argc < 3) {
+ puts("Error: original & degraded filename required");
+ puts(desc);
+ return 1;
+ }
+
+ /* Set log level. */
+ pj_log_set_level(3);
+
+ /* Must init PJLIB first: */
+ status = pj_init();
+ PJ_ASSERT_RETURN(status == PJ_SUCCESS, 1);
+
+ /* Must create a pool factory before we can allocate any memory. */
+ pj_caching_pool_init(&cp, &pj_pool_factory_default_policy, 0);
+
+ /*
+ * Initialize media endpoint.
+ * This will implicitly initialize PJMEDIA too.
+ */
+ status = pjmedia_endpt_create(&cp.factory, NULL, 1, &med_endpt);
+ PJ_ASSERT_RETURN(status == PJ_SUCCESS, 1);
+
+ /* Create memory pool for our file player */
+ pool = pj_pool_create( &cp.factory, /* pool factory */
+ "wav", /* pool name. */
+ 4000, /* init size */
+ 4000, /* increment size */
+ NULL /* callback on error */
+ );
+
+ /* Create file media port from the original WAV file */
+ status = pjmedia_wav_player_port_create( pool, /* memory pool */
+ argv[1], /* file to play */
+ 40, /* ptime. */
+ PJMEDIA_FILE_NO_LOOP, /* flags */
+ 0, /* default buffer */
+ &file_ori_port/* returned port */
+ );
+ if (status != PJ_SUCCESS) {
+ app_perror(THIS_FILE, "Unable to use WAV file", status);
+ return 1;
+ }
+
+ /* Create file media port from the degraded WAV file */
+ status = pjmedia_wav_player_port_create( pool, /* memory pool */
+ argv[2], /* file to play */
+ 40, /* ptime. */
+ PJMEDIA_FILE_NO_LOOP, /* flags */
+ 0, /* default buffer */
+ &file_deg_port/* returned port */
+ );
+ if (status != PJ_SUCCESS) {
+ app_perror(THIS_FILE, "Unable to use WAV file", status);
+ return 1;
+ }
+
+ if (file_ori_port->info.clock_rate != file_deg_port->info.clock_rate) {
+ app_perror(THIS_FILE, "Clock rates must be same.", PJ_EINVAL);
+ return 1;
+ }
+
+ if (argc > 3)
+ first_nsamples = atoi(argv[3]) * file_ori_port->info.clock_rate / 1000;
+
+ if (argc > 4)
+ detail = atoi(argv[4]);
+
+ while (1) {
+ pjmedia_frame f1, f2;
+
+ f1.buf = buf1;
+ f1.size = BYTES_PER_FRAME;
+ f2.buf = buf2;
+ f2.size = BYTES_PER_FRAME;
+
+ status = pjmedia_port_get_frame(file_ori_port, &f1);
+ if (status == PJ_EEOF) {
+ break;
+ } else if (status != PJ_SUCCESS) {
+ app_perror(THIS_FILE, "Error occured while reading file", status);
+ break;
+ }
+ status = pjmedia_port_get_frame(file_deg_port, &f2);
+ if (status == PJ_EEOF) {
+ break;
+ } else if (status != PJ_SUCCESS) {
+ app_perror(THIS_FILE, "Error occured while reading file", status);
+ break;
+ }
+
+ /* Calculate magnitudes */
+ ref_mag += sum_mult_sig(f1.buf, f1.buf, BYTES_PER_FRAME >> 1);
+ deg_mag += sum_mult_sig(f2.buf, f2.buf, BYTES_PER_FRAME >> 1);
+ mix_mag += sum_mult_sig(f1.buf, f2.buf, BYTES_PER_FRAME >> 1);
+
+ samples_compared += BYTES_PER_FRAME >> 1;
+ if (first_nsamples && samples_compared >= first_nsamples)
+ break;
+ }
+
+ /* Degraded magnitude compared to reference magnitude
+ */
+ res_deg = (int) (deg_mag / ref_mag * 100.0);
+ if (res_deg < 0)
+ res_deg = -1;
+ else if (res_deg >= 81)
+ res_deg = 9;
+ else
+ res_deg = pj_isqrt(res_deg);
+
+ /* Mixed magnitude (don't know what this is actually :D) compared to
+ * reference magnitude
+ */
+ res_mix = (int) (mix_mag / ref_mag * 100.0);
+ if (res_mix < 0)
+ res_mix = -1;
+ else if (res_mix >= 81)
+ res_mix = 9;
+ else
+ res_mix = pj_isqrt(res_mix);
+
+ /* Overall score.
+ * If mixed score is -1, then overall score should be -1 as well.
+ * Apply no weighting (1:1) for now.
+ */
+ if (res_mix == -1)
+ res_overall = -1;
+ else
+ res_overall = (res_mix*1 + res_deg*1) / 2;
+
+ if (detail) {
+ printf("Reference = %.0f\n", ref_mag);
+ printf("Degraded = %.0f\n", deg_mag);
+ printf("Mixed = %.0f\n", mix_mag);
+
+ printf("\n");
+
+ printf("Score 1 = %d\n", res_deg);
+ printf("Score 2 = %d\n", res_mix);
+
+ printf("\n");
+ }
+
+ printf("Overall = %d\n", res_overall);
+
+ /* Destroy file port */
+ status = pjmedia_port_destroy( file_ori_port );
+ PJ_ASSERT_RETURN(status == PJ_SUCCESS, 1);
+
+ status = pjmedia_port_destroy( file_deg_port );
+ PJ_ASSERT_RETURN(status == PJ_SUCCESS, 1);
+
+ /* Release application pool */
+ pj_pool_release( pool );
+
+ /* Destroy media endpoint. */
+ pjmedia_endpt_destroy( med_endpt );
+
+ /* Destroy pool factory */
+ pj_caching_pool_destroy( &cp );
+
+ /* Shutdown PJLIB */
+ pj_shutdown();
+
+
+ /* Done. */
+ return 0;
+}
+
generated by cgit v1.2.3 (git 2.39.1) at 2024-06-14 03:12:55 +0000