/*
  * VC3/DNxHD decoder.

  * Copyright (c) 2007 SmartJog S.A., Baptiste Coudurier <baptiste dot coudurier at smartjog dot com>

  *
  * This file is part of FFmpeg.
  *
  * FFmpeg 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.
  *
  * FFmpeg 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 FFmpeg; if not, write to the Free Software
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */
 

 #ifndef AVCODEC_DNXHDDATA_H
 #define AVCODEC_DNXHDDATA_H

 #include <stdint.h>

 #include "avcodec.h"

 #include "libavutil/internal.h"

 #include "libavutil/intreadwrite.h"

 /** Additional profile info flags */
 #define DNXHD_INTERLACED   (1<<0)

 #define DNXHD_MBAFF        (1<<1)
 #define DNXHD_444          (1<<2)

 /** Frame headers, extra 0x00 added to end for parser */
 #define DNXHD_HEADER_INITIAL 0x000002800100
 #define DNXHD_HEADER_444     0x000002800200
 

 /** Indicate that a CIDEntry value must be read in the bitstream */
 #define DNXHD_VARIABLE 0
 

 typedef struct CIDEntry {

     int cid;
     unsigned int width, height;
     unsigned int frame_size;
     unsigned int coding_unit_size;

     uint16_t flags;

     int index_bits;
     int bit_depth;

     int eob_index;

     const uint8_t *luma_weight, *chroma_weight;

     const uint8_t *dc_codes, *dc_bits;
     const uint16_t *ac_codes;

     const uint8_t *ac_bits, *ac_info;

     const uint16_t *run_codes;
     const uint8_t *run_bits, *run;

     int bit_rates[5]; ///< Helper to choose variants, rounded to nearest 5Mb/s

     AVRational packet_scale;

 } CIDEntry;
 

 extern const CIDEntry ff_dnxhd_cid_table[];

 
 int ff_dnxhd_get_cid_table(int cid);

 int ff_dnxhd_find_cid(AVCodecContext *avctx, int bit_depth);

 void ff_dnxhd_print_profiles(AVCodecContext *avctx, int loglevel);

 static av_always_inline uint64_t ff_dnxhd_check_header_prefix_hr(uint64_t prefix)
 {
     uint64_t data_offset = prefix >> 16;
     if ((prefix & 0xFFFF0000FFFFLL) == 0x0300 &&
          data_offset >= 0x0280 && data_offset <= 0x2170 &&

          (data_offset & 3) == 0)

         return prefix;
     return 0;
 }
 

 static av_always_inline uint64_t ff_dnxhd_check_header_prefix(uint64_t prefix)
 {
     if (prefix == DNXHD_HEADER_INITIAL ||
         prefix == DNXHD_HEADER_444     ||

         ff_dnxhd_check_header_prefix_hr(prefix))

         return prefix;
     return 0;
 }
 

 static av_always_inline uint64_t ff_dnxhd_parse_header_prefix(const uint8_t *buf)
 {
     uint64_t prefix = AV_RB32(buf);
     prefix = (prefix << 16) | buf[4] << 8;
     return ff_dnxhd_check_header_prefix(prefix);
 }
 

 static av_always_inline int ff_dnxhd_get_hr_frame_size(int cid, int w, int h)
 {
     int result, i = ff_dnxhd_get_cid_table(cid);
 
     if (i < 0)
         return i;
 
     result = ((h + 15) / 16) * ((w + 15) / 16) * (int64_t)ff_dnxhd_cid_table[i].packet_scale.num / ff_dnxhd_cid_table[i].packet_scale.den;
     result = (result + 2048) / 4096 * 4096;
 
     return FFMAX(result, 8192);
 }
 

 int avpriv_dnxhd_get_frame_size(int cid);

 int avpriv_dnxhd_get_interlaced(int cid);

 #endif /* AVCODEC_DNXHDDATA_H */