Hu Chunming / FFNvDecoder

Commit 00b0fbdb28050ddecec6db2f21a697ec6373e30a

Authored by Hu Chunming
1 parent d248da62

编译nvdec

Inline Side-by-side
Showing 29 changed files with 1103 additions and 646 deletions
.vscode/launch.json
  View file @00b0fbd
@@ -19,6 +19,24 @@ @@ -19,6 +19,24 @@
19 "ignoreFailures": true 19 "ignoreFailures": true
20 } 20 }
21 ] 21 ]
  22 + },{
  23 + "name": "nvdec",
  24 + "type": "cppdbg",
  25 + "request": "launch",
  26 + "program": "${workspaceFolder}/bin/lib/demo",
  27 + "args": ["/mnt/data/cmhu/data/caishenkezhan.mp4","0", "0"],
  28 + "stopAtEntry": false,
  29 + "cwd": "${workspaceFolder}/bin/lib",
  30 + "environment": [],
  31 + "externalConsole": false,
  32 + "MIMode": "gdb",
  33 + "setupCommands": [
  34 + {
  35 + "description": "Enable pretty-printing for gdb",
  36 + "text": "-enable-pretty-printing",
  37 + "ignoreFailures": true
  38 + }
  39 + ]
22 } 40 }
23 ] 41 ]
24 } 42 }
25 \ No newline at end of file 43 \ No newline at end of file
src/Makefile
  View file @00b0fbd
@@ -7,7 +7,7 @@ BUILD_DIR:=$(TOP_DIR)/build @@ -7,7 +7,7 @@ BUILD_DIR:=$(TOP_DIR)/build
7 BIN_DIR:=$(BUILD_DIR)/bin 7 BIN_DIR:=$(BUILD_DIR)/bin
8 export BUILD_DIR 8 export BUILD_DIR
9 9
10 -PROJECT_ROOT= /home/huchunming/FFNvDecoder 10 +PROJECT_ROOT= /mnt/data/cmhu/FFNvDecoder
11 11
12 DEPEND_DIR = $(PROJECT_ROOT)/bin 12 DEPEND_DIR = $(PROJECT_ROOT)/bin
13 SRC_ROOT = $(PROJECT_ROOT)/src 13 SRC_ROOT = $(PROJECT_ROOT)/src
@@ -29,11 +29,12 @@ LDFLAGS:= @@ -29,11 +29,12 @@ LDFLAGS:=
29 LIBS:= -L $(SPDLOG_ROOT)/lib -l:libspdlog.a \ 29 LIBS:= -L $(SPDLOG_ROOT)/lib -l:libspdlog.a \
30 30
31 # 各个模块 31 # 各个模块
32 -MODULES:= dvpp interface demo 32 +MODULES:= nvdecoder gb28181 interface demo
33 33
34 # 各个模块对应的库 34 # 各个模块对应的库
35 -MODULE_LIBS:=$(BUILD_DIR)/dvpp/lib/libdvpp.a\  
36 - $(BUILD_DIR)/interface/lib/interface.a\ 35 +# MODULE_LIBS:=$(BUILD_DIR)/nvdecoder/lib/nvdecoder.a\
  36 +# $(BUILD_DIR)/nvdecoder/lib/gb28181.a\
  37 +# $(BUILD_DIR)/interface/lib/interface.a\
37 38
38 # 最终目标文件 39 # 最终目标文件
39 TARGET:=$(BIN_DIR)/test 40 TARGET:=$(BIN_DIR)/test
@@ -45,7 +46,7 @@ all:$(TARGET) @@ -45,7 +46,7 @@ all:$(TARGET)
45 # 最终目标依赖关系 46 # 最终目标依赖关系
46 $(TARGET):FORCE | $(BIN_DIR) 47 $(TARGET):FORCE | $(BIN_DIR)
47 @for n in $(MODULES); do make -s -f $(TOP_DIR)/$$n/Makefile MODULE=$$n || exit "$$?"; done 48 @for n in $(MODULES); do make -s -f $(TOP_DIR)/$$n/Makefile MODULE=$$n || exit "$$?"; done
48 - @echo -e "\e[32m""Linking executable $(TARGET)""\e[0m" 49 +# @echo -e "\e[32m""Linking executable $(TARGET)""\e[0m"
49 #@$(LD) $(LDFLAGS) -o $@ $(MODULE_LIBS) $(LIBS) 50 #@$(LD) $(LDFLAGS) -o $@ $(MODULE_LIBS) $(LIBS)
50 51
51 # 若没有bin目录则自动生成 52 # 若没有bin目录则自动生成
src/Makefile.bak
  View file @00b0fbd
@@ -8,12 +8,10 @@ PROJECT_ROOT= /home/huchunming/FFNvDecoder @@ -8,12 +8,10 @@ PROJECT_ROOT= /home/huchunming/FFNvDecoder
8 8
9 DEPEND_DIR = $(PROJECT_ROOT)/bin 9 DEPEND_DIR = $(PROJECT_ROOT)/bin
10 SRC_ROOT = $(PROJECT_ROOT)/src 10 SRC_ROOT = $(PROJECT_ROOT)/src
11 -THIRDPARTY_ROOT = $(PROJECT_ROOT)/3rdparty  
12 -  
13 11
14 TARGET= $(DEPEND_DIR)/lib/test 12 TARGET= $(DEPEND_DIR)/lib/test
15 13
16 - 14 +THIRDPARTY_ROOT = $(PROJECT_ROOT)/3rdparty
17 SPDLOG_ROOT = $(THIRDPARTY_ROOT)/spdlog-1.9.2/release 15 SPDLOG_ROOT = $(THIRDPARTY_ROOT)/spdlog-1.9.2/release
18 JRTP_ROOT = $(THIRDPARTY_ROOT)/jrtp_export 16 JRTP_ROOT = $(THIRDPARTY_ROOT)/jrtp_export
19 17
src/Makefile.bak0308 deleted
View file @d248da6
1 -# 项目根目录  
2 -TOP_DIR:=$(patsubst %/, %, $(dir $(abspath $(lastword $(MAKEFILE_LIST)))))  
3 -export TOP_DIR  
4 -  
5 -# 各项目录  
6 -BUILD_DIR:=$(TOP_DIR)/build  
7 -BIN_DIR:=$(BUILD_DIR)/bin  
8 -export BUILD_DIR  
9 -  
10 -PROJECT_ROOT= /home/huchunming/FFNvDecoder  
11 -  
12 -DEPEND_DIR = $(PROJECT_ROOT)/bin  
13 -SRC_ROOT = $(PROJECT_ROOT)/src  
14 -THIRDPARTY_ROOT = $(PROJECT_ROOT)/3rdparty  
15 -SPDLOG_ROOT = $(THIRDPARTY_ROOT)/spdlog-1.9.2/release  
16 -  
17 -#编译器  
18 -CXX:=g++  
19 -CXXFLAGS:=-std=c++11 -Wall -Wextra -Og -g  
20 -INCS:=-I $(TOP_DIR) \  
21 - -I $(SPDLOG_ROOT)/include \  
22 -  
23 -MACROS:=  
24 -export CXX CXXFLAGS INCS MACROS  
25 -  
26 -# 链接器  
27 -LD:=g++  
28 -LDFLAGS:=  
29 -LIBS:= -L $(SPDLOG_ROOT)/lib -l:libspdlog.a \  
30 -  
31 -# 各个模块  
32 -MODULES:= dvpp interface  
33 -  
34 -# 各个模块对应的库  
35 -MODULE_LIBS:=$(BUILD_DIR)/dvpp/lib/libdvpp.a\  
36 - $(BUILD_DIR)/interface/lib/interface.a\  
37 -  
38 -# 最终目标文件  
39 -TARGET:=$(BIN_DIR)/test  
40 -  
41 -# 默认最终目标  
42 -.PHONY:all  
43 -all:$(TARGET)  
44 -  
45 -# 最终目标依赖关系  
46 -$(TARGET):FORCE | $(BIN_DIR)  
47 - @for n in $(MODULES); do make -s -f $(TOP_DIR)/$$n/Makefile MODULE=$$n || exit "$$?"; done  
48 - @echo -e "\e[32m""Linking executable $(TARGET)""\e[0m"  
49 -#@$(LD) $(LDFLAGS) -o $@ $(MODULE_LIBS) $(LIBS)  
50 -  
51 -# 若没有bin目录则自动生成  
52 -$(BIN_DIR):  
53 - @mkdir -p $@  
54 -  
55 -# 强制执行命令  
56 -.PHONY:FORCE  
57 -FORCE:  
58 -  
59 -# make clean直接删除整个build目录  
60 -.PHONY:clean  
61 -clean:  
62 - @rm -rf $(BUILD_DIR)  
src/demo/Makefile
  View file @00b0fbd
1 XX = g++ 1 XX = g++
2 2
3 3
4 -PROJECT_ROOT= /home/huchunming/FFNvDecoder 4 +PROJECT_ROOT= /mnt/data/cmhu/FFNvDecoder
  5 +
  6 +CUDA_ROOT = /usr/local/cuda-11.1
5 7
6 DEPEND_DIR = $(PROJECT_ROOT)/bin 8 DEPEND_DIR = $(PROJECT_ROOT)/bin
7 -SRC_ROOT = $(PROJECT_ROOT)/src  
8 THIRDPARTY_ROOT = $(PROJECT_ROOT)/3rdparty 9 THIRDPARTY_ROOT = $(PROJECT_ROOT)/3rdparty
9 -  
10 -  
11 -TARGET= /home/huchunming/FFNvDecoder/src/build/bin/demo  
12 -  
13 -  
14 SPDLOG_ROOT = $(THIRDPARTY_ROOT)/spdlog-1.9.2/release 10 SPDLOG_ROOT = $(THIRDPARTY_ROOT)/spdlog-1.9.2/release
15 JRTP_ROOT = $(THIRDPARTY_ROOT)/jrtp_export 11 JRTP_ROOT = $(THIRDPARTY_ROOT)/jrtp_export
16 12
  13 +SRC_ROOT = $(PROJECT_ROOT)/src
17 14
18 -include_dir=-I/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/runtime/include  
19 -lib_dir=-L/usr/lib \  
20 - -L/usr/local/lib \  
21 - -L/usr/local/Ascend/driver/lib64 \  
22 - -L/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/atc/lib64\  
23 - -L/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/runtime/lib64 \  
24 - -L/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/runtime/lib64/stub \  
25 - -L/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/lib64 \  
26 - -L/usr/local/Ascend/driver/lib64/driver  
27 -  
28 -lib=-lacl_dvpp -lascendcl -lmmpa -lglog -lgflags -lpthread -lz -lacl_dvpp_mpi -lruntime -lascendalog -lc_sec -lmsprofiler -lgert -lge_executor -lge_common \  
29 - -lgraph -lascend_protobuf -lprofapi -lerror_manager -lexe_graph -lregister -lplatform 15 +TARGET= $(PROJECT_ROOT)/bin/lib/demo
30 16
31 DEFS = -DENABLE_DVPP_INTERFACE 17 DEFS = -DENABLE_DVPP_INTERFACE
32 18
33 INCLUDE= -I $(SRC_ROOT)/interface \ 19 INCLUDE= -I $(SRC_ROOT)/interface \
34 -I $(SRC_ROOT)/dvpp \ 20 -I $(SRC_ROOT)/dvpp \
  21 + -I $(SRC_ROOT)/gb28181 \
  22 + -I $(SRC_ROOT)/nvdecoder \
  23 + -I $(DEPEND_DIR)/include \
  24 + -I $(CUDA_ROOT)/include \
  25 + -I $(TOP_DIR)/common/inc \
  26 + -I $(TOP_DIR)/common/UtilNPP \
  27 + -I $(TOP_DIR)/ \
  28 + -I $(SPDLOG_ROOT)/include \
  29 + -I $(JRTP_ROOT)/jrtplib/include/jrtplib3 \
  30 + -I $(JRTP_ROOT)/jthread/include/jthread
  31 +
  32 +LIBSPATH= -L $(DEPEND_DIR)/lib -lavformat -lavcodec -lswscale -lavutil -lavfilter -lswresample -lavdevice \
  33 + -L $(CUDA_ROOT)/lib64 -lcuda -lcudart -lnvcuvid -lcurand -lcublas -lnvjpeg \
  34 + -L $(SPDLOG_ROOT) -l:libspdlog.a \
  35 + -L $(JRTP_ROOT)/jthread/lib -l:libjthread.a \
  36 + -L $(JRTP_ROOT)/jrtplib/lib -l:libjrtp.a
  37 +
  38 +
  39 +# include_dir=-I/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/runtime/include
  40 +
  41 +# lib_dir=-L/usr/lib \
  42 +# -L/usr/local/lib \
  43 +# -L/usr/local/Ascend/driver/lib64 \
  44 +# -L/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/atc/lib64\
  45 +# -L/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/runtime/lib64 \
  46 +# -L/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/runtime/lib64/stub \
  47 +# -L/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/lib64 \
  48 +# -L/usr/local/Ascend/driver/lib64/driver
  49 +
  50 +# lib=-lacl_dvpp -lascendcl -lmmpa -lglog -lgflags -lpthread -lz -lacl_dvpp_mpi -lruntime -lascendalog -lc_sec -lmsprofiler -lgert -lge_executor -lge_common \
  51 +# -lgraph -lascend_protobuf -lprofapi -lerror_manager -lexe_graph -lregister -lplatform
  52 +# LIBS= -lavformat -lavcodec -lswscale -lavutil -lavfilter -lswresample -lavdevice
  53 +# CXXFLAGS= -g -O0 -fPIC $(INCLUDE) $(include_dir) $(DEFS) -lpthread -lrt -lz -fexceptions -std=c++11 -fvisibility=hidden -Wl,-Bsymbolic -ldl
