// -*- Mode: c++ -*-
// Copyright (c) 2003-2004, Daniel Thor Kristjansson
#ifndef _MPEG_DESCRIPTORS_H_
#define _MPEG_DESCRIPTORS_H_

#include <cassert>
#include "iso639.h"

typedef vector<const unsigned char*> desc_list_t;

class DescriptorID
{
  public:
    enum
    {
        // MPEG
        video                       = 0x02,
        audio                       = 0x03,
        hierarchy                   = 0x04,
        registration                = 0x05,
        data_stream_alignment       = 0x06,
        target_background_grid      = 0x07,
        video_window                = 0x08,
        conditional_access          = 0x09,
        ISO_639_language            = 0x0A,
        system_clock                = 0x0B,
        multiplex_buffer_utilization= 0x0C,
        copyright                   = 0x0D,
        maximum_bitrate             = 0x0E,
        private_data_indicator      = 0x0F,
        smoothing_buffer            = 0x10,
        STD                         = 0x11,
        IBP                         = 0x12,
        mpeg4_video                 = 0x1B,
        mpeg4_audio                 = 0x1C,
        IOD                         = 0x1D,
        SL                          = 0x1E,
        FMC                         = 0x1F,
        external_es_id              = 0x20,
        mux_code                    = 0x21,
        fmx_buffer_size             = 0x22,
        multiplex_buffer            = 0x23,
        content_labeling            = 0x24,
        metadata_pointer            = 0x25,
        metadata                    = 0x26,
        metadata_std                = 0x27,
        avc_video                   = 0x28,
        ipmp                        = 0x29,
        avc_timing__hrd             = 0x30,

        // DVB
        network_name                = 0x40,
        service_list                = 0x41,
        dvb_stuffing                = 0x42,
        satellite_delivery_system   = 0x43,
        cable_delivery_system       = 0x44,
        VBI_data                    = 0x45,
        VBI_teletext                = 0x46,
        bouquet_name                = 0x47,
        service                     = 0x48,
        country_availability        = 0x49,
        linkage                     = 0x4A,
        NVOD_reference              = 0x4B,
        dvb_time_shifted_service    = 0x4C,
        short_event                 = 0x4D,
        extended_event              = 0x4E,
        time_shifted_event          = 0x4F,
        component                   = 0x50,
        mosaic                      = 0x51,
        stream_identifier           = 0x52,
        CA_identifier               = 0x53,
        content                     = 0x54,
        parental_rating             = 0x55,
        teletext                    = 0x56,
        telephone                   = 0x57,
        local_time_offset           = 0x58,
        subtitling                  = 0x59,
        terrestrial_delivery_system = 0x5A,
        multilingual_network_name   = 0x5B,
        multilingual_bouquet_name   = 0x5C,
        multilingual_service_name   = 0x5D,
        multilingual_component      = 0x5E,
        private_data_specifier      = 0x5F,
        service_move                = 0x60,
        short_smoothing_buffer      = 0x61,
        frequency_list              = 0x62,
        partial_transport_stream    = 0x63,
        data_broadcast              = 0x64,
        scrambling                  = 0x65,
        data_broadcast_id           = 0x66,
        transport_stream            = 0x67,
        DSNG                        = 0x68,
        PDC                         = 0x69,
        AC3                         = 0x6A,
        ancillary_data              = 0x6B,
        cell_list                   = 0x6C,
        cell_frequency_link         = 0x6D,
        announcement_support        = 0x6E,
        application_signalling      = 0x6F,
        adaptation_field_data       = 0x70,
        service_identifier          = 0x71,
        service_availability        = 0x72,
        default_authority           = 0x73,
        related_content             = 0x74,
        TVA_id                      = 0x75,
        dvb_content_identifier      = 0x76,
        time_slice_fec_identifier   = 0x77,
        ECM_repetition_rate         = 0x78,

        // private
        dvb_uk_channel_list         = 0x83,

        // ATSC
        atsc_stuffing               = 0x80,
        audio_stream                = 0x81,
        caption_service             = 0x86,
        content_advisory            = 0x87,

        // Dish Network
        dish_event_name             = 0x91,
        dish_event_description      = 0x92,

        // ATSC
        extended_channel_name       = 0xA0,
        service_location            = 0xA1,
        atsc_time_shifted_service   = 0xA2,
        component_name              = 0xA3,
        DCC_departing_request       = 0xA8,
        DCC_arriving_request        = 0xA9,
        DRM_control                 = 0xAA,
        atsc_content_identifier     = 0xB6,

        // private premiere.de
        premiere_content_order        = 0xF0,
        premiere_parental_information = 0xF1,
        premiere_content_transmission = 0xF2,
    };
};

class MPEGDescriptor
{
  public:
    operator const unsigned char*() const { return _data; }

    MPEGDescriptor(const unsigned char* data) : _data(data) { ; }
    virtual ~MPEGDescriptor() {}

    uint DescriptorTag() const    { return _data[0]; }
    QString DescriptorTagString() const;
    uint DescriptorLength() const { return _data[1]; }
    static desc_list_t Parse(const unsigned char* data, uint len);
    static desc_list_t ParseAndExclude(const unsigned char* data, uint len,
                                       int descriptorid);
    static desc_list_t ParseOnlyInclude(const unsigned char* data, uint len,
                                        int descriptorid);

    static const unsigned char* Find(const desc_list_t &parsed, uint desc_tag);
    static desc_list_t FindAll(const desc_list_t &parsed, uint desc_tag);

