diff options
Diffstat (limited to 'src/dabInputMpegFile.cpp')
| -rw-r--r-- | src/dabInputMpegFile.cpp | 201 |
1 files changed, 201 insertions, 0 deletions
diff --git a/src/dabInputMpegFile.cpp b/src/dabInputMpegFile.cpp new file mode 100644 index 0000000..0a06b2f --- /dev/null +++ b/src/dabInputMpegFile.cpp @@ -0,0 +1,201 @@ +/* + Copyright (C) 2009 Her Majesty the Queen in Right of Canada (Communications + Research Center Canada) + */ +/* + This file is part of CRC-DabMux. + + CRC-DabMux 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. + + CRC-DabMux 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 CRC-DabMux. If not, see <http://www.gnu.org/licenses/>. + */ + +#include "dabInputMpegFile.h" +#include "dabInputFile.h" +#include "mpeg.h" + +#include <stdio.h> +#include <errno.h> +#include <string.h> + + +#ifdef HAVE_FORMAT_MPEG +# ifdef HAVE_INPUT_FILE + + +struct dabInputOperations dabInputMpegFileOperations = { + dabInputFileInit, + dabInputFileOpen, + dabInputSetbuf, + dabInputFileRead, + NULL, + NULL, + dabInputMpegFileRead, + dabInputMpegSetbitrate, + dabInputFileClose, + dabInputFileClean, + dabInputFileRewind +}; + + +#define MPEG_FREQUENCY -2 +#define MPEG_PADDING -3 +#define MPEG_COPYRIGHT -4 +#define MPEG_ORIGINAL -5 +#define MPEG_EMPHASIS -6 +int checkDabMpegFrame(void* data) { + mpegHeader* header = (mpegHeader*)data; + unsigned long* headerData = (unsigned long*)data; + if ((*headerData & 0x0f0ffcff) == 0x0004fcff) return 0; + if ((*headerData & 0x0f0ffcff) == 0x0004f4ff) return 0; + if (getMpegFrequency(header) != 48000) { + if (getMpegFrequency(header) != 24000) { + return MPEG_FREQUENCY; + } + } + if (header->padding != 0) { + return MPEG_PADDING; + } + if (header->copyright != 0) { + return MPEG_COPYRIGHT; + } + if (header->original != 0) { + return MPEG_ORIGINAL; + } + if (header->emphasis != 0) { + return MPEG_EMPHASIS; + } + return -1; +} + + +int dabInputMpegFileRead(dabInputOperations* ops, void* args, void* buffer, int size) +{ + dabInputFileData* data = (dabInputFileData*)args; + int result; + bool rewind = false; +READ_SUBCHANNEL: + if (data->parity) { + result = readData(data->file, buffer, size, 2); + data->parity = false; + return 0; + } else { + result = readMpegHeader(data->file, buffer, size); + if (result > 0) { + result = readMpegFrame(data->file, buffer, size); + if (result < 0 && getMpegFrequency(buffer) == 24000) { + data->parity = true; + result = size; + } + } + } + switch (result) { + case MPEG_BUFFER_UNDERFLOW: + etiLog.print(TcpLog::WARNING, "data underflow -> frame muted\n"); + goto MUTE_SUBCHANNEL; + case MPEG_BUFFER_OVERFLOW: + etiLog.print(TcpLog::WARNING, "bitrate too high -> frame muted\n"); + goto MUTE_SUBCHANNEL; + case MPEG_FILE_EMPTY: + if (rewind) { + etiLog.print(TcpLog::ERR, "file rewinded and still empty " + "-> frame muted\n"); + goto MUTE_SUBCHANNEL; + } else { + rewind = true; + etiLog.print(TcpLog::NOTICE, "reach end of file -> rewinding\n"); + lseek(data->file, 0, SEEK_SET); + goto READ_SUBCHANNEL; + } + case MPEG_FILE_ERROR: + etiLog.print(TcpLog::CRIT, "can't read file (%i) -> frame muted\n", errno); + perror(""); + goto MUTE_SUBCHANNEL; + case MPEG_SYNC_NOT_FOUND: + etiLog.print(TcpLog::CRIT, "mpeg sync not found, maybe is not a valid file " + "-> frame muted\n"); + goto MUTE_SUBCHANNEL; + case MPEG_INVALID_FRAME: + etiLog.print(TcpLog::CRIT, "file is not a valid mpeg file " + "-> frame muted\n"); + goto MUTE_SUBCHANNEL; + default: + if (result < 0) { + etiLog.print(TcpLog::CRIT, + "unknown error (code = %i) -> frame muted\n", + result); +MUTE_SUBCHANNEL: + memset(buffer, 0, size); + } else { + if (result < size) { + etiLog.print(TcpLog::WARNING, "bitrate too low from file " + "-> frame padded\n"); + memset((char*)buffer + result, 0, size - result); + } + result = checkDabMpegFrame(buffer); + switch (result) { + case MPEG_FREQUENCY: + etiLog.print(TcpLog::ERR, "file has a frame with an invalid " + "frequency: %i, should be 48000 or 24000\n", + getMpegFrequency(buffer)); + break; + case MPEG_PADDING: + etiLog.print(TcpLog::WARNING, + "file has a frame with padding bit set\n"); + break; + case MPEG_COPYRIGHT: + etiLog.print(TcpLog::WARNING, + "file has a frame with copyright bit set\n"); + break; + case MPEG_ORIGINAL: + etiLog.print(TcpLog::WARNING, + "file has a frame with original bit set\n"); + break; + case MPEG_EMPHASIS: + etiLog.print(TcpLog::WARNING, + "file has a frame with emphasis bits set\n"); + break; + default: + if (result < 0) { + etiLog.print(TcpLog::CRIT, "mpeg file has an invalid DAB " + "mpeg frame (unknown reason: %i)\n", result); + } + break; + } + } + } + return result; +} + + +int dabInputMpegSetbitrate(dabInputOperations* ops, void* args, int bitrate) +{ + //dabInputFileData* data = (dabInputFileData*)args; + if (bitrate == 0) { + char buffer[4]; + + if (ops->readFrame(ops, args, buffer, 4) == 0) { + bitrate = getMpegBitrate(buffer); + } else { + bitrate = -1; + } + ops->rewind(args); + } + if (ops->setbuf(args, bitrate * 3) != 0) { + bitrate = -1; + } + return bitrate; +} + + +# endif +#endif |