如何将AVFrame转换为类

cv::Mat mat;

int width = avFrame->width;
int height = avFrame->height;
AVPixelFormat pixelFormat = avFrame->format;

int cvPixelFormat;
switch (pixelFormat) {
    case AV_PIX_FMT_BGR24:
        cvPixelFormat = CV_8UC3;
        break;
    case AV_PIX_FMT_GRAY8:
        cvPixelFormat = CV_8UC1;
        break;
    // 其他像素格式的处理
    default:
        // 不支持的像素格式
        return;
}

mat.create(height, width, cvPixelFormat);

uint8_t* data = avFrame->data[0];
int linesize = avFrame->linesize[0];

for (int i = 0; i < height; i++) {
    memcpy(mat.ptr(i), data + i * linesize, width * mat.elemSize());
}