    static const unsigned char* FindBestMatch(
        const desc_list_t &parsed, uint desc_tag, QMap<uint,uint> &langPref);
    static desc_list_t FindBestMatches(
        const desc_list_t &parsed, uint desc_tag, QMap<uint,uint> &langPref);

    virtual QString toString() const;
  protected:
    const unsigned char* _data;
};

// a_52a.pdf p119, Table A1
class RegistrationDescriptor : public MPEGDescriptor
{
  public:
    RegistrationDescriptor(const unsigned char* data) : MPEGDescriptor(data)
    {
        assert(DescriptorID::registration == DescriptorTag());
        if (0x04 != DescriptorLength())
        {
            //cerr<<"Registration Descriptor length != 4 !!!!"<<endl;
        }
    }

    uint FormatIdentifier(void) const
        { return (_data[2]<<24) | (_data[3]<<16) | (_data[4]<<8) | _data[5]; }
    QString FormatIdentifierString(void) const
    {
        return QString("") + QChar(_data[2]) + QChar(_data[3]) +
            QChar(_data[4]) + QChar(_data[5]);
    }
    QString toString() const;
};

class ConditionalAccessDescriptor : public MPEGDescriptor
{
  public:
    ConditionalAccessDescriptor(const unsigned char* data) : MPEGDescriptor(data)
    {
        assert(DescriptorID::conditional_access == DescriptorTag());
    }
    uint SystemID(void) const { return  _data[2] << 8 | _data[3]; }
    uint PID(void) const      { return (_data[4] & 0x1F) << 8 | _data[5]; }
    uint DataSize(void) const { return DescriptorLength() - 4; }
    const unsigned char* Data(void) const { return _data+6; }
    QString toString() const;
};

class ISO639LanguageDescriptor : public MPEGDescriptor
{
  public:
    ISO639LanguageDescriptor(const unsigned char* data) : MPEGDescriptor(data)
    {
        assert(DescriptorID::ISO_639_language == DescriptorTag());
    }
    const unsigned char* CodeRaw() const { return &_data[2]; }

    int LanguageKey(void) const
        { return iso639_str3_to_key(&_data[2]); }
    QString LanguageString(void) const
        { return iso639_key_to_str3(LanguageKey()); }
    int CanonicalLanguageKey(void) const
        { return iso639_key_to_canonical_key(LanguageKey()); }
    QString CanonicalLanguageString(void) const
        { return iso639_key_to_str3(CanonicalLanguageKey()); }
    QString toString() const;
};

/// ISO 13818-1:2000/Amd.3:2004 page 11
class AVCVideoDescriptor : public MPEGDescriptor
{
  public:
    AVCVideoDescriptor(const unsigned char* data) : MPEGDescriptor(data)
    {
    //       Name             bits  loc  expected value
    // descriptor_tag           8   0.0       0x
        assert(DescriptorID::avc_video == DescriptorTag());
    // descriptor_length        8   1.0
    }
    // profile_idc              8   2.0
    uint ProfileIDC(void)         const { return _data[2]; }
    // constraint_set0_flag     1   3.0
    bool ConstaintSet0(void)      const { return _data[3]&0x80; }
    // constraint_set1_flag     1   3.1
    bool ConstaintSet1(void)      const { return _data[3]&0x40; }
    // constraint_set2_flag     1   3.2
    bool ConstaintSet2(void)      const { return _data[3]&0x20; }
    // AVC_compatible_flags     5   3.3
    uint AVCCompatible(void)      const { return _data[3]&0x1f; }
    // level_idc                8   4.0
    uint LevelIDC(void)           const { return _data[4]; }
    // AVC_still_present        1   5.0
    bool AVCStill(void)           const { return _data[5]&0x80; }
    // AVC_24_hour_picture_flag 1   5.1
    bool AVC24HourPicture(void)   const { return _data[5]&0x40; }
    // reserved 6 bslbf
    QString toString() const;
};

/// ISO 13818-1:2000/Amd.3:2004 page 12
class AVCTimingAndHRDDescriptor : public MPEGDescriptor
{
    AVCTimingAndHRDDescriptor(const unsigned char* data) : MPEGDescriptor(data)
    {
    //       Name             bits  loc  expected value
    // descriptor_tag           8   0.0       0x
        assert(DescriptorID::avc_timing__hrd == DescriptorTag());
    // descriptor_length        8   1.0
    }
    // hrd_management_valid     1   2.0
    bool HRDManagementValid(void)      const { return _data[2]&0x80; }
    // reserved                 6   2.1
    // picture_and_timing_info_present 1 2.7
    bool HasPictureAndTimingInfo(void) const { return _data[2]&0x01;}
    // if (picture_and_timing_info_present) {
    //   90kHz_flag             1   3.0
    //   reserved               7   3.1
    //   if (90kHz_flag == '0') {
    //     N                   32   4.0 uimsbf
    //     K                   32   8.0 uimsbf
    //   }
    //   num_units_in_tick     32 (90khz)?4.0:12.0 uimsbf
    // }
    // fixed_frame_rate_flag    1 (pict_info)?((90khz)?8.0:16.0):3.0
    // temporal_poc_flag        1 (pict_info)?((90khz)?8.1:16.1):3.1
    // picture_to_display_conversion_flag 1 (pict_info)?((90khz)?8.2:16.2):3.2
    // reserved 5 bslbf
};

#endif // _MPEG_DESCRIPTORS_H_
