Commit 78c6c9d6 authored by Anton Khirnov's avatar Anton Khirnov

pngenc: use the AVFrame API properly.

parent 0ea430c7
...@@ -37,7 +37,6 @@ typedef struct PNGEncContext { ...@@ -37,7 +37,6 @@ typedef struct PNGEncContext {
uint8_t *bytestream; uint8_t *bytestream;
uint8_t *bytestream_start; uint8_t *bytestream_start;
uint8_t *bytestream_end; uint8_t *bytestream_end;
AVFrame picture;
int filter_type; int filter_type;
...@@ -231,7 +230,7 @@ static int encode_frame(AVCodecContext *avctx, AVPacket *pkt, ...@@ -231,7 +230,7 @@ static int encode_frame(AVCodecContext *avctx, AVPacket *pkt,
const AVFrame *pict, int *got_packet) const AVFrame *pict, int *got_packet)
{ {
PNGEncContext *s = avctx->priv_data; PNGEncContext *s = avctx->priv_data;
AVFrame * const p= &s->picture; const AVFrame * const p = pict;
int bit_depth, color_type, y, len, row_size, ret, is_progressive; int bit_depth, color_type, y, len, row_size, ret, is_progressive;
int bits_per_pixel, pass_row_size, enc_row_size, max_packet_size; int bits_per_pixel, pass_row_size, enc_row_size, max_packet_size;
int compression_level; int compression_level;
...@@ -241,10 +240,6 @@ static int encode_frame(AVCodecContext *avctx, AVPacket *pkt, ...@@ -241,10 +240,6 @@ static int encode_frame(AVCodecContext *avctx, AVPacket *pkt,
uint8_t *rgba_buf = NULL; uint8_t *rgba_buf = NULL;
uint8_t *top_buf = NULL; uint8_t *top_buf = NULL;
*p = *pict;
p->pict_type= AV_PICTURE_TYPE_I;
p->key_frame= 1;
is_progressive = !!(avctx->flags & CODEC_FLAG_INTERLACED_DCT); is_progressive = !!(avctx->flags & CODEC_FLAG_INTERLACED_DCT);
switch(avctx->pix_fmt) { switch(avctx->pix_fmt) {
case AV_PIX_FMT_RGB32: case AV_PIX_FMT_RGB32:
...@@ -444,8 +439,13 @@ static int encode_frame(AVCodecContext *avctx, AVPacket *pkt, ...@@ -444,8 +439,13 @@ static int encode_frame(AVCodecContext *avctx, AVPacket *pkt,
static av_cold int png_enc_init(AVCodecContext *avctx){ static av_cold int png_enc_init(AVCodecContext *avctx){
PNGEncContext *s = avctx->priv_data; PNGEncContext *s = avctx->priv_data;
avcodec_get_frame_defaults(&s->picture); avctx->coded_frame = av_frame_alloc();
avctx->coded_frame= &s->picture; if (!avctx->coded_frame)
return AVERROR(ENOMEM);
avctx->coded_frame->pict_type = AV_PICTURE_TYPE_I;
avctx->coded_frame->key_frame = 1;
ff_dsputil_init(&s->dsp, avctx); ff_dsputil_init(&s->dsp, avctx);
s->filter_type = av_clip(avctx->prediction_method, PNG_FILTER_VALUE_NONE, PNG_FILTER_VALUE_MIXED); s->filter_type = av_clip(avctx->prediction_method, PNG_FILTER_VALUE_NONE, PNG_FILTER_VALUE_MIXED);
...@@ -455,6 +455,12 @@ static av_cold int png_enc_init(AVCodecContext *avctx){ ...@@ -455,6 +455,12 @@ static av_cold int png_enc_init(AVCodecContext *avctx){
return 0; return 0;
} }
static av_cold int png_enc_close(AVCodecContext *avctx)
{
av_frame_free(&avctx->coded_frame);
return 0;
}
AVCodec ff_png_encoder = { AVCodec ff_png_encoder = {
.name = "png", .name = "png",
.long_name = NULL_IF_CONFIG_SMALL("PNG (Portable Network Graphics) image"), .long_name = NULL_IF_CONFIG_SMALL("PNG (Portable Network Graphics) image"),
...@@ -462,6 +468,7 @@ AVCodec ff_png_encoder = { ...@@ -462,6 +468,7 @@ AVCodec ff_png_encoder = {
.id = AV_CODEC_ID_PNG, .id = AV_CODEC_ID_PNG,
.priv_data_size = sizeof(PNGEncContext), .priv_data_size = sizeof(PNGEncContext),
.init = png_enc_init, .init = png_enc_init,
.close = png_enc_close,
.encode2 = encode_frame, .encode2 = encode_frame,
.pix_fmts = (const enum AVPixelFormat[]){ .pix_fmts = (const enum AVPixelFormat[]){
AV_PIX_FMT_RGB24, AV_PIX_FMT_RGB32, AV_PIX_FMT_PAL8, AV_PIX_FMT_GRAY8, AV_PIX_FMT_RGB24, AV_PIX_FMT_RGB32, AV_PIX_FMT_PAL8, AV_PIX_FMT_GRAY8,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment