mythccextractorplayer.h

Go to the documentation of this file.

// -*- Mode: c++ -*-

#ifndef MYTHCCEXTRACTORPLAYER_H
#define MYTHCCEXTRACTORPLAYER_H

#include <stdint.h>

#include <QStringList>
#include <QImage>
#include <QPoint>
#include <QHash>

#include "mythplayer.h"

class SRTWriter;

enum sub_types
{
    kSubCC608,
    kSubCC708,
    kSubDVB,
    kSubTeletext,
};

class MTV_PUBLIC OneSubtitle
{
  public:
    int64_t start_time;
    int length;
    bool is_text;
    QStringList text;
    QImage img;
    QPoint img_shift;

    OneSubtitle() :
        start_time(),
        length(-1),
        is_text(true),
        text(),
        img_shift(0, 0)
    {}

    static const int kDefaultLength;
};

typedef QHash<int, QList<OneSubtitle> > CC608StreamType;
typedef QHash<int, QList<OneSubtitle> > CC708StreamType;
typedef QHash<int, QList<OneSubtitle> > TeletextStreamType;
typedef QList<OneSubtitle> DVBStreamType;

class SRTStuff
{
  public:
    SRTStuff() {}
    virtual ~SRTStuff();
    QHash<int, SRTWriter*> srtwriters;
    QHash<int,int>         subs_num;
};

class CC608Stuff : public SRTStuff
{
  public:
    CC608Stuff() : reader(NULL) { }
    ~CC608Stuff();
    CC608Reader *reader;
    CC608StreamType subs;
};
typedef QHash<uint, CC608Stuff> CC608Info;

class CC708Stuff : public SRTStuff
{
  public:
    CC708Stuff() : reader(NULL) { }
    ~CC708Stuff();
    CC708Reader *reader;
    CC708StreamType subs;
};
typedef QHash<uint, CC708Stuff> CC708Info;

class TeletextExtractorReader;
class TeletextStuff : public SRTStuff
{
  public:
    TeletextStuff() : reader(NULL) { }
    ~TeletextStuff();
    TeletextExtractorReader *reader;
    TeletextStreamType subs;
};
typedef QHash<uint, TeletextStuff> TeletextInfo;

class DVBSubStuff
{
  public:
    DVBSubStuff() : reader(NULL), subs_num(0) { }
    ~DVBSubStuff();
    SubtitleReader *reader;
    int             subs_num;
    DVBStreamType   subs;
};
typedef QHash<uint, DVBSubStuff> DVBSubInfo;

typedef QHash<uint, SubtitleReader*> SubtitleReaders;

class MTV_PUBLIC MythCCExtractorPlayer : public MythPlayer
{
  public:
    MythCCExtractorPlayer(PlayerFlags flags, bool showProgress,
                          const QString &fileName);
    ~MythCCExtractorPlayer() {}

    bool run(void);

    virtual CC708Reader    *GetCC708Reader(uint id=0);
    virtual CC608Reader    *GetCC608Reader(uint id=0);
    virtual SubtitleReader *GetSubReader(uint id=0);
    virtual TeletextReader *GetTeletextReader(uint id=0);

  private:
    void IngestSubtitle(QList<OneSubtitle>&, const QStringList&);
    void IngestSubtitle(QList<OneSubtitle>&, const OneSubtitle&);

    enum { kProcessNormal = 0, kProcessFinalize = 1 };
    void Ingest608Captions(void);
    void Process608Captions(uint flags);

    void Ingest708Captions(void);
    void Ingest708Caption(uint streamId, uint serviceIdx, uint windowIdx,
                          uint start_row, uint start_column,
                          const CC708Window &win,
                          const vector<CC708String*> &content);
    void Process708Captions(uint flags);

    void IngestTeletext(void);
    void ProcessTeletext(void);

    void IngestDVBSubtitles(void);
    void ProcessDVBSubtitles(uint flags);

    void OnGotNewFrame(void);

  protected:
    CC608Info       m_cc608_info;
    CC708Info       m_cc708_info;
    TeletextInfo    m_ttx_info;
    DVBSubInfo      m_dvbsub_info;

    class Window
    {
      public:
        uint row;
        uint column;
        QStringList text;
    };
    typedef QHash<uint, QMap<int, Window> > WindowsOnService;
    QHash<uint, WindowsOnService > m_cc708_windows;

    double  m_curTime;
    uint64_t m_myFramesPlayed;
    bool    m_showProgress;
    QString m_fileName;
    QDir    m_workingDir;
    QString m_baseName;
};

#endif // MYTHCCEXTRACTORPLAYER_H

/* vim: set expandtab tabstop=4 shiftwidth=4: */

---