嵌入式linux驱动 视频

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/videodev2.h>

static int __init my_video_init(void) {
    printk(KERN_INFO "Loading video driver\n");
    return 0;
}

static void __exit my_video_exit(void) {
    printk(KERN_INFO "Unloading video driver\n");
}

module_init(my_video_init);
module_exit(my_video_exit);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Your Name");
MODULE_DESCRIPTION("A simple video driver");

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libavutil/imgutils.h>
#include <libswscale/swscale.h>

int main(int argc, char *argv[]) {
    AVFormatContext *fmt_ctx = NULL;
    AVCodecContext *codec_ctx = NULL;
    AVPacket pkt;
    AVFrame *frame, *frame_rgb;
    struct SwsContext *sws_ctx;

    av_register_all();
    avformat_open_input(&fmt_ctx, argv[1], NULL, NULL);
    avformat_find_stream_info(fmt_ctx, NULL);

    int video_stream_idx = -1;
    for (int i = 0; i < fmt_ctx->nb_streams; i++) {
        if (fmt_ctx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
            video_stream_idx = i;
            break;
        }
    }

    AVCodec *codec = avcodec_find_decoder(fmt_ctx->streams[video_stream_idx]->codecpar->codec_id);
    codec_ctx = avcodec_alloc_context3(codec);
    avcodec_parameters_to_context(codec_ctx, fmt_ctx->streams[video_stream_idx]->codecpar);
    avcodec_open2(codec_ctx, codec, NULL);

    frame = av_frame_alloc();
    frame_rgb = av_frame_alloc();
    sws_ctx = sws_getContext(codec_ctx->width, codec_ctx->height, codec_ctx->pix_fmt,
                            codec_ctx->width, codec_ctx->height, AV_PIX_FMT_RGB24,
                            SWS_BILINEAR, NULL, NULL, NULL);

    while (av_read_frame(fmt_ctx, &pkt) >= 0) {
        if (pkt.stream_index == video_stream_idx) {
            avcodec_send_packet(codec_ctx, &pkt);
            while (avcodec_receive_frame(codec_ctx, frame) == 0) {
                sws_scale(sws_ctx, (const uint8_t * const *)frame->data, frame->linesize,
                          0, codec_ctx->height, frame_rgb->data, frame_rgb->linesize);
                // 处理RGB帧数据
            }
        }
        av_packet_unref(&pkt);
    }

    av_frame_free(&frame);
    av_frame_free(&frame_rgb);
    sws_freeContext(sws_ctx);
    avcodec_close(codec_ctx);
    avformat_close_input(&fmt_ctx);

    return 0;
}