From 4673583e61c883e0b681bdb82d1e7afa12a27348 Mon Sep 17 00:00:00 2001 From: "Matthias P. Braendli" Date: Sun, 9 Mar 2014 16:41:25 +0100 Subject: Move sources to src, rename all programs --- .gitignore | 8 +- Makefile.am | 50 ++-- aac-enc-dabplus-zmq.c | 502 ------------------------------------ aac-enc-dabplus.c | 425 ------------------------------ aac-enc.c | 234 ----------------- mot-encoder.c | 638 ---------------------------------------------- src/aac-enc-dabplus-zmq.c | 502 ++++++++++++++++++++++++++++++++++++ src/aac-enc-dabplus.c | 425 ++++++++++++++++++++++++++++++ src/aac-enc.c | 234 +++++++++++++++++ src/mot-encoder.c | 638 ++++++++++++++++++++++++++++++++++++++++++++++ src/wavreader.c | 163 ++++++++++++ src/wavreader.h | 37 +++ wavreader.c | 163 ------------ wavreader.h | 37 --- 14 files changed, 2029 insertions(+), 2027 deletions(-) delete mode 100644 aac-enc-dabplus-zmq.c delete mode 100644 aac-enc-dabplus.c delete mode 100644 aac-enc.c delete mode 100644 mot-encoder.c create mode 100644 src/aac-enc-dabplus-zmq.c create mode 100644 src/aac-enc-dabplus.c create mode 100644 src/aac-enc.c create mode 100644 src/mot-encoder.c create mode 100644 src/wavreader.c create mode 100644 src/wavreader.h delete mode 100644 wavreader.c delete mode 100644 wavreader.h diff --git a/.gitignore b/.gitignore index fe88485..2014f3a 100644 --- a/.gitignore +++ b/.gitignore @@ -26,9 +26,7 @@ m4/ltversion.m4 m4/lt~obsolete.m4 missing stamp-h1 -aac-enc mot-encoder -aac-enc-dabplus -aac-enc-dabplus-zmq -dabplus-zmq -alsa-dabplus-zmq +dabplus-enc-file +dabplus-enc-file-zmq +dabplus-enc-alsa-zmq diff --git a/Makefile.am b/Makefile.am index f0c5de4..1fdddb8 100644 --- a/Makefile.am +++ b/Makefile.am @@ -33,36 +33,40 @@ libfdk_aac_la_CXXFLAGS = -fno-exceptions -fno-rtti libfdk_aac_la_LDFLAGS = -version-info @FDK_AAC_VERSION@ -no-undefined \ -export-symbols $(top_srcdir)/fdk-aac.sym -aac_enc_LDADD = libfdk-aac.la -aac_enc_SOURCES = aac-enc.c wavreader.c +#aac_enc_LDADD = libfdk-aac.la +#aac_enc_SOURCES = src/aac-enc.c \ +# src/wavreader.c -aac_enc_dabplus_LDADD = libfdk-aac.la -lfec -aac_enc_dabplus_SOURCES = aac-enc-dabplus.c wavreader.c +dabplus_enc_file_LDADD = libfdk-aac.la -lfec +dabplus_enc_file_SOURCES = src/aac-enc-dabplus.c \ + src/wavreader.c -aac_enc_dabplus_zmq_LDADD = libfdk-aac.la -lfec -lzmq -aac_enc_dabplus_zmq_SOURCES = aac-enc-dabplus-zmq.c wavreader.c \ - contrib/lib_crc.c +dabplus_enc_file_zmq_LDADD = libfdk-aac.la -lfec -lzmq +dabplus_enc_file_zmq_SOURCES = src/aac-enc-dabplus-zmq.c \ + src/wavreader.c \ + contrib/lib_crc.c -mot_encoder_CFLAGS = @MAGICKWAND_CFLAGS@ -Icontrib -mot_encoder_LDADD = @MAGICKWAND_LDADD@ -mot_encoder_SOURCES = mot-encoder.c contrib/lib_crc.c +dabplus_enc_alsa_zmq_LDFLAGS = -no-install +dabplus_enc_alsa_zmq_LDADD = libfdk-aac.la -lfec -lzmq -lasound \ + -lrt -lboost_thread +dabplus_enc_alsa_zmq_SOURCES = src/AlsaDabplus.cpp \ + src/AlsaInput.cpp \ + src/AlsaInput.h \ + src/SampleQueue.h \ + src/zmq.hpp -dabplus_enc_LDFLAGS = -no-install -dabplus_enc_LDADD = libfdk-aac.la -lfec -lzmq -lasound \ - -lrt -lboost_thread -dabplus_enc_SOURCES = src/AlsaDabplus.cpp \ - src/AlsaInput.cpp \ - src/AlsaInput.h \ - src/SampleQueue.h \ - src/zmq.hpp +mot_encoder_CFLAGS = @MAGICKWAND_CFLAGS@ -Icontrib +mot_encoder_LDADD = @MAGICKWAND_LDADD@ +mot_encoder_SOURCES = src/mot-encoder.c \ + contrib/lib_crc.c -bin_PROGRAMS = aac-enc-dabplus$(EXEEXT) \ - aac-enc-dabplus-zmq$(EXEEXT) \ - dabplus-enc$(EXEEXT) \ - mot-encoder$(EXEEXT) +bin_PROGRAMS = dabplus-enc-file$(EXEEXT) \ + dabplus-enc-file-zmq$(EXEEXT) \ + dabplus-enc-alsa-zmq$(EXEEXT) \ + mot-encoder$(EXEEXT) -noinst_HEADERS = wavreader.h +noinst_HEADERS = src/wavreader.h AACDEC_SRC = \ libAACdec/src/aacdec_drc.cpp \ diff --git a/aac-enc-dabplus-zmq.c b/aac-enc-dabplus-zmq.c deleted file mode 100644 index febbdd9..0000000 --- a/aac-enc-dabplus-zmq.c +++ /dev/null @@ -1,502 +0,0 @@ -/* ------------------------------------------------------------------ - * Copyright (C) 2011 Martin Storsjo - * Copyright (C) 2013,2014 Matthias P. Braendli - * Copyright (C) 2014 CSP Innovazione nelle ICT s.c.a r.l. - * http://rd.csp.it/ - * - * http://opendigitalradio.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 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. - * - * You should have received a copy of the GNU General Public License - * along with this program If not, see . - * ------------------------------------------------------------------- - */ - -#include -#include -#include -#include -#include -#include -#include -#include -#include "libAACenc/include/aacenc_lib.h" -#include "wavreader.h" - -#include -#include -#include -#include -#include - -#include "contrib/lib_crc.h" - -#define MIN(a,b) (((a)<(b))?(a):(b)) -#define MAX(a,b) (((a)>(b))?(a):(b)) - -void usage(const char* name) { - fprintf(stderr, "fdk-aac-dabplus HE-AACv2 encoder for DAB+\n\n"); - fprintf(stderr, "With ZeroMQ output for ODR-DabMux\n"); - fprintf(stderr, "and PAD (DLS and MOT Slideshow) by http://rd.csp.it\n"); - fprintf(stderr, "\n"); - fprintf(stderr, "http://opendigitalradio.org\n"); - fprintf(stderr, "\nUsage:\n"); - fprintf(stderr, "%s [OPTION...]\n", name); - fprintf(stderr, -" -b, --bitrate={ 8, 16, ..., 192 } Output bitrate in kbps. Must be 8 multiple.\n" -//" -d, --data=FILENAME Set data filename.\n" -//" -g, --fs-bug Turn on FS bug mitigation.\n" -" -i, --input=FILENAME Input filename (default: stdin).\n" -" -o, --output=URI Output zmq uri. (e.g. 'tcp://*:9000')\n" -" -a, --afterburner Turn on AAC encoder quality increaser.\n" -//" -m, --message Turn on AAC frame messages.\n" -" -p, --pad=BYTES Set PAD size in bytes.\n" -" -f, --format={ wav, raw } Set input file format (default: wav).\n" -" -c, --channels={ 1, 2 } Nb of input channels for raw input (default: 2).\n" -" -r, --rate={ 32000, 48000 } Sample rate for raw input (default: 48000).\n" -//" -t, --type=TYPE Set data type (dls|pad|packet|dg).\n" -//" -v, --verbose=LEVEL Set verbosity level.\n" -//" -V, --version Print version and exit.\n" -//" --mi=[ 0, ... ] Set AAC frame messages interval in milliseconds.\n" -//" --ma=[ 0, ... ] Set AAC frame messages attack time in milliseconds.\n" -//" -l, --lp Set frame size to 1024 instead of 960.\n" -"\n" -"Only the tcp:// zeromq transport has been tested until now.\n" - -); - -} - - -#define no_argument 0 -#define required_argument 1 -#define optional_argument 2 - -int main(int argc, char *argv[]) { - int subchannel_index = 8; //64kbps subchannel - int ch=0; - const char *infile = NULL; - const char *outuri = NULL; - FILE *in_fh; - void *wav; - int wav_format, bits_per_sample, sample_rate=48000, channels=2; - uint8_t* input_buf; - int16_t* convert_buf; - void *rs_handler = NULL; - int aot = AOT_DABPLUS_AAC_LC; - int afterburner = 0, raw_input=0; - HANDLE_AACENCODER handle; - CHANNEL_MODE mode; - AACENC_InfoStruct info = { 0 }; - - int pad_fd; - unsigned char pad_buf[128]; - int padlen; - - void *zmq_context = zmq_ctx_new(); - void *zmq_sock = NULL; - - const struct option longopts[] = { - {"bitrate", required_argument, 0, 'b'}, - {"input", required_argument, 0, 'i'}, - {"output", required_argument, 0, 'o'}, - {"format", required_argument, 0, 'f'}, - {"rate", required_argument, 0, 'r'}, - {"channels", required_argument, 0, 'c'}, - {"pad", required_argument, 0, 'p'}, - //{"lp", no_argument, 0, 'l'}, - {"afterburner", no_argument, 0, 'a'}, - {"help", no_argument, 0, 'h'}, - {0,0,0,0}, - }; - - if (argc == 1) { - usage(argv[0]); - return 0; - } - - int index; - while(ch != -1) { - ch = getopt_long(argc, argv, "tlhab:c:i:o:r:f:p:", longopts, &index); - switch (ch) { - case 'f': - if(strcmp(optarg, "raw")==0) { - raw_input = 1; - } else if(strcmp(optarg, "wav")!=0) - usage(argv[0]); - break; - case 'a': - afterburner = 1; - break; - case 'b': - subchannel_index = atoi(optarg) / 8; - break; - case 'c': - channels = atoi(optarg); - break; - case 'r': - sample_rate = atoi(optarg); - break; - case 'i': - infile = optarg; - break; - case 'o': - outuri = optarg; - break; - case 'p': - padlen = atoi(optarg); - break; - case '?': - case 'h': - usage(argv[0]); - return 1; - } - } - - if(subchannel_index < 1 || subchannel_index > 24) { - fprintf(stderr, "Bad subchannels number: %d, try other bitrate.\n", - subchannel_index); - return 1; - } - if(padlen != 0) { - int flags; - if (mkfifo("/tmp/pad.fifo", S_IWUSR | S_IRUSR | S_IRGRP | S_IROTH) != 0) { - if (errno != EEXIST) { - fprintf(stderr, "Can't create pad file: %d!\n", errno); - return 1; - } - } - pad_fd = open("/tmp/pad.fifo", O_RDONLY | O_NONBLOCK); - if (pad_fd == -1) { - fprintf(stderr, "Can't open pad file!\n"); - return 1; - } - flags = fcntl(pad_fd, F_GETFL, 0); - if (fcntl(pad_fd, F_SETFL, flags | O_NONBLOCK)) { - fprintf(stderr, "Can't set non-blocking mode in pad file!\n"); - return 1; - } - } - - if(raw_input) { - if(infile && strcmp(infile, "-")) { - in_fh = fopen(infile, "rb"); - if(!in_fh) { - fprintf(stderr, "Can't open input file!\n"); - return 1; - } - } else { - in_fh = stdin; - } - } else { - wav = wav_read_open(infile); - if (!wav) { - fprintf(stderr, "Unable to open wav file %s\n", infile); - return 1; - } - if (!wav_get_header(wav, &wav_format, &channels, &sample_rate, &bits_per_sample, NULL)) { - fprintf(stderr, "Bad wav file %s\n", infile); - return 1; - } - if (wav_format != 1) { - fprintf(stderr, "Unsupported WAV format %d\n", wav_format); - return 1; - } - if (bits_per_sample != 16) { - fprintf(stderr, "Unsupported WAV sample depth %d\n", bits_per_sample); - return 1; - } - if (channels > 2) { - fprintf(stderr, "Unsupported WAV channels %d\n", channels); - return 1; - } - } - - if (outuri) { - zmq_sock = zmq_socket(zmq_context, ZMQ_PUB); - if (zmq_sock == NULL) { - fprintf(stderr, "Error occurred during zmq_socket: %s\n", zmq_strerror(errno)); - return 2; - } - if (zmq_connect(zmq_sock, outuri) != 0) { - fprintf(stderr, "Error occurred during zmq_connect: %s\n", zmq_strerror(errno)); - return 2; - } - } else { - fprintf(stderr, "Output URI not defined\n"); - return 1; - } - - - switch (channels) { - case 1: mode = MODE_1; break; - case 2: mode = MODE_2; break; - default: - fprintf(stderr, "Unsupported channels number %d\n", channels); - return 1; - } - - - if (aacEncOpen(&handle, 0x01|0x02|0x04, channels) != AACENC_OK) { - fprintf(stderr, "Unable to open encoder\n"); - return 1; - } - - - if(channels == 2 && subchannel_index <= 6) - aot = AOT_DABPLUS_PS; - else if((channels == 1 && subchannel_index <= 8) || subchannel_index <= 10) - aot = AOT_DABPLUS_SBR; - - fprintf(stderr, "Using %d subchannels. AAC type: %s%s%s. channels=%d, sample_rate=%d\n", - subchannel_index, - aot == AOT_DABPLUS_PS ? "HE-AAC v2" : "", - aot == AOT_DABPLUS_SBR ? "HE-AAC" : "", - aot == AOT_DABPLUS_AAC_LC ? "AAC-LC" : "", - channels, sample_rate); - - if (aacEncoder_SetParam(handle, AACENC_AOT, aot) != AACENC_OK) { - fprintf(stderr, "Unable to set the AOT\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_SAMPLERATE, sample_rate) != AACENC_OK) { - fprintf(stderr, "Unable to set the samplerate\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_CHANNELMODE, mode) != AACENC_OK) { - fprintf(stderr, "Unable to set the channel mode\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_CHANNELORDER, 1) != AACENC_OK) { - fprintf(stderr, "Unable to set the wav channel order\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_GRANULE_LENGTH, 960) != AACENC_OK) { - fprintf(stderr, "Unable to set the granule length\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_TRANSMUX, TT_DABPLUS) != AACENC_OK) { - fprintf(stderr, "Unable to set the RAW transmux\n"); - return 1; - } - - /*if (aacEncoder_SetParam(handle, AACENC_BITRATEMODE, 7 *AACENC_BR_MODE_SFR*) != AACENC_OK) { - fprintf(stderr, "Unable to set the bitrate mode\n"); - return 1; - }*/ - - - fprintf(stderr, "AAC bitrate set to: %d\n", subchannel_index*8000); - if (aacEncoder_SetParam(handle, AACENC_BITRATE, subchannel_index*8000) != AACENC_OK) { - fprintf(stderr, "Unable to set the bitrate\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_AFTERBURNER, afterburner) != AACENC_OK) { - fprintf(stderr, "Unable to set the afterburner mode\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_ANCILLARY_BITRATE, 0) != AACENC_OK) { - fprintf(stderr, "Unable to set the ancillary bitrate\n"); - return 1; - } - if (aacEncEncode(handle, NULL, NULL, NULL, NULL) != AACENC_OK) { - fprintf(stderr, "Unable to initialize the encoder\n"); - return 1; - } - if (aacEncInfo(handle, &info) != AACENC_OK) { - fprintf(stderr, "Unable to get the encoder info\n"); - return 1; - } - - fprintf(stderr, "DAB+ Encoding: framelen=%d\n", info.frameLength); - - int input_size = channels*2*info.frameLength; - input_buf = (uint8_t*) malloc(input_size); - convert_buf = (int16_t*) malloc(input_size); - - /* symsize=8, gfpoly=0x11d, fcr=0, prim=1, nroots=10, pad=135 */ - rs_handler = init_rs_char(8, 0x11d, 0, 1, 10, 135); - if (rs_handler == NULL) { - perror("init_rs_char failed"); - return 0; - } - - int loops = 0; - int outbuf_size = subchannel_index*120; - uint8_t outbuf[20480]; - - if(outbuf_size % 5 != 0) { - fprintf(stderr, "(outbuf_size mod 5) = %d\n", outbuf_size % 5); - } - - fprintf(stderr, "outbuf_size: %d\n", outbuf_size); - //outbuf_size += (4 * subchannel_index * (8*8)/8) - outbuf_size/5; - fprintf(stderr, "outbuf_size: %d\n", outbuf_size); - - int frame=0; - int send_error_count = 0; - while (1) { - memset(outbuf, 0x00, outbuf_size); - - AACENC_BufDesc in_buf = { 0 }, out_buf = { 0 }; - AACENC_InArgs in_args = { 0 }; - AACENC_OutArgs out_args = { 0 }; - int in_identifier[] = {IN_AUDIO_DATA, IN_ANCILLRY_DATA}; - int in_size[2], in_elem_size[2]; - int out_identifier = OUT_BITSTREAM_DATA; - int out_size, out_elem_size; - int pcmread=0, i, ret; - int send_error; - void *in_ptr[2], *out_ptr; - AACENC_ERROR err; - - // Read data from the PAD fifo - if (padlen != 0) { - ret = read(pad_fd, pad_buf, padlen); - } - else { - ret = 0; - } - - - if(ret < 0 && errno == EAGAIN) { - // If this condition passes, there is no data to be read - in_buf.numBufs = 1; // Samples; - } - else if(ret >= 0) { - // Otherwise, you're good to go and buffer should contain "count" bytes. - in_buf.numBufs = 2; // Samples + Data; - } - else { - // Some other error occurred during read. - fprintf(stderr, "Unable to read from PAD!\n"); - break; - } - - if(raw_input) { - if(fread(input_buf, input_size, 1, in_fh) == 1) { - pcmread = input_size; - } else { - fprintf(stderr, "Unable to read from input!\n"); - break; - } - } else { - pcmread = wav_read_data(wav, input_buf, input_size); - } - - for (i = 0; i < pcmread/2; i++) { - const uint8_t* in = &input_buf[2*i]; - convert_buf[i] = in[0] | (in[1] << 8); - } - - if (pcmread <= 0) { - in_args.numInSamples = -1; - } else { - in_ptr[0] = convert_buf; - in_ptr[1] = pad_buf; - in_size[0] = pcmread; - in_size[1] = padlen; - - in_elem_size[0] = 2; - in_elem_size[1] = sizeof(UCHAR); - - in_args.numInSamples = pcmread/2; - in_args.numAncBytes = padlen; - - //in_buf.numBufs = 2; // Samples + Data - - in_buf.bufs = (void**)&in_ptr; - in_buf.bufferIdentifiers = in_identifier; - in_buf.bufSizes = in_size; - in_buf.bufElSizes = in_elem_size; - - } - out_ptr = outbuf; - out_size = sizeof(outbuf); - out_elem_size = 1; - out_buf.numBufs = 1; - out_buf.bufs = &out_ptr; - out_buf.bufferIdentifiers = &out_identifier; - out_buf.bufSizes = &out_size; - out_buf.bufElSizes = &out_elem_size; - - if ((err = aacEncEncode(handle, &in_buf, &out_buf, &in_args, &out_args)) != AACENC_OK) { - if (err == AACENC_ENCODE_EOF) - break; - fprintf(stderr, "Encoding failed\n"); - return 1; - } - if (out_args.numOutBytes == 0) - continue; -#if 0 - unsigned char au_start[6]; - unsigned char* sfbuf = outbuf; - au_start[0] = 6; - au_start[1] = (*(sfbuf + 3) << 4) + ((*(sfbuf + 4)) >> 4); - au_start[2] = ((*(sfbuf + 4) & 0x0f) << 8) + *(sfbuf + 5); - fprintf (stderr, "au_start[0] = %d\n", au_start[0]); - fprintf (stderr, "au_start[1] = %d\n", au_start[1]); - fprintf (stderr, "au_start[2] = %d\n", au_start[2]); -#endif - - int row, col; - unsigned char buf_to_rs_enc[110]; - unsigned char rs_enc[10]; - for(row=0; row < subchannel_index; row++) { - for(col=0;col < 110; col++) { - buf_to_rs_enc[col] = outbuf[subchannel_index * col + row]; - } - - encode_rs_char(rs_handler, buf_to_rs_enc, rs_enc); - - for(col=110; col<120; col++) { - outbuf[subchannel_index * col + row] = rs_enc[col-110]; - assert(subchannel_index * col + row < outbuf_size); - } - } - - send_error = zmq_send(zmq_sock, outbuf, outbuf_size, ZMQ_DONTWAIT); - if (send_error < 0) { - fprintf(stderr, "ZeroMQ send failed! %s\n", zmq_strerror(errno)); - send_error_count ++; - } - - if (send_error_count > 10) - { - fprintf(stderr, "ZeroMQ send failed ten times, aborting!\n"); - break; - } - //fwrite(outbuf, 1, /*out_args.numOutBytes*/ outbuf_size, out_fh); - //fprintf(stderr, "Written %d/%d bytes!\n", out_args.numOutBytes + row*10, outbuf_size); - if(out_args.numOutBytes + row*10 == outbuf_size) - fprintf(stderr, "."); - -// if(frame > 10) -// break; - frame++; - } - free(input_buf); - free(convert_buf); - if(raw_input) { - fclose(in_fh); - } else { - wav_read_close(wav); - } - zmq_close(zmq_sock); - free_rs_char(rs_handler); - - aacEncClose(&handle); - - zmq_ctx_term(zmq_context); - return 0; -} - diff --git a/aac-enc-dabplus.c b/aac-enc-dabplus.c deleted file mode 100644 index 6a0b882..0000000 --- a/aac-enc-dabplus.c +++ /dev/null @@ -1,425 +0,0 @@ -/* ------------------------------------------------------------------ - * Copyright (C) 2011 Martin Storsjo - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either - * express or implied. - * See the License for the specific language governing permissions - * and limitations under the License. - * ------------------------------------------------------------------- - */ - -#include -#include -#include -#include -#include -#include -#include -#include "libAACenc/include/aacenc_lib.h" -#include "wavreader.h" - -#include - -void usage(const char* name) { - fprintf(stderr, "%s [OPTION...]\n", name); - fprintf(stderr, -" -b, --bitrate={ 8, 16, ..., 192 } Output bitrate in kbps. Must be 8 multiple.\n" -//" -d, --data=FILENAME Set data filename.\n" -//" -g, --fs-bug Turn on FS bug mitigation.\n" -" -i, --input=FILENAME Input filename (default: stdin).\n" -" -o, --output=FILENAME Output filename (default: stdout).\n" -" -a, --afterburner Turn on AAC encoder quality increaser.\n" -//" -m, --message Turn on AAC frame messages.\n" -//" -p, --pad=BYTES Set PAD size in bytes.\n" -" -f, --format={ wav, raw } Set input file format (default: wav).\n" -" -c, --channels={ 1, 2 } Nb of input channels for raw input (default: 2).\n" -" -r, --rate={ 32000, 48000 } Sample rate for raw input (default: 48000).\n" -//" -t, --type=TYPE Set data type (dls|pad|packet|dg).\n" -//" -v, --verbose=LEVEL Set verbosity level.\n" -//" -V, --version Print version and exit.\n" -//" --mi=[ 0, ... ] Set AAC frame messages interval in milliseconds.\n" -//" --ma=[ 0, ... ] Set AAC frame messages attack time in milliseconds.\n" -//" -t, --adts Set ADTS output format (for debugging).\n" -//" -l, --lp Set frame size to 1024 instead of 960.\n" - -); - -} - - -#define no_argument 0 -#define required_argument 1 -#define optional_argument 2 - -#define ADTS_HEADER_SIZE 7 -#define ADTS_MPEG_ID 1 /* 0: MPEG-4, 1: MPEG-2 */ -#define ADTS_MPEG_PROFILE 1 -const int mpeg4audio_sample_rates[16] = { - 96000, 88200, 64000, 48000, 44100, 32000, - 24000, 22050, 16000, 12000, 11025, 8000, 7350 -}; - -int FindSRIndex(int sr) -{ - int i; - for (i = 0; i < 16; i++) { - if (sr == mpeg4audio_sample_rates[i]) - return i; - } - return 16 - 1; -} - -void adts_hdr_up(char *buff, int size) -{ - unsigned short len = size + ADTS_HEADER_SIZE; - unsigned short buffer_fullness = 0x07FF; - - /* frame length, 13 bits */ - buff[3] &= 0xFC; - buff[3] |= ((len >> 11) & 0x03); /* 2b: aac_frame_length */ - buff[4] = len >> 3; /* 8b: aac_frame_length */ - buff[5] = (len << 5) & 0xE0; /* 3b: aac_frame_length */ - /* buffer fullness, 11 bits */ - buff[5] |= ((buffer_fullness >> 6) & 0x1F); /* 5b: adts_buffer_fullness */ - buff[6] = (buffer_fullness << 2) & 0xFC; /* 6b: adts_buffer_fullness */ - /* 2b: num_raw_data_blocks */ -} - -int main(int argc, char *argv[]) { - int subchannel_index = 8; //64kbps subchannel - int ch=0; - const char *infile, *outfile; - FILE *in_fh, *out_fh; - void *wav; - int wav_format, bits_per_sample, sample_rate=48000, channels=2; - uint8_t* input_buf; - int16_t* convert_buf; - void *rs_handler = NULL; - int aot = AOT_DABPLUS_AAC_LC; - int afterburner = 0, raw_input=0; - HANDLE_AACENCODER handle; - CHANNEL_MODE mode; - AACENC_InfoStruct info = { 0 }; - - const struct option longopts[] = { - {"bitrate", required_argument, 0, 'b'}, - {"input", required_argument, 0, 'i'}, - {"output", required_argument, 0, 'o'}, - {"format", required_argument, 0, 'f'}, - {"rate", required_argument, 0, 'r'}, - {"channels", required_argument, 0, 'c'}, - //{"lp", no_argument, 0, 'l'}, - //{"adts", no_argument, 0, 't'}, - {"afterburner", no_argument, 0, 'a'}, - {"help", no_argument, 0, 'h'}, - {0,0,0,0}, - }; - - int index; - while(ch != -1) { - ch = getopt_long(argc, argv, "tlhab:c:i:o:r:f:", longopts, &index); - switch (ch) { - case 'f': - if(strcmp(optarg, "raw")==0) { - raw_input = 1; - } else if(strcmp(optarg, "wav")!=0) - usage(argv[0]); - break; - case 'a': - afterburner = 1; - break; - case 'b': - subchannel_index = atoi(optarg) / 8; - break; - case 'c': - channels = atoi(optarg); - break; - case 'r': - sample_rate = atoi(optarg); - break; - case 'i': - infile = optarg; - break; - case 'o': - outfile = optarg; - break; - case '?': - case 'h': - usage(argv[0]); - return 1; - } - } - - if(subchannel_index < 1 || subchannel_index > 24) { - fprintf(stderr, "Bad subchannels number: %d, try other bitrate.\n", subchannel_index); - return 1; - } - - if(raw_input) { - if(infile && strcmp(infile, "-")) { - in_fh = fopen(infile, "rb"); - if(!in_fh) { - fprintf(stderr, "Can't open input file!\n"); - return 1; - } - } else { - in_fh = stdin; - } - } else { - wav = wav_read_open(infile); - if (!wav) { - fprintf(stderr, "Unable to open wav file %s\n", infile); - return 1; - } - if (!wav_get_header(wav, &wav_format, &channels, &sample_rate, &bits_per_sample, NULL)) { - fprintf(stderr, "Bad wav file %s\n", infile); - return 1; - } - if (wav_format != 1) { - fprintf(stderr, "Unsupported WAV format %d\n", wav_format); - return 1; - } - if (bits_per_sample != 16) { - fprintf(stderr, "Unsupported WAV sample depth %d\n", bits_per_sample); - return 1; - } - if (channels > 2) { - fprintf(stderr, "Unsupported WAV channels %d\n", channels); - return 1; - } - } - - if(outfile && strcmp(outfile, "-")) { - out_fh = fopen(outfile, "wb"); - if(!out_fh) { - fprintf(stderr, "Can't open output file!\n"); - return 1; - } - } else { - out_fh = stdout; - } - - - switch (channels) { - case 1: mode = MODE_1; break; - case 2: mode = MODE_2; break; - default: - fprintf(stderr, "Unsupported channels number %d\n", channels); - return 1; - } - - - if (aacEncOpen(&handle, 0x01|0x02|0x04, channels) != AACENC_OK) { - fprintf(stderr, "Unable to open encoder\n"); - return 1; - } - - - if(channels == 2 && subchannel_index <= 6) - aot = AOT_DABPLUS_PS; - else if((channels == 1 && subchannel_index <= 8) || subchannel_index <= 10) - aot = AOT_DABPLUS_SBR; - - fprintf(stderr, "Using %d subchannels. AAC type: %s%s%s. channels=%d, sample_rate=%d\n", - subchannel_index, - aot == AOT_DABPLUS_PS ? "HE-AAC v2" : "", - aot == AOT_DABPLUS_SBR ? "HE-AAC" : "", - aot == AOT_DABPLUS_AAC_LC ? "AAC-LC" : "", - channels, sample_rate); - - if (aacEncoder_SetParam(handle, AACENC_AOT, aot) != AACENC_OK) { - fprintf(stderr, "Unable to set the AOT\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_SAMPLERATE, sample_rate) != AACENC_OK) { - fprintf(stderr, "Unable to set the AOT\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_CHANNELMODE, mode) != AACENC_OK) { - fprintf(stderr, "Unable to set the channel mode\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_CHANNELORDER, 1) != AACENC_OK) { - fprintf(stderr, "Unable to set the wav channel order\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_GRANULE_LENGTH, 960) != AACENC_OK) { - fprintf(stderr, "Unable to set the AOT\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_TRANSMUX, TT_DABPLUS) != AACENC_OK) { - fprintf(stderr, "Unable to set the RAW transmux\n"); - return 1; - } - - /*if (aacEncoder_SetParam(handle, AACENC_BITRATEMODE, 7 *AACENC_BR_MODE_SFR*) != AACENC_OK) { - fprintf(stderr, "Unable to set the bitrate mode\n"); - return 1; - }*/ - - - fprintf(stderr, "AAC bitrate set to: %d\n", subchannel_index*8000); - if (aacEncoder_SetParam(handle, AACENC_BITRATE, subchannel_index*8000) != AACENC_OK) { - fprintf(stderr, "Unable to set the bitrate\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_AFTERBURNER, afterburner) != AACENC_OK) { - fprintf(stderr, "Unable to set the afterburner mode\n"); - return 1; - } - if (aacEncEncode(handle, NULL, NULL, NULL, NULL) != AACENC_OK) { - fprintf(stderr, "Unable to initialize the encoder\n"); - return 1; - } - if (aacEncInfo(handle, &info) != AACENC_OK) { - fprintf(stderr, "Unable to get the encoder info\n"); - return 1; - } - - fprintf(stderr, "DAB+ Encoding: framelen=%d\n", info.frameLength); - - int input_size = channels*2*info.frameLength; - input_buf = (uint8_t*) malloc(input_size); - convert_buf = (int16_t*) malloc(input_size); - - /* symsize=8, gfpoly=0x11d, fcr=0, prim=1, nroots=10, pad=135 */ - rs_handler = init_rs_char(8, 0x11d, 0, 1, 10, 135); - if (rs_handler == NULL) { - perror("init_rs_char failed"); - return 0; - } - - int loops = 0; - int outbuf_size = subchannel_index*120; - uint8_t outbuf[20480]; - - if(outbuf_size % 5 != 0) { - fprintf(stderr, "(outbuf_size mod 5) = %d\n", outbuf_size % 5); - } - - fprintf(stderr, "outbuf_size: %d\n", outbuf_size); - //outbuf_size += (4 * subchannel_index * (8*8)/8) - outbuf_size/5; - fprintf(stderr, "outbuf_size: %d\n", outbuf_size); - - int frame=0; - while (1) { - memset(outbuf, 0x00, outbuf_size); - - AACENC_BufDesc in_buf = { 0 }, out_buf = { 0 }; - AACENC_InArgs in_args = { 0 }; - AACENC_OutArgs out_args = { 0 }; - int in_identifier = IN_AUDIO_DATA; - int in_size, in_elem_size; - int out_identifier = OUT_BITSTREAM_DATA; - int out_size, out_elem_size; - int read=0, i; - void *in_ptr, *out_ptr; - AACENC_ERROR err; - - if(raw_input) { - if(fread(input_buf, input_size, 1, in_fh) == 1) { - read = input_size; - } else { - fprintf(stderr, "Unable to read from input!\n"); - break; - } - } else { - read = wav_read_data(wav, input_buf, input_size); - // returns bytes read - } - - for (i = 0; i < read/2; i++) { - const uint8_t* in = &input_buf[2*i]; - convert_buf[i] = in[0] | (in[1] << 8); - } - - if (read <= 0) { - in_args.numInSamples = -1; - } else { - in_ptr = convert_buf; - in_size = read; - in_elem_size = 2; - - in_args.numInSamples = read/2; - in_buf.numBufs = 1; - in_buf.bufs = &in_ptr; - in_buf.bufferIdentifiers = &in_identifier; - in_buf.bufSizes = &in_size; - in_buf.bufElSizes = &in_elem_size; - } - out_ptr = outbuf; - out_size = sizeof(outbuf); - out_elem_size = 1; - out_buf.numBufs = 1; - out_buf.bufs = &out_ptr; - out_buf.bufferIdentifiers = &out_identifier; - out_buf.bufSizes = &out_size; - out_buf.bufElSizes = &out_elem_size; - - if ((err = aacEncEncode(handle, &in_buf, &out_buf, &in_args, &out_args)) != AACENC_OK) { - if (err == AACENC_ENCODE_EOF) - break; - fprintf(stderr, "Encoding failed\n"); - return 1; - } - if (out_args.numOutBytes == 0) - continue; -#if 0 - unsigned char au_start[6]; - unsigned char* sfbuf = outbuf; - au_start[0] = 6; - au_start[1] = (*(sfbuf + 3) << 4) + ((*(sfbuf + 4)) >> 4); - au_start[2] = ((*(sfbuf + 4) & 0x0f) << 8) + *(sfbuf + 5); - fprintf (stderr, "au_start[0] = %d\n", au_start[0]); - fprintf (stderr, "au_start[1] = %d\n", au_start[1]); - fprintf (stderr, "au_start[2] = %d\n", au_start[2]); -#endif - - int row, col; - char buf_to_rs_enc[110]; - char rs_enc[10]; - for(row=0; row < subchannel_index; row++) { - for(col=0;col < 110; col++) { - buf_to_rs_enc[col] = outbuf[subchannel_index * col + row]; - } - - encode_rs_char(rs_handler, buf_to_rs_enc, rs_enc); - - for(col=110; col<120; col++) { - outbuf[subchannel_index * col + row] = rs_enc[col-110]; - assert(subchannel_index * col + row < outbuf_size); - } - } - - fwrite(outbuf, 1, /*out_args.numOutBytes*/ outbuf_size, out_fh); - //fprintf(stderr, "Written %d/%d bytes!\n", out_args.numOutBytes + row*10, outbuf_size); - if(out_args.numOutBytes + row*10 == outbuf_size) - fprintf(stderr, "."); - -// if(frame > 10) -// break; - frame++; - } - free(input_buf); - free(convert_buf); - if(raw_input) { - fclose(in_fh); - } else { - wav_read_close(wav); - } - fclose(out_fh); - free_rs_char(rs_handler); - - aacEncClose(&handle); - - return 0; -} - diff --git a/aac-enc.c b/aac-enc.c deleted file mode 100644 index 1528881..0000000 --- a/aac-enc.c +++ /dev/null @@ -1,234 +0,0 @@ -/* ------------------------------------------------------------------ - * Copyright (C) 2011 Martin Storsjo - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either - * express or implied. - * See the License for the specific language governing permissions - * and limitations under the License. - * ------------------------------------------------------------------- - */ - -#include -#include -#include -#include -#include "libAACenc/include/aacenc_lib.h" -#include "wavreader.h" - -void usage(const char* name) { - fprintf(stderr, "%s [-r bitrate] [-t aot] [-a afterburner] [-s sbr] [-v vbr] in.wav out.aac\n", name); - fprintf(stderr, "Supported AOTs:\n"); - fprintf(stderr, "\t2\tAAC-LC\n"); - fprintf(stderr, "\t5\tHE-AAC\n"); - fprintf(stderr, "\t29\tHE-AAC v2\n"); - fprintf(stderr, "\t23\tAAC-LD\n"); - fprintf(stderr, "\t39\tAAC-ELD\n"); -} - -int main(int argc, char *argv[]) { - int bitrate = 64000; - int ch; - const char *infile, *outfile; - FILE *out; - void *wav; - int format, sample_rate, channels, bits_per_sample; - int input_size; - uint8_t* input_buf; - int16_t* convert_buf; - int aot = 2; - int afterburner = 1; - int eld_sbr = 0; - int vbr = 0; - HANDLE_AACENCODER handle; - CHANNEL_MODE mode; - AACENC_InfoStruct info = { 0 }; - while ((ch = getopt(argc, argv, "r:t:a:s:v:")) != -1) { - switch (ch) { - case 'r': - bitrate = atoi(optarg); - break; - case 't': - aot = atoi(optarg); - break; - case 'a': - afterburner = atoi(optarg); - break; - case 's': - eld_sbr = atoi(optarg); - break; - case 'v': - vbr = atoi(optarg); - break; - case '?': - default: - usage(argv[0]); - return 1; - } - } - if (argc - optind < 2) { - usage(argv[0]); - return 1; - } - infile = argv[optind]; - outfile = argv[optind + 1]; - - wav = wav_read_open(infile); - if (!wav) { - fprintf(stderr, "Unable to open wav file %s\n", infile); - return 1; - } - if (!wav_get_header(wav, &format, &channels, &sample_rate, &bits_per_sample, NULL)) { - fprintf(stderr, "Bad wav file %s\n", infile); - return 1; - } - if (format != 1) { - fprintf(stderr, "Unsupported WAV format %d\n", format); - return 1; - } - if (bits_per_sample != 16) { - fprintf(stderr, "Unsupported WAV sample depth %d\n", bits_per_sample); - return 1; - } - switch (channels) { - case 1: mode = MODE_1; break; - case 2: mode = MODE_2; break; - case 3: mode = MODE_1_2; break; - case 4: mode = MODE_1_2_1; break; - case 5: mode = MODE_1_2_2; break; - case 6: mode = MODE_1_2_2_1; break; - default: - fprintf(stderr, "Unsupported WAV channels %d\n", channels); - return 1; - } - if (aacEncOpen(&handle, 0, channels) != AACENC_OK) { - fprintf(stderr, "Unable to open encoder\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_AOT, aot) != AACENC_OK) { - fprintf(stderr, "Unable to set the AOT\n"); - return 1; - } - if (aot == 39 && eld_sbr) { - if (aacEncoder_SetParam(handle, AACENC_SBR_MODE, 1) != AACENC_OK) { - fprintf(stderr, "Unable to set SBR mode for ELD\n"); - return 1; - } - } - if (aacEncoder_SetParam(handle, AACENC_SAMPLERATE, sample_rate) != AACENC_OK) { - fprintf(stderr, "Unable to set the AOT\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_CHANNELMODE, mode) != AACENC_OK) { - fprintf(stderr, "Unable to set the channel mode\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_CHANNELORDER, 1) != AACENC_OK) { - fprintf(stderr, "Unable to set the wav channel order\n"); - return 1; - } - if (vbr) { - if (aacEncoder_SetParam(handle, AACENC_BITRATEMODE, vbr) != AACENC_OK) { - fprintf(stderr, "Unable to set the VBR bitrate mode\n"); - return 1; - } - } else { - if (aacEncoder_SetParam(handle, AACENC_BITRATE, bitrate) != AACENC_OK) { - fprintf(stderr, "Unable to set the bitrate\n"); - return 1; - } - } - if (aacEncoder_SetParam(handle, AACENC_TRANSMUX, 2) != AACENC_OK) { - fprintf(stderr, "Unable to set the ADTS transmux\n"); - return 1; - } - if (aacEncoder_SetParam(handle, AACENC_AFTERBURNER, afterburner) != AACENC_OK) { - fprintf(stderr, "Unable to set the afterburner mode\n"); - return 1; - } - if (aacEncEncode(handle, NULL, NULL, NULL, NULL) != AACENC_OK) { - fprintf(stderr, "Unable to initialize the encoder\n"); - return 1; - } - if (aacEncInfo(handle, &info) != AACENC_OK) { - fprintf(stderr, "Unable to get the encoder info\n"); - return 1; - } - - out = fopen(outfile, "wb"); - if (!out) { - perror(outfile); - return 1; - } - - input_size = channels*2*info.frameLength; - input_buf = (uint8_t*) malloc(input_size); - convert_buf = (int16_t*) malloc(input_size); - - while (1) { - AACENC_BufDesc in_buf = { 0 }, out_buf = { 0 }; - AACENC_InArgs in_args = { 0 }; - AACENC_OutArgs out_args = { 0 }; - int in_identifier = IN_AUDIO_DATA; - int in_size, in_elem_size; - int out_identifier = OUT_BITSTREAM_DATA; - int out_size, out_elem_size; - int read, i; - void *in_ptr, *out_ptr; - uint8_t outbuf[20480]; - AACENC_ERROR err; - - read = wav_read_data(wav, input_buf, input_size); - for (i = 0; i < read/2; i++) { - const uint8_t* in = &input_buf[2*i]; - convert_buf[i] = in[0] | (in[1] << 8); - } - if (read <= 0) { - in_args.numInSamples = -1; - } else { - in_ptr = convert_buf; - in_size = read; - in_elem_size = 2; - - in_args.numInSamples = read/2; - in_buf.numBufs = 1; - in_buf.bufs = &in_ptr; - in_buf.bufferIdentifiers = &in_identifier; - in_buf.bufSizes = &in_size; - in_buf.bufElSizes = &in_elem_size; - } - out_ptr = outbuf; - out_size = sizeof(outbuf); - out_elem_size = 1; - out_buf.numBufs = 1; - out_buf.bufs = &out_ptr; - out_buf.bufferIdentifiers = &out_identifier; - out_buf.bufSizes = &out_size; - out_buf.bufElSizes = &out_elem_size; - - if ((err = aacEncEncode(handle, &in_buf, &out_buf, &in_args, &out_args)) != AACENC_OK) { - if (err == AACENC_ENCODE_EOF) - break; - fprintf(stderr, "Encoding failed\n"); - return 1; - } - if (out_args.numOutBytes == 0) - continue; - fwrite(outbuf, 1, out_args.numOutBytes, out); - } - free(input_buf); - free(convert_buf); - fclose(out); - wav_read_close(wav); - aacEncClose(&handle); - - return 0; -} - diff --git a/mot-encoder.c b/mot-encoder.c deleted file mode 100644 index a60a39a..0000000 --- a/mot-encoder.c +++ /dev/null @@ -1,638 +0,0 @@ -/* - Copyright (C) 2014 CSP Innovazione nelle ICT s.c.a r.l. (http://rd.csp.it/) - - 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 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. - - You should have received a copy of the GNU General Public License - along with this program. If not, see . - - MotEncoder.c - Generete PAD data for MOT Slideshow and DLS - - Authors: - Sergio Sagliocco -*/ - -#include -#include -#include -#include -#include -#include -#include -#include -#include - -#include "lib_crc.h" - -#define MIN(a,b) (((a)<(b))?(a):(b)) -#define MAX(a,b) (((a)>(b))?(a):(b)) - -#define MAXSEGLEN 8179 -#define MAXDLS 129 - -typedef unsigned char UCHAR; -typedef unsigned short int USHORT; - -typedef struct { - // MSC Data Group Header (extension field not supported) - unsigned char extflag; // 1 bit - unsigned char crcflag; // 1 bit - unsigned char segflag; // 1 bit - unsigned char accflag; // 1 bit - unsigned char dgtype; // 4 bits - unsigned char cindex; // 4 bits - unsigned char rindex; // 4 bits - /// Session header - Segment field - unsigned char last; // 1 bit - unsigned short int segnum; // 16 bits - // Session header - User access field - unsigned char rfa; // 3 bits - unsigned char tidflag; // 1 bit - unsigned char lenid; // 4 bits - Fixed to value 2 in this implemntation - unsigned short int tid; // 16 bits - // MSC data group data field - // Mot Segmentation header - unsigned char rcount; // 3 bits - unsigned short int seglen; // 13 bits - // Mot segment - unsigned char* segdata; - // MSC data group CRC - unsigned short int crc; // 16 bits -} MSCDG; -/* - typedef struct { -// MOT HEADER CUSTOMIZED FOR SLIDESHOW APP -unsigned int bodysize; // 28 bits -unsigned short int headsize; // 13 bits -unsigned char ctype; // 6 bits -unsigned char sctype; // 9 bits -unsigned char triggertime[5]; // 0x85 0x00 0x00 0x00 0x00 => NOW -unsigned char contname[14]; // 0xCC 0x0C 0x00 imgXXXX.jpg -} MOTSLIDEHDR; -*/ -int encodeFile (char* fname, int fidx, int padlen); -void createMotHeader(size_t blobsize, int fidx, char* mothdr, int* mothdrlen); -void createMscDG(MSCDG* msc, unsigned short int dgtype, unsigned short int cindex, unsigned short int lastseg, unsigned short int tid, unsigned char* data, unsigned short int datalen); -void packMscDG(unsigned char* mscblob, MSCDG* msc, unsigned short int *bsize); -void writeMotPAD(unsigned char* mscdg, unsigned short int mscdgsize, unsigned short int padlen); -void create_dls_datagroup (char* text, int padlen, unsigned char*** p_dlsdg, int* p_numdg); -void writeDLS(int padlen); - -void usage(char* name) -{ - fprintf(stderr, "DAB MOT encoder\n" - "for slideshow and DLS\n\n" - "By CSP Innovazione nelle ICT s.c.a r.l. (http://rd.csp.it/)\n\n" - "Reads image data from the specified directory, and outputs PAD data\n" - "on standard output\n" - "Reads DLS from /tmp/dls.file\n"); - fprintf(stderr, "Usage: %s \n", name); -} - -int main (int argc, char *argv[]) -{ - int len,fidx,ret; - struct dirent *pDirent; - DIR *pDir; - char imagepath[128]; - char dlstext[MAXDLS],dlstextprev[MAXDLS]; - int padlen=53; - - - if (argc < 2) { - usage(argv[0]); - return 0; - } - - MagickWandGenesis(); - - fidx=0; - while(1) { - pDir = opendir (argv[1]); - if (pDir == NULL) { - printf ("Cannot open directory '%s'\n", argv[1]); - return 1; - } - if (fidx == 9999) fidx=0; - while ((pDirent = readdir(pDir)) != NULL) { - if (pDirent->d_name[0]!='.') { - sprintf (imagepath,"%s/%s",argv[1],pDirent->d_name); - ret=encodeFile(imagepath, fidx, padlen); - if (ret!=1) { - fprintf(stderr,"Error - Cannot encode file %s\n",pDirent->d_name); - } else { - fidx++; - writeDLS(padlen); - sleep(10); - } - } - } - closedir (pDir); - } - return 1; -} - -int encodeFile (char* fname, int fidx, int padlen) -{ - int fd=0,ret,mothdrlen,nseg,lastseglen,i,last,curseglen; - unsigned char mothdr[32]; - MagickWand *m_wand = NULL; - PixelWand *p_wand = NULL; - size_t blobsize,height,width; - unsigned char *blob = NULL, *curseg = NULL; - MagickBooleanType err; - MSCDG msc; - unsigned char mscblob[8200]; - unsigned short int mscblobsize; - //float aspectRatio; - - - m_wand = NewMagickWand(); - p_wand = NewPixelWand(); - PixelSetColor(p_wand,"black"); - - err = MagickReadImage(m_wand,fname); - if (err==MagickFalse) { - fprintf(stderr,"Error - Unable to load image %s\n",fname); - ret=0; - goto RETURN; - } - - height=MagickGetImageHeight(m_wand); - width=MagickGetImageWidth(m_wand); - //aspectRatio = (width * 1.0)/height; - - fprintf(stderr,"Image: %s (id=%d). Original size: %d x %d. ",fname,fidx,width,height); - - while (height>240 || width>320) { - if (height/240.0 > width/320.0) { - //width = height * aspectRatio; - width = width * 240.0 / height; - height = 240; - } - else { - //height = width * (1.0/aspectRatio); - height = height * 320.0 / width; - width = 320; - } - MagickResizeImage(m_wand,width,height,LanczosFilter,1); - } - - - - height=MagickGetImageHeight(m_wand); - width=MagickGetImageWidth(m_wand); - - MagickBorderImage(m_wand,p_wand,(320-width)/2,(240-height)/2); - - MagickSetImageCompressionQuality(m_wand,75); - MagickSetImageFormat(m_wand,"jpg"); - blob=MagickGetImagesBlob(m_wand, &blobsize); - fprintf(stderr,"Resized to %d x %d. Size after compression %d bytes\n",width,height,blobsize); - - nseg=blobsize / MAXSEGLEN; - lastseglen=blobsize % MAXSEGLEN; - if (lastseglen !=0) nseg++; - - createMotHeader(blobsize,fidx,mothdr,&mothdrlen); - // Create the MSC Data Group C-Structure - createMscDG(&msc,3,0,1,fidx,mothdr,mothdrlen); - // Generate the MSC DG frame (Figure 9 en 300 401) - packMscDG(mscblob,&msc,&mscblobsize); - writeMotPAD(mscblob,mscblobsize,padlen); - - for (i=0;i>24; - MotHeaderCore[1]=(blobsize<<4 & 0x00FF0000)>>16; - MotHeaderCore[2]=(blobsize<<4 & 0x0000FF00)>>8; - MotHeaderCore[3]=(blobsize<<4 & 0x000000FF); - - sprintf(cntemp,"img%04d.jpg",fidx); - for (i=0; iextflag=0; - msc->crcflag=1; - msc->segflag=1; - msc->accflag=1; - msc->dgtype=dgtype; - msc->cindex=cindex; - msc->rindex=0; - msc->last=lastseg; - msc->segnum=cindex; - msc->rfa=0; - msc->tidflag=1; - msc->lenid=2; - msc->tid=tid; - msc->segdata=data; - msc->rcount=0; - msc->seglen=datalen; - - return; -} - - -void packMscDG(unsigned char* b, MSCDG* msc, unsigned short int* bsize) { - - int i; - unsigned short int crc=0xFFFF; - - b[0]=(msc->extflag<<7)|(msc->crcflag<<6)|(msc->segflag<<5)|(msc->accflag<<4)|msc->dgtype; - b[1]=(msc->cindex<<4)|msc->rindex; - b[2]=(msc->last<<7)|((msc->segnum & 0x7F00)>>8); - b[3]=msc->segnum & 0x00FF; - b[4]=0; - b[4]=(msc->rfa<<5)|(msc->tidflag<<4)|msc->lenid; - b[5]=(msc->tid & 0xFF00)>>8; - b[6]=msc->tid & 0x00FF; - b[7]=(msc->rcount<<5)|((msc->seglen & 0x1F00)>>8); - b[8]=msc->seglen & 0x00FF; - - for (i=0;i<9;i++) - crc = update_crc_ccitt(crc, b[i]); - - for(i=0;iseglen;i++) { - b[i+9]=(msc->segdata)[i]; - crc = update_crc_ccitt(crc, b[i+9]); - } - - crc=~crc; - b[9+msc->seglen] = (crc & 0xFF00) >> 8; // HI CRC - b[9+msc->seglen+1] = crc & 0x00FF; // LO CRC - - *bsize=9+msc->seglen+1+1; - - //write(1,b,9+msc->seglen+1+1); - -} - - -void writeMotPAD(unsigned char* mscdg, unsigned short int mscdgsize, unsigned short int padlen) { - - unsigned char pad[128]; - int xpadlengthmask,i,j,numseg,lastseglen; - unsigned short int crc; - - - if (padlen==17) - xpadlengthmask=3; - else if (padlen==21) - xpadlengthmask=4; - else if (padlen==29) - xpadlengthmask=5; - else if (padlen==37) - xpadlengthmask=6; - else if (padlen==53) - xpadlengthmask=7; - - - // Write Data Group Length Indicator - crc = 0xffff; - pad[padlen-1]=0x02; // FF-PAD Byte L (CI=1) - pad[padlen-2]=0x20; // FF-PAD Byte L-1 (Variable size X_PAD) - pad[padlen-3]=(xpadlengthmask<<5) | 0x01; // CI => data length = 12 (011) - Application Type=2 (DLS - start of X-PAD data group) - pad[padlen-4]=0x00; // End of CI list - pad[padlen-5]=(mscdgsize & 0x3F00)>>8; // RFA+HI Data group length - pad[padlen-6]=(mscdgsize & 0x00FF); - crc = update_crc_ccitt(crc, pad[padlen-5]); - crc = update_crc_ccitt(crc, pad[padlen-6]); - crc = ~crc; - pad[padlen-7]=(crc & 0xFF00) >> 8; // HI CRC - pad[padlen-8]=(crc & 0x00FF); // LO CRC - for (i=padlen-9;i>=0;i--) { // NULL PADDING - pad[i]=0x00; - } - write(STDOUT_FILENO,pad,padlen); - //fprintf(stderr,"Data Group Length Indicator: "); - //for (i=0;i 0) { - numseg++; // The last incomplete segment - } - - for (i=0;i data length = 12 (011) - Application Type=12 (start of MOT) - else - pad[padlen-3]=(xpadlengthmask<<5) | 13; // CI => data length = 12 (011) - Application Type=13 (MOT) - - pad[padlen-4]=0x00; // End of CI list - - for (j=0; j=0;j--) - pad[j]=0x00; - - write(STDOUT_FILENO,pad,padlen); - //fprintf(stderr,"MSC Data Group - Segment %d: ",i); - //for (j=0;j= 16) { - if (lastseglen > 0) numseg++; // The last incomplete segment - numdg=numseg; // The PAD can contain the full segmnet and overhead (9 bytes) - } else { - numdg=numseg*2; // Each 16 char segment span over 2 dg - if (lastseglen > 0) { - numseg++; // The last incomplete segment - if (lastseglen <= padlen-9) - numdg+=1; - else - numdg+=2; - - } - } - *p_numdg=numdg; - fprintf(stderr,"PAD Length: %d\n",padlen); - fprintf(stderr,"DLS text: %s\n",text); - fprintf(stderr,"Number od DLS segments: %d\n",numseg); - fprintf(stderr,"Number od DLS data grupus: %d\n",numdg); - - if (padlen==17) - xpadlengthmask=3; - else if (padlen==21) - xpadlengthmask=4; - else if (padlen==29) - xpadlengthmask=5; - else if (padlen==37) - xpadlengthmask=6; - else if (padlen==53) - xpadlengthmask=7; - - *p_dlsdg = (UCHAR**) malloc(numdg*sizeof(UCHAR*)); - dlsdg=*p_dlsdg; - - i=0; - for (z=0;z data length = 12 (011) - Application Type=2 (DLS - start of X-PAD data group) - dlsdg[i][padlen-4]=0x00; // End of CI list - - dlsdg[i][padlen-5]=((toggle*8+firstseg*4+lastseg*2+0)<<4) | (curseglen-1); // DLS Prefix (T=1,Only one segment,segment length-1) - - if (firstseg==1) - dlsdg[i][padlen-6]=0x00; // DLS Prefix (Charset standard) - else - dlsdg[i][padlen-6]=z<<4; // DLS SegNum - - idx_start_crc = padlen-5; // CRC start from prefix - for (j=0;j=idx_stop_crc;j--) { - dlscrc = update_crc_ccitt(dlscrc, dlsdg[i][j]); - } - - dlscrc = ~dlscrc; - fprintf(stderr,"crc=%x ~crc=%x\n",~dlscrc,dlscrc); - - dlsdg[i][padlen-7-curseglen] = (dlscrc & 0xFF00) >> 8; // HI CRC - dlsdg[i][padlen-7-curseglen-1] = dlscrc & 0x00FF; // LO CRC - - for (j=padlen-7-curseglen-2;j>=0;j--) { // NULL PADDING - dlsdg[i][j]=0x00; - } - fprintf(stderr,"Data group: "); - for (j=0;j data length = 12 (011) - Application Type=2 (DLS - start of X-PAD data group) - dlsdg[i][padlen-4]=0x00; // End of CI list - dlsdg[i][padlen-5]=((toggle*8+firstseg*4+lastseg*2+0)<<4) | (curseglen-1); // DLS Prefix (T=1,Only one segment,segment length-1) - - if (firstseg==1) - dlsdg[i][padlen-6]=0x00; // DLS Prefix (Charset standard) - else - dlsdg[i][padlen-6]=(i-1)<<4; // DLS SegNum - - dlscrc = update_crc_ccitt(dlscrc, dlsdg[i][padlen-5]); - dlscrc = update_crc_ccitt(dlscrc, dlsdg[i][padlen-6]); - - for (j=0;j < MIN(curseglen,padlen-7); j++) { // DLS text - dlsdg[i][padlen-7-j]=curseg[j]; - dlscrc = update_crc_ccitt(dlscrc, dlsdg[i][padlen-7-j]); - } - k=j; - if (curseglen==padlen-8) { // end of segment - dlscrc = ~dlscrc; - dlsdg[i][1] = (dlscrc & 0xFF00) >> 8; // HI CRC - fprintf(stderr,"crc=%x ~crc=%x\n",~dlscrc,dlscrc); - } else if (curseglen==padlen-7) { - dlscrc = ~dlscrc; - fprintf(stderr,"crc=%x ~crc=%x\n",~dlscrc,dlscrc); - } - dlsdg[i][0]=0x00; - - fprintf(stderr,"First Data group: "); - for (j=0;j> 8; // HI CRC - dlsdg[i][padlen-4] = dlscrc & 0x00FF; // LO CRC - } else { - for (j=0;j < curseglen-k; j++) { // DLS text - dlsdg[i][padlen-3-j]=curseg[k+j]; - dlscrc = update_crc_ccitt(dlscrc, dlsdg[i][padlen-3-j]); - } - dlscrc = ~dlscrc; - dlsdg[i][padlen-3-curseglen+k]=(dlscrc & 0xFF00) >> 8; // HI CRC - dlsdg[i][padlen-3-curseglen+k-1]=dlscrc & 0x00FF; // LO CRC - } - - fprintf(stderr,"Second Data group: "); - for (j=0;j. + * ------------------------------------------------------------------- + */ + +#include +#include +#include +#include +#include +#include +#include +#include +#include "libAACenc/include/aacenc_lib.h" +#include "wavreader.h" + +#include +#include +#include +#include +#include + +#include "contrib/lib_crc.h" + +#define MIN(a,b) (((a)<(b))?(a):(b)) +#define MAX(a,b) (((a)>(b))?(a):(b)) + +void usage(const char* name) { + fprintf(stderr, "fdk-aac-dabplus HE-AACv2 encoder for DAB+\n\n"); + fprintf(stderr, "With ZeroMQ output for ODR-DabMux\n"); + fprintf(stderr, "and PAD (DLS and MOT Slideshow) by http://rd.csp.it\n"); + fprintf(stderr, "\n"); + fprintf(stderr, "http://opendigitalradio.org\n"); + fprintf(stderr, "\nUsage:\n"); + fprintf(stderr, "%s [OPTION...]\n", name); + fprintf(stderr, +" -b, --bitrate={ 8, 16, ..., 192 } Output bitrate in kbps. Must be 8 multiple.\n" +//" -d, --data=FILENAME Set data filename.\n" +//" -g, --fs-bug Turn on FS bug mitigation.\n" +" -i, --input=FILENAME Input filename (default: stdin).\n" +" -o, --output=URI Output zmq uri. (e.g. 'tcp://*:9000')\n" +" -a, --afterburner Turn on AAC encoder quality increaser.\n" +//" -m, --message Turn on AAC frame messages.\n" +" -p, --pad=BYTES Set PAD size in bytes.\n" +" -f, --format={ wav, raw } Set input file format (default: wav).\n" +" -c, --channels={ 1, 2 } Nb of input channels for raw input (default: 2).\n" +" -r, --rate={ 32000, 48000 } Sample rate for raw input (default: 48000).\n" +//" -t, --type=TYPE Set data type (dls|pad|packet|dg).\n" +//" -v, --verbose=LEVEL Set verbosity level.\n" +//" -V, --version Print version and exit.\n" +//" --mi=[ 0, ... ] Set AAC frame messages interval in milliseconds.\n" +//" --ma=[ 0, ... ] Set AAC frame messages attack time in milliseconds.\n" +//" -l, --lp Set frame size to 1024 instead of 960.\n" +"\n" +"Only the tcp:// zeromq transport has been tested until now.\n" + +); + +} + + +#define no_argument 0 +#define required_argument 1 +#define optional_argument 2 + +int main(int argc, char *argv[]) { + int subchannel_index = 8; //64kbps subchannel + int ch=0; + const char *infile = NULL; + const char *outuri = NULL; + FILE *in_fh; + void *wav; + int wav_format, bits_per_sample, sample_rate=48000, channels=2; + uint8_t* input_buf; + int16_t* convert_buf; + void *rs_handler = NULL; + int aot = AOT_DABPLUS_AAC_LC; + int afterburner = 0, raw_input=0; + HANDLE_AACENCODER handle; + CHANNEL_MODE mode; + AACENC_InfoStruct info = { 0 }; + + int pad_fd; + unsigned char pad_buf[128]; + int padlen; + + void *zmq_context = zmq_ctx_new(); + void *zmq_sock = NULL; + + const struct option longopts[] = { + {"bitrate", required_argument, 0, 'b'}, + {"input", required_argument, 0, 'i'}, + {"output", required_argument, 0, 'o'}, + {"format", required_argument, 0, 'f'}, + {"rate", required_argument, 0, 'r'}, + {"channels", required_argument, 0, 'c'}, + {"pad", required_argument, 0, 'p'}, + //{"lp", no_argument, 0, 'l'}, + {"afterburner", no_argument, 0, 'a'}, + {"help", no_argument, 0, 'h'}, + {0,0,0,0}, + }; + + if (argc == 1) { + usage(argv[0]); + return 0; + } + + int index; + while(ch != -1) { + ch = getopt_long(argc, argv, "tlhab:c:i:o:r:f:p:", longopts, &index); + switch (ch) { + case 'f': + if(strcmp(optarg, "raw")==0) { + raw_input = 1; + } else if(strcmp(optarg, "wav")!=0) + usage(argv[0]); + break; + case 'a': + afterburner = 1; + break; + case 'b': + subchannel_index = atoi(optarg) / 8; + break; + case 'c': + channels = atoi(optarg); + break; + case 'r': + sample_rate = atoi(optarg); + break; + case 'i': + infile = optarg; + break; + case 'o': + outuri = optarg; + break; + case 'p': + padlen = atoi(optarg); + break; + case '?': + case 'h': + usage(argv[0]); + return 1; + } + } + + if(subchannel_index < 1 || subchannel_index > 24) { + fprintf(stderr, "Bad subchannels number: %d, try other bitrate.\n", + subchannel_index); + return 1; + } + if(padlen != 0) { + int flags; + if (mkfifo("/tmp/pad.fifo", S_IWUSR | S_IRUSR | S_IRGRP | S_IROTH) != 0) { + if (errno != EEXIST) { + fprintf(stderr, "Can't create pad file: %d!\n", errno); + return 1; + } + } + pad_fd = open("/tmp/pad.fifo", O_RDONLY | O_NONBLOCK); + if (pad_fd == -1) { + fprintf(stderr, "Can't open pad file!\n"); + return 1; + } + flags = fcntl(pad_fd, F_GETFL, 0); + if (fcntl(pad_fd, F_SETFL, flags | O_NONBLOCK)) { + fprintf(stderr, "Can't set non-blocking mode in pad file!\n"); + return 1; + } + } + + if(raw_input) { + if(infile && strcmp(infile, "-")) { + in_fh = fopen(infile, "rb"); + if(!in_fh) { + fprintf(stderr, "Can't open input file!\n"); + return 1; + } + } else { + in_fh = stdin; + } + } else { + wav = wav_read_open(infile); + if (!wav) { + fprintf(stderr, "Unable to open wav file %s\n", infile); + return 1; + } + if (!wav_get_header(wav, &wav_format, &channels, &sample_rate, &bits_per_sample, NULL)) { + fprintf(stderr, "Bad wav file %s\n", infile); + return 1; + } + if (wav_format != 1) { + fprintf(stderr, "Unsupported WAV format %d\n", wav_format); + return 1; + } + if (bits_per_sample != 16) { + fprintf(stderr, "Unsupported WAV sample depth %d\n", bits_per_sample); + return 1; + } + if (channels > 2) { + fprintf(stderr, "Unsupported WAV channels %d\n", channels); + return 1; + } + } + + if (outuri) { + zmq_sock = zmq_socket(zmq_context, ZMQ_PUB); + if (zmq_sock == NULL) { + fprintf(stderr, "Error occurred during zmq_socket: %s\n", zmq_strerror(errno)); + return 2; + } + if (zmq_connect(zmq_sock, outuri) != 0) { + fprintf(stderr, "Error occurred during zmq_connect: %s\n", zmq_strerror(errno)); + return 2; + } + } else { + fprintf(stderr, "Output URI not defined\n"); + return 1; + } + + + switch (channels) { + case 1: mode = MODE_1; break; + case 2: mode = MODE_2; break; + default: + fprintf(stderr, "Unsupported channels number %d\n", channels); + return 1; + } + + + if (aacEncOpen(&handle, 0x01|0x02|0x04, channels) != AACENC_OK) { + fprintf(stderr, "Unable to open encoder\n"); + return 1; + } + + + if(channels == 2 && subchannel_index <= 6) + aot = AOT_DABPLUS_PS; + else if((channels == 1 && subchannel_index <= 8) || subchannel_index <= 10) + aot = AOT_DABPLUS_SBR; + + fprintf(stderr, "Using %d subchannels. AAC type: %s%s%s. channels=%d, sample_rate=%d\n", + subchannel_index, + aot == AOT_DABPLUS_PS ? "HE-AAC v2" : "", + aot == AOT_DABPLUS_SBR ? "HE-AAC" : "", + aot == AOT_DABPLUS_AAC_LC ? "AAC-LC" : "", + channels, sample_rate); + + if (aacEncoder_SetParam(handle, AACENC_AOT, aot) != AACENC_OK) { + fprintf(stderr, "Unable to set the AOT\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_SAMPLERATE, sample_rate) != AACENC_OK) { + fprintf(stderr, "Unable to set the samplerate\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_CHANNELMODE, mode) != AACENC_OK) { + fprintf(stderr, "Unable to set the channel mode\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_CHANNELORDER, 1) != AACENC_OK) { + fprintf(stderr, "Unable to set the wav channel order\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_GRANULE_LENGTH, 960) != AACENC_OK) { + fprintf(stderr, "Unable to set the granule length\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_TRANSMUX, TT_DABPLUS) != AACENC_OK) { + fprintf(stderr, "Unable to set the RAW transmux\n"); + return 1; + } + + /*if (aacEncoder_SetParam(handle, AACENC_BITRATEMODE, 7 *AACENC_BR_MODE_SFR*) != AACENC_OK) { + fprintf(stderr, "Unable to set the bitrate mode\n"); + return 1; + }*/ + + + fprintf(stderr, "AAC bitrate set to: %d\n", subchannel_index*8000); + if (aacEncoder_SetParam(handle, AACENC_BITRATE, subchannel_index*8000) != AACENC_OK) { + fprintf(stderr, "Unable to set the bitrate\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_AFTERBURNER, afterburner) != AACENC_OK) { + fprintf(stderr, "Unable to set the afterburner mode\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_ANCILLARY_BITRATE, 0) != AACENC_OK) { + fprintf(stderr, "Unable to set the ancillary bitrate\n"); + return 1; + } + if (aacEncEncode(handle, NULL, NULL, NULL, NULL) != AACENC_OK) { + fprintf(stderr, "Unable to initialize the encoder\n"); + return 1; + } + if (aacEncInfo(handle, &info) != AACENC_OK) { + fprintf(stderr, "Unable to get the encoder info\n"); + return 1; + } + + fprintf(stderr, "DAB+ Encoding: framelen=%d\n", info.frameLength); + + int input_size = channels*2*info.frameLength; + input_buf = (uint8_t*) malloc(input_size); + convert_buf = (int16_t*) malloc(input_size); + + /* symsize=8, gfpoly=0x11d, fcr=0, prim=1, nroots=10, pad=135 */ + rs_handler = init_rs_char(8, 0x11d, 0, 1, 10, 135); + if (rs_handler == NULL) { + perror("init_rs_char failed"); + return 0; + } + + int loops = 0; + int outbuf_size = subchannel_index*120; + uint8_t outbuf[20480]; + + if(outbuf_size % 5 != 0) { + fprintf(stderr, "(outbuf_size mod 5) = %d\n", outbuf_size % 5); + } + + fprintf(stderr, "outbuf_size: %d\n", outbuf_size); + //outbuf_size += (4 * subchannel_index * (8*8)/8) - outbuf_size/5; + fprintf(stderr, "outbuf_size: %d\n", outbuf_size); + + int frame=0; + int send_error_count = 0; + while (1) { + memset(outbuf, 0x00, outbuf_size); + + AACENC_BufDesc in_buf = { 0 }, out_buf = { 0 }; + AACENC_InArgs in_args = { 0 }; + AACENC_OutArgs out_args = { 0 }; + int in_identifier[] = {IN_AUDIO_DATA, IN_ANCILLRY_DATA}; + int in_size[2], in_elem_size[2]; + int out_identifier = OUT_BITSTREAM_DATA; + int out_size, out_elem_size; + int pcmread=0, i, ret; + int send_error; + void *in_ptr[2], *out_ptr; + AACENC_ERROR err; + + // Read data from the PAD fifo + if (padlen != 0) { + ret = read(pad_fd, pad_buf, padlen); + } + else { + ret = 0; + } + + + if(ret < 0 && errno == EAGAIN) { + // If this condition passes, there is no data to be read + in_buf.numBufs = 1; // Samples; + } + else if(ret >= 0) { + // Otherwise, you're good to go and buffer should contain "count" bytes. + in_buf.numBufs = 2; // Samples + Data; + } + else { + // Some other error occurred during read. + fprintf(stderr, "Unable to read from PAD!\n"); + break; + } + + if(raw_input) { + if(fread(input_buf, input_size, 1, in_fh) == 1) { + pcmread = input_size; + } else { + fprintf(stderr, "Unable to read from input!\n"); + break; + } + } else { + pcmread = wav_read_data(wav, input_buf, input_size); + } + + for (i = 0; i < pcmread/2; i++) { + const uint8_t* in = &input_buf[2*i]; + convert_buf[i] = in[0] | (in[1] << 8); + } + + if (pcmread <= 0) { + in_args.numInSamples = -1; + } else { + in_ptr[0] = convert_buf; + in_ptr[1] = pad_buf; + in_size[0] = pcmread; + in_size[1] = padlen; + + in_elem_size[0] = 2; + in_elem_size[1] = sizeof(UCHAR); + + in_args.numInSamples = pcmread/2; + in_args.numAncBytes = padlen; + + //in_buf.numBufs = 2; // Samples + Data + + in_buf.bufs = (void**)&in_ptr; + in_buf.bufferIdentifiers = in_identifier; + in_buf.bufSizes = in_size; + in_buf.bufElSizes = in_elem_size; + + } + out_ptr = outbuf; + out_size = sizeof(outbuf); + out_elem_size = 1; + out_buf.numBufs = 1; + out_buf.bufs = &out_ptr; + out_buf.bufferIdentifiers = &out_identifier; + out_buf.bufSizes = &out_size; + out_buf.bufElSizes = &out_elem_size; + + if ((err = aacEncEncode(handle, &in_buf, &out_buf, &in_args, &out_args)) != AACENC_OK) { + if (err == AACENC_ENCODE_EOF) + break; + fprintf(stderr, "Encoding failed\n"); + return 1; + } + if (out_args.numOutBytes == 0) + continue; +#if 0 + unsigned char au_start[6]; + unsigned char* sfbuf = outbuf; + au_start[0] = 6; + au_start[1] = (*(sfbuf + 3) << 4) + ((*(sfbuf + 4)) >> 4); + au_start[2] = ((*(sfbuf + 4) & 0x0f) << 8) + *(sfbuf + 5); + fprintf (stderr, "au_start[0] = %d\n", au_start[0]); + fprintf (stderr, "au_start[1] = %d\n", au_start[1]); + fprintf (stderr, "au_start[2] = %d\n", au_start[2]); +#endif + + int row, col; + unsigned char buf_to_rs_enc[110]; + unsigned char rs_enc[10]; + for(row=0; row < subchannel_index; row++) { + for(col=0;col < 110; col++) { + buf_to_rs_enc[col] = outbuf[subchannel_index * col + row]; + } + + encode_rs_char(rs_handler, buf_to_rs_enc, rs_enc); + + for(col=110; col<120; col++) { + outbuf[subchannel_index * col + row] = rs_enc[col-110]; + assert(subchannel_index * col + row < outbuf_size); + } + } + + send_error = zmq_send(zmq_sock, outbuf, outbuf_size, ZMQ_DONTWAIT); + if (send_error < 0) { + fprintf(stderr, "ZeroMQ send failed! %s\n", zmq_strerror(errno)); + send_error_count ++; + } + + if (send_error_count > 10) + { + fprintf(stderr, "ZeroMQ send failed ten times, aborting!\n"); + break; + } + //fwrite(outbuf, 1, /*out_args.numOutBytes*/ outbuf_size, out_fh); + //fprintf(stderr, "Written %d/%d bytes!\n", out_args.numOutBytes + row*10, outbuf_size); + if(out_args.numOutBytes + row*10 == outbuf_size) + fprintf(stderr, "."); + +// if(frame > 10) +// break; + frame++; + } + free(input_buf); + free(convert_buf); + if(raw_input) { + fclose(in_fh); + } else { + wav_read_close(wav); + } + zmq_close(zmq_sock); + free_rs_char(rs_handler); + + aacEncClose(&handle); + + zmq_ctx_term(zmq_context); + return 0; +} + diff --git a/src/aac-enc-dabplus.c b/src/aac-enc-dabplus.c new file mode 100644 index 0000000..6a0b882 --- /dev/null +++ b/src/aac-enc-dabplus.c @@ -0,0 +1,425 @@ +/* ------------------------------------------------------------------ + * Copyright (C) 2011 Martin Storsjo + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either + * express or implied. + * See the License for the specific language governing permissions + * and limitations under the License. + * ------------------------------------------------------------------- + */ + +#include +#include +#include +#include +#include +#include +#include +#include "libAACenc/include/aacenc_lib.h" +#include "wavreader.h" + +#include + +void usage(const char* name) { + fprintf(stderr, "%s [OPTION...]\n", name); + fprintf(stderr, +" -b, --bitrate={ 8, 16, ..., 192 } Output bitrate in kbps. Must be 8 multiple.\n" +//" -d, --data=FILENAME Set data filename.\n" +//" -g, --fs-bug Turn on FS bug mitigation.\n" +" -i, --input=FILENAME Input filename (default: stdin).\n" +" -o, --output=FILENAME Output filename (default: stdout).\n" +" -a, --afterburner Turn on AAC encoder quality increaser.\n" +//" -m, --message Turn on AAC frame messages.\n" +//" -p, --pad=BYTES Set PAD size in bytes.\n" +" -f, --format={ wav, raw } Set input file format (default: wav).\n" +" -c, --channels={ 1, 2 } Nb of input channels for raw input (default: 2).\n" +" -r, --rate={ 32000, 48000 } Sample rate for raw input (default: 48000).\n" +//" -t, --type=TYPE Set data type (dls|pad|packet|dg).\n" +//" -v, --verbose=LEVEL Set verbosity level.\n" +//" -V, --version Print version and exit.\n" +//" --mi=[ 0, ... ] Set AAC frame messages interval in milliseconds.\n" +//" --ma=[ 0, ... ] Set AAC frame messages attack time in milliseconds.\n" +//" -t, --adts Set ADTS output format (for debugging).\n" +//" -l, --lp Set frame size to 1024 instead of 960.\n" + +); + +} + + +#define no_argument 0 +#define required_argument 1 +#define optional_argument 2 + +#define ADTS_HEADER_SIZE 7 +#define ADTS_MPEG_ID 1 /* 0: MPEG-4, 1: MPEG-2 */ +#define ADTS_MPEG_PROFILE 1 +const int mpeg4audio_sample_rates[16] = { + 96000, 88200, 64000, 48000, 44100, 32000, + 24000, 22050, 16000, 12000, 11025, 8000, 7350 +}; + +int FindSRIndex(int sr) +{ + int i; + for (i = 0; i < 16; i++) { + if (sr == mpeg4audio_sample_rates[i]) + return i; + } + return 16 - 1; +} + +void adts_hdr_up(char *buff, int size) +{ + unsigned short len = size + ADTS_HEADER_SIZE; + unsigned short buffer_fullness = 0x07FF; + + /* frame length, 13 bits */ + buff[3] &= 0xFC; + buff[3] |= ((len >> 11) & 0x03); /* 2b: aac_frame_length */ + buff[4] = len >> 3; /* 8b: aac_frame_length */ + buff[5] = (len << 5) & 0xE0; /* 3b: aac_frame_length */ + /* buffer fullness, 11 bits */ + buff[5] |= ((buffer_fullness >> 6) & 0x1F); /* 5b: adts_buffer_fullness */ + buff[6] = (buffer_fullness << 2) & 0xFC; /* 6b: adts_buffer_fullness */ + /* 2b: num_raw_data_blocks */ +} + +int main(int argc, char *argv[]) { + int subchannel_index = 8; //64kbps subchannel + int ch=0; + const char *infile, *outfile; + FILE *in_fh, *out_fh; + void *wav; + int wav_format, bits_per_sample, sample_rate=48000, channels=2; + uint8_t* input_buf; + int16_t* convert_buf; + void *rs_handler = NULL; + int aot = AOT_DABPLUS_AAC_LC; + int afterburner = 0, raw_input=0; + HANDLE_AACENCODER handle; + CHANNEL_MODE mode; + AACENC_InfoStruct info = { 0 }; + + const struct option longopts[] = { + {"bitrate", required_argument, 0, 'b'}, + {"input", required_argument, 0, 'i'}, + {"output", required_argument, 0, 'o'}, + {"format", required_argument, 0, 'f'}, + {"rate", required_argument, 0, 'r'}, + {"channels", required_argument, 0, 'c'}, + //{"lp", no_argument, 0, 'l'}, + //{"adts", no_argument, 0, 't'}, + {"afterburner", no_argument, 0, 'a'}, + {"help", no_argument, 0, 'h'}, + {0,0,0,0}, + }; + + int index; + while(ch != -1) { + ch = getopt_long(argc, argv, "tlhab:c:i:o:r:f:", longopts, &index); + switch (ch) { + case 'f': + if(strcmp(optarg, "raw")==0) { + raw_input = 1; + } else if(strcmp(optarg, "wav")!=0) + usage(argv[0]); + break; + case 'a': + afterburner = 1; + break; + case 'b': + subchannel_index = atoi(optarg) / 8; + break; + case 'c': + channels = atoi(optarg); + break; + case 'r': + sample_rate = atoi(optarg); + break; + case 'i': + infile = optarg; + break; + case 'o': + outfile = optarg; + break; + case '?': + case 'h': + usage(argv[0]); + return 1; + } + } + + if(subchannel_index < 1 || subchannel_index > 24) { + fprintf(stderr, "Bad subchannels number: %d, try other bitrate.\n", subchannel_index); + return 1; + } + + if(raw_input) { + if(infile && strcmp(infile, "-")) { + in_fh = fopen(infile, "rb"); + if(!in_fh) { + fprintf(stderr, "Can't open input file!\n"); + return 1; + } + } else { + in_fh = stdin; + } + } else { + wav = wav_read_open(infile); + if (!wav) { + fprintf(stderr, "Unable to open wav file %s\n", infile); + return 1; + } + if (!wav_get_header(wav, &wav_format, &channels, &sample_rate, &bits_per_sample, NULL)) { + fprintf(stderr, "Bad wav file %s\n", infile); + return 1; + } + if (wav_format != 1) { + fprintf(stderr, "Unsupported WAV format %d\n", wav_format); + return 1; + } + if (bits_per_sample != 16) { + fprintf(stderr, "Unsupported WAV sample depth %d\n", bits_per_sample); + return 1; + } + if (channels > 2) { + fprintf(stderr, "Unsupported WAV channels %d\n", channels); + return 1; + } + } + + if(outfile && strcmp(outfile, "-")) { + out_fh = fopen(outfile, "wb"); + if(!out_fh) { + fprintf(stderr, "Can't open output file!\n"); + return 1; + } + } else { + out_fh = stdout; + } + + + switch (channels) { + case 1: mode = MODE_1; break; + case 2: mode = MODE_2; break; + default: + fprintf(stderr, "Unsupported channels number %d\n", channels); + return 1; + } + + + if (aacEncOpen(&handle, 0x01|0x02|0x04, channels) != AACENC_OK) { + fprintf(stderr, "Unable to open encoder\n"); + return 1; + } + + + if(channels == 2 && subchannel_index <= 6) + aot = AOT_DABPLUS_PS; + else if((channels == 1 && subchannel_index <= 8) || subchannel_index <= 10) + aot = AOT_DABPLUS_SBR; + + fprintf(stderr, "Using %d subchannels. AAC type: %s%s%s. channels=%d, sample_rate=%d\n", + subchannel_index, + aot == AOT_DABPLUS_PS ? "HE-AAC v2" : "", + aot == AOT_DABPLUS_SBR ? "HE-AAC" : "", + aot == AOT_DABPLUS_AAC_LC ? "AAC-LC" : "", + channels, sample_rate); + + if (aacEncoder_SetParam(handle, AACENC_AOT, aot) != AACENC_OK) { + fprintf(stderr, "Unable to set the AOT\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_SAMPLERATE, sample_rate) != AACENC_OK) { + fprintf(stderr, "Unable to set the AOT\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_CHANNELMODE, mode) != AACENC_OK) { + fprintf(stderr, "Unable to set the channel mode\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_CHANNELORDER, 1) != AACENC_OK) { + fprintf(stderr, "Unable to set the wav channel order\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_GRANULE_LENGTH, 960) != AACENC_OK) { + fprintf(stderr, "Unable to set the AOT\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_TRANSMUX, TT_DABPLUS) != AACENC_OK) { + fprintf(stderr, "Unable to set the RAW transmux\n"); + return 1; + } + + /*if (aacEncoder_SetParam(handle, AACENC_BITRATEMODE, 7 *AACENC_BR_MODE_SFR*) != AACENC_OK) { + fprintf(stderr, "Unable to set the bitrate mode\n"); + return 1; + }*/ + + + fprintf(stderr, "AAC bitrate set to: %d\n", subchannel_index*8000); + if (aacEncoder_SetParam(handle, AACENC_BITRATE, subchannel_index*8000) != AACENC_OK) { + fprintf(stderr, "Unable to set the bitrate\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_AFTERBURNER, afterburner) != AACENC_OK) { + fprintf(stderr, "Unable to set the afterburner mode\n"); + return 1; + } + if (aacEncEncode(handle, NULL, NULL, NULL, NULL) != AACENC_OK) { + fprintf(stderr, "Unable to initialize the encoder\n"); + return 1; + } + if (aacEncInfo(handle, &info) != AACENC_OK) { + fprintf(stderr, "Unable to get the encoder info\n"); + return 1; + } + + fprintf(stderr, "DAB+ Encoding: framelen=%d\n", info.frameLength); + + int input_size = channels*2*info.frameLength; + input_buf = (uint8_t*) malloc(input_size); + convert_buf = (int16_t*) malloc(input_size); + + /* symsize=8, gfpoly=0x11d, fcr=0, prim=1, nroots=10, pad=135 */ + rs_handler = init_rs_char(8, 0x11d, 0, 1, 10, 135); + if (rs_handler == NULL) { + perror("init_rs_char failed"); + return 0; + } + + int loops = 0; + int outbuf_size = subchannel_index*120; + uint8_t outbuf[20480]; + + if(outbuf_size % 5 != 0) { + fprintf(stderr, "(outbuf_size mod 5) = %d\n", outbuf_size % 5); + } + + fprintf(stderr, "outbuf_size: %d\n", outbuf_size); + //outbuf_size += (4 * subchannel_index * (8*8)/8) - outbuf_size/5; + fprintf(stderr, "outbuf_size: %d\n", outbuf_size); + + int frame=0; + while (1) { + memset(outbuf, 0x00, outbuf_size); + + AACENC_BufDesc in_buf = { 0 }, out_buf = { 0 }; + AACENC_InArgs in_args = { 0 }; + AACENC_OutArgs out_args = { 0 }; + int in_identifier = IN_AUDIO_DATA; + int in_size, in_elem_size; + int out_identifier = OUT_BITSTREAM_DATA; + int out_size, out_elem_size; + int read=0, i; + void *in_ptr, *out_ptr; + AACENC_ERROR err; + + if(raw_input) { + if(fread(input_buf, input_size, 1, in_fh) == 1) { + read = input_size; + } else { + fprintf(stderr, "Unable to read from input!\n"); + break; + } + } else { + read = wav_read_data(wav, input_buf, input_size); + // returns bytes read + } + + for (i = 0; i < read/2; i++) { + const uint8_t* in = &input_buf[2*i]; + convert_buf[i] = in[0] | (in[1] << 8); + } + + if (read <= 0) { + in_args.numInSamples = -1; + } else { + in_ptr = convert_buf; + in_size = read; + in_elem_size = 2; + + in_args.numInSamples = read/2; + in_buf.numBufs = 1; + in_buf.bufs = &in_ptr; + in_buf.bufferIdentifiers = &in_identifier; + in_buf.bufSizes = &in_size; + in_buf.bufElSizes = &in_elem_size; + } + out_ptr = outbuf; + out_size = sizeof(outbuf); + out_elem_size = 1; + out_buf.numBufs = 1; + out_buf.bufs = &out_ptr; + out_buf.bufferIdentifiers = &out_identifier; + out_buf.bufSizes = &out_size; + out_buf.bufElSizes = &out_elem_size; + + if ((err = aacEncEncode(handle, &in_buf, &out_buf, &in_args, &out_args)) != AACENC_OK) { + if (err == AACENC_ENCODE_EOF) + break; + fprintf(stderr, "Encoding failed\n"); + return 1; + } + if (out_args.numOutBytes == 0) + continue; +#if 0 + unsigned char au_start[6]; + unsigned char* sfbuf = outbuf; + au_start[0] = 6; + au_start[1] = (*(sfbuf + 3) << 4) + ((*(sfbuf + 4)) >> 4); + au_start[2] = ((*(sfbuf + 4) & 0x0f) << 8) + *(sfbuf + 5); + fprintf (stderr, "au_start[0] = %d\n", au_start[0]); + fprintf (stderr, "au_start[1] = %d\n", au_start[1]); + fprintf (stderr, "au_start[2] = %d\n", au_start[2]); +#endif + + int row, col; + char buf_to_rs_enc[110]; + char rs_enc[10]; + for(row=0; row < subchannel_index; row++) { + for(col=0;col < 110; col++) { + buf_to_rs_enc[col] = outbuf[subchannel_index * col + row]; + } + + encode_rs_char(rs_handler, buf_to_rs_enc, rs_enc); + + for(col=110; col<120; col++) { + outbuf[subchannel_index * col + row] = rs_enc[col-110]; + assert(subchannel_index * col + row < outbuf_size); + } + } + + fwrite(outbuf, 1, /*out_args.numOutBytes*/ outbuf_size, out_fh); + //fprintf(stderr, "Written %d/%d bytes!\n", out_args.numOutBytes + row*10, outbuf_size); + if(out_args.numOutBytes + row*10 == outbuf_size) + fprintf(stderr, "."); + +// if(frame > 10) +// break; + frame++; + } + free(input_buf); + free(convert_buf); + if(raw_input) { + fclose(in_fh); + } else { + wav_read_close(wav); + } + fclose(out_fh); + free_rs_char(rs_handler); + + aacEncClose(&handle); + + return 0; +} + diff --git a/src/aac-enc.c b/src/aac-enc.c new file mode 100644 index 0000000..1528881 --- /dev/null +++ b/src/aac-enc.c @@ -0,0 +1,234 @@ +/* ------------------------------------------------------------------ + * Copyright (C) 2011 Martin Storsjo + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either + * express or implied. + * See the License for the specific language governing permissions + * and limitations under the License. + * ------------------------------------------------------------------- + */ + +#include +#include +#include +#include +#include "libAACenc/include/aacenc_lib.h" +#include "wavreader.h" + +void usage(const char* name) { + fprintf(stderr, "%s [-r bitrate] [-t aot] [-a afterburner] [-s sbr] [-v vbr] in.wav out.aac\n", name); + fprintf(stderr, "Supported AOTs:\n"); + fprintf(stderr, "\t2\tAAC-LC\n"); + fprintf(stderr, "\t5\tHE-AAC\n"); + fprintf(stderr, "\t29\tHE-AAC v2\n"); + fprintf(stderr, "\t23\tAAC-LD\n"); + fprintf(stderr, "\t39\tAAC-ELD\n"); +} + +int main(int argc, char *argv[]) { + int bitrate = 64000; + int ch; + const char *infile, *outfile; + FILE *out; + void *wav; + int format, sample_rate, channels, bits_per_sample; + int input_size; + uint8_t* input_buf; + int16_t* convert_buf; + int aot = 2; + int afterburner = 1; + int eld_sbr = 0; + int vbr = 0; + HANDLE_AACENCODER handle; + CHANNEL_MODE mode; + AACENC_InfoStruct info = { 0 }; + while ((ch = getopt(argc, argv, "r:t:a:s:v:")) != -1) { + switch (ch) { + case 'r': + bitrate = atoi(optarg); + break; + case 't': + aot = atoi(optarg); + break; + case 'a': + afterburner = atoi(optarg); + break; + case 's': + eld_sbr = atoi(optarg); + break; + case 'v': + vbr = atoi(optarg); + break; + case '?': + default: + usage(argv[0]); + return 1; + } + } + if (argc - optind < 2) { + usage(argv[0]); + return 1; + } + infile = argv[optind]; + outfile = argv[optind + 1]; + + wav = wav_read_open(infile); + if (!wav) { + fprintf(stderr, "Unable to open wav file %s\n", infile); + return 1; + } + if (!wav_get_header(wav, &format, &channels, &sample_rate, &bits_per_sample, NULL)) { + fprintf(stderr, "Bad wav file %s\n", infile); + return 1; + } + if (format != 1) { + fprintf(stderr, "Unsupported WAV format %d\n", format); + return 1; + } + if (bits_per_sample != 16) { + fprintf(stderr, "Unsupported WAV sample depth %d\n", bits_per_sample); + return 1; + } + switch (channels) { + case 1: mode = MODE_1; break; + case 2: mode = MODE_2; break; + case 3: mode = MODE_1_2; break; + case 4: mode = MODE_1_2_1; break; + case 5: mode = MODE_1_2_2; break; + case 6: mode = MODE_1_2_2_1; break; + default: + fprintf(stderr, "Unsupported WAV channels %d\n", channels); + return 1; + } + if (aacEncOpen(&handle, 0, channels) != AACENC_OK) { + fprintf(stderr, "Unable to open encoder\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_AOT, aot) != AACENC_OK) { + fprintf(stderr, "Unable to set the AOT\n"); + return 1; + } + if (aot == 39 && eld_sbr) { + if (aacEncoder_SetParam(handle, AACENC_SBR_MODE, 1) != AACENC_OK) { + fprintf(stderr, "Unable to set SBR mode for ELD\n"); + return 1; + } + } + if (aacEncoder_SetParam(handle, AACENC_SAMPLERATE, sample_rate) != AACENC_OK) { + fprintf(stderr, "Unable to set the AOT\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_CHANNELMODE, mode) != AACENC_OK) { + fprintf(stderr, "Unable to set the channel mode\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_CHANNELORDER, 1) != AACENC_OK) { + fprintf(stderr, "Unable to set the wav channel order\n"); + return 1; + } + if (vbr) { + if (aacEncoder_SetParam(handle, AACENC_BITRATEMODE, vbr) != AACENC_OK) { + fprintf(stderr, "Unable to set the VBR bitrate mode\n"); + return 1; + } + } else { + if (aacEncoder_SetParam(handle, AACENC_BITRATE, bitrate) != AACENC_OK) { + fprintf(stderr, "Unable to set the bitrate\n"); + return 1; + } + } + if (aacEncoder_SetParam(handle, AACENC_TRANSMUX, 2) != AACENC_OK) { + fprintf(stderr, "Unable to set the ADTS transmux\n"); + return 1; + } + if (aacEncoder_SetParam(handle, AACENC_AFTERBURNER, afterburner) != AACENC_OK) { + fprintf(stderr, "Unable to set the afterburner mode\n"); + return 1; + } + if (aacEncEncode(handle, NULL, NULL, NULL, NULL) != AACENC_OK) { + fprintf(stderr, "Unable to initialize the encoder\n"); + return 1; + } + if (aacEncInfo(handle, &info) != AACENC_OK) { + fprintf(stderr, "Unable to get the encoder info\n"); + return 1; + } + + out = fopen(outfile, "wb"); + if (!out) { + perror(outfile); + return 1; + } + + input_size = channels*2*info.frameLength; + input_buf = (uint8_t*) malloc(input_size); + convert_buf = (int16_t*) malloc(input_size); + + while (1) { + AACENC_BufDesc in_buf = { 0 }, out_buf = { 0 }; + AACENC_InArgs in_args = { 0 }; + AACENC_OutArgs out_args = { 0 }; + int in_identifier = IN_AUDIO_DATA; + int in_size, in_elem_size; + int out_identifier = OUT_BITSTREAM_DATA; + int out_size, out_elem_size; + int read, i; + void *in_ptr, *out_ptr; + uint8_t outbuf[20480]; + AACENC_ERROR err; + + read = wav_read_data(wav, input_buf, input_size); + for (i = 0; i < read/2; i++) { + const uint8_t* in = &input_buf[2*i]; + convert_buf[i] = in[0] | (in[1] << 8); + } + if (read <= 0) { + in_args.numInSamples = -1; + } else { + in_ptr = convert_buf; + in_size = read; + in_elem_size = 2; + + in_args.numInSamples = read/2; + in_buf.numBufs = 1; + in_buf.bufs = &in_ptr; + in_buf.bufferIdentifiers = &in_identifier; + in_buf.bufSizes = &in_size; + in_buf.bufElSizes = &in_elem_size; + } + out_ptr = outbuf; + out_size = sizeof(outbuf); + out_elem_size = 1; + out_buf.numBufs = 1; + out_buf.bufs = &out_ptr; + out_buf.bufferIdentifiers = &out_identifier; + out_buf.bufSizes = &out_size; + out_buf.bufElSizes = &out_elem_size; + + if ((err = aacEncEncode(handle, &in_buf, &out_buf, &in_args, &out_args)) != AACENC_OK) { + if (err == AACENC_ENCODE_EOF) + break; + fprintf(stderr, "Encoding failed\n"); + return 1; + } + if (out_args.numOutBytes == 0) + continue; + fwrite(outbuf, 1, out_args.numOutBytes, out); + } + free(input_buf); + free(convert_buf); + fclose(out); + wav_read_close(wav); + aacEncClose(&handle); + + return 0; +} + diff --git a/src/mot-encoder.c b/src/mot-encoder.c new file mode 100644 index 0000000..a60a39a --- /dev/null +++ b/src/mot-encoder.c @@ -0,0 +1,638 @@ +/* + Copyright (C) 2014 CSP Innovazione nelle ICT s.c.a r.l. (http://rd.csp.it/) + + 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 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. + + You should have received a copy of the GNU General Public License + along with this program. If not, see . + + MotEncoder.c + Generete PAD data for MOT Slideshow and DLS + + Authors: + Sergio Sagliocco +*/ + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "lib_crc.h" + +#define MIN(a,b) (((a)<(b))?(a):(b)) +#define MAX(a,b) (((a)>(b))?(a):(b)) + +#define MAXSEGLEN 8179 +#define MAXDLS 129 + +typedef unsigned char UCHAR; +typedef unsigned short int USHORT; + +typedef struct { + // MSC Data Group Header (extension field not supported) + unsigned char extflag; // 1 bit + unsigned char crcflag; // 1 bit + unsigned char segflag; // 1 bit + unsigned char accflag; // 1 bit + unsigned char dgtype; // 4 bits + unsigned char cindex; // 4 bits + unsigned char rindex; // 4 bits + /// Session header - Segment field + unsigned char last; // 1 bit + unsigned short int segnum; // 16 bits + // Session header - User access field + unsigned char rfa; // 3 bits + unsigned char tidflag; // 1 bit + unsigned char lenid; // 4 bits - Fixed to value 2 in this implemntation + unsigned short int tid; // 16 bits + // MSC data group data field + // Mot Segmentation header + unsigned char rcount; // 3 bits + unsigned short int seglen; // 13 bits + // Mot segment + unsigned char* segdata; + // MSC data group CRC + unsigned short int crc; // 16 bits +} MSCDG; +/* + typedef struct { +// MOT HEADER CUSTOMIZED FOR SLIDESHOW APP +unsigned int bodysize; // 28 bits +unsigned short int headsize; // 13 bits +unsigned char ctype; // 6 bits +unsigned char sctype; // 9 bits +unsigned char triggertime[5]; // 0x85 0x00 0x00 0x00 0x00 => NOW +unsigned char contname[14]; // 0xCC 0x0C 0x00 imgXXXX.jpg +} MOTSLIDEHDR; +*/ +int encodeFile (char* fname, int fidx, int padlen); +void createMotHeader(size_t blobsize, int fidx, char* mothdr, int* mothdrlen); +void createMscDG(MSCDG* msc, unsigned short int dgtype, unsigned short int cindex, unsigned short int lastseg, unsigned short int tid, unsigned char* data, unsigned short int datalen); +void packMscDG(unsigned char* mscblob, MSCDG* msc, unsigned short int *bsize); +void writeMotPAD(unsigned char* mscdg, unsigned short int mscdgsize, unsigned short int padlen); +void create_dls_datagroup (char* text, int padlen, unsigned char*** p_dlsdg, int* p_numdg); +void writeDLS(int padlen); + +void usage(char* name) +{ + fprintf(stderr, "DAB MOT encoder\n" + "for slideshow and DLS\n\n" + "By CSP Innovazione nelle ICT s.c.a r.l. (http://rd.csp.it/)\n\n" + "Reads image data from the specified directory, and outputs PAD data\n" + "on standard output\n" + "Reads DLS from /tmp/dls.file\n"); + fprintf(stderr, "Usage: %s \n", name); +} + +int main (int argc, char *argv[]) +{ + int len,fidx,ret; + struct dirent *pDirent; + DIR *pDir; + char imagepath[128]; + char dlstext[MAXDLS],dlstextprev[MAXDLS]; + int padlen=53; + + + if (argc < 2) { + usage(argv[0]); + return 0; + } + + MagickWandGenesis(); + + fidx=0; + while(1) { + pDir = opendir (argv[1]); + if (pDir == NULL) { + printf ("Cannot open directory '%s'\n", argv[1]); + return 1; + } + if (fidx == 9999) fidx=0; + while ((pDirent = readdir(pDir)) != NULL) { + if (pDirent->d_name[0]!='.') { + sprintf (imagepath,"%s/%s",argv[1],pDirent->d_name); + ret=encodeFile(imagepath, fidx, padlen); + if (ret!=1) { + fprintf(stderr,"Error - Cannot encode file %s\n",pDirent->d_name); + } else { + fidx++; + writeDLS(padlen); + sleep(10); + } + } + } + closedir (pDir); + } + return 1; +} + +int encodeFile (char* fname, int fidx, int padlen) +{ + int fd=0,ret,mothdrlen,nseg,lastseglen,i,last,curseglen; + unsigned char mothdr[32]; + MagickWand *m_wand = NULL; + PixelWand *p_wand = NULL; + size_t blobsize,height,width; + unsigned char *blob = NULL, *curseg = NULL; + MagickBooleanType err; + MSCDG msc; + unsigned char mscblob[8200]; + unsigned short int mscblobsize; + //float aspectRatio; + + + m_wand = NewMagickWand(); + p_wand = NewPixelWand(); + PixelSetColor(p_wand,"black"); + + err = MagickReadImage(m_wand,fname); + if (err==MagickFalse) { + fprintf(stderr,"Error - Unable to load image %s\n",fname); + ret=0; + goto RETURN; + } + + height=MagickGetImageHeight(m_wand); + width=MagickGetImageWidth(m_wand); + //aspectRatio = (width * 1.0)/height; + + fprintf(stderr,"Image: %s (id=%d). Original size: %d x %d. ",fname,fidx,width,height); + + while (height>240 || width>320) { + if (height/240.0 > width/320.0) { + //width = height * aspectRatio; + width = width * 240.0 / height; + height = 240; + } + else { + //height = width * (1.0/aspectRatio); + height = height * 320.0 / width; + width = 320; + } + MagickResizeImage(m_wand,width,height,LanczosFilter,1); + } + + + + height=MagickGetImageHeight(m_wand); + width=MagickGetImageWidth(m_wand); + + MagickBorderImage(m_wand,p_wand,(320-width)/2,(240-height)/2); + + MagickSetImageCompressionQuality(m_wand,75); + MagickSetImageFormat(m_wand,"jpg"); + blob=MagickGetImagesBlob(m_wand, &blobsize); + fprintf(stderr,"Resized to %d x %d. Size after compression %d bytes\n",width,height,blobsize); + + nseg=blobsize / MAXSEGLEN; + lastseglen=blobsize % MAXSEGLEN; + if (lastseglen !=0) nseg++; + + createMotHeader(blobsize,fidx,mothdr,&mothdrlen); + // Create the MSC Data Group C-Structure + createMscDG(&msc,3,0,1,fidx,mothdr,mothdrlen); + // Generate the MSC DG frame (Figure 9 en 300 401) + packMscDG(mscblob,&msc,&mscblobsize); + writeMotPAD(mscblob,mscblobsize,padlen); + + for (i=0;i>24; + MotHeaderCore[1]=(blobsize<<4 & 0x00FF0000)>>16; + MotHeaderCore[2]=(blobsize<<4 & 0x0000FF00)>>8; + MotHeaderCore[3]=(blobsize<<4 & 0x000000FF); + + sprintf(cntemp,"img%04d.jpg",fidx); + for (i=0; iextflag=0; + msc->crcflag=1; + msc->segflag=1; + msc->accflag=1; + msc->dgtype=dgtype; + msc->cindex=cindex; + msc->rindex=0; + msc->last=lastseg; + msc->segnum=cindex; + msc->rfa=0; + msc->tidflag=1; + msc->lenid=2; + msc->tid=tid; + msc->segdata=data; + msc->rcount=0; + msc->seglen=datalen; + + return; +} + + +void packMscDG(unsigned char* b, MSCDG* msc, unsigned short int* bsize) { + + int i; + unsigned short int crc=0xFFFF; + + b[0]=(msc->extflag<<7)|(msc->crcflag<<6)|(msc->segflag<<5)|(msc->accflag<<4)|msc->dgtype; + b[1]=(msc->cindex<<4)|msc->rindex; + b[2]=(msc->last<<7)|((msc->segnum & 0x7F00)>>8); + b[3]=msc->segnum & 0x00FF; + b[4]=0; + b[4]=(msc->rfa<<5)|(msc->tidflag<<4)|msc->lenid; + b[5]=(msc->tid & 0xFF00)>>8; + b[6]=msc->tid & 0x00FF; + b[7]=(msc->rcount<<5)|((msc->seglen & 0x1F00)>>8); + b[8]=msc->seglen & 0x00FF; + + for (i=0;i<9;i++) + crc = update_crc_ccitt(crc, b[i]); + + for(i=0;iseglen;i++) { + b[i+9]=(msc->segdata)[i]; + crc = update_crc_ccitt(crc, b[i+9]); + } + + crc=~crc; + b[9+msc->seglen] = (crc & 0xFF00) >> 8; // HI CRC + b[9+msc->seglen+1] = crc & 0x00FF; // LO CRC + + *bsize=9+msc->seglen+1+1; + + //write(1,b,9+msc->seglen+1+1); + +} + + +void writeMotPAD(unsigned char* mscdg, unsigned short int mscdgsize, unsigned short int padlen) { + + unsigned char pad[128]; + int xpadlengthmask,i,j,numseg,lastseglen; + unsigned short int crc; + + + if (padlen==17) + xpadlengthmask=3; + else if (padlen==21) + xpadlengthmask=4; + else if (padlen==29) + xpadlengthmask=5; + else if (padlen==37) + xpadlengthmask=6; + else if (padlen==53) + xpadlengthmask=7; + + + // Write Data Group Length Indicator + crc = 0xffff; + pad[padlen-1]=0x02; // FF-PAD Byte L (CI=1) + pad[padlen-2]=0x20; // FF-PAD Byte L-1 (Variable size X_PAD) + pad[padlen-3]=(xpadlengthmask<<5) | 0x01; // CI => data length = 12 (011) - Application Type=2 (DLS - start of X-PAD data group) + pad[padlen-4]=0x00; // End of CI list + pad[padlen-5]=(mscdgsize & 0x3F00)>>8; // RFA+HI Data group length + pad[padlen-6]=(mscdgsize & 0x00FF); + crc = update_crc_ccitt(crc, pad[padlen-5]); + crc = update_crc_ccitt(crc, pad[padlen-6]); + crc = ~crc; + pad[padlen-7]=(crc & 0xFF00) >> 8; // HI CRC + pad[padlen-8]=(crc & 0x00FF); // LO CRC + for (i=padlen-9;i>=0;i--) { // NULL PADDING + pad[i]=0x00; + } + write(STDOUT_FILENO,pad,padlen); + //fprintf(stderr,"Data Group Length Indicator: "); + //for (i=0;i 0) { + numseg++; // The last incomplete segment + } + + for (i=0;i data length = 12 (011) - Application Type=12 (start of MOT) + else + pad[padlen-3]=(xpadlengthmask<<5) | 13; // CI => data length = 12 (011) - Application Type=13 (MOT) + + pad[padlen-4]=0x00; // End of CI list + + for (j=0; j=0;j--) + pad[j]=0x00; + + write(STDOUT_FILENO,pad,padlen); + //fprintf(stderr,"MSC Data Group - Segment %d: ",i); + //for (j=0;j= 16) { + if (lastseglen > 0) numseg++; // The last incomplete segment + numdg=numseg; // The PAD can contain the full segmnet and overhead (9 bytes) + } else { + numdg=numseg*2; // Each 16 char segment span over 2 dg + if (lastseglen > 0) { + numseg++; // The last incomplete segment + if (lastseglen <= padlen-9) + numdg+=1; + else + numdg+=2; + + } + } + *p_numdg=numdg; + fprintf(stderr,"PAD Length: %d\n",padlen); + fprintf(stderr,"DLS text: %s\n",text); + fprintf(stderr,"Number od DLS segments: %d\n",numseg); + fprintf(stderr,"Number od DLS data grupus: %d\n",numdg); + + if (padlen==17) + xpadlengthmask=3; + else if (padlen==21) + xpadlengthmask=4; + else if (padlen==29) + xpadlengthmask=5; + else if (padlen==37) + xpadlengthmask=6; + else if (padlen==53) + xpadlengthmask=7; + + *p_dlsdg = (UCHAR**) malloc(numdg*sizeof(UCHAR*)); + dlsdg=*p_dlsdg; + + i=0; + for (z=0;z data length = 12 (011) - Application Type=2 (DLS - start of X-PAD data group) + dlsdg[i][padlen-4]=0x00; // End of CI list + + dlsdg[i][padlen-5]=((toggle*8+firstseg*4+lastseg*2+0)<<4) | (curseglen-1); // DLS Prefix (T=1,Only one segment,segment length-1) + + if (firstseg==1) + dlsdg[i][padlen-6]=0x00; // DLS Prefix (Charset standard) + else + dlsdg[i][padlen-6]=z<<4; // DLS SegNum + + idx_start_crc = padlen-5; // CRC start from prefix + for (j=0;j=idx_stop_crc;j--) { + dlscrc = update_crc_ccitt(dlscrc, dlsdg[i][j]); + } + + dlscrc = ~dlscrc; + fprintf(stderr,"crc=%x ~crc=%x\n",~dlscrc,dlscrc); + + dlsdg[i][padlen-7-curseglen] = (dlscrc & 0xFF00) >> 8; // HI CRC + dlsdg[i][padlen-7-curseglen-1] = dlscrc & 0x00FF; // LO CRC + + for (j=padlen-7-curseglen-2;j>=0;j--) { // NULL PADDING + dlsdg[i][j]=0x00; + } + fprintf(stderr,"Data group: "); + for (j=0;j data length = 12 (011) - Application Type=2 (DLS - start of X-PAD data group) + dlsdg[i][padlen-4]=0x00; // End of CI list + dlsdg[i][padlen-5]=((toggle*8+firstseg*4+lastseg*2+0)<<4) | (curseglen-1); // DLS Prefix (T=1,Only one segment,segment length-1) + + if (firstseg==1) + dlsdg[i][padlen-6]=0x00; // DLS Prefix (Charset standard) + else + dlsdg[i][padlen-6]=(i-1)<<4; // DLS SegNum + + dlscrc = update_crc_ccitt(dlscrc, dlsdg[i][padlen-5]); + dlscrc = update_crc_ccitt(dlscrc, dlsdg[i][padlen-6]); + + for (j=0;j < MIN(curseglen,padlen-7); j++) { // DLS text + dlsdg[i][padlen-7-j]=curseg[j]; + dlscrc = update_crc_ccitt(dlscrc, dlsdg[i][padlen-7-j]); + } + k=j; + if (curseglen==padlen-8) { // end of segment + dlscrc = ~dlscrc; + dlsdg[i][1] = (dlscrc & 0xFF00) >> 8; // HI CRC + fprintf(stderr,"crc=%x ~crc=%x\n",~dlscrc,dlscrc); + } else if (curseglen==padlen-7) { + dlscrc = ~dlscrc; + fprintf(stderr,"crc=%x ~crc=%x\n",~dlscrc,dlscrc); + } + dlsdg[i][0]=0x00; + + fprintf(stderr,"First Data group: "); + for (j=0;j> 8; // HI CRC + dlsdg[i][padlen-4] = dlscrc & 0x00FF; // LO CRC + } else { + for (j=0;j < curseglen-k; j++) { // DLS text + dlsdg[i][padlen-3-j]=curseg[k+j]; + dlscrc = update_crc_ccitt(dlscrc, dlsdg[i][padlen-3-j]); + } + dlscrc = ~dlscrc; + dlsdg[i][padlen-3-curseglen+k]=(dlscrc & 0xFF00) >> 8; // HI CRC + dlsdg[i][padlen-3-curseglen+k-1]=dlscrc & 0x00FF; // LO CRC + } + + fprintf(stderr,"Second Data group: "); + for (j=0;j +#include +#include +#include + +#define TAG(a, b, c, d) (((a) << 24) | ((b) << 16) | ((c) << 8) | (d)) + +struct wav_reader { + FILE *wav; + uint32_t data_length; + + int format; + int sample_rate; + int bits_per_sample; + int channels; + int byte_rate; + int block_align; +}; + +static uint32_t read_tag(struct wav_reader* wr) { + uint32_t tag = 0; + tag = (tag << 8) | fgetc(wr->wav); + tag = (tag << 8) | fgetc(wr->wav); + tag = (tag << 8) | fgetc(wr->wav); + tag = (tag << 8) | fgetc(wr->wav); + return tag; +} + +static uint32_t read_int32(struct wav_reader* wr) { + uint32_t value = 0; + value |= fgetc(wr->wav) << 0; + value |= fgetc(wr->wav) << 8; + value |= fgetc(wr->wav) << 16; + value |= fgetc(wr->wav) << 24; + return value; +} + +static uint16_t read_int16(struct wav_reader* wr) { + uint16_t value = 0; + value |= fgetc(wr->wav) << 0; + value |= fgetc(wr->wav) << 8; + return value; +} + +void* wav_read_open(const char *filename) { + struct wav_reader* wr = (struct wav_reader*) malloc(sizeof(*wr)); + long data_pos = 0; + memset(wr, 0, sizeof(*wr)); + + wr->wav = fopen(filename, "rb"); + if (wr->wav == NULL) { + free(wr); + return NULL; + } + + while (1) { + uint32_t tag, tag2, length; + tag = read_tag(wr); + if (feof(wr->wav)) + break; + length = read_int32(wr); + if (tag != TAG('R', 'I', 'F', 'F') || length < 4) { + fseek(wr->wav, length, SEEK_CUR); + continue; + } + tag2 = read_tag(wr); + length -= 4; + if (tag2 != TAG('W', 'A', 'V', 'E')) { + fseek(wr->wav, length, SEEK_CUR); + continue; + } + // RIFF chunk found, iterate through it + while (length >= 8) { + uint32_t subtag, sublength; + subtag = read_tag(wr); + if (feof(wr->wav)) + break; + sublength = read_int32(wr); + length -= 8; + if (length < sublength) + break; + if (subtag == TAG('f', 'm', 't', ' ')) { + if (sublength < 16) { + // Insufficient data for 'fmt ' + break; + } + wr->format = read_int16(wr); + wr->channels = read_int16(wr); + wr->sample_rate = read_int32(wr); + wr->byte_rate = read_int32(wr); + wr->block_align = read_int16(wr); + wr->bits_per_sample = read_int16(wr); + fseek(wr->wav, sublength - 16, SEEK_CUR); + } else if (subtag == TAG('d', 'a', 't', 'a')) { + data_pos = ftell(wr->wav); + wr->data_length = sublength; + fseek(wr->wav, sublength, SEEK_CUR); + } else { + fseek(wr->wav, sublength, SEEK_CUR); + } + length -= sublength; + } + if (length > 0) { + // Bad chunk? + fseek(wr->wav, length, SEEK_CUR); + } + } + fseek(wr->wav, data_pos, SEEK_SET); + return wr; +} + +void wav_read_close(void* obj) { + struct wav_reader* wr = (struct wav_reader*) obj; + fclose(wr->wav); + free(wr); +} + +int wav_get_header(void* obj, int* format, int* channels, int* sample_rate, int* bits_per_sample, unsigned int* data_length) { + struct wav_reader* wr = (struct wav_reader*) obj; + if (format) + *format = wr->format; + if (channels) + *channels = wr->channels; + if (sample_rate) + *sample_rate = wr->sample_rate; + if (bits_per_sample) + *bits_per_sample = wr->bits_per_sample; + if (data_length) + *data_length = wr->data_length; + return wr->format && wr->sample_rate; +} + +int wav_read_data(void* obj, unsigned char* data, unsigned int length) { + struct wav_reader* wr = (struct wav_reader*) obj; + int n; + if (wr->wav == NULL) + return -1; + if (length > wr->data_length) + length = wr->data_length; + n = fread(data, 1, length, wr->wav); + wr->data_length -= length; + return n; +} + diff --git a/src/wavreader.h b/src/wavreader.h new file mode 100644 index 0000000..57a13ff --- /dev/null +++ b/src/wavreader.h @@ -0,0 +1,37 @@ +/* ------------------------------------------------------------------ + * Copyright (C) 2009 Martin Storsjo + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either + * express or implied. + * See the License for the specific language governing permissions + * and limitations under the License. + * ------------------------------------------------------------------- + */ + +#ifndef WAVREADER_H +#define WAVREADER_H + +#ifdef __cplusplus +extern "C" { +#endif + +void* wav_read_open(const char *filename); +void wav_read_close(void* obj); + +int wav_get_header(void* obj, int* format, int* channels, int* sample_rate, int* bits_per_sample, unsigned int* data_length); +int wav_read_data(void* obj, unsigned char* data, unsigned int length); + +#ifdef __cplusplus +} +#endif + +#endif + diff --git a/wavreader.c b/wavreader.c deleted file mode 100644 index c62a885..0000000 --- a/wavreader.c +++ /dev/null @@ -1,163 +0,0 @@ -/* ------------------------------------------------------------------ - * Copyright (C) 2009 Martin Storsjo - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either - * express or implied. - * See the License for the specific language governing permissions - * and limitations under the License. - * ------------------------------------------------------------------- - */ - -#include "wavreader.h" -#include -#include -#include -#include - -#define TAG(a, b, c, d) (((a) << 24) | ((b) << 16) | ((c) << 8) | (d)) - -struct wav_reader { - FILE *wav; - uint32_t data_length; - - int format; - int sample_rate; - int bits_per_sample; - int channels; - int byte_rate; - int block_align; -}; - -static uint32_t read_tag(struct wav_reader* wr) { - uint32_t tag = 0; - tag = (tag << 8) | fgetc(wr->wav); - tag = (tag << 8) | fgetc(wr->wav); - tag = (tag << 8) | fgetc(wr->wav); - tag = (tag << 8) | fgetc(wr->wav); - return tag; -} - -static uint32_t read_int32(struct wav_reader* wr) { - uint32_t value = 0; - value |= fgetc(wr->wav) << 0; - value |= fgetc(wr->wav) << 8; - value |= fgetc(wr->wav) << 16; - value |= fgetc(wr->wav) << 24; - return value; -} - -static uint16_t read_int16(struct wav_reader* wr) { - uint16_t value = 0; - value |= fgetc(wr->wav) << 0; - value |= fgetc(wr->wav) << 8; - return value; -} - -void* wav_read_open(const char *filename) { - struct wav_reader* wr = (struct wav_reader*) malloc(sizeof(*wr)); - long data_pos = 0; - memset(wr, 0, sizeof(*wr)); - - wr->wav = fopen(filename, "rb"); - if (wr->wav == NULL) { - free(wr); - return NULL; - } - - while (1) { - uint32_t tag, tag2, length; - tag = read_tag(wr); - if (feof(wr->wav)) - break; - length = read_int32(wr); - if (tag != TAG('R', 'I', 'F', 'F') || length < 4) { - fseek(wr->wav, length, SEEK_CUR); - continue; - } - tag2 = read_tag(wr); - length -= 4; - if (tag2 != TAG('W', 'A', 'V', 'E')) { - fseek(wr->wav, length, SEEK_CUR); - continue; - } - // RIFF chunk found, iterate through it - while (length >= 8) { - uint32_t subtag, sublength; - subtag = read_tag(wr); - if (feof(wr->wav)) - break; - sublength = read_int32(wr); - length -= 8; - if (length < sublength) - break; - if (subtag == TAG('f', 'm', 't', ' ')) { - if (sublength < 16) { - // Insufficient data for 'fmt ' - break; - } - wr->format = read_int16(wr); - wr->channels = read_int16(wr); - wr->sample_rate = read_int32(wr); - wr->byte_rate = read_int32(wr); - wr->block_align = read_int16(wr); - wr->bits_per_sample = read_int16(wr); - fseek(wr->wav, sublength - 16, SEEK_CUR); - } else if (subtag == TAG('d', 'a', 't', 'a')) { - data_pos = ftell(wr->wav); - wr->data_length = sublength; - fseek(wr->wav, sublength, SEEK_CUR); - } else { - fseek(wr->wav, sublength, SEEK_CUR); - } - length -= sublength; - } - if (length > 0) { - // Bad chunk? - fseek(wr->wav, length, SEEK_CUR); - } - } - fseek(wr->wav, data_pos, SEEK_SET); - return wr; -} - -void wav_read_close(void* obj) { - struct wav_reader* wr = (struct wav_reader*) obj; - fclose(wr->wav); - free(wr); -} - -int wav_get_header(void* obj, int* format, int* channels, int* sample_rate, int* bits_per_sample, unsigned int* data_length) { - struct wav_reader* wr = (struct wav_reader*) obj; - if (format) - *format = wr->format; - if (channels) - *channels = wr->channels; - if (sample_rate) - *sample_rate = wr->sample_rate; - if (bits_per_sample) - *bits_per_sample = wr->bits_per_sample; - if (data_length) - *data_length = wr->data_length; - return wr->format && wr->sample_rate; -} - -int wav_read_data(void* obj, unsigned char* data, unsigned int length) { - struct wav_reader* wr = (struct wav_reader*) obj; - int n; - if (wr->wav == NULL) - return -1; - if (length > wr->data_length) - length = wr->data_length; - n = fread(data, 1, length, wr->wav); - wr->data_length -= length; - return n; -} - diff --git a/wavreader.h b/wavreader.h deleted file mode 100644 index 57a13ff..0000000 --- a/wavreader.h +++ /dev/null @@ -1,37 +0,0 @@ -/* ------------------------------------------------------------------ - * Copyright (C) 2009 Martin Storsjo - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either - * express or implied. - * See the License for the specific language governing permissions - * and limitations under the License. - * ------------------------------------------------------------------- - */ - -#ifndef WAVREADER_H -#define WAVREADER_H - -#ifdef __cplusplus -extern "C" { -#endif - -void* wav_read_open(const char *filename); -void wav_read_close(void* obj); - -int wav_get_header(void* obj, int* format, int* channels, int* sample_rate, int* bits_per_sample, unsigned int* data_length); -int wav_read_data(void* obj, unsigned char* data, unsigned int length); - -#ifdef __cplusplus -} -#endif - -#endif - -- cgit v1.2.3