From bd04b2e2abff19b8f2b1fc1546a29c4967311678 Mon Sep 17 00:00:00 2001 From: xiongziliang <771730766@qq.com> Date: Tue, 17 Apr 2018 21:38:13 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=BF=E6=8D=A2=E6=88=90sdl2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cmake/FindSDL2.cmake | 15 ++++ tests/CMakeLists.txt | 19 ++-- tests/YuvDisplayer.h | 199 ++++++++++++++++++++++++++++++++---------- tests/test_player.cpp | 87 +++++++++--------- 4 files changed, 219 insertions(+), 101 deletions(-) create mode 100644 cmake/FindSDL2.cmake diff --git a/cmake/FindSDL2.cmake b/cmake/FindSDL2.cmake new file mode 100644 index 00000000..cc608fbe --- /dev/null +++ b/cmake/FindSDL2.cmake @@ -0,0 +1,15 @@ +find_path(SDL2_INCLUDE_DIR + NAMES SDL2/SDL.h + HINTS SDL2 + PATHS $ENV{HOME}/sdl2/include) + +find_library(SDL2_LIBRARY + NAMES SDL2 + PATHS $ENV{HOME}/sdl2/lib/x86) + +set(SDL2_LIBRARIES ${SDL2_LIBRARY}) +set(SDL2_INCLUDE_DIRS ${SDL2_INCLUDE_DIR}) + +include(FindPackageHandleStandardArgs) + +find_package_handle_standard_args(SDL2 DEFAULT_MSG SDL2_LIBRARY SDL2_INCLUDE_DIR) \ No newline at end of file diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index f4fe9618..d90f431c 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -1,10 +1,10 @@ -#查找SDL是否安装 -find_package(SDL QUIET) -if(SDL_FOUND) - include_directories(${SDL_INCLUDE_DIR}) - list(APPEND LINK_LIB_LIST ${SDL_LIBRARY}) - message(STATUS " found SDL") -endif() +#查找SDL2是否安装 +find_package(SDL2 QUIET) +if (SDL2_FOUND) + include_directories(${SDL2_INCLUDE_DIR}) + list(APPEND LINK_LIB_LIST ${SDL2_LIBRARY}) + message(STATUS "found SDL2:${SDL2_INCLUDE_DIR},${SDL2_LIBRARY}") +endif (SDL2_FOUND) #查找ffmpeg/libutil是否安装 find_package(AVUTIL QUIET) @@ -24,10 +24,10 @@ endif() aux_source_directory(. TEST_SRC_LIST) #如果ffmpeg/libavcodec ffmpeg/libavcodec SDL 都安装了则编译 test_player -if(SDL_FOUND AND AVCODEC_FOUND AND AVUTIL_FOUND) +if(SDL2_FOUND AND AVCODEC_FOUND AND AVUTIL_FOUND) message(STATUS "test_player will be compiled") else() - message(STATUS "test_player ingored, please install sdl ffmpeg/libavcodec ffmpeg/libavcodec") + message(STATUS "test_player ingored, please install sdl2 ffmpeg/libavcodec ffmpeg/libavutil") list(REMOVE_ITEM TEST_SRC_LIST ./test_player.cpp) endif() @@ -50,4 +50,3 @@ endforeach() - diff --git a/tests/YuvDisplayer.h b/tests/YuvDisplayer.h index b2da7374..22f41898 100644 --- a/tests/YuvDisplayer.h +++ b/tests/YuvDisplayer.h @@ -1,9 +1,7 @@ /* * MIT License * - * Copyright (c) 2016 xiongziliang <771730766@qq.com> - * - * This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit). + * Copyright (c) 2017 xiongziliang <771730766@qq.com> * * Permission is hereby granted, free of charge, to any person obtaining a copy * of this software and associated documentation files (the "Software"), to deal @@ -23,7 +21,6 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ - #ifndef YUVDISPLAYER_H_ #define YUVDISPLAYER_H_ #include @@ -31,68 +28,176 @@ #ifdef __cplusplus extern "C" { #endif -#include -#include +#include "SDL2/SDL.h" +#include "libavcodec/avcodec.h" #ifdef __cplusplus } #endif -namespace ZL { -namespace Screen { +#if defined(_WIN32) +#pragma comment(lib,"SDL2.lib") +#endif //defined(_WIN32) + +using namespace ZL::Util; + +#define REFRESH_EVENT (SDL_USEREVENT + 1) + +class SDLDisplayerHelper +{ +public: + static SDLDisplayerHelper &Instance(){ + static SDLDisplayerHelper *instance(new SDLDisplayerHelper); + return *instance; + } + static void Destory(){ + delete &Instance(); + } + template + void doTask(FUN &&f){ + { + lock_guard lck(_mtxTask); + _taskList.emplace_back(f); + } + SDL_Event event; + event.type = REFRESH_EVENT; + SDL_PushEvent(&event); + } + + +private: + SDLDisplayerHelper(){ + _loopThread.reset(new std::thread(&SDLDisplayerHelper::runLoop,this)); + }; + ~SDLDisplayerHelper(){ + doTask([](){return false;}); + _loopThread->join(); + _loopThread.reset(); + }; + + void runLoop(){ + bool flag = true; + std::function task; + SDL_Event event; + while(flag){ + SDL_WaitEvent(&event); + if (event.type == REFRESH_EVENT) + { + { + lock_guard lck(_mtxTask); + if(_taskList.empty()){ + //not reachable + continue; + } + task = _taskList.front(); + _taskList.pop_front(); + } + flag = task(); + } + + } + } + + + std::deque > _taskList; + std::shared_ptr _loopThread; + std::mutex _mtxTask; + +}; class YuvDisplayer { public: - YuvDisplayer(){ - static onceToken token([]() { - if(SDL_Init(SDL_INIT_EVERYTHING) == -1) { - throw std::runtime_error("初始化SDL失败"); - } - }, []() { - SDL_Quit(); - }); + YuvDisplayer(void *hwnd = nullptr,const char *title = "untitled"){ + + static onceToken token([]() { + if(SDL_Init(SDL_INIT_VIDEO|SDL_INIT_AUDIO) == -1) { + string err = "初始化SDL失败:"; + err+= SDL_GetError(); + ErrorL << err; + throw std::runtime_error(err); + } + SDL_LogSetAllPriority(SDL_LOG_PRIORITY_CRITICAL); + SDL_LogSetOutputFunction([](void *userdata, int category, SDL_LogPriority priority, const char *message){ + DebugL << category << " " << priority << message; + },nullptr); + InfoL << "SDL_Init"; + }, []() { + SDLDisplayerHelper::Destory(); + SDL_Quit(); + InfoL << "SDL_Quit"; + }); + + _title = title; + _hwnd = hwnd; } virtual ~YuvDisplayer(){ - if(m_pOverlay){ - SDL_FreeYUVOverlay(m_pOverlay); + if (_texture) { + SDL_DestroyTexture(_texture); + _texture = nullptr; + } + if (_render) { + SDL_DestroyRenderer(_render); + _render = nullptr; + } + if (_win) { + SDL_DestroyWindow(_win); + _win = nullptr; } } bool displayYUV(AVFrame *pFrame){ - if (!m_pScreen) { - /* Set up the screen */ - m_pScreen = SDL_SetVideoMode(1366, 768, 16, SDL_SWSURFACE); + if (!_win) { + if (_hwnd) { + _win = SDL_CreateWindowFrom(_hwnd); + }else { + _win = SDL_CreateWindow(_title.data(), + SDL_WINDOWPOS_UNDEFINED, + SDL_WINDOWPOS_UNDEFINED, + pFrame->width, + pFrame->height, + SDL_WINDOW_OPENGL); + } } - - if (!m_pOverlay && m_pScreen) { - /* Create a YUV overlay */ - m_pOverlay = SDL_CreateYUVOverlay(pFrame->width, pFrame->height, SDL_YV12_OVERLAY, m_pScreen); - /* Set the window caption */ - SDL_WM_SetCaption("YUV Window", NULL); + if (_win && ! _render){ +#if 0 + SDL_RENDERER_SOFTWARE = 0x00000001, /**< The renderer is a software fallback */ + SDL_RENDERER_ACCELERATED = 0x00000002, /**< The renderer uses hardware + acceleration */ + SDL_RENDERER_PRESENTVSYNC = 0x00000004, /**< Present is synchronized + with the refresh rate */ + SDL_RENDERER_TARGETTEXTURE = 0x00000008 /**< The renderer supports + rendering to texture */ +#endif +#if defined(__linux__) + _render = SDL_CreateRenderer(_win, -1, SDL_RENDERER_SOFTWARE); +#else + _render = SDL_CreateRenderer(_win, -1, SDL_RENDERER_ACCELERATED); +#endif } - if (m_pOverlay) { - /* Apply the image to the screen */ - m_pOverlay->pixels[0] = pFrame->data[0]; - m_pOverlay->pixels[2] = pFrame->data[1]; - m_pOverlay->pixels[1] = pFrame->data[2]; + if (_render && !_texture) { + _texture = SDL_CreateTexture(_render, SDL_PIXELFORMAT_IYUV, + SDL_TEXTUREACCESS_STREAMING, + pFrame->width, + pFrame->height); + } + if (_texture) { + SDL_UpdateYUVTexture(_texture, nullptr, + pFrame->data[0], pFrame->linesize[0], + pFrame->data[1], pFrame->linesize[1], + pFrame->data[2], pFrame->linesize[2]); - m_pOverlay->pitches[0] = pFrame->linesize[0]; - m_pOverlay->pitches[2] = pFrame->linesize[1]; - m_pOverlay->pitches[1] = pFrame->linesize[2]; - - /* Update the screen */ - SDL_Rect rect = { 0 ,0 ,1366,768}; - SDL_LockYUVOverlay(m_pOverlay); - SDL_DisplayYUVOverlay(m_pOverlay, &rect); - SDL_UnlockYUVOverlay(m_pOverlay); + //SDL_UpdateTexture(_texture, nullptr, pFrame->data[0], pFrame->linesize[0]); + SDL_RenderClear(_render); + SDL_RenderCopy(_render, _texture, nullptr, nullptr); + SDL_RenderPresent(_render); return true; } return false; } private: - SDL_Surface* m_pScreen = nullptr; - SDL_Overlay* m_pOverlay = nullptr; + string _title; + SDL_Window *_win = nullptr; + SDL_Renderer *_render = nullptr; + SDL_Texture *_texture = nullptr; + void *_hwnd = nullptr; }; -} /* namespace Screen */ -} /* namespace ZL */ - -#endif /* YUVDISPLAYER_H_ */ +#endif /* YUVDISPLAYER_H_ */ \ No newline at end of file diff --git a/tests/test_player.cpp b/tests/test_player.cpp index 4bf2f085..9ce06563 100644 --- a/tests/test_player.cpp +++ b/tests/test_player.cpp @@ -36,7 +36,6 @@ #include "Network/sockutil.h" using namespace std; -using namespace ZL::Screen; using namespace ZL::Codec; using namespace ZL::Util; using namespace ZL::Thread; @@ -44,55 +43,55 @@ using namespace ZL::Network; using namespace ZL::Rtsp; using namespace ZL::Player; -int main(int argc, char *argv[]){ - //设置退出信号处理函数 - signal(SIGINT, [](int){EventPoller::Instance().shutdown();}); - //设置日志 - Logger::Instance().add(std::make_shared("stdout", LTrace)); - Logger::Instance().setWriter(std::make_shared()); +int main(int argc, char *argv[]) { + //设置退出信号处理函数 + signal(SIGINT, [](int) { EventPoller::Instance().shutdown(); }); + //设置日志 + Logger::Instance().add(std::make_shared("stdout", LTrace)); + Logger::Instance().setWriter(std::make_shared()); - if(argc != 3){ - ErrorL << "\r\n测试方法：./test_player rtxp_url rtp_type\r\n" - << "例如：./test_player rtsp://admin:123456@127.0.0.1/live/0 0\r\n" - <setOnPlayResult([](const SockException &ex) { - InfoL << "OnPlayResult:" << ex.what(); - }); - player->setOnShutdown([](const SockException &ex) { - ErrorL << "OnShutdown:" << ex.what(); - }); - (*player)[RtspPlayer::kRtpType] = atoi(argv[2]); - player->play(argv[1]); + MediaPlayer::Ptr player(new MediaPlayer()); + player->setOnPlayResult([](const SockException &ex) { + InfoL << "OnPlayResult:" << ex.what(); + }); + player->setOnShutdown([](const SockException &ex) { + ErrorL << "OnShutdown:" << ex.what(); + }); + (*player)[RtspPlayer::kRtpType] = atoi(argv[2]); + player->play(argv[1]); - H264Decoder decoder; - YuvDisplayer displayer; - ThreadPool pool(1); - player->setOnVideoCB([&](const H264Frame &frame){ - pool.async([&,frame]() { - AVFrame *pFrame = nullptr; - bool flag = decoder.inputVideo((unsigned char *)frame.data.data(), frame.data.size() ,frame.timeStamp, &pFrame); - if(flag) { - //DebugL << pFrame->pkt_pts; - displayer.displayYUV(pFrame); - } - }); - }); + H264Decoder decoder; + YuvDisplayer displayer; + player->setOnVideoCB([&](const H264Frame &frame) { + SDLDisplayerHelper::Instance().doTask([&, frame]() { + AVFrame *pFrame = nullptr; + bool flag = decoder.inputVideo((unsigned char *) frame.data.data(), frame.data.size(), frame.timeStamp, &pFrame); + if (flag) { + //DebugL << pFrame->pkt_pts; + displayer.displayYUV(pFrame); + } + return true; + }); + }); - EventPoller::Instance().runLoop(); + EventPoller::Instance().runLoop(); - static onceToken token(nullptr, []() { - UDPServer::Destory(); - EventPoller::Destory(); - AsyncTaskThread::Destory(); - Logger::Destory(); - }); - return 0; + static onceToken token(nullptr, []() { + UDPServer::Destory(); + EventPoller::Destory(); + AsyncTaskThread::Destory(); + Logger::Destory(); + }); + return 0; }