diff options
Diffstat (limited to 'sbc/sbcenc.c')
-rw-r--r-- | sbc/sbcenc.c | 280 |
1 files changed, 280 insertions, 0 deletions
diff --git a/sbc/sbcenc.c b/sbc/sbcenc.c new file mode 100644 index 000000000..e17f8fdbc --- /dev/null +++ b/sbc/sbcenc.c @@ -0,0 +1,280 @@ +/* + * + * 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 3 + +struct 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; +} |