rbutil/rbutilqt/base/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 #include "rbsettings.h"
  22
  23 TalkFileCreator::TalkFileCreator(QObject* parent): QObject(parent)
  24 {
  25
  26 }
  27
  28 //! \brief Creates Talkfiles.
  29 //!
  30 //! \param logger A pointer to a Loggerobject
  31 bool TalkFileCreator::createTalkFiles()
  32 {
  33     m_abort = false;
  34     QString errStr;
  35
  36     emit logItem(tr("Starting Talk file generation"),LOGINFO);
  37     emit logProgress(0,0);
  38     QCoreApplication::processEvents();
  39
  40     // read in Maps of paths - file/dirnames
  41     emit logItem(tr("Reading Filelist..."),LOGINFO);
  42     if(createTalkList(m_dir) == false)
  43     {
  44         emit logItem(tr("Talk file creation aborted"),LOGERROR);
  45         doAbort();
  46         return false;
  47     }
  48     QCoreApplication::processEvents();
  49
  50     // generate entries
  51     {
  52         TalkGenerator generator(this);
  53         connect(&generator,SIGNAL(done(bool)),this,SIGNAL(done(bool)));
  54         connect(&generator,SIGNAL(logItem(QString,int)),this,SIGNAL(logItem(QString,int)));
  55         connect(&generator,SIGNAL(logProgress(int,int)),this,SIGNAL(logProgress(int,int)));
  56         connect(this,SIGNAL(aborted()),&generator,SLOT(abort()));
  57
  58         if(generator.process(&m_talkList) == TalkGenerator::eERROR)
  59         {
  60             doAbort();
  61             return false;
  62         }
  63     }
  64
  65     // Copying talk files
  66     emit logItem(tr("Copying Talkfiles..."),LOGINFO);
  67     if(copyTalkFiles(&errStr) == false)
  68     {
  69         emit logItem(errStr,LOGERROR);
  70         doAbort();
  71         return false;
  72     }
  73
  74     // Deleting left overs
  75     if( !cleanup())
  76         return false;
  77
  78     emit logItem(tr("Finished creating Talk files"),LOGOK);
  79     emit logProgress(1,1);
  80     emit done(false);
  81
  82     return true;
  83 }
  84
  85 //! \brief Strips everything after and including the last dot in a string. If there is no dot, nothing is changed
  86 //!
  87 //! \param filename The filename from which to strip the Extension
  88 //! \returns the modified string
  89 QString TalkFileCreator::stripExtension(QString filename)
  90 {
  91     if(filename.lastIndexOf(".") != -1)
  92         return filename.left(filename.lastIndexOf("."));
  93     else
  94         return filename;
  95 }
  96
  97 //! \brief Does needed Tasks when we need to abort. Cleans up Files. Stops the Logger, Stops TTS and Encoder
  98 //!
  99 void TalkFileCreator::doAbort()
 100 {
 101     cleanup();
 102     emit logProgress(0,1);
 103     emit done(true);
 104 }
 105 //! \brief creates a list of what to generate
 106 //!
 107 //! \param startDir The directory from which to start scanning
 108 bool TalkFileCreator::createTalkList(QDir startDir)
 109 {
 110     m_talkList.clear();
 111
 112      // create Iterator
 113     QDirIterator::IteratorFlags flags = QDirIterator::NoIteratorFlags;
 114     if(m_recursive)
 115         flags = QDirIterator::Subdirectories;
 116
 117     QDirIterator it(startDir,flags);
 118
 119     //create temp directory
 120     QDir tempDir(QDir::tempPath()+ "/talkfiles/");
 121     if(!tempDir.exists())
 122         tempDir.mkpath(QDir::tempPath()+ "/talkfiles/");
 123
 124     // read in Maps of paths - file/dirnames
 125     while (it.hasNext())
 126     {
 127         it.next();
 128         if(m_abort)
 129         {
 130             return false;
 131         }
 132
 133         QFileInfo fileInf = it.fileInfo();
 134
 135         // its a dir
 136         if(fileInf.isDir())
 137         {
 138             QDir dir = fileInf.dir();
 139
 140             // insert into List
 141             if(!dir.dirName().isEmpty() && m_talkFolders)
 142             {
 143                 TalkGenerator::TalkEntry entry;
 144                 entry.toSpeak = dir.dirName();
 145                 entry.wavfilename = QDir::tempPath()+ "/talkfiles/" + QCryptographicHash::hash(entry.toSpeak.toUtf8(),
 146                                         QCryptographicHash::Md5).toHex() + ".wav";
 147                 entry.talkfilename = QDir::tempPath()+ "/talkfiles/" + QCryptographicHash::hash(entry.toSpeak.toUtf8(),
 148                                         QCryptographicHash::Md5).toHex() + ".talk";
 149                 entry.target = dir.path() + "/_dirname.talk";
 150                 entry.voiced = false;
 151                 entry.encoded = false;
 152                 qDebug() << "toSpeak: " << entry.toSpeak << " target: " << entry.target << " intermediates: " <<
 153                             entry.wavfilename << entry.talkfilename;
 154                 m_talkList.append(entry);
 155             }
 156         }
 157         else  // its a File
 158         {
 159             // insert into List
 160             if( !fileInf.fileName().isEmpty() && !fileInf.fileName().endsWith(".talk") && m_talkFiles)
 161             {
 162                 //test if we should ignore this file
 163                 bool match = false;
 164                 for(int i=0; i < m_ignoreFiles.size();i++)
 165                 {
 166                     QRegExp rx(m_ignoreFiles[i].trimmed());
 167                     rx.setPatternSyntax(QRegExp::Wildcard);
 168                     if(rx.exactMatch(fileInf.fileName()))
 169                         match = true;
 170                 }
 171                 if(match)
 172                     continue;
 173
 174                 //generate entry
 175                 TalkGenerator::TalkEntry entry;
 176                 if(m_stripExtensions)
 177                     entry.toSpeak = stripExtension(fileInf.fileName());
 178                 else
 179                     entry.toSpeak = fileInf.fileName();
 180                 entry.wavfilename = QDir::tempPath()+ "/talkfiles/" + QCryptographicHash::hash(entry.toSpeak.toUtf8(),
 181                                         QCryptographicHash::Md5).toHex() + ".wav";
 182                 entry.talkfilename = QDir::tempPath()+ "/talkfiles/" + QCryptographicHash::hash(entry.toSpeak.toUtf8(),
 183                                         QCryptographicHash::Md5).toHex() + ".talk";
 184                 entry.target =  fileInf.path() + "/" + fileInf.fileName() + ".talk";
 185                 entry.voiced = false;
 186                 entry.encoded = false;
 187                 qDebug() << "toSpeak: " << entry.toSpeak << " target: " << entry.target << " intermediates: " <<
 188                             entry.wavfilename << entry.talkfilename;
 189                 m_talkList.append(entry);
 190             }
 191         }
 192         QCoreApplication::processEvents();
 193     }
 194     return true;
 195 }
 196
 197
 198 //! \brief copys Talkfiles from the temp dir to the target. Progress and installlog is handled inside
 199 //!
 200 //! \param errString Pointer to a QString where the error cause is written.
 201 //! \returns true on success, false on error or user abort
 202 bool TalkFileCreator::copyTalkFiles(QString* errString)
 203 {
 204     int progressMax = m_talkList.size();
 205     int m_progress = 0;
 206     emit logProgress(m_progress,progressMax);
 207
 208     QSettings installlog(m_mountpoint + "/.rockbox/rbutil.log", QSettings::IniFormat, 0);
 209     installlog.beginGroup("talkfiles");
 210
 211     for(int i=0; i < m_talkList.size(); i++)
 212     {
 213         if(m_abort)
 214         {
 215             *errString = tr("File copy aborted");
 216             return false;
 217         }
 218
 219         // skip not encoded files
 220         if(m_talkList[i].encoded == false)
 221         {
 222             emit logProgress(++m_progress,progressMax);
 223             continue; // this file was skipped in one of the previous steps
 224         }
 225         // remove target if it exists, and if we should overwrite it
 226         if(m_overwriteTalk && QFile::exists(m_talkList[i].target))
 227             QFile::remove(m_talkList[i].target);
 228
 229         // copying
 230         qDebug() << "copying " << m_talkList[i].talkfilename << "to" << m_talkList[i].target;
 231         if(!QFile::copy(m_talkList[i].talkfilename,m_talkList[i].target))
 232         {
 233             *errString = tr("Copying of %1 to %2 failed").arg(m_talkList[i].talkfilename).arg(m_talkList[i].target);
 234             return false;
 235         }
 236
 237         // add to installlog
 238         QString now = QDate::currentDate().toString("yyyyMMdd");
 239         installlog.setValue(m_talkList[i].target.remove(0,m_mountpoint.length()),now);
 240
 241         emit logProgress(++m_progress,progressMax);
 242         QCoreApplication::processEvents();
 243     }
 244     installlog.endGroup();
 245     installlog.sync();
 246     return true;
 247 }
 248
 249
 250 //! \brief Cleans up Files potentially left in the temp dir
 251 //!
 252 bool TalkFileCreator::cleanup()
 253 {
 254     emit logItem(tr("Cleaning up.."),LOGINFO);
 255
 256     for(int i=0; i < m_talkList.size(); i++)
 257     {
 258         if(QFile::exists(m_talkList[i].wavfilename))
 259                 QFile::remove(m_talkList[i].wavfilename);
 260         if(QFile::exists(m_talkList[i].talkfilename))
 261                 QFile::remove(m_talkList[i].talkfilename);
 262
 263         QCoreApplication::processEvents();
 264     }
 265     emit logItem(tr("Finished"),LOGINFO);
 266     return true;
 267 }
 268
 269 //! \brief slot, which is connected to the abort of the Logger. Sets a flag, so Creating Talkfiles ends at the next possible position
 270 //!
 271 void TalkFileCreator::abort()
 272 {
 273     m_abort = true;
 274     emit aborted();
 275 }
 276