zcy
/
qt_rtmp_demo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

no message

master
zcy 2023-11-30 00:22:43 +08:00
parent 02696e243c
commit cf02c1cb4b
11 changed files with 99 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.cpp
View File

@ -41,7 +41,7 @@ int main(int argc, char *argv[])
MainWindow main;
main.setWindowTitle("직첵竟꿎桿묏야");
main.setFixedSize(1920,1080);
main.setFixedSize(1280,769);
main.show();
return app.exec();
}

5
mainwindow.cpp
View File

@ -97,13 +97,14 @@ void MainWindow::on_pushButton_2_clicked()
GUIDToAvFormat(mCamera->MediaType()));
}
mCamera->SetObserver(mVideoCoder);
if(mPusher == nullptr)
mPusher = new H264RtmpPuser();
// todo 根据返回结果判断是否推流
qDebug()<<"连接RTMP服务器"<<ui->lineEdit->text();
if (!mPusher->IfConnect()) {
const char* address = ui->lineEdit->text().toLocal8Bit().data();
qDebug()<<address;
if (0 == mPusher->RTMP264_Connect("rtmp://127.0.0.1:1939/live/1")) {
if (0 == mPusher->RTMP264_Connect("rtmp://hyrtc.net:2530/live/123")) {
ToastWidget::showTip("已经连接上RTMP服务器",this->parentWidget());
mVideoCoder->SetOberver(mPusher);
mPusher->StartPush();

2
media/AudioCapture.h
View File

@ -1,7 +1,7 @@
#ifndef __CAPTUREAUDIO_H__
#define __CAPTUREAUDIO_H__
#include "stdint.h"
#include "../third/portaudio/portaudio.h"
#include "../third/portaudio/include/portaudio.h"
#include <vector>
#include <string>
//Windows

14
media/RtmpPusher.cpp
View File

@ -208,6 +208,8 @@ int H264RtmpPuser::sortAndSendNal(uint8_t * data, int len)
uint8_t * nalhead = nullptr;
uint8_t * naltail = nullptr;
uint32_t size = 0;
static uint32_t time = 0;
if(0 == mStartTime){
mStartTime = RTMP_GetTime();
}
@ -221,11 +223,12 @@ int H264RtmpPuser::sortAndSendNal(uint8_t * data, int len)
&& ((data[i + 2] == 0x00) && (data[i + 3] == 0x01) || (data[i + 2] == 0x01))) {
if ((nalhead == nullptr) && (i == 0) ) {
if ((data[i + 3] == 0x01) && (data[i + 4] == 0x41)) { //p ÖĄÖą˝Óˇ˘
time += 100;
nalhead = data;
naltail = data + (len);
size = naltail - nalhead;
this->SendH264Packet(nalhead, size, 0, RTMP_GetTime() - mStartTime);
this->SendH264Packet(nalhead, size, 0, time - mStartTime);
return 0;
}
//sps ÖĄ˝řĐĐ˝â°ü
@ -241,21 +244,22 @@ int H264RtmpPuser::sortAndSendNal(uint8_t * data, int len)
else {
// i frame
if ((data[i + 2] == 0x01) && (data[i + 3] == 0x65)) {
time += 100;
naltail = data + i;
size = naltail - nalhead;
this->SendH264Packet(nalhead, size, 0, RTMP_GetTime() - mStartTime);
this->SendH264Packet(nalhead, size, 0, time - mStartTime);
nalhead = data + i;
naltail = data + (len);
size = naltail - nalhead;
this->SendH264Packet(nalhead, size, 0, RTMP_GetTime() - mStartTime);
this->SendH264Packet(nalhead, size, 0, time - mStartTime);
return 0;
}
//pps
if ((data[i + 3] == 0x01) && (data[i + 4] == 0x68)) { // sps or pps or sei
naltail = data + i;
size = naltail - nalhead;
this->SendH264Packet(nalhead, size, 0, RTMP_GetTime() - mStartTime);
this->SendH264Packet(nalhead, size, 0, time - mStartTime);
nalhead = data + i;
i += 3;
}//sps
@ -267,7 +271,7 @@ int H264RtmpPuser::sortAndSendNal(uint8_t * data, int len)
if ((data[i + 3] == 0x01) && (data[i + 4] == 0x06)) { // sps or pps or sei
naltail = data + i;
size = naltail - nalhead;
this->SendH264Packet(nalhead, size, 0, RTMP_GetTime() - mStartTime);
this->SendH264Packet(nalhead, size, 0, time - mStartTime);
nalhead = data + i;
i += 3;
}

22
media/VideoCoder.cpp
View File

@ -74,8 +74,8 @@ VideoCoder::VideoCoder(int width, int height, AVPixelFormat formt):
printf("Could not allocate video frame\n");
}
mFrame->format = mCodecCtx->pix_fmt;
mFrame->width = mCodecCtx->width/2;
mFrame->height = mCodecCtx->height/2;
mFrame->width = mCodecCtx->width;
mFrame->height = mCodecCtx->height;
mFrame->pts = 0;
int ret = av_image_alloc(mFrame->data, mFrame->linesize, mCodecCtx->width, mCodecCtx->height,
mCodecCtx->pix_fmt, 8);
@ -125,14 +125,17 @@ void VideoCoder::Encode(uint8_t * src, int size, enum AVPixelFormat format) {
mFrame->pts++;
int got_picture = 0;
//Encode
avcodec_encode_video2(mCodecCtx, &mAVPack, mFrame, &got_picture);
if (got_picture > 0) {
ret = avcodec_encode_video2(mCodecCtx, &mAVPack, mFrame, &got_picture);
if (got_picture > 0) {
fwrite(mAVPack.data, 1, mAVPack.size, p);
fflush(p);
if(nullptr != this->mObserver)
this->mObserver->OnGetCodeFrame(mAVPack.data, mAVPack.size);
}
//Debuger::Debug(L"Succeed to encode frame: %5d\tsize:%5d\n", 1, mAVPack.size);
fwrite(mAVPack.data, 1, mAVPack.size, p);
fflush(p);
Debuger::Debug(L"Succeed to encode frame: %5d\tsize:%5d\n", 1, mAVPack.size);
// 刷新coder，防止包挤压
av_packet_unref(&mAVPack);
av_freep(&pFrame[0]);
@ -208,7 +211,7 @@ int VideoCoder::forceYUV420P(uint8_t * src, int size,
//Show AVOption
//av_opt_show2(img_convert_ctx, stdout, AV_OPT_FLAG_VIDEO_PARAM, 0);
//Set Value
av_opt_set_int(img_convert_ctx, "sws_flags", SWS_BICUBIC | SWS_PRINT_INFO, 0);
av_opt_set_int(img_convert_ctx, "sws_flags", SWS_BICUBIC | SWS_PRINT_INFO, 0);
av_opt_set_int(img_convert_ctx, "srcw", mWidth, 0);
av_opt_set_int(img_convert_ctx, "srch", mHeight, 0);
av_opt_set_int(img_convert_ctx, "src_format", format, 0);
@ -252,7 +255,7 @@ int VideoCoder::forceYUV420P(uint8_t * src, int size,
memcpy(src_data[0], src, mWidth*mHeight * 3); //Packed
break;
}
case AV_PIX_FMT_RGB32: {
case AV_PIX_FMT_BGRA: {
memcpy(src_data[0], src, mWidth*mHeight *4); //Packed
break;
}
@ -273,7 +276,6 @@ int VideoCoder::forceYUV420P(uint8_t * src, int size,
*len = mDestWidth*mDestHeight + mDestWidth*mDestHeight / 2;
// source此时就不需要了，但是dst要在外面free
av_freep(&src_data[0]);
av_freep(&dst_data[0]);
sws_freeContext(img_convert_ctx);
return 0;

2
media/audiocaptureff.h
View File

@ -2,7 +2,7 @@
#define AUDIOCAPTUREFF_H
#include "stdint.h"
#include "../third/portaudio/portaudio.h"
#include "../third/portaudio/include/portaudio.h"
#include <vector>
#include <string>
//Windows

6
readme.md
View File

@ -9,7 +9,7 @@
3. opengl用于本地摄像头的渲染。
4. directshow的相关接口用于获取本地的摄像头数据。
5. 界面风格基于!qssWrpter库 https://gitee.com/290198252/qsswraper
6. librtmp需要openssl 1.10 以下的版本
6. librtmp需要openssl 1.10以下版本
### 界面:
@ -19,4 +19,6 @@
#### 发行版
- win32
- win32

3
utils/utils.cpp
View File

@ -20,5 +20,8 @@ AVPixelFormat GUIDToAvFormat(GUID mediatype){
if(IsEqualIID(MEDIASUBTYPE_RGB32,mediatype)){
return AV_PIX_FMT_BGRA;
}
if(IsEqualIID(MEDIASUBTYPE_ARGB32,mediatype)){
return AV_PIX_FMT_BGRA;
}
return AV_PIX_FMT_NONE;
}

30
utils/utils.h Normal file
View File

@ -0,0 +1,30 @@
#pragma once
#include <windows.h>
#include <string>
#include <vector>
#include <functional>
#include <dshow.h>
#include <windows.h>
#include "qedit.h"
#include <mutex>
#include <vector>
#include "guiddef.h"
extern "C"
{
#include "libavcodec/avcodec.h"
#include "libavformat/avformat.h"
#include "libavutil/avutil.h"
#include "libswscale/swscale.h"
#include "libavutil/opt.h"
#include "libavutil/imgutils.h"
}
using namespace std;
wstring char2wchar(const char* cchar);
AVPixelFormat GUIDToAvFormat(GUID mediatype);

38
yuvgl.pro
View File

@ -11,8 +11,8 @@ QT += multimedia
TARGET = yuvgl
INCLUDEPATH += $$[QT_INSTALL_HEADERS]/QtZlib "C:/Program Files/OpenSSL-Win64/include" \
$$PWD//third//ffmpeg//include
INCLUDEPATH += $$[QT_INSTALL_HEADERS]/QtZlib \
media/
include(G:\\project\\c++qt\\qsswraper\\qsswraper.pri)
@ -49,7 +49,6 @@ SOURCES += \
media/audiocaptureff.cpp \
media/screen_capture.cpp \
media/sps_decode.cpp \
utils/Base64.cpp \
utils/Debuger.cpp \
utils/utils.cpp
HEADERS += \
@ -57,7 +56,8 @@ HEADERS += \
librtmp/strncasecmp.h \
mainwindow.h \
cplaywidget.h \
media/screen_capture.h
media/screen_capture.h \
utils/utils.h
FORMS += \
@ -65,9 +65,6 @@ FORMS += \
mainwindow.ui
INCLUDEPATH += media/ \
C:\\Program Files\\OpenSSL-Win64\\include
contains(DEFINES, __MINGW32__){
message("mingw")
INCLUDEPATH += media/ inc/
@ -79,7 +76,8 @@ contains(DEFINES, __MINGW32__){
} else {
message("64-bit")
}
}else{
}else{
message("msvc")
@ -88,7 +86,7 @@ contains(DEFINES, __MINGW32__){
contains(QT_ARCH, i386) {
INCLUDEPATH += inc $$PWD/third/msvc32/fdk-aac/include \
INCLUDEPATH += inc $$PWD/third/fdk-aac/include \
$$PWD/third/msvc32/libx264/include \
$$PWD/third/msvc32/ffmpeg/include \
$$PWD/third/msvc32/openssl/include
@ -101,12 +99,32 @@ contains(DEFINES, __MINGW32__){
LIBS += libavfilter.a libavdevice.a libavcodec.a libpostproc.a \
libavformat.a libavutil.a \
libswresample.a libswscale.a fdk-aac.lib ws2_32.lib libeay32.lib ssleay32.lib \
shell32.lib gdi32.lib crypt32.lib User32.lib GDI32.lib Advapi32.lib zlibstaticd.lib Secur32.lib \
shell32.lib gdi32.lib crypt32.lib User32.lib GDI32.lib Advapi32.lib Secur32.lib \
Bcrypt.lib Kernel32.lib portaudio_x86.lib ole32.lib oleaut32.lib strmiids.lib libx264.lib d3d9.lib
}
else{
message("64-bit")
INCLUDEPATH += inc $$PWD/third/fdk-aac/include \
$$PWD/third/ffmpeg/include \
$$PWD/third/openssl_1_0_2/include \
$$PWD/third/libx264/include \
include
LIBS += -L$$PWD/third/openssl_1_0_2/lib
LIBS += -L$$PWD/third/ffmpeg/lib
LIBS += -L$$PWD/third/fdk-aac/lib
LIBS += -L$$PWD/third/portaudio/lib
LIBS += -L$$PWD/third/libx264/lib/x64
LIBS += avfilter.lib avdevice.lib libavcodec.dll.a postproc.lib \
libavformat.dll.a libavutil.dll.a \
libswresample.dll.a libswscale.dll.a fdk-aac.lib ws2_32.lib libeay32.lib ssleay32.lib \
shell32.lib gdi32.lib crypt32.lib User32.lib GDI32.lib Advapi32.lib Secur32.lib \
Bcrypt.lib Kernel32.lib portaudio_x64.lib ole32.lib oleaut32.lib strmiids.lib libx264.lib d3d9.lib
QMAKE_CXXFLAGS_RELEASE += -Zi
QMAKE_LFLAGS_RELEASE += /DEBUG /OPT:REF

11
yuvgl.pro.user
View File

@ -1,6 +1,6 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE QtCreatorProject>
<!-- Written by QtCreator 11.0.3, 2023-11-13T01:02:28. -->
<!-- Written by QtCreator 11.0.3, 2023-11-29T00:51:06. -->
<qtcreator>
<data>
<variable>EnvironmentId</variable>
@ -82,6 +82,9 @@
<valuelist type="QVariantList" key="ClangTools.SuppressedDiagnostics"/>
<value type="bool" key="ClangTools.UseGlobalSettings">true</value>
</valuemap>
<valuemap type="QVariantMap" key="CppEditor.QuickFix">
<value type="bool" key="UseGlobalSettings">true</value>
</valuemap>
</valuemap>
</data>
<data>
@ -238,12 +241,12 @@
<value type="int" key="PE.EnvironmentAspect.Base">2</value>
<valuelist type="QVariantList" key="PE.EnvironmentAspect.Changes"/>
<value type="bool" key="PE.EnvironmentAspect.PrintOnRun">false</value>
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName">yuvgl2</value>
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.Qt4RunConfiguration:D:/project/rtmp/yuvgl.pro</value>
<value type="QString" key="ProjectExplorer.RunConfiguration.BuildKey">D:/project/rtmp/yuvgl.pro</value>
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.Qt4RunConfiguration:G:/project/multimedia/client/rtmp_demo/yuvgl/yuvgl.pro</value>
<value type="QString" key="ProjectExplorer.RunConfiguration.BuildKey">G:/project/multimedia/client/rtmp_demo/yuvgl/yuvgl.pro</value>
<value type="bool" key="RunConfiguration.UseCppDebuggerAuto">true</value>
<value type="bool" key="RunConfiguration.UseLibrarySearchPath">true</value>
<value type="bool" key="RunConfiguration.UseQmlDebuggerAuto">true</value>
<value type="QString" key="RunConfiguration.WorkingDirectory.default">G:/project/multimedia/client/rtmp_demo/build-yuvgl-Desktop_Qt_5_15_2_MSVC2019_64bit-Debug</value>
</valuemap>
<value type="qlonglong" key="ProjectExplorer.Target.RunConfigurationCount">1</value>
</valuemap>