ffmpeg avframe yuv 旋转

最新推荐文章于 2025-05-26 22:40:59 发布

满衣兄

最新推荐文章于 2025-05-26 22:40:59 发布

阅读量2.8k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： C++学习

本文链接：https://blog.youkuaiyun.com/yfqvip/article/details/79351126

C++学习专栏收录该内容

23 篇文章

订阅专栏

本文详细介绍了如何使用FFmpeg库中的AVFrame结构实现视频帧的顺时针旋转功能，包括90度、180度及270度旋转的具体算法实现，并提供了完整的代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

// 顺时针旋转 ffmpeg avframe yuv  rotation


enum VideoRotationMode {
  kRotateNone = 0,
  kRotate90 = 90,
  kRotate180 = 180,
  kRotate270 = 270,
};






VideoRotationMode rotation_ = kRotateNone；






bool Rotate(const AVFrame* src, AVFrame** dst)
{
    if (rotation_ == kRotateNone) {
        return false;
    }
    AVFrame* tmp = *dst;
    if (!tmp) {
        tmp = av_frame_alloc();
    }
// width_、height_ 为src的宽高，若是90或者270则宽高交换值
    if (tmp->width != width_ || tmp->height != height_) {
        av_frame_free(&tmp);
        tmp = av_frame_alloc();
        uint8_t* buffer = (uint8_t*)av_malloc(P_av_image_get_buffer_size(AV_PIX_FMT_YUV420P, width_, height_, 1));
        av_image_fill_arrays(tmp->data, tmp->linesize, buffer, AV_PIX_FMT_YUV420P, width_, height_, 1);
    }




    if (rotation_ == kRotate90) {
        Rotate90(src, tmp);
    }
    else if (rotation_ == kRotate180) {
        Rotate180(src, tmp);
    }
    else if (rotation_ == kRotate270) {
        Rotate270(src, tmp);
    }
    else {
        return false;
    }


    tmp->format     = src->format;
    tmp->pts        = src->pts;
    tmp->pkt_pts    = src->pkt_pts;
    tmp->pkt_dts    = src->pkt_dts;
    tmp->key_frame  = src->key_frame;


    *dst = tmp;


    return true;
}




void Rotate90(const AVFrame* src, AVFrame* dst)
{
    int half_width = src->width >> 1;
    int half_height = src->height >> 1;




    int size = src->linesize[0] * src->height;
    int half_size = size >> 2;




    for (int j = 0, n = 0; j < src->width; j++) {
        int pos = size;
        for (int i = src->height - 1; i >= 0; i--) {
            pos -= src->linesize[0];
            dst->data[0][n++] = src->data[0][pos + j];
        }
    }




    for (int j = 0, n = 0; j < half_width; j++) {
        int pos = half_size;
        for (int i = half_height - 1; i >= 0; i--) {
            pos -= src->linesize[1];
            dst->data[1][n] = src->data[1][pos + j];
            dst->data[2][n++] = src->data[2][pos + j];
        }
    }




    dst->height = src->width;
    dst->width  = src->height;
}




void Rotate180(const AVFrame* src, AVFrame* dst)
{
    int half_width = src->width >> 1;
    int half_height = src->height >> 1;




    int pos = src->linesize[0] * src->height;
    for (int i = 0, n = 0; i < src->height; i++) {
        pos -= src->linesize[0];
        for (int j = src->width - 1; j >= 0; j--) {
            dst->data[0][n++] = src->data[0][pos + j];
        }
    }




    pos = src->linesize[0] * src->height >> 2;
    for (int i = 0, n = 0; i < half_height; i++) {
        pos -= src->linesize[1];
        for (int j = half_width - 1; j >= 0; j--) {
            dst->data[1][n] = src->data[1][pos + j];
            dst->data[2][n++] = src->data[2][pos + j];
        }
    }




    dst->width  = src->width;
    dst->height = src->height;
}




void Rotate270(const AVFrame* src, AVFrame* dst)
{
    int half_width = src->linesize[0] >> 1;
    int half_height = src->height >> 1;




    for (int i = src->width - 1, n = 0; i >= 0; i--) {
        for (int j = 0, pos = 0; j < src->height; j++) {
            dst->data[0][n++] = src->data[0][pos + i];
            pos += src->linesize[0];
        }
    }




    for (int i = (src->width >> 1) - 1, n = 0; i >= 0; i--) {
        for (int j = 0, pos = 0; j < half_height; j++) {
            dst->data[1][n] = src->data[1][pos + i];
            dst->data[2][n++] = src->data[2][pos + i];
            pos += half_width;
        }
    }




    dst->width = src->height;
    dst->height = src->width;
}

参考：http://blog.youkuaiyun.com/veilling/article/details/52200326