rbutil/rbutilqt/voicefile.cpp

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  *
   9  *   Copyright (C) 2007 by Dominik Wenger
  10  *   $Id: voicefile.h 15932 2007-12-15 13:13:57Z domonoky $
  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 "voicefile.h"
  21
  22 #define STATE_INVALID 0
  23 #define STATE_PHRASE 1
  24 #define STATE_VOICE 2
  25
  26
  27 VoiceFileCreator::VoiceFileCreator(QObject* parent) :QObject(parent)
  28 {
  29     m_wavtrimThreshold=500;
  30 }
  31
  32 void VoiceFileCreator::abort()
  33 {
  34     m_abort = true;
  35 }
  36
  37 bool VoiceFileCreator::createVoiceFile(ProgressloggerInterface* logger)
  38 {
  39     m_abort = false;
  40     m_logger = logger;
  41     m_logger->addItem(tr("Starting Voicefile generation"),LOGINFO);
  42
  43     // test if tempdir exists
  44     if(!QDir(QDir::tempPath()+"/rbvoice/").exists())
  45     {
  46         QDir(QDir::tempPath()).mkdir("rbvoice");
  47     }
  48
  49     m_path = QDir::tempPath() + "/rbvoice/";
  50
  51     // read rockbox-info.txt
  52     QFile info(m_mountpoint+"/.rockbox/rockbox-info.txt");
  53     if(!info.open(QIODevice::ReadOnly))
  54     {
  55         m_logger->addItem(tr("failed to open rockbox-info.txt"),LOGERROR);
  56         return false;
  57     }
  58
  59     QString target, features,version;
  60     while (!info.atEnd()) {
  61         QString line = info.readLine();
  62
  63         if(line.contains("Target:"))
  64         {
  65             target = line.remove("Target:").trimmed();
  66         }
  67         else if(line.contains("Features:"))
  68         {
  69             features = line.remove("Features:").trimmed();
  70         }
  71         else if(line.contains("Version:"))
  72         {
  73             version = line.remove("Version:").trimmed();
  74             version = version.left(version.indexOf("-")).remove(0,1);
  75         }
  76     }
  77     info.close();
  78
  79     //prepare download url
  80     QUrl genlangUrl = deviceSettings->value("genlang_url").toString() +"?lang=" +m_lang+"&t="+target+"&rev="+version+"&f="+features;
  81
  82     qDebug() << "downloading " << genlangUrl;
  83
  84     //download the correct genlang output
  85     QTemporaryFile *downloadFile = new QTemporaryFile(this);
  86     downloadFile->open();
  87     filename = downloadFile->fileName();
  88     downloadFile->close();
  89     // get the real file.
  90     getter = new HttpGet(this);
  91     getter->setProxy(m_proxy);
  92     getter->setFile(downloadFile);
  93     getter->getFile(genlangUrl);
  94
  95     connect(getter, SIGNAL(done(bool)), this, SLOT(downloadDone(bool)));
  96     connect(getter, SIGNAL(downloadDone(int, bool)), this, SLOT(downloadRequestFinished(int, bool)));
  97     connect(getter, SIGNAL(dataReadProgress(int, int)), this, SLOT(updateDataReadProgress(int, int)));
  98     connect(m_logger, SIGNAL(aborted()), getter, SLOT(abort()));
  99     return true;
 100  }
 101
 102
 103 void VoiceFileCreator::downloadRequestFinished(int id, bool error)
 104 {
 105     qDebug() << "Install::downloadRequestFinished" << id << error;
 106     qDebug() << "error:" << getter->errorString();
 107
 108     downloadDone(error);
 109 }
 110
 111
 112 void VoiceFileCreator::downloadDone(bool error)
 113 {
 114     qDebug() << "Voice creator::downloadDone, error:" << error;
 115
 116     // update progress bar
 117     int max = m_logger->getProgressMax();
 118     if(max == 0) {
 119         max = 100;
 120         m_logger->setProgressMax(max);
 121     }
 122     m_logger->setProgressValue(max);
 123     if(getter->httpResponse() != 200 && !getter->isCached()) {
 124         m_logger->addItem(tr("Download error: received HTTP error %1.").arg(getter->httpResponse()),LOGERROR);
 125         m_logger->abort();
 126         return;
 127     }
 128     if(getter->isCached()) m_logger->addItem(tr("Cached file used."), LOGINFO);
 129     if(error) {
 130         m_logger->addItem(tr("Download error: %1").arg(getter->errorString()),LOGERROR);
 131         m_logger->abort();
 132         return;
 133     }
 134     else m_logger->addItem(tr("Download finished."),LOGOK);
 135     QApplication::processEvents();
 136
 137
 138     m_logger->setProgressMax(0);
 139     //open downloaded file
 140     QFile genlang(filename);
 141     if(!genlang.open(QIODevice::ReadOnly))
 142     {
 143         m_logger->addItem(tr("failed to open downloaded file"),LOGERROR);
 144         m_logger->abort();
 145         return;
 146     }
 147
 148     //tts
 149     m_tts = getTTS(userSettings->value("tts").toString());
 150     m_tts->setCfg(userSettings,deviceSettings);
 151
 152     QString errStr;
 153     if(!m_tts->start(&errStr))
 154     {
 155         m_logger->addItem(errStr,LOGERROR);
 156         m_logger->addItem(tr("Init of TTS engine failed"),LOGERROR);
 157         m_logger->abort();
 158         return;
 159     }
 160
 161     // Encoder
 162     m_enc = getEncoder(userSettings->value("encoder").toString());
 163     m_enc->setUserCfg(userSettings);
 164
 165     if(!m_enc->start())
 166     {
 167         m_logger->addItem(tr("Init of Encoder engine failed"),LOGERROR);
 168         m_tts->stop();
 169         m_logger->abort();
 170         return;
 171     }
 172
 173     QApplication::processEvents();
 174     connect(m_logger,SIGNAL(aborted()),this,SLOT(abort()));
 175
 176     //read in downloaded file
 177     QList<QPair<QString,QString> > voicepairs;
 178     QTextStream in(&genlang);
 179     in.setCodec("UTF-8");
 180     QString id, voice;
 181     bool idfound = false;
 182     bool voicefound=false;
 183     while (!in.atEnd())
 184     {
 185         QString line = in.readLine();
 186         if(line.contains("id:"))  //ID found
 187         {
 188             id = line.remove("id:").remove('"').trimmed();
 189             idfound = true;
 190         }
 191         else if(line.contains("voice:"))  // voice found
 192         {
 193             voice = line.remove("voice:").remove('"').trimmed();
 194             voicefound=true;
 195         }
 196
 197         if(idfound && voicefound)
 198         {
 199             voicepairs.append(QPair<QString,QString>(id,voice));
 200             idfound=false;
 201             voicefound=false;
 202         }
 203     }
 204     genlang.close();
 205
 206     // check for empty list
 207     if(voicepairs.size() == 0)
 208     {
 209         m_logger->addItem(tr("The downloaded file was empty!"),LOGERROR);
 210         m_logger->abort();
 211         m_tts->stop();
 212         return;
 213     }
 214
 215     m_logger->setProgressMax(voicepairs.size());
 216     m_logger->setProgressValue(0);
 217
 218     // create voice clips
 219     QStringList mp3files;
 220     for(int i=0; i< voicepairs.size(); i++)
 221     {
 222         if(m_abort)
 223         {
 224             m_logger->addItem("aborted.",LOGERROR);
 225             m_logger->abort();
 226             m_tts->stop();
 227             return;
 228         }
 229
 230         m_logger->setProgressValue(i);
 231
 232         QString wavname = m_path + "/" + voicepairs.at(i).first + ".wav";
 233         QString toSpeak = voicepairs.at(i).second;
 234         QString encodedname = m_path + "/" + voicepairs.at(i).first +".mp3";
 235
 236         // todo PAUSE
 237         if(voicepairs.at(i).first == "VOICE_PAUSE")
 238         {
 239             QFile::copy(":/builtin/builtin/VOICE_PAUSE.wav",m_path + "/VOICE_PAUSE.wav");
 240
 241         }
 242         else
 243         {
 244             if(toSpeak == "") continue;
 245
 246             m_logger->addItem(tr("creating ")+toSpeak,LOGINFO);
 247             QApplication::processEvents();
 248             m_tts->voice(toSpeak,wavname); // generate wav
 249         }
 250
 251         // todo strip
 252         char buffer[255];
 253
 254         wavtrim((char*)qPrintable(wavname),m_wavtrimThreshold,buffer,255);
 255
 256         // encode wav
 257         m_enc->encode(wavname,encodedname);
 258         // remove the wav file
 259         QFile::remove(wavname);
 260         // remember the mp3 file for later removing
 261         mp3files << encodedname;
 262     }
 263
 264
 265     //make voicefile
 266     FILE* ids2 = fopen(filename.toUtf8(), "r");
 267     if (ids2 == NULL)
 268     {
 269         m_logger->addItem(tr("Error opening downloaded file"),LOGERROR);
 270         m_logger->abort();
 271         return;
 272     }
 273
 274     FILE* output = fopen(QString(m_mountpoint + "/.rockbox/langs/" + m_lang + ".voice").toUtf8(), "wb");
 275     if (output == NULL)
 276     {
 277         m_logger->addItem(tr("Error opening output file"),LOGERROR);
 278         return;
 279     }
 280
 281     voicefont(ids2,m_targetid,(char*)(const char*)m_path.toUtf8(), output);
 282
 283     //remove .mp3 files
 284     for(int i=0;i< mp3files.size(); i++)
 285     {
 286         QFile::remove(mp3files.at(i));
 287     }
 288
 289     m_logger->setProgressMax(100);
 290     m_logger->setProgressValue(100);
 291     m_logger->addItem(tr("successfully created."),LOGOK);
 292     m_logger->abort();
 293 }
 294
 295 void VoiceFileCreator::updateDataReadProgress(int read, int total)
 296 {
 297     m_logger->setProgressMax(total);
 298     m_logger->setProgressValue(read);
 299     //qDebug() << "progress:" << read << "/" << total;
 300
 301 }
 302