[libav.git] / libavcodec / libdiracenc.c

/*
 * Dirac encoding support via libdirac library
 * Copyright (c) 2005 BBC, Andrew Kennedy <dirac at rd dot bbc dot co dot uk>
 * Copyright (c) 2006-2008 BBC, Anuradha Suraparaju <asuraparaju at gmail dot com >
 *
 * 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
 */

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

#include "libdirac_libschro.h"
#include "libdirac.h"

#undef NDEBUG
#include <assert.h>


#include <libdirac_encoder/dirac_encoder.h>

/** Dirac encoder private data */
typedef struct FfmpegDiracEncoderParams
{
    /** Dirac encoder context */
    dirac_encoder_context_t enc_ctx;

    /** frame being encoded */
    AVFrame picture;

    /** frame size */
    int frame_size;

    /** Dirac encoder handle */
    dirac_encoder_t* p_encoder;

    /** input frame buffer */
    unsigned char *p_in_frame_buf;

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

    /** size of encoder buffer */
    int enc_buf_size;

    /** queue storing encoded frames */
    FfmpegDiracSchroQueue enc_frame_queue;

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

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

/**
* Works out Dirac-compatible chroma format.
*/
static dirac_chroma_t GetDiracChromaFormat(enum PixelFormat ff_pix_fmt)
{
    int num_formats = sizeof(ffmpeg_dirac_pixel_format_map) /
                      sizeof(ffmpeg_dirac_pixel_format_map[0]);
    int idx;

    for (idx = 0; idx < num_formats; ++idx) {
        if (ffmpeg_dirac_pixel_format_map[idx].ff_pix_fmt == ff_pix_fmt) {
            return ffmpeg_dirac_pixel_format_map[idx].dirac_pix_fmt;
        }
    }
    return formatNK;
}

/**
* Dirac video preset table. Ensure that this tables matches up correctly
* with the ff_dirac_schro_video_format_info table in libdirac_libschro.c.
*/
static const VideoFormat ff_dirac_video_formats[]={
    VIDEO_FORMAT_CUSTOM           ,
    VIDEO_FORMAT_QSIF525          ,
    VIDEO_FORMAT_QCIF             ,
    VIDEO_FORMAT_SIF525           ,
    VIDEO_FORMAT_CIF              ,
    VIDEO_FORMAT_4SIF525          ,
    VIDEO_FORMAT_4CIF             ,
    VIDEO_FORMAT_SD_480I60        ,
    VIDEO_FORMAT_SD_576I50        ,
    VIDEO_FORMAT_HD_720P60        ,
    VIDEO_FORMAT_HD_720P50        ,
    VIDEO_FORMAT_HD_1080I60       ,
    VIDEO_FORMAT_HD_1080I50       ,
    VIDEO_FORMAT_HD_1080P60       ,
    VIDEO_FORMAT_HD_1080P50       ,
    VIDEO_FORMAT_DIGI_CINEMA_2K24 ,
    VIDEO_FORMAT_DIGI_CINEMA_4K24 ,
};

/**
* Returns the video format preset matching the input video dimensions and
* time base.
*/
static VideoFormat GetDiracVideoFormatPreset (AVCodecContext *avccontext)
{
    unsigned int num_formats = sizeof(ff_dirac_video_formats) /
                               sizeof(ff_dirac_video_formats[0]);

    unsigned int idx = ff_dirac_schro_get_video_format_idx (avccontext);

    return (idx < num_formats) ?
                 ff_dirac_video_formats[idx] : VIDEO_FORMAT_CUSTOM;
}

static av_cold int libdirac_encode_init(AVCodecContext *avccontext)
{

    FfmpegDiracEncoderParams* p_dirac_params = avccontext->priv_data;
    int no_local = 1;
    int verbose = avccontext->debug;
    VideoFormat preset;

    /* get Dirac preset */
    preset = GetDiracVideoFormatPreset(avccontext);

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

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

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

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

    p_dirac_params->enc_ctx.src_params.width  = avccontext->width;
    p_dirac_params->enc_ctx.src_params.height = avccontext->height;

    p_dirac_params->frame_size = avpicture_get_size(avccontext->pix_fmt,
                                                    avccontext->width,
                                                    avccontext->height);

    avccontext->coded_frame = &p_dirac_params->picture;

    if (no_local) {
        p_dirac_params->enc_ctx.decode_flag = 0;
        p_dirac_params->enc_ctx.instr_flag  = 0;
    } else {
        p_dirac_params->enc_ctx.decode_flag = 1;
        p_dirac_params->enc_ctx.instr_flag  = 1;
    }

    /* Intra-only sequence */
    if (!avccontext->gop_size) {
        p_dirac_params->enc_ctx.enc_params.num_L1 = 0;
        if (avccontext->coder_type == FF_CODER_TYPE_VLC)
            p_dirac_params->enc_ctx.enc_params.using_ac = 0;
    } else
        avccontext->has_b_frames = 1;

    if (avccontext->flags & CODEC_FLAG_QSCALE) {
        if (avccontext->global_quality) {
            p_dirac_params->enc_ctx.enc_params.qf =
                            avccontext->global_quality / (FF_QP2LAMBDA*10.0);
            /* if it is not default bitrate then send target rate. */
            if (avccontext->bit_rate >= 1000 &&
                avccontext->bit_rate != 200000) {
               p_dirac_params->enc_ctx.enc_params.trate =
                             avccontext->bit_rate / 1000;
            }
        } else
            p_dirac_params->enc_ctx.enc_params.lossless = 1;
    } else if (avccontext->bit_rate >= 1000)
        p_dirac_params->enc_ctx.enc_params.trate = avccontext->bit_rate / 1000;

    if ((preset > VIDEO_FORMAT_QCIF || preset < VIDEO_FORMAT_QSIF525) &&
         avccontext->bit_rate == 200000) {
        p_dirac_params->enc_ctx.enc_params.trate = 0;
    }

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

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

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

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

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

    return 0 ;
}

static void DiracFreeFrame (void *data)
{
    FfmpegDiracSchroEncodedFrame *enc_frame = data;

    av_freep (&(enc_frame->p_encbuf));
    av_free(enc_frame);
}

static int libdirac_encode_frame(AVCodecContext *avccontext,
                                 unsigned char *frame,
                                 int buf_size, void *data)
{
    int enc_size = 0;
    dirac_encoder_state_t state;
    FfmpegDiracEncoderParams* p_dirac_params = avccontext->priv_data;
    FfmpegDiracSchroEncodedFrame* p_frame_output = NULL;
    FfmpegDiracSchroEncodedFrame* p_next_output_frame = NULL;
    int go = 1;
    int last_frame_in_sequence = 0;

    if (!data) {
        /* push end of sequence if not already signalled */
        if (!p_dirac_params->eos_signalled) {
            dirac_encoder_end_sequence( p_dirac_params->p_encoder );
            p_dirac_params->eos_signalled = 1;
        }
    } else {

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

        /* load next frame */
        if (dirac_encoder_load (p_dirac_params->p_encoder,
                                p_dirac_params->p_in_frame_buf,
                                p_dirac_params->frame_size ) < 0) {
            av_log(avccontext, AV_LOG_ERROR, "Unrecoverable Encoder Error."
                   " dirac_encoder_load failed...\n");
            return -1;
        }
    }

    if (p_dirac_params->eos_pulled)
        go = 0;

    while(go) {
        p_dirac_params->p_encoder->enc_buf.buffer = frame;
        p_dirac_params->p_encoder->enc_buf.size   = buf_size;
        /* process frame */
        state = dirac_encoder_output ( p_dirac_params->p_encoder );

        switch (state)
        {
        case ENC_STATE_AVAIL:
        case ENC_STATE_EOS:
            assert (p_dirac_params->p_encoder->enc_buf.size > 0);

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

            p_dirac_params->enc_buf = av_realloc (
                                      p_dirac_params->enc_buf,
                                      p_dirac_params->enc_buf_size +
                                      p_dirac_params->p_encoder->enc_buf.size
                                      );
            memcpy(p_dirac_params->enc_buf + p_dirac_params->enc_buf_size,
                   p_dirac_params->p_encoder->enc_buf.buffer,
                   p_dirac_params->p_encoder->enc_buf.size);

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

            if (state == ENC_STATE_EOS) {
                p_dirac_params->eos_pulled = 1;
                go = 0;
            }

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

            /* create output frame */
            p_frame_output = av_mallocz(sizeof(FfmpegDiracSchroEncodedFrame));
            /* set output data */
            p_frame_output->size     = p_dirac_params->enc_buf_size;
            p_frame_output->p_encbuf = p_dirac_params->enc_buf;
            p_frame_output->frame_num =
                            p_dirac_params->p_encoder->enc_pparams.pnum;

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

            ff_dirac_schro_queue_push_back (&p_dirac_params->enc_frame_queue,
                                            p_frame_output);

            p_dirac_params->enc_buf_size = 0;
            p_dirac_params->enc_buf      = NULL;
            break;

        case ENC_STATE_BUFFER:
            go = 0;
            break;

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

        default:
            av_log(avccontext, AV_LOG_ERROR, "Unknown Dirac Encoder state\n");
            return -1;
        }
    }

    /* copy 'next' frame in queue */

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

    p_next_output_frame =
          ff_dirac_schro_queue_pop(&p_dirac_params->enc_frame_queue);

    if (!p_next_output_frame)
        return 0;

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

    /* Append the end of sequence information to the last frame in the
     * sequence. */
    if (last_frame_in_sequence && p_dirac_params->enc_buf_size > 0)
    {
        memcpy (frame + enc_size, p_dirac_params->enc_buf,
                p_dirac_params->enc_buf_size);
        enc_size += p_dirac_params->enc_buf_size;
        av_freep (&p_dirac_params->enc_buf);
        p_dirac_params->enc_buf_size = 0;
    }

    /* free frame */
    DiracFreeFrame(p_next_output_frame);

    return enc_size;
}

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

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

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

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

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

    return 0 ;
}


