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 bool TalkFileCreator::createTalkFiles(ProgressloggerInterface* logger)
  28 {
  29     m_abort = false;
  30     m_logger = logger;
  31     m_logger->addItem(tr("Starting Talk file generation"),LOGINFO);
  32
  33     //tts
  34     m_tts = TTSBase::getTTS(settings->curTTS());
  35     m_tts->setCfg(settings);
  36
  37     QString errStr;
  38     if(!m_tts->start(&errStr))
  39     {
  40         m_logger->addItem(errStr,LOGERROR);
  41         m_logger->addItem(tr("Init of TTS engine failed"),LOGERROR);
  42         m_logger->abort();
  43         return false;
  44     }
  45
  46     // Encoder
  47     m_enc = EncBase::getEncoder(settings->curEncoder());
  48     m_enc->setCfg(settings);
  49
  50     if(!m_enc->start())
  51     {
  52         m_logger->addItem(tr("Init of Encoder engine failed"),LOGERROR);
  53         m_logger->abort();
  54         m_tts->stop();
  55         return false;
  56     }
  57
  58     QCoreApplication::processEvents();
  59
  60     connect(logger,SIGNAL(aborted()),this,SLOT(abort()));
  61     m_logger->setProgressMax(0);
  62
  63     QDirIterator::IteratorFlags flags = QDirIterator::NoIteratorFlags;
  64     if(m_recursive)
  65         flags = QDirIterator::Subdirectories;
  66
  67     QDirIterator it(m_dir,flags);
  68     QSettings installlog(m_mountpoint + "/.rockbox/rbutil.log", QSettings::IniFormat, 0);
  69     installlog.beginGroup("talkfiles");
  70     // iterate over all entrys
  71     while (it.hasNext())
  72     {
  73         if(m_abort)
  74         {
  75             m_logger->addItem(tr("Talk file creation aborted"),LOGERROR);
  76             m_logger->abort();
  77             m_tts->stop();
  78             return false;
  79         }
  80
  81         QCoreApplication::processEvents();
  82         QFileInfo fileInf = it.fileInfo();
  83         QString toSpeak;
  84         QString filename;
  85         QString wavfilename;
  86         QString filepath;
  87
  88         QString path = fileInf.filePath();
  89         qDebug() << path;
  90
  91         if( path.endsWith("..") || path.endsWith(".") || path.endsWith(".talk") )
  92         {
  93             it.next();
  94             continue;
  95         }
  96
  97         //! if it is a dir
  98         if(fileInf.isDir())
  99         {
 100             // skip entry if folder talking isnt enabled
 101             if(m_talkFolders == false)
 102             {
 103               it.next();
 104               continue;
 105             }
 106
 107             toSpeak = fileInf.fileName();
 108
 109             filepath = fileInf.filePath() + "/";
 110             filename = "_dirname.talk";
 111             qDebug() << "toSpeak: " << toSpeak << "filename: " << filename << " path: " <<filepath;
 112         }
 113         else   // if it is a file
 114         {
 115             // skip entry if file talking isnt enabled
 116             if(m_talkFiles == false)
 117             {
 118                it.next();
 119                continue;
 120             }
 121
 122             // create toSpeak string
 123             if(m_stripExtensions)
 124                 toSpeak = fileInf.baseName();
 125             else
 126                 toSpeak = fileInf.fileName();
 127             // create filename and path
 128             filepath = fileInf.absolutePath();
 129             filename = fileInf.fileName() + ".talk";
 130
 131         }
 132
 133         wavfilename = QDir::tempPath()+ "/"+ filename + ".wav";
 134
 135         QFileInfo filenameInf(filepath+filename);
 136         QFileInfo wavfilenameInf(wavfilename);
 137
 138         //! the actual generation of the .talk files
 139         if(!filenameInf.exists() || m_overwriteTalk)
 140         {
 141             if(!wavfilenameInf.exists() || m_overwriteWav)
 142             {
 143                 m_logger->addItem(tr("Voicing of %1").arg(toSpeak),LOGINFO);
 144                 if(!m_tts->voice(toSpeak,wavfilename))
 145                 {
 146                     m_logger->addItem(tr("Voicing of %s failed").arg(toSpeak),LOGERROR);
 147                     m_logger->abort();
 148                     m_tts->stop();
 149                     m_enc->stop();
 150
 151                     return false;
 152                 }
 153                 QCoreApplication::processEvents();
 154             }
 155             m_logger->addItem(tr("Encoding of %1").arg(toSpeak),LOGINFO);
 156             if(!m_enc->encode(wavfilename,filepath+filename))
 157             {
 158                 m_logger->addItem(tr("Encoding of %1 failed").arg(wavfilename),LOGERROR);
 159                 m_logger->abort();
 160                 m_tts->stop();
 161                 m_enc->stop();
 162
 163                 return false;
 164             }
 165             QCoreApplication::processEvents();
 166         }
 167
 168         //! remove the intermedia wav file, if requested
 169         QString now = QDate::currentDate().toString("yyyyMMdd");
 170         if(m_removeWav)
 171         {
 172             QFile wavfile(wavfilename);
 173             wavfile.remove();
 174             installlog.remove(wavfilename);
 175         }
 176         else
 177             installlog.setValue(wavfilename.remove(0,m_mountpoint.length()),now);
 178
 179         //! add the .talk file to the install log
 180         installlog.setValue(QString(filepath+filename).remove(0,m_mountpoint.length()),now);
 181         it.next();
 182     }
 183
 184     installlog.endGroup();
 185     m_tts->stop();
 186     m_logger->addItem(tr("Finished creating Talk files"),LOGOK);
 187     m_logger->setProgressMax(1);
 188     m_logger->setProgressValue(1);
 189     m_logger->abort();
 190
 191     return true;
 192
 193 }
 194
 195 void TalkFileCreator::abort()
 196 {
 197     m_abort = true;
 198 }
 199