1/*
2 * nut
3 * Copyright (c) 2004-2007 Michael Niedermayer
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 "libavutil/mathematics.h"
23#include "libavutil/tree.h"
24#include "nut.h"
25#include "riff.h"
26#include "internal.h"
27
28const AVCodecTag ff_nut_subtitle_tags[] = {
29    { AV_CODEC_ID_TEXT,             MKTAG('U', 'T', 'F', '8') },
30    { AV_CODEC_ID_SSA,              MKTAG('S', 'S', 'A',  0 ) },
31    { AV_CODEC_ID_DVD_SUBTITLE,     MKTAG('D', 'V', 'D', 'S') },
32    { AV_CODEC_ID_DVB_SUBTITLE,     MKTAG('D', 'V', 'B', 'S') },
33    { AV_CODEC_ID_DVB_TELETEXT,     MKTAG('D', 'V', 'B', 'T') },
34    { AV_CODEC_ID_NONE,             0                         }
35};
36
37const AVCodecTag ff_nut_data_tags[] = {
38    { AV_CODEC_ID_TEXT,             MKTAG('U', 'T', 'F', '8') },
39    { AV_CODEC_ID_NONE,             0 }
40};
41
42const AVCodecTag ff_nut_video_tags[] = {
43    { AV_CODEC_ID_VP9,              MKTAG('V', 'P', '9', '0') },
44    { AV_CODEC_ID_RAWVIDEO,         MKTAG('R', 'G', 'B', 15 ) },
45    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', 'G', 'R', 15 ) },
46    { AV_CODEC_ID_RAWVIDEO,         MKTAG('R', 'G', 'B', 16 ) },
47    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', 'G', 'R', 16 ) },
48    { AV_CODEC_ID_RAWVIDEO,         MKTAG(15 , 'B', 'G', 'R') },
49    { AV_CODEC_ID_RAWVIDEO,         MKTAG(15 , 'R', 'G', 'B') },
50    { AV_CODEC_ID_RAWVIDEO,         MKTAG(16 , 'B', 'G', 'R') },
51    { AV_CODEC_ID_RAWVIDEO,         MKTAG(16 , 'R', 'G', 'B') },
52    { AV_CODEC_ID_RAWVIDEO,         MKTAG('R', 'G', 'B', 12 ) },
53    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', 'G', 'R', 12 ) },
54    { AV_CODEC_ID_RAWVIDEO,         MKTAG(12 , 'B', 'G', 'R') },
55    { AV_CODEC_ID_RAWVIDEO,         MKTAG(12 , 'R', 'G', 'B') },
56    { AV_CODEC_ID_RAWVIDEO,         MKTAG('R', 'G', 'B', 'A') },
57    { AV_CODEC_ID_RAWVIDEO,         MKTAG('R', 'G', 'B',  0 ) },
58    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', 'G', 'R', 'A') },
59    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', 'G', 'R',  0 ) },
60    { AV_CODEC_ID_RAWVIDEO,         MKTAG('A', 'B', 'G', 'R') },
61    { AV_CODEC_ID_RAWVIDEO,         MKTAG( 0 , 'B', 'G', 'R') },
62    { AV_CODEC_ID_RAWVIDEO,         MKTAG('A', 'R', 'G', 'B') },
63    { AV_CODEC_ID_RAWVIDEO,         MKTAG( 0 , 'R', 'G', 'B') },
64    { AV_CODEC_ID_RAWVIDEO,         MKTAG('R', 'G', 'B', 24 ) },
65    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', 'G', 'R', 24 ) },
66    { AV_CODEC_ID_RAWVIDEO,         MKTAG('4', '1', '1', 'P') },
67    { AV_CODEC_ID_RAWVIDEO,         MKTAG('4', '2', '2', 'P') },
68    { AV_CODEC_ID_RAWVIDEO,         MKTAG('4', '2', '2', 'P') },
69    { AV_CODEC_ID_RAWVIDEO,         MKTAG('4', '4', '0', 'P') },
70    { AV_CODEC_ID_RAWVIDEO,         MKTAG('4', '4', '0', 'P') },
71    { AV_CODEC_ID_RAWVIDEO,         MKTAG('4', '4', '4', 'P') },
72    { AV_CODEC_ID_RAWVIDEO,         MKTAG('4', '4', '4', 'P') },
73    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', '1', 'W', '0') },
74    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', '0', 'W', '1') },
75    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', 'G', 'R',  8 ) },
76    { AV_CODEC_ID_RAWVIDEO,         MKTAG('R', 'G', 'B',  8 ) },
77    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', 'G', 'R',  4 ) },
78    { AV_CODEC_ID_RAWVIDEO,         MKTAG('R', 'G', 'B',  4 ) },
79    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', '4', 'B', 'Y') },
80    { AV_CODEC_ID_RAWVIDEO,         MKTAG('R', '4', 'B', 'Y') },
81    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', 'G', 'R', 48 ) },
82    { AV_CODEC_ID_RAWVIDEO,         MKTAG('R', 'G', 'B', 48 ) },
83    { AV_CODEC_ID_RAWVIDEO,         MKTAG(48 , 'B', 'G', 'R') },
84    { AV_CODEC_ID_RAWVIDEO,         MKTAG(48 , 'R', 'G', 'B') },
85    { AV_CODEC_ID_RAWVIDEO,         MKTAG('R', 'B', 'A', 64 ) },
86    { AV_CODEC_ID_RAWVIDEO,         MKTAG('B', 'R', 'A', 64 ) },
87    { AV_CODEC_ID_RAWVIDEO,         MKTAG(64 , 'R', 'B', 'A') },
88    { AV_CODEC_ID_RAWVIDEO,         MKTAG(64 , 'B', 'R', 'A') },
89    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '3', 11 , 10 ) },
90    { AV_CODEC_ID_RAWVIDEO,         MKTAG(10 , 11 , '3', 'Y') },
91    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '3', 10 , 10 ) },
92    { AV_CODEC_ID_RAWVIDEO,         MKTAG(10 , 10 , '3', 'Y') },
93    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '3',  0 , 10 ) },
94    { AV_CODEC_ID_RAWVIDEO,         MKTAG(10 ,  0 , '3', 'Y') },
95    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '3', 11 , 12 ) },
96    { AV_CODEC_ID_RAWVIDEO,         MKTAG(12 , 11 , '3', 'Y') },
97    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '3', 10 , 12 ) },
98    { AV_CODEC_ID_RAWVIDEO,         MKTAG(12 , 10 , '3', 'Y') },
99    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '3',  0 , 12 ) },
100    { AV_CODEC_ID_RAWVIDEO,         MKTAG(12 ,  0 , '3', 'Y') },
101    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '3', 11 , 14 ) },
102    { AV_CODEC_ID_RAWVIDEO,         MKTAG(14 , 11 , '3', 'Y') },
103    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '3', 10 , 14 ) },
104    { AV_CODEC_ID_RAWVIDEO,         MKTAG(14 , 10 , '3', 'Y') },
105    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '3',  0 , 14 ) },
106    { AV_CODEC_ID_RAWVIDEO,         MKTAG(14 ,  0 , '3', 'Y') },
107    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '1',  0 , 16 ) },
108    { AV_CODEC_ID_RAWVIDEO,         MKTAG(16 ,  0 , '1', 'Y') },
109    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '3', 11 , 16 ) },
110    { AV_CODEC_ID_RAWVIDEO,         MKTAG(16 , 11 , '3', 'Y') },
111    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '3', 10 , 16 ) },
112    { AV_CODEC_ID_RAWVIDEO,         MKTAG(16 , 10 , '3', 'Y') },
113    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '3',  0 , 16 ) },
114    { AV_CODEC_ID_RAWVIDEO,         MKTAG(16 ,  0 , '3', 'Y') },
115    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '4', 11 ,  8 ) },
116    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '4', 10 ,  8 ) },
117    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '4',  0 ,  8 ) },
118    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '2',  0 ,  8 ) },
119
120    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '1',   0,   9) },
121    { AV_CODEC_ID_RAWVIDEO,         MKTAG(9,     0, '1', 'Y') },
122    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '4',  11,   9) },
123    { AV_CODEC_ID_RAWVIDEO,         MKTAG(9,    11, '4', 'Y') },
124    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '4',  10,   9) },
125    { AV_CODEC_ID_RAWVIDEO,         MKTAG(9,    10, '4', 'Y') },
126    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '4',   0,   9) },
127    { AV_CODEC_ID_RAWVIDEO,         MKTAG(9,     0, '4', 'Y') },
128
129    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '1',   0,  10) },
130    { AV_CODEC_ID_RAWVIDEO,         MKTAG(10,    0, '1', 'Y') },
131    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '4',  11,  10) },
132    { AV_CODEC_ID_RAWVIDEO,         MKTAG(10,   11, '4', 'Y') },
133    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '4',  10,  10) },
134    { AV_CODEC_ID_RAWVIDEO,         MKTAG(10,   10, '4', 'Y') },
135    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '4',   0,  10) },
136    { AV_CODEC_ID_RAWVIDEO,         MKTAG(10,    0, '4', 'Y') },
137
138    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '1',   0,  16) },
139    { AV_CODEC_ID_RAWVIDEO,         MKTAG(16,    0, '1', 'Y') },
140    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '4',  11,  16) },
141    { AV_CODEC_ID_RAWVIDEO,         MKTAG(16,   11, '4', 'Y') },
142    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '4',  10,  16) },
143    { AV_CODEC_ID_RAWVIDEO,         MKTAG(16,   10, '4', 'Y') },
144    { AV_CODEC_ID_RAWVIDEO,         MKTAG('Y', '4',   0,  16) },
145    { AV_CODEC_ID_RAWVIDEO,         MKTAG(16,    0, '4', 'Y') },
146
147    { AV_CODEC_ID_RAWVIDEO,         MKTAG('G', '3',   0,   8) },
148
149    { AV_CODEC_ID_RAWVIDEO,         MKTAG('G', '3',   0,   9) },
150    { AV_CODEC_ID_RAWVIDEO,         MKTAG( 9,    0, '3', 'G') },
151    { AV_CODEC_ID_RAWVIDEO,         MKTAG('G', '3',   0,  10) },
152    { AV_CODEC_ID_RAWVIDEO,         MKTAG(10,    0, '3', 'G') },
153    { AV_CODEC_ID_RAWVIDEO,         MKTAG('G', '3',   0,  12) },
154    { AV_CODEC_ID_RAWVIDEO,         MKTAG(12,    0, '3', 'G') },
155    { AV_CODEC_ID_RAWVIDEO,         MKTAG('G', '3',   0,  14) },
156    { AV_CODEC_ID_RAWVIDEO,         MKTAG(14,    0, '3', 'G') },
157    { AV_CODEC_ID_RAWVIDEO,         MKTAG('G', '3',   0,  16) },
158    { AV_CODEC_ID_RAWVIDEO,         MKTAG(16,    0, '3', 'G') },
159
160    { AV_CODEC_ID_RAWVIDEO,         MKTAG('X', 'Y', 'Z' , 36 ) },
161    { AV_CODEC_ID_RAWVIDEO,         MKTAG(36 , 'Z' , 'Y', 'X') },
162
163    { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'B', 'G', 8   ) },
164    { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'B', 'G', 16  ) },
165    { AV_CODEC_ID_RAWVIDEO, MKTAG(16  , 'G', 'B', 0xBA) },
166    { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'R', 'G', 8   ) },
167    { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'R', 'G', 16  ) },
168    { AV_CODEC_ID_RAWVIDEO, MKTAG(16  , 'G', 'R', 0xBA) },
169    { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'G', 'B', 8   ) },
170    { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'G', 'B', 16  ) },
171    { AV_CODEC_ID_RAWVIDEO, MKTAG(16,   'B', 'G', 0xBA) },
172    { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'G', 'R', 8   ) },
173    { AV_CODEC_ID_RAWVIDEO, MKTAG(0xBA, 'G', 'R', 16  ) },
174    { AV_CODEC_ID_RAWVIDEO, MKTAG(16,   'R', 'G', 0xBA) },
175
176    { AV_CODEC_ID_NONE,             0 }
177};
178
179const AVCodecTag ff_nut_audio_extra_tags[] = {
180    { AV_CODEC_ID_COMFORT_NOISE,    MKTAG('3', '3', '8', '9') },
181    { AV_CODEC_ID_PCM_ALAW,         MKTAG('A', 'L', 'A', 'W') },
182    { AV_CODEC_ID_PCM_MULAW,        MKTAG('U', 'L', 'A', 'W') },
183    { AV_CODEC_ID_MP3,              MKTAG('M', 'P', '3', ' ') },
184    { AV_CODEC_ID_NONE,             0                         }
185};
186
187const AVCodecTag ff_nut_audio_tags[] = {
188    { AV_CODEC_ID_PCM_F32BE,        MKTAG(32 , 'D', 'F', 'P') },
189    { AV_CODEC_ID_PCM_F32LE,        MKTAG('P', 'F', 'D', 32 ) },
190    { AV_CODEC_ID_PCM_F64BE,        MKTAG(64 , 'D', 'F', 'P') },
191    { AV_CODEC_ID_PCM_F64LE,        MKTAG('P', 'F', 'D', 64 ) },
192    { AV_CODEC_ID_PCM_S16BE,        MKTAG(16 , 'D', 'S', 'P') },
193    { AV_CODEC_ID_PCM_S16LE,        MKTAG('P', 'S', 'D', 16 ) },
194    { AV_CODEC_ID_PCM_S24BE,        MKTAG(24 , 'D', 'S', 'P') },
195    { AV_CODEC_ID_PCM_S24LE,        MKTAG('P', 'S', 'D', 24 ) },
196    { AV_CODEC_ID_PCM_S32BE,        MKTAG(32 , 'D', 'S', 'P') },
197    { AV_CODEC_ID_PCM_S32LE,        MKTAG('P', 'S', 'D', 32 ) },
198    { AV_CODEC_ID_PCM_S8,           MKTAG('P', 'S', 'D',  8 ) },
199    { AV_CODEC_ID_PCM_U16BE,        MKTAG(16 , 'D', 'U', 'P') },
200    { AV_CODEC_ID_PCM_U16LE,        MKTAG('P', 'U', 'D', 16 ) },
201    { AV_CODEC_ID_PCM_U24BE,        MKTAG(24 , 'D', 'U', 'P') },
202    { AV_CODEC_ID_PCM_U24LE,        MKTAG('P', 'U', 'D', 24 ) },
203    { AV_CODEC_ID_PCM_U32BE,        MKTAG(32 , 'D', 'U', 'P') },
204    { AV_CODEC_ID_PCM_U32LE,        MKTAG('P', 'U', 'D', 32 ) },
205    { AV_CODEC_ID_PCM_U8,           MKTAG('P', 'U', 'D',  8 ) },
206    { AV_CODEC_ID_PCM_S8_PLANAR,    MKTAG('P', 'S', 'P',  8 ) },
207    { AV_CODEC_ID_PCM_S16BE_PLANAR, MKTAG(16 , 'P', 'S', 'P') },
208    { AV_CODEC_ID_PCM_S16LE_PLANAR, MKTAG('P', 'S', 'P', 16 ) },
209    { AV_CODEC_ID_PCM_S24LE_PLANAR, MKTAG('P', 'S', 'P', 24 ) },
210    { AV_CODEC_ID_PCM_S32LE_PLANAR, MKTAG('P', 'S', 'P', 32 ) },
211    { AV_CODEC_ID_NONE,             0                         }
212};
213
214const AVCodecTag * const ff_nut_codec_tags[] = {
215    ff_nut_video_tags, ff_nut_audio_tags, ff_nut_subtitle_tags,
216    ff_codec_bmp_tags, ff_codec_wav_tags, ff_nut_audio_extra_tags, ff_nut_data_tags, 0
217};
218
219void ff_nut_reset_ts(NUTContext *nut, AVRational time_base, int64_t val)
220{
221    int i;
222    for (i = 0; i < nut->avf->nb_streams; i++)
223        nut->stream[i].last_pts =
224            av_rescale_rnd(val,
225                           time_base.num * (int64_t)nut->stream[i].time_base->den,
226                           time_base.den * (int64_t)nut->stream[i].time_base->num,
227                           AV_ROUND_DOWN);
228}
229
230int64_t ff_lsb2full(StreamContext *stream, int64_t lsb)
231{
232    int64_t mask  = (1ULL << stream->msb_pts_shift) - 1;
233    int64_t delta = stream->last_pts - mask / 2;
234    return ((lsb - delta) & mask) + delta;
235}
236
237int ff_nut_sp_pos_cmp(const Syncpoint *a, const Syncpoint *b)
238{
239    return ((a->pos - b->pos) >> 32) - ((b->pos - a->pos) >> 32);
240}
241
242int ff_nut_sp_pts_cmp(const Syncpoint *a, const Syncpoint *b)
243{
244    return ((a->ts - b->ts) >> 32) - ((b->ts - a->ts) >> 32);
245}
246
247int ff_nut_add_sp(NUTContext *nut, int64_t pos, int64_t back_ptr, int64_t ts)
248{
249    Syncpoint *sp           = av_mallocz(sizeof(Syncpoint));
250    struct AVTreeNode *node = av_tree_node_alloc();
251
252    if (!sp || !node) {
253        av_freep(&sp);
254        av_freep(&node);
255        return AVERROR(ENOMEM);
256    }
257
258    nut->sp_count++;
259
260    sp->pos      = pos;
261    sp->back_ptr = back_ptr;
262    sp->ts       = ts;
263    av_tree_insert(&nut->syncpoints, sp, (void *) ff_nut_sp_pos_cmp, &node);
264    if (node) {
265        av_free(sp);
266        av_free(node);
267    }
268
269    return 0;
270}
271
272static int enu_free(void *opaque, void *elem)
273{
274    av_free(elem);
275    return 0;
276}
277
278void ff_nut_free_sp(NUTContext *nut)
279{
280    av_tree_enumerate(nut->syncpoints, NULL, NULL, enu_free);
281    av_tree_destroy(nut->syncpoints);
282}
283
284const Dispositions ff_nut_dispositions[] = {
285    { "default",        AV_DISPOSITION_DEFAULT  },
286    { "dub",            AV_DISPOSITION_DUB      },
287    { "original",       AV_DISPOSITION_ORIGINAL },
288    { "comment",        AV_DISPOSITION_COMMENT  },
289    { "lyrics",         AV_DISPOSITION_LYRICS   },
290    { "karaoke",        AV_DISPOSITION_KARAOKE  },
291    { "",               0                       }
292};
293
294const AVMetadataConv ff_nut_metadata_conv[] = {
295    { "Author",         "artist"      },
296    { "X-CreationTime", "date"        },
297    { "CreationTime",   "date"        },
298    { "SourceFilename", "filename"    },
299    { "X-Language",     "language"    },
300    { "X-Disposition",  "disposition" },
301    { "X-Replaces",     "replaces"    },
302    { "X-Depends",      "depends"     },
303    { "X-Uses",         "uses"        },
304    { "X-UsesFont",     "usesfont"    },
305    { 0 },
306};
307