C#利用ffmpeg借助NVIDIA GPU实现实时RTSP硬解码+硬编码录制MP4

 

目录

说明

效果

项目 

代码

下载

说明

利用周杰的开源项目 Sdcb.FFmpeg

项目地址：https://github.com/sdcb/Sdcb.FFmpeg/

代码实现参考：https://github.com/sdcb/ffmpeg-muxing-video-demo

效果

C#利用ffmpeg借助NVIDIA GPU实现实时RTSP硬解码+硬编码录制MP4

项目 

代码

using Sdcb.FFmpeg.Codecs;
using Sdcb.FFmpeg.Formats;
using Sdcb.FFmpeg.Raw;
using Sdcb.FFmpeg.Toolboxs.Extensions;
using System;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using System.Windows.Forms;

namespace Sdcb.FFmpegDemo
{
    public partial class Form1 : Form
    {
        public Form1()
        {
            InitializeComponent();
        }

        CancellationTokenSource cts;

        ///


        /// 播放
        ///
        ///
        ///
        private void button1_Click(object sender, EventArgs e)
        {
            button1.Enabled = false;
            button2.Enabled = true;
            cts = new CancellationTokenSource();
            string rtsp_url = txtURL.Text;
            //输出视频文件的名称。
            string outputFile = "output.mp4";
            Task.Run(() => Recording(rtsp_url, outputFile, cts.Token));
        }

        void Recording(string url, string outputFile, CancellationToken cancellationToken)
        {
            //输出视频的帧率,帧率设置为每秒25帧
            AVRational frameRate = new AVRational(25, 1);
            
            //输出视频的比特率。
            long bitRate = 16 * 1024 * 1024; // 16M

            //从文件夹读取
            //该字符串指定了源图像的文件夹和命名模式。%03d部分表示图像以三位数字命名（例如，001.jpg，002.jpg等）。
            //string sourceFolder = @".\src\%03d.jpg";
            //FormatContext srcFc = FormatContext.OpenInputUrl(sourceFolder, options: new MediaDictionary
            //{
            //    ["framerate"] = frameRate.ToString()
            //});

            FormatContext srcFc = FormatContext.OpenInputUrl(url);
            srcFc.LoadStreamInfo();
            MediaStream srcVideo = srcFc.GetVideoStream();
            CodecParameters srcCodecParameters = srcVideo.Codecpar;

            CodecContext videoDecoder = new CodecContext(Codec.FindDecoderByName("h264_cuvid"));
            {
            };
            videoDecoder.FillParameters(srcCodecParameters);
            videoDecoder.Open();

            //var d=  Codec.FindDecoders(AVCodecID.H264).Select(x => x.Name);
            //var e = Codec.FindEncoders(AVCodecID.H264).Select(x => x.Name);

            FormatContext dstFc = FormatContext.AllocOutput(OutputFormat.Guess("mp4"));

            dstFc.VideoCodec = Codec.FindEncoderByName("h264_nvenc");
            MediaStream vstream = dstFc.NewStream(dstFc.VideoCodec);

            CodecContext vcodec = new CodecContext(dstFc.VideoCodec)
            {
                Width = srcCodecParameters.Width,
                Height = srcCodecParameters.Height,
                TimeBase = frameRate.Inverse(),
                PixelFormat = AVPixelFormat.Yuv420p,
                Flags = AV_CODEC_FLAG.GlobalHeader,
                BitRate = bitRate,
            };
            vcodec.Open(dstFc.VideoCodec);
            vstream.Codecpar.CopyFrom(vcodec);
            vstream.TimeBase = vcodec.TimeBase;

            IOContext io = IOContext.OpenWrite(outputFile);
            dstFc.Pb = io;
            dstFc.WriteHeader();

            // src        -- srcFc.ReadPackets()          -->
            // src Packet -- DecodePackets(videoDecoder)  -->
            // src Frame  -- ConvertFrames(vcodec)        -->
            // dst Frame  -- EncodeFrames(vcodec)         -->
            // dst Packet -- dstFc.InterleavedWritePacket -->
            // dst

            foreach (Packet packet in srcFc
                .ReadPackets().Where(x => x.StreamIndex == srcVideo.Index)
                .DecodePackets(videoDecoder)
                .ConvertFrames(vcodec)
                .EncodeFrames(vcodec)
                )
            {
                try
                {
                    packet.RescaleTimestamp(vcodec.TimeBase, vstream.TimeBase);
                    packet.StreamIndex = vstream.Index;
                    dstFc.InterleavedWritePacket(packet);

                    if (cancellationToken.IsCancellationRequested) break;
                }
                finally
                {
                    packet.Unref();
                }
            }
            dstFc.WriteTrailer();

            io.Dispose();
            vcodec.Dispose();
            dstFc.Dispose();
            videoDecoder.Dispose();
            srcFc.Dispose();

        }

