rbutil/rbutilqt/base/encoders.cpp

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  *
   9  *   Copyright (C) 2007 by Dominik Wenger
  10  *   $Id$
  11  *
  12  * All files in this archive are subject to the GNU General Public License.
  13  * See the file COPYING in the source tree root for full license agreement.
  14  *
  15  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  16  * KIND, either express or implied.
  17  *
  18  ****************************************************************************/
  19
  20 #include "encoders.h"
  21 #include "utils.h"
  22 #include "rbsettings.h"
  23
  24 /*********************************************************************
  25 * Encoder Base
  26 **********************************************************************/
  27 QMap<QString,QString> EncBase::encoderList;
  28
  29 EncBase::EncBase(QObject *parent): EncTtsSettingInterface(parent)
  30 {
  31
  32 }
  33
  34 // initialize list of encoders
  35 void EncBase::initEncodernamesList()
  36 {
  37     encoderList["rbspeex"] = "Rockbox Speex Encoder";
  38     encoderList["lame"] = "Lame Mp3 Encoder";
  39 }
  40
  41
  42 // get nice name for a specific encoder
  43 QString EncBase::getEncoderName(QString encoder)
  44 {
  45     if(encoderList.isEmpty())
  46         initEncodernamesList();
  47     return encoderList.value(encoder);
  48 }
  49
  50
  51 // get a specific encoder object
  52 EncBase* EncBase::getEncoder(QObject* parent,QString encoder)
  53 {
  54     EncBase* enc;
  55     if(encoder == "lame")
  56     {
  57         enc = new EncExes(encoder,parent);
  58         return enc;
  59     }
  60     else  // rbspeex is default
  61     {
  62         enc = new EncRbSpeex(parent);
  63         return enc;
  64     }
  65 }
  66
  67
  68 QStringList EncBase::getEncoderList()
  69 {
  70     if(encoderList.isEmpty())
  71         initEncodernamesList();
  72     return encoderList.keys();
  73 }
  74
  75
  76 /*********************************************************************
  77 *  GEneral Exe Encoder
  78 **********************************************************************/
  79 EncExes::EncExes(QString name,QObject *parent) : EncBase(parent)
  80 {
  81     m_name = name;
  82
  83     m_TemplateMap["lame"] = "\"%exe\" %options \"%input\" \"%output\"";
  84
  85 }
  86
  87
  88
  89 void EncExes::generateSettings()
  90 {
  91     QString exepath =RbSettings::subValue(m_name,RbSettings::EncoderPath).toString();
  92     if(exepath == "") exepath = Utils::findExecutable(m_name);
  93
  94     insertSetting(eEXEPATH,new EncTtsSetting(this,EncTtsSetting::eSTRING,
  95             tr("Path to Encoder:"),exepath,EncTtsSetting::eBROWSEBTN));
  96     insertSetting(eEXEOPTIONS,new EncTtsSetting(this,EncTtsSetting::eSTRING,
  97             tr("Encoder options:"),RbSettings::subValue(m_name,RbSettings::EncoderOptions)));
  98 }
  99
 100 void EncExes::saveSettings()
 101 {
 102     RbSettings::setSubValue(m_name,RbSettings::EncoderPath,getSetting(eEXEPATH)->current().toString());
 103     RbSettings::setSubValue(m_name,RbSettings::EncoderOptions,getSetting(eEXEOPTIONS)->current().toString());
 104     RbSettings::sync();
 105 }
 106
 107 bool EncExes::start()
 108 {
 109     m_EncExec = RbSettings::subValue(m_name, RbSettings::EncoderPath).toString();
 110     m_EncOpts = RbSettings::subValue(m_name, RbSettings::EncoderOptions).toString();
 111
 112     m_EncTemplate = m_TemplateMap.value(m_name);
 113
 114     QFileInfo enc(m_EncExec);
 115     if(enc.exists())
 116     {
 117         return true;
 118     }
 119     else
 120     {
 121         return false;
 122     }
 123 }
 124
 125 bool EncExes::encode(QString input,QString output)
 126 {
 127     //qDebug() << "encoding..";
 128     QString execstring = m_EncTemplate;
 129
 130     execstring.replace("%exe",m_EncExec);
 131     execstring.replace("%options",m_EncOpts);
 132     execstring.replace("%input",input);
 133     execstring.replace("%output",output);
 134     qDebug() << "[EncExes] cmd: " << execstring;
 135     int result = QProcess::execute(execstring);
 136     return (result == 0) ? true : false;
 137 }
 138
 139
 140 bool EncExes::configOk()
 141 {
 142     QString path = RbSettings::subValue(m_name, RbSettings::EncoderPath).toString();
 143
 144     if (QFileInfo(path).exists())
 145         return true;
 146
 147     return false;
 148 }
 149
 150 /*********************************************************************
 151 *  RB SPEEX ENCODER
 152 **********************************************************************/
 153 EncRbSpeex::EncRbSpeex(QObject *parent) : EncBase(parent)
 154 {
 155
 156 }
 157
 158 void EncRbSpeex::generateSettings()
 159 {
 160     insertSetting(eVOLUME,new EncTtsSetting(this,EncTtsSetting::eDOUBLE,
 161         tr("Volume:"),RbSettings::subValue("rbspeex",RbSettings::EncoderVolume),1.0,10.0));
 162     insertSetting(eQUALITY,new EncTtsSetting(this,EncTtsSetting::eDOUBLE,
 163         tr("Quality:"),RbSettings::subValue("rbspeex",RbSettings::EncoderQuality),0,10.0));
 164     insertSetting(eCOMPLEXITY,new EncTtsSetting(this,EncTtsSetting::eINT,
 165         tr("Complexity:"),RbSettings::subValue("rbspeex",RbSettings::EncoderComplexity),0,10));
 166     insertSetting(eNARROWBAND,new EncTtsSetting(this,EncTtsSetting::eBOOL,
 167         tr("Use Narrowband:"),RbSettings::subValue("rbspeex",RbSettings::EncoderNarrowBand)));
 168 }
 169
 170 void EncRbSpeex::saveSettings()
 171 {
 172     //save settings in user config
 173     RbSettings::setSubValue("rbspeex",RbSettings::EncoderVolume,
 174                             getSetting(eVOLUME)->current().toDouble());
 175     RbSettings::setSubValue("rbspeex",RbSettings::EncoderQuality,
 176                             getSetting(eQUALITY)->current().toDouble());
 177     RbSettings::setSubValue("rbspeex",RbSettings::EncoderComplexity,
 178                             getSetting(eCOMPLEXITY)->current().toInt());
 179     RbSettings::setSubValue("rbspeex",RbSettings::EncoderNarrowBand,
 180                             getSetting(eNARROWBAND)->current().toBool());
 181
 182     RbSettings::sync();
 183 }
 184
 185 bool EncRbSpeex::start()
 186 {
 187
 188     // try to get config from settings
 189     quality = RbSettings::subValue("rbspeex", RbSettings::EncoderQuality).toDouble();
 190     complexity = RbSettings::subValue("rbspeex", RbSettings::EncoderComplexity).toInt();
 191     volume = RbSettings::subValue("rbspeex", RbSettings::EncoderVolume).toDouble();
 192     narrowband = RbSettings::subValue("rbspeex", RbSettings::EncoderNarrowBand).toBool();
 193
 194
 195     return true;
 196 }
 197
 198 bool EncRbSpeex::encode(QString input,QString output)
 199 {
 200     qDebug() << "[RbSpeex] Encoding " << input << " to "<< output;
 201     char errstr[512];
 202
 203     FILE *fin,*fout;
 204     if ((fin = fopen(input.toLocal8Bit(), "rb")) == NULL) {
 205         qDebug() << "[RbSpeex] Error: could not open input file\n";
 206         return false;
 207     }
 208     if ((fout = fopen(output.toLocal8Bit(), "wb")) == NULL) {
 209         qDebug() << "[RbSpeex] Error: could not open output file\n";
 210         fclose(fin);
 211         return false;
 212     }
 213
 214     int ret = encode_file(fin, fout, quality, complexity, narrowband, volume,
 215                       errstr, sizeof(errstr));
 216     fclose(fout);
 217     fclose(fin);
 218
 219     if (!ret) {
 220         /* Attempt to delete unfinished output */
 221         qDebug() << "[RbSpeex] Error:" << errstr;
 222         QFile(output).remove();
 223         return false;
 224     }
 225     return true;
 226 }
 227
 228 bool EncRbSpeex::configOk()
 229 {
 230     bool result=true;
 231     // check config
 232
 233     if(RbSettings::subValue("rbspeex", RbSettings::EncoderVolume).toDouble() <= 0)
 234         result =false;
 235
 236     if(RbSettings::subValue("rbspeex", RbSettings::EncoderQuality).toDouble() <= 0)
 237         result =false;
 238
 239     if(RbSettings::subValue("rbspeex", RbSettings::EncoderComplexity).toInt() <= 0)
 240         result =false;
 241
 242     return result;
 243 }
 244