Fred Gleason c7c5e6727e 2019-01-17 Fred Gleason <fredg@paravelsystems.com>
* Fixed a bug in rdrender(1) that caused corruption when printing
	UTF-8 characters.
2019-01-17 18:19:39 -05:00

344 lines
9.8 KiB
C++

// rdrender.cpp
//
// Render a Rivendell log.
//
// (C) Copyright 2017-2018 Fred Gleason <fredg@paravelsystems.com>
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License version 2 as
// published by the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public
// License along with this program; if not, write to the Free Software
// Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
//
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <qapplication.h>
#include <qdir.h>
#include <qfile.h>
#include <rd.h>
#include <rdapplication.h>
#include <rdaudioconvert.h>
#include <rdaudioexport.h>
#include <rdaudioimport.h>
#include <rdcart.h>
#include <rdcmd_switch.h>
#include <rdconf.h>
#include <rdcut.h>
#include <rdescape_string.h>
#include <rddbheartbeat.h>
#include <rdlog.h>
#include <rdrenderer.h>
#include <rdsettings.h>
#include "rdrender.h"
MainObject::MainObject(QObject *parent)
:QObject(parent)
{
QString err_msg;
render_verbose=false;
render_first_line=-1;
render_last_line=-1;
render_ignore_stops=false;
render_cart_number=0;
render_cut_number=-1;
render_start_time=QTime::currentTime();
//
// Initialize Audio Settings
//
render_settings.setChannels(RDRENDER_DEFAULT_CHANNELS);
render_settings.setSampleRate(0);
render_settings.setFormat(RDRENDER_DEFAULT_FORMAT);
render_settings.setBitRate(RDRENDER_DEFAULT_BITRATE);
render_settings.setQuality(RDRENDER_DEFAULT_BITRATE);
render_settings.setNormalizationLevel(RDRENDER_DEFAULT_NORMALIZATION_LEVEL);
//
// Open the Database
//
rda=new RDApplication("rdrender","rdrender",RDRENDER_USAGE,this);
if(!rda->open(&err_msg)) {
fprintf(stderr,"rdrender: %s\n",(const char *)err_msg.toUtf8());
exit(1);
}
//
// Read Command Options
//
if(rda->cmdSwitch()->keys()<1) {
fprintf(stderr,"rdrender: you must specify a logname\n");
exit(256);
}
for(unsigned i=0;i<rda->cmdSwitch()->keys()-1;i++) {
bool ok=false;
if(rda->cmdSwitch()->key(i)=="--verbose") {
render_verbose=true;
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--bitrate") {
render_settings.setBitRate(rda->cmdSwitch()->value(i).toUInt(&ok));
if(!ok) {
fprintf(stderr,"rdrender: invalid --bitrate argument\n");
exit(1);
}
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--channels") {
render_settings.setChannels(rda->cmdSwitch()->value(i).toUInt(&ok));
if((!ok)||
(render_settings.channels()>2)||(render_settings.channels()==0)) {
fprintf(stderr,"rdrender: invalid --channels argument\n");
exit(1);
}
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--format") {
QString format=rda->cmdSwitch()->value(i);
ok=false;
if(format.lower()=="flac") {
render_settings.setFormat(RDSettings::Flac);
ok=true;
}
if(format.lower()=="mp2") {
render_settings.setFormat(RDSettings::MpegL2);
ok=true;
}
if(format.lower()=="mp3") {
render_settings.setFormat(RDSettings::MpegL3);
ok=true;
}
if(format.lower()=="pcm16") {
render_settings.setFormat(RDSettings::Pcm16);
ok=true;
}
if(format.lower()=="pcm24") {
render_settings.setFormat(RDSettings::Pcm24);
ok=true;
}
if(format.lower()=="vorbis") {
render_settings.setFormat(RDSettings::OggVorbis);
ok=true;
}
if(!ok) {
fprintf(stderr,"rdrender: unknown --format \"%s\"\n",
(const char *)format);
exit(1);
}
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--first-line") {
render_first_line=rda->cmdSwitch()->value(i).toInt(&ok);
if((!ok)|(render_first_line<0)) {
fprintf(stderr,"rdrender: invalid --first-line argument\n");
exit(1);
}
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--first-time") {
render_first_time=QTime::fromString(rda->cmdSwitch()->value(i));
if(!render_first_time.isValid()) {
fprintf(stderr,"rdrender: invalid --first-time argument\n");
exit(1);
}
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--ignore-stops") {
render_ignore_stops=true;
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--last-time") {
render_last_time=QTime::fromString(rda->cmdSwitch()->value(i));
if(!render_last_time.isValid()) {
fprintf(stderr,"rdrender: invalid --last-time argument\n");
exit(1);
}
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--last-line") {
render_last_line=rda->cmdSwitch()->value(i).toInt(&ok);
if((!ok)||(render_last_line<0)) {
fprintf(stderr,"rdrender: invalid --last-line argument\n");
exit(1);
}
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--normalization-level") {
render_settings.setNormalizationLevel(rda->cmdSwitch()->value(i).toInt(&ok));
if(!ok) {
fprintf(stderr,"rdrender: invalid --normalization-level argument\n");
exit(1);
}
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--quality") {
render_settings.setQuality(rda->cmdSwitch()->value(i).toUInt(&ok));
if(!ok) {
fprintf(stderr,"rdrender: invalid --quality argument\n");
exit(1);
}
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--samplerate") {
render_settings.setSampleRate(rda->cmdSwitch()->value(i).toUInt(&ok));
if(!ok) {
fprintf(stderr,"rdrender: invalid --samplerate argument\n");
exit(1);
}
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--start-time") {
render_start_time=QTime::fromString(rda->cmdSwitch()->value(i));
if(!render_start_time.isValid()) {
fprintf(stderr,"rdrender: invalid --start-time\n");
exit(1);
}
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--to-cart") {
QStringList f0=rda->cmdSwitch()->value(i).split(":");
if(f0.size()!=2) {
fprintf(stderr,"rdrender: invalid --to-cart argument\n");
exit(1);
}
render_cart_number=f0[0].toUInt(&ok);
if((!ok)||(render_cart_number>RD_MAX_CART_NUMBER)) {
fprintf(stderr,"rdrender: invalid cart number in --to-cart argument\n");
exit(1);
}
render_cut_number=f0[1].toInt(&ok);
if((!ok)||(render_cut_number>RD_MAX_CUT_NUMBER)||(render_cut_number<1)) {
fprintf(stderr,"rdrender: invalid cut number in --to-cart argument\n");
exit(1);
}
rda->cmdSwitch()->setProcessed(i,true);
}
if(rda->cmdSwitch()->key(i)=="--to-file") {
render_to_file=rda->cmdSwitch()->value(i);
rda->cmdSwitch()->setProcessed(i,true);
}
if(!rda->cmdSwitch()->processed(i)) {
fprintf(stderr,"rdrepld: unknown command option \"%s\"\n",
(const char *)rda->cmdSwitch()->key(i));
exit(2);
}
}
//
// Sanity Checks
//
if((render_last_line>=0)&&(render_first_line>=0)&&
(render_last_line<render_last_line)) {
fprintf(stderr,"rdrender: --last-line must be greater than --first-line\n");
exit(1);
}
if(!RDAudioConvert::settingsValid(&render_settings)) {
fprintf(stderr,"rdrender: invalid audio settings\n");
exit(1);
}
if(render_to_file.isEmpty()&&
((render_cart_number==0)||(render_cut_number==-1))) {
fprintf(stderr,"rdrender: you must specify exactly one --to-cart or --to-file option\n");
exit(1);
}
render_logname=rda->cmdSwitch()->key(rda->cmdSwitch()->keys()-1);
if(render_start_time.isNull()) {
render_start_time=QTime(0,0,0,1);
}
//
// RIPC Connection
//
connect(rda,SIGNAL(userChanged()),this,SLOT(userData()));
rda->ripc()->
connectHost("localhost",RIPCD_TCP_PORT,rda->config()->password());
//
// System Configuration
//
if(render_settings.sampleRate()==0) {
render_settings.setSampleRate(rda->system()->sampleRate());
}
}
void MainObject::userData()
{
//
// Get User Context
//
disconnect(rda->ripc(),SIGNAL(userChanged()),this,SLOT(userData()));
//
// Open Log
//
RDLog *log=new RDLog(render_logname);
if(!log->exists()) {
fprintf(stderr,"rdrender: no such log\n");
exit(1);
}
RDLogEvent *log_event=new RDLogEvent(render_logname);
log_event->load();
//
// Render It
//
QString err_msg;
RDRenderer *r=new RDRenderer(this);
connect(r,SIGNAL(progressMessageSent(const QString &)),
this,SLOT(printProgressMessage(const QString &)));
if(render_to_file.isEmpty()) {
if(!r->renderToCart(render_cart_number,render_cut_number,log_event,
&render_settings,render_start_time,
render_ignore_stops,&err_msg,render_first_line,
render_last_line,render_first_time,render_last_time)) {
fprintf(stderr,"rdrender: %s\n",(const char *)err_msg);
exit(1);
}
}
else {
if(!r->renderToFile(render_to_file,log_event,&render_settings,
render_start_time,render_ignore_stops,
&err_msg,render_first_line,render_last_line,
render_first_time,render_last_time)) {
fprintf(stderr,"rdrender: %s\n",(const char *)err_msg);
exit(1);
}
}
QStringList warnings=r->warnings();
for(int i=0;i<warnings.size();i++) {
fprintf(stderr,"%s: %s\n",(const char *)tr("WARNING"),
(const char *)warnings[i]);
}
exit(0);
}
void MainObject::printProgressMessage(const QString &msg)
{
if(render_verbose) {
fprintf(stderr,"%s\n",(const char *)msg.toUtf8());
}
}
int main(int argc,char *argv[])
{
QApplication a(argc,argv,false);
new MainObject();
return a.exec();
}