/* * *  Bluetooth low-complexity, subband codec (SBC) library * *  Copyright (C) 2007-2008  Marcel Holtmann <marcel@holtmann.org> *  Copyright (C) 2007-2008  Frederic Dalleau <fdalleau@free.fr> * * *  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., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA * */#ifdef HAVE_CONFIG_H#include <config.h>#endif#include <stdio.h>#include <stdlib.h>#include <sndfile.h>#include <math.h>#include <string.h>#define MAXCHANNELS 2#define DEFACCURACY 7static double sampletobits(short sample16, int verbose){	double bits = 0;	unsigned short bit;	int i;	if (verbose)		printf("===> sampletobits(%hd, %04hX)\n", sample16, sample16);	/* Bit 0 is MSB */	if (sample16 < 0)		bits = -1;	if (verbose)		printf("%d", (sample16 < 0) ? 1 : 0);	/* Bit 15 is LSB */	for (i = 1; i < 16; i++) {		bit = (unsigned short) sample16;		bit >>= 15 - i;		bit %= 2;		if (verbose)			printf("%d", bit);		if (bit)			bits += (1.0 / pow(2.0, i));	}	if (verbose)		printf("\n");	return bits;}static int calculate_rms_level(SNDFILE * sndref, SF_INFO * infosref,				SNDFILE * sndtst, SF_INFO * infostst,						int accuracy, char *csvname){	short refsample[MAXCHANNELS], tstsample[MAXCHANNELS];	double refbits, tstbits;	double rms_accu[MAXCHANNELS];	double rms_level[MAXCHANNELS];	double rms_limit = 1.0 / (pow(2.0, accuracy - 1) * pow(12.0, 0.5));	FILE *csv = NULL;	int i, j, r1, r2, verdict;	if (csvname)		csv = fopen(csvname, "wt");	if (csv) {		fprintf(csv, "num;");		for (j = 0; j < infostst->channels; j++)			fprintf(csv, "ref channel %d;tst channel %d;", j, j);		fprintf(csv, "\r\n");	}	sf_seek(sndref, 0, SEEK_SET);	sf_seek(sndtst, 0, SEEK_SET);	memset(rms_accu, 0, sizeof(rms_accu));	memset(rms_level, 0, sizeof(rms_level));	for (i = 0; i < infostst->frames; i++) {		if (csv)			fprintf(csv, "%d;", i);		r1 = sf_read_short(sndref, refsample, infostst->channels);		if (r1 != infostst->channels) {			printf("Failed to read reference data: %s "					"(r1=%d, channels=%d)",					sf_strerror(sndref), r1,					infostst->channels);			if (csv)				fclose(csv);			return -1;		}		r2 = sf_read_short(sndtst, tstsample, infostst->channels);		if (r2 != infostst->channels) {			printf("Failed to read test data: %s "					"(r2=%d, channels=%d)\n",					sf_strerror(sndtst), r2,					infostst->channels);			if (csv)				fclose(csv);			return -1;		}		for (j = 0; j < infostst->channels; j++) {			if (csv)				fprintf(csv, "%d;%d;", refsample[j],						tstsample[j]);			refbits = sampletobits(refsample[j], 0);			tstbits = sampletobits(tstsample[j], 0);			rms_accu[j] += pow(tstbits - refbits, 2.0);		}		if (csv)			fprintf(csv, "\r\n");	}	printf("Limit: %f\n", rms_limit);	for (j = 0; j < infostst->channels; j++) {		printf("Channel %d\n", j);		printf("Accumulated %f\n", rms_accu[j]);		rms_accu[j] /= (double) infostst->frames;		printf("Accumulated / %f = %f\n", (double) infostst->frames,				rms_accu[j]);		rms_level[j] = sqrt(rms_accu[j]);		printf("Level = %f (%f x %f = %f)\n",				rms_level[j], rms_level[j], rms_level[j],						rms_level[j] * rms_level[j]);	}	verdict = 1;	for (j = 0; j < infostst->channels; j++) {		printf("Channel %d: %f\n", j, rms_level[j]);		if (rms_level[j] > rms_limit)			verdict = 0;	}	printf("%s return %d\n", __FUNCTION__, verdict);	return verdict;}static int check_absolute_diff(SNDFILE * sndref, SF_INFO * infosref,				SNDFILE * sndtst, SF_INFO * infostst,				int accuracy){	short refsample[MAXCHANNELS], tstsample[MAXCHANNELS];	short refmax[MAXCHANNELS], tstmax[MAXCHANNELS];	double refbits, tstbits;	double rms_absolute = 1.0 / (pow(2, accuracy - 2));	double calc_max[MAXCHANNELS];	int calc_count = 0;	short r1, r2;	double cur_diff;	int i, j, verdict;	memset(&refmax, 0, sizeof(refmax));	memset(&tstmax, 0, sizeof(tstmax));	memset(&calc_max, 0, sizeof(calc_max));	memset(&refsample, 0, sizeof(refsample));	memset(&tstsample, 0, sizeof(tstsample));	sf_seek(sndref, 0, SEEK_SET);	sf_seek(sndtst, 0, SEEK_SET);	verdict = 1;	printf("Absolute max: %f\n", rms_absolute);	for (i = 0; i < infostst->frames; i++) {		r1 = sf_read_short(sndref, refsample, infostst->channels);		if (r1 != infostst->channels) {			printf("Failed to read reference data: %s "					"(r1=%d, channels=%d)",					sf_strerror(sndref), r1,					infostst->channels);			return -1;		}		r2 = sf_read_short(sndtst, tstsample, infostst->channels);		if (r2 != infostst->channels) {			printf("Failed to read test data: %s "					"(r2=%d, channels=%d)\n",					sf_strerror(sndtst), r2,					infostst->channels);			return -1;		}		for (j = 0; j < infostst->channels; j++) {			refbits = sampletobits(refsample[j], 0);			tstbits = sampletobits(tstsample[j], 0);			cur_diff = fabs(tstbits - refbits);			if (cur_diff > rms_absolute) {				calc_count++;				/* printf("Channel %d exceeded : fabs(%f - %f) = %f > %f\n", j, tstbits, refbits, cur_diff, rms_absolute); */				verdict = 0;			}			if (cur_diff > calc_max[j]) {				calc_max[j] = cur_diff;				refmax[j] = refsample[j];				tstmax[j] = tstsample[j];			}		}	}	for (j = 0; j < infostst->channels; j++) {		printf("Calculated max: %f (%hd-%hd=%hd)\n",			calc_max[j], tstmax[j], refmax[j],			tstmax[j] - refmax[j]);	}	printf("%s return %d\n", __FUNCTION__, verdict);	return verdict;}static void usage(){	printf("SBC conformance test ver %s\n", VERSION);	printf("Copyright (c) 2007-2008 Marcel Holtmann\n");	printf("Copyright (c) 2007-2008 Frederic Dalleau\n\n");	printf("Usage:\n"		"\tsbctester reference.wav checkfile.wav\n"		"\tsbctester integer\n"		"\n");	printf("To test the encoder:\n");	printf("\tUse a reference codec to encode original.wav to reference.sbc\n");	printf("\tUse sbcenc to encode original.wav to checkfile.sbc\n");	printf("\tDecode both file using the reference decoder\n");	printf("\tRun sbctester with these two wav files to get the result\n\n");	printf("\tA file called out.csv is generated to use the data in a\n");	printf("\tspreadsheet application or database.\n\n");}int main(int argc, char *argv[]){	SNDFILE *sndref = NULL;	SNDFILE *sndtst = NULL;	SF_INFO infosref;	SF_INFO infostst;	char *ref;	char *tst;	int pass_rms, pass_absolute, pass, accuracy;	if (argc == 2) {		double db;		printf("Test sampletobits\n");		db = sampletobits((short) atoi(argv[1]), 1);		printf("db = %f\n", db);		exit(0);	}	if (argc < 3) {		usage();		exit(1);	}	ref = argv[1];	tst = argv[2];	printf("opening reference %s\n", ref);	sndref = sf_open(ref, SFM_READ, &infosref);	if (!sndref) {		printf("Failed to open reference file\n");		exit(1);	}	printf("opening testfile %s\n", tst);	sndtst = sf_open(tst, SFM_READ, &infostst);	if (!sndtst) {		printf("Failed to open test file\n");		sf_close(sndref);		exit(1);	}	printf("reference:\n\t%d frames,\n\t%d hz,\n\t%d channels\n",		(int) infosref.frames, (int) infosref.samplerate,		(int) infosref.channels);	printf("testfile:\n\t%d frames,\n\t%d hz,\n\t%d channels\n",		(int) infostst.frames, (int) infostst.samplerate,		(int) infostst.channels);	/* check number of channels */	if (infosref.channels > 2 || infostst.channels > 2) {		printf("Too many channels\n");		goto error;	}	/* compare number of samples */	if (infosref.samplerate != infostst.samplerate ||				infosref.channels != infostst.channels) {		printf("Cannot compare files with different charasteristics\n");		goto error;	}	accuracy = DEFACCURACY;	printf("Accuracy: %d\n", accuracy);	/* Condition 1 rms level */	pass_rms = calculate_rms_level(sndref, &infosref, sndtst, &infostst,					accuracy, "out.csv");	if (pass_rms < 0)		goto error;	/* Condition 2 absolute difference */	pass_absolute = check_absolute_diff(sndref, &infosref, sndtst,						&infostst, accuracy);	if (pass_absolute < 0)		goto error;	/* Verdict */	pass = pass_rms && pass_absolute;	printf("Verdict: %s\n", pass ? "pass" : "fail");	return 0;error:	sf_close(sndref);	sf_close(sndtst);	exit(1);}