cc608decoder.h

Go to the documentation of this file.

// -*- Mode: c++ -*-
 
#ifndef CCDECODER_H_
#define CCDECODER_H_
 
#include <cstdint>
#include <ctime>
 
#include <array>
#include <vector>
 
#include <QString>
#include <QMutex>
#include <QChar>
 
#include "format.h"
 
using CC608Seen        = std::array<bool,4>;
using CC608ProgramType = std::array<QString,96>;
using CC608PerField    = std::array<int,2>;
using CC608PerMode     = std::array<int,8>;
 
class CC608Input
{
  public:
    virtual ~CC608Input() = default;
    virtual void AddTextData(unsigned char *buf, int len,
                             int64_t timecode, char type) = 0;
};
 
enum
{
    kHasMPAA       = 0x1,
    kHasTPG        = 0x2,
    kHasCanEnglish = 0x4,
    kHasCanFrench  = 0x8,
};
enum
{
    kRatingMPAA = 0,
    kRatingTPG,
    kRatingCanEnglish,
    kRatingCanFrench,
};
 
class CC608Decoder
{
  public:
    explicit CC608Decoder(CC608Input *ccr);
    CC608Decoder(const CC608Decoder& rhs);
    ~CC608Decoder();
 
    void FormatCC(int tc, int code1, int code2);
    void FormatCCField(int tc, int field, int data);
    bool FalseDup(int tc, int field, int data);
 
    void DecodeVPS(const unsigned char *buf);
    void DecodeWSS(const unsigned char *buf);
 
    void SetIgnoreTimecode(bool val) { m_ignoreTimeCode = val; }
 
    uint    GetRatingSystems(bool future) const;
    uint    GetRating(uint i, bool future) const;
    QString GetRatingString(uint i, bool future) const;
    QString GetProgramName(bool future) const;
    QString GetProgramType(bool future) const;
    QString GetXDS(const QString &key) const;
 
    void GetServices(uint seconds, CC608Seen& seen) const;
 
    static QString ToASCII(const QString &cc608, bool suppress_unknown);
 
  private:
    QChar CharCC(int code) const { return m_stdChar[code]; }
    void ResetCC(int mode);
    void BufferCC(int mode, int len, int clr);
    int NewRowCC(int mode, int len);
 
    QString XDSDecodeString(const std::vector<unsigned char>&buf,
                            uint start, uint end) const;
    bool XDSDecode(int field, int b1, int b2);
 
    bool XDSPacketParseProgram(const std::vector<unsigned char> &xds_buf,
                               bool future);
    bool XDSPacketParseChannel(const std::vector<unsigned char> &xds_buf);
    void XDSPacketParse(const std::vector<unsigned char> &xds_buf);
    bool XDSPacketCRC(const std::vector<unsigned char> &xds_buf);
 
    CC608Input     *m_reader                {nullptr};
 
    bool            m_ignoreTimeCode        {false};
 
    std::array<time_t,4> m_lastSeen         {0};
 
    // per-field
    CC608PerField   m_badVbi                { 0,  0};
    CC608PerField   m_lastTc                { 0,  0};
    CC608PerField   m_lastCode              {-1, -1};
    CC608PerField   m_lastCodeTc            { 0,  0};
    CC608PerField   m_ccMode                {-1, -1}; // 0=cc1/txt1, 1=cc2/txt2
    CC608PerField   m_xds                   { 0,  0};
    std::array<int,4> m_txtMode             { 0,  0,  0,  0};
 
    // per-mode state
    CC608PerMode    m_lastRow               {0};
    CC608PerMode    m_newRow                {0};
    CC608PerMode    m_newCol                {0};
    CC608PerMode    m_newAttr               {0}; // color+italic+underline
    CC608PerMode    m_timeCode              {0};
    CC608PerMode    m_row                   {0};
    CC608PerMode    m_col                   {0};
    CC608PerMode    m_rowCount              {0};
    CC608PerMode    m_style                 {0};
    CC608PerMode    m_lineCont              {0};
    CC608PerMode    m_resumeText            {0};
    CC608PerMode    m_lastClr               {0};
    std::array<QString,8> m_ccBuf;
 
    // translation table
    std::array<QChar,128> m_stdChar;
 
    // temporary buffer
    unsigned char  *m_rbuf                  {nullptr};
    std::array<int,2>  m_lastFormatTc       {0, 0};
    std::array<int,2>  m_lastFormatData     {0, 0};
 
    // VPS data
    std::array<char,20> m_vpsPrLabel        {0};
    std::array<char,20> m_vpsLabel          {0};
    int             m_vpsL                  {0};
 
    // WSS data
    uint            m_wssFlags              {0};
    bool            m_wssValid              {false};
 
    int             m_xdsCurService         {-1};
    std::array<std::vector<unsigned char>,7> m_xdsBuf;
    uint            m_xdsCrcPassed          {0};
    uint            m_xdsCrcFailed          {0};
 
    mutable QMutex  m_xdsLock               {QMutex::Recursive};
    std::array<uint,2> m_xdsRatingSystems   {0};
    std::array<std::array<uint,4>,2> m_xdsRating       {{}};
    std::array<QString,2>            m_xdsProgramName;
    std::array<std::vector<uint>,2>  m_xdsProgramType;
 
    QString         m_xdsNetCall;
    QString         m_xdsNetName;
    uint            m_xdsTsid               {0};
 
    CC608ProgramType  m_xdsProgramTypeString;
};
 
#endif