From 5dc1859db41790e9dcbbc87c160601e489761bcb Mon Sep 17 00:00:00 2001
From: xiongziliang <771730766@qq.com>
Date: Mon, 29 Oct 2018 13:56:10 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E6=88=90=E6=92=AD=E6=94=BE=E5=99=A8?=
 =?UTF-8?q?=E9=80=82=E9=85=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/YuvDisplayer.h  | 61 +++++++++++++++++----------------------
 tests/test_player.cpp | 67 +++++++++++++++++++++----------------------
 2 files changed, 59 insertions(+), 69 deletions(-)

diff --git a/tests/YuvDisplayer.h b/tests/YuvDisplayer.h
index f818dab3..476b57ee 100644
--- a/tests/YuvDisplayer.h
+++ b/tests/YuvDisplayer.h
@@ -44,7 +44,6 @@ using namespace mediakit;
 #define REFRESH_EVENT   (SDL_USEREVENT + 1)
 
 
-#ifndef __MACH__
 class SDLDisplayerHelper
 {
 public:
@@ -66,48 +65,44 @@ public:
         SDL_PushEvent(&event);
     }
 
+	void runLoop(){
+		bool flag = true;
+		std::function<bool ()> task;
+		SDL_Event event;
+		while(flag){
+			SDL_WaitEvent(&event);
+			if (event.type == REFRESH_EVENT)
+			{
+				{
+					lock_guard<mutex> lck(_mtxTask);
+					if(_taskList.empty()){
+						//not reachable
+						continue;
+					}
+					task = _taskList.front();
+					_taskList.pop_front();
+				}
+				flag = task();
+			}
 
+		}
+	}
+
+	void shutdown(){
+		doTask([](){return false;});
+	}
 private:
     SDLDisplayerHelper(){
-        _loopThread.reset(new std::thread(&SDLDisplayerHelper::runLoop,this));
     };
     ~SDLDisplayerHelper(){
-        doTask([](){return false;});
-        _loopThread->join();
-        _loopThread.reset();
+		shutdown();
     };
-
-    void runLoop(){
-        bool flag = true;
-        std::function<bool ()> task;
-        SDL_Event event;
-        while(flag){
-            SDL_WaitEvent(&event);
-            if (event.type == REFRESH_EVENT)
-            {
-                {
-                    lock_guard<mutex> lck(_mtxTask);
-                    if(_taskList.empty()){
-                        //not reachable
-                        continue;
-                    }
-                    task = _taskList.front();
-                    _taskList.pop_front();
-                }
-                flag = task();
-            }
-
-        }
-    }
-
-
+private:
     std::deque<std::function<bool ()> > _taskList;
-    std::shared_ptr<thread> _loopThread;
     std::mutex _mtxTask;
 
 };
 
-#endif //__MACH__
 
 class YuvDisplayer {
 public:
@@ -126,9 +121,7 @@ public:
             },nullptr);
             InfoL << "SDL_Init";
         }, []() {
-#ifndef __MACH__
             SDLDisplayerHelper::Destory();
-#endif
             SDL_Quit();
         });
 
diff --git a/tests/test_player.cpp b/tests/test_player.cpp
index 76c218d9..bb10dfe1 100644
--- a/tests/test_player.cpp
+++ b/tests/test_player.cpp
@@ -41,10 +41,9 @@ using namespace mediakit;
 
 int main(int argc, char *argv[]) {
     //设置退出信号处理函数
-    signal(SIGINT, [](int) { EventPoller::Instance().shutdown(); });
+    signal(SIGINT, [](int) { SDLDisplayerHelper::Instance().shutdown(); });
     //设置日志
     Logger::Instance().add(std::make_shared<ConsoleChannel>("stdout", LTrace));
-    Logger::Instance().setWriter(std::make_shared<AsyncLogWriter>());
 
     if (argc != 3) {
         ErrorL << "\r\n测试方法：./test_player rtxp_url rtp_type\r\n"
@@ -57,46 +56,44 @@ int main(int argc, char *argv[]) {
 
     {
         MediaPlayer::Ptr player(new MediaPlayer());
-        player->setOnPlayResult([](const SockException &ex) {
+        weak_ptr<MediaPlayer> weakPlayer = player;
+        player->setOnPlayResult([weakPlayer](const SockException &ex) {
             InfoL << "OnPlayResult:" << ex.what();
+            auto strongPlayer = weakPlayer.lock();
+            if (ex || !strongPlayer) {
+                return;
+            }
+
+            auto viedoTrack = strongPlayer->getTrack(TrackVideo);
+            if (!viedoTrack || viedoTrack->getCodecId() != CodecH264) {
+                WarnL << "没有视频或者视频不是264编码!";
+                return;
+            }
+            SDLDisplayerHelper::Instance().doTask([viedoTrack]() {
+                std::shared_ptr<H264Decoder> decoder(new H264Decoder);
+                std::shared_ptr<YuvDisplayer> displayer(new YuvDisplayer);
+                viedoTrack->addDelegate(std::make_shared<FrameWriterInterfaceHelper>([decoder, displayer](const Frame::Ptr &frame) {
+                    SDLDisplayerHelper::Instance().doTask([decoder, displayer, frame]() {
+                        AVFrame *pFrame = nullptr;
+                        bool flag = decoder->inputVideo((unsigned char *) frame->data(), frame->size(),
+                                                        frame->stamp(), &pFrame);
+                        if (flag) {
+                            displayer->displayYUV(pFrame);
+                        }
+                        return true;
+                    });
+                }));
+                return true;
+            });
         });
+
+
         player->setOnShutdown([](const SockException &ex) {
             ErrorL << "OnShutdown:" << ex.what();
         });
         (*player)[RtspPlayer::kRtpType] = atoi(argv[2]);
         player->play(argv[1]);
-
-        H264Decoder decoder;
-        YuvDisplayer displayer;
-        //todo(xzl) 修复此处
-#if 0
-        player->setOnVideoCB([&](const H264Frame &frame) {
-#ifndef __MACH__
-            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;
-                    EventPoller::Instance().sync([&](){
-                        displayer.displayYUV(pFrame);
-                    });
-                }
-                return true;
-            });
-#else
-            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);
-            }
-#endif
-
-        });
-#endif
-        EventPoller::Instance().runLoop();
+        SDLDisplayerHelper::Instance().runLoop();
     }
     UDPServer::Destory();
     EventPoller::Destory();