File: prores_parser.c

package info (click to toggle)
ffmpeg 7%3A8.1-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 113,964 kB
  • sloc: ansic: 1,376,567; asm: 153,816; sh: 9,602; makefile: 5,414; cpp: 5,172; lisp: 1,771; perl: 1,463; objc: 1,058; python: 120; awk: 56; ruby: 51
file content (129 lines) | stat: -rw-r--r-- 4,349 bytes parent folder | download | duplicates (6)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/*
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include "libavutil/intreadwrite.h"
#include "bytestream.h"

#include "avcodec.h"
#include "parser_internal.h"

static int parse(AVCodecParserContext *s,
                 AVCodecContext *avctx,
                 const uint8_t **poutbuf, int *poutbuf_size,
                 const uint8_t *buf, int buf_size)
{
    GetByteContext gb;
    uint8_t flags, depth, chroma_format, alpha_channel_type;

    *poutbuf      = buf;
    *poutbuf_size = buf_size;

    /* Frame fields + frame header size */
    if (buf_size < 28)
        return buf_size;

    bytestream2_init(&gb, buf, buf_size);

    /* Frame size */
    if (bytestream2_get_be32(&gb) != buf_size)
        return buf_size;

    /* Frame identifier */
    if (bytestream2_get_le32(&gb) != MKTAG('i','c','p','f'))
        return buf_size;

    /* Frame header size */
    if (bytestream2_get_be16(&gb) < 20)
        return buf_size;

    bytestream2_skip(&gb, 6); /* Bitstream version, encoder identifier */

    s->key_frame = 1;
    s->pict_type = AV_PICTURE_TYPE_I;

    s->width  = bytestream2_get_be16(&gb);
    s->height = bytestream2_get_be16(&gb);
    s->coded_width  = FFALIGN(s->width,  16);
    s->coded_height = FFALIGN(s->height, 16);

    flags = bytestream2_get_byte(&gb);

    /* Interlace mode */
    switch (flags >> 2 & 3) {
        case 0:
            s->field_order       = AV_FIELD_PROGRESSIVE;
            s->picture_structure = AV_PICTURE_STRUCTURE_FRAME;
            break;
        case 1:
            s->field_order       = AV_FIELD_TT;
            s->picture_structure = AV_PICTURE_STRUCTURE_TOP_FIELD;
            break;
        case 2:
            s->field_order       = AV_FIELD_BB;
            s->picture_structure = AV_PICTURE_STRUCTURE_BOTTOM_FIELD;
            break;
        default:
            break;
    }

    bytestream2_skip(&gb, 4); /* Aspect ratio information, frame rate code, color primaries, transfer characteristic, matrix coefficients */

    /* Determine pixel format based on color depth, chroma format and alpha type */
    switch (avctx->codec_tag) {
        case MKTAG('a','p','c','o'):
        case MKTAG('a','p','c','s'):
        case MKTAG('a','p','c','n'):
        case MKTAG('a','p','c','h'):
            depth = 10;
            break;
        case MKTAG('a','p','4','h'):
        case MKTAG('a','p','4','x'):
            depth = 12;
            break;
        default:
            return buf_size;
    }

    chroma_format = flags >> 6 & 3;
    if (chroma_format < 2)
        return buf_size;

    alpha_channel_type = bytestream2_get_byte(&gb) & 0xf;

    switch (depth | (chroma_format << 4) | (alpha_channel_type << 8)) {
        case 10 | (2 << 4) | (0 << 8): s->format = AV_PIX_FMT_YUV422P10;  break;
        case 10 | (2 << 4) | (1 << 8):
        case 10 | (2 << 4) | (2 << 8): s->format = AV_PIX_FMT_YUVA422P10; break;
        case 10 | (3 << 4) | (0 << 8): s->format = AV_PIX_FMT_YUV444P10;  break;
        case 10 | (3 << 4) | (1 << 8):
        case 10 | (3 << 4) | (2 << 8): s->format = AV_PIX_FMT_YUVA444P10; break;
        case 12 | (2 << 4) | (0 << 8): s->format = AV_PIX_FMT_YUV422P12;  break;
        case 12 | (2 << 4) | (1 << 8):
        case 12 | (2 << 4) | (2 << 8): s->format = AV_PIX_FMT_YUVA422P12; break;
        case 12 | (3 << 4) | (0 << 8): s->format = AV_PIX_FMT_YUV444P12;  break;
        case 12 | (3 << 4) | (1 << 8):
        case 12 | (3 << 4) | (2 << 8): s->format = AV_PIX_FMT_YUVA444P12; break;
    }

    return buf_size;
}

const FFCodecParser ff_prores_parser = {
    PARSER_CODEC_LIST(AV_CODEC_ID_PRORES),
    .parse        = parse,
};