从AVFrame复制矩形区域 - ffmpeg

时间:2013-02-04 21:48:03

标签: ffmpeg rgb libav

我试图拉出一个AVFrame的矩形区域,并开始使用这样做的功能。我只对使用格式为PIX_FMT_RGB24的AVFrame感兴趣。我也许在这里重新发明轮子,所以如果已经有一个功能,请跳进去。到目前为止,我的功能看起来像这样:

AVFrame * getRGBsection(AVFrame *pFrameRGB, const int start_x, const int start_y, const int w, const int h) {

AVFrame *pFrameSect;
int numBytes;
uint8_t *mb_buffer;

pFrameSect = avcodec_alloc_frame();
numBytes = avpicture_get_size(PIX_FMT_RGB24, w, h);
mb_buffer = (uint8_t *) av_malloc(numBytes * sizeof(uint8_t));
avpicture_fill((AVPicture *) pFrameSect, mb_buffer, PIX_FMT_RGB24, w, h);

int curY, curX, i = 0;
for (curY = start_y ; curY < (start_y + h); curY++) {

    for (curX = start_x; curX < (start_x + w); curX++) {

        int curIndex = curX * 3 + curY * pFrameRGB->linesize[0];

        pFrameSect->data[0][i] = pFrameRGB->data[0][curIndex];
        pFrameSect->data[0][i + 1] = pFrameRGB->data[0][curIndex + 1];
        pFrameSect->data[0][i + 2] = pFrameRGB->data[0][curIndex + 2];

        i += 3;

    }

}

return pFrameSect;

}

当我从(0,0)(我认为)开始时,该功能似乎有效,但当我移动到图像中的其他地方时,它输出的颜色与应该存在的颜色相似,但是不对。我想我在这附近很近,有人可以提供指导吗?

2 个答案:

答案 0 :(得分:1)

  • 有两个选项

    1. 用户视频过滤器(vf_crop)。 (filtering_video.c提供了实用的裁剪示例)
    2. imgconvert.c中的函数av_picture_crop()。此功能尚未完成,但您可以对其进行修改以供您使用。

答案 1 :(得分:1)

此代码适用于我(仅限RGB24)

#include <libavutil/imgutils.h>

// .....
// left, top
const int crop_left = 500;
const int crop_top = 500;

// output width, height
const int crop_width = 300;
const int crop_height = 200;

AVFrame * rgb24picure;
AVFrame * output;
/// .... initialize ....

const int one_pixel = av_image_get_linesize(PIX_FMT_RGB24, 1, 0);
const int src_line_size = av_image_get_linesize(PIX_FMT_RGB24, source_width, 0);
const int copy_line_size = av_image_get_linesize(PIX_FMT_RGB24, crop_width, 0);

for (int h = crop_top; h < crop_top + crop_height; ++h)
{
    unsigned char * src = rgb24picure->data[0] + src_line_size * h + one_pixel * crop_left;
    unsigned char * dst = output->data[0] + copy_line_size * (h - crop_top);
    memcpy(dst, src, copy_line_size);
}