Home | History | Annotate | Download | only in sbc
      1 /*
      2  *
      3  *  Bluetooth low-complexity, subband codec (SBC) decoder
      4  *
      5  *  Copyright (C) 2008-2010  Nokia Corporation
      6  *  Copyright (C) 2004-2010  Marcel Holtmann <marcel (at) holtmann.org>
      7  *
      8  *
      9  *  This program is free software; you can redistribute it and/or modify
     10  *  it under the terms of the GNU General Public License as published by
     11  *  the Free Software Foundation; either version 2 of the License, or
     12  *  (at your option) any later version.
     13  *
     14  *  This program is distributed in the hope that it will be useful,
     15  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
     16  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     17  *  GNU General Public License for more details.
     18  *
     19  *  You should have received a copy of the GNU General Public License
     20  *  along with this program; if not, write to the Free Software
     21  *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
     22  *
     23  */
     24 
     25 #ifdef HAVE_CONFIG_H
     26 #include <config.h>
     27 #endif
     28 
     29 #include <stdio.h>
     30 #include <errno.h>
     31 #include <fcntl.h>
     32 #include <unistd.h>
     33 #include <stdlib.h>
     34 #include <string.h>
     35 #include <getopt.h>
     36 #include <sys/stat.h>
     37 #include <sys/ioctl.h>
     38 #include <sys/soundcard.h>
     39 
     40 #include "sbc.h"
     41 #include "formats.h"
     42 
     43 #define BUF_SIZE 8192
     44 
     45 static int verbose = 0;
     46 
     47 static void decode(char *filename, char *output, int tofile)
     48 {
     49 	unsigned char buf[BUF_SIZE], *stream;
     50 	struct stat st;
     51 	sbc_t sbc;
     52 	int fd, ad, pos, streamlen, framelen, count;
     53 	size_t len;
     54 	int format = AFMT_S16_BE, frequency, channels;
     55 	ssize_t written;
     56 
     57 	if (stat(filename, &st) < 0) {
     58 		fprintf(stderr, "Can't get size of file %s: %s\n",
     59 						filename, strerror(errno));
     60 		return;
     61 	}
     62 
     63 	stream = malloc(st.st_size);
     64 
     65 	if (!stream) {
     66 		fprintf(stderr, "Can't allocate memory for %s: %s\n",
     67 						filename, strerror(errno));
     68 		return;
     69 	}
     70 
     71 	fd = open(filename, O_RDONLY);
     72 	if (fd < 0) {
     73 		fprintf(stderr, "Can't open file %s: %s\n",
     74 						filename, strerror(errno));
     75 		goto free;
     76 	}
     77 
     78 	if (read(fd, stream, st.st_size) != st.st_size) {
     79 		fprintf(stderr, "Can't read content of %s: %s\n",
     80 						filename, strerror(errno));
     81 		close(fd);
     82 		goto free;
     83 	}
     84 
     85 	close(fd);
     86 
     87 	pos = 0;
     88 	streamlen = st.st_size;
     89 
     90 	if (tofile)
     91 		ad = open(output, O_WRONLY | O_CREAT | O_TRUNC, 0644);
     92 	else
     93 		ad = open(output, O_WRONLY, 0);
     94 
     95 	if (ad < 0) {
     96 		fprintf(stderr, "Can't open output %s: %s\n",
     97 						output, strerror(errno));
     98 		goto free;
     99 	}
    100 
    101 	sbc_init(&sbc, 0L);
    102 	sbc.endian = SBC_BE;
    103 
    104 	framelen = sbc_decode(&sbc, stream, streamlen, buf, sizeof(buf), &len);
    105 	channels = sbc.mode == SBC_MODE_MONO ? 1 : 2;
    106 	switch (sbc.frequency) {
    107 	case SBC_FREQ_16000:
    108 		frequency = 16000;
    109 		break;
    110 
    111 	case SBC_FREQ_32000:
    112 		frequency = 32000;
    113 		break;
    114 
    115 	case SBC_FREQ_44100:
    116 		frequency = 44100;
    117 		break;
    118 
    119 	case SBC_FREQ_48000:
    120 		frequency = 48000;
    121 		break;
    122 	default:
    123 		frequency = 0;
    124 	}
    125 
    126 	if (verbose) {
    127 		fprintf(stderr,"decoding %s with rate %d, %d subbands, "
    128 			"%d bits, allocation method %s and mode %s\n",
    129 			filename, frequency, sbc.subbands * 4 + 4, sbc.bitpool,
    130 			sbc.allocation == SBC_AM_SNR ? "SNR" : "LOUDNESS",
    131 			sbc.mode == SBC_MODE_MONO ? "MONO" :
    132 					sbc.mode == SBC_MODE_STEREO ?
    133 						"STEREO" : "JOINTSTEREO");
    134 	}
    135 
    136 	if (tofile) {
    137 		struct au_header au_hdr;
    138 
    139 		au_hdr.magic       = AU_MAGIC;
    140 		au_hdr.hdr_size    = BE_INT(24);
    141 		au_hdr.data_size   = BE_INT(0);
    142 		au_hdr.encoding    = BE_INT(AU_FMT_LIN16);
    143 		au_hdr.sample_rate = BE_INT(frequency);
    144 		au_hdr.channels    = BE_INT(channels);
    145 
    146 		written = write(ad, &au_hdr, sizeof(au_hdr));
    147 		if (written < (ssize_t) sizeof(au_hdr)) {
    148 			fprintf(stderr, "Failed to write header\n");
    149 			goto close;
    150 		}
    151 	} else {
    152 		if (ioctl(ad, SNDCTL_DSP_SETFMT, &format) < 0) {
    153 			fprintf(stderr, "Can't set audio format on %s: %s\n",
    154 						output, strerror(errno));
    155 			goto close;
    156 		}
    157 
    158 		if (ioctl(ad, SNDCTL_DSP_CHANNELS, &channels) < 0) {
    159 			fprintf(stderr, "Can't set number of channels on %s: %s\n",
    160 						output, strerror(errno));
    161 			goto close;
    162 		}
    163 
    164 		if (ioctl(ad, SNDCTL_DSP_SPEED, &frequency) < 0) {
    165 			fprintf(stderr, "Can't set audio rate on %s: %s\n",
    166 						output, strerror(errno));
    167 			goto close;
    168 		}
    169 	}
    170 
    171 	count = len;
    172 
    173 	while (framelen > 0) {
    174 		/* we have completed an sbc_decode at this point sbc.len is the
    175 		 * length of the frame we just decoded count is the number of
    176 		 * decoded bytes yet to be written */
    177 
    178 		if (count + len >= BUF_SIZE) {
    179 			/* buffer is too full to stuff decoded audio in so it
    180 			 * must be written to the device */
    181 			written = write(ad, buf, count);
    182 			if (written > 0)
    183 				count -= written;
    184 		}
    185 
    186 		/* sanity check */
    187 		if (count + len >= BUF_SIZE) {
    188 			fprintf(stderr,
    189 				"buffer size of %d is too small for decoded"
    190 				" data (%lu)\n", BUF_SIZE, (unsigned long) (len + count));
    191 			exit(1);
    192 		}
    193 
    194 		/* push the pointer in the file forward to the next bit to be
    195 		 * decoded tell the decoder to decode up to the remaining
    196 		 * length of the file (!) */
    197 		pos += framelen;
    198 		framelen = sbc_decode(&sbc, stream + pos, streamlen - pos,
    199 					buf + count, sizeof(buf) - count, &len);
    200 
    201 		/* increase the count */
    202 		count += len;
    203 	}
    204 
    205 	if (count > 0) {
    206 		written = write(ad, buf, count);
    207 		if (written > 0)
    208 			count -= written;
    209 	}
    210 
    211 close:
    212 	sbc_finish(&sbc);
    213 
    214 	close(ad);
    215 
    216 free:
    217 	free(stream);
    218 }
    219 
    220 static void usage(void)
    221 {
    222 	printf("SBC decoder utility ver %s\n", VERSION);
    223 	printf("Copyright (c) 2004-2010  Marcel Holtmann\n\n");
    224 
    225 	printf("Usage:\n"
    226 		"\tsbcdec [options] file(s)\n"
    227 		"\n");
    228 
    229 	printf("Options:\n"
    230 		"\t-h, --help           Display help\n"
    231 		"\t-v, --verbose        Verbose mode\n"
    232 		"\t-d, --device <dsp>   Sound device\n"
    233 		"\t-f, --file <file>    Decode to a file\n"
    234 		"\n");
    235 }
    236 
    237 static struct option main_options[] = {
    238 	{ "help",	0, 0, 'h' },
    239 	{ "device",	1, 0, 'd' },
    240 	{ "verbose",	0, 0, 'v' },
    241 	{ "file",	1, 0, 'f' },
    242 	{ 0, 0, 0, 0 }
    243 };
    244 
    245 int main(int argc, char *argv[])
    246 {
    247 	char *output = NULL;
    248 	int i, opt, tofile = 0;
    249 
    250 	while ((opt = getopt_long(argc, argv, "+hvd:f:",
    251 						main_options, NULL)) != -1) {
    252 		switch(opt) {
    253 		case 'h':
    254 			usage();
    255 			exit(0);
    256 
    257 		case 'v':
    258 			verbose = 1;
    259 			break;
    260 
    261 		case 'd':
    262 			free(output);
    263 			output = strdup(optarg);
    264 			tofile = 0;
    265 			break;
    266 
    267 		case 'f' :
    268 			free(output);
    269 			output = strdup(optarg);
    270 			tofile = 1;
    271 			break;
    272 
    273 		default:
    274 			exit(1);
    275 		}
    276 	}
    277 
    278 	argc -= optind;
    279 	argv += optind;
    280 	optind = 0;
    281 
    282 	if (argc < 1) {
    283 		usage();
    284 		exit(1);
    285 	}
    286 
    287 	for (i = 0; i < argc; i++)
    288 		decode(argv[i], output ? output : "/dev/dsp", tofile);
    289 
    290 	free(output);
    291 
    292 	return 0;
    293 }
    294