        ///


        /// 停止
        ///
        ///
        ///
        private void button2_Click(object sender, EventArgs e)
        {
            button2.Enabled = false;
            button1.Enabled = true;
            cts.Cancel();
        }

        private void Form1_Load(object sender, EventArgs e)
        {
            button2.Enabled = false;
            button1.Enabled = true;
            Sdcb.FFmpeg.Utils.FFmpegLogger.LogWriter = (level, msg) => Console.WriteLine(msg);
        }
    }

}
 

  1. using Sdcb.FFmpeg.Codecs;
  2. using Sdcb.FFmpeg.Formats;
  3. using Sdcb.FFmpeg.Raw;
  4. using Sdcb.FFmpeg.Toolboxs.Extensions;
  5. using System;
  6. using System.Linq;
  7. using System.Threading;
  8. using System.Threading.Tasks;
  9. using System.Windows.Forms;
  10.  
  11. namespace Sdcb.FFmpegDemo
  12. {
  13.     public partial class Form1 : Form
  14.     {
  15.         public Form1()
  16.         {
  17.             InitializeComponent();
  18.         }
  19.  
  20.         CancellationTokenSource cts;
  21.  
  22.         /// <summary>
  23.         /// 播放
  24.         /// </summary>
  25.         /// <param name="sender"></param>
  26.         /// <param name="e"></param>
  27.         private void button1_Click(object sender, EventArgs e)
  28.         {
  29.             button1.Enabled = false;
  30.             button2.Enabled = true;
  31.             cts = new CancellationTokenSource();
  32.             string rtsp_url = txtURL.Text;
  33.             //输出视频文件的名称。
  34.             string outputFile = "output.mp4";
  35.             Task.Run(() => Recording(rtsp_url, outputFile, cts.Token));
  36.         }
  37.  
  38.         void Recording(string url, string outputFile, CancellationToken cancellationToken)
  39.         {
  40.             //输出视频的帧率,帧率设置为每秒25帧
  41.             AVRational frameRate = new AVRational(25, 1);
  42.             
  43.             //输出视频的比特率。
  44.             long bitRate = 16 * 1024 * 1024; // 16M
  45.  
  46.             //从文件夹读取
  47.             //该字符串指定了源图像的文件夹和命名模式。%03d部分表示图像以三位数字命名（例如，001.jpg，002.jpg等）。
  48.             //string sourceFolder = @".\src\%03d.jpg";
  49.             //FormatContext srcFc = FormatContext.OpenInputUrl(sourceFolder, options: new MediaDictionary
  50.             //{
  51.             //    ["framerate"] = frameRate.ToString()
  52.             //});
  53.  
  54.             FormatContext srcFc = FormatContext.OpenInputUrl(url);
  55.             srcFc.LoadStreamInfo();
  56.             MediaStream srcVideo = srcFc.GetVideoStream();
  57.             CodecParameters srcCodecParameters = srcVideo.Codecpar;
  58.  
  59.             CodecContext videoDecoder = new CodecContext(Codec.FindDecoderByName("h264_cuvid"));
  60.             {
  61.             };
  62.             videoDecoder.FillParameters(srcCodecParameters);
  63.             videoDecoder.Open();
  64.  
  65.             //var d=  Codec.FindDecoders(AVCodecID.H264).Select(x => x.Name);
  66.             //var e = Codec.FindEncoders(AVCodecID.H264).Select(x => x.Name);
  67.  
  68.             FormatContext dstFc = FormatContext.AllocOutput(OutputFormat.Guess("mp4"));
  69.  
  70.             dstFc.VideoCodec = Codec.FindEncoderByName("h264_nvenc");
  71.             MediaStream vstream = dstFc.NewStream(dstFc.VideoCodec);
  72.  
  73.             CodecContext vcodec = new CodecContext(dstFc.VideoCodec)
  74.             {
  75.                 Width = srcCodecParameters.Width,
  76.                 Height = srcCodecParameters.Height,
  77.                 TimeBase = frameRate.Inverse(),
  78.                 PixelFormat = AVPixelFormat.Yuv420p,
  79.                 Flags = AV_CODEC_FLAG.GlobalHeader,
  80.                 BitRate = bitRate,
  81.             };
  82.             vcodec.Open(dstFc.VideoCodec);
  83.             vstream.Codecpar.CopyFrom(vcodec);
  84.             vstream.TimeBase = vcodec.TimeBase;
  85.  
  86.             IOContext io = IOContext.OpenWrite(outputFile);
  87.             dstFc.Pb = io;
  88.             dstFc.WriteHeader();
  89.  
  90.             // src        -- srcFc.ReadPackets()          -->
  91.             // src Packet -- DecodePackets(videoDecoder)  -->
  92.             // src Frame  -- ConvertFrames(vcodec)        -->
  93.             // dst Frame  -- EncodeFrames(vcodec)         -->
  94.             // dst Packet -- dstFc.InterleavedWritePacket -->
  95.             // dst
  96.  
  97.             foreach (Packet packet in srcFc
  98.                 .ReadPackets().Where(x => x.StreamIndex == srcVideo.Index)
  99.                 .DecodePackets(videoDecoder)
  100.                 .ConvertFrames(vcodec)
  101.                 .EncodeFrames(vcodec)
  102.                 )
  103.             {
  104.                 try
  105.                 {
  106.                     packet.RescaleTimestamp(vcodec.TimeBase, vstream.TimeBase);
  107.                     packet.StreamIndex = vstream.Index;
  108.                     dstFc.InterleavedWritePacket(packet);
  109.  
  110.                     if (cancellationToken.IsCancellationRequested) break;
  111.                 }
  112.                 finally
  113.                 {
  114.                     packet.Unref();
  115.                 }
  116.             }
  117.             dstFc.WriteTrailer();
  118.  
  119.             io.Dispose();
  120.             vcodec.Dispose();
  121.             dstFc.Dispose();
  122.             videoDecoder.Dispose();
  123.             srcFc.Dispose();
  124.  
  125.         }
  126.  
  127.         /// <summary>
  128.         /// 停止
  129.         /// </summary>
  130.         /// <param name="sender"></param>
  131.         /// <param name="e"></param>
  132.         private void button2_Click(object sender, EventArgs e)
  133.         {
  134.             button2.Enabled = false;
  135.             button1.Enabled = true;
  136.             cts.Cancel();
  137.         }
  138.  
  139.         private void Form1_Load(object sender, EventArgs e)
  140.         {
  141.             button2.Enabled = false;
  142.             button1.Enabled = true;
  143.             Sdcb.FFmpeg.Utils.FFmpegLogger.LogWriter = (level, msg) => Console.WriteLine(msg);
  144.         }
  145.     }
  146.  
  147. }

下载

源码下载

天天代码码天天
微信公众号
.NET 人工智能实践
注：本文转载自blog.csdn.net的天天代码码天天的文章"https://lw112190.blog.csdn.net/article/details/141641143"。版权归原作者所有，此博客不拥有其著作权，亦不承担相应法律责任。如有侵权，请联系我们删除。
复制链接
相关推荐
发表评论

评论记录：

未查询到任何数据！