/*****************************************************************************

 * avci.h : AVC Intra SPS/PPS tables

 *****************************************************************************

 * Copyright (C) 2015 VLC authors and VideoLAN

 *

 * This program is free software; you can redistribute it and/or modify it

 * under the terms of the GNU Lesser General Public License as published by

 * the Free Software Foundation; either version 2.1 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

 * GNU Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public License

 * along with this program; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.

 *****************************************************************************/

#ifndef MP4_AVCI_H

# define MP4_AVCI_H

/* AVCi Tables derived from ffmpeg ff_generate_AVCi_extradata */

static const uint8_t AVCi_HIGH_422_1080p_sps[] = {

    0x7a, 0x10, 0x29, 0xb6, 0xd4, 0x20, 0x22, 0x33,

    0x19, 0xc6, 0x63, 0x23, 0x21, 0x01, 0x11, 0x98,

    0xce, 0x33, 0x19, 0x18, 0x21, 0x02, 0x56, 0xb9,

    0x3d, 0x7d, 0x7e, 0x4f, 0xe3, 0x3f, 0x11, 0xf1,

    0x9e, 0x08, 0xb8, 0x8c, 0x54, 0x43, 0xc0, 0x78,

    0x02, 0x27, 0xe2, 0x70, 0x1e, 0x30, 0x10, 0x10,

    0x14, 0x00, 0x00, 0x03, 0x00, 0x04, 0x00, 0x00,

    0x03, 0x00, 0xca, 0x10

};

static const uint8_t AVCi_HIGH_422_1080i_sps[] = {

    0x7a, 0x10, 0x29, 0xb6, 0xd4, 0x20, 0x22, 0x33,

    0x19, 0xc6, 0x63, 0x23, 0x21, 0x01, 0x11, 0x98,

    0xce, 0x33, 0x19, 0x18, 0x21, 0x03, 0x3a, 0x46,

    0x65, 0x6a, 0x65, 0x24, 0xad, 0xe9, 0x12, 0x32,

    0x14, 0x1a, 0x26, 0x34, 0xad, 0xa4, 0x41, 0x82,

    0x23, 0x01, 0x50, 0x2b, 0x1a, 0x24, 0x69, 0x48,

    0x30, 0x40, 0x2e, 0x11, 0x12, 0x08, 0xc6, 0x8c,

    0x04, 0x41, 0x28, 0x4c, 0x34, 0xf0, 0x1e, 0x01,

    0x13, 0xf2, 0xe0, 0x3c, 0x60, 0x20, 0x20, 0x28,

    0x00, 0x00, 0x03, 0x00, 0x08, 0x00, 0x00, 0x03,

    0x01, 0x94, 0x20

};

static const uint8_t AVCi_HIGH_422_720p_sps[] = {

    0x7a, 0x10, 0x29, 0xb6, 0xd4, 0x20, 0x2a, 0x33,

    0x1d, 0xc7, 0x62, 0xa1, 0x08, 0x40, 0x54, 0x66,

    0x3b, 0x8e, 0xc5, 0x42, 0x02, 0x10, 0x25, 0x64,

    0x2c, 0x89, 0xe8, 0x85, 0xe4, 0x21, 0x4b, 0x90,

    0x83, 0x06, 0x95, 0xd1, 0x06, 0x46, 0x97, 0x20,

    0xc8, 0xd7, 0x43, 0x08, 0x11, 0xc2, 0x1e, 0x4c,

    0x91, 0x0f, 0x01, 0x40, 0x16, 0xec, 0x07, 0x8c,

    0x04, 0x04, 0x05, 0x00, 0x00, 0x03, 0x00, 0x01,

    0x00, 0x00, 0x03, 0x00, 0x64, 0x84

};