AVCodec libdirac_encoder = {
    "libdirac",
    CODEC_TYPE_VIDEO,
    CODEC_ID_DIRAC,
    sizeof(FfmpegDiracEncoderParams),
    libdirac_encode_init,
    libdirac_encode_frame,
    libdirac_encode_close,
   .capabilities= CODEC_CAP_DELAY,
   .pix_fmts= (enum PixelFormat[]){PIX_FMT_YUV420P, PIX_FMT_YUV422P, PIX_FMT_YUV444P, PIX_FMT_NONE},
   .long_name= NULL_IF_CONFIG_SMALL("libdirac Dirac 2.2"),
} ;
Commit	Line	Data
690bfceb AS	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	/**
bad5537e	24	* @file libavcodec/libdiracenc.c
690bfceb AS	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	{
	43	/** Dirac encoder context */
	44	dirac_encoder_context_t enc_ctx;
	45
	46	/** frame being encoded */
	47	AVFrame picture;
	48
	49	/** frame size */
	50	int frame_size;
	51
	52	/** Dirac encoder handle */
	53	dirac_encoder_t* p_encoder;
	54
	55	/** input frame buffer */
	56	unsigned char *p_in_frame_buf;
	57
f73b2040 AS	58	/** buffer to store encoder output before writing it to the frame queue */
	59	unsigned char *enc_buf;
	60
	61	/** size of encoder buffer */
	62	int enc_buf_size;
	63
