dnxhddata.h 3.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697
  1. /*
  2. * VC3/DNxHD decoder.
  3. * Copyright (c) 2007 SmartJog S.A., Baptiste Coudurier <baptiste dot coudurier at smartjog dot com>
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public
  9. * License as published by the Free Software Foundation; either
  10. * version 2.1 of the License, or (at your option) any later version.
  11. *
  12. * FFmpeg is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. #ifndef AVCODEC_DNXHDDATA_H
  22. #define AVCODEC_DNXHDDATA_H
  23. #include <stdint.h>
  24. #include "avcodec.h"
  25. #include "libavutil/internal.h"
  26. #include "libavutil/intreadwrite.h"
  27. /** Additional profile info flags */
  28. #define DNXHD_INTERLACED (1<<0)
  29. #define DNXHD_MBAFF (1<<1)
  30. #define DNXHD_444 (1<<2)
  31. /** Frame headers, extra 0x00 added to end for parser */
  32. #define DNXHD_HEADER_INITIAL 0x000002800100
  33. #define DNXHD_HEADER_444 0x000002800200
  34. /** Indicate that a CIDEntry value must be read in the bitstream */
  35. #define DNXHD_VARIABLE 0
  36. typedef struct CIDEntry {
  37. int cid;
  38. unsigned int width, height;
  39. unsigned int frame_size;
  40. unsigned int coding_unit_size;
  41. uint16_t flags;
  42. int index_bits;
  43. int bit_depth;
  44. int eob_index;
  45. const uint8_t *luma_weight, *chroma_weight;
  46. const uint8_t *dc_codes, *dc_bits;
  47. const uint16_t *ac_codes;
  48. const uint8_t *ac_bits, *ac_info;
  49. const uint16_t *run_codes;
  50. const uint8_t *run_bits, *run;
  51. int bit_rates[5]; ///< Helper to choose variants, rounded to nearest 5Mb/s
  52. AVRational packet_scale;
  53. } CIDEntry;
  54. extern const CIDEntry ff_dnxhd_cid_table[];
  55. int ff_dnxhd_get_cid_table(int cid);
  56. int ff_dnxhd_find_cid(AVCodecContext *avctx, int bit_depth);
  57. void ff_dnxhd_print_profiles(AVCodecContext *avctx, int loglevel);
  58. static av_always_inline uint64_t ff_dnxhd_check_header_prefix_hr(uint64_t prefix)
  59. {
  60. uint64_t data_offset = prefix >> 16;
  61. if ((prefix & 0xFFFF0000FFFFLL) == 0x0300 &&
  62. data_offset >= 0x0280 && data_offset <= 0x2170 &&
  63. (data_offset & 3) == 0)
  64. return prefix;
  65. return 0;
  66. }
  67. static av_always_inline uint64_t ff_dnxhd_check_header_prefix(uint64_t prefix)
  68. {
  69. if (prefix == DNXHD_HEADER_INITIAL ||
  70. prefix == DNXHD_HEADER_444 ||
  71. ff_dnxhd_check_header_prefix_hr(prefix))
  72. return prefix;
  73. return 0;
  74. }
  75. static av_always_inline uint64_t ff_dnxhd_parse_header_prefix(const uint8_t *buf)
  76. {
  77. uint64_t prefix = AV_RB32(buf);
  78. prefix = (prefix << 16) | buf[4] << 8;
  79. return ff_dnxhd_check_header_prefix(prefix);
  80. }
  81. int avpriv_dnxhd_get_frame_size(int cid);
  82. int avpriv_dnxhd_get_hr_frame_size(int cid, int w, int h);
  83. int avpriv_dnxhd_get_interlaced(int cid);
  84. #endif /* AVCODEC_DNXHDDATA_H */