X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~mdw/git/disorder/blobdiff_plain/f8f8039fc44104b43b94467d17a8d71fc40f219b..34fb8c61ac9d00d64a82facbfc16113cc5b6cfd1:/server/decode.c diff --git a/server/decode.c b/server/decode.c index 434ad00..8a09013 100644 --- a/server/decode.c +++ b/server/decode.c @@ -1,43 +1,32 @@ /* * This file is part of DisOrder - * Copyright (C) 2007 Richard Kettlewell + * Copyright (C) 2007-2010 Richard Kettlewell * - * This program is free software; you can redistribute it and/or modify + * 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 + * the Free Software Foundation, either version 3 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. - * + * 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 - * USA + * along with this program. If not, see . */ /** @file server/decode.c * @brief General-purpose decoder for use by speaker process */ +#include "decode.h" -#include -#include "types.h" - -#include -#include -#include -#include -#include -#include -#include -#include #include -#include +#include + +#include + +#include "wav.h" -#include "log.h" -#include "syscalls.h" -#include "defs.h" /** @brief Encoding lookup table type */ struct decoder { @@ -47,207 +36,55 @@ struct decoder { void (*decode)(void); }; -/** @brief Input file */ -static int inputfd; - -/** @brief Output file */ -static FILE *outputfp; - -/** @brief Filename */ -static const char *path; - -/** @brief Input buffer */ -static unsigned char buffer[1048576]; - -/** @brief Open the input file */ -static void open_input(void) { - if((inputfd = open(path, O_RDONLY)) < 0) - fatal(errno, "opening %s", path); -} - -/** @brief Fill the buffer - * @return Number of bytes read - */ -static size_t fill(void) { - int n = read(inputfd, buffer, sizeof buffer); - - if(n < 0) - fatal(errno, "reading from %s", path); - return n; -} - -/** @brief Write a 16-bit word in bigendian format */ -static inline void output_16(uint16_t n) { - if(putc(n >> 8, outputfp) < 0 - || putc(n & 0xFF, outputfp) < 0) - fatal(errno, "decoding %s: output error", path); -} - -/** @brief Write the header - * If called more than once, either does nothing (if you kept the same - * output encoding) or fails (if you changed it). +FILE *outputfp; +const char *path; +char input_buffer[INPUT_BUFFER_SIZE]; +int input_count; + +/** @brief Write a block header + * @param rate Sample rate in Hz + * @param channels Channel count (currently only 1 or 2 supported) + * @param bits Bits per sample (must be a multiple of 8, no more than 64) + * @param nbytes Total number of data bytes + * @param endian @ref ENDIAN_BIG or @ref ENDIAN_LITTLE + * + * Checks that the sample format is a supported one (so other calls do not have + * to) and calls disorder_fatal() on error. */ -static void output_header(int rate, - int channels, - int bits) { - static int already_written_header; - struct ao_sample_format format; - - if(!already_written_header) { - format.rate = rate; - format.bits = bits; - format.channels = channels; - format.byte_format = AO_FMT_BIG; - if(fwrite(&format, sizeof format, 1, outputfp) < 1) - fatal(errno, "decoding %s: writing format header", path); - already_written_header = 1; - } -} - -/** @brief Dithering state - * Filched from mpg321, which credits it to Robert Leslie */ -struct audio_dither { - mad_fixed_t error[3]; - mad_fixed_t random; -}; - -/** @brief 32-bit PRNG - * Filched from mpg321, which credits it to Robert Leslie */ -static inline unsigned long prng(unsigned long state) -{ - return (state * 0x0019660dL + 0x3c6ef35fL) & 0xffffffffL; -} - -/** @brief Generic linear sample quantize and dither routine - * Filched from mpg321, which credits it to Robert Leslie */ -#define bits 16 -static long audio_linear_dither(mad_fixed_t sample, - struct audio_dither *dither) { - unsigned int scalebits; - mad_fixed_t output, mask, rnd; - - enum { - MIN = -MAD_F_ONE, - MAX = MAD_F_ONE - 1 - }; - - /* noise shape */ - sample += dither->error[0] - dither->error[1] + dither->error[2]; - - dither->error[2] = dither->error[1]; - dither->error[1] = dither->error[0] / 2; - - /* bias */ - output = sample + (1L << (MAD_F_FRACBITS + 1 - bits - 1)); - - scalebits = MAD_F_FRACBITS + 1 - bits; - mask = (1L << scalebits) - 1; - - /* dither */ - rnd = prng(dither->random); - output += (rnd & mask) - (dither->random & mask); - - dither->random = rnd; - - /* clip */ - if (output > MAX) { - output = MAX; - - if (sample > MAX) - sample = MAX; - } - else if (output < MIN) { - output = MIN; - - if (sample < MIN) - sample = MIN; - } - - /* quantize */ - output &= ~mask; - - /* error feedback */ - dither->error[0] = sample - output; - - /* scale */ - return output >> scalebits; -} -#undef bits - -/** @brief MP3 output callback */ -static enum mad_flow mp3_output(void attribute((unused)) *data, - struct mad_header const *header, - struct mad_pcm *pcm) { - size_t n = pcm->length; - const mad_fixed_t *l = pcm->samples[0], *r = pcm->samples[1]; - static struct audio_dither ld[1], rd[1]; - - output_header(header->samplerate, - pcm->channels, - 16); - switch(pcm->channels) { - case 1: - while(n--) - output_16(audio_linear_dither(*l++, ld)); - break; - case 2: - while(n--) { - output_16(audio_linear_dither(*l++, ld)); - output_16(audio_linear_dither(*r++, rd)); - } - break; - default: - fatal(0, "decoding %s: unsupported channel count %d", path, pcm->channels); - } - return MAD_FLOW_CONTINUE; -} - -/** @brief MP3 input callback */ -static enum mad_flow mp3_input(void attribute((unused)) *data, - struct mad_stream *stream) { - const size_t n = fill(); - fprintf(stderr, "n=%zu\n", n); - if(!n) - return MAD_FLOW_STOP; - mad_stream_buffer(stream, buffer, n); - return MAD_FLOW_CONTINUE; -} - - -/** @brief MP3 error callback */ -static enum mad_flow mp3_error(void attribute((unused)) *data, - struct mad_stream *stream, - struct mad_frame attribute((unused)) *frame) { - error(0, "decoding %s: %s (%#04x)", - path, mad_stream_errorstr(stream), stream->error); - return MAD_FLOW_CONTINUE; -} - -/** @brief MP3 header callback */ -static enum mad_flow mp3_header(void attribute((unused)) *data, - struct mad_header const *header) { - output_header(header->samplerate, - MAD_NCHANNELS(header), - 16); - return MAD_FLOW_CONTINUE; -} - -/** @brief MP3 decoder */ -static void decode_mp3(void) { - struct mad_decoder mad[1]; - - open_input(); - mad_decoder_init(mad, 0/*data*/, mp3_input, mp3_header, 0/*filter*/, - mp3_output, mp3_error, 0/*message*/); - if(mad_decoder_run(mad, MAD_DECODER_MODE_SYNC)) - exit(1); - mad_decoder_finish(mad); +void output_header(int rate, + int channels, + int bits, + int nbytes, + int endian) { + struct stream_header header; + + if(bits <= 0 || bits % 8 || bits > 64) + disorder_fatal(0, "decoding %s: unsupported sample size %d bits", + path, bits); + if(channels <= 0 || channels > 2) + disorder_fatal(0, "decoding %s: unsupported channel count %d", + path, channels); + if(rate <= 0) + disorder_fatal(0, "decoding %s: nonsensical sample rate %dHz", path, rate); + header.rate = rate; + header.bits = bits; + header.channels = channels; + header.endian = endian; + header.nbytes = nbytes; + if(fwrite(&header, sizeof header, 1, outputfp) < 1) + disorder_fatal(errno, "decoding %s: writing format header", path); } /** @brief Lookup table of decoders */ static const struct decoder decoders[] = { { "*.mp3", decode_mp3 }, { "*.MP3", decode_mp3 }, + { "*.ogg", decode_ogg }, + { "*.OGG", decode_ogg }, + { "*.flac", decode_flac }, + { "*.FLAC", decode_flac }, + { "*.wav", decode_wav }, + { "*.WAV", decode_wav }, { 0, 0 } }; @@ -271,33 +108,26 @@ static void help(void) { exit(0); } -/* Display version number and terminate. */ -static void version(void) { - xprintf("disorder-decode version %s\n", disorder_version_string); - xfclose(stdout); - exit(0); -} - int main(int argc, char **argv) { int n; const char *e; set_progname(argv); - if(!setlocale(LC_CTYPE, "")) fatal(errno, "calling setlocale"); + if(!setlocale(LC_CTYPE, "")) disorder_fatal(errno, "calling setlocale"); while((n = getopt_long(argc, argv, "hV", options, 0)) >= 0) { switch(n) { case 'h': help(); - case 'V': version(); - default: fatal(0, "invalid option"); + case 'V': version("disorder-decode"); + default: disorder_fatal(0, "invalid option"); } } if(optind >= argc) - fatal(0, "missing filename"); + disorder_fatal(0, "missing filename"); if(optind + 1 < argc) - fatal(0, "excess arguments"); + disorder_fatal(0, "excess arguments"); if((e = getenv("DISORDER_RAW_FD"))) { if(!(outputfp = fdopen(atoi(e), "wb"))) - fatal(errno, "fdopen"); + disorder_fatal(errno, "fdopen"); } else outputfp = stdout; path = argv[optind]; @@ -307,7 +137,7 @@ int main(int argc, char **argv) { ++n) ; if(!decoders[n].pattern) - fatal(0, "cannot determine file type for %s", path); + disorder_fatal(0, "cannot determine file type for %s", path); decoders[n].decode(); xfclose(outputfp); return 0;