File: prores_parser.c

package info (click to toggle)
chromium 145.0.7632.159-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 5,976,224 kB
  • sloc: cpp: 36,198,469; ansic: 7,634,080; javascript: 3,564,060; python: 1,649,622; xml: 838,470; asm: 717,087; pascal: 185,708; sh: 88,786; perl: 88,718; objc: 79,984; sql: 59,811; cs: 42,452; fortran: 24,101; makefile: 21,144; tcl: 15,277; php: 14,022; yacc: 9,066; ruby: 7,553; awk: 3,720; lisp: 3,233; lex: 1,328; ada: 727; jsp: 228; sed: 36
file content (129 lines) | stat: -rw-r--r-- 4,349 bytes parent folder | download | duplicates (3)
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,
};