From a88698ced8bcd58f0f1918b10380bc66b0bfcbbc Mon Sep 17 00:00:00 2001
From: houxiao <houxiao@454eff88-639b-444f-9e54-f578c98de674>
Date: 星期五, 30 十二月 2016 18:32:15 +0800
Subject: [PATCH] add pl_scale pl_fork

---
 RtspFace/main.cpp |   48 +++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 39 insertions(+), 9 deletions(-)

diff --git a/RtspFace/main.cpp b/RtspFace/main.cpp
index 8b87ae0..412c1ac 100644
--- a/RtspFace/main.cpp
+++ b/RtspFace/main.cpp
@@ -6,14 +6,15 @@
 #include "PL_AVFrameYUV420.h"
 #include "PL_AVFrameBGRA.h"
 #include "PL_Queue.h"
+#include "PL_Scale.h"
 
-#include "PL_SensetimeFaceDetect.h"
+#include "PL_SensetimeFaceTrack.h"
 
 #include "logger.h"
 
 int main(int argc, char** argv)
 {
-	initLogger(2);
+	initLogger(LV_DEBUG);
 
 	PipeLine pipeLine;
 	
@@ -23,8 +24,9 @@
 	pipeLine.register_elem_creator("PL_AVFrameYUV420", create_PL_AVFrameYUV420);
 	pipeLine.register_elem_creator("PL_H264Encoder", create_PL_H264Encoder);
 	pipeLine.register_elem_creator("PL_Queue", create_PL_Queue);
+	pipeLine.register_elem_creator("PL_Scale", create_PL_Scale);
 	
-	pipeLine.register_elem_creator("PL_SensetimeFaceDetect", create_PL_SensetimeFaceDetect);
+	pipeLine.register_elem_creator("PL_SensetimeFaceTrack", create_PL_SensetimeFaceTrack);
 	
 	{
 		PL_RTSPClient* rtspClient = (PL_RTSPClient*)pipeLine.push_elem("PL_RTSPClient");
@@ -45,18 +47,41 @@
 
 	{
 		PL_H264Decoder* h264Decoder = (PL_H264Decoder*)pipeLine.push_elem("PL_H264Decoder");
-		h264Decoder->init(nullptr);
+		bool ret = h264Decoder->init(nullptr);
+		if (!ret)
+		{
+			LOG_ERROR << "PL_H264Decoder.init error";
+			exit(EXIT_FAILURE);
+		}
 	}
 
 	{
 		PL_AVFrameYUV420* avFrameYUV420 = (PL_AVFrameYUV420*)pipeLine.push_elem("PL_AVFrameYUV420");
-		avFrameYUV420->init(nullptr);
+		bool ret = avFrameYUV420->init(nullptr);
+		if (!ret)
+		{
+			LOG_ERROR << "PL_AVFrameYUV420.init error";
+			exit(EXIT_FAILURE);
+		}
+	}
+	
+	{
+		PL_Scale_Config config;
+		config.toWidth = 800;
+		config.toHeight = 600;
+		PL_Scale* pl = (PL_Scale*)pipeLine.push_elem("PL_Scale");
+		bool ret = pl->init(&config);
+		if (!ret)
+		{
+			LOG_ERROR << "PL_Scale.init error";
+			exit(EXIT_FAILURE);
+		}
 	}
 
 	{
-		SensetimeFaceDetectConfig config;
-		PL_SensetimeFaceDetect* stFaceDetect = (PL_SensetimeFaceDetect*)pipeLine.push_elem("PL_SensetimeFaceDetect");
-		stFaceDetect->init(&config);
+		SensetimeFaceTrackConfig config;
+		PL_SensetimeFaceTrack* pl = (PL_SensetimeFaceTrack*)pipeLine.push_elem("PL_SensetimeFaceTrack");
+		pl->init(&config);
 	}
 
 	//{//#todo queue should support deep copy
@@ -72,7 +97,12 @@
 
 	{
 		PL_H264Encoder* h264Encoder = (PL_H264Encoder*)pipeLine.push_elem("PL_H264Encoder");
-		h264Encoder->init(nullptr);
+		bool ret = h264Encoder->init(nullptr);
+		if (!ret)
+		{
+			LOG_ERROR << "PL_H264Encoder.init error";
+			exit(EXIT_FAILURE);
+		}
 	}
 	
 	{

--
Gitblit v1.8.0