[FFmpeg-cvslog] flashsv2enc: Replace a VLA with a heap alloc
Derek Buitenhuis
git at videolan.org
Fri Sep 7 21:56:40 CEST 2012
ffmpeg | branch: master | Derek Buitenhuis <derek.buitenhuis at gmail.com> | Fri Sep 7 10:38:15 2012 -0400| [3174987b4263651286c8d17d346d224750c368fd] | committer: Michael Niedermayer
flashsv2enc: Replace a VLA with a heap alloc
Signed-off-by: Derek Buitenhuis <derek.buitenhuis at gmail.com>
Signed-off-by: Michael Niedermayer <michaelni at gmx.at>
> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=3174987b4263651286c8d17d346d224750c368fd
---
libavcodec/flashsv2enc.c | 25 ++++++++++++++++++++-----
1 file changed, 20 insertions(+), 5 deletions(-)
diff --git a/libavcodec/flashsv2enc.c b/libavcodec/flashsv2enc.c
index 008ebe0..5a1d594 100644
--- a/libavcodec/flashsv2enc.c
+++ b/libavcodec/flashsv2enc.c
@@ -92,6 +92,9 @@ typedef struct FlashSV2Context {
uint8_t *keybuffer;
uint8_t *databuffer;
+ uint8_t *blockbuffer;
+ int blockbuffer_size;
+
Block *frame_blocks;
Block *key_blocks;
int frame_size;
@@ -127,6 +130,7 @@ static av_cold void cleanup(FlashSV2Context * s)
av_freep(&s->encbuffer);
av_freep(&s->keybuffer);
av_freep(&s->databuffer);
+ av_freep(&s->blockbuffer);
av_freep(&s->current_frame);
av_freep(&s->key_frame);
@@ -229,6 +233,9 @@ static av_cold int flashsv2_encode_init(AVCodecContext * avctx)
s->frame_blocks = av_mallocz(s->blocks_size);
s->key_blocks = av_mallocz(s->blocks_size);
+ s->blockbuffer = NULL;
+ s->blockbuffer_size = 0;
+
init_blocks(s, s->frame_blocks, s->encbuffer, s->databuffer);
init_blocks(s, s->key_blocks, s->keybuffer, 0);
reset_stats(s);
@@ -542,13 +549,21 @@ static int encode_15_7(Palette * palette, Block * b, const uint8_t * src,
return b->enc_size;
}
-static int encode_block(Palette * palette, Block * b, Block * prev,
- const uint8_t * src, int stride, int comp, int dist,
- int keyframe)
+static int encode_block(FlashSV2Context *s, Palette * palette, Block * b,
+ Block * prev, const uint8_t * src, int stride, int comp,
+ int dist, int keyframe)
{
unsigned buf_size = b->width * b->height * 6;
- uint8_t buf[buf_size];
+ uint8_t *buf;
int res;
+
+ av_fast_malloc(&s->blockbuffer, &s->blockbuffer_size, buf_size);
+ if (!s->blockbuffer) {
+ av_log(s->avctx, AV_LOG_ERROR, "Could not allocate block buffer.\n");
+ return AVERROR(ENOMEM);
+ }
+ buf = s->blockbuffer;
+
if (b->flags & COLORSPACE_15_7) {
encode_15_7(palette, b, src, stride, dist);
} else {
@@ -640,7 +655,7 @@ static int encode_all_blocks(FlashSV2Context * s, int keyframe)
b->flags = s->use15_7 ? COLORSPACE_15_7 | HAS_DIFF_BLOCKS : HAS_DIFF_BLOCKS;
}
data = s->current_frame + s->image_width * 3 * s->block_height * row + s->block_width * col * 3;
- res = encode_block(&s->palette, b, prev, data, s->image_width * 3, s->comp, s->dist, keyframe);
+ res = encode_block(s, &s->palette, b, prev, data, s->image_width * 3, s->comp, s->dist, keyframe);
#ifndef FLASHSV2_DUMB
if (b->dirty)
s->diff_blocks++;
More information about the ffmpeg-cvslog
mailing list