summaryrefslogtreecommitdiffstats
path: root/rbutil/rbutilqt/base/talkgenerator.h
blob: 5fe036e2f1068a8e3333e234865562a6377e883d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/***************************************************************************
 *             __________               __   ___.
 *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
 *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
 *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
 *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
 *                     \/            \/     \/    \/            \/
 *
 *   Copyright (C) 2007 by Dominik Wenger
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
 * KIND, either express or implied.
 *
 ****************************************************************************/


#ifndef TALKGENERATOR_H
#define TALKGENERATOR_H

#include <QtCore>
#include "progressloggerinterface.h"

#include "encoderbase.h"
#include "ttsbase.h"

//! \brief Talk generator, generates .wav and .talk files out of a list.
class TalkGenerator :public QObject
{
    Q_OBJECT
public:
    enum Status
    {
        eOK,
        eWARNING,
        eERROR
    };

    struct TalkEntry
    {
        QString toSpeak;
        QString wavfilename;
        QString talkfilename;
        QString target;
        bool voiced;
        bool encoded;

      /* We need the following members because 
       * 1) the QtConcurrent entry points are all static methods (and we 
       * need to communicate with the TalkGenerator)
       * 2) we are not guaranteed to go through the list in any 
       * particular order, so we can't use the progress slot 
       * for error checking */
      struct
      {
        EncoderBase* encoder;
        TTSBase* tts;
        TalkGenerator* generator;
        int wavtrim;
      } refs;
    };

    TalkGenerator(QObject* parent);
    Status process(QList<TalkEntry>* list,int wavtrimth = -1);
    QString correctString(QString s);

public slots:
    void abort();
    void encProgress(int value);
    void ttsProgress(int value);
    void setLang(QString name);

signals:
    void done(bool);
    void logItem(QString, int); //! set logger item
    void logProgress(int, int); //! set progress bar.

private:
    QFutureWatcher<void> encFutureWatcher;
    QFutureWatcher<void> ttsFutureWatcher;
    void encFailEntry(const TalkEntry& entry);
    void ttsFailEntry(const TalkEntry& entry, TTSStatus status, QString error);

    Status voiceList(QList<TalkEntry>* list,int wavetrimth);
    Status encodeList(QList<TalkEntry>* list);

    static void encEntryPoint(TalkEntry& entry);
    static void ttsEntryPoint(TalkEntry& entry);

    TTSBase* m_tts;
    EncoderBase* m_enc;

    bool m_ttsWarnings;
    bool m_userAborted;
    QString m_lang;

    struct CorrectionItems
    {
        QString search;
        QString replace;
        QString modifier;
    };
    QList<struct CorrectionItems> m_corrections;
};


#endif