movenc: support cenc (common encryption)
[ffmpeg.git] / libavformat / movenc.h
1 /*
2  * MOV, 3GP, MP4 muxer
3  * Copyright (c) 2003 Thomas Raivio
4  * Copyright (c) 2004 Gildas Bazin <gbazin at videolan dot org>
5  * Copyright (c) 2009 Baptiste Coudurier <baptiste dot coudurier at gmail dot com>
6  *
7  * This file is part of FFmpeg.
8  *
9  * FFmpeg is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * FFmpeg is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with FFmpeg; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22  */
23
24 #ifndef AVFORMAT_MOVENC_H
25 #define AVFORMAT_MOVENC_H
26
27 #include "avformat.h"
28 #include "movenccenc.h"
29
30 #define MOV_FRAG_INFO_ALLOC_INCREMENT 64
31 #define MOV_INDEX_CLUSTER_SIZE 1024
32 #define MOV_TIMESCALE 1000
33
34 #define RTP_MAX_PACKET_SIZE 1450
35
36 #define MODE_MP4  0x01
37 #define MODE_MOV  0x02
38 #define MODE_3GP  0x04
39 #define MODE_PSP  0x08 // example working PSP command line:
40 // ffmpeg -i testinput.avi  -f psp -r 14.985 -s 320x240 -b 768 -ar 24000 -ab 32 M4V00001.MP4
41 #define MODE_3G2  0x10
42 #define MODE_IPOD 0x20
43 #define MODE_ISM  0x40
44 #define MODE_F4V  0x80
45
46 typedef struct MOVIentry {
47     uint64_t     pos;
48     int64_t      dts;
49     unsigned int size;
50     unsigned int samples_in_chunk;
51     unsigned int chunkNum;              ///< Chunk number if the current entry is a chunk start otherwise 0
52     unsigned int entries;
53     int          cts;
54 #define MOV_SYNC_SAMPLE         0x0001
55 #define MOV_PARTIAL_SYNC_SAMPLE 0x0002
56     uint32_t     flags;
57 } MOVIentry;
58
59 typedef struct HintSample {
60     uint8_t *data;
61     int size;
62     int sample_number;
63     int offset;
64     int own_data;
65 } HintSample;
66
67 typedef struct HintSampleQueue {
68     int size;
69     int len;
70     HintSample *samples;
71 } HintSampleQueue;
72
73 typedef struct MOVFragmentInfo {
74     int64_t offset;
75     int64_t time;
76     int64_t duration;
77     int64_t tfrf_offset;
78     int size;
79 } MOVFragmentInfo;
80
81 typedef struct MOVTrack {
82     int         mode;
83     int         entry;
84     unsigned    timescale;
85     uint64_t    time;
86     int64_t     track_duration;
87     int         last_sample_is_subtitle_end;
88     long        sample_count;
89     long        sample_size;
90     long        chunkCount;
91     int         has_keyframes;
92 #define MOV_TRACK_CTTS         0x0001
93 #define MOV_TRACK_STPS         0x0002
94 #define MOV_TRACK_ENABLED      0x0004
95     uint32_t    flags;
96 #define MOV_TIMECODE_FLAG_DROPFRAME     0x0001
97 #define MOV_TIMECODE_FLAG_24HOURSMAX    0x0002
98 #define MOV_TIMECODE_FLAG_ALLOWNEGATIVE 0x0004
99     uint32_t    timecode_flags;
100     int         language;
101     int         track_id;
102     int         tag; ///< stsd fourcc
103     AVStream        *st;
104     AVCodecContext *enc;
105     int multichannel_as_mono;
106
107     int         vos_len;
108     uint8_t     *vos_data;
109     MOVIentry   *cluster;
110     unsigned    cluster_capacity;
111     int         audio_vbr;
112     int         height; ///< active picture (w/o VBI) height for D-10/IMX
113     uint32_t    tref_tag;
114     int         tref_id; ///< trackID of the referenced track
115     int64_t     start_dts;
116     int64_t     start_cts;
117     int64_t     end_pts;
118
119     int         hint_track;   ///< the track that hints this track, -1 if no hint track is set
120     int         src_track;    ///< the track that this hint (or tmcd) track describes
121     AVFormatContext *rtp_ctx; ///< the format context for the hinting rtp muxer
122     uint32_t    prev_rtp_ts;
123     int64_t     cur_rtp_ts_unwrapped;
124     uint32_t    max_packet_size;
125
126     int64_t     default_duration;
127     uint32_t    default_sample_flags;
128     uint32_t    default_size;
129
130     HintSampleQueue sample_queue;
131
132     AVIOContext *mdat_buf;
133     int64_t     data_offset;
134     int64_t     frag_start;
135     int         frag_discont;
136     int         entries_flushed;
137
138     int         nb_frag_info;
139     MOVFragmentInfo *frag_info;
140     unsigned    frag_info_capacity;
141
142     struct {
143         int     first_packet_seq;
144         int     first_packet_entry;
145         int     first_packet_seen;
146         int     first_frag_written;
147         int     packet_seq;
148         int     packet_entry;
149         int     slices;
150     } vc1_info;
151
152     void       *eac3_priv;
153
154     MOVMuxCencContext cenc;
155 } MOVTrack;
156
157 typedef enum {
158     MOV_ENC_NONE = 0,
159     MOV_ENC_CENC_AES_CTR,
160 } MOVEncryptionScheme;
161
162 typedef struct MOVMuxContext {
163     const AVClass *av_class;
164     int     mode;
165     int64_t time;
166     int     nb_streams;
167     int     nb_meta_tmcd;  ///< number of new created tmcd track based on metadata (aka not data copy)
168     int     chapter_track; ///< qt chapter track number
169     int64_t mdat_pos;
170     uint64_t mdat_size;
171     MOVTrack *tracks;
172
173     int flags;
174     int rtp_flags;
175
176     int iods_skip;
177     int iods_video_profile;
178     int iods_audio_profile;
179
180     int moov_written;
181     int fragments;
182     int max_fragment_duration;
183     int min_fragment_duration;
184     int max_fragment_size;
185     int ism_lookahead;
186     AVIOContext *mdat_buf;
187     int first_trun;
188
189     int video_track_timescale;
190
191     int reserved_moov_size; ///< 0 for disabled, -1 for automatic, size otherwise
192     int64_t reserved_header_pos;
193
194     char *major_brand;
195
196     int per_stream_grouping;
197     AVFormatContext *fc;
198
199     int use_editlist;
200     float gamma;
201
202     int frag_interleave;
203     int missing_duration_warned;
204
205     char *encryption_scheme_str;
206     MOVEncryptionScheme encryption_scheme;
207     uint8_t *encryption_key;
208     int encryption_key_len;
209     uint8_t *encryption_kid;
210     int encryption_kid_len;
211
212 } MOVMuxContext;
213
214 #define FF_MOV_FLAG_RTP_HINT              (1 <<  0)
215 #define FF_MOV_FLAG_FRAGMENT              (1 <<  1)
216 #define FF_MOV_FLAG_EMPTY_MOOV            (1 <<  2)
217 #define FF_MOV_FLAG_FRAG_KEYFRAME         (1 <<  3)
218 #define FF_MOV_FLAG_SEPARATE_MOOF         (1 <<  4)
219 #define FF_MOV_FLAG_FRAG_CUSTOM           (1 <<  5)
220 #define FF_MOV_FLAG_ISML                  (1 <<  6)
221 #define FF_MOV_FLAG_FASTSTART             (1 <<  7)
222 #define FF_MOV_FLAG_OMIT_TFHD_OFFSET      (1 <<  8)
223 #define FF_MOV_FLAG_DISABLE_CHPL          (1 <<  9)
224 #define FF_MOV_FLAG_DEFAULT_BASE_MOOF     (1 << 10)
225 #define FF_MOV_FLAG_DASH                  (1 << 11)
226 #define FF_MOV_FLAG_FRAG_DISCONT          (1 << 12)
227 #define FF_MOV_FLAG_DELAY_MOOV            (1 << 13)
228 #define FF_MOV_FLAG_GLOBAL_SIDX           (1 << 14)
229 #define FF_MOV_FLAG_WRITE_COLR            (1 << 15)
230 #define FF_MOV_FLAG_WRITE_GAMA            (1 << 16)
231
232 int ff_mov_write_packet(AVFormatContext *s, AVPacket *pkt);
233
234 int ff_mov_init_hinting(AVFormatContext *s, int index, int src_index);
235 int ff_mov_add_hinted_packet(AVFormatContext *s, AVPacket *pkt,
236                              int track_index, int sample,
237                              uint8_t *sample_data, int sample_size);
238 void ff_mov_close_hinting(MOVTrack *track);
239
240 #endif /* AVFORMAT_MOVENC_H */