Use kcp as QoS module

2025-10-26 20:25:34 +08:00 · 2023-08-30 17:44:22 +08:00
parent a4cd77dcb0
commit 3c1f7973d0
79 changed files with 14442 additions and 3150 deletions
--- a/thirdparty/nvcodec/Samples/Utils/FFmpegStreamer.h
+++ b/thirdparty/nvcodec/Samples/Utils/FFmpegStreamer.h
@@ -0,0 +1,109 @@
+/*
+* Copyright 2017-2020 NVIDIA Corporation.  All rights reserved.
+*
+* Please refer to the NVIDIA end user license agreement (EULA) associated
+* with this source code for terms and conditions that govern your use of
+* this software. Any use, reproduction, disclosure, or distribution of
+* this software and related documentation outside the terms of the EULA
+* is strictly prohibited.
+*
+*/
+#pragma once
+
+#include <thread>
+#include <mutex>
+extern "C" {
+#include <libavformat/avformat.h>
+#include <libavutil/opt.h>
+#include <libswresample/swresample.h>
+};
+#include "Logger.h"
+
+extern simplelogger::Logger *logger;
+
+class FFmpegStreamer {
+private:
+    AVFormatContext *oc = NULL;
+    AVStream *vs = NULL;
+    int nFps = 0;
+
+public:
+    FFmpegStreamer(AVCodecID eCodecId, int nWidth, int nHeight, int nFps, const char *szInFilePath) : nFps(nFps) {
+        avformat_network_init();
+        oc = avformat_alloc_context();
+        if (!oc) {
+            LOG(ERROR) << "FFMPEG: avformat_alloc_context error";
+            return;
+        }
+
+        // Set format on oc
+        AVOutputFormat *fmt = av_guess_format("mpegts", NULL, NULL);
+        if (!fmt) {
+            LOG(ERROR) << "Invalid format";
+            return;
+        }
+        fmt->video_codec = eCodecId;
+
+        oc->oformat = fmt;
+        oc->url = av_strdup(szInFilePath);
+        LOG(INFO) << "Streaming destination: " << oc->url;
+
+        // Add video stream to oc
+        vs = avformat_new_stream(oc, NULL);
+        if (!vs) {
+            LOG(ERROR) << "FFMPEG: Could not alloc video stream";
+            return;
+        }
+        vs->id = 0;
+
+        // Set video parameters
+        AVCodecParameters *vpar = vs->codecpar;
+        vpar->codec_id = fmt->video_codec;
+        vpar->codec_type = AVMEDIA_TYPE_VIDEO;
+        vpar->width = nWidth;
+        vpar->height = nHeight;
+
+        // Everything is ready. Now open the output stream.
+        if (avio_open(&oc->pb, oc->url, AVIO_FLAG_WRITE) < 0) {
+            LOG(ERROR) << "FFMPEG: Could not open " << oc->url;
+            return ;
+        }
+
+        // Write the container header
+        if (avformat_write_header(oc, NULL)) {
+            LOG(ERROR) << "FFMPEG: avformat_write_header error!";
+            return;
+        }
+    }
+    ~FFmpegStreamer() {
+        if (oc) {
+            av_write_trailer(oc);
+            avio_close(oc->pb);
+            avformat_free_context(oc);
+        }
+    }
+
+    bool Stream(uint8_t *pData, int nBytes, int nPts) {
+        AVPacket pkt = {0};
+        av_init_packet(&pkt);
+        pkt.pts = av_rescale_q(nPts++, AVRational {1, nFps}, vs->time_base);
+        // No B-frames
+        pkt.dts = pkt.pts;
+        pkt.stream_index = vs->index;
+        pkt.data = pData;
+        pkt.size = nBytes;
+
+        if(!memcmp(pData, "\x00\x00\x00\x01\x67", 5)) {
+            pkt.flags |= AV_PKT_FLAG_KEY;
+        }
+
+        // Write the compressed frame into the output
+        int ret = av_write_frame(oc, &pkt);
+        av_write_frame(oc, NULL);
+        if (ret < 0) {
+            LOG(ERROR) << "FFMPEG: Error while writing video frame";
+        }
+
+        return true;
+    }
+};