690bfceb AS	64	/** queue storing encoded frames */
	65	FfmpegDiracSchroQueue enc_frame_queue;
	66
	67	/** end of sequence signalled by user, 0 - false, 1 - true */
	68	int eos_signalled;
	69
	70	/** end of sequence returned by encoder, 0 - false, 1 - true */
	71	int eos_pulled;
	72	} FfmpegDiracEncoderParams;
	73
	74	/**
	75	* Works out Dirac-compatible chroma format.
	76	*/
	77	static dirac_chroma_t GetDiracChromaFormat(enum PixelFormat ff_pix_fmt)
	78	{
	79	int num_formats = sizeof(ffmpeg_dirac_pixel_format_map) /
	80	sizeof(ffmpeg_dirac_pixel_format_map[0]);
	81	int idx;
	82
	83	for (idx = 0; idx < num_formats; ++idx) {
	84	if (ffmpeg_dirac_pixel_format_map[idx].ff_pix_fmt == ff_pix_fmt) {
	85	return ffmpeg_dirac_pixel_format_map[idx].dirac_pix_fmt;
	86	}
	87	}
	88	return formatNK;
	89	}
	90
	91	/**
	92	* Dirac video preset table. Ensure that this tables matches up correctly
	93	* with the ff_dirac_schro_video_format_info table in libdirac_libschro.c.
	94	*/
	95	static const VideoFormat ff_dirac_video_formats[]={
	96	VIDEO_FORMAT_CUSTOM ,
	97	VIDEO_FORMAT_QSIF525 ,
	98	VIDEO_FORMAT_QCIF ,
	99	VIDEO_FORMAT_SIF525 ,
	100	VIDEO_FORMAT_CIF ,
	101	VIDEO_FORMAT_4SIF525 ,
	102	VIDEO_FORMAT_4CIF ,
	103	VIDEO_FORMAT_SD_480I60 ,
	104	VIDEO_FORMAT_SD_576I50 ,
	105	VIDEO_FORMAT_HD_720P60 ,
	106	VIDEO_FORMAT_HD_720P50 ,
	107	VIDEO_FORMAT_HD_1080I60 ,
	108	VIDEO_FORMAT_HD_1080I50 ,
	109	VIDEO_FORMAT_HD_1080P60 ,
	110	VIDEO_FORMAT_HD_1080P50 ,
	111	VIDEO_FORMAT_DIGI_CINEMA_2K24 ,
	112	VIDEO_FORMAT_DIGI_CINEMA_4K24 ,
	113	};
	114
	115	/**
	116	* Returns the video format preset matching the input video dimensions and
	117	* time base.
	118	*/
	119	static VideoFormat GetDiracVideoFormatPreset (AVCodecContext *avccontext)
	120	{
	121	unsigned int num_formats = sizeof(ff_dirac_video_formats) /
	122	sizeof(ff_dirac_video_formats[0]);
	123
	124	unsigned int idx = ff_dirac_schro_get_video_format_idx (avccontext);
	125
	126	return (idx < num_formats) ?
	127	ff_dirac_video_formats[idx] : VIDEO_FORMAT_CUSTOM;
128	}
129
5ef251e5	130	static av_cold int libdirac_encode_init(AVCodecContext *avccontext)
690bfceb AS	131	{
	132
	133	FfmpegDiracEncoderParams* p_dirac_params = avccontext->priv_data;
	134	int no_local = 1;
	135	int verbose = avccontext->debug;
	136	VideoFormat preset;
	137
	138	/* get Dirac preset */
	139	preset = GetDiracVideoFormatPreset(avccontext);
	140
	141	/* initialize the encoder context */
	142	dirac_encoder_context_init (&(p_dirac_params->enc_ctx), preset);
	143
	144	p_dirac_params->enc_ctx.src_params.chroma =
	145	GetDiracChromaFormat(avccontext->pix_fmt);
	146
	147	if (p_dirac_params->enc_ctx.src_params.chroma == formatNK) {
	148	av_log (avccontext, AV_LOG_ERROR,
	149	"Unsupported pixel format %d. This codec supports only "
	150	"Planar YUV formats (yuv420p, yuv422p, yuv444p\n",
	151	avccontext->pix_fmt);
	152	return -1;
	153	}
	154
	155	p_dirac_params->enc_ctx.src_params.frame_rate.numerator =
	156	avccontext->time_base.den;
	157	p_dirac_params->enc_ctx.src_params.frame_rate.denominator =
	158	avccontext->time_base.num;
	159
	160	p_dirac_params->enc_ctx.src_params.width = avccontext->width;
	161	p_dirac_params->enc_ctx.src_params.height = avccontext->height;
	162
	163	p_dirac_params->frame_size = avpicture_get_size(avccontext->pix_fmt,
	164	avccontext->width,
	165	avccontext->height);
	166
	167	avccontext->coded_frame = &p_dirac_params->picture;
	168
	169	if (no_local) {
	170	p_dirac_params->enc_ctx.decode_flag = 0;
	171	p_dirac_params->enc_ctx.instr_flag = 0;
	172	} else {
	173	p_dirac_params->enc_ctx.decode_flag = 1;
	174	p_dirac_params->enc_ctx.instr_flag = 1;
	175	}
	176
	177	/* Intra-only sequence */
7c809dc3	178	if (!avccontext->gop_size) {
690bfceb	179	p_dirac_params->enc_ctx.enc_params.num_L1 = 0;
d7c96d66 AS	180	if (avccontext->coder_type == FF_CODER_TYPE_VLC)
	181	p_dirac_params->enc_ctx.enc_params.using_ac = 0;
	182	} else
690bfceb AS	183	avccontext->has_b_frames = 1;
	184
	185	if (avccontext->flags & CODEC_FLAG_QSCALE) {
7c809dc3	186	if (avccontext->global_quality) {
690bfceb AS	187	p_dirac_params->enc_ctx.enc_params.qf =
	188	avccontext->global_quality / (FF_QP2LAMBDA*10.0);
	189	/* if it is not default bitrate then send target rate. */
	190	if (avccontext->bit_rate >= 1000 &&
	191	avccontext->bit_rate != 200000) {
	192	p_dirac_params->enc_ctx.enc_params.trate =
	193	avccontext->bit_rate / 1000;
	194	}
	195	} else
	196	p_dirac_params->enc_ctx.enc_params.lossless = 1;
	197	} else if (avccontext->bit_rate >= 1000)
	198	p_dirac_params->enc_ctx.enc_params.trate = avccontext->bit_rate / 1000;
	199
	200	if ((preset > VIDEO_FORMAT_QCIF \|\| preset < VIDEO_FORMAT_QSIF525) &&
	201	avccontext->bit_rate == 200000) {
	202	p_dirac_params->enc_ctx.enc_params.trate = 0;
	203	}
	204
	205	if (avccontext->flags & CODEC_FLAG_INTERLACED_ME) {
	206	/* all material can be coded as interlaced or progressive
	207	* irrespective of the type of source material */
	208	p_dirac_params->enc_ctx.enc_params.picture_coding_mode = 1;
	209	}
	210
	211	p_dirac_params->p_encoder = dirac_encoder_init (&(p_dirac_params->enc_ctx),
	212	verbose );
	213
	214	if (!p_dirac_params->p_encoder) {
	215	av_log(avccontext, AV_LOG_ERROR,
	216	"Unrecoverable Error: dirac_encoder_init failed. ");
	217	return EXIT_FAILURE;
	218	}
	219
	220	/* allocate enough memory for the incoming data */
	221	p_dirac_params->p_in_frame_buf = av_malloc(p_dirac_params->frame_size);
	222
	223	/* initialize the encoded frame queue */
	224	ff_dirac_schro_queue_init(&p_dirac_params->enc_frame_queue);
	225
	226	return 0 ;
	227	}
	228
	229	static void DiracFreeFrame (void *data)
	230	{
	231	FfmpegDiracSchroEncodedFrame *enc_frame = data;
	232
	233	av_freep (&(enc_frame->p_encbuf));
	234	av_free(enc_frame);
	235	}
	236
	237	static int libdirac_encode_frame(AVCodecContext *avccontext,
	238	unsigned char *frame,
	239	int buf_size, void *data)
	240	{
	241	int enc_size = 0;
	242	dirac_encoder_state_t state;
	243	FfmpegDiracEncoderParams* p_dirac_params = avccontext->priv_data;
	244	FfmpegDiracSchroEncodedFrame* p_frame_output = NULL;
	245	FfmpegDiracSchroEncodedFrame* p_next_output_frame = NULL;
	246	int go = 1;
f73b2040	247	int last_frame_in_sequence = 0;
690bfceb	248
7c809dc3	249	if (!data) {
690bfceb AS	250	/* push end of sequence if not already signalled */
	251	if (!p_dirac_params->eos_signalled) {
	252	dirac_encoder_end_sequence( p_dirac_params->p_encoder );
	253	p_dirac_params->eos_signalled = 1;
	254	}
	255	} else {
	256
	257	/* Allocate frame data to Dirac input buffer.
	258	* Input line size may differ from what the codec supports,
	259	* especially when transcoding from one format to another.
	260	* So use avpicture_layout to copy the frame. */
	261	avpicture_layout ((AVPicture *)data, avccontext->pix_fmt,
	262	avccontext->width, avccontext->height,
	263	p_dirac_params->p_in_frame_buf,
	264	p_dirac_params->frame_size);
	265
	266	/* load next frame */
	267	if (dirac_encoder_load (p_dirac_params->p_encoder,
	268	p_dirac_params->p_in_frame_buf,
	269	p_dirac_params->frame_size ) < 0) {
	270	av_log(avccontext, AV_LOG_ERROR, "Unrecoverable Encoder Error."
	271	" dirac_encoder_load failed...\n");
	272	return -1;
	273	}
	274	}
	275
	276	if (p_dirac_params->eos_pulled)
	277	go = 0;
	278
	279	while(go) {
	280	p_dirac_params->p_encoder->enc_buf.buffer = frame;
	281	p_dirac_params->p_encoder->enc_buf.size = buf_size;
	282	/* process frame */
	283	state = dirac_encoder_output ( p_dirac_params->p_encoder );
	284
	285	switch (state)
	286	{
	287	case ENC_STATE_AVAIL:
	288	case ENC_STATE_EOS:
	289	assert (p_dirac_params->p_encoder->enc_buf.size > 0);
690bfceb	290
f73b2040 AS	291	/* All non-frame data is prepended to actual frame data to
	292	* be able to set the pts correctly. So we don't write data
	293	* to the frame output queue until we actually have a frame
	294	*/
	295
	296	p_dirac_params->enc_buf = av_realloc (
	297	p_dirac_params->enc_buf,
	298	p_dirac_params->enc_buf_size +
	299	p_dirac_params->p_encoder->enc_buf.size
	300	);
	301	memcpy(p_dirac_params->enc_buf + p_dirac_params->enc_buf_size,
690bfceb AS	302	p_dirac_params->p_encoder->enc_buf.buffer,
	303	p_dirac_params->p_encoder->enc_buf.size);
	304
f73b2040 AS	305	p_dirac_params->enc_buf_size +=
	306	p_dirac_params->p_encoder->enc_buf.size;
	307
	308	if (state == ENC_STATE_EOS) {
	309	p_dirac_params->eos_pulled = 1;
	310	go = 0;
	311	}
	312
	313	/* If non-frame data, don't output it until it we get an
	314	* encoded frame back from the encoder. */
	315	if (p_dirac_params->p_encoder->enc_pparams.pnum == -1)
	316	break;
690bfceb	317
f73b2040 AS	318	/* create output frame */
	319	p_frame_output = av_mallocz(sizeof(FfmpegDiracSchroEncodedFrame));
	320	/* set output data */
	321	p_frame_output->size = p_dirac_params->enc_buf_size;
	322	p_frame_output->p_encbuf = p_dirac_params->enc_buf;
690bfceb AS	323	p_frame_output->frame_num =
	324	p_dirac_params->p_encoder->enc_pparams.pnum;
	325
	326	if (p_dirac_params->p_encoder->enc_pparams.ptype == INTRA_PICTURE &&
	327	p_dirac_params->p_encoder->enc_pparams.rtype == REFERENCE_PICTURE)
	328	p_frame_output->key_frame = 1;
	329
	330	ff_dirac_schro_queue_push_back (&p_dirac_params->enc_frame_queue,
	331	p_frame_output);
	332
f73b2040 AS	333	p_dirac_params->enc_buf_size = 0;
f73b2040 AS	334	p_dirac_params->enc_buf = NULL;
690bfceb AS	335	break;
	336
	337	case ENC_STATE_BUFFER:
	338	go = 0;
	339	break;
	340
	341	case ENC_STATE_INVALID:
	342	av_log(avccontext, AV_LOG_ERROR,
	343	"Unrecoverable Dirac Encoder Error. Quitting...\n");
	344	return -1;
	345
	346	default:
	347	av_log(avccontext, AV_LOG_ERROR, "Unknown Dirac Encoder state\n");
	348	return -1;
	349	}
	350	}
	351
	352	/* copy 'next' frame in queue */
f73b2040 AS	353
	354	if (p_dirac_params->enc_frame_queue.size == 1 &&
	355	p_dirac_params->eos_pulled)
	356	last_frame_in_sequence = 1;
	357
690bfceb AS	358	p_next_output_frame =
	359	ff_dirac_schro_queue_pop(&p_dirac_params->enc_frame_queue);
	360
7c809dc3	361	if (!p_next_output_frame)
690bfceb AS	362	return 0;
	363
	364	memcpy(frame, p_next_output_frame->p_encbuf, p_next_output_frame->size);
	365	avccontext->coded_frame->key_frame = p_next_output_frame->key_frame;
	366	/* Use the frame number of the encoded frame as the pts. It is OK to do
	367	* so since Dirac is a constant framerate codec. It expects input to be
	368	* of constant framerate. */
	369	avccontext->coded_frame->pts = p_next_output_frame->frame_num;
	370	enc_size = p_next_output_frame->size;
	371
f73b2040 AS	372	/* Append the end of sequence information to the last frame in the
	373	* sequence. */
	374	if (last_frame_in_sequence && p_dirac_params->enc_buf_size > 0)
	375	{
	376	memcpy (frame + enc_size, p_dirac_params->enc_buf,
	377	p_dirac_params->enc_buf_size);
	378	enc_size += p_dirac_params->enc_buf_size;
	379	av_freep (&p_dirac_params->enc_buf);
	380	p_dirac_params->enc_buf_size = 0;
	381	}
	382
690bfceb AS	383	/* free frame */
	384	DiracFreeFrame(p_next_output_frame);
	385
	386	return enc_size;
	387	}
	388
