/src/ffmpeg/libavcodec/dvd_nav_parser.c
Line | Count | Source |
1 | | /* |
2 | | * DVD navigation block parser for FFmpeg |
3 | | * Copyright (c) 2013 The FFmpeg Project |
4 | | * |
5 | | * This file is part of FFmpeg. |
6 | | * |
7 | | * FFmpeg is free software; you can redistribute it and/or |
8 | | * modify it under the terms of the GNU Lesser General Public |
9 | | * License as published by the Free Software Foundation; either |
10 | | * version 2.1 of the License, or (at your option) any later version. |
11 | | * |
12 | | * FFmpeg is distributed in the hope that it will be useful, |
13 | | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 | | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
15 | | * Lesser General Public License for more details. |
16 | | * |
17 | | * You should have received a copy of the GNU Lesser General Public |
18 | | * License along with FFmpeg; if not, write to the Free Software |
19 | | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
20 | | */ |
21 | | |
22 | | #include "avcodec.h" |
23 | | #include "parser_internal.h" |
24 | | #include "libavutil/intreadwrite.h" |
25 | | |
26 | 117 | #define PCI_SIZE 980 |
27 | 13 | #define DSI_SIZE 1018 |
28 | | |
29 | | /* parser definition */ |
30 | | typedef struct DVDNavParseContext { |
31 | | uint32_t lba; |
32 | | uint8_t buffer[PCI_SIZE+DSI_SIZE]; |
33 | | int copied; |
34 | | } DVDNavParseContext; |
35 | | |
36 | | static av_cold int dvd_nav_parse_init(AVCodecParserContext *s) |
37 | 33 | { |
38 | 33 | DVDNavParseContext *pc = s->priv_data; |
39 | | |
40 | 33 | pc->lba = 0xFFFFFFFF; |
41 | 33 | pc->copied = 0; |
42 | 33 | return 0; |
43 | 33 | } |
44 | | |
45 | | static int dvd_nav_parse(AVCodecParserContext *s, |
46 | | AVCodecContext *avctx, |
47 | | const uint8_t **poutbuf, int *poutbuf_size, |
48 | | const uint8_t *buf, int buf_size) |
49 | 134 | { |
50 | 134 | DVDNavParseContext *pc1 = s->priv_data; |
51 | 134 | int lastPacket = 0; |
52 | 134 | int valid = 0; |
53 | | |
54 | 134 | s->pict_type = AV_PICTURE_TYPE_NONE; |
55 | | |
56 | 134 | avctx->time_base.num = 1; |
57 | 134 | avctx->time_base.den = 90000; |
58 | | |
59 | 134 | if (buf && buf_size) { |
60 | 101 | switch(buf[0]) { |
61 | 86 | case 0x00: |
62 | 86 | if (buf_size == PCI_SIZE) { |
63 | | /* PCI */ |
64 | 60 | uint32_t lba = AV_RB32(&buf[0x01]); |
65 | 60 | uint32_t startpts = AV_RB32(&buf[0x0D]); |
66 | 60 | uint32_t endpts = AV_RB32(&buf[0x11]); |
67 | | |
68 | 60 | if (endpts > startpts) { |
69 | 12 | pc1->lba = lba; |
70 | 12 | s->pts = (int64_t)startpts; |
71 | 12 | s->duration = endpts - startpts; |
72 | | |
73 | 12 | memcpy(pc1->buffer, buf, PCI_SIZE); |
74 | 12 | pc1->copied = PCI_SIZE; |
75 | 12 | valid = 1; |
76 | 12 | } |
77 | 60 | } |
78 | 86 | break; |
79 | | |
80 | 7 | case 0x01: |
81 | 7 | if ((buf_size == DSI_SIZE) && (pc1->copied == PCI_SIZE)) { |
82 | | /* DSI */ |
83 | 6 | uint32_t lba = AV_RB32(&buf[0x05]); |
84 | | |
85 | 6 | if (lba == pc1->lba) { |
86 | 6 | memcpy(pc1->buffer + pc1->copied, buf, DSI_SIZE); |
87 | 6 | lastPacket = 1; |
88 | 6 | valid = 1; |
89 | 6 | } |
90 | 6 | } |
91 | 7 | break; |
92 | 101 | } |
93 | 101 | } |
94 | | |
95 | 134 | if (!valid || lastPacket) { |
96 | 122 | pc1->copied = 0; |
97 | 122 | pc1->lba = 0xFFFFFFFF; |
98 | 122 | } |
99 | | |
100 | 134 | if (lastPacket) { |
101 | 6 | *poutbuf = pc1->buffer; |
102 | 6 | *poutbuf_size = sizeof(pc1->buffer); |
103 | 128 | } else { |
104 | 128 | *poutbuf = NULL; |
105 | 128 | *poutbuf_size = 0; |
106 | 128 | } |
107 | | |
108 | 134 | return buf_size; |
109 | 134 | } |
110 | | |
111 | | const FFCodecParser ff_dvd_nav_parser = { |
112 | | PARSER_CODEC_LIST(AV_CODEC_ID_DVD_NAV), |
113 | | .priv_data_size = sizeof(DVDNavParseContext), |
114 | | .init = dvd_nav_parse_init, |
115 | | .parse = dvd_nav_parse, |
116 | | }; |