/*
Copyright (C) 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
Her Majesty the Queen in Right of Canada (Communications Research
Center Canada)
Copyright (C) 2014, 2015
Matthias P. Braendli, matthias.braendli@mpb.li
http://opendigitalradio.org
*/
/*
This file is part of ODR-DabMod.
ODR-DabMod 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.
ODR-DabMod 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 ODR-DabMod. If not, see .
*/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include "porting.h"
#include "Utils.h"
#include "Log.h"
#include "DabModulator.h"
#include "InputMemory.h"
#include "OutputFile.h"
#include "FormatConverter.h"
#if defined(HAVE_OUTPUT_UHD)
# include "OutputUHD.h"
#endif
#include "OutputZeroMQ.h"
#include "InputReader.h"
#include "PcDebug.h"
#include "TimestampDecoder.h"
#include "FIRFilter.h"
#include "RemoteControl.h"
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#if HAVE_NETINET_IN_H
# include
#endif
#if HAVE_DECL__MM_MALLOC
# include
#else
# define memalign(a, b) malloc(b)
#endif
#define ZMQ_INPUT_MAX_FRAME_QUEUE 50
typedef std::complex complexf;
volatile sig_atomic_t running = 1;
void signalHandler(int signalNb)
{
PDEBUG("signalHandler(%i)\n", signalNb);
running = 0;
}
int main(int argc, char* argv[])
{
int ret = 0;
bool loop = false;
std::string inputName = "";
std::string inputTransport = "file";
unsigned inputMaxFramesQueued = ZMQ_INPUT_MAX_FRAME_QUEUE;
std::string outputName;
int useZeroMQOutput = 0;
int useFileOutput = 0;
std::string fileOutputFormat = "complexf";
int useUHDOutput = 0;
uint64_t frame = 0;
size_t outputRate = 2048000;
size_t clockRate = 0;
unsigned dabMode = 0;
float digitalgain = 1.0f;
float normalise = 1.0f;
GainMode gainMode = GAIN_VAR;
Buffer data;
/* UHD requires the input I and Q samples to be in the interval
* [-1.0,1.0], otherwise they get truncated, which creates very
* wide-spectrum spikes. Depending on the Transmission Mode, the
* Gain Mode and the sample rate (and maybe other parameters), the
* samples can have peaks up to about 48000. The value of 50000
* should guarantee that with a digital gain of 1.0, UHD never clips
* our samples.
*/
const float normalise_factor = 50000.0f;
std::string filterTapsFilename = "";
// Two configuration sources exist: command line and (new) INI file
bool use_configuration_cmdline = false;
bool use_configuration_file = false;
std::string configuration_file;
#if defined(HAVE_OUTPUT_UHD)
OutputUHDConfig outputuhd_conf;
#endif
// To handle the timestamp offset of the modulator
struct modulator_offset_config modconf;
modconf.use_offset_file = false;
modconf.use_offset_fixed = false;
modconf.delay_calculation_pipeline_stages = 0;
Flowgraph* flowgraph = NULL;
DabModulator* modulator = NULL;
InputMemory* input = NULL;
FormatConverter* format_converter = NULL;
ModOutput* output = NULL;
RemoteControllers rcs;
Logger logger;
InputFileReader inputFileReader(logger);
#if defined(HAVE_ZEROMQ)
InputZeroMQReader inputZeroMQReader(logger);
#endif
InputReader* inputReader;
struct sigaction sa;
memset(&sa, 0, sizeof(struct sigaction));
sa.sa_handler = &signalHandler;
if (sigaction(SIGINT, &sa, NULL) == -1) {
perror("sigaction");
return EXIT_FAILURE;
}
// Set timezone to UTC
setenv("TZ", "", 1);
tzset();
while (true) {
int c = getopt(argc, argv, "a:C:c:f:F:g:G:hlm:o:O:r:T:u:V");
if (c == -1) {
break;
}
if (c != 'C') {
use_configuration_cmdline = true;
}
switch (c) {
case 'C':
use_configuration_file = true;
configuration_file = optarg;
break;
case 'a':
digitalgain = strtof(optarg, NULL);
break;
case 'c':
clockRate = strtol(optarg, NULL, 0);
break;
case 'f':
#if defined(HAVE_OUTPUT_UHD)
if (useUHDOutput) {
fprintf(stderr, "Options -u and -f are mutually exclusive\n");
goto END_MAIN;
}
#endif
outputName = optarg;
useFileOutput = 1;
break;
case 'F':
#if defined(HAVE_OUTPUT_UHD)
outputuhd_conf.frequency = strtof(optarg, NULL);
#endif
break;
case 'g':
gainMode = (GainMode)strtol(optarg, NULL, 0);
break;
case 'G':
#if defined(HAVE_OUTPUT_UHD)
outputuhd_conf.txgain = strtod(optarg, NULL);
#endif
break;
case 'l':
loop = true;
break;
case 'o':
if (modconf.use_offset_file)
{
fprintf(stderr, "Options -o and -O are mutually exclusive\n");
goto END_MAIN;
}
modconf.use_offset_fixed = true;
modconf.offset_fixed = strtod(optarg, NULL);
#if defined(HAVE_OUTPUT_UHD)
outputuhd_conf.enableSync = true;
#endif
break;
case 'O':
if (modconf.use_offset_fixed)
{
fprintf(stderr, "Options -o and -O are mutually exclusive\n");
goto END_MAIN;
}
modconf.use_offset_file = true;
modconf.offset_filename = std::string(optarg);
#if defined(HAVE_OUTPUT_UHD)
outputuhd_conf.enableSync = true;
#endif
break;
case 'm':
dabMode = strtol(optarg, NULL, 0);
break;
case 'r':
outputRate = strtol(optarg, NULL, 0);
break;
case 'T':
filterTapsFilename = optarg;
break;
case 'u':
#if defined(HAVE_OUTPUT_UHD)
if (useFileOutput) {
fprintf(stderr, "Options -u and -f are mutually exclusive\n");
goto END_MAIN;
}
outputuhd_conf.device = optarg;
useUHDOutput = 1;
#endif
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;
}
}
std::cerr << "ODR-DabMod version " <<
#if defined(GITVERSION)
GITVERSION
#else
VERSION
#endif
<< std::endl;
std::cerr << "Using FFT library " <<
#if defined(USE_FFTW)
"FFTW" <<
#endif
#if defined(USE_KISS_FFT)
"Kiss FFT" <<
#endif
#if defined(USE_SIMD)
" (with fft_simd)" <<
#endif
"\n";
std::cerr << "Compiled with features: " <<
#if defined(HAVE_ZEROMQ)
"zeromq " <<
#endif
#if defined(HAVE_OUTPUT_UHD)
"output_uhd " <<
#endif
"\n";
if (use_configuration_file && use_configuration_cmdline) {
fprintf(stderr, "Warning: configuration file and command line parameters are defined:\n\t"
"Command line parameters override settings in the configuration file !\n");
}
// No argument given ? You can't be serious ! Show usage.
if (argc == 1) {
printUsage(argv[0]);
goto END_MAIN;
}
// If only one argument is given, interpret as configuration file name
if (argc == 2) {
use_configuration_file = true;
configuration_file = argv[1];
}
if (use_configuration_file) {
// First read parameters from the file
using boost::property_tree::ptree;
ptree pt;
try {
read_ini(configuration_file, pt);
}
catch (boost::property_tree::ini_parser::ini_parser_error &e)
{
fprintf(stderr, "Error, cannot read configuration file '%s'\n", configuration_file.c_str());
goto END_MAIN;
}
// remote controller:
if (pt.get("remotecontrol.telnet", 0) == 1) {
try {
int telnetport = pt.get("remotecontrol.telnetport");
RemoteControllerTelnet* telnetrc = new RemoteControllerTelnet(telnetport);
rcs.add_controller(telnetrc);
}
catch (std::exception &e) {
std::cerr << "Error: " << e.what() << "\n";
std::cerr << " telnet remote control enabled, but no telnetport defined.\n";
goto END_MAIN;
}
}
#if defined(HAVE_ZEROMQ)
if (pt.get("remotecontrol.zmqctrl", 0) == 1) {
try {
std::string zmqCtrlEndpoint = pt.get("remotecontrol.zmqctrlendpoint", "");
std::cerr << "ZmqCtrlEndpoint: " << zmqCtrlEndpoint << std::endl;
RemoteControllerZmq* zmqrc = new RemoteControllerZmq(zmqCtrlEndpoint);
rcs.add_controller(zmqrc);
}
catch (std::exception &e) {
std::cerr << "Error: " << e.what() << "\n";
std::cerr << " zmq remote control enabled, but no endpoint defined.\n";
goto END_MAIN;
}
}
#endif
// input params:
if (pt.get("input.loop", 0) == 1) {
loop = true;
}
inputTransport = pt.get("input.transport", "file");
inputMaxFramesQueued = pt.get("input.max_frames_queued",
ZMQ_INPUT_MAX_FRAME_QUEUE);
inputName = pt.get("input.source", "/dev/stdin");
// log parameters:
if (pt.get("log.syslog", 0) == 1) {
LogToSyslog* log_syslog = new LogToSyslog();
logger.register_backend(log_syslog);
}
if (pt.get("log.filelog", 0) == 1) {
std::string logfilename;
try {
logfilename = pt.get("log.filename");
}
catch (std::exception &e) {
std::cerr << "Error: " << e.what() << "\n";
std::cerr << " Configuration enables file log, but does not specify log filename\n";
goto END_MAIN;
}
LogToFile* log_file = new LogToFile(logfilename);
logger.register_backend(log_file);
}
// modulator parameters:
gainMode = (GainMode)pt.get("modulator.gainmode", 0);
dabMode = pt.get("modulator.mode", dabMode);
clockRate = pt.get("modulator.dac_clk_rate", (size_t)0);
digitalgain = pt.get("modulator.digital_gain", digitalgain);
outputRate = pt.get("modulator.rate", outputRate);
// FIR Filter parameters:
if (pt.get("firfilter.enabled", 0) == 1) {
try {
filterTapsFilename = pt.get("firfilter.filtertapsfile");
}
catch (std::exception &e) {
std::cerr << "Error: " << e.what() << "\n";
std::cerr << " Configuration enables firfilter, but does not specify filter taps file\n";
goto END_MAIN;
}
}
// Output options
std::string output_selected;
try {
output_selected = pt.get("output.output");
}
catch (std::exception &e) {
std::cerr << "Error: " << e.what() << "\n";
std::cerr << " Configuration does not specify output\n";
goto END_MAIN;
}
if (output_selected == "file") {
try {
outputName = pt.get("fileoutput.filename");
}
catch (std::exception &e) {
std::cerr << "Error: " << e.what() << "\n";
std::cerr << " Configuration does not specify file name for file output\n";
goto END_MAIN;
}
useFileOutput = 1;
fileOutputFormat = pt.get("fileoutput.format", fileOutputFormat);
}
#if defined(HAVE_OUTPUT_UHD)
else if (output_selected == "uhd") {
outputuhd_conf.device = pt.get("uhdoutput.device", "");
outputuhd_conf.usrpType = pt.get("uhdoutput.type", "");
outputuhd_conf.subDevice = pt.get("uhdoutput.subdevice", "");
outputuhd_conf.masterClockRate = pt.get("uhdoutput.master_clock_rate", 0);
if (outputuhd_conf.device.find("master_clock_rate") != std::string::npos) {
std::cerr << "Warning:"
"setting master_clock_rate in [uhd] device is deprecated !\n";
}
if (outputuhd_conf.device.find("type=") != std::string::npos) {
std::cerr << "Warning:"
"setting type in [uhd] device is deprecated !\n";
}
outputuhd_conf.txgain = pt.get("uhdoutput.txgain", 0.0);
outputuhd_conf.frequency = pt.get("uhdoutput.frequency", 0);
std::string chan = pt.get("uhdoutput.channel", "");
outputuhd_conf.dabMode = dabMode;
if (outputuhd_conf.frequency == 0 && chan == "") {
std::cerr << " UHD output enabled, but neither frequency nor channel defined.\n";
goto END_MAIN;
}
else if (outputuhd_conf.frequency == 0) {
double freq;
if (chan == "5A") freq = 174928000;
else if (chan == "5B") freq = 176640000;
else if (chan == "5C") freq = 178352000;
else if (chan == "5D") freq = 180064000;
else if (chan == "6A") freq = 181936000;
else if (chan == "6B") freq = 183648000;
else if (chan == "6C") freq = 185360000;
else if (chan == "6D") freq = 187072000;
else if (chan == "7A") freq = 188928000;
else if (chan == "7B") freq = 190640000;
else if (chan == "7C") freq = 192352000;
else if (chan == "7D") freq = 194064000;
else if (chan == "8A") freq = 195936000;
else if (chan == "8B") freq = 197648000;
else if (chan == "8C") freq = 199360000;
else if (chan == "8D") freq = 201072000;
else if (chan == "9A") freq = 202928000;
else if (chan == "9B") freq = 204640000;
else if (chan == "9C") freq = 206352000;
else if (chan == "9D") freq = 208064000;
else if (chan == "10A") freq = 209936000;
else if (chan == "10B") freq = 211648000;
else if (chan == "10C") freq = 213360000;
else if (chan == "10D") freq = 215072000;
else if (chan == "11A") freq = 216928000;
else if (chan == "11B") freq = 218640000;
else if (chan == "11C") freq = 220352000;
else if (chan == "11D") freq = 222064000;
else if (chan == "12A") freq = 223936000;
else if (chan == "12B") freq = 225648000;
else if (chan == "12C") freq = 227360000;
else if (chan == "12D") freq = 229072000;
else if (chan == "13A") freq = 230784000;
else if (chan == "13B") freq = 232496000;
else if (chan == "13C") freq = 234208000;
else if (chan == "13D") freq = 235776000;
else if (chan == "13E") freq = 237488000;
else if (chan == "13F") freq = 239200000;
else {
std::cerr << " UHD output: channel " << chan << " does not exist in table\n";
goto END_MAIN;
}
outputuhd_conf.frequency = freq;
}
else if (outputuhd_conf.frequency != 0 && chan != "") {
std::cerr << " UHD output: cannot define both frequency and channel.\n";
goto END_MAIN;
}
outputuhd_conf.refclk_src = pt.get("uhdoutput.refclk_source", "int");
outputuhd_conf.pps_src = pt.get("uhdoutput.pps_source", "int");
outputuhd_conf.pps_polarity = pt.get("uhdoutput.pps_polarity", "pos");
std::string behave = pt.get("uhdoutput.behaviour_refclk_lock_lost", "ignore");
if (behave == "crash") {
outputuhd_conf.refclk_lock_loss_behaviour = CRASH;
}
else if (behave == "ignore") {
outputuhd_conf.refclk_lock_loss_behaviour = IGNORE;
}
else {
std::cerr << "Error: UHD output: behaviour_refclk_lock_lost invalid." << std::endl;
goto END_MAIN;
}
useUHDOutput = 1;
}
#endif
#if defined(HAVE_ZEROMQ)
else if (output_selected == "zmq") {
outputName = pt.get("zmqoutput.listen");
useZeroMQOutput = 1;
}
#endif
else {
std::cerr << "Error: Invalid output defined.\n";
goto END_MAIN;
}
#if defined(HAVE_OUTPUT_UHD)
outputuhd_conf.enableSync = (pt.get("delaymanagement.synchronous", 0) == 1);
if (outputuhd_conf.enableSync) {
try {
std::string delay_mgmt = pt.get("delaymanagement.management");
if (delay_mgmt == "fixed") {
modconf.offset_fixed = pt.get("delaymanagement.fixedoffset");
modconf.use_offset_fixed = true;
}
else if (delay_mgmt == "dynamic") {
modconf.offset_filename = pt.get("delaymanagement.dynamicoffsetfile");
modconf.use_offset_file = true;
}
else {
throw std::runtime_error("invalid management value");
}
}
catch (std::exception &e) {
std::cerr << "Error: " << e.what() << "\n";
std::cerr << " Synchronised transmission enabled, but delay management specification is incomplete.\n";
goto END_MAIN;
}
}
outputuhd_conf.muteNoTimestamps = (pt.get("delaymanagement.mutenotimestamps", 0) == 1);
#endif
}
if (rcs.get_no_controllers() == 0) {
logger.level(warn) << "No Remote-Control started";
rcs.add_controller(new RemoteControllerDummy());
}
logger.level(info) << "Starting up";
if (!(modconf.use_offset_file || modconf.use_offset_fixed)) {
logger.level(debug) << "No Modulator offset defined, setting to 0";
modconf.use_offset_fixed = true;
modconf.offset_fixed = 0;
}
// When using the FIRFilter, increase the modulator offset pipelining delay
// by the correct amount
if (filterTapsFilename != "") {
modconf.delay_calculation_pipeline_stages += FIRFILTER_PIPELINE_DELAY;
}
// Setting ETI input filename
if (use_configuration_cmdline && inputName == "") {
if (optind < argc) {
inputName = argv[optind++];
if (inputName.substr(0, 4) == "zmq+" &&
inputName.find("://") != std::string::npos) {
// if the name starts with zmq+XYZ://somewhere:port
inputTransport = "zeromq";
}
}
else {
inputName = "/dev/stdin";
}
}
// Checking unused arguments
if (use_configuration_cmdline && optind != argc) {
fprintf(stderr, "Invalid arguments:");
while (optind != argc) {
fprintf(stderr, " %s", argv[optind++]);
}
fprintf(stderr, "\n");
printUsage(argv[0]);
ret = -1;
logger.level(error) << "Received invalid command line arguments";
goto END_MAIN;
}
if (!useFileOutput && !useUHDOutput && !useZeroMQOutput) {
logger.level(error) << "Output not specified";
fprintf(stderr, "Must specify output !");
goto END_MAIN;
}
// Print settings
fprintf(stderr, "Input\n");
fprintf(stderr, " Type: %s\n", inputTransport.c_str());
fprintf(stderr, " Source: %s\n", inputName.c_str());
fprintf(stderr, "Output\n");
if (useFileOutput) {
fprintf(stderr, " Name: %s\n", outputName.c_str());
}
#if defined(HAVE_OUTPUT_UHD)
else if (useUHDOutput) {
fprintf(stderr, " UHD\n"
" Device: %s\n"
" Type: %s\n"
" master_clock_rate: %ld\n",
outputuhd_conf.device.c_str(),
outputuhd_conf.usrpType.c_str(),
outputuhd_conf.masterClockRate);
}
#endif
else if (useZeroMQOutput) {
fprintf(stderr, " ZeroMQ\n"
" Listening on: %s\n",
outputName.c_str());
}
fprintf(stderr, " Sampling rate: ");
if (outputRate > 1000) {
if (outputRate > 1000000) {
fprintf(stderr, "%.4g MHz\n", outputRate / 1000000.0f);
} else {
fprintf(stderr, "%.4g kHz\n", outputRate / 1000.0f);
}
} else {
fprintf(stderr, "%zu Hz\n", outputRate);
}
if (inputTransport == "file") {
// Opening ETI input file
if (inputFileReader.Open(inputName, loop) == -1) {
fprintf(stderr, "Unable to open input file!\n");
logger.level(error) << "Unable to open input file!";
ret = -1;
goto END_MAIN;
}
inputReader = &inputFileReader;
}
else if (inputTransport == "zeromq") {
#if !defined(HAVE_ZEROMQ)
fprintf(stderr, "Error, ZeroMQ input transport selected, but not compiled in!\n");
ret = -1;
goto END_MAIN;
#else
// The URL might start with zmq+tcp://
if (inputName.substr(0, 4) == "zmq+") {
inputZeroMQReader.Open(inputName.substr(4), inputMaxFramesQueued);
}
else {
inputZeroMQReader.Open(inputName, inputMaxFramesQueued);
}
inputReader = &inputZeroMQReader;
#endif
}
else
{
fprintf(stderr, "Error, invalid input transport %s selected!\n", inputTransport.c_str());
ret = -1;
goto END_MAIN;
}
if (useFileOutput) {
if (fileOutputFormat == "complexf") {
output = new OutputFile(outputName);
}
else if (fileOutputFormat == "s8") {
// We must normalise the samples to the interval [-127.0; 127.0]
normalise = 127.0f / normalise_factor;
format_converter = new FormatConverter();
output = new OutputFile(outputName);
}
}
#if defined(HAVE_OUTPUT_UHD)
else if (useUHDOutput) {
normalise = 1.0f / normalise_factor;
outputuhd_conf.sampleRate = outputRate;
try {
output = new OutputUHD(outputuhd_conf, logger);
((OutputUHD*)output)->enrol_at(rcs);
}
catch (std::exception& e) {
logger.level(error) << "UHD initialisation failed:" << e.what();
goto END_MAIN;
}
}
#endif
#if defined(HAVE_ZEROMQ)
else if (useZeroMQOutput) {
/* We normalise the same way as for the UHD output */
normalise = 1.0f / normalise_factor;
output = new OutputZeroMQ(outputName);
}
#endif
flowgraph = new Flowgraph();
data.setLength(6144);
input = new InputMemory(&data);
modulator = new DabModulator(modconf, &rcs, logger, outputRate, clockRate,
dabMode, gainMode, digitalgain, normalise, filterTapsFilename);
flowgraph->connect(input, modulator);
if (format_converter) {
flowgraph->connect(modulator, format_converter);
flowgraph->connect(format_converter, output);
}
else {
flowgraph->connect(modulator, output);
}
#if defined(HAVE_OUTPUT_UHD)
if (useUHDOutput) {
((OutputUHD*)output)->setETIReader(modulator->getEtiReader());
}
#endif
inputReader->PrintInfo();
try {
while (running) {
int framesize;
PDEBUG("*****************************************\n");
PDEBUG("* Starting main loop\n");
PDEBUG("*****************************************\n");
while ((framesize = inputReader->GetNextFrame(data.getData())) > 0) {
if (!running) {
break;
}
frame++;
PDEBUG("*****************************************\n");
PDEBUG("* Read frame %lu\n", frame);
PDEBUG("*****************************************\n");
////////////////////////////////////////////////////////////////
// Proccessing data
////////////////////////////////////////////////////////////////
flowgraph->run();
/* Check every once in a while if the remote control
* is still working */
if (rcs.get_no_controllers() > 0 && (frame % 250) == 0) {
rcs.check_faults();
}
}
if (framesize == 0) {
fprintf(stderr, "End of file reached.\n");
}
else {
fprintf(stderr, "Input read error.\n");
}
running = 0;
}
} catch (std::exception& e) {
fprintf(stderr, "EXCEPTION: %s\n", e.what());
ret = -1;
}
END_MAIN:
////////////////////////////////////////////////////////////////////////
// Cleaning things
////////////////////////////////////////////////////////////////////////
fprintf(stderr, "\n\n");
fprintf(stderr, "%lu 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");
logger.level(info) << "Terminating";
return ret;
}