5ef251e5	389	static av_cold int libdirac_encode_close(AVCodecContext *avccontext)
690bfceb AS	390	{
	391	FfmpegDiracEncoderParams* p_dirac_params = avccontext->priv_data;
	392
	393	/* close the encoder */
	394	dirac_encoder_close(p_dirac_params->p_encoder );
	395
	396	/* free data in the output frame queue */
	397	ff_dirac_schro_queue_free(&p_dirac_params->enc_frame_queue,
	398	DiracFreeFrame);
	399
f73b2040 AS	400	/* free the encoder buffer */
	401	if (p_dirac_params->enc_buf_size)
	402	av_freep(&p_dirac_params->enc_buf);
	403
690bfceb AS	404	/* free the input frame buffer */
	405	av_freep(&p_dirac_params->p_in_frame_buf);
	406
	407	return 0 ;
	408	}
	409
	410
	411	AVCodec libdirac_encoder = {
	412	"libdirac",
	413	CODEC_TYPE_VIDEO,
	414	CODEC_ID_DIRAC,
	415	sizeof(FfmpegDiracEncoderParams),
	416	libdirac_encode_init,
	417	libdirac_encode_frame,
	418	libdirac_encode_close,
	419	.capabilities= CODEC_CAP_DELAY,
3bd9f708	420	.pix_fmts= (enum PixelFormat[]){PIX_FMT_YUV420P, PIX_FMT_YUV422P, PIX_FMT_YUV444P, PIX_FMT_NONE},
fe4bf374	421	.long_name= NULL_IF_CONFIG_SMALL("libdirac Dirac 2.2"),
690bfceb	422	} ;