35 54
36 -LIBSPATH=  
37 -  
38 -LIBS= -lavformat -lavcodec -lswscale -lavutil -lavfilter -lswresample -lavdevice  
39 -  
40 -CXXFLAGS= -g -O0 -fPIC $(INCLUDE) $(include_dir) $(DEFS) -lpthread -lrt -lz -fexceptions -std=c++11 -fvisibility=hidden -Wl,-Bsymbolic -ldl 55 +CXXFLAGS= -g -O0 -fPIC $(INCLUDE) $(DEFS) -lpthread -lrt -lz -fexceptions -std=c++11 -fvisibility=hidden -Wl,-Bsymbolic -ldl
41 56
42 SRCS:=$(wildcard $(SRC_ROOT)/demo/*.cpp) 57 SRCS:=$(wildcard $(SRC_ROOT)/demo/*.cpp)
43 OBJS = $(patsubst %.cpp, %.o, $(notdir $(SRCS))) 58 OBJS = $(patsubst %.cpp, %.o, $(notdir $(SRCS)))
44 59
45 -OBJ_ROOT = /home/huchunming/FFNvDecoder/src/build  
46 -DVPP_SRCS:=$(wildcard $(OBJ_ROOT)/dvpp/obj/*.o) 60 +OBJ_ROOT = $(PROJECT_ROOT)/src/build
  61 +# DVPP_SRCS:=$(wildcard $(OBJ_ROOT)/dvpp/obj/*.o)
47 INTEFACE_SRCS:=$(wildcard $(OBJ_ROOT)/interface/obj/*.o) 62 INTEFACE_SRCS:=$(wildcard $(OBJ_ROOT)/interface/obj/*.o)
  63 +NVDECODER_SRCS:=$(wildcard $(OBJ_ROOT)/nvdecoder/obj/*.o)
  64 +GB28181_SRCS:=$(wildcard $(OBJ_ROOT)/gb28181/obj/*.o)
48 65
49 66
50 -$(TARGET):$(OBJS) $(INTEFACE_SRCS) $(DVPP_SRCS) 67 +$(TARGET):$(OBJS) $(INTEFACE_SRCS) $(NVDECODER_SRCS) $(GB28181_SRCS)
51 rm -f $(TARGET) 68 rm -f $(TARGET)
52 - @echo -e "\e[33m""Building object $@""\e[0m"  
53 - $(XX) -o $@ $^ $(CXXFLAGS) $(LIBS) $(lib_dir) $(lib) -Wwrite-strings 69 +# @echo -e "\e[33m""Building object $@""\e[0m"
  70 +# $(XX) -o $@ $^ $(CXXFLAGS) $(LIBS) $(lib_dir) $(lib) -Wwrite-strings
  71 + $(XX) -o $@ $^ $(CXXFLAGS) $(LIBSPATH) -Wwrite-strings
54 rm -f *.o 72 rm -f *.o
55 73
56 %.o:$(SRC_ROOT)/demo/%.cpp 74 %.o:$(SRC_ROOT)/demo/%.cpp
57 - $(XX) $(CFLAGS) -c $< 75 + $(XX) $(CXXFLAGS) -c $<
58 76
59 77
60 clean: 78 clean:
src/demo/Makefile.BK0308 deleted
View file @d248da6
1 -XX = g++  
2 -  
3 -  
4 -PROJECT_ROOT= /home/huchunming/FFNvDecoder  
5 -  
6 -DEPEND_DIR = $(PROJECT_ROOT)/bin  
7 -SRC_ROOT = $(PROJECT_ROOT)/src  
8 -THIRDPARTY_ROOT = $(PROJECT_ROOT)/3rdparty  
9 -  
10 -  
11 -TARGET= /home/huchunming/FFNvDecoder/src/build/bin/test  
12 -  
13 -  
14 -SPDLOG_ROOT = $(THIRDPARTY_ROOT)/spdlog-1.9.2/release  
15 -JRTP_ROOT = $(THIRDPARTY_ROOT)/jrtp_export  
16 -  
17 -  
18 -INCLUDE= -I $(SRC_ROOT)/interface \  
19 -  
20 -LIBSPATH= -L /home/huchunming/FFNvDecoder/src/build/interface/lib -l:interface.a \  
21 - -L /home/huchunming/FFNvDecoder/src/build/dvpp/lib -l:libdvpp.a \  
22 -  
23 -  
24 -LIBS= -lavformat -lavcodec -lswscale -lavutil -lavfilter -lswresample -lavdevice  
25 -  
26 -CFLAGS= -g -fPIC -O0 $(INCLUDE) -pthread -lrt -lz -std=c++11 -fvisibility=hidden -Wl,-Bsymbolic -ldl  
27 - # -DUNICODE -D_UNICODE  
28 -  
29 -SRCS:=$(wildcard $(SRC_ROOT)/demo/*.cpp)  
30 -OBJS = $(patsubst %.cpp, %.o, $(notdir $(SRCS)))  
31 -  
32 -  
33 -$(TARGET):$(OBJS) $(CU_OBJS)  
34 - rm -f $(TARGET)  
35 - $(XX) -o $@ $^ $(CFLAGS) $(LIBSPATH) $(LIBS) -Wwrite-strings  
36 - rm -f *.o  
37 -  
38 -%.o:$(SRC_ROOT)/demo/%.cpp  
39 - $(XX) $(CFLAGS) -c $<  
40 -  
41 -  
42 -clean:  
43 - rm -f *.o $(TARGET)  
44 \ No newline at end of file 0 \ No newline at end of file
src/demo/Makefile.dvpp 0 → 100644
  View file @00b0fbd
  1 +XX = g++
  2 +
  3 +
  4 +PROJECT_ROOT= /mnt/data/cmhu/FFNvDecoder
  5 +
  6 +SRC_ROOT = $(PROJECT_ROOT)/src
  7 +
  8 +TARGET= $(PROJECT_ROOT)/src/build/bin/demo
  9 +
  10 +DEFS = -DENABLE_DVPP_INTERFACE
  11 +
  12 +
  13 +include_dir=-I/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/runtime/include
  14 +
  15 +lib_dir=-L/usr/lib \
  16 + -L/usr/local/lib \
  17 + -L/usr/local/Ascend/driver/lib64 \
  18 + -L/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/atc/lib64\
  19 + -L/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/runtime/lib64 \
  20 + -L/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/runtime/lib64/stub \
  21 + -L/usr/local/Ascend/ascend-toolkit/6.3.RC1.alpha001/lib64 \
  22 + -L/usr/local/Ascend/driver/lib64/driver
  23 +
  24 +lib=-lacl_dvpp -lascendcl -lmmpa -lglog -lgflags -lpthread -lz -lacl_dvpp_mpi -lruntime -lascendalog -lc_sec -lmsprofiler -lgert -lge_executor -lge_common \
  25 + -lgraph -lascend_protobuf -lprofapi -lerror_manager -lexe_graph -lregister -lplatform
  26 +LIBS= -lavformat -lavcodec -lswscale -lavutil -lavfilter -lswresample -lavdevice
  27 +CXXFLAGS= -g -O0 -fPIC $(INCLUDE) $(include_dir) $(DEFS) -lpthread -lrt -lz -fexceptions -std=c++11 -fvisibility=hidden -Wl,-Bsymbolic -ldl
  28 +
  29 +SRCS:=$(wildcard $(SRC_ROOT)/demo/*.cpp)
  30 +OBJS = $(patsubst %.cpp, %.o, $(notdir $(SRCS)))
  31 +
  32 +OBJ_ROOT = $(PROJECT_ROOT)/src/build
  33 +DVPP_SRCS:=$(wildcard $(OBJ_ROOT)/dvpp/obj/*.o)
  34 +INTEFACE_SRCS:=$(wildcard $(OBJ_ROOT)/interface/obj/*.o)
  35 +
  36 +
  37 +$(TARGET):$(OBJS) $(INTEFACE_SRCS) $(NVDECODER_SRCS) $(GB28181_SRCS)
  38 + rm -f $(TARGET)
  39 + @echo -e "\e[33m""Building object $@""\e[0m"
  40 + $(XX) -o $@ $^ $(CXXFLAGS) $(LIBS) $(lib_dir) $(lib) -Wwrite-strings
  41 + rm -f *.o
  42 +
  43 +%.o:$(SRC_ROOT)/demo/%.cpp
  44 + $(XX) $(CXXFLAGS) -c $<
  45 +
  46 +
  47 +clean:
  48 + rm -f *.o $(TARGET)
0 \ No newline at end of file 49 \ No newline at end of file
src/demo/main_dvpp.cpp renamed to src/demo/main_dvpp.cpp1
  View file @00b0fbd
@@ -276,17 +276,6 @@ int main(int argc, char* argv[]){ @@ -276,17 +276,6 @@ int main(int argc, char* argv[]){
276 FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance(); 276 FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance();
277 int i = 0; 277 int i = 0;
278 278
279 - createDvppDecoder(i, gpuid, 0);  
280 - i++;  
281 - createDvppDecoder(i, gpuid, 0);  
282 - i++;  
283 - createDvppDecoder(i, gpuid, 0);  
284 - i++;  
285 -  
286 - for(;i<30;i++){  
287 - createDvppDecoder(i, gpuid, 0);  
288 - }  
289 -  
290 while (true) 279 while (true)
291 { 280 {
292 int ch = getchar(); 281 int ch = getchar();
src/demo/main_nvdec.cpp 0 → 100644
  View file @00b0fbd
  1 +#include "FFNvDecoderManager.h"
  2 +#include <iostream>
  3 +
  4 +#include "cuda_kernels.h"
  5 +
  6 +#include "NvJpegEncoder.h"
  7 +
  8 +#include <pthread.h>
  9 +#include <thread>
  10 +
  11 +#include <chrono>
  12 +
  13 +#include <unistd.h>
  14 +
  15 +
  16 +#ifdef _WIN32
  17 +#include "Winsock2.h"
  18 +#pragma comment(lib, "ws2_32.lib")
  19 +#endif
  20 +
  21 +#ifdef __linux__
  22 +#include "arpa/inet.h"
  23 +#endif
  24 +
  25 +#include "utiltools.hpp"
  26 +
  27 +#define MIN_RTP_PORT 10000
  28 +#define MAX_RTP_PORT 60000
  29 +
  30 +// ȡ MIN_RTP_PORT(10000)~MAX_RTP_PORT(60000)֮�������˿�(ż���������������˿ڿ���)
  31 +int allocRtpPort() {
  32 +
  33 + static int s_rtpPort = MIN_RTP_PORT;
  34 + if (MIN_RTP_PORT == s_rtpPort)
  35 + {
  36 + srand((unsigned int)time(NULL));
  37 + s_rtpPort = MIN_RTP_PORT + (rand() % MIN_RTP_PORT);
  38 + }
  39 +
  40 + if (s_rtpPort % 2)
  41 + ++s_rtpPort;
  42 +
  43 + while (true)
  44 + {
  45 + s_rtpPort += 2;
  46 + s_rtpPort = s_rtpPort >= MAX_RTP_PORT ? MIN_RTP_PORT : s_rtpPort;
  47 +
  48 + int i = 0;
  49 + for (; i < 2; i++)
  50 + {
  51 + sockaddr_in sRecvAddr;
  52 + int s = socket(AF_INET, SOCK_DGRAM, 0);
  53 +
  54 + sRecvAddr.sin_family = AF_INET;
  55 + sRecvAddr.sin_addr.s_addr = htonl(INADDR_ANY);
  56 + sRecvAddr.sin_port = htons(s_rtpPort + i);
  57 +
  58 + int nResult = bind(s, (sockaddr *)&sRecvAddr, sizeof(sRecvAddr));
  59 + if (nResult != 0)
  60 + {
  61 + break;
  62 + }
  63 +
  64 + nResult = close(s);
  65 + if (nResult != 0)
  66 + {
  67 + printf("closesocket failed:%d\n", nResult);
  68 + break;
  69 + }
  70 + }
  71 +
  72 + if (i == 2)
  73 + break;
  74 + }
  75 +
  76 + return s_rtpPort;
  77 +}
  78 +
  79 +
  80 +
  81 +
  82 +
  83 +unsigned char *pHwRgb[2] = {nullptr, nullptr};
  84 +
  85 +int sum1 = 0;
  86 +int sum2 = 0;
  87 +
  88 +cudaStream_t stream[2];
  89 +
  90 +string data_home = "/mnt/data/cmhu/tmp/";
  91 +
  92 +
  93 +#define checkCudaErrors(S) do {CUresult status; \
  94 + status = S; \
  95 + if (status != CUDA_SUCCESS ) std::cout << __LINE__ <<" checkCudaErrors - status = " << status << std::endl; \
  96 + } while (false)
  97 +
  98 +
  99 +static void gpu_helper(int gpuid)
  100 +{
  101 + cudaSetDevice(gpuid);
  102 +
  103 + // int *dn;
  104 + // cudaMalloc((void **)&dn, 1 * sizeof(int));
  105 +
  106 + size_t free_byte;
  107 + size_t total_byte;
  108 +
  109 + CUresult cuda_status = cuMemGetInfo(&free_byte, &total_byte);
  110 +
  111 + const char *pStr = nullptr;
  112 + if (CUDA_SUCCESS != cuda_status) {
  113 + cuGetErrorString(cuda_status, &pStr);
  114 + printf("Error: cudaMemGetInfo fails, %s \n", pStr);
  115 + return;
  116 + }
  117 +
  118 + double free_db = (double)free_byte;
  119 + double total_db = (double)total_byte;
  120 + double used_db_1 = (total_db - free_db) / 1024.0 / 1024.0;
  121 +
  122 + std::cout <<"显存已使用 " << used_db_1 << " MB\n";
  123 +
  124 + // cudaFree(dn);
  125 +}
  126 +
  127 +int CheckCUDAProperty( int devId )
  128 +{
  129 + cuInit(0);
  130 +
  131 + CUdevice dev = devId;
  132 + size_t memSize = 0;
  133 + char devName[256] = {0};
  134 + int major = 0, minor = 0;
  135 + CUresult rlt = CUDA_SUCCESS;
  136 +
  137 + rlt = cuDeviceGetAttribute(&major, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR, dev);
  138 + checkCudaErrors( rlt );
  139 +
  140 + rlt = cuDeviceGetAttribute(&minor, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR, dev);
  141 + checkCudaErrors( rlt );
  142 +
  143 + rlt = cuDeviceGetName( devName, sizeof( devName ), dev );
  144 + checkCudaErrors( rlt );
  145 +
  146 + printf( "Using GPU Device %d: %s has SM %d.%d compute capability\n",
  147 + dev, devName, major, minor );
  148 +
  149 + rlt = cuDeviceTotalMem( &memSize, dev );
  150 + checkCudaErrors( rlt );
  151 +
  152 + printf( "Total amount of global memory: %4.4f MB\n",
  153 + (float)memSize / ( 1024 * 1024 ) );
  154 +
  155 + return 0;
  156 +}
  157 +
  158 +/**
  159 + * 注意： gpuFrame 在解码器设置的显卡上，后续操作要十分注意这一点，尤其是多线程情况
  160 + * */
  161 +static long lastpts = 0;
  162 +void postDecoded(const void * userPtr, DeviceRgbMemory * gpuFrame){
  163 + AbstractDecoder* decoder = (AbstractDecoder*)userPtr;
  164 + if (decoder!= nullptr)
  165 + {
  166 + long curpts = UtilTools::get_cur_time_ms();
  167 + cout << decoder->getName() << " " << gpuFrame->getWidth() << "x" << gpuFrame->getHeight() << " " << curpts - lastpts << endl;
  168 + lastpts = curpts;
  169 + delete gpuFrame;
  170 + gpuFrame = nullptr;
  171 +
  172 + // const char* gpu_pixfmt = av_get_pix_fmt_name((AVPixelFormat)gpuFrame->format);
  173 + // cout << "pixfmt: " << gpu_pixfmt << endl;
  174 + // cout << "keyframe: " << gpuFrame->key_frame << " width: " << gpuFrame->width << " height: "<< gpuFrame->height << endl;
  175 + // cout << "decode successed ✿✿ヽ(°▽°)ノ✿ " << endl;
  176 +
  177 + // int sum = sum1;
  178 + // if (decoder->getName() == "dec0")
  179 + // {
  180 + // sum1 ++ ;
  181 + // sum = sum1;
  182 +
  183 + // if (gpuFrame->format == AV_PIX_FMT_CUDA)
  184 + // {
  185 + // // cout << "gpuid = " << atoi(decoder->m_cfg.gpuid.c_str()) << endl;
  186 + // cudaSetDevice(atoi(decoder->m_cfg.gpuid.c_str()));
  187 + // cudaError_t cudaStatus;
  188 + // if(pHwRgb[0] == nullptr){
  189 + // // cudaStreamCreate(&stream[0]);
  190 + // cuda_common::setColorSpace( ITU_709, 0 );
  191 + // cudaStatus = cudaMalloc((void **)&pHwRgb[0], 3 * gpuFrame->width * gpuFrame->height * sizeof(unsigned char));
  192 + // }
  193 + // cudaStatus = cuda_common::CUDAToBGR((CUdeviceptr)gpuFrame->data[0],(CUdeviceptr)gpuFrame->data[1], gpuFrame->linesize[0], gpuFrame->linesize[1], pHwRgb[0], gpuFrame->width, gpuFrame->height);
  194 + // cudaDeviceSynchronize();
  195 + // if (cudaStatus != cudaSuccess) {
  196 + // cout << "CUDAToBGR failed !!!" << endl;
  197 + // return;
  198 + // }
  199 +
  200 + // string path = data_home + decoder->getName() + ".jpg";
  201 + // saveJpeg(path.c_str(), pHwRgb[0], gpuFrame->width, gpuFrame->height, stream[0]); // 验证 CUDAToRGB
  202 + // }
  203 + // } else if (decoder->getName() == "dec2")
  204 + // {
  205 + // sum2 ++ ;
  206 + // sum = sum2;
  207 +
  208 + // if (gpuFrame->format == AV_PIX_FMT_CUDA)
  209 + // {
  210 + // // cout << "gpuid = " << atoi(decoder->m_cfg.gpuid.c_str()) << endl;
  211 + // cudaSetDevice(atoi(decoder->m_cfg.gpuid.c_str()));
  212 + // cudaError_t cudaStatus;
  213 + // if(pHwRgb[1] == nullptr){
  214 + // // cudaStreamCreate(&stream[1]);
  215 + // cuda_common::setColorSpace( ITU_709, 0 );
  216 + // cudaStatus = cudaMalloc((void **)&pHwRgb[1], 3 * gpuFrame->width * gpuFrame->height * sizeof(unsigned char));
  217 + // }
  218 + // cudaStatus = cuda_common::CUDAToBGR((CUdeviceptr)gpuFrame->data[0],(CUdeviceptr)gpuFrame->data[1], gpuFrame->linesize[0], gpuFrame->linesize[1], pHwRgb[1], gpuFrame->width, gpuFrame->height);
  219 + // cudaDeviceSynchronize();
  220 + // if (cudaStatus != cudaSuccess) {
  221 + // cout << "CUDAToBGR failed !!!" << endl;
  222 + // return;
  223 + // }
  224 +
  225 + // string path = data_home + decoder->getName() + ".jpg";
  226 + // saveJpeg(path.c_str(), pHwRgb[1], gpuFrame->width, gpuFrame->height, stream[1]); // 验证 CUDAToRGB
  227 + // }
  228 + // }
  229 + }
  230 +}
  231 +
  232 +long start_time = 0;
  233 +long end_time = 0;
  234 +bool count_flag = false;
  235 +int count = 0;
  236 +int count_std = 100;
  237 +
  238 +
  239 +static int sum = 0;
  240 +unsigned char *pHwData = nullptr;
  241 +
  242 +void postDecoded0(const void * userPtr, DeviceRgbMemory* gpuFrame){
  243 + // std::this_thread::sleep_for(std::chrono::milliseconds(30000));
  244 +
  245 + AbstractDecoder* decoder = (AbstractDecoder*)userPtr;
  246 + if (decoder!= nullptr)
  247 + {
  248 + // cout << "decode name: " << decoder->getName() << endl;
  249 + // if (decoder->getName() == "dec")
  250 + // {
  251 + // if (! count_flag)
  252 + // {
  253 + // count_flag = true;
  254 + // count = 0;
  255 + // end_time = start_time = UtilTools::get_cur_time_ms();
  256 + // }
  257 + // count++;
  258 + // sum ++ ;
  259 + // if (count >= count_std)
  260 + // {
  261 + // // end_time = UtilTools::get_cur_time_ms();
  262 + // // long time_using = end_time - start_time;
  263 + // // double time_per_frame = double(time_using)/count_std ;
  264 + // // cout << count_std << "帧用时:" << time_using << "ms 每帧用时：" << time_per_frame << "ms" << endl;
  265 + // cout << decoder->getName() << " keyframe: " << gpuFrame->key_frame << " width: " << gpuFrame->width << " height: "<< gpuFrame->height << endl;
  266 + // // cout << gpuFrame->pts << endl;
  267 +
  268 + // count_flag = false;
  269 + // }
  270 + // // cout << "帧数：" << sum << endl;
  271 +
  272 + // if (gpuFrame->format == AV_PIX_FMT_CUDA)
  273 + // {
  274 + // cudaSetDevice(atoi(decoder->m_cfg.gpuid.c_str()));
  275 + // // cout << "gpu id : " << decoder->m_cfg.gpuid.c_str() << endl;
  276 + // cudaError_t cudaStatus;
  277 + // if(pHwData == nullptr){
  278 + // cuda_common::setColorSpace( ITU_709, 0 );
  279 + // cudaStatus = cudaMalloc((void **)&pHwData, 3 * gpuFrame->width * gpuFrame->height * sizeof(unsigned char));
  280 + // }
  281 + // cudaStatus = cuda_common::CUDAToBGR((CUdeviceptr)gpuFrame->data[0],(CUdeviceptr)gpuFrame->data[1], gpuFrame->linesize[0], gpuFrame->linesize[1], pHwData, gpuFrame->width, gpuFrame->height);
  282 + // cudaDeviceSynchronize();
  283 + // if (cudaStatus != cudaSuccess) {
  284 + // cout << "CUDAToBGR failed !!!" << endl;
  285 + // return;
  286 + // }
  287 +
  288 + // string path = data_home + decoder->getName() + ".jpg";
  289 + // saveJpeg(path.c_str(), pHwData, gpuFrame->width, gpuFrame->height, nullptr); // 验证 CUDAToRGB
  290 + // }
  291 + // }
  292 + }
  293 +}
  294 +
  295 +void decode_finished_cbk(const void* userPtr){
  296 + cout << "当前时间戳: " << UtilTools::get_cur_time_ms() << endl;
  297 +}
  298 +
  299 +bool decode_request_stream_cbk(const char* deviceId){
  300 + cout << "需在此请求流" << endl;
  301 + return true;
  302 +}
  303 +
  304 +// string test_uri = "rtmp://192.168.10.56:1935/objecteye/1";
  305 +// string test_uri = "/home/cmhu/data/output_800x480.mp4";
  306 +// string test_uri = "/home/cmhu/data/output_1920x1080.mp4";
  307 +// string test_uri = "rtsp://176.10.0.2:8554/stream";
  308 +// string test_uri = "/mnt/f/fiss/test_data/h265.mp4";
  309 +string test_uri = "rtsp://176.10.0.4:8554/stream";
  310 +
  311 +void createDecode(int index, const char* gpu_id){
  312 + FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance();
  313 + MgrDecConfig config;
  314 + config.name = "dec" + to_string(index);
  315 + config.cfg.uri = test_uri;
  316 + config.cfg.post_decoded_cbk = postDecoded;
  317 + config.cfg.decode_finished_cbk = decode_finished_cbk;
  318 + config.cfg.force_tcp = true;
  319 + config.dec_type = DECODER_TYPE_FFMPEG;
  320 +
  321 + config.cfg.gpuid = gpu_id;
  322 + // if (index % 2 == 0)
  323 + // {
  324 + // config.cfg.gpuid = "0";
  325 + // }
  326 + // else
  327 + // {
  328 + // config.cfg.gpuid = "0";
  329 + // }
  330 +
  331 + AbstractDecoder* decoder = pDecManager->createDecoder(config);
  332 + if (!decoder)
  333 + {
  334 + return ;
  335 + }
  336 + pDecManager->setPostDecArg(config.name, decoder);
  337 + pDecManager->setFinishedDecArg(config.name, decoder);
  338 + pDecManager->startDecodeByName(config.name);
  339 +}
  340 +
  341 +void createGB28181Decode(int index, char* gpu_id, int port){
  342 + FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance();
  343 + MgrDecConfig config;
  344 + config.name = "dec" + to_string(index);
  345 + config.cfg.uri = config.name;
  346 + config.cfg.post_decoded_cbk = postDecoded;
  347 + config.cfg.decode_finished_cbk = decode_finished_cbk;
  348 + config.cfg.request_stream_cbk = decode_request_stream_cbk;
  349 + config.cfg.force_tcp = true;
  350 +
  351 + config.dec_type = DECODER_TYPE_GB28181;
  352 + config.cfg.port = port;//allocRtpPort();
  353 +
  354 + config.cfg.gpuid = gpu_id;
  355 +
  356 + AbstractDecoder* decoder = pDecManager->createDecoder(config);
  357 + if (!decoder)
  358 + {
  359 + return ;
  360 + }
  361 + pDecManager->setPostDecArg(config.name, decoder);
  362 + pDecManager->setFinishedDecArg(config.name, decoder);
  363 + pDecManager->startDecodeByName(config.name);
  364 +}
  365 +
  366 +void logFF(void *, int level, const char *fmt, va_list ap)
  367 +{
  368 + vfprintf(stdout, fmt, ap);
  369 +}
  370 +
  371 +
  372 +int main(int argc, char* argv[]){
  373 +
  374 + test_uri = "rtsp://admin:admin@123456@192.168.60.176:554/cam/realmonitor?channel=1&subtype=0";//argv[1];
  375 + char* gpuid = argv[2];
  376 + int port = atoi(argv[3]);
  377 + cout << test_uri << " gpu_id:" << gpuid << " port:" << port << endl;
  378 +
  379 + // av_log_set_callback(&logFF);
  380 +
  381 + CheckCUDAProperty(atoi(gpuid));
  382 +
  383 + pthread_t m_decode_thread;
  384 + pthread_create(&m_decode_thread,0,
  385 + [](void* arg)
  386 + {
  387 + // cudaSetDevice(atoi(gpuid));
  388 + while (true)
  389 + {
  390 + std::this_thread::sleep_for(std::chrono::minutes(1));
  391 + FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance();
  392 + int count = pDecManager->count();
  393 + cout << "当前时间：" << UtilTools::get_cur_time_ms() << " 当前运行路数： " << pDecManager->count() << endl;
  394 + }
  395 +
  396 + return (void*)0;
  397 + }
  398 + ,nullptr);
  399 +
  400 +
  401 + FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance();
  402 + int i = 0;
  403 +
  404 + while (true)
  405 + {
  406 + int ch = getchar();
  407 + if (ch == 'q')
  408 + {
  409 + break;
  410 + }
  411 +
  412 + switch (ch)
  413 + {
  414 + case 'f':
  415 + case 'F':
  416 + createDecode(i, gpuid);
  417 + i++;
  418 + break;
  419 + case 'g':
  420 + case 'G':
  421 + createGB28181Decode(i, gpuid, port);
  422 + i++;
  423 + break;
  424 + case 'r':
  425 + case 'R':
  426 + pDecManager->resumeDecoder("dec0");
  427 + break;
  428 + case 'p':
  429 + case 'P':
  430 + pDecManager->pauseDecoder("dec0");
  431 + break;
  432 +
  433 + case 'c':
  434 + case 'C':
  435 + i--;
  436 + pDecManager->closeDecoderByName("dec" + to_string(i));
  437 + break;
  438 +
  439 + case 'i':
  440 + case 'I':
  441 + {
  442 + int w,h;
  443 + pDecManager->getResolution("dec0", w,h);
  444 + printf( "%s : %dx%d\n", "dec0" , w,h );
  445 + }
  446 + break;
  447 +
  448 + default:
  449 + break;
  450 + }
  451 +
  452 + /* code */
  453 + }
  454 +
  455 + cout << "总共帧数：" << sum << endl;
  456 + pDecManager->closeAllDecoder();
  457 +}
0 \ No newline at end of file 458 \ No newline at end of file
src/demo/main_nvdec.cpp1 deleted
View file @d248da6
1 -// #include "FFNvDecoderManager.h"  
2 -// #include <iostream>  
3 -  
4 -// #include "cuda_kernels.h"  
5 -  
6 -// #include "NvJpegEncoder.h"  
7 -  
8 -// #include <pthread.h>  
9 -// #include <thread>  
10 -  
11 -// #include <chrono>  
12 -  
13 -// #include <unistd.h>  
14 -  
15 -  
16 -// #ifdef _WIN32  
17 -// #include "Winsock2.h"  
18 -// #pragma comment(lib, "ws2_32.lib")  
19 -// #endif  
20 -  
21 -// #ifdef __linux__  
22 -// #include "arpa/inet.h"  
23 -// #endif  
24 -  
25 -// #include "utiltools.hpp"  
26 -  
27 -// #define MIN_RTP_PORT 10000  
28 -// #define MAX_RTP_PORT 60000  
29 -  
30 -// // ȡ MIN_RTP_PORT(10000)~MAX_RTP_PORT(60000)֮�������˿�(ż���������������˿ڿ���)  
31 -// int allocRtpPort() {  
32 -  
33 -// static int s_rtpPort = MIN_RTP_PORT;  
34 -// if (MIN_RTP_PORT == s_rtpPort)  
35 -// {  
36 -// srand((unsigned int)time(NULL));  
37 -// s_rtpPort = MIN_RTP_PORT + (rand() % MIN_RTP_PORT);  
38 -// }  
39 -  
40 -// if (s_rtpPort % 2)  
41 -// ++s_rtpPort;  
42 -  
43 -// while (true)  
44 -// {  
45 -// s_rtpPort += 2;  
46 -// s_rtpPort = s_rtpPort >= MAX_RTP_PORT ? MIN_RTP_PORT : s_rtpPort;  
47 -  
48 -// int i = 0;  
49 -// for (; i < 2; i++)  
50 -// {  
51 -// sockaddr_in sRecvAddr;  
52 -// int s = socket(AF_INET, SOCK_DGRAM, 0);  
53 -  
54 -// sRecvAddr.sin_family = AF_INET;  
55 -// sRecvAddr.sin_addr.s_addr = htonl(INADDR_ANY);  
56 -// sRecvAddr.sin_port = htons(s_rtpPort + i);  
57 -  
58 -// int nResult = bind(s, (sockaddr *)&sRecvAddr, sizeof(sRecvAddr));  
59 -// if (nResult != 0)  
60 -// {  
61 -// break;  
62 -// }  
63 -  
64 -// nResult = close(s);  
65 -// if (nResult != 0)  
66 -// {  
67 -// printf("closesocket failed:%d\n", nResult);  
68 -// break;  
69 -// }  
70 -// }  
71 -  
72 -// if (i == 2)  
73 -// break;  
74 -// }  
75 -  
76 -// return s_rtpPort;  
77 -// }  
78 -  
79 -  
80 -  
81 -  
82 -  
83 -// unsigned char *pHwRgb[2] = {nullptr, nullptr};  
84 -  
85 -// int sum1 = 0;  
86 -// int sum2 = 0;  
87 -  
88 -// cudaStream_t stream[2];  
89 -  
90 -// string data_home = "/mnt/data/cmhu/tmp/";  
91 -  
92 -  
93 -// #define checkCudaErrors(S) do {CUresult status; \  
94 -// status = S; \  
95 -// if (status != CUDA_SUCCESS ) std::cout << __LINE__ <<" checkCudaErrors - status = " << status << std::endl; \  
96 -// } while (false)  
97 -  
98 -  
99 -// static void gpu_helper(int gpuid)  
100 -// {  
101 -// cudaSetDevice(gpuid);  
102 -  
103 -// // int *dn;  
104 -// // cudaMalloc((void **)&dn, 1 * sizeof(int));  
105 -  
106 -// size_t free_byte;  
107 -// size_t total_byte;  
108 -  
109 -// CUresult cuda_status = cuMemGetInfo(&free_byte, &total_byte);  
110 -  
111 -// const char *pStr = nullptr;  
112 -// if (CUDA_SUCCESS != cuda_status) {  
113 -// cuGetErrorString(cuda_status, &pStr);  
114 -// printf("Error: cudaMemGetInfo fails, %s \n", pStr);  
115 -// return;  
116 -// }  
117 -  
118 -// double free_db = (double)free_byte;  
119 -// double total_db = (double)total_byte;  
120 -// double used_db_1 = (total_db - free_db) / 1024.0 / 1024.0;  
121 -  
122 -// std::cout <<"显存已使用 " << used_db_1 << " MB\n";  
123 -  
124 -// // cudaFree(dn);  
125 -// }  
126 -  
127 -// int CheckCUDAProperty( int devId )  
128 -// {  
129 -// cuInit(0);  
130 -  
131 -// CUdevice dev = devId;  
132 -// size_t memSize = 0;  
133 -// char devName[256] = {0};  
134 -// int major = 0, minor = 0;  
135 -// CUresult rlt = CUDA_SUCCESS;  
136 -  
137 -// rlt = cuDeviceGetAttribute(&major, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR, dev);  
138 -// checkCudaErrors( rlt );  
139 -  
140 -// rlt = cuDeviceGetAttribute(&minor, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR, dev);  
141 -// checkCudaErrors( rlt );  
142 -  
143 -// rlt = cuDeviceGetName( devName, sizeof( devName ), dev );  
144 -// checkCudaErrors( rlt );  
145 -  
146 -// printf( "Using GPU Device %d: %s has SM %d.%d compute capability\n",  
147 -// dev, devName, major, minor );  
148 -  
149 -// rlt = cuDeviceTotalMem( &memSize, dev );  
150 -// checkCudaErrors( rlt );  
151 -  
152 -// printf( "Total amount of global memory: %4.4f MB\n",  
153 -// (float)memSize / ( 1024 * 1024 ) );  
154 -  
155 -// return 0;  
156 -// }  
157 -  
158 -// /**  
159 -// * 注意： gpuFrame 在解码器设置的显卡上，后续操作要十分注意这一点，尤其是多线程情况  
160 -// * */  
161 -// void postDecoded(const void * userPtr, AVFrame * gpuFrame){  
162 -// AbstractDecoder* decoder = (AbstractDecoder*)userPtr;  
163 -// if (decoder!= nullptr)  
164 -// {  
165 -// // cout << "decode name: " << decoder->getName() << endl;  
166 -  
167 -// // const char* gpu_pixfmt = av_get_pix_fmt_name((AVPixelFormat)gpuFrame->format);  
168 -// // cout << "pixfmt: " << gpu_pixfmt << endl;  
169 -// // cout << "keyframe: " << gpuFrame->key_frame << " width: " << gpuFrame->width << " height: "<< gpuFrame->height << endl;  
170 -// // cout << "decode successed ✿✿ヽ(°▽°)ノ✿ " << endl;  
171 -  
172 -// int sum = sum1;  
173 -// if (decoder->getName() == "dec0")  
174 -// {  
175 -// sum1 ++ ;  
176 -// sum = sum1;  
177 -  
178 -// if (gpuFrame->format == AV_PIX_FMT_CUDA)  
179 -// {  
180 -// // cout << "gpuid = " << atoi(decoder->m_cfg.gpuid.c_str()) << endl;  
181 -// cudaSetDevice(atoi(decoder->m_cfg.gpuid.c_str()));  
182 -// cudaError_t cudaStatus;  
183 -// if(pHwRgb[0] == nullptr){  
184 -// // cudaStreamCreate(&stream[0]);  
185 -// cuda_common::setColorSpace( ITU_709, 0 );  
186 -// cudaStatus = cudaMalloc((void **)&pHwRgb[0], 3 * gpuFrame->width * gpuFrame->height * sizeof(unsigned char));  
187 -// }  
188 -// cudaStatus = cuda_common::CUDAToBGR((CUdeviceptr)gpuFrame->data[0],(CUdeviceptr)gpuFrame->data[1], gpuFrame->linesize[0], gpuFrame->linesize[1], pHwRgb[0], gpuFrame->width, gpuFrame->height);  
189 -// cudaDeviceSynchronize();  
190 -// if (cudaStatus != cudaSuccess) {  
191 -// cout << "CUDAToBGR failed !!!" << endl;  
192 -// return;  
193 -// }  
194 -  
195 -// string path = data_home + decoder->getName() + ".jpg";  
196 -// saveJpeg(path.c_str(), pHwRgb[0], gpuFrame->width, gpuFrame->height, stream[0]); // 验证 CUDAToRGB  
197 -// }  
198 -// } else if (decoder->getName() == "dec2")  
199 -// {  
200 -// sum2 ++ ;  
201 -// sum = sum2;  
202 -  
203 -// if (gpuFrame->format == AV_PIX_FMT_CUDA)  
204 -// {  
205 -// // cout << "gpuid = " << atoi(decoder->m_cfg.gpuid.c_str()) << endl;  
206 -// cudaSetDevice(atoi(decoder->m_cfg.gpuid.c_str()));  
207 -// cudaError_t cudaStatus;  
208 -// if(pHwRgb[1] == nullptr){  
209 -// // cudaStreamCreate(&stream[1]);  
210 -// cuda_common::setColorSpace( ITU_709, 0 );  
211 -// cudaStatus = cudaMalloc((void **)&pHwRgb[1], 3 * gpuFrame->width * gpuFrame->height * sizeof(unsigned char));  
212 -// }  
213 -// cudaStatus = cuda_common::CUDAToBGR((CUdeviceptr)gpuFrame->data[0],(CUdeviceptr)gpuFrame->data[1], gpuFrame->linesize[0], gpuFrame->linesize[1], pHwRgb[1], gpuFrame->width, gpuFrame->height);  
214 -// cudaDeviceSynchronize();  
215 -// if (cudaStatus != cudaSuccess) {  
216 -// cout << "CUDAToBGR failed !!!" << endl;  
217 -// return;  
218 -// }  
219 -  
220 -// string path = data_home + decoder->getName() + ".jpg";  
221 -// saveJpeg(path.c_str(), pHwRgb[1], gpuFrame->width, gpuFrame->height, stream[1]); // 验证 CUDAToRGB  
222 -// }  
223 -// }  
224 -// }  
225 -// }  
226 -  
227 -// long start_time = 0;  
228 -// long end_time = 0;  
229 -// bool count_flag = false;  
230 -// int count = 0;  
231 -// int count_std = 100;  
232 -  
233 -  
234 -// static int sum = 0;  
235 -// unsigned char *pHwData = nullptr;  
236 -  
237 -// void postDecoded0(const void * userPtr, AVFrame * gpuFrame){  
238 -// // std::this_thread::sleep_for(std::chrono::milliseconds(30000));  
239 -  
240 -// AbstractDecoder* decoder = (AbstractDecoder*)userPtr;  
241 -// if (decoder!= nullptr)  
242 -// {  
243 -// // cout << "decode name: " << decoder->getName() << endl;  
244 -// if (decoder->getName() == "dec")  
245 -// {  
246 -// if (! count_flag)  
247 -// {  
248 -// count_flag = true;  
249 -// count = 0;  
250 -// end_time = start_time = UtilTools::get_cur_time_ms();  
251 -// }  
252 -// count++;  
253 -// sum ++ ;  
254 -// if (count >= count_std)  
255 -// {  
256 -// // end_time = UtilTools::get_cur_time_ms();  
257 -// // long time_using = end_time - start_time;  
258 -// // double time_per_frame = double(time_using)/count_std ;  
259 -// // cout << count_std << "帧用时:" << time_using << "ms 每帧用时：" << time_per_frame << "ms" << endl;  
260 -// cout << decoder->getName() << " keyframe: " << gpuFrame->key_frame << " width: " << gpuFrame->width << " height: "<< gpuFrame->height << endl;  
261 -// // cout << gpuFrame->pts << endl;  
262 -  
263 -// count_flag = false;  
264 -// }  
265 -// // cout << "帧数：" << sum << endl;  
266 -  
267 -// if (gpuFrame->format == AV_PIX_FMT_CUDA)  
268 -// {  
269 -// cudaSetDevice(atoi(decoder->m_cfg.gpuid.c_str()));  
270 -// // cout << "gpu id : " << decoder->m_cfg.gpuid.c_str() << endl;  
271 -// cudaError_t cudaStatus;  
272 -// if(pHwData == nullptr){  
273 -// cuda_common::setColorSpace( ITU_709, 0 );  
274 -// cudaStatus = cudaMalloc((void **)&pHwData, 3 * gpuFrame->width * gpuFrame->height * sizeof(unsigned char));  
275 -// }  
276 -// cudaStatus = cuda_common::CUDAToBGR((CUdeviceptr)gpuFrame->data[0],(CUdeviceptr)gpuFrame->data[1], gpuFrame->linesize[0], gpuFrame->linesize[1], pHwData, gpuFrame->width, gpuFrame->height);  
277 -// cudaDeviceSynchronize();  
278 -// if (cudaStatus != cudaSuccess) {  
279 -// cout << "CUDAToBGR failed !!!" << endl;  
280 -// return;  
281 -// }  
282 -  
283 -// string path = data_home + decoder->getName() + ".jpg";  
284 -// saveJpeg(path.c_str(), pHwData, gpuFrame->width, gpuFrame->height, nullptr); // 验证 CUDAToRGB  
285 -// }  
286 -// }  
287 -// }  
288 -// }  
289 -  
290 -// void decode_finished_cbk(const void* userPtr){  
291 -// cout << "当前时间戳: " << UtilTools::get_cur_time_ms() << endl;  
292 -// }  
293 -  
294 -// bool decode_request_stream_cbk(const char* deviceId){  
295 -// cout << "需在此请求流" << endl;  
296 -// return true;  
297 -// }  
298 -  
299 -// // string test_uri = "rtmp://192.168.10.56:1935/objecteye/1";  
300 -// // string test_uri = "/home/cmhu/data/output_800x480.mp4";  
301 -// // string test_uri = "/home/cmhu/data/output_1920x1080.mp4";  
302 -// // string test_uri = "rtsp://176.10.0.2:8554/stream";  
303 -// // string test_uri = "/mnt/f/fiss/test_data/h265.mp4";  
304 -// string test_uri = "rtsp://176.10.0.4:8554/stream";  
305 -  
306 -// void createDecode(int index, const char* gpu_id){  
307 -// FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance();  
308 -// MgrDecConfig config;  
309 -// config.name = "dec" + to_string(index);  
310 -// config.cfg.uri = test_uri;  
311 -// config.cfg.post_decoded_cbk = postDecoded;  
312 -// config.cfg.decode_finished_cbk = decode_finished_cbk;  
313 -// config.cfg.force_tcp = true;  
314 -// config.dec_type = DECODER_TYPE_FFMPEG;  
315 -  
316 -// config.cfg.gpuid = gpu_id;  
317 -// // if (index % 2 == 0)  
318 -// // {  
319 -// // config.cfg.gpuid = "0";  
320 -// // }  
321 -// // else  
322 -// // {  
323 -// // config.cfg.gpuid = "0";  
324 -// // }  
325 -  
326 -// AbstractDecoder* decoder = pDecManager->createDecoder(config);  
327 -// if (!decoder)  
328 -// {  
329 -// return ;  
330 -// }  
331 -// pDecManager->setPostDecArg(config.name, decoder);  
332 -// pDecManager->setFinishedDecArg(config.name, decoder);  
333 -// pDecManager->startDecodeByName(config.name);  
334 -// }  
335 -  
336 -// void createGB28181Decode(int index, char* gpu_id, int port){  
337 -// FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance();  
338 -// MgrDecConfig config;  
339 -// config.name = "dec" + to_string(index);  
340 -// config.cfg.uri = config.name;  
341 -// config.cfg.post_decoded_cbk = postDecoded;  
342 -// config.cfg.decode_finished_cbk = decode_finished_cbk;  
343 -// config.cfg.request_stream_cbk = decode_request_stream_cbk;  
344 -// config.cfg.force_tcp = true;  
345 -  
346 -// config.dec_type = DECODER_TYPE_GB28181;  
347 -// config.cfg.port = port;//allocRtpPort();  
348 -  
349 -// config.cfg.gpuid = gpu_id;  
350 -  
351 -// AbstractDecoder* decoder = pDecManager->createDecoder(config);  
352 -// if (!decoder)  
353 -// {  
354 -// return ;  
355 -// }  
356 -// pDecManager->setPostDecArg(config.name, decoder);  
357 -// pDecManager->setFinishedDecArg(config.name, decoder);  
358 -// pDecManager->startDecodeByName(config.name);  
359 -// }  
360 -  
361 -// void logFF(void *, int level, const char *fmt, va_list ap)  
362 -// {  
363 -// vfprintf(stdout, fmt, ap);  
364 -// }  
365 -  
366 -  
367 -// int main(int argc, char* argv[]){  
368 -  
369 -// test_uri = "rtsp://admin:admin@123456@192.168.60.176:554/cam/realmonitor?channel=1&subtype=0";//argv[1];  
370 -// char* gpuid = argv[2];  
371 -// int port = atoi(argv[3]);  
372 -// cout << test_uri << " gpu_id:" << gpuid << " port:" << port << endl;  
373 -  
374 -// // av_log_set_callback(&logFF);  
375 -  
376 -// CheckCUDAProperty(atoi(gpuid));  
377 -  
378 -// pthread_t m_decode_thread;  
379 -// pthread_create(&m_decode_thread,0,  
380 -// [](void* arg)  
381 -// {  
382 -// // cudaSetDevice(atoi(gpuid));  
383 -// while (true)  
384 -// {  
385 -// std::this_thread::sleep_for(std::chrono::minutes(1));  
386 -// FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance();  
387 -// int count = pDecManager->count();  
388 -// cout << "当前时间：" << UtilTools::get_cur_time_ms() << " 当前运行路数： " << pDecManager->count() << endl;  
389 -// }  
390 -  
391 -// return (void*)0;  
392 -// }  
393 -// ,nullptr);  
394 -  
395 -  
396 -// FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance();  
397 -// int i = 0;  
398 -  
399 -// while (true)  
400 -// {  
401 -// int ch = getchar();  
402 -// if (ch == 'q')  
403 -// {  
404 -// break;  
405 -// }  
406 -  
407 -// switch (ch)  
408 -// {  
409 -// case 'f':  
410 -// case 'F':  
411 -// createDecode(i, gpuid);  
412 -// i++;  
413 -// break;  
414 -// case 'g':  
415 -// case 'G':  
416 -// createGB28181Decode(i, gpuid, port);  
417 -// i++;  
418 -// break;  
419 -// case 'r':  
420 -// case 'R':  
421 -// pDecManager->resumeDecoder("dec0");  
422 -// break;  
423 -// case 'p':  
424 -// case 'P':  
425 -// pDecManager->pauseDecoder("dec0");  
426 -// break;  
427 -  
428 -// case 'c':  
429 -// case 'C':  
430 -// i--;  
431 -// pDecManager->closeDecoderByName("dec" + to_string(i));  
432 -// break;  
433 -  
434 -// case 'i':  
435 -// case 'I':  
436 -// {  
437 -// int w,h;  
438 -// pDecManager->getResolution("dec0", w,h);  
439 -// printf( "%s : %dx%d\n", "dec0" , w,h );  
440 -// }  
441 -// break;  
442 -  
443 -// default:  
444 -// break;  
445 -// }  
446 -  
447 -// /* code */  
448 -// }  
449 -  
450 -// cout << "总共帧数：" << sum << endl;  
451 -// pDecManager->closeAllDecoder();  
452 -// }  
453 \ No newline at end of file 0 \ No newline at end of file
src/dvpp/DvppDecoderApi.h
  View file @00b0fbd
@@ -8,7 +8,7 @@ using namespace std; @@ -8,7 +8,7 @@ using namespace std;
8 8
9 class DvppDecoder; 9 class DvppDecoder;
10 10
11 -class DvppDecoderApi : public AbstractDecoder{ 11 +class DvppDecoderApi : public AbstractDecoder {
12 public: 12 public:
13 DvppDecoderApi(); 13 DvppDecoderApi();
14 ~DvppDecoderApi(); 14 ~DvppDecoderApi();
src/dvpp/depend_headers.h
  View file @00b0fbd
@@ -23,10 +23,10 @@ @@ -23,10 +23,10 @@
23 23
24 // ffmpeg 是c库 所以编译的时候要加入从 extern导入的C 来声明否则连接失败 24 // ffmpeg 是c库 所以编译的时候要加入从 extern导入的C 来声明否则连接失败
25 extern "C" { 25 extern "C" {
26 -#include "libavutil/imgutils.h"  
27 -#include "libavutil/samplefmt.h"  
28 -#include "libavformat/avformat.h"  
29 -#include "libavcodec/avcodec.h" 26 + #include "libavutil/imgutils.h"
  27 + #include "libavutil/samplefmt.h"
  28 + #include "libavformat/avformat.h"
  29 + #include "libavcodec/avcodec.h"
30 } 30 }
31 31
32 32
src/gb28181/FFGB28181Decoder.cpp
  View file @00b0fbd
@@ -2,7 +2,7 @@ @@ -2,7 +2,7 @@
2 #include <iostream> 2 #include <iostream>
3 #include "FFGB28181Decoder.h" 3 #include "FFGB28181Decoder.h"
4 4
5 -#include "../FFCuContextManager.h" 5 +#include "../nvdecoder/FFCuContextManager.h"
6 6
7 extern "C" { 7 extern "C" {
8 #include "libavutil/avstring.h" 8 #include "libavutil/avstring.h"
@@ -296,7 +296,7 @@ void FFGB28181Decoder::post_decode_thread(){ @@ -296,7 +296,7 @@ void FFGB28181Decoder::post_decode_thread(){
296 m_queue_mutex.unlock(); 296 m_queue_mutex.unlock();
297 // 跳帧 297 // 跳帧
298 if (m_frameSkip == 1 || index % m_frameSkip == 0){ 298 if (m_frameSkip == 1 || index % m_frameSkip == 0){
299 - post_decoded_cbk(m_postDecArg, gpuFrame); 299 + post_decoded_cbk(m_postDecArg, convert2bgr(gpuFrame));
300 } 300 }
301 301
302 av_frame_free(&gpuFrame); 302 av_frame_free(&gpuFrame);
@@ -320,6 +320,14 @@ void FFGB28181Decoder::stream_end_callback() @@ -320,6 +320,14 @@ void FFGB28181Decoder::stream_end_callback()
320 return; 320 return;
321 } 321 }
322 322
  323 +void FFGB28181Decoder::setPostDecArg(const void* postDecArg){
  324 + m_postDecArg = postDecArg;
  325 +}
  326 +
  327 +void FFGB28181Decoder::setFinishedDecArg(const void* finishedDecArg){
  328 + m_finishedDecArg = finishedDecArg;
  329 +}
  330 +
323 void FFGB28181Decoder::pause() { 331 void FFGB28181Decoder::pause() {
324 m_status = EPAUSE; 332 m_status = EPAUSE;
325 LOG_INFO("pause --{}", m_dec_name); 333 LOG_INFO("pause --{}", m_dec_name);
@@ -370,6 +378,36 @@ int FFGB28181Decoder::getCachedQueueLength(){ @@ -370,6 +378,36 @@ int FFGB28181Decoder::getCachedQueueLength(){
370 return m_rtpPtr->GetPsFrameListSize(); 378 return m_rtpPtr->GetPsFrameListSize();
371 } 379 }
372 380
  381 +DeviceRgbMemory* FFGB28181Decoder::convert2bgr(AVFrame * gpuFrame){
  382 + if (gpuFrame != nullptr && gpuFrame->format == AV_PIX_FMT_CUDA ){
  383 + LOG_DEBUG("decode task: gpuid: {} width: {} height: {}", m_cfg.gpuid, gpuFrame->width, gpuFrame->height);
  384 + GpuRgbMemory* gpuMem = new GpuRgbMemory(3, gpuFrame->width, gpuFrame->height, getName(), m_cfg.gpuid, false, true);
  385 +
  386 + do{
  387 + if (gpuMem->getMem() == nullptr){
  388 + LOG_ERROR("new GpuRgbMemory failed !!!");
  389 + break;
  390 + }
  391 +
  392 + cudaSetDevice(atoi(m_cfg.gpuid.c_str()));
  393 + cuda_common::setColorSpace( ITU_709, 0 );
  394 + cudaError_t cudaStatus = cuda_common::CUDAToBGR((CUdeviceptr)gpuFrame->data[0],(CUdeviceptr)gpuFrame->data[1], gpuFrame->linesize[0], gpuFrame->linesize[1], gpuMem->getMem(), gpuFrame->width, gpuFrame->height);
  395 + cudaDeviceSynchronize();
  396 + if (cudaStatus != cudaSuccess) {
  397 + LOG_ERROR("CUDAToBGR failed failed !!!");
  398 + break;
  399 + }
  400 +
  401 + return gpuMem;
  402 + }while(0);
  403 +
  404 + delete gpuMem;
  405 + gpuMem = nullptr;
  406 + }
  407 +
  408 + return nullptr;
  409 +}
  410 +
373 FFImgInfo* FFGB28181Decoder::snapshot(){ 411 FFImgInfo* FFGB28181Decoder::snapshot(){
374 412
375 // 锁住停止队列消耗 413 // 锁住停止队列消耗
@@ -401,7 +439,7 @@ FFImgInfo* FFGB28181Decoder::snapshot(){ @@ -401,7 +439,7 @@ FFImgInfo* FFGB28181Decoder::snapshot(){
401 439
402 if (gpuFrame != nullptr && gpuFrame->format == AV_PIX_FMT_CUDA ){ 440 if (gpuFrame != nullptr && gpuFrame->format == AV_PIX_FMT_CUDA ){
403 LOG_DEBUG("decode task: gpuid: {} width: {} height: {}", m_cfg.gpuid, gpuFrame->width, gpuFrame->height); 441 LOG_DEBUG("decode task: gpuid: {} width: {} height: {}", m_cfg.gpuid, gpuFrame->width, gpuFrame->height);
404 - GpuRgbMemory* gpuMem = new GpuRgbMemory(3, gpuFrame->width, gpuFrame->height, getName(), m_cfg.gpuid , true); 442 + GpuRgbMemory* gpuMem = new GpuRgbMemory(3, gpuFrame->width, gpuFrame->height, getName(), m_cfg.gpuid , false, true);
405 443
406 if (gpuMem->getMem() == nullptr){ 444 if (gpuMem->getMem() == nullptr){
407 LOG_ERROR("new GpuRgbMemory failed !!!"); 445 LOG_ERROR("new GpuRgbMemory failed !!!");
src/gb28181/FFGB28181Decoder.h
  View file @00b0fbd
@@ -3,7 +3,8 @@ @@ -3,7 +3,8 @@
3 3
4 #include "RTPReceiver.h" 4 #include "RTPReceiver.h"
5 5
6 -#include "../AbstractDecoder.h" 6 +#include "common_header.h"
  7 +#include "../interface/AbstractDecoder.h"
7 8
8 #include <atomic> 9 #include <atomic>
9 #include <mutex> 10 #include <mutex>
@@ -14,6 +15,7 @@ struct AVCodec; @@ -14,6 +15,7 @@ struct AVCodec;
14 struct AVFrame; 15 struct AVFrame;
15 struct AVPacket; 16 struct AVPacket;
16 struct SwsContext; 17 struct SwsContext;
  18 +struct AVDictionary;
17 19
18 using namespace std; 20 using namespace std;
19 21
@@ -46,12 +48,29 @@ public: @@ -46,12 +48,29 @@ public:
46 48
47 FFImgInfo* snapshot(); 49 FFImgInfo* snapshot();
48 50
  51 + void setName(string nm){
  52 + m_dec_name = nm;
  53 + }
  54 +
  55 + string getName(){
  56 + return m_dec_name;
  57 + }
  58 +
  59 + void setPostDecArg(const void* postDecArg);
  60 + void setFinishedDecArg(const void* finishedDecArg);
  61 +
49 public: 62 public:
50 void stream_callback(int videoType, char* data, int len, int isKey, uint64_t pts, uint64_t localPts); 63 void stream_callback(int videoType, char* data, int len, int isKey, uint64_t pts, uint64_t localPts);
51 void stream_end_callback(); 64 void stream_end_callback();
52 void post_decode_thread(); 65 void post_decode_thread();
53 66
54 private: 67 private:
  68 + DeviceRgbMemory* convert2bgr(AVFrame * gpuFrame);
  69 +
  70 +private:
  71 + string m_dec_name;
  72 + FFDecConfig m_cfg;
  73 +
55 AVCodecContext* m_pAVCodecCtx {}; 74 AVCodecContext* m_pAVCodecCtx {};
56 const AVCodec* m_pAVCodec {}; 75 const AVCodec* m_pAVCodec {};
57 76
@@ -76,10 +95,17 @@ private: @@ -76,10 +95,17 @@ private:
76 AVDictionary *gpu_options = nullptr; 95 AVDictionary *gpu_options = nullptr;
77 96
78 pthread_t m_post_decode_thread; 97 pthread_t m_post_decode_thread;
  98 + const void * m_postDecArg;
  99 + POST_DECODE_CALLBACK post_decoded_cbk; // 解码数据回调接口
  100 +
  101 + const void * m_finishedDecArg;
  102 + DECODE_FINISHED_CALLBACK decode_finished_cbk;
79 103
80 queue<AVFrame*> mFrameQueue; 104 queue<AVFrame*> mFrameQueue;
81 mutex m_queue_mutex; 105 mutex m_queue_mutex;
82 mutex m_snapshot_mutex; 106 mutex m_snapshot_mutex;
  107 +
  108 + bool m_dec_keyframe;
83 }; 109 };
84 110
85 #endif // _GB28181_DECODER_H_ 111 #endif // _GB28181_DECODER_H_
src/gb28181/Makefile
  View file @00b0fbd
@@ -12,6 +12,34 @@ DEPS:=$(addprefix $(DEP_DIR)/, $(patsubst %.cpp, %.d,a $(SRCS))) @@ -12,6 +12,34 @@ DEPS:=$(addprefix $(DEP_DIR)/, $(patsubst %.cpp, %.d,a $(SRCS)))
12 # 自动生成头文件依赖选项 12 # 自动生成头文件依赖选项
13 DEPFLAGS=-MT $@ -MMD -MP -MF $(DEP_DIR)/$*.d 13 DEPFLAGS=-MT $@ -MMD -MP -MF $(DEP_DIR)/$*.d
14 14
  15 +CUDA_ROOT = /usr/local/cuda-11.1
  16 +
  17 +PROJECT_ROOT= /mnt/data/cmhu/FFNvDecoder
  18 +
  19 +DEPEND_DIR = $(PROJECT_ROOT)/bin
  20 +THIRDPARTY_ROOT = $(PROJECT_ROOT)/3rdparty
  21 +SPDLOG_ROOT = $(THIRDPARTY_ROOT)/spdlog-1.9.2/release
  22 +JRTP_ROOT = $(THIRDPARTY_ROOT)/jrtp_export
  23 +
  24 +
  25 +INCLUDE= -I $(DEPEND_DIR)/include \
  26 + -I $(CUDA_ROOT)/include \
  27 + -I $(TOP_DIR)/common/inc \
  28 + -I $(TOP_DIR)/common/UtilNPP \
  29 + -I $(TOP_DIR)/ \
  30 + -I $(SPDLOG_ROOT)/include \
  31 + -I $(JRTP_ROOT)/jrtplib/include/jrtplib3 \
  32 + -I $(JRTP_ROOT)/jthread/include/jthread
  33 +
  34 +LIBSPATH= -L $(DEPEND_DIR)/lib -lavformat -lavcodec -lswscale -lavutil -lavfilter -lswresample -lavdevice \
  35 + -L $(CUDA_ROOT)/lib64 -lcuda -lcudart -lnvcuvid -lcurand -lcublas -lnvjpeg \
  36 + -L $(SPDLOG_ROOT) -l:libspdlog.a \
  37 + -L $(JRTP_ROOT)/jthread/lib -l:libjthread.a \
  38 + -L $(JRTP_ROOT)/jrtplib/lib -l:libjrtp.a
  39 +
  40 +
  41 +CXXFLAGS= -g -O0 -fPIC $(INCLUDE) $(DEFS) -lpthread -lrt -lz -fexceptions -std=c++11 -fvisibility=hidden -Wl,-Bsymbolic -ldl -Wwrite-strings
  42 +
15 # 最终目标文件 43 # 最终目标文件
16 TARGET:=$(LIB_DIR)/$(MODULE).a 44 TARGET:=$(LIB_DIR)/$(MODULE).a
17 45
@@ -22,7 +50,7 @@ all:$(TARGET) @@ -22,7 +50,7 @@ all:$(TARGET)
22 # 生成最终目标 50 # 生成最终目标
23 $(TARGET):$(OBJS) | $(LIB_DIR) 51 $(TARGET):$(OBJS) | $(LIB_DIR)
24 @echo -e "\e[32m""Linking static library $(TARGET)""\e[0m" 52 @echo -e "\e[32m""Linking static library $(TARGET)""\e[0m"
25 - @ar -rc $@ $^ 53 +# @ar -rc $@ $^
26 54
27 # 若没有lib目录则自动生成 55 # 若没有lib目录则自动生成
28 $(LIB_DIR): 56 $(LIB_DIR):
@@ -31,7 +59,7 @@ $(LIB_DIR): @@ -31,7 +59,7 @@ $(LIB_DIR):
31 # 生成中间目标文件 59 # 生成中间目标文件
32 $(OBJ_DIR)/%.o:$(SRC_DIR)/%.cpp $(DEP_DIR)/%.d | $(OBJ_DIR) $(DEP_DIR) 60 $(OBJ_DIR)/%.o:$(SRC_DIR)/%.cpp $(DEP_DIR)/%.d | $(OBJ_DIR) $(DEP_DIR)
33 @echo -e "\e[33m""Building object $@""\e[0m" 61 @echo -e "\e[33m""Building object $@""\e[0m"
34 - @$(CXX) -c $(DEPFLAGS) $(CXXFLAGS) $(INCS) $(MACROS) -o $@ $< 62 + @$(CXX) -c $(DEPFLAGS) $(CXXFLAGS) $(INCS) $(LIBSPATH) $(MACROS) -o $@ $<
35 63
36 # 若没有obj目录则自动生成 64 # 若没有obj目录则自动生成
37 $(OBJ_DIR): 65 $(OBJ_DIR):
src/interface/FFNvDecoderManager.h
  View file @00b0fbd
@@ -14,8 +14,8 @@ struct MgrDecConfig @@ -14,8 +14,8 @@ struct MgrDecConfig
14 string name{""}; // 解码器名称 14 string name{""}; // 解码器名称
15 }; 15 };
16 16
17 -// #define USE_NVDEC  
18 -#define USE_DVPP 17 +#define USE_NVDEC
  18 +// #define USE_DVPP
19 /** 19 /**
20 * 解码器管理类，单例类 20 * 解码器管理类，单例类
21 * 谨防死锁 21 * 谨防死锁
src/interface/Makefile
  View file @00b0fbd
@@ -12,10 +12,41 @@ DEPS:=$(addprefix $(DEP_DIR)/, $(patsubst %.cpp, %.d,a $(SRCS))) @@ -12,10 +12,41 @@ DEPS:=$(addprefix $(DEP_DIR)/, $(patsubst %.cpp, %.d,a $(SRCS)))
12 # 自动生成头文件依赖选项 12 # 自动生成头文件依赖选项
13 DEPFLAGS=-MT $@ -MMD -MP -MF $(DEP_DIR)/$*.d 13 DEPFLAGS=-MT $@ -MMD -MP -MF $(DEP_DIR)/$*.d
14 14
  15 +
  16 +PROJECT_ROOT= /mnt/data/cmhu/FFNvDecoder
  17 +
  18 +DEPEND_DIR = $(PROJECT_ROOT)/bin
  19 +THIRDPARTY_ROOT = $(PROJECT_ROOT)/3rdparty
  20 +SPDLOG_ROOT = $(THIRDPARTY_ROOT)/spdlog-1.9.2/release
  21 +JRTP_ROOT = $(THIRDPARTY_ROOT)/jrtp_export
  22 +
  23 +
  24 +INCLUDE= -I $(DEPEND_DIR)/include \
  25 + -I $(CUDA_ROOT)/include \
  26 + -I $(TOP_DIR)/common/inc \
  27 + -I $(TOP_DIR)/common/UtilNPP \
  28 + -I $(TOP_DIR)/ \
  29 + -I $(SPDLOG_ROOT)/include \
  30 + -I $(JRTP_ROOT)/jrtplib/include/jrtplib3 \
  31 + -I $(JRTP_ROOT)/jthread/include/jthread \
  32 + -I $(TOP_DIR)/src/gb28181 \
  33 + -I $(TOP_DIR)/src/nvdecoder \
  34 +
  35 +LIBSPATH= -L $(DEPEND_DIR)/lib -lavformat -lavcodec -lswscale -lavutil -lavfilter -lswresample -lavdevice \
  36 + -L $(CUDA_ROOT)/lib64 -lcuda -lcudart -lnvcuvid -lcurand -lcublas -lnvjpeg \
  37 + -L $(SPDLOG_ROOT) -l:libspdlog.a \
  38 + -L $(JRTP_ROOT)/jthread/lib -l:libjthread.a \
  39 + -L $(JRTP_ROOT)/jrtplib/lib -l:libjrtp.a
  40 +
  41 +
  42 +CXXFLAGS= -g -O0 -fPIC $(INCLUDE) $(DEFS) -lpthread -lrt -lz -fexceptions -std=c++11 -fvisibility=hidden -Wl,-Bsymbolic -ldl -Wwrite-strings
  43 +
  44 +
15 # 最终目标文件 45 # 最终目标文件
16 TARGET:=$(LIB_DIR)/$(MODULE).a 46 TARGET:=$(LIB_DIR)/$(MODULE).a
17 47
18 -MODULE_LIBS:=$(BUILD_DIR)/dvpp/lib/libdvpp.a\ 48 +# MODULE_LIBS:=$(BUILD_DIR)/nvdecoder/lib/nvdecoder.a \
  49 +# $(BUILD_DIR)/gb28181/lib/gb28181.a\
19 50
20 # 默认最终目标 51 # 默认最终目标
21 .PHONY:all 52 .PHONY:all
@@ -23,8 +54,8 @@ all:$(TARGET) @@ -23,8 +54,8 @@ all:$(TARGET)
23 54
24 # 生成最终目标 55 # 生成最终目标
25 $(TARGET):$(OBJS) | $(LIB_DIR) 56 $(TARGET):$(OBJS) | $(LIB_DIR)
26 - @echo -e "\e[32m""Linking static library $(TARGET)""\e[0m"  
27 - @ar -rc $@ $^ 57 + # @echo -e "\e[32m""Linking static library $(TARGET)""\e[0m"
  58 +# @ar -rc $@ $^
28 59
29 # 若没有lib目录则自动生成 60 # 若没有lib目录则自动生成
30 $(LIB_DIR): 61 $(LIB_DIR):
@@ -33,7 +64,8 @@ $(LIB_DIR): @@ -33,7 +64,8 @@ $(LIB_DIR):
33 # 生成中间目标文件 64 # 生成中间目标文件
34 $(OBJ_DIR)/%.o:$(SRC_DIR)/%.cpp $(DEP_DIR)/%.d | $(OBJ_DIR) $(DEP_DIR) 65 $(OBJ_DIR)/%.o:$(SRC_DIR)/%.cpp $(DEP_DIR)/%.d | $(OBJ_DIR) $(DEP_DIR)
35 @echo -e "\e[33m""Building object $@""\e[0m" 66 @echo -e "\e[33m""Building object $@""\e[0m"
36 - @$(CXX) -c $(DEPFLAGS) $(CXXFLAGS) $(INCS) $(MACROS) -o $@ $(MODULE_LIBS) $< 67 +# @$(CXX) -c $(DEPFLAGS) $(CXXFLAGS) $(INCS) $(LIBSPATH) $(MACROS) -o $@ $(MODULE_LIBS) $<
  68 + @$(CXX) -c $(DEPFLAGS) $(CXXFLAGS) $(INCS) $(LIBSPATH) $(MACROS) -o $@ $<
37 69
38 # 若没有obj目录则自动生成 70 # 若没有obj目录则自动生成
39 $(OBJ_DIR): 71 $(OBJ_DIR):
src/interface/utiltools.hpp
  View file @00b0fbd
1 #ifndef _UTIL_TOOLS_HPP_ 1 #ifndef _UTIL_TOOLS_HPP_
2 #define _UTIL_TOOLS_HPP_ 2 #define _UTIL_TOOLS_HPP_
3 3
4 -#include<chrono> 4 +#include <chrono>
5 5
6 using namespace std; 6 using namespace std;
7 7
src/nvdecoder/DrawImageOnGPU.cu
  View file @00b0fbd
1 #include "cuda_kernels.h" 1 #include "cuda_kernels.h"
2 2
3 -#include "logger.hpp" 3 +#include "../interface/logger.hpp"
4 4
5 typedef unsigned char uchar; 5 typedef unsigned char uchar;
6 typedef unsigned int uint32; 6 typedef unsigned int uint32;
src/nvdecoder/FFCuContextManager.cpp
  View file @00b0fbd
@@ -4,6 +4,18 @@ @@ -4,6 +4,18 @@
4 4
5 using namespace std; 5 using namespace std;
6 6
  7 +extern "C"
  8 +{
  9 + #include <libavcodec/avcodec.h>
  10 + #include <libavdevice/avdevice.h>
  11 + #include <libavformat/avformat.h>
  12 + #include <libavfilter/avfilter.h>
  13 + #include <libavutil/avutil.h>
  14 + #include <libavutil/pixdesc.h>
  15 + #include <libswscale/swscale.h>
  16 + #include <libavutil/imgutils.h>
  17 +}
  18 +
7 FFCuContextManager::~FFCuContextManager() 19 FFCuContextManager::~FFCuContextManager()
8 { 20 {
9 for(auto iter = ctxMap.begin(); iter != ctxMap.end(); iter++){ 21 for(auto iter = ctxMap.begin(); iter != ctxMap.end(); iter++){
src/nvdecoder/FFCuContextManager.h
  View file @00b0fbd
@@ -2,19 +2,10 @@ @@ -2,19 +2,10 @@
2 #include<map> 2 #include<map>
3 #include<string> 3 #include<string>
4 4
5 -extern "C"  
6 -{  
7 - #include <libavcodec/avcodec.h>  
8 - #include <libavdevice/avdevice.h>  
9 - #include <libavformat/avformat.h>  
10 - #include <libavfilter/avfilter.h>  
11 - #include <libavutil/avutil.h>  
12 - #include <libavutil/pixdesc.h>  
13 - #include <libswscale/swscale.h>  
14 -}  
15 -  
16 using namespace std; 5 using namespace std;
17 6
  7 +struct AVBufferRef;
  8 +
18 class FFCuContextManager{ 9 class FFCuContextManager{
19 public: 10 public:
20 static FFCuContextManager* getInstance(){ 11 static FFCuContextManager* getInstance(){
src/nvdecoder/FFNvDecoder.cpp
  View file @00b0fbd
@@ -63,6 +63,7 @@ FFNvDecoder::~FFNvDecoder() @@ -63,6 +63,7 @@ FFNvDecoder::~FFNvDecoder()
63 bool FFNvDecoder::init(FFDecConfig& cfg) 63 bool FFNvDecoder::init(FFDecConfig& cfg)
64 { 64 {
65 m_cfg = cfg; 65 m_cfg = cfg;
  66 + m_dec_name = cfg.dec_name;
66 67
67 fstream infile(cfg.uri); 68 fstream infile(cfg.uri);
68 if (infile.is_open()){ 69 if (infile.is_open()){
@@ -315,7 +316,7 @@ void FFNvDecoder::post_decode_thread(){ @@ -315,7 +316,7 @@ void FFNvDecoder::post_decode_thread(){
315 m_queue_mutex.unlock(); 316 m_queue_mutex.unlock();
316 // 跳帧 317 // 跳帧
317 if (skip_frame == 1 || index % skip_frame == 0){ 318 if (skip_frame == 1 || index % skip_frame == 0){
318 - post_decoded_cbk(m_postDecArg, gpuFrame); 319 + post_decoded_cbk(m_postDecArg, convert2bgr(gpuFrame));
319 index = 0; 320 index = 0;
320 } 321 }
321 322
@@ -387,6 +388,44 @@ float FFNvDecoder::fps(){ @@ -387,6 +388,44 @@ float FFNvDecoder::fps(){
387 return m_fps; 388 return m_fps;
388 } 389 }
389 390
  391 +void FFNvDecoder::setPostDecArg(const void* postDecArg){
  392 + m_postDecArg = postDecArg;
  393 +}
  394 +
  395 +void FFNvDecoder::setFinishedDecArg(const void* finishedDecArg){
  396 + m_finishedDecArg = finishedDecArg;
  397 +}
  398 +
  399 +DeviceRgbMemory* FFNvDecoder::convert2bgr(AVFrame * gpuFrame){
  400 + if (gpuFrame != nullptr && gpuFrame->format == AV_PIX_FMT_CUDA ){
  401 + LOG_DEBUG("decode task: gpuid: {} width: {} height: {}", m_cfg.gpuid, gpuFrame->width, gpuFrame->height);
  402 + GpuRgbMemory* gpuMem = new GpuRgbMemory(3, gpuFrame->width, gpuFrame->height, getName(), m_cfg.gpuid, false, true);
  403 +
  404 + do{
  405 + if (gpuMem->getMem() == nullptr){
  406 + LOG_ERROR("new GpuRgbMemory failed !!!");
  407 + break;
  408 + }
  409 +
  410 + cudaSetDevice(atoi(m_cfg.gpuid.c_str()));
  411 + cuda_common::setColorSpace( ITU_709, 0 );
  412 + cudaError_t cudaStatus = cuda_common::CUDAToBGR((CUdeviceptr)gpuFrame->data[0],(CUdeviceptr)gpuFrame->data[1], gpuFrame->linesize[0], gpuFrame->linesize[1], gpuMem->getMem(), gpuFrame->width, gpuFrame->height);
  413 + cudaDeviceSynchronize();
  414 + if (cudaStatus != cudaSuccess) {
  415 + LOG_ERROR("CUDAToBGR failed failed !!!");
  416 + break;
  417 + }
  418 +
  419 + return gpuMem;
  420 + }while(0);
  421 +
  422 + delete gpuMem;
  423 + gpuMem = nullptr;
  424 + }
  425 +
  426 + return nullptr;
  427 +}
  428 +
390 FFImgInfo* FFNvDecoder::snapshot(){ 429 FFImgInfo* FFNvDecoder::snapshot(){
391 430
392 // 锁住停止队列消耗 431 // 锁住停止队列消耗
@@ -418,7 +457,7 @@ FFImgInfo* FFNvDecoder::snapshot(){ @@ -418,7 +457,7 @@ FFImgInfo* FFNvDecoder::snapshot(){
418 457
419 if (gpuFrame != nullptr && gpuFrame->format == AV_PIX_FMT_CUDA ){ 458 if (gpuFrame != nullptr && gpuFrame->format == AV_PIX_FMT_CUDA ){
420 LOG_DEBUG("decode task: gpuid: {} width: {} height: {}", m_cfg.gpuid, gpuFrame->width, gpuFrame->height); 459 LOG_DEBUG("decode task: gpuid: {} width: {} height: {}", m_cfg.gpuid, gpuFrame->width, gpuFrame->height);
421 - GpuRgbMemory* gpuMem = new GpuRgbMemory(3, gpuFrame->width, gpuFrame->height, getName(), m_cfg.gpuid , true); 460 + GpuRgbMemory* gpuMem = new GpuRgbMemory(3, gpuFrame->width, gpuFrame->height, getName(), m_cfg.gpuid, false, true);
422 461
423 if (gpuMem->getMem() == nullptr){ 462 if (gpuMem->getMem() == nullptr){
424 LOG_ERROR("new GpuRgbMemory failed !!!"); 463 LOG_ERROR("new GpuRgbMemory failed !!!");
src/nvdecoder/FFNvDecoder.h
  View file @00b0fbd
1 #include<string> 1 #include<string>
2 #include <pthread.h> 2 #include <pthread.h>
3 3
4 -#include "../AbstractDecoder.h"  
5 -  
6 #include <mutex> 4 #include <mutex>
7 5
  6 +extern "C"
  7 +{
  8 + #include <libavcodec/avcodec.h>
  9 + #include <libavdevice/avdevice.h>
  10 + #include <libavformat/avformat.h>
  11 + #include <libavfilter/avfilter.h>
  12 + #include <libavutil/avutil.h>
  13 + #include <libavutil/pixdesc.h>
  14 + #include <libswscale/swscale.h>
  15 + #include <libavutil/imgutils.h>
  16 +}
  17 +
  18 +#include "common_header.h"
  19 +
  20 +#include "../interface/AbstractDecoder.h"
  21 +
8 using namespace std; 22 using namespace std;
9 23
10 -class FFNvDecoder : public AbstractDecoder{ 24 +class FFNvDecoder : public AbstractDecoder {
11 public: 25 public:
12 FFNvDecoder(); 26 FFNvDecoder();
13 ~FFNvDecoder(); 27 ~FFNvDecoder();
@@ -34,6 +48,17 @@ public: @@ -34,6 +48,17 @@ public:
34 48
35 FFImgInfo* snapshot(); 49 FFImgInfo* snapshot();
36 50
  51 + void setName(string nm){
  52 + m_dec_name = nm;
  53 + }
  54 +
  55 + string getName(){
  56 + return m_dec_name;
  57 + }
  58 +
  59 + void setPostDecArg(const void* postDecArg);
  60 + void setFinishedDecArg(const void* finishedDecArg);
  61 +
37 public: 62 public:
38 AVPixelFormat getHwPixFmt(); 63 AVPixelFormat getHwPixFmt();
39 64
@@ -43,7 +68,12 @@ private: @@ -43,7 +68,12 @@ private:
43 bool init(const char* uri, const char* gpuid, bool force_tcp); 68 bool init(const char* uri, const char* gpuid, bool force_tcp);
44 void decode_finished(); 69 void decode_finished();
45 70
  71 + DeviceRgbMemory* convert2bgr(AVFrame * gpuFrame);
  72 +
46 private: 73 private:
  74 + string m_dec_name;
  75 + FFDecConfig m_cfg;
  76 +
47 AVStream* stream; 77 AVStream* stream;
48 AVCodecContext *avctx; 78 AVCodecContext *avctx;
49 int stream_index; 79 int stream_index;
@@ -65,4 +95,13 @@ private: @@ -65,4 +95,13 @@ private:
65 queue<AVFrame*> mFrameQueue; 95 queue<AVFrame*> mFrameQueue;
66 mutex m_queue_mutex; 96 mutex m_queue_mutex;
67 mutex m_snapshot_mutex; 97 mutex m_snapshot_mutex;
  98 + long m_index{0};
  99 +
  100 + bool m_dec_keyframe;
  101 +
  102 + const void * m_postDecArg;
  103 + POST_DECODE_CALLBACK post_decoded_cbk; // 解码数据回调接口
  104 +
  105 + const void * m_finishedDecArg;
  106 + DECODE_FINISHED_CALLBACK decode_finished_cbk;
68 }; 107 };
69 \ No newline at end of file 108 \ No newline at end of file
src/nvdecoder/GpuRgbMemory.hpp
  View file @00b0fbd
1 #include<string> 1 #include<string>
2 2
3 -#include "../DeviceRgbMemory.hpp" 3 +#include "../interface/DeviceRgbMemory.hpp"
4 #include "cuda_kernels.h" 4 #include "cuda_kernels.h"
5 #include "define.hpp" 5 #include "define.hpp"
6 -#include "utiltools.hpp" 6 +#include "common_header.h"
7 7
8 using namespace std; 8 using namespace std;
9 9
10 class GpuRgbMemory : public DeviceRgbMemory{ 10 class GpuRgbMemory : public DeviceRgbMemory{
11 11
12 public: 12 public:
13 - GpuRgbMemory(int _channel, int _width, int _height, string _id, string _gpuid, bool _isused)  
14 - :DeviceRgbMemory(_channel, _width, _height, _id, _gpuid, _isused){ 13 + GpuRgbMemory(int _channel, int _width, int _height, string _id, string _gpuid, bool _key_frame, bool _isused)
  14 + :DeviceRgbMemory(_channel, _width, _height, _id, _gpuid, _key_frame, _isused){
15 gpuid = _gpuid; 15 gpuid = _gpuid;
16 cudaSetDevice(atoi(gpuid.c_str())); 16 cudaSetDevice(atoi(gpuid.c_str()));
17 CHECK_CUDA(cudaMalloc((void **)&pHwRgb, data_size * sizeof(unsigned char))); 17 CHECK_CUDA(cudaMalloc((void **)&pHwRgb, data_size * sizeof(unsigned char)));
src/nvdecoder/Makefile 0 → 100644
  View file @00b0fbd
  1 +# 各项目录
  2 +LIB_DIR:=$(BUILD_DIR)/$(MODULE)/lib
  3 +DEP_DIR:=$(BUILD_DIR)/$(MODULE)/.dep
  4 +OBJ_DIR:=$(BUILD_DIR)/$(MODULE)/obj
  5 +SRC_DIR:=$(TOP_DIR)/$(MODULE)
  6 +
  7 +# 源文件以及中间目标文件和依赖文件
  8 +SRCS:=$(notdir $(wildcard $(SRC_DIR)/*.cpp))
  9 +OBJS:=$(addprefix $(OBJ_DIR)/, $(patsubst %.cpp, %.o, $(SRCS)))
  10 +DEPS:=$(addprefix $(DEP_DIR)/, $(patsubst %.cpp, %.d,a $(SRCS)))
  11 +
  12 +CUDA_ROOT = /usr/local/cuda-11.1
  13 +NVCC = $(CUDA_ROOT)/bin/nvcc
  14 +
  15 +# 自动生成头文件依赖选项
  16 +DEPFLAGS=-MT $@ -MMD -MP -MF $(DEP_DIR)/$*.d
  17 +
  18 +DEFS = -DENABLE_DVPP_INTERFACE
  19 +
  20 +# 最终目标文件
  21 +TARGET:=$(LIB_DIR)/$(MODULE).a
  22 +
  23 +
  24 +PROJECT_ROOT= /mnt/data/cmhu/FFNvDecoder
  25 +
  26 +DEPEND_DIR = $(PROJECT_ROOT)/bin
  27 +THIRDPARTY_ROOT = $(PROJECT_ROOT)/3rdparty
  28 +SPDLOG_ROOT = $(THIRDPARTY_ROOT)/spdlog-1.9.2/release
  29 +JRTP_ROOT = $(THIRDPARTY_ROOT)/jrtp_export
  30 +
  31 +
  32 +INCLUDE= -I $(DEPEND_DIR)/include \
  33 + -I $(CUDA_ROOT)/include \
  34 + -I $(TOP_DIR)/common/inc \
  35 + -I $(TOP_DIR)/common/UtilNPP \
  36 + -I $(TOP_DIR)/ \
  37 + -I $(SPDLOG_ROOT)/include \
  38 + -I $(JRTP_ROOT)/jrtplib/include/jrtplib3 \
  39 + -I $(JRTP_ROOT)/jthread/include/jthread
  40 +
  41 +LIBSPATH= -L $(DEPEND_DIR)/lib -lavformat -lavcodec -lswscale -lavutil -lavfilter -lswresample -lavdevice \
  42 + -L $(CUDA_ROOT)/lib64 -lcuda -lcudart -lnvcuvid -lcurand -lcublas -lnvjpeg \
  43 + -L $(SPDLOG_ROOT) -l:libspdlog.a \
  44 + -L $(JRTP_ROOT)/jthread/lib -l:libjthread.a \
  45 + -L $(JRTP_ROOT)/jrtplib/lib -l:libjrtp.a
  46 +
  47 +
  48 +CXXFLAGS= -g -O0 -fPIC $(INCLUDE) $(DEFS) -lpthread -lrt -lz -fexceptions -std=c++11 -fvisibility=hidden -Wl,-Bsymbolic -ldl -Wwrite-strings
  49 +# CFLAGS= -g -fPIC -O0 $(INCLUDE) -pthread -lrt -lz -std=c++11 -fvisibility=hidden -Wl,-Bsymbolic -ldl
  50 + # -DUNICODE -D_UNICODE
  51 +
  52 +NFLAGS_LIB=-g -c -shared -Xcompiler -fPIC -Xcompiler -fvisibility=hidden
  53 +NFLAGS = $(NFLAGS_LIB) $(INCLUDE) -std=c++11
  54 +
  55 +# CU_SOURCES = $(wildcard ${SRC_DIR}/*.cu)
  56 +# CU_OBJS = $(patsubst %.cu, %.o, $(notdir $(CU_SOURCES)))
  57 +
  58 +CU_SOURCES:=$(notdir $(wildcard $(SRC_DIR)/*.cu))
  59 +CU_OBJS:=$(addprefix $(OBJ_DIR)/, $(patsubst %.cu, %.o, $(CU_SOURCES)))
  60 +
  61 +
  62 +# 默认最终目标
  63 +.PHONY:all
  64 +all:$(TARGET)
  65 +
  66 +# 生成最终目标
  67 +$(TARGET):$(OBJS) $(CU_OBJS) | $(LIB_DIR)
  68 + @echo -e "\e[32m""Linking static library $(TARGET)""\e[0m"
  69 +# @ar -rc $@ $^
  70 +
  71 +# 若没有lib目录则自动生成
  72 +$(LIB_DIR):
  73 + @mkdir -p $@
  74 +
  75 +# 生成中间目标文件
  76 +$(OBJ_DIR)/%.o:$(SRC_DIR)/%.cpp $(DEP_DIR)/%.d | $(OBJ_DIR) $(DEP_DIR)
  77 +# @echo -e "\e[33m""Building object $@""\e[0m"
  78 + @$(CXX) -c $(DEPFLAGS) $(CXXFLAGS) $(INCS) $(LIBSPATH) $(MACROS) -o $@ $<
  79 +
  80 +$(OBJ_DIR)%.o:$(SRC_DIR)/%.cu
  81 + @echo "#######################CU_OBJS:$@###############"
  82 + $(NVCC) $(NFLAGS) -o $@ $<
  83 +
  84 +
  85 +# 若没有obj目录则自动生成
  86 +$(OBJ_DIR):
  87 + @mkdir -p $@
  88 +
  89 +# 若没有.dep目录则自动生成
  90 +$(DEP_DIR):
  91 + @mkdir -p $@
  92 +
  93 +# 依赖文件会在生成中间文件的时候自动生成，这里只是为了防止报错
  94 +$(DEPS):
  95 +
  96 +# 引入中间目标文件头文件依赖关系
  97 +include $(wildcard $(DEPS))
  98 +
  99 +# 直接删除组件build目录
  100 +.PHONY:clean
  101 +clean:
  102 + @rm -rf $(BUILD_DIR)/$(MODULE)
src/nvdecoder/NV12ToRGB.cu
  View file @00b0fbd
@@ -2,7 +2,7 @@ @@ -2,7 +2,7 @@
2 #include "cuda_kernels.h" 2 #include "cuda_kernels.h"
3 3
4 #include <builtin_types.h> 4 #include <builtin_types.h>
5 -#include "common/inc/helper_cuda_drvapi.h" 5 +#include "helper_cuda_drvapi.h"
6 6
7 typedef unsigned char uint8; 7 typedef unsigned char uint8;
8 typedef unsigned int uint32; 8 typedef unsigned int uint32;
src/nvdecoder/NvDecoderApi.cpp 0 → 100644
  View file @00b0fbd
  1 +#include "NvDecoderApi.h"
  2 +#include "FFNvDecoder.h"
  3 +
  4 +NvDecoderApi::NvDecoderApi(){
  5 + m_pDecoder = nullptr;
  6 +}
  7 +
  8 +NvDecoderApi::~NvDecoderApi(){
  9 + if(m_pDecoder != nullptr){
  10 + delete m_pDecoder;
  11 + m_pDecoder = nullptr;
  12 + }
  13 +}
  14 +
  15 +bool NvDecoderApi::init(FFDecConfig& cfg){
  16 + m_pDecoder = new FFNvDecoder();
  17 + if(m_pDecoder != nullptr){
  18 + return m_pDecoder->init(cfg);
  19 + }
  20 + return false;
  21 +}
  22 +
  23 +void NvDecoderApi::close(){
  24 + if(m_pDecoder != nullptr){
  25 + return m_pDecoder->close();
  26 + }
  27 +}
  28 +
  29 +bool NvDecoderApi::start(){
  30 + if(m_pDecoder != nullptr){
  31 + return m_pDecoder->start();
  32 + }
  33 + return false;
  34 +}
  35 +
  36 +void NvDecoderApi::pause(){
  37 + if(m_pDecoder != nullptr){
  38 + return m_pDecoder->pause();
  39 + }
  40 +}
  41 +
  42 +void NvDecoderApi::resume(){
  43 + if(m_pDecoder != nullptr){
  44 + return m_pDecoder->resume();
  45 + }
  46 +}
  47 +
  48 +void NvDecoderApi::setDecKeyframe(bool bKeyframe){
  49 + if(m_pDecoder != nullptr){
  50 + return m_pDecoder->setDecKeyframe(bKeyframe);
  51 + }
  52 +}
  53 +
  54 +bool NvDecoderApi::isRunning(){
  55 + if(m_pDecoder != nullptr){
  56 + return m_pDecoder->isRunning();
  57 + }
  58 + return false;
  59 +}
  60 +
  61 +bool NvDecoderApi::isFinished(){
  62 + if(m_pDecoder != nullptr){
  63 + return m_pDecoder->isFinished();
  64 + }
  65 + return false;
  66 +}
  67 +
  68 +bool NvDecoderApi::isPausing(){
  69 + if(m_pDecoder != nullptr){
  70 + return m_pDecoder->isPausing();
  71 + }
  72 + return false;
  73 +}
  74 +
  75 +bool NvDecoderApi::getResolution(int &width, int &height){
  76 + if(m_pDecoder != nullptr){
  77 + return m_pDecoder->getResolution(width, height);
  78 + }
  79 + return false;
  80 +}
  81 +
  82 +bool NvDecoderApi::isSurport(FFDecConfig& cfg){
  83 + if(m_pDecoder != nullptr){
  84 + return m_pDecoder->isSurport(cfg);
  85 + }
  86 + return false;
  87 +}
  88 +
  89 +float NvDecoderApi::fps(){
  90 + if(m_pDecoder != nullptr){
  91 + return m_pDecoder->fps();
  92 + }
  93 + return 0.0;
  94 +}
  95 +
  96 +int NvDecoderApi::getCachedQueueLength(){
  97 + if(m_pDecoder != nullptr){
  98 + return m_pDecoder->getCachedQueueLength();
  99 + }
  100 + return 0;
  101 +}
  102 +
  103 +void NvDecoderApi::setName(string nm){
  104 + if(m_pDecoder != nullptr){
  105 + return m_pDecoder->setName(nm);
  106 + }
  107 +}
  108 +
  109 +string NvDecoderApi::getName(){
  110 + if(m_pDecoder != nullptr){
  111 + return m_pDecoder->getName();
  112 + }
  113 + return nullptr;
  114 +}
  115 +
  116 +FFImgInfo* NvDecoderApi::snapshot(){
  117 + if(m_pDecoder != nullptr){
  118 + return m_pDecoder->snapshot();
  119 + }
  120 + return nullptr;
  121 +}
  122 +
  123 +void NvDecoderApi::setPostDecArg(const void* postDecArg){
  124 + if(m_pDecoder != nullptr){
  125 + return m_pDecoder->setPostDecArg(postDecArg);
  126 + }
  127 +}
  128 +
  129 +void NvDecoderApi::setFinishedDecArg(const void* finishedDecArg){
  130 + if(m_pDecoder != nullptr){
  131 + return m_pDecoder->setFinishedDecArg(finishedDecArg);
  132 + }
  133 +}
0 \ No newline at end of file 134 \ No newline at end of file
src/nvdecoder/NvDecoderApi.h 0 → 100644
  View file @00b0fbd
  1 +#include<string>
  2 +#include <pthread.h>
  3 +
  4 +#include "common_header.h"
  5 +#include "../interface/AbstractDecoder.h"
  6 +
  7 +using namespace std;
  8 +
  9 +class FFNvDecoder;
  10 +
  11 +class NvDecoderApi : public AbstractDecoder{
  12 +public:
  13 + NvDecoderApi();
  14 + ~NvDecoderApi();
  15 + bool init(FFDecConfig& cfg);
  16 + void close();
  17 + bool start();
  18 + void pause();
  19 + void resume();
  20 +
  21 + void setDecKeyframe(bool bKeyframe);
  22 +
  23 + bool isRunning();
  24 + bool isFinished();
  25 + bool isPausing();
  26 + bool getResolution( int &width, int &height );
  27 +
  28 + bool isSurport(FFDecConfig& cfg);
  29 +
  30 + int getCachedQueueLength();
  31 +
  32 + float fps();
  33 +
  34 + FFImgInfo* snapshot();
  35 +
  36 + DECODER_TYPE getDecoderType(){ return DECODER_TYPE_DVPP; }
  37 + void setName(string nm);
  38 + string getName();
  39 +
  40 + void setPostDecArg(const void* postDecArg);
  41 + void setFinishedDecArg(const void* finishedDecArg);
  42 +private:
  43 + FFNvDecoder* m_pDecoder;
  44 +};
0 \ No newline at end of file 45 \ No newline at end of file
src/nvdecoder/common_header.h
  View file @00b0fbd
@@ -4,5 +4,6 @@ @@ -4,5 +4,6 @@
4 4
5 #include "../interface/logger.hpp" 5 #include "../interface/logger.hpp"
6 #include "../interface/utiltools.hpp" 6 #include "../interface/utiltools.hpp"
  7 +#include "../interface/interface_headers.h"
7 8
8 #endif 9 #endif
9 \ No newline at end of file 10 \ No newline at end of file