static const uint8_t AVCi_HIGH_10_1080p_sps[] = {

    0x6e, 0x10, 0x28, 0xa6, 0xd4, 0x20, 0x32, 0x33,

    0x0c, 0x71, 0x18, 0x88, 0x62, 0x10, 0x19, 0x19,

    0x86, 0x38, 0x8c, 0x44, 0x30, 0x21, 0x02, 0x56,

    0x4e, 0x6f, 0x37, 0xcd, 0xf9, 0xbf, 0x81, 0x6b,

    0xf3, 0x7c, 0xde, 0x6e, 0x6c, 0xd3, 0x3c, 0x05,

    0xa0, 0x22, 0x7e, 0x5f, 0xfc, 0x00, 0x0c, 0x00,

    0x13, 0x8c, 0x04, 0x04, 0x05, 0x00, 0x00, 0x03,

    0x00, 0x01, 0x00, 0x00, 0x03, 0x00, 0x32, 0x84,

};

static const uint8_t AVCi_HIGH_10_1080i_sps[] = {

    0x6e, 0x10, 0x28, 0xa6, 0xd4, 0x20, 0x32, 0x33,

    0x0c, 0x71, 0x18, 0x88, 0x62, 0x10, 0x19, 0x19,

    0x86, 0x38, 0x8c, 0x44, 0x30, 0x21, 0x02, 0x56,

    0x4e, 0x6e, 0x61, 0x87, 0x3e, 0x73, 0x4d, 0x98,

    0x0c, 0x03, 0x06, 0x9c, 0x0b, 0x73, 0xe6, 0xc0,

    0xb5, 0x18, 0x63, 0x0d, 0x39, 0xe0, 0x5b, 0x02,

    0xd4, 0xc6, 0x19, 0x1a, 0x79, 0x8c, 0x32, 0x34,

    0x24, 0xf0, 0x16, 0x81, 0x13, 0xf7, 0xff, 0x80,

    0x02, 0x00, 0x01, 0xf1, 0x80, 0x80, 0x80, 0xa0,

    0x00, 0x00, 0x03, 0x00, 0x20, 0x00, 0x00, 0x06,

    0x50, 0x80

};

static const uint8_t AVCi_HIGH_10_720p_sps[] = {

    0x6e, 0x10, 0x20, 0xa6, 0xd4, 0x20, 0x32, 0x33,

    0x0c, 0x71, 0x18, 0x88, 0x62, 0x10, 0x19, 0x19,

    0x86, 0x38, 0x8c, 0x44, 0x30, 0x21, 0x02, 0x56,

    0x4e, 0x6f, 0x37, 0xcd, 0xf9, 0xbf, 0x81, 0x6b,

    0xf3, 0x7c, 0xde, 0x6e, 0x6c, 0xd3, 0x3c, 0x0f,

    0x01, 0x6e, 0xff, 0xc0, 0x00, 0xc0, 0x01, 0x38,

    0xc0, 0x40, 0x40, 0x50, 0x00, 0x00, 0x03, 0x00,

    0x10, 0x00, 0x00, 0x06, 0x48, 0x40

};

static const uint8_t AVCi_HIGH_422_1080x_pps[] = {

    0xce, 0x33, 0x48, 0xd0

};

static const uint8_t AVCi_HIGH_422_720p_pps[] = {

    0xce, 0x31, 0x12, 0x11

};

static const uint8_t AVCi_HIGH_10_pps[] = {

    0xee, 0x31, 0x12, 0x11

};

#define AVCI_ENTRY(a,b, A, B) {b, A, B, sizeof(A), sizeof(B), a}

#define AVCI_ENTRIES 6

static const struct

