rbutil/rbutilqt/talkfile.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 "talkfile.h"
  21
  22 TalkFileCreator::TalkFileCreator(QObject* parent): QObject(parent)
  23 {
  24
  25 }
  26
  27
  28 bool TalkFileCreator::initEncoder()
  29 {
  30     QFileInfo enc(m_EncExec);
  31     if(enc.exists())
  32     {
  33         return true;
  34     }
  35     else
  36     {
  37         return false;
  38     }
  39 }
  40
  41
  42 bool TalkFileCreator::createTalkFiles(ProgressloggerInterface* logger)
  43 {
  44     m_abort = false;
  45     m_logger = logger;
  46     m_logger->addItem("Starting Talk file generation",LOGINFO);
  47
  48     if(m_curTTS == "sapi")
  49         m_tts = new TTSSapi();
  50     else
  51         m_tts = new TTSExes();
  52
  53     m_tts->setTTSexe(m_TTSexec);
  54     m_tts->setTTsOpts(m_TTSOpts);
  55     m_tts->setTTsLanguage(m_TTSLanguage);
  56     m_tts->setTTsTemplate(m_curTTSTemplate);
  57
  58     if(!m_tts->start())
  59     {
  60         m_logger->addItem("Init of TTS engine failed",LOGERROR);
  61         return false;
  62     }
  63     if(!initEncoder())
  64     {
  65         m_logger->addItem("Init of encoder failed",LOGERROR);
  66         m_tts->stop();
  67         return false;
  68     }
  69     QApplication::processEvents();
  70
  71     connect(logger,SIGNAL(aborted()),this,SLOT(abort()));
  72     m_logger->setProgressMax(0);
  73     QDirIterator it(m_dir,QDirIterator::Subdirectories);
  74     QSettings installlog(m_mountpoint + "/.rockbox/rbutil.log", QSettings::IniFormat, 0);
  75     installlog.beginGroup("talkfiles");
  76     // iterate over all entrys
  77     while (it.hasNext())
  78     {
  79         if(m_abort)
  80         {
  81             m_logger->addItem("Talk file creation aborted",LOGERROR);
  82             m_tts->stop();
  83             return false;
  84         }
  85
  86         QApplication::processEvents();
  87         QFileInfo fileInf = it.fileInfo();
  88         QString toSpeak;
  89         QString filename;
  90         QString wavfilename;
  91
  92         //! skip dotdot and .talk files
  93         if(fileInf.fileName() == ".." || fileInf.suffix() == "talk")
  94         {
  95             it.next();
  96             continue;
  97         }
  98         //! if it is a dir
  99         if(fileInf.isDir())
 100         {
 101             // skip entry if folder talking isnt enabled
 102             if(m_talkFolders == false)
 103             {
 104               it.next();
 105               continue;
 106             }
 107             toSpeak = fileInf.absoluteDir().dirName();
 108             filename = fileInf.absolutePath() + "/_dirname.talk";
 109         }
 110         else   // if it is a file
 111         {
 112             // skip entry if file talking isnt enabled
 113             if(m_talkFiles == false)
 114             {
 115                it.next();
 116                continue;
 117             }
 118             if(m_stripExtensions)
 119                 toSpeak = fileInf.baseName();
 120             else
 121                 toSpeak = fileInf.fileName();
 122             filename = fileInf.absoluteFilePath() + ".talk";
 123         }
 124         wavfilename = filename + ".wav";
 125
 126         QFileInfo filenameInf(filename);
 127         QFileInfo wavfilenameInf(wavfilename);
 128
 129         //! the actual generation of the .talk files
 130         if(!filenameInf.exists() || m_overwriteTalk)
 131         {
 132             if(!wavfilenameInf.exists() || m_overwriteWav)
 133             {
 134                 m_logger->addItem("Voicing of " + toSpeak,LOGINFO);
 135                 if(!m_tts->voice(toSpeak,wavfilename))
 136                 {
 137                     m_logger->addItem("Voicing of " + toSpeak + " failed",LOGERROR);
 138                     m_logger->abort();
 139                     m_tts->stop();
 140                     return false;
 141                 }
 142             }
 143             m_logger->addItem("Encoding of " + toSpeak,LOGINFO);
 144             if(!encode(wavfilename,filename))
 145             {
 146                 m_logger->addItem("Encoding of " + wavfilename + " failed",LOGERROR);
 147                 m_logger->abort();
 148                 m_tts->stop();
 149                 return false;
 150             }
 151         }
 152
 153         //! remove the intermedia wav file, if requested
 154         QString now = QDate::currentDate().toString("yyyyMMdd");
 155         if(m_removeWav)
 156         {
 157             QFile wavfile(wavfilename);
 158             wavfile.remove();
 159             installlog.remove(wavfilename);
 160         }
 161         else
 162             installlog.setValue(wavfilename.remove(0,m_mountpoint.length()),now);
 163
 164         //! add the .talk file to the install log
 165         installlog.setValue(filename.remove(0,m_mountpoint.length()),now);
 166         it.next();
 167     }
 168
 169     installlog.endGroup();
 170     m_tts->stop();
 171     m_logger->addItem("Finished creating Talk files",LOGOK);
 172     m_logger->setProgressMax(1);
 173     m_logger->setProgressValue(1);
 174     m_logger->abort();
 175
 176     return true;
 177
 178 }
 179
 180 void TalkFileCreator::abort()
 181 {
 182     m_abort = true;
 183 }
 184
 185 bool TalkFileCreator::encode(QString input,QString output)
 186 {
 187     qDebug() << "encoding..";
 188     QString execstring = m_curEncTemplate;
 189
 190     execstring.replace("%exe",m_EncExec);
 191     execstring.replace("%options",m_EncOpts);
 192     execstring.replace("%input",input);
 193     execstring.replace("%output",output);
 194     qDebug() << execstring;
 195     QProcess::execute(execstring);
 196     return true;
 197
 198 }
 199
 200 bool TTSSapi::start()
 201 {
 202     QFileInfo tts(m_TTSexec);
 203     if(!tts.exists())
 204         return false;
 205
 206     // create the voice process
 207     QString execstring = m_TTSTemplate;
 208     execstring.replace("%exe",m_TTSexec);
 209     execstring.replace("%options",m_TTSOpts);
 210     execstring.replace("%lang",m_TTSLanguage);
 211     qDebug() << "init" << execstring;
 212     voicescript = new QProcess(NULL);
 213     voicescript->start(execstring);
 214     if(!voicescript->waitForStarted())
 215         return false;
 216     return true;
 217 }
 218
 219 bool TTSSapi::voice(QString text,QString wavfile)
 220 {
 221     QString query = "SPEAK\t"+wavfile+"\t"+text+"\r\n";
 222     qDebug() << "voicing" << query;
 223     voicescript->write(query.toLocal8Bit());
 224     voicescript->write("SYNC\tbla\r\n");
 225     voicescript->waitForReadyRead();
 226     return true;
 227 }
 228
 229 bool TTSSapi::stop()
 230 {
 231     QString query = "QUIT\r\n";
 232     voicescript->write(query.toLocal8Bit());
 233     voicescript->waitForFinished();
 234     delete voicescript;
 235     return true;
 236 }
 237
 238 bool TTSExes::start()
 239 {
 240     QFileInfo tts(m_TTSexec);
 241     qDebug() << "ttsexe init";
 242     if(tts.exists())
 243     {
 244         return true;
 245     }
 246     else
 247     {
 248         return false;
 249     }
 250 }
 251
 252 bool TTSExes::voice(QString text,QString wavfile)
 253 {
 254     QString execstring = m_TTSTemplate;
 255
 256     execstring.replace("%exe",m_TTSexec);
 257     execstring.replace("%options",m_TTSOpts);
 258     execstring.replace("%wavfile",wavfile);
 259     execstring.replace("%text",text);
 260     qDebug() << "voicing" << execstring;
 261     QProcess::execute(execstring);
 262     return true;
 263
 264 }