added unmodified mmbtools

1 files changed, 456 insertions, 0 deletions

diff --git a/src/DabMod.cpp b/src/DabMod.cpp
new file mode 100644
index 0000000..db0c712
--- /dev/null
+++ b/src/DabMod.cpp
@@ -0,0 +1,456 @@
+/*
+   Copyright (C) 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
+   Her Majesty the Queen in Right of Canada (Communications Research
+   Center Canada)
+ */
+/*
+   This file is part of CRC-DADMOD.
+
+   CRC-DADMOD 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-DADMOD 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-DADMOD.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifdef HAVE_CONFIG_H
+#   include "config.h"
+#endif
+
+#include "porting.h"
+
+#include "DabModulator.h"
+#include "InputMemory.h"
+#include "OutputFile.h"
+#include "PcDebug.h"
+
+#include <complex>
+#include <stdlib.h>
+#include <unistd.h>
+#include <stdio.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <stdexcept>
+#include <signal.h>
+#include <string.h>
+
+#ifdef HAVE_NETINET_IN_H
+#   include <netinet/in.h>
+#endif
+
+#ifdef HAVE_DECL__MM_MALLOC
+#   include <mm_malloc.h>
+#else
+#   define memalign(a, b)   malloc(b)
+#endif
+
+
+typedef std::complex<float> complexf;
+
+
+bool running = true;
+
+void signalHandler(int signalNb)
+{
+    PDEBUG("signalHandler(%i)\n", signalNb);
+
+    running = false;
+}
+
+
+void printUsage(char* progName, FILE* out = stderr)
+{
+    fprintf(out, "Usage:\n");
+    fprintf(out, "\t%s"
+            " [input [output]]"
+            " [-a amplitude]"
+            " [-c clockrate]"
+            " [-f]"
+            " [-g gainMode]"
+            " [-h]"
+            " [-l]"
+            " [-m dabMode]"
+            " [-r samplingRate]"
+            "\n", progName);
+    fprintf(out, "Where:\n");
+    fprintf(out, "input:    ETI input filename (default: stdin).\n");
+    fprintf(out, "output:   COFDM output filename (default: stdout).\n");
+    fprintf(out, "-a:       Apply amplitude gain.\n");
+    fprintf(out, "-c:       Set the DAC clock rate.\n");
+    fprintf(out, "-f:       (deprecated) Set fifo input.\n");
+    fprintf(out, "-g:       Set computation gain mode: "
+            "%u FIX, %u MAX, %u VAR\n", GAIN_FIX, GAIN_MAX, GAIN_VAR);
+    fprintf(out, "-h:       Print this help.\n");
+    fprintf(out, "-l:       Loop file when reach end of file.\n");
+    fprintf(out, "-m:       Set DAB mode: (0: auto, 1-4: force).\n");
+    fprintf(out, "-r:       Set output sampling rate (default: 2048000).\n");
+}
+
+
+void printVersion(FILE *out = stderr)
+{
+    fprintf(out, "Welcome to %s %s, compiled at %s, %s\n\n",
+            PACKAGE, VERSION, __DATE__, __TIME__);
+    fprintf(out, "CRC-DABMOD is copyright (C) Her Majesty the Queen in Right of Canada,\n"
+            "    2009, 2010, 2011, 2012 Communications Research Centre (CRC).\n"
+            "\n"
+            "    This program is available free of charge and is licensed to you on a\n"
+            "    non-exclusive basis; you may not redistribute it.\n"
+            "\n"
+            "    This program is provided \"AS IS\" in the hope that it will be useful, but\n"
+            "    WITHOUT ANY WARRANTY with respect to its accurancy or usefulness; witout\n"
+            "    even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR\n"
+            "    PURPOSE and NONINFRINGEMENT.\n"
+            "\n"
+            "    In no event shall CRC be LIABLE for any LOSS, DAMAGE or COST that may be\n"
+            "    incurred in connection with the use of this software.\n"
+            "\n"
+            "CRC-DABMOD makes use of the following open source packages:\n"
+            "    Kiss FFT v1.2.9 (Revised BSD) - http://kissfft.sourceforge.net/\n"
+           );
+
+}
+
+
+int main(int argc, char* argv[])
+{
+    int ret = 0;
+    bool loop = false;
+    char* inputName;
+    char* outputName;
+    FILE* inputFile = NULL;
+    uint32_t sync = 0;
+    uint32_t nbFrames = 0;
+    uint32_t frame = 0;
+    uint16_t frameSize = 0;
+    size_t outputRate = 2048000;
+    size_t clockRate = 0;
+    unsigned dabMode = 0;
+    float amplitude = 1.0f;
+    GainMode gainMode = GAIN_VAR;
+    Buffer data;
+
+    Flowgraph* flowgraph = NULL;
+    DabModulator* modulator = NULL;
+    InputMemory* input = NULL;
+    OutputFile* output = NULL;
+
+    signal(SIGINT, signalHandler);
+
+    while (true) {
+        int c = getopt(argc, argv, "a:c:fg:hlm:r:V");
+        if (c == -1) {
+            break;
+        }
+        switch (c) {
+        case 'a':
+            amplitude = strtof(optarg, NULL);
+            break;
+        case 'c':
+            clockRate = strtol(optarg, NULL, 0);
+            break;
+        case 'f':
+            fprintf(stderr, "Option -f deprecated!\n");
+            break;
+        case 'g':
+            gainMode = (GainMode)strtol(optarg, NULL, 0);
+            break;
+        case 'l':
+            loop = true;
+            break;
+        case 'm':
+            dabMode = strtol(optarg, NULL, 0);
+            break;
+        case 'r':
+            outputRate = strtol(optarg, NULL, 0);
+            break;
+        case 'V':
+            printVersion();
+            goto END_MAIN;
+            break;
+        case '?':
+        case 'h':
+            printUsage(argv[0]);
+            goto END_MAIN;
+            break;
+        default:
+            fprintf(stderr, "Option '%c' not coded yet!\n", c);
+            ret = -1;
+            goto END_MAIN;
+        }
+    }
+    // Setting ETI input filename
+    if (optind < argc) {
+        inputName = argv[optind++];
+    } else {
+        inputName = (char*)"/dev/stdin";
+    }
+    // Setting COFDM output filename
+    if (optind < argc) {
+        outputName = argv[optind++];
+    } else {
+        outputName = (char*)"/dev/stdout";
+    }
+    // Checking unused arguments
+    if (optind != argc) {
+        fprintf(stderr, "Invalid arguments:");
+        while (optind != argc) {
+            fprintf(stderr, " %s", argv[optind++]);
+        }
+        fprintf(stderr, "\n");
+        printUsage(argv[0]);
+        ret = -1;
+        goto END_MAIN;
+    }
+
+    // Print settings
+    fprintf(stderr, "Input\n");
+    fprintf(stderr, "  Name: %s\n", inputName);
+    fprintf(stderr, "Output\n");
+    fprintf(stderr, "  Name: %s\n", outputName);
+    fprintf(stderr, "  Sampling rate: ");
+    if (outputRate > 1000) {
+        if (outputRate > 1000000) {
+            fprintf(stderr, "%.3g mHz\n", outputRate / 1000000.0f);
+        } else {
+            fprintf(stderr, "%.3g kHz\n", outputRate / 1000.0f);
+        }
+    } else {
+        fprintf(stderr, "%zu Hz\n", outputRate);
+    }
+
+    // Opening ETI input file
+    inputFile = fopen(inputName, "r");
+    if (inputFile == NULL) {
+        fprintf(stderr, "Unable to open input file!\n");
+        perror(inputName);
+        ret = -1;
+        goto END_MAIN;
+    }
+    // Opening COFDM output file
+    if (outputName != NULL) {
+        output = new OutputFile(outputName);
+    }
+
+    flowgraph = new Flowgraph();
+    data.setLength(6144);
+    input = new InputMemory(&data);
+    modulator = new DabModulator(outputRate, clockRate, dabMode, gainMode, amplitude);
+    flowgraph->connect(input, modulator);
+    flowgraph->connect(modulator, output);
+
+    try {
+        while (running) {
+            enum EtiStreamType {
+                ETI_STREAM_TYPE_NONE = 0,
+                ETI_STREAM_TYPE_RAW,
+                ETI_STREAM_TYPE_STREAMED,
+                ETI_STREAM_TYPE_FRAMED,
+            };
+            EtiStreamType streamType = ETI_STREAM_TYPE_NONE;
+
+            struct stat inputFileStat;
+            fstat(fileno(inputFile), &inputFileStat);
+            size_t inputFileLength = inputFileStat.st_size;
+
+            if (fread(&sync, sizeof(sync), 1, inputFile) != 1) {
+                fprintf(stderr, "Unable to read sync in input file!\n");
+                perror(inputName);
+                ret = -1;
+                goto END_MAIN;
+            }
+            if ((sync == 0x49c5f8ff) || (sync == 0xb63a07ff)) {
+                streamType = ETI_STREAM_TYPE_RAW;
+                if (inputFileLength > 0) {
+                    nbFrames = inputFileLength / 6144;
+                } else {
+                    nbFrames = (uint32_t)-1;
+                }
+                if (fseek(inputFile, -sizeof(sync), SEEK_CUR) != 0) {
+                    if (fread(data.getData(), 6144 - sizeof(sync), 1, inputFile)
+                            != 1) {
+                        fprintf(stderr, "Unable to seek in input file!\n");
+                        ret = -1;
+                        goto END_MAIN;
+                    }
+                }
+                goto START;
+            }
+
+            nbFrames = sync;
+            if (fread(&frameSize, sizeof(frameSize), 1, inputFile) != 1) {
+                fprintf(stderr, "Unable to read frame size in input file!\n");
+                perror(inputName);
+                ret = -1;
+                goto END_MAIN;
+            }
+            sync >>= 16;
+            sync &= 0xffff;
+            sync |= ((uint32_t)frameSize) << 16;
+
+            if ((sync == 0x49c5f8ff) || (sync == 0xb63a07ff)) {
+                streamType = ETI_STREAM_TYPE_STREAMED;
+                frameSize = nbFrames & 0xffff;
+                if (inputFileLength > 0) {
+                    nbFrames = inputFileLength / (frameSize + 2);
+                } else {
+                    nbFrames = (uint32_t)-1;
+                }
+                if (fseek(inputFile, -6, SEEK_CUR) != 0) {
+                    if (fread(data.getData(), frameSize - 4, 1, inputFile)
+                            != 1) {
+                        fprintf(stderr, "Unable to seek in input file!\n");
+                        ret = -1;
+                        goto END_MAIN;
+                    }
+                }
+                goto START;
+            }
+
+            if (fread(&sync, sizeof(sync), 1, inputFile) != 1) {
+                fprintf(stderr, "Unable to read nb frame in input file!\n");
+                perror(inputName);
+                ret = -1;
+                goto END_MAIN;
+            }
+            if ((sync == 0x49c5f8ff) || (sync == 0xb63a07ff)) {
+                streamType = ETI_STREAM_TYPE_FRAMED;
+                if (fseek(inputFile, -6, SEEK_CUR) != 0) {
+                    if (fread(data.getData(), frameSize - 4, 1, inputFile)
+                            != 1) {
+                        fprintf(stderr, "Unable to seek in input file!\n");
+                        ret = -1;
+                        goto END_MAIN;
+                    }
+                }
+                goto START;
+            }
+
+            for (size_t i = 10; i < 6144 + 10; ++i) {
+                sync >>= 8;
+                sync &= 0xffffff;
+                if (fread((uint8_t*)&sync + 3, 1, 1, inputFile) != 1) {
+                    fprintf(stderr, "Unable to read in input file!\n");
+                    ret = 1;
+                    goto END_MAIN;
+                }
+                if ((sync == 0x49c5f8ff) || (sync == 0xb63a07ff)) {
+                    streamType = ETI_STREAM_TYPE_RAW;
+                    if (inputFileLength > 0) {
+                        nbFrames = (inputFileLength - i) / 6144;
+                    } else {
+                        nbFrames = (uint32_t)-1;
+                    }
+                    if (fseek(inputFile, -sizeof(sync), SEEK_CUR) != 0) {
+                        if (fread(data.getData(), 6144 - sizeof(sync), 1, inputFile)
+                                != 1) {
+                            fprintf(stderr, "Unable to seek in input file!\n");
+                            ret = -1;
+                            goto END_MAIN;
+                        }
+                    }
+                    goto START;
+                }
+            }
+
+            fprintf(stderr, "Bad input file format!\n");
+            ret = -1;
+            goto END_MAIN;
+
+START:
+            fprintf(stderr, "Input file format: ");
+            switch (streamType) {
+            case ETI_STREAM_TYPE_RAW:
+                fprintf(stderr, "raw");
+                break;
+            case ETI_STREAM_TYPE_STREAMED:
+                fprintf(stderr, "streamed");
+                break;
+            case ETI_STREAM_TYPE_FRAMED:
+                fprintf(stderr, "framed");
+                break;
+            default:
+                fprintf(stderr, "unknown\n");
+                ret = -1;
+                goto END_MAIN;
+            }
+            fprintf(stderr, "\n");
+            fprintf(stderr, "Input file length: %zu\n", inputFileLength);
+            fprintf(stderr, "Input file nb frames: %u\n", nbFrames);
+
+            for (frame = 0; frame < nbFrames; ++frame) {
+                if (!running) {
+                    break;
+                }
+
+                PDEBUG("*****************************************\n");
+                PDEBUG("* Reading frame %u\n", frame);
+                PDEBUG("*****************************************\n");
+                if (streamType == ETI_STREAM_TYPE_RAW) {
+                    frameSize = 6144;
+                } else {
+                    if (fread(&frameSize, sizeof(frameSize), 1, inputFile)
+                            != 1) {
+                        PDEBUG("End of file!\n");
+                        goto END_MAIN;
+                    }
+                }
+                PDEBUG("Frame size: %i\n", frameSize);
+
+                if (fread(data.getData(), frameSize, 1, inputFile) != 1) {
+                    fprintf(stderr,
+                            "Unable to read %i data bytes in input file!\n",
+                            frameSize);
+                    perror(inputName);
+                    ret = -1;
+                    goto END_MAIN;
+                }
+                memset(&((uint8_t*)data.getData())[frameSize], 0x55, 6144 - frameSize);
+
+                ////////////////////////////////////////////////////////////////
+                // Proccessing data
+                ////////////////////////////////////////////////////////////////
+                flowgraph->run();
+            }
+            fprintf(stderr, "End of file reached.\n");
+            if (!loop) {
+                running = false;
+            } else {
+                fprintf(stderr, "Rewinding file.\n");
+                rewind(inputFile);
+            }
+        }
+    } catch (std::exception& e) {
+        fprintf(stderr, "EXCEPTION: %s\n", e.what());
+        ret = -1;
+        goto END_MAIN;
+    }
+
+END_MAIN:
+    ////////////////////////////////////////////////////////////////////////
+    // Cleaning things
+    ////////////////////////////////////////////////////////////////////////
+    fprintf(stderr, "\n\n");
+    fprintf(stderr, "%u DAB frames encoded\n", frame);
+    fprintf(stderr, "%f seconds encoded\n", (float)frame * 0.024f);
+
+    fprintf(stderr, "\nCleaning flowgraph...\n");
+    delete flowgraph;
+
+    // Cif
+    fprintf(stderr, "\nCleaning buffers...\n");
+
+    fprintf(stderr, "\nClosing input file...\n");
+    if (inputFile != NULL) {
+        fclose(inputFile);
+    }
+
+    return ret;
+}