sbc/sbctester.c - platform/external/bluetooth/bluez - Git at Google

 /*
  *
  *  Bluetooth low-complexity, subband codec (SBC) library
  *
  *  Copyright (C) 2008-2010  Nokia Corporation
  *  Copyright (C) 2007-2010  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 7

 static 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-2010  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);
 }
	/*
	*
	* Bluetooth low-complexity, subband codec (SBC) library
	*
	* Copyright (C) 2008-2010 Nokia Corporation
	* Copyright (C) 2007-2010 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 7

	static 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-2010 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);
	}