/* * *  Bluetooth low-complexity, subband codec (SBC) encoder * *  Copyright (C) 2004-2008  Marcel Holtmann <marcel@holtmann.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., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA * */#ifdef HAVE_CONFIG_H#include <config.h>#endif#include <stdio.h>#include <errno.h>#include <fcntl.h>#include <unistd.h>#include <stdlib.h>#include <stdint.h>#include <string.h>#include <getopt.h>#include <byteswap.h>#include <sys/stat.h>#include "sbc.h"#if __BYTE_ORDER == __LITTLE_ENDIAN#define COMPOSE_ID(a,b,c,d)	((a) | ((b)<<8) | ((c)<<16) | ((d)<<24))#define LE_SHORT(v)		(v)#define LE_INT(v)		(v)#define BE_SHORT(v)		bswap_16(v)#define BE_INT(v)		bswap_32(v)#elif __BYTE_ORDER == __BIG_ENDIAN#define COMPOSE_ID(a,b,c,d)	((d) | ((c)<<8) | ((b)<<16) | ((a)<<24))#define LE_SHORT(v)		bswap_16(v)#define LE_INT(v)		bswap_32(v)#define BE_SHORT(v)		(v)#define BE_INT(v)		(v)#else#error "Wrong endian"#endif#define AU_MAGIC		COMPOSE_ID('.','s','n','d')#define AU_FMT_ULAW		1#define AU_FMT_LIN8		2#define AU_FMT_LIN16		3struct au_header {	uint32_t magic;		/* '.snd' */	uint32_t hdr_size;	/* size of header (min 24) */	uint32_t data_size;	/* size of data */	uint32_t encoding;	/* see to AU_FMT_XXXX */	uint32_t sample_rate;	/* sample rate */	uint32_t channels;	/* number of channels (voices) */};static ssize_t __read(int fd, void *buf, size_t count){	ssize_t len, pos = 0;	while (count > 0) {		len = read(fd, buf + pos, count);		if (len <= 0)			return len;		count -= len;		pos   += len;	}	return pos;}static ssize_t __write(int fd, const void *buf, size_t count){	ssize_t len, pos = 0;	while (count > 0) {		len = write(fd, buf + pos, count);		if (len <= 0)			return len;		count -= len;		pos   += len;	}	return pos;}static void encode(char *filename, int subbands, int joint){	struct au_header *au_hdr;	unsigned char input[2048], output[2048];	sbc_t sbc;	int fd, len, size, count, encoded;	if (strcmp(filename, "-")) {		fd = open(filename, O_RDONLY);		if (fd < 0) {			fprintf(stderr, "Can't open file %s: %s\n",						filename, strerror(errno));			return;		}	} else		fd = fileno(stdin);	len = __read(fd, input, sizeof(input));	if (len < sizeof(*au_hdr)) {		if (fd > fileno(stderr))			fprintf(stderr, "Can't read header from file %s: %s\n",						filename, strerror(errno));		else			perror("Can't read audio header");		goto done;	}	au_hdr = (struct au_header *) input;	if (au_hdr->magic != AU_MAGIC ||			BE_INT(au_hdr->hdr_size) > 128 ||			BE_INT(au_hdr->hdr_size) < 24 ||			BE_INT(au_hdr->encoding) != AU_FMT_LIN16) {		fprintf(stderr, "Data is not in Sun/NeXT audio S16_BE format\n");		goto done;	}	sbc_init(&sbc, 0L);	switch (BE_INT(au_hdr->sample_rate)) {	case 16000:		sbc.frequency = SBC_FREQ_16000;		break;	case 32000:		sbc.frequency = SBC_FREQ_32000;		break;	case 44100:		sbc.frequency = SBC_FREQ_44100;		break;	case 48000:		sbc.frequency = SBC_FREQ_48000;		break;	}	sbc.subbands = subbands == 4 ? SBC_SB_4 : SBC_SB_8;	if (BE_INT(au_hdr->channels) == 1)		sbc.mode = SBC_MODE_MONO;	else if (joint)		sbc.mode = SBC_MODE_JOINT_STEREO;	else		sbc.mode = SBC_MODE_STEREO;	sbc.endian = SBC_BE;	count = BE_INT(au_hdr->data_size);	size = len - BE_INT(au_hdr->hdr_size);	memmove(input, input + BE_INT(au_hdr->hdr_size), size);	while (1) {		if (size < sizeof(input)) {			len = __read(fd, input + size, sizeof(input) - size);			if (len == 0)				break;			if (len < 0) {				perror("Can't read audio data");				break;			}			size += len;		}		len = sbc_encode(&sbc, input, size, output, sizeof(output),					&encoded);		if (len < size)			memmove(input, input + len, size - len);		size -= len;		len = __write(fileno(stdout), output, encoded);		if (len == 0)			break;		if (len < 0 || len != encoded) {			perror("Can't write SBC output");			break;		}	}	sbc_finish(&sbc);done:	if (fd > fileno(stderr))		close(fd);}static void usage(void){	printf("SBC encoder utility ver %s\n", VERSION);	printf("Copyright (c) 2004-2008  Marcel Holtmann\n\n");	printf("Usage:\n"		"\tsbcenc [options] file(s)\n"		"\n");	printf("Options:\n"		"\t-h, --help           Display help\n"		"\t-v, --verbose        Verbose mode\n"		"\t-s, --subbands       Number of subbands to use (4 or 8)\n"		"\t-j, --joint          Joint stereo\n"		"\n");}static struct option main_options[] = {	{ "help",	0, 0, 'h' },	{ "verbose",	0, 0, 'v' },	{ "subbands",	1, 0, 's' },	{ "joint",	0, 0, 'j' },	{ 0, 0, 0, 0 }};int main(int argc, char *argv[]){	int i, opt, verbose = 0, subbands = 8, joint = 0;	while ((opt = getopt_long(argc, argv, "+hvs:j", main_options, NULL)) != -1) {		switch(opt) {		case 'h':			usage();			exit(0);		case 'v':			verbose = 1;			break;		case 's':			subbands = atoi(strdup(optarg));			if (subbands != 8 && subbands != 4) {				fprintf(stderr, "Invalid subbands %d!\n",						subbands);				exit(1);			}			break;		case 'j':			joint = 1;			break;		default:			exit(1);		}	}	argc -= optind;	argv += optind;	optind = 0;	if (argc < 1) {		usage();		exit(1);	}	for (i = 0; i < argc; i++)		encode(argv[i], subbands, joint);	return 0;}