falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78340 Commits
32 Branches
313MB
Tree: 58ded09bd1
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '58ded09bd1'
${ noResults }
FFmpeg/libavcodec/diracdec.c

2211 lines
75KB
Raw Blame History 

  1. /*

  2.  * Copyright (C) 2007 Marco Gerards <marco@gnu.org>

  3.  * Copyright (C) 2009 David Conrad

  4.  * Copyright (C) 2011 Jordi Ortiz

  5.  *

  6.  * This file is part of FFmpeg.

  7.  *

  8.  * FFmpeg is free software; you can redistribute it and/or

  9.  * modify it under the terms of the GNU Lesser General Public

  10.  * License as published by the Free Software Foundation; either

  11.  * version 2.1 of the License, or (at your option) any later version.

  12.  *

  13.  * FFmpeg is distributed in the hope that it will be useful,

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

  15.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

  16.  * Lesser General Public License for more details.

  17.  *

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

  19.  * License along with FFmpeg; if not, write to the Free Software

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

  21.  */

  22. 

  23. /**

  24.  * @file

  25.  * Dirac Decoder

  26.  * @author Marco Gerards <marco@gnu.org>, David Conrad, Jordi Ortiz <nenjordi@gmail.com>

  27.  */

  28. 

  29. #include "avcodec.h"

  30. #include "get_bits.h"

  31. #include "bytestream.h"

  32. #include "internal.h"

  33. #include "golomb.h"

  34. #include "dirac_arith.h"

  35. #include "mpeg12data.h"

  36. #include "libavcodec/mpegvideo.h"

  37. #include "mpegvideoencdsp.h"

  38. #include "dirac_dwt.h"

  39. #include "dirac.h"

  40. #include "diractab.h"

  41. #include "diracdsp.h"

  42. #include "videodsp.h"

  43. 

  44. /**

  45.  * The spec limits this to 3 for frame coding, but in practice can be as high as 6

  46.  */

  47. #define MAX_REFERENCE_FRAMES 8

  48. #define MAX_DELAY 5         /* limit for main profile for frame coding (TODO: field coding) */

  49. #define MAX_FRAMES (MAX_REFERENCE_FRAMES + MAX_DELAY + 1)

  50. #define MAX_QUANT 255        /* max quant for VC-2 */

  51. #define MAX_BLOCKSIZE 32    /* maximum xblen/yblen we support */

  52. 

  53. /**

  54.  * DiracBlock->ref flags, if set then the block does MC from the given ref

  55.  */

  56. #define DIRAC_REF_MASK_REF1   1

  57. #define DIRAC_REF_MASK_REF2   2

  58. #define DIRAC_REF_MASK_GLOBAL 4

  59. 

  60. /**

  61.  * Value of Picture.reference when Picture is not a reference picture, but

  62.  * is held for delayed output.

  63.  */

  64. #define DELAYED_PIC_REF 4

  65. 

  66. #define CALC_PADDING(size, depth)                       \

  67.     (((size + (1 << depth) - 1) >> depth) << depth)

  68. 

  69. #define DIVRNDUP(a, b) (((a) + (b) - 1) / (b))

  70. 

  71. typedef struct {

  72.     AVFrame *avframe;

  73.     int interpolated[3];    /* 1 if hpel[] is valid */

  74.     uint8_t *hpel[3][4];

  75.     uint8_t *hpel_base[3][4];

  76.     int reference;

  77. } DiracFrame;

  78. 

  79. typedef struct {

  80.     union {

  81.         int16_t mv[2][2];

  82.         int16_t dc[3];

  83.     } u; /* anonymous unions aren't in C99 :( */

  84.     uint8_t ref;

  85. } DiracBlock;

  86. 

  87. typedef struct SubBand {

  88.     int level;

  89.     int orientation;

  90.     int stride; /* in bytes */

  91.     int width;

  92.     int height;

  93.     int pshift;

  94.     int quant;

  95.     uint8_t *ibuf;

  96.     struct SubBand *parent;

  97. 

  98.     /* for low delay */

  99.     unsigned length;

  100.     const uint8_t *coeff_data;

  101. } SubBand;

  102. 

  103. typedef struct Plane {

  104.     int width;

  105.     int height;

  106.     ptrdiff_t stride;

  107. 

  108.     int idwt_width;

  109.     int idwt_height;

  110.     int idwt_stride;

  111.     uint8_t *idwt_buf;

  112.     uint8_t *idwt_buf_base;

  113.     uint8_t *idwt_tmp;

  114. 

  115.     /* block length */

  116.     uint8_t xblen;

  117.     uint8_t yblen;

  118.     /* block separation (block n+1 starts after this many pixels in block n) */

  119.     uint8_t xbsep;

  120.     uint8_t ybsep;

  121.     /* amount of overspill on each edge (half of the overlap between blocks) */

  122.     uint8_t xoffset;

  123.     uint8_t yoffset;

  124. 

  125.     SubBand band[MAX_DWT_LEVELS][4];

  126. } Plane;

  127. 

  128. typedef struct DiracContext {

  129.     AVCodecContext *avctx;

  130.     MpegvideoEncDSPContext mpvencdsp;

  131.     VideoDSPContext vdsp;

  132.     DiracDSPContext diracdsp;

  133.     DiracVersionInfo version;

  134.     GetBitContext gb;

  135.     AVDiracSeqHeader seq;

  136.     int seen_sequence_header;

  137.     int frame_number;           /* number of the next frame to display       */

  138.     Plane plane[3];

  139.     int chroma_x_shift;

  140.     int chroma_y_shift;

  141. 

  142.     int bit_depth;              /* bit depth                                 */

  143.     int pshift;                 /* pixel shift = bit_depth > 8               */

  144. 

  145.     int zero_res;               /* zero residue flag                         */

  146.     int is_arith;               /* whether coeffs use arith or golomb coding */

  147.     int core_syntax;            /* use core syntax only                      */

  148.     int low_delay;              /* use the low delay syntax                  */

  149.     int hq_picture;             /* high quality picture, enables low_delay   */

  150.     int ld_picture;             /* use low delay picture, turns on low_delay */

  151.     int dc_prediction;          /* has dc prediction                         */

  152.     int globalmc_flag;          /* use global motion compensation            */

  153.     int num_refs;               /* number of reference pictures              */

  154. 

  155.     /* wavelet decoding */

  156.     unsigned wavelet_depth;     /* depth of the IDWT                         */

  157.     unsigned wavelet_idx;

  158. 

  159.     /**

  160.      * schroedinger older than 1.0.8 doesn't store

  161.      * quant delta if only one codebook exists in a band

  162.      */

  163.     unsigned old_delta_quant;

  164.     unsigned codeblock_mode;

  165. 

  166.     unsigned num_x;              /* number of horizontal slices               */

  167.     unsigned num_y;              /* number of vertical slices                 */

  168. 

  169.     struct {

  170.         unsigned width;

  171.         unsigned height;

  172.     } codeblock[MAX_DWT_LEVELS+1];

  173. 

  174.     struct {

  175.         AVRational bytes;       /* average bytes per slice                   */

  176.         uint8_t quant[MAX_DWT_LEVELS][4]; /* [DIRAC_STD] E.1 */

  177.     } lowdelay;

  178. 

  179.     struct {

  180.         unsigned prefix_bytes;

  181.         unsigned size_scaler;

  182.     } highquality;

  183. 

  184.     struct {

  185.         int pan_tilt[2];        /* pan/tilt vector                           */

  186.         int zrs[2][2];          /* zoom/rotate/shear matrix                  */

  187.         int perspective[2];     /* perspective vector                        */

  188.         unsigned zrs_exp;

  189.         unsigned perspective_exp;

  190.     } globalmc[2];

  191. 

  192.     /* motion compensation */

  193.     uint8_t mv_precision;       /* [DIRAC_STD] REFS_WT_PRECISION             */

  194.     int16_t weight[2];          /* [DIRAC_STD] REF1_WT and REF2_WT           */

  195.     unsigned weight_log2denom;  /* [DIRAC_STD] REFS_WT_PRECISION             */

  196. 

  197.     int blwidth;                /* number of blocks (horizontally)           */

  198.     int blheight;               /* number of blocks (vertically)             */

  199.     int sbwidth;                /* number of superblocks (horizontally)      */

  200.     int sbheight;               /* number of superblocks (vertically)        */

  201. 

  202.     uint8_t *sbsplit;

  203.     DiracBlock *blmotion;

  204. 

  205.     uint8_t *edge_emu_buffer[4];

  206.     uint8_t *edge_emu_buffer_base;

  207. 

  208.     uint16_t *mctmp;            /* buffer holding the MC data multiplied by OBMC weights */

  209.     uint8_t *mcscratch;

  210.     int buffer_stride;

  211. 

  212.     DECLARE_ALIGNED(16, uint8_t, obmc_weight)[3][MAX_BLOCKSIZE*MAX_BLOCKSIZE];

  213. 

  214.     void (*put_pixels_tab[4])(uint8_t *dst, const uint8_t *src[5], int stride, int h);

  215.     void (*avg_pixels_tab[4])(uint8_t *dst, const uint8_t *src[5], int stride, int h);

  216.     void (*add_obmc)(uint16_t *dst, const uint8_t *src, int stride, const uint8_t *obmc_weight, int yblen);

  217.     dirac_weight_func weight_func;

  218.     dirac_biweight_func biweight_func;

  219. 

  220.     DiracFrame *current_picture;

  221.     DiracFrame *ref_pics[2];

  222. 

  223.     DiracFrame *ref_frames[MAX_REFERENCE_FRAMES+1];

  224.     DiracFrame *delay_frames[MAX_DELAY+1];

  225.     DiracFrame all_frames[MAX_FRAMES];

  226. } DiracContext;

  227. 

  228. enum dirac_subband {

  229.     subband_ll = 0,

  230.     subband_hl = 1,

  231.     subband_lh = 2,

  232.     subband_hh = 3,

  233.     subband_nb,

  234. };

  235. 

  236. /* magic number division by 3 from schroedinger */

  237. static inline int divide3(int x)

  238. {

  239.     return ((x+1)*21845 + 10922) >> 16;

  240. }

  241. 

  242. static DiracFrame *remove_frame(DiracFrame *framelist[], int picnum)

  243. {

  244.     DiracFrame *remove_pic = NULL;

  245.     int i, remove_idx = -1;

  246. 

  247.     for (i = 0; framelist[i]; i++)

  248.         if (framelist[i]->avframe->display_picture_number == picnum) {

  249.             remove_pic = framelist[i];

  250.             remove_idx = i;

  251.         }

  252. 

  253.     if (remove_pic)

  254.         for (i = remove_idx; framelist[i]; i++)

  255.             framelist[i] = framelist[i+1];

  256. 

  257.     return remove_pic;

  258. }

  259. 

  260. static int add_frame(DiracFrame *framelist[], int maxframes, DiracFrame *frame)

  261. {

  262.     int i;

  263.     for (i = 0; i < maxframes; i++)

  264.         if (!framelist[i]) {

  265.             framelist[i] = frame;

  266.             return 0;

  267.         }

  268.     return -1;

  269. }

  270. 

  271. static int alloc_sequence_buffers(DiracContext *s)

  272. {

  273.     int sbwidth  = DIVRNDUP(s->seq.width,  4);

  274.     int sbheight = DIVRNDUP(s->seq.height, 4);

  275.     int i, w, h, top_padding;

  276. 

  277.     /* todo: think more about this / use or set Plane here */

  278.     for (i = 0; i < 3; i++) {

  279.         int max_xblen = MAX_BLOCKSIZE >> (i ? s->chroma_x_shift : 0);

  280.         int max_yblen = MAX_BLOCKSIZE >> (i ? s->chroma_y_shift : 0);

  281.         w = s->seq.width  >> (i ? s->chroma_x_shift : 0);

  282.         h = s->seq.height >> (i ? s->chroma_y_shift : 0);

  283. 

  284.         /* we allocate the max we support here since num decompositions can

  285.          * change from frame to frame. Stride is aligned to 16 for SIMD, and

  286.          * 1<<MAX_DWT_LEVELS top padding to avoid if(y>0) in arith decoding

  287.          * MAX_BLOCKSIZE padding for MC: blocks can spill up to half of that

  288.          * on each side */

  289.         top_padding = FFMAX(1<<MAX_DWT_LEVELS, max_yblen/2);

  290.         w = FFALIGN(CALC_PADDING(w, MAX_DWT_LEVELS), 8); /* FIXME: Should this be 16 for SSE??? */

  291.         h = top_padding + CALC_PADDING(h, MAX_DWT_LEVELS) + max_yblen/2;

  292. 

  293.         s->plane[i].idwt_buf_base = av_mallocz_array((w+max_xblen), h * (2 << s->pshift));

  294.         s->plane[i].idwt_tmp      = av_malloc_array((w+16), 2 << s->pshift);

  295.         s->plane[i].idwt_buf      = s->plane[i].idwt_buf_base + (top_padding*w)*(2 << s->pshift);

  296.         if (!s->plane[i].idwt_buf_base || !s->plane[i].idwt_tmp)

  297.             return AVERROR(ENOMEM);

  298.     }

  299. 

  300.     /* fixme: allocate using real stride here */

  301.     s->sbsplit  = av_malloc_array(sbwidth, sbheight);

  302.     s->blmotion = av_malloc_array(sbwidth, sbheight * 16 * sizeof(*s->blmotion));

  303. 

  304.     if (!s->sbsplit || !s->blmotion)

  305.         return AVERROR(ENOMEM);

  306.     return 0;

  307. }

  308. 

  309. static int alloc_buffers(DiracContext *s, int stride)

  310. {

  311.     int w = s->seq.width;

  312.     int h = s->seq.height;

  313. 

  314.     av_assert0(stride >= w);

  315.     stride += 64;

  316. 

  317.     if (s->buffer_stride >= stride)

  318.         return 0;

  319.     s->buffer_stride = 0;

  320. 

  321.     av_freep(&s->edge_emu_buffer_base);

  322.     memset(s->edge_emu_buffer, 0, sizeof(s->edge_emu_buffer));

  323.     av_freep(&s->mctmp);

  324.     av_freep(&s->mcscratch);

  325. 

  326.     s->edge_emu_buffer_base = av_malloc_array(stride, MAX_BLOCKSIZE);

  327. 

  328.     s->mctmp     = av_malloc_array((stride+MAX_BLOCKSIZE), (h+MAX_BLOCKSIZE) * sizeof(*s->mctmp));

  329.     s->mcscratch = av_malloc_array(stride, MAX_BLOCKSIZE);

  330. 

  331.     if (!s->edge_emu_buffer_base || !s->mctmp || !s->mcscratch)

  332.         return AVERROR(ENOMEM);

  333. 

  334.     s->buffer_stride = stride;

  335.     return 0;

  336. }

  337. 

  338. static void free_sequence_buffers(DiracContext *s)

  339. {

  340.     int i, j, k;

  341. 

  342.     for (i = 0; i < MAX_FRAMES; i++) {

  343.         if (s->all_frames[i].avframe->data[0]) {

  344.             av_frame_unref(s->all_frames[i].avframe);

  345.             memset(s->all_frames[i].interpolated, 0, sizeof(s->all_frames[i].interpolated));

  346.         }

  347. 

  348.         for (j = 0; j < 3; j++)

  349.             for (k = 1; k < 4; k++)

  350.                 av_freep(&s->all_frames[i].hpel_base[j][k]);

  351.     }

  352. 

  353.     memset(s->ref_frames, 0, sizeof(s->ref_frames));

  354.     memset(s->delay_frames, 0, sizeof(s->delay_frames));

  355. 

  356.     for (i = 0; i < 3; i++) {

  357.         av_freep(&s->plane[i].idwt_buf_base);

  358.         av_freep(&s->plane[i].idwt_tmp);

  359.     }

  360. 

  361.     s->buffer_stride = 0;

  362.     av_freep(&s->sbsplit);

  363.     av_freep(&s->blmotion);

  364.     av_freep(&s->edge_emu_buffer_base);

  365. 

  366.     av_freep(&s->mctmp);

  367.     av_freep(&s->mcscratch);

  368. }

  369. 

  370. static av_cold int dirac_decode_init(AVCodecContext *avctx)

  371. {

  372.     DiracContext *s = avctx->priv_data;

  373.     int i;

  374. 

  375.     s->avctx = avctx;

  376.     s->frame_number = -1;

  377. 

  378.     ff_diracdsp_init(&s->diracdsp);

  379.     ff_mpegvideoencdsp_init(&s->mpvencdsp, avctx);

  380.     ff_videodsp_init(&s->vdsp, 8);

  381. 

  382.     for (i = 0; i < MAX_FRAMES; i++) {

  383.         s->all_frames[i].avframe = av_frame_alloc();

  384.         if (!s->all_frames[i].avframe) {

  385.             while (i > 0)

  386.                 av_frame_free(&s->all_frames[--i].avframe);

  387.             return AVERROR(ENOMEM);

  388.         }

  389.     }

  390. 

  391.     return 0;

  392. }

  393. 

  394. static void dirac_decode_flush(AVCodecContext *avctx)

  395. {

  396.     DiracContext *s = avctx->priv_data;

  397.     free_sequence_buffers(s);

  398.     s->seen_sequence_header = 0;

  399.     s->frame_number = -1;

  400. }

  401. 

  402. static av_cold int dirac_decode_end(AVCodecContext *avctx)

  403. {

  404.     DiracContext *s = avctx->priv_data;

  405.     int i;

  406. 

  407.     dirac_decode_flush(avctx);

  408.     for (i = 0; i < MAX_FRAMES; i++)

  409.         av_frame_free(&s->all_frames[i].avframe);

  410. 

  411.     return 0;

  412. }

  413. 

  414. #define SIGN_CTX(x) (CTX_SIGN_ZERO + ((x) > 0) - ((x) < 0))

  415. 

  416. static inline int coeff_unpack_golomb(GetBitContext *gb, int qfactor, int qoffset)

  417. {

  418.     int sign, coeff;

  419.     uint32_t buf;

  420. 

  421.     OPEN_READER(re, gb);

  422.     UPDATE_CACHE(re, gb);

  423.     buf = GET_CACHE(re, gb);

  424. 

  425.     if (buf & 0x80000000) {

  426.         LAST_SKIP_BITS(re,gb,1);

  427.         CLOSE_READER(re, gb);

  428.         return 0;

  429.     }

  430. 

  431.     if (buf & 0xAA800000) {

  432.         buf >>= 32 - 8;

  433.         SKIP_BITS(re, gb, ff_interleaved_golomb_vlc_len[buf]);

  434. 

  435.         coeff = ff_interleaved_ue_golomb_vlc_code[buf];

  436.     } else {

  437.         unsigned ret = 1;

  438. 

  439.         do {

  440.             buf >>= 32 - 8;

  441.             SKIP_BITS(re, gb,

  442.                            FFMIN(ff_interleaved_golomb_vlc_len[buf], 8));

  443. 

  444.             if (ff_interleaved_golomb_vlc_len[buf] != 9) {

  445.                 ret <<= (ff_interleaved_golomb_vlc_len[buf] - 1) >> 1;

  446.                 ret  |= ff_interleaved_dirac_golomb_vlc_code[buf];

  447.                 break;

  448.             }

  449.             ret = (ret << 4) | ff_interleaved_dirac_golomb_vlc_code[buf];

  450.             UPDATE_CACHE(re, gb);

  451.             buf = GET_CACHE(re, gb);

  452.         } while (ret<0x8000000U && BITS_AVAILABLE(re, gb));

  453. 

  454.         coeff = ret - 1;

  455.     }

  456. 

  457.     coeff = (coeff * qfactor + qoffset) >> 2;

  458.     sign  = SHOW_SBITS(re, gb, 1);

  459.     LAST_SKIP_BITS(re, gb, 1);

  460.     coeff = (coeff ^ sign) - sign;

  461. 

  462.     CLOSE_READER(re, gb);

  463.     return coeff;

  464. }

  465. 

  466. #define UNPACK_ARITH(n, type) \

  467.     static inline void coeff_unpack_arith_##n(DiracArith *c, int qfactor, int qoffset, \

  468.                                               SubBand *b, type *buf, int x, int y) \

  469.     { \

  470.         int coeff, sign, sign_pred = 0, pred_ctx = CTX_ZPZN_F1; \

  471.         const int mstride = -(b->stride >> (1+b->pshift)); \

  472.         if (b->parent) { \

  473.             const type *pbuf = (type *)b->parent->ibuf; \

  474.             const int stride = b->parent->stride >> (1+b->parent->pshift); \

  475.             pred_ctx += !!pbuf[stride * (y>>1) + (x>>1)] << 1; \

  476.         } \

  477.         if (b->orientation == subband_hl) \

  478.             sign_pred = buf[mstride]; \

  479.         if (x) { \

  480.             pred_ctx += !(buf[-1] | buf[mstride] | buf[-1 + mstride]); \

  481.             if (b->orientation == subband_lh) \

  482.                 sign_pred = buf[-1]; \

  483.         } else { \

  484.             pred_ctx += !buf[mstride]; \

  485.         } \

  486.         coeff = dirac_get_arith_uint(c, pred_ctx, CTX_COEFF_DATA); \

  487.         if (coeff) { \

  488.             coeff = (coeff * qfactor + qoffset) >> 2; \

  489.             sign  = dirac_get_arith_bit(c, SIGN_CTX(sign_pred)); \

  490.             coeff = (coeff ^ -sign) + sign; \

  491.         } \

  492.         *buf = coeff; \

  493.     } \

  494. 

  495. UNPACK_ARITH(8, int16_t)

  496. UNPACK_ARITH(10, int32_t)

  497. 

  498. /**

  499.  * Decode the coeffs in the rectangle defined by left, right, top, bottom

  500.  * [DIRAC_STD] 13.4.3.2 Codeblock unpacking loop. codeblock()

  501.  */

  502. static inline void codeblock(DiracContext *s, SubBand *b,

  503.                              GetBitContext *gb, DiracArith *c,

  504.                              int left, int right, int top, int bottom,

  505.                              int blockcnt_one, int is_arith)

  506. {

  507.     int x, y, zero_block;

  508.     int qoffset, qfactor;

  509.     uint8_t *buf;

  510. 

  511.     /* check for any coded coefficients in this codeblock */

  512.     if (!blockcnt_one) {

  513.         if (is_arith)

  514.             zero_block = dirac_get_arith_bit(c, CTX_ZERO_BLOCK);

  515.         else

  516.             zero_block = get_bits1(gb);

  517. 

  518.         if (zero_block)

  519.             return;

  520.     }

  521. 

  522.     if (s->codeblock_mode && !(s->old_delta_quant && blockcnt_one)) {

  523.         int quant = b->quant;

  524.         if (is_arith)

  525.             quant += dirac_get_arith_int(c, CTX_DELTA_Q_F, CTX_DELTA_Q_DATA);

  526.         else

  527.             quant += dirac_get_se_golomb(gb);

  528.         if (quant < 0) {

  529.             av_log(s->avctx, AV_LOG_ERROR, "Invalid quant\n");

  530.             return;

  531.         }

  532.         b->quant = quant;

  533.     }

  534. 

  535.     if (b->quant > 115) {

  536.         av_log(s->avctx, AV_LOG_ERROR, "Unsupported quant %d\n", b->quant);

  537.         b->quant = 0;

  538.         return;

  539.     }

  540. 

  541.     qfactor = ff_dirac_qscale_tab[b->quant];

  542.     /* TODO: context pointer? */

  543.     if (!s->num_refs)

  544.         qoffset = ff_dirac_qoffset_intra_tab[b->quant] + 2;

  545.     else

  546.         qoffset = ff_dirac_qoffset_inter_tab[b->quant] + 2;

  547. 

  548.     buf = b->ibuf + top * b->stride;

  549.     if (is_arith) {

  550.         for (y = top; y < bottom; y++) {

  551.             for (x = left; x < right; x++) {

  552.                 if (b->pshift) {

  553.                     coeff_unpack_arith_10(c, qfactor, qoffset, b, (int32_t*)(buf)+x, x, y);

  554.                 } else {

  555.                     coeff_unpack_arith_8(c, qfactor, qoffset, b, (int16_t*)(buf)+x, x, y);

  556.                 }

  557.             }

  558.             buf += b->stride;

  559.         }

  560.     } else {

  561.         for (y = top; y < bottom; y++) {

  562.             for (x = left; x < right; x++) {

  563.                 int val = coeff_unpack_golomb(gb, qfactor, qoffset);

  564.                 if (b->pshift) {

  565.                     AV_WN32(&buf[4*x], val);

  566.                 } else {

  567.                     AV_WN16(&buf[2*x], val);

  568.                 }

  569.             }

  570.             buf += b->stride;

  571.          }

  572.      }

  573. }

  574. 

  575. /**

  576.  * Dirac Specification ->

  577.  * 13.3 intra_dc_prediction(band)

  578.  */

  579. #define INTRA_DC_PRED(n, type) \

  580.     static inline void intra_dc_prediction_##n(SubBand *b) \

  581.     { \

  582.         type *buf = (type*)b->ibuf; \

  583.         int x, y; \

  584.         \

  585.         for (x = 1; x < b->width; x++) \

  586.             buf[x] += buf[x-1]; \

  587.         buf += (b->stride >> (1+b->pshift)); \

  588.         \

  589.         for (y = 1; y < b->height; y++) { \

  590.             buf[0] += buf[-(b->stride >> (1+b->pshift))]; \

  591.             \

  592.             for (x = 1; x < b->width; x++) { \

  593.                 int pred = buf[x - 1] + buf[x - (b->stride >> (1+b->pshift))] + buf[x - (b->stride >> (1+b->pshift))-1]; \

  594.                 buf[x]  += divide3(pred); \

  595.             } \

  596.             buf += (b->stride >> (1+b->pshift)); \

  597.         } \

  598.     } \

  599. 

  600. INTRA_DC_PRED(8, int16_t)

  601. INTRA_DC_PRED(10, int32_t)

  602. 

  603. /**

  604.  * Dirac Specification ->

  605.  * 13.4.2 Non-skipped subbands.  subband_coeffs()

  606.  */

  607. static av_always_inline void decode_subband_internal(DiracContext *s, SubBand *b, int is_arith)

  608. {

  609.     int cb_x, cb_y, left, right, top, bottom;

  610.     DiracArith c;

  611.     GetBitContext gb;

  612.     int cb_width  = s->codeblock[b->level + (b->orientation != subband_ll)].width;

  613.     int cb_height = s->codeblock[b->level + (b->orientation != subband_ll)].height;

  614.     int blockcnt_one = (cb_width + cb_height) == 2;

  615. 

  616.     if (!b->length)

  617.         return;

  618. 

  619.     init_get_bits8(&gb, b->coeff_data, b->length);

  620. 

  621.     if (is_arith)

  622.         ff_dirac_init_arith_decoder(&c, &gb, b->length);

  623. 

  624.     top = 0;

  625.     for (cb_y = 0; cb_y < cb_height; cb_y++) {

  626.         bottom = (b->height * (cb_y+1LL)) / cb_height;

  627.         left = 0;

  628.         for (cb_x = 0; cb_x < cb_width; cb_x++) {

  629.             right = (b->width * (cb_x+1LL)) / cb_width;

  630.             codeblock(s, b, &gb, &c, left, right, top, bottom, blockcnt_one, is_arith);

  631.             left = right;

  632.         }

  633.         top = bottom;

  634.     }

  635. 

  636.     if (b->orientation == subband_ll && s->num_refs == 0) {

  637.         if (s->pshift) {

  638.             intra_dc_prediction_10(b);

  639.         } else {

  640.             intra_dc_prediction_8(b);

  641.         }

  642.     }

  643. }

  644. 

  645. static int decode_subband_arith(AVCodecContext *avctx, void *b)

  646. {

  647.     DiracContext *s = avctx->priv_data;

  648.     decode_subband_internal(s, b, 1);

  649.     return 0;

  650. }

  651. 

  652. static int decode_subband_golomb(AVCodecContext *avctx, void *arg)

  653. {

  654.     DiracContext *s = avctx->priv_data;

  655.     SubBand **b     = arg;

  656.     decode_subband_internal(s, *b, 0);

  657.     return 0;

  658. }

  659. 

  660. /**

  661.  * Dirac Specification ->

  662.  * [DIRAC_STD] 13.4.1 core_transform_data()

  663.  */

  664. static void decode_component(DiracContext *s, int comp)

  665. {

  666.     AVCodecContext *avctx = s->avctx;

  667.     SubBand *bands[3*MAX_DWT_LEVELS+1];

  668.     enum dirac_subband orientation;

  669.     int level, num_bands = 0;

  670. 

  671.     /* Unpack all subbands at all levels. */

  672.     for (level = 0; level < s->wavelet_depth; level++) {

  673.         for (orientation = !!level; orientation < 4; orientation++) {

  674.             SubBand *b = &s->plane[comp].band[level][orientation];

  675.             bands[num_bands++] = b;

  676. 

  677.             align_get_bits(&s->gb);

  678.             /* [DIRAC_STD] 13.4.2 subband() */

  679.             b->length = svq3_get_ue_golomb(&s->gb);

  680.             if (b->length) {

  681.                 b->quant = svq3_get_ue_golomb(&s->gb);

  682.                 align_get_bits(&s->gb);

  683.                 b->coeff_data = s->gb.buffer + get_bits_count(&s->gb)/8;

  684.                 b->length = FFMIN(b->length, FFMAX(get_bits_left(&s->gb)/8, 0));

  685.                 skip_bits_long(&s->gb, b->length*8);

  686.             }

  687.         }

  688.         /* arithmetic coding has inter-level dependencies, so we can only execute one level at a time */

  689.         if (s->is_arith)

  690.             avctx->execute(avctx, decode_subband_arith, &s->plane[comp].band[level][!!level],

  691.                            NULL, 4-!!level, sizeof(SubBand));

  692.     }

  693.     /* golomb coding has no inter-level dependencies, so we can execute all subbands in parallel */

  694.     if (!s->is_arith)

  695.         avctx->execute(avctx, decode_subband_golomb, bands, NULL, num_bands, sizeof(SubBand*));

  696. }

  697. 

  698. #define PARSE_VALUES(type, x, gb, ebits, buf1, buf2) \

  699.     type *buf = (type *)buf1; \

  700.     buf[x] = coeff_unpack_golomb(gb, qfactor, qoffset); \

  701.     if (get_bits_count(gb) >= ebits) \

  702.         return; \

  703.     if (buf2) { \

  704.         buf = (type *)buf2; \

  705.         buf[x] = coeff_unpack_golomb(gb, qfactor, qoffset); \

  706.         if (get_bits_count(gb) >= ebits) \

  707.             return; \

  708.     } \

  709. 

  710. static void decode_subband(DiracContext *s, GetBitContext *gb, int quant,

  711.                            int slice_x, int slice_y, int bits_end,

  712.                            SubBand *b1, SubBand *b2)

  713. {

  714.     int left   = b1->width  * slice_x    / s->num_x;

  715.     int right  = b1->width  *(slice_x+1) / s->num_x;

  716.     int top    = b1->height * slice_y    / s->num_y;

  717.     int bottom = b1->height *(slice_y+1) / s->num_y;

  718. 

  719.     int qfactor, qoffset;

  720. 

  721.     uint8_t *buf1 =      b1->ibuf + top * b1->stride;

  722.     uint8_t *buf2 = b2 ? b2->ibuf + top * b2->stride: NULL;

  723.     int x, y;

  724. 

  725.     if (quant > 115) {

  726.         av_log(s->avctx, AV_LOG_ERROR, "Unsupported quant %d\n", quant);

  727.         return;

  728.     }

  729.     qfactor = ff_dirac_qscale_tab[quant & 0x7f];

  730.     qoffset = ff_dirac_qoffset_intra_tab[quant & 0x7f] + 2;

  731.     /* we have to constantly check for overread since the spec explicitly

  732.        requires this, with the meaning that all remaining coeffs are set to 0 */

  733.     if (get_bits_count(gb) >= bits_end)

  734.         return;

  735. 

  736.     if (s->pshift) {

  737.         for (y = top; y < bottom; y++) {

  738.             for (x = left; x < right; x++) {

  739.                 PARSE_VALUES(int32_t, x, gb, bits_end, buf1, buf2);

  740.             }

  741.             buf1 += b1->stride;

  742.             if (buf2)

  743.                 buf2 += b2->stride;

  744.         }

  745.     }

  746.     else {

  747.         for (y = top; y < bottom; y++) {

  748.             for (x = left; x < right; x++) {

  749.                 PARSE_VALUES(int16_t, x, gb, bits_end, buf1, buf2);

  750.             }

  751.             buf1 += b1->stride;

  752.             if (buf2)

  753.                 buf2 += b2->stride;

  754.         }

  755.     }

  756. }

  757. 

  758. /* Used by Low Delay and High Quality profiles */

  759. typedef struct DiracSlice {

  760.     GetBitContext gb;

  761.     int slice_x;

  762.     int slice_y;

  763.     int bytes;

  764. } DiracSlice;

  765. 

  766. 

  767. /**

  768.  * Dirac Specification ->

  769.  * 13.5.2 Slices. slice(sx,sy)

  770.  */

  771. static int decode_lowdelay_slice(AVCodecContext *avctx, void *arg)

  772. {

  773.     DiracContext *s = avctx->priv_data;

  774.     DiracSlice *slice = arg;

  775.     GetBitContext *gb = &slice->gb;

  776.     enum dirac_subband orientation;

  777.     int level, quant, chroma_bits, chroma_end;

  778. 

  779.     int quant_base  = get_bits(gb, 7); /*[DIRAC_STD] qindex */

  780.     int length_bits = av_log2(8 * slice->bytes)+1;

  781.     int luma_bits   = get_bits_long(gb, length_bits);

  782.     int luma_end    = get_bits_count(gb) + FFMIN(luma_bits, get_bits_left(gb));

  783. 

  784.     /* [DIRAC_STD] 13.5.5.2 luma_slice_band */

  785.     for (level = 0; level < s->wavelet_depth; level++)

  786.         for (orientation = !!level; orientation < 4; orientation++) {

  787.             quant = FFMAX(quant_base - s->lowdelay.quant[level][orientation], 0);

  788.             decode_subband(s, gb, quant, slice->slice_x, slice->slice_y, luma_end,

  789.                            &s->plane[0].band[level][orientation], NULL);

  790.         }

  791. 

  792.     /* consume any unused bits from luma */

  793.     skip_bits_long(gb, get_bits_count(gb) - luma_end);

  794. 

  795.     chroma_bits = 8*slice->bytes - 7 - length_bits - luma_bits;

  796.     chroma_end  = get_bits_count(gb) + FFMIN(chroma_bits, get_bits_left(gb));

  797.     /* [DIRAC_STD] 13.5.5.3 chroma_slice_band */

  798.     for (level = 0; level < s->wavelet_depth; level++)

  799.         for (orientation = !!level; orientation < 4; orientation++) {

  800.             quant = FFMAX(quant_base - s->lowdelay.quant[level][orientation], 0);

  801.             decode_subband(s, gb, quant, slice->slice_x, slice->slice_y, chroma_end,

  802.                            &s->plane[1].band[level][orientation],

  803.                            &s->plane[2].band[level][orientation]);

  804.         }

  805. 

  806.     return 0;

  807. }

  808. 

  809. /**

  810.  * VC-2 Specification ->

  811.  * 13.5.3 hq_slice(sx,sy)

  812.  */

  813. static int decode_hq_slice(AVCodecContext *avctx, void *arg)

  814. {

  815.     int i, quant, level, orientation, quant_idx;

  816.     uint8_t quants[MAX_DWT_LEVELS][4];

  817.     DiracContext *s = avctx->priv_data;

  818.     DiracSlice *slice = arg;

  819.     GetBitContext *gb = &slice->gb;

  820. 

  821.     skip_bits_long(gb, 8*s->highquality.prefix_bytes);

  822.     quant_idx = get_bits(gb, 8);

  823. 

  824.     /* Slice quantization (slice_quantizers() in the specs) */

  825.     for (level = 0; level < s->wavelet_depth; level++) {

  826.         for (orientation = !!level; orientation < 4; orientation++) {

  827.             quant = FFMAX(quant_idx - s->lowdelay.quant[level][orientation], 0);

  828.             quants[level][orientation] = quant;

  829.         }

  830.     }

  831. 

  832.     /* Luma + 2 Chroma planes */

  833.     for (i = 0; i < 3; i++) {

  834.         int length = s->highquality.size_scaler * get_bits(gb, 8);

  835.         int bits_left = 8 * length;

  836.         int bits_end = get_bits_count(gb) + bits_left;

  837.         for (level = 0; level < s->wavelet_depth; level++) {

  838.             for (orientation = !!level; orientation < 4; orientation++) {

  839.                 decode_subband(s, gb, quants[level][orientation], slice->slice_x, slice->slice_y, bits_end,

  840.                                &s->plane[i].band[level][orientation], NULL);

  841.             }

  842.         }

  843.         skip_bits_long(gb, bits_end - get_bits_count(gb));

  844.     }

  845. 

  846.     return 0;

  847. }

  848. 

  849. /**

  850.  * Dirac Specification ->

  851.  * 13.5.1 low_delay_transform_data()

  852.  */

  853. static int decode_lowdelay(DiracContext *s)

  854. {

  855.     AVCodecContext *avctx = s->avctx;

  856.     int slice_x, slice_y, bytes = 0, bufsize;

  857.     const uint8_t *buf;

  858.     DiracSlice *slices;

  859.     int slice_num = 0;

  860. 

  861.     slices = av_mallocz_array(s->num_x, s->num_y * sizeof(DiracSlice));

  862.     if (!slices)

  863.         return AVERROR(ENOMEM);

  864. 

  865.     align_get_bits(&s->gb);

  866.     /*[DIRAC_STD] 13.5.2 Slices. slice(sx,sy) */

  867.     buf = s->gb.buffer + get_bits_count(&s->gb)/8;

  868.     bufsize = get_bits_left(&s->gb);

  869. 

  870.     if (s->hq_picture) {

  871.         int i;

  872. 

  873.         for (slice_y = 0; bufsize > 0 && slice_y < s->num_y; slice_y++) {

  874.             for (slice_x = 0; bufsize > 0 && slice_x < s->num_x; slice_x++) {

  875.                 bytes = s->highquality.prefix_bytes + 1;

  876.                 for (i = 0; i < 3; i++) {

  877.                     if (bytes <= bufsize/8)

  878.                         bytes += buf[bytes] * s->highquality.size_scaler + 1;

  879.                 }

  880. 

  881.                 slices[slice_num].bytes   = bytes;

  882.                 slices[slice_num].slice_x = slice_x;

  883.                 slices[slice_num].slice_y = slice_y;

  884.                 init_get_bits(&slices[slice_num].gb, buf, bufsize);

  885.                 slice_num++;

  886. 

  887.                 buf     += bytes;

  888.                 if (bufsize/8 >= bytes)

  889.                     bufsize -= bytes*8;

  890.                 else

  891.                     bufsize = 0;

  892.             }

  893.         }

  894.         avctx->execute(avctx, decode_hq_slice, slices, NULL, slice_num,

  895.                        sizeof(DiracSlice));

  896.     } else {

  897.         for (slice_y = 0; bufsize > 0 && slice_y < s->num_y; slice_y++) {

  898.             for (slice_x = 0; bufsize > 0 && slice_x < s->num_x; slice_x++) {

  899.                 bytes = (slice_num+1) * s->lowdelay.bytes.num / s->lowdelay.bytes.den

  900.                     - slice_num    * s->lowdelay.bytes.num / s->lowdelay.bytes.den;

  901.                 slices[slice_num].bytes   = bytes;

  902.                 slices[slice_num].slice_x = slice_x;

  903.                 slices[slice_num].slice_y = slice_y;

  904.                 init_get_bits(&slices[slice_num].gb, buf, bufsize);

  905.                 slice_num++;

  906. 

  907.                 buf     += bytes;

  908.                 if (bufsize/8 >= bytes)

  909.                     bufsize -= bytes*8;

  910.                 else

  911.                     bufsize = 0;

  912.             }

  913.         }

  914.         avctx->execute(avctx, decode_lowdelay_slice, slices, NULL, slice_num,

  915.                        sizeof(DiracSlice)); /* [DIRAC_STD] 13.5.2 Slices */

  916.     }

  917. 

  918.     if (s->dc_prediction) {

  919.         if (s->pshift) {

  920.             intra_dc_prediction_10(&s->plane[0].band[0][0]); /* [DIRAC_STD] 13.3 intra_dc_prediction() */

  921.             intra_dc_prediction_10(&s->plane[1].band[0][0]); /* [DIRAC_STD] 13.3 intra_dc_prediction() */

  922.             intra_dc_prediction_10(&s->plane[2].band[0][0]); /* [DIRAC_STD] 13.3 intra_dc_prediction() */

  923.         } else {

  924.             intra_dc_prediction_8(&s->plane[0].band[0][0]);

  925.             intra_dc_prediction_8(&s->plane[1].band[0][0]);

  926.             intra_dc_prediction_8(&s->plane[2].band[0][0]);

  927.         }

  928.     }

  929.     av_free(slices);

  930.     return 0;

  931. }

  932. 

  933. static void init_planes(DiracContext *s)

  934. {

  935.     int i, w, h, level, orientation;

  936. 

  937.     for (i = 0; i < 3; i++) {

  938.         Plane *p = &s->plane[i];

  939. 

  940.         p->width       = s->seq.width  >> (i ? s->chroma_x_shift : 0);

  941.         p->height      = s->seq.height >> (i ? s->chroma_y_shift : 0);

  942.         p->idwt_width  = w = CALC_PADDING(p->width , s->wavelet_depth);

  943.         p->idwt_height = h = CALC_PADDING(p->height, s->wavelet_depth);

  944.         p->idwt_stride = FFALIGN(p->idwt_width, 8) << (1 + s->pshift);

  945. 

  946.         for (level = s->wavelet_depth-1; level >= 0; level--) {

  947.             w = w>>1;

  948.             h = h>>1;

  949.             for (orientation = !!level; orientation < 4; orientation++) {

  950.                 SubBand *b = &p->band[level][orientation];

  951. 

  952.                 b->pshift = s->pshift;

  953.                 b->ibuf   = p->idwt_buf;

  954.                 b->level  = level;

  955.                 b->stride = p->idwt_stride << (s->wavelet_depth - level);

  956.                 b->width  = w;

  957.                 b->height = h;

  958.                 b->orientation = orientation;

  959. 

  960.                 if (orientation & 1)

  961.                     b->ibuf += w << (1+b->pshift);

  962.                 if (orientation > 1)

  963.                     b->ibuf += (b->stride>>1);

  964. 

  965.                 if (level)

  966.                     b->parent = &p->band[level-1][orientation];

  967.             }

  968.         }

  969. 

  970.         if (i > 0) {

  971.             p->xblen = s->plane[0].xblen >> s->chroma_x_shift;

  972.             p->yblen = s->plane[0].yblen >> s->chroma_y_shift;

  973.             p->xbsep = s->plane[0].xbsep >> s->chroma_x_shift;

  974.             p->ybsep = s->plane[0].ybsep >> s->chroma_y_shift;

  975.         }

  976. 

  977.         p->xoffset = (p->xblen - p->xbsep)/2;

  978.         p->yoffset = (p->yblen - p->ybsep)/2;

  979.     }

  980. }

  981. 

  982. /**

  983.  * Unpack the motion compensation parameters

  984.  * Dirac Specification ->

  985.  * 11.2 Picture prediction data. picture_prediction()

  986.  */

  987. static int dirac_unpack_prediction_parameters(DiracContext *s)

  988. {

  989.     static const uint8_t default_blen[] = { 4, 12, 16, 24 };

  990. 

  991.     GetBitContext *gb = &s->gb;

  992.     unsigned idx, ref;

  993. 

  994.     align_get_bits(gb);

  995.     /* [DIRAC_STD] 11.2.2 Block parameters. block_parameters() */

  996.     /* Luma and Chroma are equal. 11.2.3 */

  997.     idx = svq3_get_ue_golomb(gb); /* [DIRAC_STD] index */

  998. 

  999.     if (idx > 4) {

  1000.         av_log(s->avctx, AV_LOG_ERROR, "Block prediction index too high\n");

  1001.         return AVERROR_INVALIDDATA;

  1002.     }

  1003. 

  1004.     if (idx == 0) {

  1005.         s->plane[0].xblen = svq3_get_ue_golomb(gb);

  1006.         s->plane[0].yblen = svq3_get_ue_golomb(gb);

  1007.         s->plane[0].xbsep = svq3_get_ue_golomb(gb);

  1008.         s->plane[0].ybsep = svq3_get_ue_golomb(gb);

  1009.     } else {

  1010.         /*[DIRAC_STD] preset_block_params(index). Table 11.1 */

  1011.         s->plane[0].xblen = default_blen[idx-1];

  1012.         s->plane[0].yblen = default_blen[idx-1];

  1013.         s->plane[0].xbsep = 4 * idx;

  1014.         s->plane[0].ybsep = 4 * idx;

  1015.     }

  1016.     /*[DIRAC_STD] 11.2.4 motion_data_dimensions()

  1017.       Calculated in function dirac_unpack_block_motion_data */

  1018. 

  1019.     if (s->plane[0].xblen % (1 << s->chroma_x_shift) != 0 ||

  1020.         s->plane[0].yblen % (1 << s->chroma_y_shift) != 0 ||

  1021.         !s->plane[0].xblen || !s->plane[0].yblen) {

  1022.         av_log(s->avctx, AV_LOG_ERROR,

  1023.                "invalid x/y block length (%d/%d) for x/y chroma shift (%d/%d)\n",

  1024.                s->plane[0].xblen, s->plane[0].yblen, s->chroma_x_shift, s->chroma_y_shift);

  1025.         return AVERROR_INVALIDDATA;

  1026.     }

  1027.     if (!s->plane[0].xbsep || !s->plane[0].ybsep || s->plane[0].xbsep < s->plane[0].xblen/2 || s->plane[0].ybsep < s->plane[0].yblen/2) {

  1028.         av_log(s->avctx, AV_LOG_ERROR, "Block separation too small\n");

  1029.         return AVERROR_INVALIDDATA;

  1030.     }

  1031.     if (s->plane[0].xbsep > s->plane[0].xblen || s->plane[0].ybsep > s->plane[0].yblen) {

  1032.         av_log(s->avctx, AV_LOG_ERROR, "Block separation greater than size\n");

  1033.         return AVERROR_INVALIDDATA;

  1034.     }

  1035.     if (FFMAX(s->plane[0].xblen, s->plane[0].yblen) > MAX_BLOCKSIZE) {

  1036.         av_log(s->avctx, AV_LOG_ERROR, "Unsupported large block size\n");

  1037.         return AVERROR_PATCHWELCOME;

  1038.     }

  1039. 

  1040.     /*[DIRAC_STD] 11.2.5 Motion vector precision. motion_vector_precision()

  1041.       Read motion vector precision */

  1042.     s->mv_precision = svq3_get_ue_golomb(gb);

  1043.     if (s->mv_precision > 3) {

  1044.         av_log(s->avctx, AV_LOG_ERROR, "MV precision finer than eighth-pel\n");

  1045.         return AVERROR_INVALIDDATA;

  1046.     }

  1047. 

  1048.     /*[DIRAC_STD] 11.2.6 Global motion. global_motion()

  1049.       Read the global motion compensation parameters */

  1050.     s->globalmc_flag = get_bits1(gb);

  1051.     if (s->globalmc_flag) {

  1052.         memset(s->globalmc, 0, sizeof(s->globalmc));

  1053.         /* [DIRAC_STD] pan_tilt(gparams) */

  1054.         for (ref = 0; ref < s->num_refs; ref++) {

  1055.             if (get_bits1(gb)) {

  1056.                 s->globalmc[ref].pan_tilt[0] = dirac_get_se_golomb(gb);

  1057.                 s->globalmc[ref].pan_tilt[1] = dirac_get_se_golomb(gb);

  1058.             }

  1059.             /* [DIRAC_STD] zoom_rotate_shear(gparams)

  1060.                zoom/rotation/shear parameters */

  1061.             if (get_bits1(gb)) {

  1062.                 s->globalmc[ref].zrs_exp   = svq3_get_ue_golomb(gb);

  1063.                 s->globalmc[ref].zrs[0][0] = dirac_get_se_golomb(gb);

  1064.                 s->globalmc[ref].zrs[0][1] = dirac_get_se_golomb(gb);

  1065.                 s->globalmc[ref].zrs[1][0] = dirac_get_se_golomb(gb);

  1066.                 s->globalmc[ref].zrs[1][1] = dirac_get_se_golomb(gb);

  1067.             } else {

  1068.                 s->globalmc[ref].zrs[0][0] = 1;

  1069.                 s->globalmc[ref].zrs[1][1] = 1;

  1070.             }

  1071.             /* [DIRAC_STD] perspective(gparams) */

  1072.             if (get_bits1(gb)) {

  1073.                 s->globalmc[ref].perspective_exp = svq3_get_ue_golomb(gb);

  1074.                 s->globalmc[ref].perspective[0]  = dirac_get_se_golomb(gb);

  1075.                 s->globalmc[ref].perspective[1]  = dirac_get_se_golomb(gb);

  1076.             }

  1077.         }

  1078.     }

  1079. 

  1080.     /*[DIRAC_STD] 11.2.7 Picture prediction mode. prediction_mode()

  1081.       Picture prediction mode, not currently used. */

  1082.     if (svq3_get_ue_golomb(gb)) {

  1083.         av_log(s->avctx, AV_LOG_ERROR, "Unknown picture prediction mode\n");

  1084.         return AVERROR_INVALIDDATA;

  1085.     }

  1086. 

  1087.     /* [DIRAC_STD] 11.2.8 Reference picture weight. reference_picture_weights()

  1088.        just data read, weight calculation will be done later on. */

  1089.     s->weight_log2denom = 1;

  1090.     s->weight[0]        = 1;

  1091.     s->weight[1]        = 1;

  1092. 

  1093.     if (get_bits1(gb)) {

  1094.         s->weight_log2denom = svq3_get_ue_golomb(gb);

  1095.         s->weight[0] = dirac_get_se_golomb(gb);

  1096.         if (s->num_refs == 2)

  1097.             s->weight[1] = dirac_get_se_golomb(gb);

  1098.     }

  1099.     return 0;

  1100. }

  1101. 

  1102. /**

  1103.  * Dirac Specification ->

  1104.  * 11.3 Wavelet transform data. wavelet_transform()

  1105.  */

  1106. static int dirac_unpack_idwt_params(DiracContext *s)

  1107. {

  1108.     GetBitContext *gb = &s->gb;

  1109.     int i, level;

  1110.     unsigned tmp;

  1111. 

  1112. #define CHECKEDREAD(dst, cond, errmsg) \

  1113.     tmp = svq3_get_ue_golomb(gb); \

  1114.     if (cond) { \

  1115.         av_log(s->avctx, AV_LOG_ERROR, errmsg); \

  1116.         return AVERROR_INVALIDDATA; \

  1117.     }\

  1118.     dst = tmp;

  1119. 

  1120.     align_get_bits(gb);

  1121. 

  1122.     s->zero_res = s->num_refs ? get_bits1(gb) : 0;

  1123.     if (s->zero_res)

  1124.         return 0;

  1125. 

  1126.     /*[DIRAC_STD] 11.3.1 Transform parameters. transform_parameters() */

  1127.     CHECKEDREAD(s->wavelet_idx, tmp > 6, "wavelet_idx is too big\n")

  1128. 

  1129.     CHECKEDREAD(s->wavelet_depth, tmp > MAX_DWT_LEVELS || tmp < 1, "invalid number of DWT decompositions\n")

  1130. 

  1131.     if (!s->low_delay) {

  1132.         /* Codeblock parameters (core syntax only) */

  1133.         if (get_bits1(gb)) {

  1134.             for (i = 0; i <= s->wavelet_depth; i++) {

  1135.                 CHECKEDREAD(s->codeblock[i].width , tmp < 1 || tmp > (s->avctx->width >>s->wavelet_depth-i), "codeblock width invalid\n")

  1136.                 CHECKEDREAD(s->codeblock[i].height, tmp < 1 || tmp > (s->avctx->height>>s->wavelet_depth-i), "codeblock height invalid\n")

  1137.             }

  1138. 

  1139.             CHECKEDREAD(s->codeblock_mode, tmp > 1, "unknown codeblock mode\n")

  1140.         }

  1141.         else {

  1142.             for (i = 0; i <= s->wavelet_depth; i++)

  1143.                 s->codeblock[i].width = s->codeblock[i].height = 1;

  1144.         }

  1145.     }

  1146.     else {

  1147.         s->num_x        = svq3_get_ue_golomb(gb);

  1148.         s->num_y        = svq3_get_ue_golomb(gb);

  1149.         if (s->ld_picture) {

  1150.             s->lowdelay.bytes.num = svq3_get_ue_golomb(gb);

  1151.             s->lowdelay.bytes.den = svq3_get_ue_golomb(gb);

  1152.             if (s->lowdelay.bytes.den <= 0) {

  1153.                 av_log(s->avctx,AV_LOG_ERROR,"Invalid lowdelay.bytes.den\n");

  1154.                 return AVERROR_INVALIDDATA;

  1155.             }

  1156.         } else if (s->hq_picture) {

  1157.             s->highquality.prefix_bytes = svq3_get_ue_golomb(gb);

  1158.             s->highquality.size_scaler  = svq3_get_ue_golomb(gb);

  1159.         }

  1160. 

  1161.         /* [DIRAC_STD] 11.3.5 Quantisation matrices (low-delay syntax). quant_matrix() */

  1162.         if (get_bits1(gb)) {

  1163.             av_log(s->avctx,AV_LOG_DEBUG,"Low Delay: Has Custom Quantization Matrix!\n");

  1164.             /* custom quantization matrix */

  1165.             s->lowdelay.quant[0][0] = svq3_get_ue_golomb(gb);

  1166.             for (level = 0; level < s->wavelet_depth; level++) {

  1167.                 s->lowdelay.quant[level][1] = svq3_get_ue_golomb(gb);

  1168.                 s->lowdelay.quant[level][2] = svq3_get_ue_golomb(gb);

  1169.                 s->lowdelay.quant[level][3] = svq3_get_ue_golomb(gb);

  1170.             }

  1171.         } else {

  1172.             if (s->wavelet_depth > 4) {

  1173.                 av_log(s->avctx,AV_LOG_ERROR,"Mandatory custom low delay matrix missing for depth %d\n", s->wavelet_depth);

  1174.                 return AVERROR_INVALIDDATA;

  1175.             }

  1176.             /* default quantization matrix */

  1177.             for (level = 0; level < s->wavelet_depth; level++)

  1178.                 for (i = 0; i < 4; i++) {

  1179.                     s->lowdelay.quant[level][i] = ff_dirac_default_qmat[s->wavelet_idx][level][i];

  1180.                     /* haar with no shift differs for different depths */

  1181.                     if (s->wavelet_idx == 3)

  1182.                         s->lowdelay.quant[level][i] += 4*(s->wavelet_depth-1 - level);

  1183.                 }

  1184.         }

  1185.     }

  1186.     return 0;

  1187. }

  1188. 

  1189. static inline int pred_sbsplit(uint8_t *sbsplit, int stride, int x, int y)

  1190. {

  1191.     static const uint8_t avgsplit[7] = { 0, 0, 1, 1, 1, 2, 2 };

  1192. 

  1193.     if (!(x|y))

  1194.         return 0;

  1195.     else if (!y)

  1196.         return sbsplit[-1];

  1197.     else if (!x)

  1198.         return sbsplit[-stride];

  1199. 

  1200.     return avgsplit[sbsplit[-1] + sbsplit[-stride] + sbsplit[-stride-1]];

  1201. }

  1202. 

  1203. static inline int pred_block_mode(DiracBlock *block, int stride, int x, int y, int refmask)

  1204. {

  1205.     int pred;

  1206. 

  1207.     if (!(x|y))

  1208.         return 0;

  1209.     else if (!y)

  1210.         return block[-1].ref & refmask;

  1211.     else if (!x)

  1212.         return block[-stride].ref & refmask;

  1213. 

  1214.     /* return the majority */

  1215.     pred = (block[-1].ref & refmask) + (block[-stride].ref & refmask) + (block[-stride-1].ref & refmask);

  1216.     return (pred >> 1) & refmask;

  1217. }

  1218. 

  1219. static inline void pred_block_dc(DiracBlock *block, int stride, int x, int y)

  1220. {

  1221.     int i, n = 0;

  1222. 

  1223.     memset(block->u.dc, 0, sizeof(block->u.dc));

  1224. 

  1225.     if (x && !(block[-1].ref & 3)) {

  1226.         for (i = 0; i < 3; i++)

  1227.             block->u.dc[i] += block[-1].u.dc[i];

  1228.         n++;

  1229.     }

  1230. 

  1231.     if (y && !(block[-stride].ref & 3)) {

  1232.         for (i = 0; i < 3; i++)

  1233.             block->u.dc[i] += block[-stride].u.dc[i];

  1234.         n++;

  1235.     }

  1236. 

  1237.     if (x && y && !(block[-1-stride].ref & 3)) {

  1238.         for (i = 0; i < 3; i++)

  1239.             block->u.dc[i] += block[-1-stride].u.dc[i];

  1240.         n++;

  1241.     }

  1242. 

  1243.     if (n == 2) {

  1244.         for (i = 0; i < 3; i++)

  1245.             block->u.dc[i] = (block->u.dc[i]+1)>>1;

  1246.     } else if (n == 3) {

  1247.         for (i = 0; i < 3; i++)

  1248.             block->u.dc[i] = divide3(block->u.dc[i]);

  1249.     }

  1250. }

  1251. 

  1252. static inline void pred_mv(DiracBlock *block, int stride, int x, int y, int ref)

  1253. {

  1254.     int16_t *pred[3];

  1255.     int refmask = ref+1;

  1256.     int mask = refmask | DIRAC_REF_MASK_GLOBAL; /*  exclude gmc blocks */

  1257.     int n = 0;

  1258. 

  1259.     if (x && (block[-1].ref & mask) == refmask)

  1260.         pred[n++] = block[-1].u.mv[ref];

  1261. 

  1262.     if (y && (block[-stride].ref & mask) == refmask)

  1263.         pred[n++] = block[-stride].u.mv[ref];

  1264. 

  1265.     if (x && y && (block[-stride-1].ref & mask) == refmask)

  1266.         pred[n++] = block[-stride-1].u.mv[ref];

  1267. 

  1268.     switch (n) {

  1269.     case 0:

  1270.         block->u.mv[ref][0] = 0;

  1271.         block->u.mv[ref][1] = 0;

  1272.         break;

  1273.     case 1:

  1274.         block->u.mv[ref][0] = pred[0][0];

  1275.         block->u.mv[ref][1] = pred[0][1];

  1276.         break;

  1277.     case 2:

  1278.         block->u.mv[ref][0] = (pred[0][0] + pred[1][0] + 1) >> 1;

  1279.         block->u.mv[ref][1] = (pred[0][1] + pred[1][1] + 1) >> 1;

  1280.         break;

  1281.     case 3:

  1282.         block->u.mv[ref][0] = mid_pred(pred[0][0], pred[1][0], pred[2][0]);

  1283.         block->u.mv[ref][1] = mid_pred(pred[0][1], pred[1][1], pred[2][1]);

  1284.         break;

  1285.     }

  1286. }

  1287. 

  1288. static void global_mv(DiracContext *s, DiracBlock *block, int x, int y, int ref)

  1289. {

  1290.     int ez      = s->globalmc[ref].zrs_exp;

  1291.     int ep      = s->globalmc[ref].perspective_exp;

  1292.     int (*A)[2] = s->globalmc[ref].zrs;

  1293.     int *b      = s->globalmc[ref].pan_tilt;

  1294.     int *c      = s->globalmc[ref].perspective;

  1295. 

  1296.     int m       = (1<<ep) - (c[0]*x + c[1]*y);

  1297.     int mx      = m * ((A[0][0] * x + A[0][1]*y) + (1<<ez) * b[0]);

  1298.     int my      = m * ((A[1][0] * x + A[1][1]*y) + (1<<ez) * b[1]);

  1299. 

  1300.     block->u.mv[ref][0] = (mx + (1<<(ez+ep))) >> (ez+ep);

  1301.     block->u.mv[ref][1] = (my + (1<<(ez+ep))) >> (ez+ep);

  1302. }

  1303. 

  1304. static void decode_block_params(DiracContext *s, DiracArith arith[8], DiracBlock *block,

  1305.                                 int stride, int x, int y)

  1306. {

  1307.     int i;

  1308. 

  1309.     block->ref  = pred_block_mode(block, stride, x, y, DIRAC_REF_MASK_REF1);

  1310.     block->ref ^= dirac_get_arith_bit(arith, CTX_PMODE_REF1);

  1311. 

  1312.     if (s->num_refs == 2) {

  1313.         block->ref |= pred_block_mode(block, stride, x, y, DIRAC_REF_MASK_REF2);

  1314.         block->ref ^= dirac_get_arith_bit(arith, CTX_PMODE_REF2) << 1;

  1315.     }

  1316. 

  1317.     if (!block->ref) {

  1318.         pred_block_dc(block, stride, x, y);

  1319.         for (i = 0; i < 3; i++)

  1320.             block->u.dc[i] += dirac_get_arith_int(arith+1+i, CTX_DC_F1, CTX_DC_DATA);

  1321.         return;

  1322.     }

  1323. 

  1324.     if (s->globalmc_flag) {

  1325.         block->ref |= pred_block_mode(block, stride, x, y, DIRAC_REF_MASK_GLOBAL);

  1326.         block->ref ^= dirac_get_arith_bit(arith, CTX_GLOBAL_BLOCK) << 2;

  1327.     }

  1328. 

  1329.     for (i = 0; i < s->num_refs; i++)

  1330.         if (block->ref & (i+1)) {

  1331.             if (block->ref & DIRAC_REF_MASK_GLOBAL) {

  1332.                 global_mv(s, block, x, y, i);

  1333.             } else {

  1334.                 pred_mv(block, stride, x, y, i);

  1335.                 block->u.mv[i][0] += dirac_get_arith_int(arith + 4 + 2 * i, CTX_MV_F1, CTX_MV_DATA);

  1336.                 block->u.mv[i][1] += dirac_get_arith_int(arith + 5 + 2 * i, CTX_MV_F1, CTX_MV_DATA);

  1337.             }

  1338.         }

  1339. }

  1340. 

  1341. /**

  1342.  * Copies the current block to the other blocks covered by the current superblock split mode

  1343.  */

  1344. static void propagate_block_data(DiracBlock *block, int stride, int size)

  1345. {

  1346.     int x, y;

  1347.     DiracBlock *dst = block;

  1348. 

  1349.     for (x = 1; x < size; x++)

  1350.         dst[x] = *block;

  1351. 

  1352.     for (y = 1; y < size; y++) {

  1353.         dst += stride;

  1354.         for (x = 0; x < size; x++)

  1355.             dst[x] = *block;

  1356.     }

  1357. }

  1358. 

  1359. /**

  1360.  * Dirac Specification ->

  1361.  * 12. Block motion data syntax

  1362.  */

  1363. static int dirac_unpack_block_motion_data(DiracContext *s)

  1364. {

  1365.     GetBitContext *gb = &s->gb;

  1366.     uint8_t *sbsplit = s->sbsplit;

  1367.     int i, x, y, q, p;

  1368.     DiracArith arith[8];

  1369. 

  1370.     align_get_bits(gb);

  1371. 

  1372.     /* [DIRAC_STD] 11.2.4 and 12.2.1 Number of blocks and superblocks */

  1373.     s->sbwidth  = DIVRNDUP(s->seq.width,  4*s->plane[0].xbsep);

  1374.     s->sbheight = DIVRNDUP(s->seq.height, 4*s->plane[0].ybsep);

  1375.     s->blwidth  = 4 * s->sbwidth;

  1376.     s->blheight = 4 * s->sbheight;

  1377. 

  1378.     /* [DIRAC_STD] 12.3.1 Superblock splitting modes. superblock_split_modes()

  1379.        decode superblock split modes */

  1380.     ff_dirac_init_arith_decoder(arith, gb, svq3_get_ue_golomb(gb));     /* svq3_get_ue_golomb(gb) is the length */

  1381.     for (y = 0; y < s->sbheight; y++) {

  1382.         for (x = 0; x < s->sbwidth; x++) {

  1383.             unsigned int split  = dirac_get_arith_uint(arith, CTX_SB_F1, CTX_SB_DATA);

  1384.             if (split > 2)

  1385.                 return AVERROR_INVALIDDATA;

  1386.             sbsplit[x] = (split + pred_sbsplit(sbsplit+x, s->sbwidth, x, y)) % 3;

  1387.         }

  1388.         sbsplit += s->sbwidth;

  1389.     }

  1390. 

  1391.     /* setup arith decoding */

  1392.     ff_dirac_init_arith_decoder(arith, gb, svq3_get_ue_golomb(gb));

  1393.     for (i = 0; i < s->num_refs; i++) {

  1394.         ff_dirac_init_arith_decoder(arith + 4 + 2 * i, gb, svq3_get_ue_golomb(gb));

  1395.         ff_dirac_init_arith_decoder(arith + 5 + 2 * i, gb, svq3_get_ue_golomb(gb));

  1396.     }

  1397.     for (i = 0; i < 3; i++)

  1398.         ff_dirac_init_arith_decoder(arith+1+i, gb, svq3_get_ue_golomb(gb));

  1399. 

  1400.     for (y = 0; y < s->sbheight; y++)

  1401.         for (x = 0; x < s->sbwidth; x++) {

  1402.             int blkcnt = 1 << s->sbsplit[y * s->sbwidth + x];

  1403.             int step   = 4 >> s->sbsplit[y * s->sbwidth + x];

  1404. 

  1405.             for (q = 0; q < blkcnt; q++)

  1406.                 for (p = 0; p < blkcnt; p++) {

  1407.                     int bx = 4 * x + p*step;

  1408.                     int by = 4 * y + q*step;

  1409.                     DiracBlock *block = &s->blmotion[by*s->blwidth + bx];

  1410.                     decode_block_params(s, arith, block, s->blwidth, bx, by);

  1411.                     propagate_block_data(block, s->blwidth, step);

  1412.                 }

  1413.         }

  1414. 

  1415.     return 0;

  1416. }

  1417. 

  1418. static int weight(int i, int blen, int offset)

  1419. {

  1420. #define ROLLOFF(i) offset == 1 ? ((i) ? 5 : 3) :        \

  1421.     (1 + (6*(i) + offset - 1) / (2*offset - 1))

  1422. 

  1423.     if (i < 2*offset)

  1424.         return ROLLOFF(i);

  1425.     else if (i > blen-1 - 2*offset)

  1426.         return ROLLOFF(blen-1 - i);

  1427.     return 8;

  1428. }

  1429. 

  1430. static void init_obmc_weight_row(Plane *p, uint8_t *obmc_weight, int stride,

  1431.                                  int left, int right, int wy)

  1432. {

  1433.     int x;

  1434.     for (x = 0; left && x < p->xblen >> 1; x++)

  1435.         obmc_weight[x] = wy*8;

  1436.     for (; x < p->xblen >> right; x++)

  1437.         obmc_weight[x] = wy*weight(x, p->xblen, p->xoffset);

  1438.     for (; x < p->xblen; x++)

  1439.         obmc_weight[x] = wy*8;

  1440.     for (; x < stride; x++)

  1441.         obmc_weight[x] = 0;

  1442. }

  1443. 

  1444. static void init_obmc_weight(Plane *p, uint8_t *obmc_weight, int stride,

  1445.                              int left, int right, int top, int bottom)

  1446. {

  1447.     int y;

  1448.     for (y = 0; top && y < p->yblen >> 1; y++) {

  1449.         init_obmc_weight_row(p, obmc_weight, stride, left, right, 8);

  1450.         obmc_weight += stride;

  1451.     }

  1452.     for (; y < p->yblen >> bottom; y++) {

  1453.         int wy = weight(y, p->yblen, p->yoffset);

  1454.         init_obmc_weight_row(p, obmc_weight, stride, left, right, wy);

  1455.         obmc_weight += stride;

  1456.     }

  1457.     for (; y < p->yblen; y++) {

  1458.         init_obmc_weight_row(p, obmc_weight, stride, left, right, 8);

  1459.         obmc_weight += stride;

  1460.     }

  1461. }

  1462. 

  1463. static void init_obmc_weights(DiracContext *s, Plane *p, int by)

  1464. {

  1465.     int top = !by;

  1466.     int bottom = by == s->blheight-1;

  1467. 

  1468.     /* don't bother re-initing for rows 2 to blheight-2, the weights don't change */

  1469.     if (top || bottom || by == 1) {

  1470.         init_obmc_weight(p, s->obmc_weight[0], MAX_BLOCKSIZE, 1, 0, top, bottom);

  1471.         init_obmc_weight(p, s->obmc_weight[1], MAX_BLOCKSIZE, 0, 0, top, bottom);

  1472.         init_obmc_weight(p, s->obmc_weight[2], MAX_BLOCKSIZE, 0, 1, top, bottom);

  1473.     }

  1474. }

  1475. 

  1476. static const uint8_t epel_weights[4][4][4] = {

  1477.     {{ 16,  0,  0,  0 },

  1478.      { 12,  4,  0,  0 },

  1479.      {  8,  8,  0,  0 },

  1480.      {  4, 12,  0,  0 }},

  1481.     {{ 12,  0,  4,  0 },

  1482.      {  9,  3,  3,  1 },

  1483.      {  6,  6,  2,  2 },

  1484.      {  3,  9,  1,  3 }},

  1485.     {{  8,  0,  8,  0 },

  1486.      {  6,  2,  6,  2 },

  1487.      {  4,  4,  4,  4 },

  1488.      {  2,  6,  2,  6 }},

  1489.     {{  4,  0, 12,  0 },

  1490.      {  3,  1,  9,  3 },

  1491.      {  2,  2,  6,  6 },

  1492.      {  1,  3,  3,  9 }}

  1493. };

  1494. 

  1495. /**

  1496.  * For block x,y, determine which of the hpel planes to do bilinear

  1497.  * interpolation from and set src[] to the location in each hpel plane

  1498.  * to MC from.

  1499.  *

  1500.  * @return the index of the put_dirac_pixels_tab function to use

  1501.  *  0 for 1 plane (fpel,hpel), 1 for 2 planes (qpel), 2 for 4 planes (qpel), and 3 for epel

  1502.  */

  1503. static int mc_subpel(DiracContext *s, DiracBlock *block, const uint8_t *src[5],

  1504.                      int x, int y, int ref, int plane)

  1505. {

  1506.     Plane *p = &s->plane[plane];

  1507.     uint8_t **ref_hpel = s->ref_pics[ref]->hpel[plane];

  1508.     int motion_x = block->u.mv[ref][0];

  1509.     int motion_y = block->u.mv[ref][1];

  1510.     int mx, my, i, epel, nplanes = 0;

  1511. 

  1512.     if (plane) {

  1513.         motion_x >>= s->chroma_x_shift;

  1514.         motion_y >>= s->chroma_y_shift;

  1515.     }

  1516. 

  1517.     mx         = motion_x & ~(-1U << s->mv_precision);

  1518.     my         = motion_y & ~(-1U << s->mv_precision);

  1519.     motion_x >>= s->mv_precision;

  1520.     motion_y >>= s->mv_precision;

  1521.     /* normalize subpel coordinates to epel */

  1522.     /* TODO: template this function? */

  1523.     mx      <<= 3 - s->mv_precision;

  1524.     my      <<= 3 - s->mv_precision;

  1525. 

  1526.     x += motion_x;

  1527.     y += motion_y;

  1528.     epel = (mx|my)&1;

  1529. 

  1530.     /* hpel position */

  1531.     if (!((mx|my)&3)) {

  1532.         nplanes = 1;

  1533.         src[0] = ref_hpel[(my>>1)+(mx>>2)] + y*p->stride + x;

  1534.     } else {

  1535.         /* qpel or epel */

  1536.         nplanes = 4;

  1537.         for (i = 0; i < 4; i++)

  1538.             src[i] = ref_hpel[i] + y*p->stride + x;

  1539. 

  1540.         /* if we're interpolating in the right/bottom halves, adjust the planes as needed

  1541.            we increment x/y because the edge changes for half of the pixels */

  1542.         if (mx > 4) {

  1543.             src[0] += 1;

  1544.             src[2] += 1;

  1545.             x++;

  1546.         }

  1547.         if (my > 4) {

  1548.             src[0] += p->stride;

  1549.             src[1] += p->stride;

  1550.             y++;

  1551.         }

  1552. 

  1553.         /* hpel planes are:

  1554.            [0]: F  [1]: H

  1555.            [2]: V  [3]: C */

  1556.         if (!epel) {

  1557.             /* check if we really only need 2 planes since either mx or my is

  1558.                a hpel position. (epel weights of 0 handle this there) */

  1559.             if (!(mx&3)) {

  1560.                 /* mx == 0: average [0] and [2]

  1561.                    mx == 4: average [1] and [3] */

  1562.                 src[!mx] = src[2 + !!mx];

  1563.                 nplanes = 2;

  1564.             } else if (!(my&3)) {

  1565.                 src[0] = src[(my>>1)  ];

  1566.                 src[1] = src[(my>>1)+1];

  1567.                 nplanes = 2;

  1568.             }

  1569.         } else {

  1570.             /* adjust the ordering if needed so the weights work */

  1571.             if (mx > 4) {

  1572.                 FFSWAP(const uint8_t *, src[0], src[1]);

  1573.                 FFSWAP(const uint8_t *, src[2], src[3]);

  1574.             }

  1575.             if (my > 4) {

  1576.                 FFSWAP(const uint8_t *, src[0], src[2]);

  1577.                 FFSWAP(const uint8_t *, src[1], src[3]);

  1578.             }

  1579.             src[4] = epel_weights[my&3][mx&3];

  1580.         }

  1581.     }

  1582. 

  1583.     /* fixme: v/h _edge_pos */

  1584.     if (x + p->xblen > p->width +EDGE_WIDTH/2 ||

  1585.         y + p->yblen > p->height+EDGE_WIDTH/2 ||

  1586.         x < 0 || y < 0) {

  1587.         for (i = 0; i < nplanes; i++) {

  1588.             s->vdsp.emulated_edge_mc(s->edge_emu_buffer[i], src[i],

  1589.                                      p->stride, p->stride,

  1590.                                      p->xblen, p->yblen, x, y,

  1591.                                      p->width+EDGE_WIDTH/2, p->height+EDGE_WIDTH/2);

  1592.             src[i] = s->edge_emu_buffer[i];

  1593.         }

  1594.     }

  1595.     return (nplanes>>1) + epel;

  1596. }

  1597. 

  1598. static void add_dc(uint16_t *dst, int dc, int stride,

  1599.                    uint8_t *obmc_weight, int xblen, int yblen)

  1600. {

  1601.     int x, y;

  1602.     dc += 128;

  1603. 

  1604.     for (y = 0; y < yblen; y++) {

  1605.         for (x = 0; x < xblen; x += 2) {

  1606.             dst[x  ] += dc * obmc_weight[x  ];

  1607.             dst[x+1] += dc * obmc_weight[x+1];

  1608.         }

  1609.         dst          += stride;

  1610.         obmc_weight  += MAX_BLOCKSIZE;

  1611.     }

  1612. }

  1613. 

  1614. static void block_mc(DiracContext *s, DiracBlock *block,

  1615.                      uint16_t *mctmp, uint8_t *obmc_weight,

  1616.                      int plane, int dstx, int dsty)

  1617. {

  1618.     Plane *p = &s->plane[plane];

  1619.     const uint8_t *src[5];

  1620.     int idx;

  1621. 

  1622.     switch (block->ref&3) {

  1623.     case 0: /* DC */

  1624.         add_dc(mctmp, block->u.dc[plane], p->stride, obmc_weight, p->xblen, p->yblen);

  1625.         return;

  1626.     case 1:

  1627.     case 2:

  1628.         idx = mc_subpel(s, block, src, dstx, dsty, (block->ref&3)-1, plane);

  1629.         s->put_pixels_tab[idx](s->mcscratch, src, p->stride, p->yblen);

  1630.         if (s->weight_func)

  1631.             s->weight_func(s->mcscratch, p->stride, s->weight_log2denom,

  1632.                            s->weight[0] + s->weight[1], p->yblen);

  1633.         break;

  1634.     case 3:

  1635.         idx = mc_subpel(s, block, src, dstx, dsty, 0, plane);

  1636.         s->put_pixels_tab[idx](s->mcscratch, src, p->stride, p->yblen);

  1637.         idx = mc_subpel(s, block, src, dstx, dsty, 1, plane);

  1638.         if (s->biweight_func) {

  1639.             /* fixme: +32 is a quick hack */

  1640.             s->put_pixels_tab[idx](s->mcscratch + 32, src, p->stride, p->yblen);

  1641.             s->biweight_func(s->mcscratch, s->mcscratch+32, p->stride, s->weight_log2denom,

  1642.                              s->weight[0], s->weight[1], p->yblen);

  1643.         } else

  1644.             s->avg_pixels_tab[idx](s->mcscratch, src, p->stride, p->yblen);

  1645.         break;

  1646.     }

  1647.     s->add_obmc(mctmp, s->mcscratch, p->stride, obmc_weight, p->yblen);

  1648. }

  1649. 

  1650. static void mc_row(DiracContext *s, DiracBlock *block, uint16_t *mctmp, int plane, int dsty)

  1651. {

  1652.     Plane *p = &s->plane[plane];

  1653.     int x, dstx = p->xbsep - p->xoffset;

  1654. 

  1655.     block_mc(s, block, mctmp, s->obmc_weight[0], plane, -p->xoffset, dsty);

  1656.     mctmp += p->xbsep;

  1657. 

  1658.     for (x = 1; x < s->blwidth-1; x++) {

  1659.         block_mc(s, block+x, mctmp, s->obmc_weight[1], plane, dstx, dsty);

  1660.         dstx  += p->xbsep;

  1661.         mctmp += p->xbsep;

  1662.     }

  1663.     block_mc(s, block+x, mctmp, s->obmc_weight[2], plane, dstx, dsty);

  1664. }

  1665. 

  1666. static void select_dsp_funcs(DiracContext *s, int width, int height, int xblen, int yblen)

  1667. {

  1668.     int idx = 0;

  1669.     if (xblen > 8)

  1670.         idx = 1;

  1671.     if (xblen > 16)

  1672.         idx = 2;

  1673. 

  1674.     memcpy(s->put_pixels_tab, s->diracdsp.put_dirac_pixels_tab[idx], sizeof(s->put_pixels_tab));

  1675.     memcpy(s->avg_pixels_tab, s->diracdsp.avg_dirac_pixels_tab[idx], sizeof(s->avg_pixels_tab));

  1676.     s->add_obmc = s->diracdsp.add_dirac_obmc[idx];

  1677.     if (s->weight_log2denom > 1 || s->weight[0] != 1 || s->weight[1] != 1) {

  1678.         s->weight_func   = s->diracdsp.weight_dirac_pixels_tab[idx];

  1679.         s->biweight_func = s->diracdsp.biweight_dirac_pixels_tab[idx];

  1680.     } else {

  1681.         s->weight_func   = NULL;

  1682.         s->biweight_func = NULL;

  1683.     }

  1684. }

  1685. 

  1686. static int interpolate_refplane(DiracContext *s, DiracFrame *ref, int plane, int width, int height)

  1687. {

  1688.     /* chroma allocates an edge of 8 when subsampled

  1689.        which for 4:2:2 means an h edge of 16 and v edge of 8

  1690.        just use 8 for everything for the moment */

  1691.     int i, edge = EDGE_WIDTH/2;

  1692. 

  1693.     ref->hpel[plane][0] = ref->avframe->data[plane];

  1694.     s->mpvencdsp.draw_edges(ref->hpel[plane][0], ref->avframe->linesize[plane], width, height, edge, edge, EDGE_TOP | EDGE_BOTTOM); /* EDGE_TOP | EDGE_BOTTOM values just copied to make it build, this needs to be ensured */

  1695. 

  1696.     /* no need for hpel if we only have fpel vectors */

  1697.     if (!s->mv_precision)

  1698.         return 0;

  1699. 

  1700.     for (i = 1; i < 4; i++) {

  1701.         if (!ref->hpel_base[plane][i])

  1702.             ref->hpel_base[plane][i] = av_malloc((height+2*edge) * ref->avframe->linesize[plane] + 32);

  1703.         if (!ref->hpel_base[plane][i]) {

  1704.             return AVERROR(ENOMEM);

  1705.         }

  1706.         /* we need to be 16-byte aligned even for chroma */

  1707.         ref->hpel[plane][i] = ref->hpel_base[plane][i] + edge*ref->avframe->linesize[plane] + 16;

  1708.     }

  1709. 

  1710.     if (!ref->interpolated[plane]) {

  1711.         s->diracdsp.dirac_hpel_filter(ref->hpel[plane][1], ref->hpel[plane][2],

  1712.                                       ref->hpel[plane][3], ref->hpel[plane][0],

  1713.                                       ref->avframe->linesize[plane], width, height);

  1714.         s->mpvencdsp.draw_edges(ref->hpel[plane][1], ref->avframe->linesize[plane], width, height, edge, edge, EDGE_TOP | EDGE_BOTTOM);

  1715.         s->mpvencdsp.draw_edges(ref->hpel[plane][2], ref->avframe->linesize[plane], width, height, edge, edge, EDGE_TOP | EDGE_BOTTOM);

  1716.         s->mpvencdsp.draw_edges(ref->hpel[plane][3], ref->avframe->linesize[plane], width, height, edge, edge, EDGE_TOP | EDGE_BOTTOM);

  1717.     }

  1718.     ref->interpolated[plane] = 1;

  1719. 

  1720.     return 0;

  1721. }

  1722. 

  1723. /**

  1724.  * Dirac Specification ->

  1725.  * 13.0 Transform data syntax. transform_data()

  1726.  */

  1727. static int dirac_decode_frame_internal(DiracContext *s)

  1728. {

  1729.     DWTContext d;

  1730.     int y, i, comp, dsty;

  1731.     int ret;

  1732. 

  1733.     if (s->low_delay) {

  1734.         /* [DIRAC_STD] 13.5.1 low_delay_transform_data() */

  1735.         for (comp = 0; comp < 3; comp++) {

  1736.             Plane *p = &s->plane[comp];

  1737.             memset(p->idwt_buf, 0, p->idwt_stride * p->idwt_height);

  1738.         }

  1739.         if (!s->zero_res) {

  1740.             if ((ret = decode_lowdelay(s)) < 0)

  1741.                 return ret;

  1742.         }

  1743.     }

  1744. 

  1745.     for (comp = 0; comp < 3; comp++) {

  1746.         Plane *p       = &s->plane[comp];

  1747.         uint8_t *frame = s->current_picture->avframe->data[comp];

  1748. 

  1749.         /* FIXME: small resolutions */

  1750.         for (i = 0; i < 4; i++)

  1751.             s->edge_emu_buffer[i] = s->edge_emu_buffer_base + i*FFALIGN(p->width, 16);

  1752. 

  1753.         if (!s->zero_res && !s->low_delay)

  1754.         {

  1755.             memset(p->idwt_buf, 0, p->idwt_stride * p->idwt_height);

  1756.             decode_component(s, comp); /* [DIRAC_STD] 13.4.1 core_transform_data() */

  1757.         }

  1758.         ret = ff_spatial_idwt_init(&d, p->idwt_buf, p->idwt_width, p->idwt_height, p->idwt_stride,

  1759.                                    s->wavelet_idx+2, s->wavelet_depth, p->idwt_tmp, s->bit_depth);

  1760.         if (ret < 0)

  1761.             return ret;

  1762. 

  1763.         if (!s->num_refs) { /* intra */

  1764.             for (y = 0; y < p->height; y += 16) {

  1765.                 int idx = (s->bit_depth - 8) >> 1;

  1766.                 ff_spatial_idwt_slice2(&d, y+16); /* decode */

  1767.                 s->diracdsp.put_signed_rect_clamped[idx](frame + y*p->stride,

  1768.                                                          p->stride,

  1769.                                                          p->idwt_buf + y*p->idwt_stride,

  1770.                                                          p->idwt_stride, p->width, 16);

  1771.             }

  1772.         } else { /* inter */

  1773.             int rowheight = p->ybsep*p->stride;

  1774. 

  1775.             select_dsp_funcs(s, p->width, p->height, p->xblen, p->yblen);

  1776. 

  1777.             for (i = 0; i < s->num_refs; i++) {

  1778.                 int ret = interpolate_refplane(s, s->ref_pics[i], comp, p->width, p->height);

  1779.                 if (ret < 0)

  1780.                     return ret;

  1781.             }

  1782. 

  1783.             memset(s->mctmp, 0, 4*p->yoffset*p->stride);

  1784. 

  1785.             dsty = -p->yoffset;

  1786.             for (y = 0; y < s->blheight; y++) {

  1787.                 int h     = 0,

  1788.                     start = FFMAX(dsty, 0);

  1789.                 uint16_t *mctmp    = s->mctmp + y*rowheight;

  1790.                 DiracBlock *blocks = s->blmotion + y*s->blwidth;

  1791. 

  1792.                 init_obmc_weights(s, p, y);

  1793. 

  1794.                 if (y == s->blheight-1 || start+p->ybsep > p->height)

  1795.                     h = p->height - start;

  1796.                 else

  1797.                     h = p->ybsep - (start - dsty);

  1798.                 if (h < 0)

  1799.                     break;

  1800. 

  1801.                 memset(mctmp+2*p->yoffset*p->stride, 0, 2*rowheight);

  1802.                 mc_row(s, blocks, mctmp, comp, dsty);

  1803. 

  1804.                 mctmp += (start - dsty)*p->stride + p->xoffset;

  1805.                 ff_spatial_idwt_slice2(&d, start + h); /* decode */

  1806.                 /* NOTE: add_rect_clamped hasn't been templated hence the shifts.

  1807.                  * idwt_stride is passed as pixels, not in bytes as in the rest of the decoder */

  1808.                 s->diracdsp.add_rect_clamped(frame + start*p->stride, mctmp, p->stride,

  1809.                                              (int16_t*)(p->idwt_buf) + start*(p->idwt_stride >> 1), (p->idwt_stride >> 1), p->width, h);

  1810. 

  1811.                 dsty += p->ybsep;

  1812.             }

  1813.         }

  1814.     }

  1815. 

  1816. 

  1817.     return 0;

  1818. }

  1819. 

  1820. static int get_buffer_with_edge(AVCodecContext *avctx, AVFrame *f, int flags)

  1821. {

  1822.     int ret, i;

  1823.     int chroma_x_shift, chroma_y_shift;

  1824.     avcodec_get_chroma_sub_sample(avctx->pix_fmt, &chroma_x_shift, &chroma_y_shift);

  1825. 

  1826.     f->width  = avctx->width  + 2 * EDGE_WIDTH;

  1827.     f->height = avctx->height + 2 * EDGE_WIDTH + 2;

  1828.     ret = ff_get_buffer(avctx, f, flags);

  1829.     if (ret < 0)

  1830.         return ret;

  1831. 

  1832.     for (i = 0; f->data[i]; i++) {

  1833.         int offset = (EDGE_WIDTH >> (i && i<3 ? chroma_y_shift : 0)) *

  1834.                      f->linesize[i] + 32;

  1835.         f->data[i] += offset;

  1836.     }

  1837.     f->width  = avctx->width;

  1838.     f->height = avctx->height;

  1839. 

  1840.     return 0;

  1841. }

  1842. 

  1843. /**

  1844.  * Dirac Specification ->

  1845.  * 11.1.1 Picture Header. picture_header()

  1846.  */

  1847. static int dirac_decode_picture_header(DiracContext *s)

  1848. {

  1849.     unsigned retire, picnum;

  1850.     int i, j, ret;

  1851.     int64_t refdist, refnum;

  1852.     GetBitContext *gb = &s->gb;

  1853. 

  1854.     /* [DIRAC_STD] 11.1.1 Picture Header. picture_header() PICTURE_NUM */

  1855.     picnum = s->current_picture->avframe->display_picture_number = get_bits_long(gb, 32);

  1856. 

  1857. 

  1858.     av_log(s->avctx,AV_LOG_DEBUG,"PICTURE_NUM: %d\n",picnum);

  1859. 

  1860.     /* if this is the first keyframe after a sequence header, start our

  1861.        reordering from here */

  1862.     if (s->frame_number < 0)

  1863.         s->frame_number = picnum;

  1864. 

  1865.     s->ref_pics[0] = s->ref_pics[1] = NULL;

  1866.     for (i = 0; i < s->num_refs; i++) {

  1867.         refnum = (picnum + dirac_get_se_golomb(gb)) & 0xFFFFFFFF;

  1868.         refdist = INT64_MAX;

  1869. 

  1870.         /* find the closest reference to the one we want */

  1871.         /* Jordi: this is needed if the referenced picture hasn't yet arrived */

  1872.         for (j = 0; j < MAX_REFERENCE_FRAMES && refdist; j++)

  1873.             if (s->ref_frames[j]

  1874.                 && FFABS(s->ref_frames[j]->avframe->display_picture_number - refnum) < refdist) {

  1875.                 s->ref_pics[i] = s->ref_frames[j];

  1876.                 refdist = FFABS(s->ref_frames[j]->avframe->display_picture_number - refnum);

  1877.             }

  1878. 

  1879.         if (!s->ref_pics[i] || refdist)

  1880.             av_log(s->avctx, AV_LOG_DEBUG, "Reference not found\n");

  1881. 

  1882.         /* if there were no references at all, allocate one */

  1883.         if (!s->ref_pics[i])

  1884.             for (j = 0; j < MAX_FRAMES; j++)

  1885.                 if (!s->all_frames[j].avframe->data[0]) {

  1886.                     s->ref_pics[i] = &s->all_frames[j];

  1887.                     get_buffer_with_edge(s->avctx, s->ref_pics[i]->avframe, AV_GET_BUFFER_FLAG_REF);

  1888.                     break;

  1889.                 }

  1890. 

  1891.         if (!s->ref_pics[i]) {

  1892.             av_log(s->avctx, AV_LOG_ERROR, "Reference could not be allocated\n");

  1893.             return AVERROR_INVALIDDATA;

  1894.         }

  1895. 

  1896.     }

  1897. 

  1898.     /* retire the reference frames that are not used anymore */

  1899.     if (s->current_picture->reference) {

  1900.         retire = (picnum + dirac_get_se_golomb(gb)) & 0xFFFFFFFF;

  1901.         if (retire != picnum) {

  1902.             DiracFrame *retire_pic = remove_frame(s->ref_frames, retire);

  1903. 

  1904.             if (retire_pic)

  1905.                 retire_pic->reference &= DELAYED_PIC_REF;

  1906.             else

  1907.                 av_log(s->avctx, AV_LOG_DEBUG, "Frame to retire not found\n");

  1908.         }

  1909. 

  1910.         /* if reference array is full, remove the oldest as per the spec */

  1911.         while (add_frame(s->ref_frames, MAX_REFERENCE_FRAMES, s->current_picture)) {

  1912.             av_log(s->avctx, AV_LOG_ERROR, "Reference frame overflow\n");

  1913.             remove_frame(s->ref_frames, s->ref_frames[0]->avframe->display_picture_number)->reference &= DELAYED_PIC_REF;

  1914.         }

  1915.     }

  1916. 

  1917.     if (s->num_refs) {

  1918.         ret = dirac_unpack_prediction_parameters(s);  /* [DIRAC_STD] 11.2 Picture Prediction Data. picture_prediction() */

  1919.         if (ret < 0)

  1920.             return ret;

  1921.         ret = dirac_unpack_block_motion_data(s);      /* [DIRAC_STD] 12. Block motion data syntax                       */

  1922.         if (ret < 0)

  1923.             return ret;

  1924.     }

  1925.     ret = dirac_unpack_idwt_params(s);                /* [DIRAC_STD] 11.3 Wavelet transform data                        */

  1926.     if (ret < 0)

  1927.         return ret;

  1928. 

  1929.     init_planes(s);

  1930.     return 0;

  1931. }

  1932. 

  1933. static int get_delayed_pic(DiracContext *s, AVFrame *picture, int *got_frame)

  1934. {

  1935.     DiracFrame *out = s->delay_frames[0];

  1936.     int i, out_idx  = 0;

  1937.     int ret;

  1938. 

  1939.     /* find frame with lowest picture number */

  1940.     for (i = 1; s->delay_frames[i]; i++)

  1941.         if (s->delay_frames[i]->avframe->display_picture_number < out->avframe->display_picture_number) {

  1942.             out     = s->delay_frames[i];

  1943.             out_idx = i;

  1944.         }

  1945. 

  1946.     for (i = out_idx; s->delay_frames[i]; i++)

  1947.         s->delay_frames[i] = s->delay_frames[i+1];

  1948. 

  1949.     if (out) {

  1950.         out->reference ^= DELAYED_PIC_REF;

  1951.         *got_frame = 1;

  1952.         if((ret = av_frame_ref(picture, out->avframe)) < 0)

  1953.             return ret;

  1954.     }

  1955. 

  1956.     return 0;

  1957. }

  1958. 

  1959. /**

  1960.  * Dirac Specification ->

  1961.  * 9.6 Parse Info Header Syntax. parse_info()

  1962.  * 4 byte start code + byte parse code + 4 byte size + 4 byte previous size

  1963.  */

  1964. #define DATA_UNIT_HEADER_SIZE 13

  1965. 

  1966. /* [DIRAC_STD] dirac_decode_data_unit makes reference to the while defined in 9.3

  1967.    inside the function parse_sequence() */

  1968. static int dirac_decode_data_unit(AVCodecContext *avctx, const uint8_t *buf, int size)

  1969. {

  1970.     DiracContext *s   = avctx->priv_data;

  1971.     DiracFrame *pic   = NULL;

  1972.     AVDiracSeqHeader *dsh;

  1973.     int ret, i;

  1974.     uint8_t parse_code;

  1975.     unsigned tmp;

  1976. 

  1977.     if (size < DATA_UNIT_HEADER_SIZE)

  1978.         return AVERROR_INVALIDDATA;

  1979. 

  1980.     parse_code = buf[4];

  1981. 

  1982.     init_get_bits(&s->gb, &buf[13], 8*(size - DATA_UNIT_HEADER_SIZE));

  1983. 

  1984.     if (parse_code == DIRAC_PCODE_SEQ_HEADER) {

  1985.         if (s->seen_sequence_header)

  1986.             return 0;

  1987. 

  1988.         /* [DIRAC_STD] 10. Sequence header */

  1989.         ret = av_dirac_parse_sequence_header(&dsh, buf + DATA_UNIT_HEADER_SIZE, size - DATA_UNIT_HEADER_SIZE, avctx);

  1990.         if (ret < 0) {

  1991.             av_log(avctx, AV_LOG_ERROR, "error parsing sequence header");

  1992.             return ret;

  1993.         }

  1994. 

  1995.         ret = ff_set_dimensions(avctx, dsh->width, dsh->height);

  1996.         if (ret < 0) {

  1997.             av_freep(&dsh);

  1998.             return ret;

  1999.         }

  2000. 

  2001.         ff_set_sar(avctx, dsh->sample_aspect_ratio);

  2002.         avctx->pix_fmt         = dsh->pix_fmt;

  2003.         avctx->color_range     = dsh->color_range;

  2004.         avctx->color_trc       = dsh->color_trc;

  2005.         avctx->color_primaries = dsh->color_primaries;

  2006.         avctx->colorspace      = dsh->colorspace;

  2007.         avctx->profile         = dsh->profile;

  2008.         avctx->level           = dsh->level;

  2009.         avctx->framerate       = dsh->framerate;

  2010.         s->bit_depth           = dsh->bit_depth;

  2011.         s->version.major       = dsh->version.major;

  2012.         s->version.minor       = dsh->version.minor;

  2013.         s->seq                 = *dsh;

  2014.         av_freep(&dsh);

  2015. 

  2016.         s->pshift = s->bit_depth > 8;

  2017. 

  2018.         avcodec_get_chroma_sub_sample(avctx->pix_fmt, &s->chroma_x_shift, &s->chroma_y_shift);

  2019. 

  2020.         ret = alloc_sequence_buffers(s);

  2021.         if (ret < 0)

  2022.             return ret;

  2023. 

  2024.         s->seen_sequence_header = 1;

  2025.     } else if (parse_code == DIRAC_PCODE_END_SEQ) { /* [DIRAC_STD] End of Sequence */

  2026.         free_sequence_buffers(s);

  2027.         s->seen_sequence_header = 0;

  2028.     } else if (parse_code == DIRAC_PCODE_AUX) {

  2029.         if (buf[13] == 1) {     /* encoder implementation/version */

  2030.             int ver[3];

  2031.             /* versions older than 1.0.8 don't store quant delta for

  2032.                subbands with only one codeblock */

  2033.             if (sscanf(buf+14, "Schroedinger %d.%d.%d", ver, ver+1, ver+2) == 3)

  2034.                 if (ver[0] == 1 && ver[1] == 0 && ver[2] <= 7)

  2035.                     s->old_delta_quant = 1;

  2036.         }

  2037.     } else if (parse_code & 0x8) {  /* picture data unit */

  2038.         if (!s->seen_sequence_header) {

  2039.             av_log(avctx, AV_LOG_DEBUG, "Dropping frame without sequence header\n");

  2040.             return AVERROR_INVALIDDATA;

  2041.         }

  2042. 

  2043.         /* find an unused frame */

  2044.         for (i = 0; i < MAX_FRAMES; i++)

  2045.             if (s->all_frames[i].avframe->data[0] == NULL)

  2046.                 pic = &s->all_frames[i];

  2047.         if (!pic) {

  2048.             av_log(avctx, AV_LOG_ERROR, "framelist full\n");

  2049.             return AVERROR_INVALIDDATA;

  2050.         }

  2051. 

  2052.         av_frame_unref(pic->avframe);

  2053. 

  2054.         /* [DIRAC_STD] Defined in 9.6.1 ... */

  2055.         tmp            =  parse_code & 0x03;                   /* [DIRAC_STD] num_refs()      */

  2056.         if (tmp > 2) {

  2057.             av_log(avctx, AV_LOG_ERROR, "num_refs of 3\n");

  2058.             return AVERROR_INVALIDDATA;

  2059.         }

  2060.         s->num_refs      = tmp;

  2061.         s->is_arith      = (parse_code & 0x48) == 0x08;          /* [DIRAC_STD] using_ac()            */

  2062.         s->low_delay     = (parse_code & 0x88) == 0x88;          /* [DIRAC_STD] is_low_delay()        */

  2063.         s->core_syntax   = (parse_code & 0x88) == 0x08;          /* [DIRAC_STD] is_core_syntax()      */

  2064.         s->ld_picture    = (parse_code & 0xF8) == 0xC8;          /* [DIRAC_STD] is_ld_picture()       */

  2065.         s->hq_picture    = (parse_code & 0xF8) == 0xE8;          /* [DIRAC_STD] is_hq_picture()       */

  2066.         s->dc_prediction = (parse_code & 0x28) == 0x08;          /* [DIRAC_STD] using_dc_prediction() */

  2067.         pic->reference   = (parse_code & 0x0C) == 0x0C;          /* [DIRAC_STD] is_reference()        */

  2068.         pic->avframe->key_frame = s->num_refs == 0;              /* [DIRAC_STD] is_intra()            */

  2069.         pic->avframe->pict_type = s->num_refs + 1;               /* Definition of AVPictureType in avutil.h */

  2070. 

  2071.         /* VC-2 Low Delay has a different parse code than the Dirac Low Delay */

  2072.         if (s->version.minor == 2 && parse_code == 0x88)

  2073.             s->ld_picture = 1;

  2074. 

  2075.         if (s->low_delay && !(s->ld_picture || s->hq_picture) ) {

  2076.             av_log(avctx, AV_LOG_ERROR, "Invalid low delay flag\n");

  2077.             return AVERROR_INVALIDDATA;

  2078.         }

  2079. 

  2080.         if ((ret = get_buffer_with_edge(avctx, pic->avframe, (parse_code & 0x0C) == 0x0C ? AV_GET_BUFFER_FLAG_REF : 0)) < 0)

  2081.             return ret;

  2082.         s->current_picture = pic;

  2083.         s->plane[0].stride = pic->avframe->linesize[0];

  2084.         s->plane[1].stride = pic->avframe->linesize[1];

  2085.         s->plane[2].stride = pic->avframe->linesize[2];

  2086. 

  2087.         if (alloc_buffers(s, FFMAX3(FFABS(s->plane[0].stride), FFABS(s->plane[1].stride), FFABS(s->plane[2].stride))) < 0)

  2088.             return AVERROR(ENOMEM);

  2089. 

  2090.         /* [DIRAC_STD] 11.1 Picture parse. picture_parse() */

  2091.         ret = dirac_decode_picture_header(s);

  2092.         if (ret < 0)

  2093.             return ret;

  2094. 

  2095.         /* [DIRAC_STD] 13.0 Transform data syntax. transform_data() */

  2096.         ret = dirac_decode_frame_internal(s);

  2097.         if (ret < 0)

  2098.             return ret;

  2099.     }

  2100.     return 0;

  2101. }

  2102. 

  2103. static int dirac_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *pkt)

  2104. {

  2105.     DiracContext *s     = avctx->priv_data;

  2106.     AVFrame *picture    = data;

  2107.     uint8_t *buf        = pkt->data;

  2108.     int buf_size        = pkt->size;

  2109.     int i, buf_idx      = 0;

  2110.     int ret;

  2111.     unsigned data_unit_size;

  2112. 

  2113.     /* release unused frames */

  2114.     for (i = 0; i < MAX_FRAMES; i++)

  2115.         if (s->all_frames[i].avframe->data[0] && !s->all_frames[i].reference) {

  2116.             av_frame_unref(s->all_frames[i].avframe);

  2117.             memset(s->all_frames[i].interpolated, 0, sizeof(s->all_frames[i].interpolated));

  2118.         }

  2119. 

  2120.     s->current_picture = NULL;

  2121.     *got_frame = 0;

  2122. 

  2123.     /* end of stream, so flush delayed pics */

  2124.     if (buf_size == 0)

  2125.         return get_delayed_pic(s, (AVFrame *)data, got_frame);

  2126. 

  2127.     for (;;) {

  2128.         /*[DIRAC_STD] Here starts the code from parse_info() defined in 9.6

  2129.           [DIRAC_STD] PARSE_INFO_PREFIX = "BBCD" as defined in ISO/IEC 646

  2130.           BBCD start code search */

  2131.         for (; buf_idx + DATA_UNIT_HEADER_SIZE < buf_size; buf_idx++) {

  2132.             if (buf[buf_idx  ] == 'B' && buf[buf_idx+1] == 'B' &&

  2133.                 buf[buf_idx+2] == 'C' && buf[buf_idx+3] == 'D')

  2134.                 break;

  2135.         }

  2136.         /* BBCD found or end of data */

  2137.         if (buf_idx + DATA_UNIT_HEADER_SIZE >= buf_size)

  2138.             break;

  2139. 

  2140.         data_unit_size = AV_RB32(buf+buf_idx+5);

  2141.         if (data_unit_size > buf_size - buf_idx || !data_unit_size) {

  2142.             if(data_unit_size > buf_size - buf_idx)

  2143.             av_log(s->avctx, AV_LOG_ERROR,

  2144.                    "Data unit with size %d is larger than input buffer, discarding\n",

  2145.                    data_unit_size);

  2146.             buf_idx += 4;

  2147.             continue;

  2148.         }

  2149.         /* [DIRAC_STD] dirac_decode_data_unit makes reference to the while defined in 9.3 inside the function parse_sequence() */

  2150.         ret = dirac_decode_data_unit(avctx, buf+buf_idx, data_unit_size);

  2151.         if (ret < 0)

  2152.         {

  2153.             av_log(s->avctx, AV_LOG_ERROR,"Error in dirac_decode_data_unit\n");

  2154.             return ret;

  2155.         }

  2156.         buf_idx += data_unit_size;

  2157.     }

  2158. 

  2159.     if (!s->current_picture)

  2160.         return buf_size;

  2161. 

  2162.     if (s->current_picture->avframe->display_picture_number > s->frame_number) {

  2163.         DiracFrame *delayed_frame = remove_frame(s->delay_frames, s->frame_number);

  2164. 

  2165.         s->current_picture->reference |= DELAYED_PIC_REF;

  2166. 

  2167.         if (add_frame(s->delay_frames, MAX_DELAY, s->current_picture)) {

  2168.             int min_num = s->delay_frames[0]->avframe->display_picture_number;

  2169.             /* Too many delayed frames, so we display the frame with the lowest pts */

  2170.             av_log(avctx, AV_LOG_ERROR, "Delay frame overflow\n");

  2171. 

  2172.             for (i = 1; s->delay_frames[i]; i++)

  2173.                 if (s->delay_frames[i]->avframe->display_picture_number < min_num)

  2174.                     min_num = s->delay_frames[i]->avframe->display_picture_number;

  2175. 

  2176.             delayed_frame = remove_frame(s->delay_frames, min_num);

  2177.             add_frame(s->delay_frames, MAX_DELAY, s->current_picture);

  2178.         }

  2179. 

  2180.         if (delayed_frame) {

  2181.             delayed_frame->reference ^= DELAYED_PIC_REF;

  2182.             if((ret=av_frame_ref(data, delayed_frame->avframe)) < 0)

  2183.                 return ret;

  2184.             *got_frame = 1;

  2185.         }

  2186.     } else if (s->current_picture->avframe->display_picture_number == s->frame_number) {

  2187.         /* The right frame at the right time :-) */

  2188.         if((ret=av_frame_ref(data, s->current_picture->avframe)) < 0)

  2189.             return ret;

  2190.         *got_frame = 1;

  2191.     }

  2192. 

  2193.     if (*got_frame)

  2194.         s->frame_number = picture->display_picture_number + 1;

  2195. 

  2196.     return buf_idx;

  2197. }

  2198. 

  2199. AVCodec ff_dirac_decoder = {

  2200.     .name           = "dirac",

  2201.     .long_name      = NULL_IF_CONFIG_SMALL("BBC Dirac VC-2"),

  2202.     .type           = AVMEDIA_TYPE_VIDEO,

  2203.     .id             = AV_CODEC_ID_DIRAC,

  2204.     .priv_data_size = sizeof(DiracContext),

  2205.     .init           = dirac_decode_init,

  2206.     .close          = dirac_decode_end,

  2207.     .decode         = dirac_decode_frame,

  2208.     .capabilities   = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS | AV_CODEC_CAP_DR1,

  2209.     .flush          = dirac_decode_flush,

  2210. };