/* * Copyright (C) 2008-2011 Teluu Inc. (http://www.teluu.com) * Copyright (C) 2003-2008 Benny Prijono * * 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 #include #include #include #include #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.fmt.det.aud.clock_rate != file_deg_port->info.fmt.det.aud.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.fmt.det.aud.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; }