Statistics
| Branch: | Revision:

ffmpeg / libavcodec / libdiracenc.c @ 72415b2a

History | View | Annotate | Download (14.1 KB)

1
/*
2
 * Dirac encoding support via libdirac library
3
 * Copyright (c) 2005 BBC, Andrew Kennedy <dirac at rd dot bbc dot co dot uk>
4
 * Copyright (c) 2006-2008 BBC, Anuradha Suraparaju <asuraparaju at gmail dot com >
5
 *
6
 * This file is part of FFmpeg.
7
 *
8
 * FFmpeg is free software; you can redistribute it and/or
9
 * modify it under the terms of the GNU Lesser General Public
10
 * License as published by the Free Software Foundation; either
11
 * version 2.1 of the License, or (at your option) any later version.
12
 *
13
 * FFmpeg is distributed in the hope that it will be useful,
14
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16
 * Lesser General Public License for more details.
17
 *
18
 * You should have received a copy of the GNU Lesser General Public
19
 * License along with FFmpeg; if not, write to the Free Software
20
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21
 */
22

    
23
/**
24
* @file libavcodec/libdiracenc.c
25
* Dirac encoding support via libdirac library; more details about the
26
* Dirac project can be found at http://dirac.sourceforge.net/.
27
* The libdirac_encoder library implements Dirac specification version 2.2
28
* (http://dirac.sourceforge.net/specification.html).
29
*/
30

    
31
#include "libdirac_libschro.h"
32
#include "libdirac.h"
33

    
34
#undef NDEBUG
35
#include <assert.h>
36

    
37

    
38
#include <libdirac_encoder/dirac_encoder.h>
39

    
40
/** Dirac encoder private data */
41
typedef struct FfmpegDiracEncoderParams {
42
    /** Dirac encoder context */
43
    dirac_encoder_context_t enc_ctx;
44

    
45
    /** frame being encoded */
46
    AVFrame picture;
47

    
48
    /** frame size */
49
    int frame_size;
50

    
51
    /** Dirac encoder handle */
52
    dirac_encoder_t* p_encoder;
53

    
54
    /** input frame buffer */
55
    unsigned char *p_in_frame_buf;
56

    
57
    /** buffer to store encoder output before writing it to the frame queue */
58
    unsigned char *enc_buf;
59

    
60
    /** size of encoder buffer */
61
    int enc_buf_size;
62

    
63
    /** queue storing encoded frames */
64
    FfmpegDiracSchroQueue enc_frame_queue;
65

    
66
    /** end of sequence signalled by user, 0 - false, 1 - true */
67
    int eos_signalled;
68

    
69
    /** end of sequence returned by encoder, 0 - false, 1 - true */
70
    int eos_pulled;
71
} FfmpegDiracEncoderParams;
72

    
73
/**
74
* Works out Dirac-compatible chroma format.
75
*/
76
static dirac_chroma_t GetDiracChromaFormat(enum PixelFormat ff_pix_fmt)
77
{
78
    int num_formats = sizeof(ffmpeg_dirac_pixel_format_map) /
79
                      sizeof(ffmpeg_dirac_pixel_format_map[0]);
80
    int idx;
81

    
82
    for (idx = 0; idx < num_formats; ++idx)
83
        if (ffmpeg_dirac_pixel_format_map[idx].ff_pix_fmt == ff_pix_fmt)
84
            return ffmpeg_dirac_pixel_format_map[idx].dirac_pix_fmt;
85
    return formatNK;
86
}
87

    
88
/**
89
* Dirac video preset table. Ensure that this tables matches up correctly
90
* with the ff_dirac_schro_video_format_info table in libdirac_libschro.c.
91
*/
92
static const VideoFormat ff_dirac_video_formats[]={
93
    VIDEO_FORMAT_CUSTOM           ,
94
    VIDEO_FORMAT_QSIF525          ,
95
    VIDEO_FORMAT_QCIF             ,
96
    VIDEO_FORMAT_SIF525           ,
97
    VIDEO_FORMAT_CIF              ,
98
    VIDEO_FORMAT_4SIF525          ,
99
    VIDEO_FORMAT_4CIF             ,
100
    VIDEO_FORMAT_SD_480I60        ,
101
    VIDEO_FORMAT_SD_576I50        ,
102
    VIDEO_FORMAT_HD_720P60        ,
103
    VIDEO_FORMAT_HD_720P50        ,
104
    VIDEO_FORMAT_HD_1080I60       ,
105
    VIDEO_FORMAT_HD_1080I50       ,
106
    VIDEO_FORMAT_HD_1080P60       ,
107
    VIDEO_FORMAT_HD_1080P50       ,
108
    VIDEO_FORMAT_DIGI_CINEMA_2K24 ,
109
    VIDEO_FORMAT_DIGI_CINEMA_4K24 ,
110
};
111

    
112
/**
113
* Returns the video format preset matching the input video dimensions and
114
* time base.
115
*/
116
static VideoFormat GetDiracVideoFormatPreset(AVCodecContext *avccontext)
117
{
118
    unsigned int num_formats = sizeof(ff_dirac_video_formats) /
119
                               sizeof(ff_dirac_video_formats[0]);
120

    
121
    unsigned int idx = ff_dirac_schro_get_video_format_idx(avccontext);
122

    
123
    return (idx < num_formats) ?
124
                 ff_dirac_video_formats[idx] : VIDEO_FORMAT_CUSTOM;
125
}
126

    
127
static av_cold int libdirac_encode_init(AVCodecContext *avccontext)
128
{
129

    
130
    FfmpegDiracEncoderParams* p_dirac_params = avccontext->priv_data;
131
    int no_local = 1;
132
    int verbose  = avccontext->debug;
133
    VideoFormat preset;
134

    
135
    /* get Dirac preset */
136
    preset = GetDiracVideoFormatPreset(avccontext);
137

    
138
    /* initialize the encoder context */
139
    dirac_encoder_context_init(&(p_dirac_params->enc_ctx), preset);
140

    
141
    p_dirac_params->enc_ctx.src_params.chroma = GetDiracChromaFormat(avccontext->pix_fmt);
142

    
143
    if (p_dirac_params->enc_ctx.src_params.chroma == formatNK) {
144
        av_log(avccontext, AV_LOG_ERROR,
145
               "Unsupported pixel format %d. This codec supports only "
146
               "Planar YUV formats (yuv420p, yuv422p, yuv444p\n",
147
               avccontext->pix_fmt);
148
        return -1;
149
    }
150

    
151
    p_dirac_params->enc_ctx.src_params.frame_rate.numerator   = avccontext->time_base.den;
152
    p_dirac_params->enc_ctx.src_params.frame_rate.denominator = avccontext->time_base.num;
153

    
154
    p_dirac_params->enc_ctx.src_params.width  = avccontext->width;
155
    p_dirac_params->enc_ctx.src_params.height = avccontext->height;
156

    
157
    p_dirac_params->frame_size = avpicture_get_size(avccontext->pix_fmt,
158
                                                    avccontext->width,
159
                                                    avccontext->height);
160

    
161
    avccontext->coded_frame = &p_dirac_params->picture;
162

    
163
    if (no_local) {
164
        p_dirac_params->enc_ctx.decode_flag = 0;
165
        p_dirac_params->enc_ctx.instr_flag  = 0;
166
    } else {
167
        p_dirac_params->enc_ctx.decode_flag = 1;
168
        p_dirac_params->enc_ctx.instr_flag  = 1;
169
    }
170

    
171
    /* Intra-only sequence */
172
    if (!avccontext->gop_size) {
173
        p_dirac_params->enc_ctx.enc_params.num_L1 = 0;
174
        if (avccontext->coder_type == FF_CODER_TYPE_VLC)
175
            p_dirac_params->enc_ctx.enc_params.using_ac = 0;
176
    } else
177
        avccontext->has_b_frames = 1;
178

    
179
    if (avccontext->flags & CODEC_FLAG_QSCALE) {
180
        if (avccontext->global_quality) {
181
            p_dirac_params->enc_ctx.enc_params.qf = avccontext->global_quality
182
                                                    / (FF_QP2LAMBDA * 10.0);
183
            /* if it is not default bitrate then send target rate. */
184
            if (avccontext->bit_rate >= 1000 &&
185
                avccontext->bit_rate != 200000)
186
                p_dirac_params->enc_ctx.enc_params.trate = avccontext->bit_rate
187
                                                           / 1000;
188
        } else
189
            p_dirac_params->enc_ctx.enc_params.lossless = 1;
190
    } else if (avccontext->bit_rate >= 1000)
191
        p_dirac_params->enc_ctx.enc_params.trate = avccontext->bit_rate / 1000;
192

    
193
    if ((preset > VIDEO_FORMAT_QCIF || preset < VIDEO_FORMAT_QSIF525) &&
194
         avccontext->bit_rate == 200000)
195
        p_dirac_params->enc_ctx.enc_params.trate = 0;
196

    
197
    if (avccontext->flags & CODEC_FLAG_INTERLACED_ME)
198
        /* all material can be coded as interlaced or progressive
199
         * irrespective of the type of source material */
200
        p_dirac_params->enc_ctx.enc_params.picture_coding_mode = 1;
201

    
202
    p_dirac_params->p_encoder = dirac_encoder_init(&(p_dirac_params->enc_ctx),
203
                                                   verbose);
204

    
205
    if (!p_dirac_params->p_encoder) {
206
        av_log(avccontext, AV_LOG_ERROR,
207
               "Unrecoverable Error: dirac_encoder_init failed. ");
208
        return EXIT_FAILURE;
209
    }
210

    
211
    /* allocate enough memory for the incoming data */
212
    p_dirac_params->p_in_frame_buf = av_malloc(p_dirac_params->frame_size);
213

    
214
    /* initialize the encoded frame queue */
215
    ff_dirac_schro_queue_init(&p_dirac_params->enc_frame_queue);
216

    
217
    return 0;
218
}
219

    
220
static void DiracFreeFrame(void *data)
221
{
222
    FfmpegDiracSchroEncodedFrame *enc_frame = data;
223

    
224
    av_freep(&(enc_frame->p_encbuf));
225
    av_free(enc_frame);
226
}
227

    
228
static int libdirac_encode_frame(AVCodecContext *avccontext,
229
                                 unsigned char *frame,
230
                                 int buf_size, void *data)
