vp9: initialize color space and range properties
[ffmpeg.git] / libavcodec / vp9.c
index 9048700..50b84ae 100644 (file)
@@ -1040,6 +1040,18 @@ static int vp9_decode_frame(AVCodecContext *avctx, AVFrame *frame,
     s->cur_frame->pict_type = s->keyframe ? AV_PICTURE_TYPE_I
                                           : AV_PICTURE_TYPE_P;
 
+    if (s->fullrange)
+        avctx->color_range = AVCOL_RANGE_JPEG;
+    else
+        avctx->color_range = AVCOL_RANGE_MPEG;
+
+    switch (s->colorspace) {
+    case 1: avctx->colorspace = AVCOL_SPC_BT470BG; break;
+    case 2: avctx->colorspace = AVCOL_SPC_BT709; break;
+    case 3: avctx->colorspace = AVCOL_SPC_SMPTE170M; break;
+    case 4: avctx->colorspace = AVCOL_SPC_SMPTE240M; break;
+    }
+
     // main tile decode loop
     memset(s->above_partition_ctx, 0, s->cols);
     memset(s->above_skip_ctx, 0, s->cols);
@@ -1144,8 +1156,17 @@ static int vp9_decode_frame(AVCodecContext *avctx, AVFrame *frame,
     // probability maintenance between frames)
     if (s->refreshctx) {
         if (s->parallelmode) {
-            memcpy(s->prob_ctx[s->framectxid].coef, s->prob.coef,
-                   sizeof(s->prob.coef));
+            int j, k, l, m;
+            for (i = 0; i < 4; i++) {
+                for (j = 0; j < 2; j++)
+                    for (k = 0; k < 2; k++)
+                        for (l = 0; l < 6; l++)
+                            for (m = 0; m < 6; m++)
+                                memcpy(s->prob_ctx[s->framectxid].coef[i][j][k][l][m],
+                                       s->prob.coef[i][j][k][l][m], 3);
+                if (s->txfmmode == i)
+                    break;
+            }
             s->prob_ctx[s->framectxid].p = s->prob.p;
         } else {
             ff_vp9_adapt_probs(s);
@@ -1189,7 +1210,7 @@ static int vp9_decode_packet(AVCodecContext *avctx, void *frame,
             const uint8_t *idx = data + size + 1 - idx_sz;
 
             while (n_frames--) {
-                int sz = AV_RL32(idx);
+                unsigned sz = AV_RL32(idx);
 
                 if (nbytes < 4)
                     sz &= (1 << (8 * nbytes)) - 1;
@@ -1197,7 +1218,7 @@ static int vp9_decode_packet(AVCodecContext *avctx, void *frame,
 
                 if (sz > size) {
                     av_log(avctx, AV_LOG_ERROR,
-                           "Superframe packet size too big: %d > %d\n",
+                           "Superframe packet size too big: %u > %d\n",
                            sz, size);
                     return AVERROR_INVALIDDATA;
                 }