| @@ -21,22 +21,24 @@ | |||||
| #include "avcodec.h" | #include "avcodec.h" | ||||
| #include "bytestream.h" | #include "bytestream.h" | ||||
| #include "internal.h" | |||||
| #include "pnm.h" | #include "pnm.h" | ||||
| static int pam_encode_frame(AVCodecContext *avctx, unsigned char *outbuf, | |||||
| int buf_size, void *data) | |||||
| static int pam_encode_frame(AVCodecContext *avctx, AVPacket *pkt, | |||||
| const AVFrame *pict, int *got_packet) | |||||
| { | { | ||||
| PNMContext *s = avctx->priv_data; | PNMContext *s = avctx->priv_data; | ||||
| AVFrame *pict = data; | |||||
| AVFrame * const p = (AVFrame*)&s->picture; | AVFrame * const p = (AVFrame*)&s->picture; | ||||
| int i, h, w, n, linesize, depth, maxval; | |||||
| int i, h, w, n, linesize, depth, maxval, ret; | |||||
| const char *tuple_type; | const char *tuple_type; | ||||
| uint8_t *ptr; | uint8_t *ptr; | ||||
| if (buf_size < avpicture_get_size(avctx->pix_fmt, avctx->width, avctx->height) + 200) { | |||||
| if ((ret = ff_alloc_packet(pkt, avpicture_get_size(avctx->pix_fmt, | |||||
| avctx->width, | |||||
| avctx->height) + 200)) < 0) { | |||||
| av_log(avctx, AV_LOG_ERROR, "encoded frame too large\n"); | av_log(avctx, AV_LOG_ERROR, "encoded frame too large\n"); | ||||
| return -1; | |||||
| return ret; | |||||
| } | } | ||||
| *p = *pict; | *p = *pict; | ||||
| @@ -44,8 +46,8 @@ static int pam_encode_frame(AVCodecContext *avctx, unsigned char *outbuf, | |||||
| p->key_frame = 1; | p->key_frame = 1; | ||||
| s->bytestream_start = | s->bytestream_start = | ||||
| s->bytestream = outbuf; | |||||
| s->bytestream_end = outbuf+buf_size; | |||||
| s->bytestream = pkt->data; | |||||
| s->bytestream_end = pkt->data + pkt->size; | |||||
| h = avctx->height; | h = avctx->height; | ||||
| w = avctx->width; | w = avctx->width; | ||||
| @@ -104,7 +106,11 @@ static int pam_encode_frame(AVCodecContext *avctx, unsigned char *outbuf, | |||||
| ptr += linesize; | ptr += linesize; | ||||
| } | } | ||||
| } | } | ||||
| return s->bytestream - s->bytestream_start; | |||||
| pkt->size = s->bytestream - s->bytestream_start; | |||||
| pkt->flags |= AV_PKT_FLAG_KEY; | |||||
| *got_packet = 1; | |||||
| return 0; | |||||
| } | } | ||||
| @@ -114,7 +120,7 @@ AVCodec ff_pam_encoder = { | |||||
| .id = CODEC_ID_PAM, | .id = CODEC_ID_PAM, | ||||
| .priv_data_size = sizeof(PNMContext), | .priv_data_size = sizeof(PNMContext), | ||||
| .init = ff_pnm_init, | .init = ff_pnm_init, | ||||
| .encode = pam_encode_frame, | |||||
| .encode2 = pam_encode_frame, | |||||
| .pix_fmts = (const enum PixelFormat[]){PIX_FMT_RGB24, PIX_FMT_RGB32, PIX_FMT_GRAY8, PIX_FMT_MONOWHITE, PIX_FMT_NONE}, | .pix_fmts = (const enum PixelFormat[]){PIX_FMT_RGB24, PIX_FMT_RGB32, PIX_FMT_GRAY8, PIX_FMT_MONOWHITE, PIX_FMT_NONE}, | ||||
| .long_name = NULL_IF_CONFIG_SMALL("PAM (Portable AnyMap) image"), | .long_name = NULL_IF_CONFIG_SMALL("PAM (Portable AnyMap) image"), | ||||
| }; | }; | ||||