231
{
232
    int enc_size = 0;
233
    dirac_encoder_state_t state;
234
    FfmpegDiracEncoderParams* p_dirac_params = avccontext->priv_data;
235
    FfmpegDiracSchroEncodedFrame* p_frame_output      = NULL;
236
    FfmpegDiracSchroEncodedFrame* p_next_output_frame = NULL;
237
    int go = 1;
238
    int last_frame_in_sequence = 0;
239

    
240
    if (!data) {
241
        /* push end of sequence if not already signalled */
242
        if (!p_dirac_params->eos_signalled) {
243
            dirac_encoder_end_sequence(p_dirac_params->p_encoder);
244
            p_dirac_params->eos_signalled = 1;
245
        }
246
    } else {
247

    
248
        /* Allocate frame data to Dirac input buffer.
249
         * Input line size may differ from what the codec supports,
250
         * especially when transcoding from one format to another.
251
         * So use avpicture_layout to copy the frame. */
252
        avpicture_layout((AVPicture *)data, avccontext->pix_fmt,
253
                         avccontext->width, avccontext->height,
254
                         p_dirac_params->p_in_frame_buf,
255
                         p_dirac_params->frame_size);
256

    
257
        /* load next frame */
258
        if (dirac_encoder_load(p_dirac_params->p_encoder,
259
                               p_dirac_params->p_in_frame_buf,
260
                               p_dirac_params->frame_size) < 0) {
261
            av_log(avccontext, AV_LOG_ERROR, "Unrecoverable Encoder Error."
262
                   " dirac_encoder_load failed...\n");
263
            return -1;
264
        }
265
    }
266

    
267
    if (p_dirac_params->eos_pulled)
268
        go = 0;
269

    
270
    while (go) {
271
        p_dirac_params->p_encoder->enc_buf.buffer = frame;
272
        p_dirac_params->p_encoder->enc_buf.size   = buf_size;
273
        /* process frame */
274
        state = dirac_encoder_output(p_dirac_params->p_encoder);
275

    
276
        switch (state) {
277
        case ENC_STATE_AVAIL:
278
        case ENC_STATE_EOS:
279
            assert(p_dirac_params->p_encoder->enc_buf.size > 0);
280

    
281
            /* All non-frame data is prepended to actual frame data to
282
             * be able to set the pts correctly. So we don't write data
283
             * to the frame output queue until we actually have a frame
284
             */
285

    
286
            p_dirac_params->enc_buf = av_realloc(p_dirac_params->enc_buf,
287
                                                 p_dirac_params->enc_buf_size +
288
                                                 p_dirac_params->p_encoder->enc_buf.size);
289
            memcpy(p_dirac_params->enc_buf + p_dirac_params->enc_buf_size,
290
                   p_dirac_params->p_encoder->enc_buf.buffer,
291
                   p_dirac_params->p_encoder->enc_buf.size);
292

    
293
            p_dirac_params->enc_buf_size += p_dirac_params->p_encoder->enc_buf.size;
294

    
295
            if (state == ENC_STATE_EOS) {
296
                p_dirac_params->eos_pulled = 1;
297
                go = 0;
298
            }
299

    
300
            /* If non-frame data, don't output it until it we get an
301
             * encoded frame back from the encoder. */
302
            if (p_dirac_params->p_encoder->enc_pparams.pnum == -1)
303
                break;
304

    
305
            /* create output frame */
306
            p_frame_output = av_mallocz(sizeof(FfmpegDiracSchroEncodedFrame));
307
            /* set output data */
308
            p_frame_output->size      = p_dirac_params->enc_buf_size;
309
            p_frame_output->p_encbuf  = p_dirac_params->enc_buf;
310
            p_frame_output->frame_num = p_dirac_params->p_encoder->enc_pparams.pnum;
311

    
312
            if (p_dirac_params->p_encoder->enc_pparams.ptype == INTRA_PICTURE &&
313
                p_dirac_params->p_encoder->enc_pparams.rtype == REFERENCE_PICTURE)
314
                p_frame_output->key_frame = 1;
315

    
316
            ff_dirac_schro_queue_push_back(&p_dirac_params->enc_frame_queue,
317
                                           p_frame_output);
318

    
319
            p_dirac_params->enc_buf_size = 0;
320
            p_dirac_params->enc_buf      = NULL;
321
            break;
322

    
323
        case ENC_STATE_BUFFER:
324
            go = 0;
325
            break;
326

    
327
        case ENC_STATE_INVALID:
328
            av_log(avccontext, AV_LOG_ERROR,
329
                   "Unrecoverable Dirac Encoder Error. Quitting...\n");
330
            return -1;
331

    
332
        default:
333
            av_log(avccontext, AV_LOG_ERROR, "Unknown Dirac Encoder state\n");
334
            return -1;
335
        }
336
    }
337

    
338
    /* copy 'next' frame in queue */
339

    
340
    if (p_dirac_params->enc_frame_queue.size == 1 && p_dirac_params->eos_pulled)
341
        last_frame_in_sequence = 1;
342

    
343
    p_next_output_frame = ff_dirac_schro_queue_pop(&p_dirac_params->enc_frame_queue);
344

    
345
    if (!p_next_output_frame)
346
        return 0;
347

    
348
    memcpy(frame, p_next_output_frame->p_encbuf, p_next_output_frame->size);
349
    avccontext->coded_frame->key_frame = p_next_output_frame->key_frame;
350
    /* Use the frame number of the encoded frame as the pts. It is OK to do
351
     * so since Dirac is a constant framerate codec. It expects input to be
352
     * of constant framerate. */
353
    avccontext->coded_frame->pts = p_next_output_frame->frame_num;
354
    enc_size = p_next_output_frame->size;
355

    
356
    /* Append the end of sequence information to the last frame in the
357
     * sequence. */
358
    if (last_frame_in_sequence && p_dirac_params->enc_buf_size > 0) {
359
        memcpy(frame + enc_size, p_dirac_params->enc_buf,
360
               p_dirac_params->enc_buf_size);
361
        enc_size += p_dirac_params->enc_buf_size;
362
        av_freep(&p_dirac_params->enc_buf);
363
        p_dirac_params->enc_buf_size = 0;
364
    }
365

    
366
    /* free frame */
367
    DiracFreeFrame(p_next_output_frame);
368

    
369
    return enc_size;
370
}
371

    
372
static av_cold int libdirac_encode_close(AVCodecContext *avccontext)
373
{
374
    FfmpegDiracEncoderParams* p_dirac_params  = avccontext->priv_data;
375

    
376
    /* close the encoder */
377
    dirac_encoder_close(p_dirac_params->p_encoder);
378

    
379
    /* free data in the output frame queue */
380
    ff_dirac_schro_queue_free(&p_dirac_params->enc_frame_queue,
381
                              DiracFreeFrame);
382

    
383
    /* free the encoder buffer */
384
    if (p_dirac_params->enc_buf_size)
385
        av_freep(&p_dirac_params->enc_buf);
386

    
387
    /* free the input frame buffer */
388
    av_freep(&p_dirac_params->p_in_frame_buf);
389

    
390
    return 0;
391
}
392

    
393

    
394
AVCodec libdirac_encoder = {
395
    "libdirac",
396
    AVMEDIA_TYPE_VIDEO,
397
    CODEC_ID_DIRAC,
398
    sizeof(FfmpegDiracEncoderParams),
399
    libdirac_encode_init,
400
    libdirac_encode_frame,
401
    libdirac_encode_close,
402
   .capabilities = CODEC_CAP_DELAY,
403
   .pix_fmts = (const enum PixelFormat[]){PIX_FMT_YUV420P, PIX_FMT_YUV422P, PIX_FMT_YUV444P, PIX_FMT_NONE},
404
   .long_name = NULL_IF_CONFIG_SMALL("libdirac Dirac 2.2"),
405
};