{

    bool b_interlace;

    const uint8_t *p_sps;

    const uint8_t *p_pps;

    uint8_t  i_sps;

    uint8_t  i_pps;

    uint16_t i_res;

} AVCi_lookup_table[AVCI_ENTRIES] = {

    AVCI_ENTRY(1920, true,  AVCi_HIGH_422_1080i_sps, AVCi_HIGH_422_1080x_pps),

    AVCI_ENTRY(1920, false, AVCi_HIGH_422_1080p_sps, AVCi_HIGH_422_1080x_pps),

    AVCI_ENTRY(1280, false, AVCi_HIGH_422_720p_sps,  AVCi_HIGH_422_720p_pps),

    AVCI_ENTRY(1440, true,  AVCi_HIGH_10_1080i_sps,  AVCi_HIGH_10_pps),

    AVCI_ENTRY(1440, false, AVCi_HIGH_10_1080p_sps,  AVCi_HIGH_10_pps),

    AVCI_ENTRY(960,  false, AVCi_HIGH_10_720p_sps,   AVCi_HIGH_10_pps),

};

static inline bool AVCi_lookup(uint16_t i_res, bool b_i,

                               const uint8_t **pp_sps, uint8_t *pi_sps,

                               const uint8_t **pp_pps, uint8_t *pi_pps)

{

    for(int i=0; i<AVCI_ENTRIES; i++)

    {

        if(AVCi_lookup_table[i].b_interlace == b_i && AVCi_lookup_table[i].i_res == i_res)

        {

            *pp_sps = AVCi_lookup_table[i].p_sps;

            *pi_sps = AVCi_lookup_table[i].i_sps;

            *pp_pps = AVCi_lookup_table[i].p_pps;

            *pi_pps = AVCi_lookup_table[i].i_pps;

            return true;

        }

    }

    return false;

}

#undef AVCI_ENTRY

#undef AVCI_ENTRIES

#if 0

static uint8_t * AVCi_create_avcC( uint16_t i_res, bool b_interlaced, int *pi_avcC )

{

    const uint8_t *p_pps, *p_sps;

    uint8_t i_sps, i_pps;

    uint8_t *p_data = NULL;

    if( AVCi_lookup( i_res, b_interlaced,

                     &p_sps, &i_sps, &p_pps, &i_pps ) )

    {

        int i_size = 5 + i_pps + i_sps + 2 + 2 * 3;

        if( (p_data = (uint8_t *) malloc(i_size)) )

        {

            *pi_avcC = i_size;

            /* Merge everything into avc decoder config record s4,1s,1p */

            p_data[0] = 0x01;

            memcpy(&p_data[1], p_sps, 3);

            p_data[4] = 0xff;

            p_data[5] = 0xe1; /* 1 sps */

            p_data[6] = 0x00; p_data[7] = i_sps + 1; /* sps NAL size */

            p_data[8] = 0x67; /* SPS_NAL Header */

            memcpy(&p_data[9], p_sps, i_sps);

            p_data[9 + i_sps] = 0x01; /* 1 sps */

            p_data[10 + i_sps] = 0x00; p_data[11 + i_sps] = i_pps + 1; /* pps NAL size */

            p_data[12 + i_sps] = 0x68; /* PPS_NAL Header */

            memcpy(&p_data[13 + i_sps], p_pps, i_pps);

        }

    }

    return p_data;

}

#endif

static uint8_t * AVCi_create_AnnexB( uint16_t i_res, bool b_interlaced, int *pi_avcC )

{

    const uint8_t *p_pps, *p_sps;

    uint8_t i_sps, i_pps;

    uint8_t *p_data = NULL;

    const uint8_t rgi_startcode[] = {0,0,0,1};

    if( AVCi_lookup( i_res, b_interlaced,

                     &p_sps, &i_sps, &p_pps, &i_pps ) )

    {

        int i_size = i_pps + i_sps + 2 * 5;

        if( (p_data = (uint8_t *) malloc(i_size)) )

        {

            *pi_avcC = i_size;

            memcpy(p_data, rgi_startcode, 4);

            p_data[4] = 0x67; /* SPS_NAL Header */

            memcpy(&p_data[5], p_sps, i_sps);

            memcpy(&p_data[5 + i_sps], rgi_startcode, 4);

            p_data[9 + i_sps] = 0x68; /* PPS_NAL Header */

            memcpy(&p_data[10 + i_sps], p_pps, i_pps);

        }

    }

    return p_data;

}

#endif